mirror of
https://github.com/crazy-max/diun.git
synced 2024-12-22 19:38:28 +00:00
58 lines
1.7 KiB
Go
58 lines
1.7 KiB
Go
package kong
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"reflect"
|
|
)
|
|
|
|
// ConfigFlag uses the configured (via kong.Configuration(loader)) configuration loader to load configuration
|
|
// from a file specified by a flag.
|
|
//
|
|
// Use this as a flag value to support loading of custom configuration via a flag.
|
|
type ConfigFlag string
|
|
|
|
// BeforeResolve adds a resolver.
|
|
func (c ConfigFlag) BeforeResolve(kong *Kong, ctx *Context, trace *Path) error {
|
|
if kong.loader == nil {
|
|
return fmt.Errorf("kong must be configured with kong.Configuration(...)")
|
|
}
|
|
path := string(ctx.FlagValue(trace.Flag).(ConfigFlag)) //nolint
|
|
resolver, err := kong.LoadConfig(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ctx.AddResolver(resolver)
|
|
return nil
|
|
}
|
|
|
|
// VersionFlag is a flag type that can be used to display a version number, stored in the "version" variable.
|
|
type VersionFlag bool
|
|
|
|
// BeforeReset writes the version variable and terminates with a 0 exit status.
|
|
func (v VersionFlag) BeforeReset(app *Kong, vars Vars) error {
|
|
fmt.Fprintln(app.Stdout, vars["version"])
|
|
app.Exit(0)
|
|
return nil
|
|
}
|
|
|
|
// ChangeDirFlag changes the current working directory to a path specified by a flag
|
|
// early in the parsing process, changing how other flags resolve relative paths.
|
|
//
|
|
// Use this flag to provide a "git -C" like functionality.
|
|
//
|
|
// It is not compatible with custom named decoders, e.g., existingdir.
|
|
type ChangeDirFlag string
|
|
|
|
// Decode is used to create a side effect of changing the current working directory.
|
|
func (c ChangeDirFlag) Decode(ctx *DecodeContext) error {
|
|
var path string
|
|
err := ctx.Scan.PopValueInto("string", &path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
path = ExpandPath(path)
|
|
ctx.Value.Target.Set(reflect.ValueOf(ChangeDirFlag(path)))
|
|
return os.Chdir(path)
|
|
}
|