mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-02 04:20:50 +08:00
cf734f7e7b
- Implemented a global logger with structured formatting for consistent log output. - Added support for rotating log files using Lumberjack. - Integrated new logging functionality with Gin HTTP server for unified log handling. - Replaced direct `log.Info` calls with `fmt.Printf` in non-critical paths to simplify core functionality.
68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/auth/gemini"
|
|
// legacy client removed
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/config"
|
|
)
|
|
|
|
// GeminiAuthenticator implements the login flow for Google Gemini CLI accounts.
|
|
type GeminiAuthenticator struct{}
|
|
|
|
// NewGeminiAuthenticator constructs a Gemini authenticator.
|
|
func NewGeminiAuthenticator() *GeminiAuthenticator {
|
|
return &GeminiAuthenticator{}
|
|
}
|
|
|
|
func (a *GeminiAuthenticator) Provider() string {
|
|
return "gemini"
|
|
}
|
|
|
|
func (a *GeminiAuthenticator) RefreshLead() *time.Duration {
|
|
return nil
|
|
}
|
|
|
|
func (a *GeminiAuthenticator) Login(ctx context.Context, cfg *config.Config, opts *LoginOptions) (*TokenRecord, error) {
|
|
if cfg == nil {
|
|
return nil, fmt.Errorf("cliproxy auth: configuration is required")
|
|
}
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
if opts == nil {
|
|
opts = &LoginOptions{}
|
|
}
|
|
|
|
var ts gemini.GeminiTokenStorage
|
|
if opts.ProjectID != "" {
|
|
ts.ProjectID = opts.ProjectID
|
|
}
|
|
|
|
geminiAuth := gemini.NewGeminiAuth()
|
|
_, err := geminiAuth.GetAuthenticatedClient(ctx, &ts, cfg, opts.NoBrowser)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("gemini authentication failed: %w", err)
|
|
}
|
|
|
|
// Skip onboarding here; rely on upstream configuration
|
|
|
|
fileName := fmt.Sprintf("%s-%s.json", ts.Email, ts.ProjectID)
|
|
metadata := map[string]string{
|
|
"email": ts.Email,
|
|
"project_id": ts.ProjectID,
|
|
}
|
|
|
|
fmt.Println("Gemini authentication successful")
|
|
|
|
return &TokenRecord{
|
|
Provider: a.Provider(),
|
|
FileName: fileName,
|
|
Storage: &ts,
|
|
Metadata: metadata,
|
|
}, nil
|
|
}
|