69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
// nolint:gochecknoglobals,gochecknoinits
|
|
package cmds
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
"go.xsfx.dev/don"
|
|
)
|
|
|
|
const (
|
|
defaultTimeout = 10 * time.Second
|
|
defaultRetry = time.Second
|
|
defaultProgressbar = true
|
|
)
|
|
|
|
var (
|
|
command string
|
|
timeout time.Duration
|
|
retry time.Duration
|
|
progressbar bool
|
|
)
|
|
|
|
var (
|
|
version = "dev"
|
|
commit = "none"
|
|
date = "unknown"
|
|
)
|
|
|
|
var rootCmd = &cobra.Command{
|
|
Use: "don [command]",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if err := don.Ready(don.Cmd(command), timeout, retry, progressbar); err != nil {
|
|
log.Fatal().Err(err).Msg("received error")
|
|
}
|
|
|
|
log.Info().Msg("ready")
|
|
},
|
|
}
|
|
|
|
var versionCmd = &cobra.Command{
|
|
Use: "version",
|
|
Short: "Print version informations",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
// nolint: forbidigo
|
|
fmt.Printf("don %s, commit %s, build on %s\n", version, commit, date)
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(versionCmd)
|
|
rootCmd.Flags().StringVarP(&command, "command", "c", "", "command to run (required)")
|
|
|
|
if err := rootCmd.MarkFlagRequired("command"); err != nil {
|
|
log.Fatal().Err(err).Msg("needs command flag")
|
|
}
|
|
|
|
rootCmd.Flags().DurationVarP(&timeout, "timeout", "t", defaultTimeout, "timeout")
|
|
rootCmd.Flags().DurationVarP(&retry, "retry", "r", defaultRetry, "retry")
|
|
rootCmd.Flags().BoolVarP(&progressbar, "progressbar", "p", defaultProgressbar, "progressbar")
|
|
}
|
|
|
|
func Execute() error {
|
|
// nolint:wrapcheck
|
|
return rootCmd.Execute()
|
|
}
|