Add Qwen support

internal/cmd/qwen_login.go

@@ -0,0 +1,85 @@
+package cmd
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/luispater/CLIProxyAPI/internal/auth/qwen"
+	"github.com/luispater/CLIProxyAPI/internal/browser"
+	"github.com/luispater/CLIProxyAPI/internal/client"
+	"github.com/luispater/CLIProxyAPI/internal/config"
+	log "github.com/sirupsen/logrus"
+)
+
+// DoQwenLogin handles the Qwen OAuth login process
+func DoQwenLogin(cfg *config.Config, options *LoginOptions) {
+	if options == nil {
+		options = &LoginOptions{}
+	}
+
+	ctx := context.Background()
+
+	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")
+			}
+		}
+	} 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)
+	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)
+		return
+	}
+
+	log.Info("Authentication successful!")
+	log.Info("You can now use Qwen services through this CLI")
+}

internal/cmd/run.go

@@ -22,6 +22,7 @@ import (
 	"github.com/luispater/CLIProxyAPI/internal/auth/claude"
 	"github.com/luispater/CLIProxyAPI/internal/auth/codex"
 	"github.com/luispater/CLIProxyAPI/internal/auth/gemini"
+	"github.com/luispater/CLIProxyAPI/internal/auth/qwen"
 	"github.com/luispater/CLIProxyAPI/internal/client"
 	"github.com/luispater/CLIProxyAPI/internal/config"
 	"github.com/luispater/CLIProxyAPI/internal/util"
@@ -102,6 +103,15 @@ func StartService(cfg *config.Config, configPath string) {
 					log.Info("Authentication successful.")
 					cliClients = append(cliClients, claudeClient)
 				}
+			} else if tokenType == "qwen" {
+				var ts qwen.QwenTokenStorage
+				if err = json.Unmarshal(data, &ts); err == nil {
+					// For each valid token, create an authenticated client.
+					log.Info("Initializing qwen authentication for token...")
+					qwenClient := client.NewQwenClient(cfg, &ts)
+					log.Info("Authentication successful.")
+					cliClients = append(cliClients, qwenClient)
+				}
 			}
 		}
 		return nil
@@ -200,12 +210,23 @@ func StartService(cfg *config.Config, configPath string) {
 						if ts != nil && ts.Expire != "" {
 							if expTime, errParse := time.Parse(time.RFC3339, ts.Expire); errParse == nil {
 								if time.Until(expTime) <= 4*time.Hour {
-									log.Debugf("refreshing codex tokens for %s", claudeCli.GetEmail())
+									log.Debugf("refreshing claude tokens for %s", claudeCli.GetEmail())
 									_ = claudeCli.RefreshTokens(ctxRefresh)
 								}
 							}
 						}
 					}
+				} else if qwenCli, isQwenOK := cliClients[i].(*client.QwenClient); isQwenOK {
+					if ts, isQwenTS := qwenCli.TokenStorage().(*qwen.QwenTokenStorage); isQwenTS {
+						if ts != nil && ts.Expire != "" {
+							if expTime, errParse := time.Parse(time.RFC3339, ts.Expire); errParse == nil {
+								if time.Until(expTime) <= 3*time.Hour {
+									log.Debugf("refreshing qwen tokens for %s", qwenCli.GetEmail())
+									_ = qwenCli.RefreshTokens(ctxRefresh)
+								}
+							}
+						}
+					}
 				}
 			}
 		}