|
|
@@ -1,18 +1,16 @@
|
|
|
-// Package cmd defines the command-line interface commands
|
|
|
package cmd
|
|
|
|
|
|
import (
|
|
|
+ "flag"
|
|
|
"fmt"
|
|
|
"log/slog"
|
|
|
"os"
|
|
|
|
|
|
"github.com/ayn2op/discordo/internal/app"
|
|
|
"github.com/ayn2op/discordo/internal/config"
|
|
|
- "github.com/ayn2op/discordo/internal/consts"
|
|
|
"github.com/ayn2op/discordo/internal/keyring"
|
|
|
"github.com/ayn2op/discordo/internal/logger"
|
|
|
"github.com/diamondburned/arikawa/v3/utils/ws"
|
|
|
- "github.com/spf13/cobra"
|
|
|
)
|
|
|
|
|
|
var (
|
|
|
@@ -20,56 +18,47 @@ var (
|
|
|
configPath string
|
|
|
logPath string
|
|
|
logLevel string
|
|
|
+)
|
|
|
|
|
|
- rootCmd = &cobra.Command{
|
|
|
- Use: consts.Name,
|
|
|
- RunE: func(cmd *cobra.Command, args []string) error {
|
|
|
- var level slog.Level
|
|
|
- switch logLevel {
|
|
|
- case "debug":
|
|
|
- ws.EnableRawEvents = true
|
|
|
- level = slog.LevelDebug
|
|
|
- case "info":
|
|
|
- level = slog.LevelInfo
|
|
|
- case "warn":
|
|
|
- level = slog.LevelWarn
|
|
|
- case "error":
|
|
|
- level = slog.LevelError
|
|
|
- }
|
|
|
-
|
|
|
- if err := logger.Load(logPath, level); err != nil {
|
|
|
- return fmt.Errorf("failed to load logger: %w", err)
|
|
|
- }
|
|
|
-
|
|
|
- cfg, err := config.Load(configPath)
|
|
|
- if err != nil {
|
|
|
- return fmt.Errorf("failed to load config: %w", err)
|
|
|
- }
|
|
|
-
|
|
|
- if token == "" {
|
|
|
- token = os.Getenv("DISCORDO_TOKEN")
|
|
|
- }
|
|
|
+func Run() error {
|
|
|
+ flag.StringVar(&token, "token", "", "authentication token (default: $DISCORDO_TOKEN or keyring)")
|
|
|
+ flag.StringVar(&configPath, "config-path", config.DefaultPath(), "path of the configuration file")
|
|
|
+ flag.StringVar(&logPath, "log-path", logger.DefaultPath(), "path of the log file")
|
|
|
+ flag.StringVar(&logLevel, "log-level", "info", "log level")
|
|
|
+ flag.Parse()
|
|
|
|
|
|
- if token == "" {
|
|
|
- token, err = keyring.GetToken()
|
|
|
- if err != nil {
|
|
|
- slog.Info("failed to retrieve token from keyring", "err", err)
|
|
|
- }
|
|
|
- }
|
|
|
+ var level slog.Level
|
|
|
+ switch logLevel {
|
|
|
+ case "debug":
|
|
|
+ ws.EnableRawEvents = true
|
|
|
+ level = slog.LevelDebug
|
|
|
+ case "info":
|
|
|
+ level = slog.LevelInfo
|
|
|
+ case "warn":
|
|
|
+ level = slog.LevelWarn
|
|
|
+ case "error":
|
|
|
+ level = slog.LevelError
|
|
|
+ }
|
|
|
|
|
|
- return app.New(cfg).Run(token)
|
|
|
- },
|
|
|
+ if err := logger.Load(logPath, level); err != nil {
|
|
|
+ return fmt.Errorf("failed to load logger: %w", err)
|
|
|
}
|
|
|
|
|
|
- Execute = rootCmd.Execute
|
|
|
-)
|
|
|
+ cfg, err := config.Load(configPath)
|
|
|
+ if err != nil {
|
|
|
+ return fmt.Errorf("failed to load config: %w", err)
|
|
|
+ }
|
|
|
|
|
|
-func init() {
|
|
|
- flags := rootCmd.Flags()
|
|
|
- flags.StringVar(&token, "token", "", "authentication token (default: $DISCORDO_TOKEN or keyring)")
|
|
|
+ if token == "" {
|
|
|
+ token = os.Getenv("DISCORDO_TOKEN")
|
|
|
+ }
|
|
|
|
|
|
- flags.StringVar(&configPath, "config-path", config.DefaultPath(), "path of the configuration file")
|
|
|
+ if token == "" {
|
|
|
+ token, err = keyring.GetToken()
|
|
|
+ if err != nil {
|
|
|
+ slog.Info("failed to retrieve token from keyring", "err", err)
|
|
|
+ }
|
|
|
+ }
|
|
|
|
|
|
- flags.StringVar(&logPath, "log-path", logger.DefaultPath(), "path of the log file")
|
|
|
- flags.StringVar(&logLevel, "log-level", "info", "log level")
|
|
|
+ return app.New(cfg).Run(token)
|
|
|
}
|
ayn2op