v6 version first commit

internal/cmd/qwen_login.go

@@ -1,95 +1,54 @@
-// Package cmd provides command-line interface functionality for the CLI Proxy API.
-// It implements the main application commands including login/authentication
-// and server startup, handling the complete user onboarding and service lifecycle.
 package cmd
 
 import (
 	"context"
+	"errors"
 	"fmt"
-	"os"
 
-	"github.com/luispater/CLIProxyAPI/v5/internal/auth/qwen"
-	"github.com/luispater/CLIProxyAPI/v5/internal/browser"
-	"github.com/luispater/CLIProxyAPI/v5/internal/client"
-	"github.com/luispater/CLIProxyAPI/v5/internal/config"
+	"github.com/router-for-me/CLIProxyAPI/v6/internal/config"
+	sdkAuth "github.com/router-for-me/CLIProxyAPI/v6/sdk/auth"
 	log "github.com/sirupsen/logrus"
 )
 
-// DoQwenLogin handles the Qwen OAuth login process for Alibaba Qwen services.
-// It initializes the OAuth flow, opens the user's browser for authentication,
-// waits for the callback, exchanges the authorization code for tokens,
-// and saves the authentication information to a file.
-//
-// Parameters:
-//   - cfg: The application configuration
-//   - options: The login options containing browser preferences
+// DoQwenLogin handles the Qwen device flow using the shared authentication manager.
 func DoQwenLogin(cfg *config.Config, options *LoginOptions) {
 	if options == nil {
 		options = &LoginOptions{}
 	}
 
-	ctx := context.Background()
+	manager := newAuthManager()
 
-	log.Info("Initializing Qwen authentication...")
-
-	// Initialize Qwen auth service
-	qwenAuth := qwen.NewQwenAuth(cfg)
-
-	// Generate authorization URL
-	deviceFlow, err := qwenAuth.InitiateDeviceFlow(ctx)
-	if err != nil {
-		log.Fatalf("Failed to generate authorization URL: %v", err)
-		return
-	}
-	authURL := deviceFlow.VerificationURIComplete
-
-	// Open browser or display URL
-	if !options.NoBrowser {
-		log.Info("Opening browser for authentication...")
-
-		// Check if browser is available
-		if !browser.IsAvailable() {
-			log.Warn("No browser available on this system")
-			log.Infof("Please manually open this URL in your browser:\n\n%s\n", authURL)
-		} else {
-			if err = browser.OpenURL(authURL); err != nil {
-				log.Infof("Please manually open this URL in your browser:\n\n%s\n", authURL)
-
-				// Log platform info for debugging
-				platformInfo := browser.GetPlatformInfo()
-				log.Debugf("Browser platform info: %+v", platformInfo)
-			} else {
-				log.Debug("Browser opened successfully")
-			}
+	promptFn := options.Prompt
+	if promptFn == nil {
+		promptFn = func(prompt string) (string, error) {
+			fmt.Println()
+			fmt.Println(prompt)
+			var value string
+			_, err := fmt.Scanln(&value)
+			return value, err
 		}
-	} else {
-		log.Infof("Please open this URL in your browser:\n\n%s\n", authURL)
 	}
 
-	log.Info("Waiting for authentication...")
-	tokenData, err := qwenAuth.PollForToken(deviceFlow.DeviceCode, deviceFlow.CodeVerifier)
+	authOpts := &sdkAuth.LoginOptions{
+		NoBrowser: options.NoBrowser,
+		Metadata:  map[string]string{},
+		Prompt:    promptFn,
+	}
+
+	_, savedPath, err := manager.Login(context.Background(), "qwen", cfg, authOpts)
 	if err != nil {
-		fmt.Printf("Authentication failed: %v\n", err)
-		os.Exit(1)
-	}
-
-	// Create token storage
-	tokenStorage := qwenAuth.CreateTokenStorage(tokenData)
-
-	// Initialize Qwen client
-	qwenClient := client.NewQwenClient(cfg, tokenStorage)
-
-	fmt.Println("\nPlease input your email address or any alias:")
-	var email string
-	_, _ = fmt.Scanln(&email)
-	tokenStorage.Email = email
-
-	// Save token storage
-	if err = qwenClient.SaveTokenToFile(); err != nil {
-		log.Fatalf("Failed to save authentication tokens: %v", err)
+		var emailErr *sdkAuth.EmailRequiredError
+		if errors.As(err, &emailErr) {
+			log.Error(emailErr.Error())
+			return
+		}
+		log.Fatalf("Qwen authentication failed: %v", err)
 		return
 	}
 
-	log.Info("Authentication successful!")
-	log.Info("You can now use Qwen services through this CLI")
+	if savedPath != "" {
+		log.Infof("Authentication saved to %s", savedPath)
+	}
+
+	log.Info("Qwen authentication successful!")
 }