From cc4cc806eab1dd43487583f5f6e82fd2dd91fab1 Mon Sep 17 00:00:00 2001
From: IanShaw027 <131567472+IanShaw027@users.noreply.github.com>
Date: Sun, 4 Jan 2026 10:44:07 +0800
Subject: [PATCH] =?UTF-8?q?feat(backend):=20=E5=A2=9E=E5=8A=A0=20Google=20?=
 =?UTF-8?q?One=20tier=20=E5=88=A4=E6=96=AD=E7=9A=84=E8=AF=A6=E7=BB=86?=
 =?UTF-8?q?=E8=B0=83=E8=AF=95=E6=97=A5=E5=BF=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

**目的：**
排查 Google One 账户 tier 判断不准确的问题（2TB 存储空间应显示 AI Premium，实际显示 Personal）

**新增日志：**
1. FetchGoogleOneTier:
   - LoadCodeAssist API 调用结果（是否返回 tier）
   - Drive API 调用结果（存储空间大小、TB 单位）
   - 最终推断的 tier

2. inferGoogleOneTier:
   - 输入的存储空间（bytes 和 TB）
   - 匹配的存储层级和返回的 tier
   - 每个判断分支的详细信息

**调试信息包含：**
- LoadCodeAssist vs Drive API 的使用情况
- 存储空间 bytes → TB 转换
- tier 推断的完整过程
- 每个存储层级的阈值检查

用户可以重新授权 Google One 账户，后端日志将显示详细的 tier 判断过程。
---
 .../internal/service/gemini_oauth_service.go  | 32 ++++++++++++++++---
 1 file changed, 28 insertions(+), 4 deletions(-)

diff --git a/backend/internal/service/gemini_oauth_service.go b/backend/internal/service/gemini_oauth_service.go
index 79d93a91..fdf7bee6 100644
--- a/backend/internal/service/gemini_oauth_service.go
+++ b/backend/internal/service/gemini_oauth_service.go
@@ -229,57 +229,81 @@ func extractTierIDFromAllowedTiers(allowedTiers []geminicli.AllowedTier) string
 
 // inferGoogleOneTier infers Google One tier from Drive storage limit
 func inferGoogleOneTier(storageBytes int64) string {
+	log.Printf("[GeminiOAuth] inferGoogleOneTier - input: %d bytes (%.2f TB)", storageBytes, float64(storageBytes)/float64(TB))
+
 	if storageBytes <= 0 {
+		log.Printf("[GeminiOAuth] inferGoogleOneTier - storageBytes <= 0, returning UNKNOWN")
 		return TierGoogleOneUnknown
 	}
 
 	if storageBytes > StorageTierUnlimited {
+		log.Printf("[GeminiOAuth] inferGoogleOneTier - > %d bytes (100TB), returning UNLIMITED", StorageTierUnlimited)
 		return TierGoogleOneUnlimited
 	}
 	if storageBytes >= StorageTierAIPremium {
+		log.Printf("[GeminiOAuth] inferGoogleOneTier - >= %d bytes (2TB), returning AI_PREMIUM", StorageTierAIPremium)
 		return TierAIPremium
 	}
 	if storageBytes >= StorageTierStandard {
+		log.Printf("[GeminiOAuth] inferGoogleOneTier - >= %d bytes (200GB), returning STANDARD", StorageTierStandard)
 		return TierGoogleOneStandard
 	}
 	if storageBytes >= StorageTierBasic {
+		log.Printf("[GeminiOAuth] inferGoogleOneTier - >= %d bytes (100GB), returning BASIC", StorageTierBasic)
 		return TierGoogleOneBasic
 	}
 	if storageBytes >= StorageTierFree {
+		log.Printf("[GeminiOAuth] inferGoogleOneTier - >= %d bytes (15GB), returning FREE", StorageTierFree)
 		return TierFree
 	}
+
+	log.Printf("[GeminiOAuth] inferGoogleOneTier - < %d bytes (15GB), returning UNKNOWN", StorageTierFree)
 	return TierGoogleOneUnknown
 }
 
 // fetchGoogleOneTier fetches Google One tier from Drive API or LoadCodeAssist API
 func (s *GeminiOAuthService) FetchGoogleOneTier(ctx context.Context, accessToken, proxyURL string) (string, *geminicli.DriveStorageInfo, error) {
+	log.Printf("[GeminiOAuth] Starting FetchGoogleOneTier")
+
 	// First try LoadCodeAssist API (works for accounts with GCP projects)
 	if s.codeAssist != nil {
+		log.Printf("[GeminiOAuth] Trying LoadCodeAssist API...")
 		loadResp, err := s.codeAssist.LoadCodeAssist(ctx, accessToken, proxyURL, nil)
-		if err == nil && loadResp != nil {
+		if err != nil {
+			log.Printf("[GeminiOAuth] LoadCodeAssist failed: %v", err)
+		} else if loadResp != nil {
 			if tier := loadResp.GetTier(); tier != "" {
-				fmt.Printf("[GeminiOAuth] Got tier from LoadCodeAssist: %s\n", tier)
+				log.Printf("[GeminiOAuth] Got tier from LoadCodeAssist: %s (skipping Drive API)", tier)
 				return tier, nil, nil
+			} else {
+				log.Printf("[GeminiOAuth] LoadCodeAssist returned no tier, falling back to Drive API")
 			}
 		}
 	}
 
 	// Fallback to Drive API (requires drive.readonly scope)
+	log.Printf("[GeminiOAuth] Calling Drive API for storage quota...")
 	driveClient := geminicli.NewDriveClient()
 
 	storageInfo, err := driveClient.GetStorageQuota(ctx, accessToken, proxyURL)
 	if err != nil {
 		// Check if it's a 403 (scope not granted)
 		if strings.Contains(err.Error(), "status 403") {
-			fmt.Printf("[GeminiOAuth] Drive API scope not available: %v\n", err)
+			log.Printf("[GeminiOAuth] Drive API scope not available (403): %v", err)
 			return TierGoogleOneUnknown, nil, err
 		}
 		// Other errors
-		fmt.Printf("[GeminiOAuth] Failed to fetch Drive storage: %v\n", err)
+		log.Printf("[GeminiOAuth] Failed to fetch Drive storage: %v", err)
 		return TierGoogleOneUnknown, nil, err
 	}
 
+	log.Printf("[GeminiOAuth] Drive API response - Limit: %d bytes (%.2f TB), Usage: %d bytes (%.2f GB)",
+		storageInfo.Limit, float64(storageInfo.Limit)/float64(TB),
+		storageInfo.Usage, float64(storageInfo.Usage)/float64(GB))
+
 	tierID := inferGoogleOneTier(storageInfo.Limit)
+	log.Printf("[GeminiOAuth] Inferred tier from storage: %s", tierID)
+
 	return tierID, storageInfo, nil
 }