1ef456579a
All checks were successful
Gitea Actions Demo / Explore-Gitea-Actions (push) Successful in 8s
Signed-off-by: Jan Tytgat <jan.tytgat@corelayer.eu>
108 lines
3.8 KiB
Go
108 lines
3.8 KiB
Go
package application
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log/slog"
|
|
"os"
|
|
"reflect"
|
|
"runtime"
|
|
"syscall"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/pflag"
|
|
|
|
"git.flexabyte.io/flexabyte/go-kit/slogd"
|
|
)
|
|
|
|
var DefaultShutdownSignals = []os.Signal{syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGQUIT}
|
|
|
|
var (
|
|
appName string
|
|
appCmd *cobra.Command
|
|
persistentPreRunE []func(cmd *cobra.Command, args []string) error // collection of PreRunE functions
|
|
persistentPostRunE []func(cmd *cobra.Command, args []string) error // collection of PostRunE functions
|
|
outWriter io.Writer = os.Stdout
|
|
)
|
|
|
|
func SubCommandInitializePrintNameFunc(cmd *cobra.Command) {
|
|
fmt.Println("Initializing:", cmd.Name())
|
|
}
|
|
|
|
func HelpFuncE(cmd *cobra.Command, args []string) error {
|
|
return cmd.Help()
|
|
}
|
|
|
|
func normalizeFunc(f *pflag.FlagSet, name string) pflag.NormalizedName {
|
|
return pflag.NormalizedName(name)
|
|
}
|
|
|
|
func persistentPreRunFuncE(cmd *cobra.Command, args []string) error {
|
|
slogd.SetLevel(slogd.Level(logLevelFlag))
|
|
|
|
if slogd.ActiveHandler() != slogd.HandlerJSON && noColorFlag {
|
|
slogd.UseHandler(slogd.HandlerText)
|
|
cmd.SetContext(slogd.WithContext(cmd.Context()))
|
|
}
|
|
|
|
slogd.FromContext(cmd.Context()).Log(cmd.Context(), slogd.LevelTrace, "starting application", slog.String("command", cmd.CommandPath()))
|
|
slogd.FromContext(cmd.Context()).Log(cmd.Context(), slogd.LevelTrace, "executing PersistentPreRun")
|
|
|
|
// Make sure we can always get the version
|
|
if versionFlag || cmd.Use == versionName {
|
|
slogd.FromContext(cmd.Context()).LogAttrs(cmd.Context(), slogd.LevelTrace, "overriding command", slog.String("old_function", runtime.FuncForPC(reflect.ValueOf(cmd.RunE).Pointer()).Name()), slog.String("new_function", runtime.FuncForPC(reflect.ValueOf(versionRunFuncE).Pointer()).Name()))
|
|
cmd.RunE = versionRunFuncE
|
|
return nil
|
|
}
|
|
|
|
// Make sure that we show the app help if no commands or flags are passed
|
|
if cmd.CalledAs() == appName && runtime.FuncForPC(reflect.ValueOf(cmd.RunE).Pointer()).Name() == runtime.FuncForPC(reflect.ValueOf(RunCatchFuncE).Pointer()).Name() {
|
|
slogd.FromContext(cmd.Context()).LogAttrs(cmd.Context(), slogd.LevelTrace, "overriding command", slog.String("old_function", runtime.FuncForPC(reflect.ValueOf(cmd.RunE).Pointer()).Name()), slog.String("new_function", runtime.FuncForPC(reflect.ValueOf(HelpFuncE).Pointer()).Name()))
|
|
|
|
cmd.RunE = HelpFuncE
|
|
return nil
|
|
}
|
|
|
|
// TODO move to front??
|
|
if quietFlag {
|
|
slogd.FromContext(cmd.Context()).LogAttrs(cmd.Context(), slogd.LevelTrace, "activating quiet mode")
|
|
outWriter = io.Discard
|
|
}
|
|
|
|
if persistentPreRunE == nil {
|
|
return nil
|
|
}
|
|
|
|
var err error
|
|
for _, preRun := range persistentPreRunE {
|
|
slogd.FromContext(cmd.Context()).Log(cmd.Context(), slogd.LevelTrace, "executing PersistentPreRun function", slog.String("function", runtime.FuncForPC(reflect.ValueOf(preRun).Pointer()).Name()))
|
|
if err = preRun(cmd, args); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func persistentPostRunFuncE(cmd *cobra.Command, args []string) error {
|
|
defer slogd.FromContext(cmd.Context()).Log(cmd.Context(), slogd.LevelTrace, "stopping application", slog.String("command", cmd.CommandPath()))
|
|
slogd.FromContext(cmd.Context()).Log(cmd.Context(), slogd.LevelTrace, "executing PersistentPostRunE")
|
|
|
|
if persistentPostRunE == nil {
|
|
return nil
|
|
}
|
|
|
|
var err error
|
|
for _, postRun := range persistentPostRunE {
|
|
slogd.FromContext(cmd.Context()).Log(cmd.Context(), slogd.LevelTrace, "executing PersistentPostRun function", slog.String("function", runtime.FuncForPC(reflect.ValueOf(postRun).Pointer()).Name()))
|
|
if err = postRun(cmd, args); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// RunCatchFuncE is an empty catch function to allow overrides through persistentPreRunE
|
|
func RunCatchFuncE(cmd *cobra.Command, args []string) error {
|
|
return nil
|
|
}
|