♻️ Refactor flag usage and mcp logging

issue #64

Author: OpenWaygate

cmd/auth.go

@@ -5,15 +5,22 @@ import (
 	"github.com/spf13/cobra"
 )
 
+const (
+	authShort  = "Authenticate with YouTube API"
+	authLong   = "Authenticate with YouTube API to access and manage YouTube resources."
+	credUsage  = "Path to client secret file, or base64 encoded string, or json string"
+	cacheUsage = "Path to token cache file, or base64 encoded string, or json string"
+)
+
 var (
 	credential string
 	cacheToken string
 )
 
 var authCmd = &cobra.Command{
 	Use:   "auth",
-	Short: "Authenticate with YouTube API",
-	Long:  "Authenticate with YouTube API",
+	Short: authShort,
+	Long:  authLong,
 	Run: func(cmd *cobra.Command, args []string) {
 		auth.NewY2BService(
 			auth.WithCredential(credential),
@@ -27,11 +34,9 @@ func init() {
 	RootCmd.AddCommand(authCmd)
 
 	authCmd.Flags().StringVarP(
-		&credential, "credential", "c", "client_secret.json",
-		"Path to client secret file, or base64 encoded string, or json string",
+		&credential, "credential", "c", "client_secret.json", credUsage,
 	)
 	authCmd.Flags().StringVarP(
-		&cacheToken, "cacheToken", "t", "youtube.token.json",
-		"Path to token cache file, or base64 encoded string, or json string",
+		&cacheToken, "cacheToken", "t", "youtube.token.json", cacheUsage,
 	)
 }

cmd/mcp.go

@@ -9,6 +9,8 @@ import (
 )
 
 const (
+	mcpShort  = "Start MCP server"
+	mcpLong   = "Start MCP server to handle requests from clients"
 	modeUsage = "stdio, http, or sse"
 	portUsage = "Port to listen on for HTTP or SSE mode"
 )
@@ -21,19 +23,20 @@ var (
 var MCP = server.NewMCPServer(
 	"yutu", Version,
 	server.WithToolCapabilities(true),
+	server.WithLogging(),
 	server.WithRecovery(),
 )
 
 var mcpCmd = &cobra.Command{
 	Use:   "mcp",
-	Short: "Start mcp server",
-	Long:  "Start mcp server to handle requests from clients",
+	Short: mcpShort,
+	Long:  mcpLong,
 	Run: func(cmd *cobra.Command, args []string) {
 		var err error
 		interval := 13 * time.Second
 		addr := fmt.Sprintf(":%d", port)
 		baseURL := fmt.Sprintf("http://localhost:%d", port)
-		message := fmt.Sprintf("Starting MCP server: %s", baseURL)
+		message := fmt.Sprintf("%s server listening on %s", mode, addr)
 
 		switch mode {
 		case "stdio":
@@ -56,7 +59,7 @@ var mcpCmd = &cobra.Command{
 		}
 
 		if err != nil {
-			fmt.Printf("Server error: %v\n", err)
+			log.Fatalf("Server error: %v\n", err)
 		}
 	},
 }

cmd/root.go

@@ -7,10 +7,15 @@ import (
 	"github.com/spf13/viper"
 )
 
+const (
+	short = "A fully functional CLI for YouTube"
+	long  = "yutu is a fully functional CLI for YouTube, which can be used to manipulate YouTube videos, playlists, channels, etc"
+)
+
 var RootCmd = &cobra.Command{
 	Use:   "yutu",
-	Short: "A fully functional CLI for YouTube",
-	Long:  "yutu is a fully functional CLI for YouTube, which can be used to manipulate YouTube videos, playlists, channels, etc.",
+	Short: short,
+	Long:  long,
 
 	Run: func(cmd *cobra.Command, args []string) {
 		_ = cmd.Help()

cmd/version.go

@@ -5,6 +5,11 @@ import (
 	"github.com/spf13/cobra"
 )
 
+const (
+	versionShort = "Show the version of yutu"
+	versionLong  = "Show the version of yutu"
+)
+
 var (
 	Version    = ""
 	Commit     = ""
@@ -16,8 +21,8 @@ var (
 
 var versionCmd = &cobra.Command{
 	Use:   "version",
-	Short: "Show the version of yutu",
-	Long:  "Show the version of yutu",
+	Short: versionShort,
+	Long:  versionLong,
 	Run: func(cmd *cobra.Command, args []string) {
 		cmd.Printf("yutu🐰 version %s %s/%s", Version, Os, Arch)
 		if Commit != "" && CommitDate != "" {