chuan/CLIProxyAPI
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPI.git synced 2026-02-03 04:50:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(auth): export Gemini constants and use in handler

Browse Source
This commit is contained in:
hkfires
2026-01-23 21:55:44 +08:00
parent 405df58f72
commit 8c0eaa1f71
2 changed files with 24 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
internal/api/handlers/management/auth_files.go
View File

@@ -1020,17 +1020,13 @@ func (h *Handler) RequestGeminiCLIToken(c *gin.Context) {
fmt.Println("Initializing Google authentication...") fmt.Println("Initializing Google authentication...")
// OAuth2 configuration (mirrors internal/auth/gemini) // OAuth2 configuration using exported constants from internal/auth/gemini
conf := &oauth2.Config{ conf := &oauth2.Config{
ClientID: "681255809395-oo8ft2oprdrnp9e3aqf6av3hmdib135j.apps.googleusercontent.com", ClientID: geminiAuth.ClientID,
ClientSecret: "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", ClientSecret: geminiAuth.ClientSecret,
RedirectURL: "http://localhost:8085/oauth2callback", RedirectURL: fmt.Sprintf("http://localhost:%d/oauth2callback", geminiAuth.DefaultCallbackPort),
Scopes: []string{ Scopes: geminiAuth.Scopes,
"https://www.googleapis.com/auth/cloud-platform", Endpoint: google.Endpoint,
"https://www.googleapis.com/auth/userinfo.email",
"https://www.googleapis.com/auth/userinfo.profile",
},
Endpoint: google.Endpoint,
} }
// Build authorization URL and return it immediately // Build authorization URL and return it immediately

36
internal/auth/gemini/gemini_auth.go
View File

@@ -28,19 +28,19 @@ import (
"golang.org/x/oauth2/google" "golang.org/x/oauth2/google"
) )
// OAuth configuration constants for Gemini
const ( const (
geminiOauthClientID = "681255809395-oo8ft2oprdrnp9e3aqf6av3hmdib135j.apps.googleusercontent.com" ClientID = "681255809395-oo8ft2oprdrnp9e3aqf6av3hmdib135j.apps.googleusercontent.com"
geminiOauthClientSecret = "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" ClientSecret = "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"
geminiDefaultCallbackPort = 8085 DefaultCallbackPort = 8085
) )
var ( // OAuth scopes for Gemini authentication
geminiOauthScopes = []string{ var Scopes = []string{
"https://www.googleapis.com/auth/cloud-platform", "https://www.googleapis.com/auth/cloud-platform",
"https://www.googleapis.com/auth/userinfo.email", "https://www.googleapis.com/auth/userinfo.email",
"https://www.googleapis.com/auth/userinfo.profile", "https://www.googleapis.com/auth/userinfo.profile",
} }
)
// GeminiAuth provides methods for handling the Gemini OAuth2 authentication flow. // GeminiAuth provides methods for handling the Gemini OAuth2 authentication flow.
// It encapsulates the logic for obtaining, storing, and refreshing authentication tokens // It encapsulates the logic for obtaining, storing, and refreshing authentication tokens
@@ -74,7 +74,7 @@ func NewGeminiAuth() *GeminiAuth {
// - *http.Client: An HTTP client configured with authentication // - *http.Client: An HTTP client configured with authentication
// - error: An error if the client configuration fails, nil otherwise // - error: An error if the client configuration fails, nil otherwise
func (g *GeminiAuth) GetAuthenticatedClient(ctx context.Context, ts *GeminiTokenStorage, cfg *config.Config, opts *WebLoginOptions) (*http.Client, error) { func (g *GeminiAuth) GetAuthenticatedClient(ctx context.Context, ts *GeminiTokenStorage, cfg *config.Config, opts *WebLoginOptions) (*http.Client, error) {
callbackPort := geminiDefaultCallbackPort callbackPort := DefaultCallbackPort
if opts != nil && opts.CallbackPort > 0 { if opts != nil && opts.CallbackPort > 0 {
callbackPort = opts.CallbackPort callbackPort = opts.CallbackPort
} }
@@ -112,10 +112,10 @@ func (g *GeminiAuth) GetAuthenticatedClient(ctx context.Context, ts *GeminiToken
// Configure the OAuth2 client. // Configure the OAuth2 client.
conf := &oauth2.Config{ conf := &oauth2.Config{
ClientID: geminiOauthClientID, ClientID: ClientID,
ClientSecret: geminiOauthClientSecret, ClientSecret: ClientSecret,
RedirectURL: callbackURL, // This will be used by the local server. RedirectURL: callbackURL, // This will be used by the local server.
Scopes: geminiOauthScopes, Scopes: Scopes,
Endpoint: google.Endpoint, Endpoint: google.Endpoint,
} }
@@ -198,9 +198,9 @@ func (g *GeminiAuth) createTokenStorage(ctx context.Context, config *oauth2.Conf
} }
ifToken["token_uri"] = "https://oauth2.googleapis.com/token" ifToken["token_uri"] = "https://oauth2.googleapis.com/token"
ifToken["client_id"] = geminiOauthClientID ifToken["client_id"] = ClientID
ifToken["client_secret"] = geminiOauthClientSecret ifToken["client_secret"] = ClientSecret
ifToken["scopes"] = geminiOauthScopes ifToken["scopes"] = Scopes
ifToken["universe_domain"] = "googleapis.com" ifToken["universe_domain"] = "googleapis.com"
ts := GeminiTokenStorage{ ts := GeminiTokenStorage{
@@ -226,7 +226,7 @@ func (g *GeminiAuth) createTokenStorage(ctx context.Context, config *oauth2.Conf
// - *oauth2.Token: The OAuth2 token obtained from the authorization flow // - *oauth2.Token: The OAuth2 token obtained from the authorization flow
// - error: An error if the token acquisition fails, nil otherwise // - error: An error if the token acquisition fails, nil otherwise
func (g *GeminiAuth) getTokenFromWeb(ctx context.Context, config *oauth2.Config, opts *WebLoginOptions) (*oauth2.Token, error) { func (g *GeminiAuth) getTokenFromWeb(ctx context.Context, config *oauth2.Config, opts *WebLoginOptions) (*oauth2.Token, error) {
callbackPort := geminiDefaultCallbackPort callbackPort := DefaultCallbackPort
if opts != nil && opts.CallbackPort > 0 { if opts != nil && opts.CallbackPort > 0 {
callbackPort = opts.CallbackPort callbackPort = opts.CallbackPort
} }