CLIProxyAPI/internal/cmd/gemini-web_auth.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)
}