97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
logger "git.towk2.me/towk/makeshift/pkg/log"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
// host string
|
|
// path string
|
|
// outputPath string
|
|
// rootPath string
|
|
// profile string
|
|
// plugins []string
|
|
// timeout int
|
|
// logFile string
|
|
loglevel logger.LogLevel = logger.INFO
|
|
)
|
|
|
|
var rootCmd = cobra.Command{
|
|
Use: "makeshift",
|
|
Short: "Extensible file cobbler",
|
|
PersistentPreRun: func(cmd *cobra.Command, args []string) {
|
|
var (
|
|
logFile string
|
|
err error
|
|
)
|
|
|
|
// initialize the logger
|
|
logFile, _ = cmd.Flags().GetString("log-file")
|
|
err = logger.InitWithLogLevel(loglevel, logFile)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to initialize logger")
|
|
os.Exit(1)
|
|
}
|
|
},
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
// try and set flags using env vars
|
|
setenv(cmd, "log-file", "MAKESHIFT_LOG_FILE")
|
|
if len(args) == 0 {
|
|
err := cmd.Help()
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to print help")
|
|
}
|
|
os.Exit(0)
|
|
}
|
|
},
|
|
PostRun: func(cmd *cobra.Command, args []string) {
|
|
log.Debug().Msg("closing log file")
|
|
err := logger.LogFile.Close()
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to close log file")
|
|
}
|
|
},
|
|
}
|
|
|
|
func Execute() {
|
|
// run the main program
|
|
if err := rootCmd.Execute(); err != nil {
|
|
fmt.Fprintln(os.Stderr, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
cobra.OnInitialize(
|
|
initLogger,
|
|
// initializeConfig,
|
|
)
|
|
// initialize the config a single time
|
|
rootCmd.PersistentFlags().VarP(&loglevel, "log-level", "l", "Set the log level output")
|
|
rootCmd.PersistentFlags().String("log-file", "", "Set the log file path (can be set with MAKESHIFT_LOG_FILE)")
|
|
}
|
|
|
|
func setenv(cmd *cobra.Command, varname string, envvar string) {
|
|
if cmd.Flags().Changed(varname) {
|
|
return
|
|
}
|
|
val := os.Getenv(envvar)
|
|
if val != "" {
|
|
cmd.Flags().Set(varname, val)
|
|
}
|
|
}
|
|
|
|
func initLogger() {
|
|
// initialize the logger
|
|
logfile, _ := rootCmd.PersistentFlags().GetString("log-file")
|
|
err := logger.InitWithLogLevel(loglevel, logfile)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to initialize logger")
|
|
os.Exit(1)
|
|
}
|
|
}
|