move main.go logic out of commands (#38555)

* move main.go logic out of commands
2025-08-01 03:21:51 +08:00 · 2021-08-26 08:42:22 -05:00
parent 2847f25781
commit 1ced2bc6df
6 changed files with 324 additions and 285 deletions
--- a/pkg/cmd/grafana-cli/commands/cli.go
+++ b/pkg/cmd/grafana-cli/commands/cli.go
@ -0,0 +1,102 @@
+package commands
+
+import (
+	"fmt"
+	"os"
+	"runtime"
+
+	"github.com/fatih/color"
+	"github.com/grafana/grafana/pkg/cmd/grafana-cli/logger"
+	"github.com/grafana/grafana/pkg/cmd/grafana-cli/services"
+	"github.com/grafana/grafana/pkg/cmd/grafana-cli/utils"
+	"github.com/urfave/cli/v2"
+)
+
+// RunCLI is the entrypoint for the grafana-cli command. It returns the exit code for the grafana-cli program.
+func RunCLI(version string) int {
+	setupLogging()
+
+	app := &cli.App{
+		Name: "Grafana CLI",
+		Authors: []*cli.Author{
+			{
+				Name:  "Grafana Project",
+				Email: "hello@grafana.com",
+			},
+		},
+		Version: version,
+		Flags: []cli.Flag{
+			&cli.StringFlag{
+				Name:    "pluginsDir",
+				Usage:   "Path to the Grafana plugin directory",
+				Value:   utils.GetGrafanaPluginDir(runtime.GOOS),
+				EnvVars: []string{"GF_PLUGIN_DIR"},
+			},
+			&cli.StringFlag{
+				Name:    "repo",
+				Usage:   "URL to the plugin repository",
+				Value:   "https://grafana.com/api/plugins",
+				EnvVars: []string{"GF_PLUGIN_REPO"},
+			},
+			&cli.StringFlag{
+				Name:    "pluginUrl",
+				Usage:   "Full url to the plugin zip file instead of downloading the plugin from grafana.com/api",
+				Value:   "",
+				EnvVars: []string{"GF_PLUGIN_URL"},
+			},
+			&cli.BoolFlag{
+				Name:  "insecure",
+				Usage: "Skip TLS verification (insecure)",
+			},
+			&cli.BoolFlag{
+				Name:  "debug, d",
+				Usage: "Enable debug logging",
+			},
+			&cli.StringFlag{
+				Name:  "configOverrides",
+				Usage: "Configuration options to override defaults as a string. e.g. cfg:default.paths.log=/dev/null",
+			},
+			&cli.StringFlag{
+				Name:  "homepath",
+				Usage: "Path to Grafana install/home path, defaults to working directory",
+			},
+			&cli.StringFlag{
+				Name:  "config",
+				Usage: "Path to config file",
+			},
+		},
+		Commands:        Commands,
+		CommandNotFound: cmdNotFound,
+	}
+
+	app.Before = func(c *cli.Context) error {
+		services.Init(version, c.Bool("insecure"), c.Bool("debug"))
+		return nil
+	}
+
+	if err := app.Run(os.Args); err != nil {
+		logger.Errorf("%s: %s %s\n", color.RedString("Error"), color.RedString("✗"), err)
+		return 1
+	}
+
+	return 0
+}
+
+func setupLogging() {
+	for _, f := range os.Args {
+		if f == "-d" || f == "--debug" || f == "-debug" {
+			logger.SetDebug(true)
+		}
+	}
+}
+
+func cmdNotFound(c *cli.Context, command string) {
+	fmt.Printf(
+		"%s: '%s' is not a %s command. See '%s --help'.\n",
+		c.App.Name,
+		command,
+		c.App.Name,
+		os.Args[0],
+	)
+	os.Exit(1)
+}