discordo-plus/cmd/root.go

package cmd

import (
	"log/slog"

	"github.com/ayn2op/discordo/internal/config"
	"github.com/ayn2op/discordo/internal/consts"
	"github.com/ayn2op/discordo/internal/logger"
	"github.com/gdamore/tcell/v2"
	"github.com/rivo/tview"
	"github.com/spf13/cobra"
	"github.com/zalando/go-keyring"
)

var (
	discordState *State
	app          *App
)

var (
	rootCmd = &cobra.Command{
		PreRunE: func(cmd *cobra.Command, args []string) error {
			var level slog.Level
			switch str, _ := cmd.Flags().GetString("log-level"); str {
			case "debug":
				level = slog.LevelDebug
			case "info":
				level = slog.LevelInfo
			case "warn":
				level = slog.LevelWarn
			case "error":
				level = slog.LevelError
			}

			return logger.Load(level)
		},

		RunE: func(cmd *cobra.Command, args []string) error {
			token, _ := cmd.Flags().GetString("token")
			if token == "" {
				var err error
				token, err = keyring.Get(consts.Name, "token")
				if err != nil {
					slog.Info("failed to retrieve token from keyring", "err", err)
				}
			}

			cfg, err := config.Load()
			if err != nil {
				return err
			}

			tview.Styles.PrimitiveBackgroundColor = tcell.GetColor(cfg.Theme.BackgroundColor)

			tview.Borders.Horizontal = cfg.Theme.Border.Preset.Horizontal
			tview.Borders.Vertical = cfg.Theme.Border.Preset.Vertical
			tview.Borders.TopLeft = cfg.Theme.Border.Preset.TopLeft
			tview.Borders.TopRight = cfg.Theme.Border.Preset.TopRight
			tview.Borders.BottomLeft = cfg.Theme.Border.Preset.BottomLeft
			tview.Borders.BottomRight = cfg.Theme.Border.Preset.BottomRight

			tview.Borders.HorizontalFocus = tview.Borders.Horizontal
			tview.Borders.VerticalFocus = tview.Borders.Vertical
			tview.Borders.TopLeftFocus = tview.Borders.TopLeft
			tview.Borders.TopRightFocus = tview.Borders.TopRight
			tview.Borders.BottomLeftFocus = tview.Borders.BottomLeft
			tview.Borders.BottomRightFocus = tview.Borders.BottomRight

			app = newApp(cfg)
			return app.run(token)
		},
	}

	Execute = rootCmd.Execute
)

func init() {
	flags := rootCmd.Flags()
	flags.StringP("token", "t", "", "the authentication token")
	flags.StringP("log-level", "l", "info", "log level")
}