108 lines
2.2 KiB
Go
108 lines
2.2 KiB
Go
|
package command
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"sort"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"github.com/urfave/cli/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Main(buildDate, projectVersion, gitRef, defaultConfigPath string, commands ...*cli.Command) {
|
||
|
|
ctx := context.Background()
|
||
|
|
|
||
|
|
compiled, err := time.Parse(time.RFC3339, buildDate)
|
||
|
|
if err != nil {
|
||
|
|
panic(errors.Wrapf(err, "could not parse build date '%s'", buildDate))
|
||
|
|
}
|
||
|
|
|
||
|
|
app := &cli.App{
|
||
|
|
Version: fmt.Sprintf("%s (%s, %s)", projectVersion, gitRef, buildDate),
|
||
|
|
Compiled: compiled,
|
||
|
|
Name: "bouncer",
|
||
|
|
Usage: "reverse proxy server with dynamic queuing management",
|
||
|
|
Commands: commands,
|
||
|
|
Before: func(ctx *cli.Context) error {
|
||
|
|
workdir := ctx.String("workdir")
|
||
|
|
// Switch to new working directory if defined
|
||
|
|
if workdir != "" {
|
||
|
|
if err := os.Chdir(workdir); err != nil {
|
||
|
|
return errors.Wrap(err, "could not change working directory")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := ctx.Set("projectVersion", projectVersion); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := ctx.Set("gitRef", gitRef); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := ctx.Set("buildDate", buildDate); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
},
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "workdir",
|
||
|
|
Value: "",
|
||
|
|
Usage: "The working directory",
|
||
|
|
},
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "projectVersion",
|
||
|
|
Value: "",
|
||
|
|
Hidden: true,
|
||
|
|
},
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "gitRef",
|
||
|
|
Value: "",
|
||
|
|
Hidden: true,
|
||
|
|
},
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "buildDate",
|
||
|
|
Value: "",
|
||
|
|
Hidden: true,
|
||
|
|
},
|
||
|
|
&cli.BoolFlag{
|
||
|
|
Name: "debug",
|
||
|
|
EnvVars: []string{"BOUNCER_DEBUG"},
|
||
|
|
Value: false,
|
||
|
|
},
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "config",
|
||
|
|
Aliases: []string{"c"},
|
||
|
|
EnvVars: []string{"BOUNCER_CONFIG"},
|
||
|
|
Value: defaultConfigPath,
|
||
|
|
TakesFile: true,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
app.ExitErrHandler = func(ctx *cli.Context, err error) {
|
||
|
|
if err == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
debug := ctx.Bool("debug")
|
||
|
|
|
||
|
|
if !debug {
|
||
|
|
fmt.Printf("[ERROR] %v\n", err)
|
||
|
|
} else {
|
||
|
|
fmt.Printf("%+v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
sort.Sort(cli.FlagsByName(app.Flags))
|
||
|
|
sort.Sort(cli.CommandsByName(app.Commands))
|
||
|
|
|
||
|
|
if err := app.RunContext(ctx, os.Args); err != nil {
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
}
|