mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-28 10:24:27 +08:00
105 lines
3.6 KiB
Go
105 lines
3.6 KiB
Go
// Package gemini provides authentication and token management functionality
|
|
// for Google's Gemini AI services. It handles OAuth2 token storage, serialization,
|
|
// and retrieval for maintaining authenticated sessions with the Gemini API.
|
|
package gemini
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/misc"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// GeminiTokenStorage stores OAuth2 token information for Google Gemini API authentication.
|
|
// It maintains compatibility with the existing auth system while adding Gemini-specific fields
|
|
// for managing access tokens, refresh tokens, and user account information.
|
|
type GeminiTokenStorage struct {
|
|
// Token holds the raw OAuth2 token data, including access and refresh tokens.
|
|
Token any `json:"token"`
|
|
|
|
// ProjectID is the Google Cloud Project ID associated with this token.
|
|
ProjectID string `json:"project_id"`
|
|
|
|
// Email is the email address of the authenticated user.
|
|
Email string `json:"email"`
|
|
|
|
// Auto indicates if the project ID was automatically selected.
|
|
Auto bool `json:"auto"`
|
|
|
|
// Checked indicates if the associated Cloud AI API has been verified as enabled.
|
|
Checked bool `json:"checked"`
|
|
|
|
// Type indicates the authentication provider type, always "gemini" for this storage.
|
|
Type string `json:"type"`
|
|
|
|
// Metadata holds arbitrary key-value pairs injected via hooks.
|
|
// It is not exported to JSON directly to allow flattening during serialization.
|
|
Metadata map[string]any `json:"-"`
|
|
}
|
|
|
|
// SetMetadata allows external callers to inject metadata into the storage before saving.
|
|
func (ts *GeminiTokenStorage) SetMetadata(meta map[string]any) {
|
|
ts.Metadata = meta
|
|
}
|
|
|
|
// SaveTokenToFile serializes the Gemini token storage to a JSON file.
|
|
// This method creates the necessary directory structure and writes the token
|
|
// data in JSON format to the specified file path for persistent storage.
|
|
// It merges any injected metadata into the top-level JSON object.
|
|
//
|
|
// Parameters:
|
|
// - authFilePath: The full path where the token file should be saved
|
|
//
|
|
// Returns:
|
|
// - error: An error if the operation fails, nil otherwise
|
|
func (ts *GeminiTokenStorage) SaveTokenToFile(authFilePath string) error {
|
|
misc.LogSavingCredentials(authFilePath)
|
|
ts.Type = "gemini"
|
|
// Merge metadata using helper
|
|
data, errMerge := misc.MergeMetadata(ts, ts.Metadata)
|
|
if errMerge != nil {
|
|
return fmt.Errorf("failed to merge metadata: %w", errMerge)
|
|
}
|
|
if err := os.MkdirAll(filepath.Dir(authFilePath), os.ModePerm); err != nil {
|
|
return fmt.Errorf("failed to create directory: %v", err)
|
|
}
|
|
|
|
f, err := os.Create(authFilePath)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create token file: %w", err)
|
|
}
|
|
defer func() {
|
|
if errClose := f.Close(); errClose != nil {
|
|
log.Errorf("failed to close file: %v", errClose)
|
|
}
|
|
}()
|
|
|
|
enc := json.NewEncoder(f)
|
|
enc.SetIndent("", " ")
|
|
if err := enc.Encode(data); err != nil {
|
|
return fmt.Errorf("failed to encode token to file: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CredentialFileName returns the filename used to persist Gemini CLI credentials.
|
|
// When projectID represents multiple projects (comma-separated or literal ALL),
|
|
// the suffix is normalized to "all" and a "gemini-" prefix is enforced to keep
|
|
// web and CLI generated files consistent.
|
|
func CredentialFileName(email, projectID string, includeProviderPrefix bool) string {
|
|
email = strings.TrimSpace(email)
|
|
project := strings.TrimSpace(projectID)
|
|
if strings.EqualFold(project, "all") || strings.Contains(project, ",") {
|
|
return fmt.Sprintf("gemini-%s-all.json", email)
|
|
}
|
|
prefix := ""
|
|
if includeProviderPrefix {
|
|
prefix = "gemini-"
|
|
}
|
|
return fmt.Sprintf("%s%s-%s.json", prefix, email, project)
|
|
}
|