package console import ( "context" "os" "gitoa.ru/go-4devs/console/input" "gitoa.ru/go-4devs/console/input/value" "gitoa.ru/go-4devs/console/output" ) // WithOutput sets outpu,^ by default output os.Stdout. func WithOutput(out output.Output) func(*App) { return func(a *App) { a.out = out } } // WithInput sets input, by default creates inpur by os.Args. func WithInput(in input.Input) func(*App) { return func(a *App) { a.in = in } } // WithSkipArgs sets how many arguments are passed. For example, you don't need to pass the name of a single command. func WithSkipArgs(l int) func(*App) { return WithInput(input.NewArgs(l)) } // WithExit sets exit callback by default os.Exit. func WithExit(f func(int)) func(*App) { return func(a *App) { a.exit = f } } // New creates and configure new console app. func New(opts ...func(*App)) *App { app := &App{ out: output.Stdout(), exit: os.Exit, in: input.NewArgs(0), } for _, opt := range opts { opt(app) } return app } // App is collection of command and configure env. type App struct { cmds []*Command out output.Output in input.Input exit func(int) } // Add add or replace command. func (a *App) Add(cmds ...*Command) *App { a.cmds = append(a.cmds, cmds...) return a } // Execute run the command by name and arguments. func (a *App) Execute(ctx context.Context) { for _, cmd := range a.cmds { register(cmd) } cmd, err := a.find(ctx) if err != nil { a.printError(ctx, err) if err := a.list(ctx); err != nil { a.printError(ctx, err) } a.exit(1) } a.exec(ctx, cmd) } func (a *App) exec(ctx context.Context, cmd *Command) { if err := Run(ctx, cmd, a.in, a.out); err != nil { a.printError(ctx, err) a.exit(1) } a.exit(0) } func (a *App) find(_ context.Context) (*Command, error) { if len(os.Args) < 2 || os.Args[1][1] == '-' { return Find(CommandList) } name := os.Args[1] return Find(name) } func (a *App) list(ctx context.Context) error { cmd, err := Find(CommandHelp) if err != nil { return err } arr := &input.Array{} arr.SetArgument("command_name", value.New(CommandList)) in := input.Chain(arr, a.in) return Run(ctx, cmd, in, a.out) } func (a *App) printError(ctx context.Context, err error) { ansi(ctx, a.in, a.out).Println(ctx, "\n\n ", err, "\n") }