瀏覽代碼

refactor: move token retrieval login to run.go

ayn2op 1 年之前
父節點
當前提交
e386077b96
共有 2 個文件被更改,包括 13 次插入12 次删除
  1. 13 0
      cmd/run.go
  2. 0 12
      main.go

+ 13 - 0
cmd/run.go

@@ -1,8 +1,11 @@
 package cmd
 
 import (
+	"log/slog"
+
 	"github.com/ayn2op/discordo/internal/config"
 	"github.com/ayn2op/discordo/internal/logger"
+	"github.com/zalando/go-keyring"
 )
 
 var (
@@ -15,6 +18,16 @@ func Run(token string) error {
 		return err
 	}
 
+	// If no token was provided, look it up in the keyring.
+	if token == "" {
+		tok, err := keyring.Get(config.Name, "token")
+		if err != nil {
+			slog.Info("failed to get token from keyring", "err", err)
+		} else {
+			token = tok
+		}
+	}
+
 	cfg, err := config.Load()
 	if err != nil {
 		return err

+ 0 - 12
main.go

@@ -5,24 +5,12 @@ import (
 	"log/slog"
 
 	"github.com/ayn2op/discordo/cmd"
-	"github.com/ayn2op/discordo/internal/config"
-	"github.com/zalando/go-keyring"
 )
 
 func main() {
 	token := flag.String("token", "", "authentication token")
 	flag.Parse()
 
-	// If no token was provided, look it up in the keyring
-	if *token == "" {
-		t, err := keyring.Get(config.Name, "token")
-		if err != nil {
-			slog.Info("failed to get token from keyring", "err", err)
-		} else {
-			*token = t
-		}
-	}
-
 	if err := cmd.Run(*token); err != nil {
 		slog.Error("failed to run", "err", err)
 	}