mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-02 04:20:50 +08:00
e3be548e8d
- Updated all authentication commands to use `fmt` instead of `log` for errors, info, and success messages. - Ensured message formatting aligns across commands for consistent user experience.
66 lines
1.8 KiB
Go
66 lines
1.8 KiB
Go
// Package cmd provides command-line interface functionality for the CLI Proxy API.
|
|
package cmd
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/auth/gemini"
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/config"
|
|
sdkAuth "github.com/router-for-me/CLIProxyAPI/v6/sdk/auth"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// DoGeminiWebAuth handles the process of creating a Gemini Web token file.
|
|
// It prompts the user for their cookie values and saves them to a JSON file.
|
|
func DoGeminiWebAuth(cfg *config.Config) {
|
|
reader := bufio.NewReader(os.Stdin)
|
|
|
|
fmt.Print("Enter your __Secure-1PSID cookie value: ")
|
|
secure1psid, _ := reader.ReadString('\n')
|
|
secure1psid = strings.TrimSpace(secure1psid)
|
|
|
|
if secure1psid == "" {
|
|
log.Fatal("The __Secure-1PSID value cannot be empty.")
|
|
return
|
|
}
|
|
|
|
fmt.Print("Enter your __Secure-1PSIDTS cookie value: ")
|
|
secure1psidts, _ := reader.ReadString('\n')
|
|
secure1psidts = strings.TrimSpace(secure1psidts)
|
|
|
|
if secure1psidts == "" {
|
|
fmt.Println("The __Secure-1PSIDTS value cannot be empty.")
|
|
return
|
|
}
|
|
|
|
tokenStorage := &gemini.GeminiWebTokenStorage{
|
|
Secure1PSID: secure1psid,
|
|
Secure1PSIDTS: secure1psidts,
|
|
}
|
|
|
|
// Generate a filename based on the SHA256 hash of the PSID
|
|
hasher := sha256.New()
|
|
hasher.Write([]byte(secure1psid))
|
|
hash := hex.EncodeToString(hasher.Sum(nil))
|
|
fileName := fmt.Sprintf("gemini-web-%s.json", hash[:16])
|
|
record := &sdkAuth.TokenRecord{
|
|
Provider: "gemini-web",
|
|
FileName: fileName,
|
|
Storage: tokenStorage,
|
|
}
|
|
store := sdkAuth.GetTokenStore()
|
|
savedPath, err := store.Save(context.Background(), cfg, record)
|
|
if err != nil {
|
|
fmt.Printf("Failed to save Gemini Web token to file: %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("Successfully saved Gemini Web token to: %s\n", savedPath)
|
|
}
|