Method BaseCommandSet

cmd/river/rivercli/river_cli.go:59–372

BaseCommandSet provides a base River CLI command set which may be further augmented with additional commands.

()

// BaseCommandSet provides a base River CLI command set which may be further
// augmented with additional commands.
func (c *CLI) BaseCommandSet() *cobra.Command {
	ctx := context.Background()

	var globalOpts struct {
		Debug            bool
		StatementTimeout time.Duration
		Verbose          bool
	}
	var rootCmd *cobra.Command

	makeLogger := func() *slog.Logger {
		switch {
		case globalOpts.Debug:
			return slog.New(tint.NewHandler(os.Stdout, &tint.Options{Level: slog.LevelDebug}))
		case globalOpts.Verbose:
			return slog.New(tint.NewHandler(os.Stdout, nil))
		default:
			return slog.New(tint.NewHandler(os.Stdout, &tint.Options{Level: slog.LevelWarn}))
		}
	}

	statementTimeoutFlagSet := func() bool {
		return rootCmd.PersistentFlags().Changed("statement-timeout")
	}

	validateGlobalOpts := func() error {
		if statementTimeoutFlagSet() && globalOpts.StatementTimeout <= time.Millisecond {
			return errors.New("`--statement-timeout` must be greater than 1ms when set")
		}

		return nil
	}

	// Make a bundle for RunCommand. Takes a database URL pointer because not every command is required to take a database URL.
	makeCommandBundle := func(databaseURL *string, schema string) *RunCommandBundle {
		var statementTimeout *time.Duration
		if statementTimeoutFlagSet() {
			statementTimeout = &globalOpts.StatementTimeout
		}

		return &RunCommandBundle{
			DatabaseURL:      databaseURL,
			DriverProcurer:   c.driverProcurer,
			Logger:           makeLogger(),
			OutStd:           c.out,
			Schema:           schema,
			StatementTimeout: statementTimeout,
		}
	}

	{
		var rootOpts struct {
			Version bool
		}

		rootCmd = &cobra.Command{
			Use:   "river",
			Short: "Provides command line facilities for the River job queue",