mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-02 04:20:50 +08:00
a1da6ff5ac
feat(oauth): add support for customizable OAuth callback ports - Introduced `oauth-callback-port` flag to override default callback ports. - Updated SDK and login flows for `iflow`, `gemini`, `antigravity`, `codex`, `claude`, and `openai` to respect configurable callback ports. - Refactored internal OAuth servers to dynamically assign ports based on the provided options. - Revised tests and documentation to reflect the new flag and behavior.
74 lines
1.8 KiB
Go
74 lines
1.8 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"
|
|
coreauth "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/auth"
|
|
)
|
|
|
|
// 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) (*coreauth.Auth, 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, &gemini.WebLoginOptions{
|
|
NoBrowser: opts.NoBrowser,
|
|
CallbackPort: opts.CallbackPort,
|
|
Prompt: opts.Prompt,
|
|
})
|
|
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]any{
|
|
"email": ts.Email,
|
|
"project_id": ts.ProjectID,
|
|
}
|
|
|
|
fmt.Println("Gemini authentication successful")
|
|
|
|
return &coreauth.Auth{
|
|
ID: fileName,
|
|
Provider: a.Provider(),
|
|
FileName: fileName,
|
|
Storage: &ts,
|
|
Metadata: metadata,
|
|
}, nil
|
|
}
|