forked from LaconicNetwork/kompose
951b46e3c2
* cleanup debug logs * avoid duplicate error messages and this makes main.go use os and fmt package instead of log to avoid touching global state
105 lines
3.6 KiB
Go
105 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.`,
|
|
SilenceErrors: true,
|
|
// 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")
|
|
}
|