kompose/cmd/root.go
namusyaka 45864ed624
make kompose exit with 1 if error (#1343)
Old code passes -1 to os.Exit and the value is directly passed to syscall.
The behavior with -1 depends on environments (most cases should convert
into 255 though)

This commit is to remove unnecessary the ambiguity and change it to use 1 instead of 255.
2020-11-04 08:24:52 +08:00

106 lines
3.6 KiB
Go

/*
Copyright 2017 The Kubernetes Authors All rights reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package cmd
import (
"strings"
log "github.com/sirupsen/logrus"
"github.com/spf13/cobra"
)
// Logrus hooks
// Hook for error and exit out on warning
type errorOnWarningHook struct{}
func (errorOnWarningHook) Levels() []log.Level {
return []log.Level{log.WarnLevel}
}
func (errorOnWarningHook) Fire(entry *log.Entry) error {
log.Fatalln(entry.Message)
return nil
}
// TODO: comment
var (
GlobalBundle string
GlobalProvider string
GlobalVerbose bool
GlobalSuppressWarnings bool
GlobalErrorOnWarning bool
GlobalFiles []string
)
// RootCmd root level flags and commands
var RootCmd = &cobra.Command{
Use: "kompose",
Short: "A tool helping Docker Compose users move to Kubernetes",
Long: `Kompose is a tool to help users who are familiar with docker-compose move to Kubernetes.`,
// PersistentPreRun will be "inherited" by all children and ran before *every* command unless
// the child has overridden the functionality. This functionality was implemented to check / modify
// all global flag calls regardless of app call.
PersistentPreRun: func(cmd *cobra.Command, args []string) {
// Add extra logging when verbosity is passed
if GlobalVerbose {
log.SetLevel(log.DebugLevel)
}
// Disable the timestamp (Kompose is too fast!)
formatter := new(log.TextFormatter)
formatter.DisableTimestamp = true
formatter.ForceColors = true
log.SetFormatter(formatter)
// Set the appropriate suppress warnings and error on warning flags
if GlobalSuppressWarnings {
log.SetLevel(log.ErrorLevel)
} else if GlobalErrorOnWarning {
hook := errorOnWarningHook{}
log.AddHook(hook)
}
// Error out of the user has not chosen Kubernetes or OpenShift
provider := strings.ToLower(GlobalProvider)
if provider != "kubernetes" && provider != "openshift" {
log.Fatalf("%s is an unsupported provider. Supported providers are: 'kubernetes', 'openshift'.", GlobalProvider)
}
},
}
// Execute executes the root level command.
// It returns an erorr if any.
func Execute() error {
return RootCmd.Execute()
}
func init() {
RootCmd.PersistentFlags().BoolVarP(&GlobalVerbose, "verbose", "v", false, "verbose output")
RootCmd.PersistentFlags().BoolVar(&GlobalSuppressWarnings, "suppress-warnings", false, "Suppress all warnings")
RootCmd.PersistentFlags().BoolVar(&GlobalErrorOnWarning, "error-on-warning", false, "Treat any warning as an error")
RootCmd.PersistentFlags().StringArrayVarP(&GlobalFiles, "file", "f", []string{}, "Specify an alternative compose file")
RootCmd.PersistentFlags().StringVarP(&GlobalBundle, "bundle", "b", "", "Specify a Distributed Application Bundle (DAB) file")
RootCmd.PersistentFlags().StringVar(&GlobalProvider, "provider", "kubernetes", "Specify a provider. Kubernetes or OpenShift.")
// Mark DAB / bundle as deprecated, see issue: https://github.com/kubernetes/kompose/issues/390
// As DAB is still EXPERIMENTAL
RootCmd.PersistentFlags().MarkDeprecated("bundle", "DAB / Bundle is deprecated, see: https://github.com/kubernetes/kompose/issues/390")
}