mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-03 04:50:52 +08:00
61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
// Package cmd provides command-line interface functionality for the CLI Proxy API.
|
|
package cmd
|
|
|
|
import (
|
|
"bufio"
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/auth/gemini"
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/config"
|
|
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 == "" {
|
|
log.Fatal("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])
|
|
filePath := filepath.Join(cfg.AuthDir, fileName)
|
|
|
|
err := tokenStorage.SaveTokenToFile(filePath)
|
|
if err != nil {
|
|
log.Fatalf("Failed to save Gemini Web token to file: %v", err)
|
|
return
|
|
}
|
|
|
|
log.Infof("Successfully saved Gemini Web token to: %s", filePath)
|
|
}
|