refactor: make Cosmovisor use cobra

cosmovisor/CHANGELOG.md

@@ -37,7 +37,10 @@ Ref: https://keepachangelog.com/en/1.0.0/
 ## [Unreleased]
 <!-- NOTE: when creating a new release, update cosmovisor/cmd/cosmovisor/cmd/version.go:Version -->
 
-* [\#11731](https://github.com/cosmos/cosmos-sdk/pull/11731) `cosmovisor version --json` returns the cosmovisor version and the result of `simd --output json --long` in one JSON object.
+### Features
+
+* [\#11823](https://github.com/cosmos/cosmos-sdk/pull/11823) Refactor `cosmovisor` CLI to use `cobra`
+* [\#11731](https://github.com/cosmos/cosmos-sdk/pull/11731) `cosmovisor version -o json` returns the cosmovisor version and the result of `simd --output json --long` in one JSON object.
 
 ## v1.1.0 2022-10-02
 

cosmovisor/cmd/cosmovisor/cmd/help.go

@@ -6,24 +6,6 @@ import (
 	"github.com/cosmos/cosmos-sdk/cosmovisor"
 )
 
-// HelpArgs are the strings that indicate a cosmovisor help command.
-var HelpArgs = []string{"help", "--help", "-h"}
-
-// ShouldGiveHelp checks the env and provided args to see if help is needed or being requested.
-// Help is needed if either cosmovisor.EnvName and/or cosmovisor.EnvHome env vars aren't set.
-// Help is requested if the first arg is "help", "--help", or "-h".
-func ShouldGiveHelp(arg string) bool {
-	return isOneOf(arg, HelpArgs)
-}
-
-// DoHelp outputs help text
-func DoHelp() error {
-	// Not using the logger for this output because the header and footer look weird for help text.
-	fmt.Println(GetHelpText())
-
-	return nil
-}
-
 // GetHelpText creates the help text multi-line string.
 func GetHelpText() string {
 	return fmt.Sprintf(`Cosmosvisor - A process manager for Cosmos SDK application binaries.
@@ -36,14 +18,5 @@ the proposal. Cosmovisor interprets that data to perform an update: switch a cur
 and restart the App.
 
 Configuration of Cosmovisor is done through environment variables, which are
-documented in: https://github.com/cosmos/cosmos-sdk/tree/main/cosmovisor/README.md
-
-To get help for the configured binary:
-  cosmovisor run help
-
-Available Commands:
-  help     This help message
-  run      Runs app passing all subsequent parameters
-  version  Prints version of cosmovisor and the associated app.
-`, cosmovisor.EnvName, cosmovisor.EnvHome)
+documented in: https://github.com/cosmos/cosmos-sdk/tree/main/cosmovisor/README.md`, cosmovisor.EnvName, cosmovisor.EnvHome)
 }

cosmovisor/cmd/cosmovisor/cmd/help_test.go

@@ -5,7 +5,6 @@ import (
 	"os"
 	"testing"
 
-	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"github.com/stretchr/testify/suite"
 
@@ -88,72 +87,6 @@ func (s *HelpTestSuite) setEnv(t *testing.T, env *cosmovisorHelpEnv) {
 	}
 }
 
-func (s HelpTestSuite) TestShouldGiveHelpArg() {
-	initialEnv := s.clearEnv()
-	defer s.setEnv(nil, initialEnv)
-
-	s.setEnv(s.T(), &cosmovisorHelpEnv{"/testhome", "testname"})
-
-	tests := []struct {
-		name     string
-		arg      string
-		expected bool
-	}{
-		{
-			name:     "empty string",
-			arg:      "",
-			expected: false,
-		},
-		{
-			name:     "random",
-			arg:      "random",
-			expected: false,
-		},
-		{
-			name:     "help",
-			arg:      "help",
-			expected: true,
-		},
-		{
-			name:     "-h",
-			arg:      "-h",
-			expected: true,
-		},
-		{
-			name:     "--help",
-			arg:      "--help",
-			expected: true,
-		},
-		{
-			name:     "help weird casing",
-			arg:      "hELP",
-			expected: true,
-		},
-		{
-			name:     "version",
-			arg:      "version",
-			expected: false,
-		},
-		{
-			name:     "--version",
-			arg:      "--version",
-			expected: false,
-		},
-		{
-			name:     "run",
-			arg:      "run",
-			expected: false,
-		},
-	}
-
-	for _, tc := range tests {
-		s.T().Run(fmt.Sprintf("%s - %t", tc.name, tc.expected), func(t *testing.T) {
-			actual := ShouldGiveHelp(tc.arg)
-			assert.Equal(t, tc.expected, actual)
-		})
-	}
-}
-
 func (s *HelpTestSuite) TestGetHelpText() {
 	expectedPieces := []string{
 		"Cosmosvisor",

cosmovisor/cmd/cosmovisor/cmd/root.go

@@ -1,44 +1,28 @@
 package cmd
 
 import (
-	"strings"
+	"os"
 
 	"github.com/rs/zerolog"
-)
-
-// RunCosmovisorCommand executes the desired cosmovisor command.
-func RunCosmovisorCommand(logger *zerolog.Logger, args []string) error {
-	arg0 := ""
-	if len(args) > 0 {
-		arg0 = strings.TrimSpace(args[0])
-	}
-
-	switch {
-	case IsVersionCommand(arg0):
-		return PrintVersion(logger, args[1:])
-
-	case ShouldGiveHelp(arg0):
-		return DoHelp()
+	"github.com/spf13/cobra"
 
-	case IsRunCommand(arg0):
-		return Run(logger, args[1:])
-	}
+	cverrors "github.com/cosmos/cosmos-sdk/cosmovisor/errors"
+)
 
-	warnRun := func() {
-		logger.Warn().Msg("use of cosmovisor without the 'run' command is deprecated. Use: cosmovisor run [args]")
-	}
-	warnRun()
-	defer warnRun()
+var logger *zerolog.Logger
 
-	return Run(logger, args)
+var rootCmd = &cobra.Command{
+	Use:   "cosmovisor",
+	Short: "A process manager for Cosmos SDK application binaries.",
+	Long:  GetHelpText(),
 }
 
-// isOneOf returns true if the given arg equals one of the provided options (ignoring case).
-func isOneOf(arg string, options []string) bool {
-	for _, opt := range options {
-		if strings.EqualFold(arg, opt) {
-			return true
-		}
+// Execute the CLI application.
+func Execute(log *zerolog.Logger) {
+	logger = log
+
+	if err := rootCmd.Execute(); err != nil {
+		cverrors.LogErrors(logger, "", err)
+		os.Exit(1)
 	}
-	return false
 }

cosmovisor/cmd/cosmovisor/cmd/run.go

@@ -2,19 +2,25 @@ package cmd
 
 import (
 	"github.com/cosmos/cosmos-sdk/cosmovisor"
-	"github.com/rs/zerolog"
+	"github.com/spf13/cobra"
 )
 
-// RunArgs are the strings that indicate a cosmovisor run command.
-var RunArgs = []string{"run"}
+func init() {
+	rootCmd.AddCommand(runCmd)
+}
 
-// IsRunCommand checks if the given args indicate that a run is desired.
-func IsRunCommand(arg string) bool {
-	return isOneOf(arg, RunArgs)
+var runCmd = &cobra.Command{
+	Use:                "run",
+	Short:              "Run an APP command.",
+	SilenceUsage:       true,
+	DisableFlagParsing: true,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		return Run(args)
+	},
 }
 
 // Run runs the configured program with the given args and monitors it for upgrades.
-func Run(logger *zerolog.Logger, args []string, options ...RunOption) error {
+func Run(args []string, options ...RunOption) error {
 	cfg, err := cosmovisor.GetConfigFromEnv()
 	if err != nil {
 		return err

cosmovisor/cmd/cosmovisor/cmd/run_test.go

@@ -1,76 +1,3 @@
 package cmd
 
-import (
-	"fmt"
-	"testing"
-
-	"github.com/stretchr/testify/require"
-)
-
-func TestIsRunCommand(t *testing.T) {
-	cases := []struct {
-		name     string
-		arg      string
-		expected bool
-	}{
-		{
-			name:     "empty string",
-			arg:      "",
-			expected: false,
-		},
-		{
-			name:     "random",
-			arg:      "random",
-			expected: false,
-		},
-		{
-			name:     "run",
-			arg:      "run",
-			expected: true,
-		},
-		{
-			name:     "run weird casing",
-			arg:      "RUn",
-			expected: true,
-		},
-		{
-			name:     "--run",
-			arg:      "--run",
-			expected: false,
-		},
-		{
-			name:     "help",
-			arg:      "help",
-			expected: false,
-		},
-		{
-			name:     "-h",
-			arg:      "-h",
-			expected: false,
-		},
-		{
-			name:     "--help",
-			arg:      "--help",
-			expected: false,
-		},
-		{
-			name:     "version",
-			arg:      "version",
-			expected: false,
-		},
-		{
-			name:     "--version",
-			arg:      "--version",
-			expected: false,
-		},
-	}
-
-	for _, tc := range cases {
-		t.Run(fmt.Sprintf("%s - %t", tc.name, tc.expected), func(t *testing.T) {
-			actual := IsRunCommand(tc.arg)
-			require.Equal(t, tc.expected, actual)
-		})
-	}
-}
-
 // TODO: Write tests for func Run(args []string) error