129 lines
2.6 KiB
Go
129 lines
2.6 KiB
Go
package cli
|
|
|
|
import (
|
|
"github.com/cortezaproject/corteza-server/pkg/logger"
|
|
"github.com/cortezaproject/corteza-server/pkg/version"
|
|
"github.com/spf13/cobra"
|
|
"go.uber.org/zap"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
rootCommand = cobra.Command{
|
|
Use: "corteza-server",
|
|
Aliases: []string{"corteza", "server"},
|
|
TraverseChildren: true,
|
|
}
|
|
|
|
serveApiCommand = cobra.Command{
|
|
Use: "serve-api",
|
|
Aliases: []string{"serve"},
|
|
|
|
Short: "Start HTTP server with REST API",
|
|
}
|
|
|
|
upgradeCommand = cobra.Command{
|
|
Use: "upgrade",
|
|
Short: "Upgrade tasks",
|
|
}
|
|
|
|
provisionCommand = cobra.Command{
|
|
Use: "provision",
|
|
Short: "Provision tasks",
|
|
}
|
|
|
|
versionCommand = cobra.Command{
|
|
Use: "version",
|
|
Short: "Version",
|
|
}
|
|
|
|
// List of commands that do not
|
|
// run initialization (db connection, etc...)
|
|
light = map[string]bool{
|
|
"help": true,
|
|
versionCommand.Use: true,
|
|
}
|
|
)
|
|
|
|
// RootCommand creates root command with a simple persistent-pre-run callback
|
|
//
|
|
// Callback is called when not executed with help subcommand
|
|
func RootCommand(ppRunEfn func() error) *cobra.Command {
|
|
// Make a copy
|
|
var (
|
|
cmd = rootCommand
|
|
|
|
silent, debug bool
|
|
)
|
|
|
|
cmd.PersistentPreRunE = func(cmd *cobra.Command, args []string) (err error) {
|
|
if light[cmd.Name()] {
|
|
// Do not run hooks on parts on help command
|
|
return nil
|
|
}
|
|
|
|
if debug {
|
|
logger.DefaultLevel.SetLevel(zap.DebugLevel)
|
|
} else if silent {
|
|
logger.DefaultLevel.SetLevel(zap.FatalLevel)
|
|
}
|
|
|
|
return ppRunEfn()
|
|
}
|
|
|
|
cmd.Flags().BoolVarP(&silent, "silent", "s", false, "No output")
|
|
cmd.Flags().BoolVarP(&debug, "debug", "d", false, "Debug")
|
|
|
|
return &cmd
|
|
}
|
|
|
|
func ServeCommand(runEfn func() error) *cobra.Command {
|
|
// Make a copy
|
|
var cmd = serveApiCommand
|
|
|
|
cmd.RunE = func(cmd *cobra.Command, args []string) (err error) {
|
|
if _, set := os.LookupEnv("LOG_LEVEL"); !set {
|
|
// If LOG_LEVEL is not explicitly set, let's
|
|
// set it to INFO so that it
|
|
logger.DefaultLevel.SetLevel(zap.InfoLevel)
|
|
}
|
|
|
|
return runEfn()
|
|
}
|
|
|
|
return &cmd
|
|
}
|
|
|
|
func UpgradeCommand(dbfn func() error) *cobra.Command {
|
|
// Make a copies
|
|
var dbCmd = upgradeCommand
|
|
|
|
dbCmd.RunE = func(cmd *cobra.Command, args []string) (err error) {
|
|
return dbfn()
|
|
}
|
|
|
|
return &dbCmd
|
|
}
|
|
|
|
func ProvisionCommand(cffn func() error) *cobra.Command {
|
|
// Make a copies
|
|
var cfCmd = provisionCommand
|
|
|
|
cfCmd.RunE = func(cmd *cobra.Command, args []string) (err error) {
|
|
return cffn()
|
|
}
|
|
|
|
return &cfCmd
|
|
}
|
|
|
|
func VersionCommand() *cobra.Command {
|
|
// Make a copies
|
|
var cfCmd = versionCommand
|
|
|
|
cfCmd.Run = func(cmd *cobra.Command, args []string) {
|
|
cmd.Println(version.Version)
|
|
}
|
|
|
|
return &cfCmd
|
|
}
|