Delete login_client.py

login_client.py

@@ -1,375 +0,0 @@
-"""
-登录客户端模块
-处理Warp登录相关功能，包括获取refresh token
-"""
-import re
-import requests
-import urllib.parse
-from typing import Optional, Dict, List, Tuple
-from concurrent.futures import ThreadPoolExecutor
-from loguru import logger
-
-from config import Config
-from protobuf_manager import ProtobufManager
-
-
-class LoginClient:
-    """Warp登录客户端"""
-    
-    def __init__(self):
-        self.session = requests.Session()
-        
-    def create_protobuf_data(self, login_url: str) -> Optional[bytes]:
-        """创建登录请求的protobuf数据"""
-        return ProtobufManager.create_login_request(login_url)
-
-    def get_oob_code(self, login_url: str) -> Optional[str]:
-        """获取oob_code"""
-        logger.debug(f"🔄 开始获取oob_code，URL: {login_url[:50]}...")
-        
-        protobuf_data = self.create_protobuf_data(login_url)
-        if not protobuf_data:
-            return None
-            
-        headers = {
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/137.0.0.0 Safari/537.36",
-            "Connection": "keep-alive",
-            "Accept": "*/*",
-            "Accept-Encoding": "gzip, deflate, br, zstd",
-            "Content-Type": "application/octet-stream",
-            "x-goog-api-key": "AIzaSyA2KlwBX3mkFo30om9LUFYQhpqLoa_BNhE",
-            "content-type": "application/x-protobuf",
-            "x-client-data": "CK+1yQEIjLbJAQimtskBCKmdygEI74zLAQiVocsBCJKjywEIhaDNAQjY7c4BCOLwzgEIpfLOAQi88s4BCJP2zgEYnPPOAQ==",
-            "sec-fetch-site": "none",
-            "sec-fetch-mode": "no-cors",
-            "sec-fetch-dest": "empty",
-            "accept-language": "zh-CN,zh;q=0.9",
-            "priority": "u=4, i"
-        }
-        
-        try:
-            response = self.session.post(
-                Config.OPTIMIZATION_GUIDE_URL,
-                headers=headers,
-                data=protobuf_data,
-                timeout=60
-            )
-            
-            if response.status_code == 200:
-                logger.info(f"🔄 响应内容: {response.text}")
-                
-                # 使用ProtobufManager解析响应
-                decoded_url = ProtobufManager.parse_login_response(response.content)
-                logger.info(f"🔄 解码后的URL: {decoded_url}")
-                if decoded_url == "":
-                    logger.error("❌ 解码后的URL为空")
-                    decoded_url = response.text
-                
-                # 提取oob_code
-                pattern = r'oobCode=([^&]+)'
-                match = re.search(pattern, decoded_url)
-                if match:
-                    oob_code = match.group(1)
-                    logger.success(f"✅ 获取到oob_code: {oob_code[:20]}...")
-                    return oob_code
-                else:
-                    logger.error("❌ 未能从响应中提取oob_code")
-                    return None
-            else:
-                logger.error(f"❌ 请求失败: {response.status_code}")
-                return None
-                
-        except Exception as e:
-            logger.error(f"❌ 获取oob_code时出错: {e}")
-            return None
-
-    def verify_id_token(self, id_token: str) -> bool:
-        """验证idToken并使refresh_token生效"""
-        logger.debug(f"🔍 开始验证idToken: {id_token[:20]}...")
-        
-        headers = {
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36',
-            'Accept-Encoding': 'gzip, deflate, br, zstd',
-            'Content-Type': 'application/json',
-            'sec-ch-ua-platform': '"Windows"',
-            'x-client-version': 'Chrome/JsCore/11.3.1/FirebaseCore-web',
-            'sec-ch-ua': '"Not)A;Brand";v="8", "Chromium";v="138", "Google Chrome";v="138"',
-            'sec-ch-ua-mobile': '?0',
-            'x-firebase-gmpid': '1:13153726198:web:1cc16bca7287752f8e06d7',
-            'origin': 'https://app.warp.dev',
-            'x-browser-channel': 'stable',
-            'x-browser-year': '2025',
-            'x-browser-validation': '6h3XF8YcD8syi2FF2BbuE2KllQo=',
-            'x-browser-copyright': 'Copyright 2025 Google LLC. All rights reserved.',
-            'x-client-data': 'CK+1yQEIjLbJAQimtskBCKmdygEI74zLAQiWocsBCJKjywEIhaDNAQji8M4BCJP2zgE=',
-            'sec-fetch-site': 'cross-site',
-            'sec-fetch-mode': 'cors',
-            'sec-fetch-dest': 'empty',
-            'accept-language': 'zh-CN,zh;q=0.9',
-            'priority': 'u=1, i'
-        }
-        
-        payload = {
-            'idToken': id_token
-        }
-        
-        try:
-            response = self.session.post(
-                'https://identitytoolkit.googleapis.com/v1/accounts:lookup?key=AIzaSyBdy3O3S9hrdayLJxJ7mriBR4qgUaUygAs',
-                headers=headers,
-                json=payload,
-                timeout=30
-            )
-            
-            if response.status_code == 200:
-                logger.success("✅ idToken验证成功，refresh_token已生效")
-                return True
-            else:
-                logger.error(f"❌ idToken验证失败，状态码: {response.status_code}")
-                logger.debug(f"❌ 响应内容: {response.text}")
-                return False
-                
-        except Exception as e:
-            logger.error(f"❌ 验证idToken时出错: {e}")
-            return False
-
-    def create_user_with_access_token(self, access_token: str) -> bool:
-        """使用access_token创建用户"""
-        logger.info(f"👤 开始创建用户，access_token: {access_token[:20]}...")
-        
-        headers = {
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36',
-            'Accept-Encoding': 'gzip, deflate, br, zstd',
-            'Content-Type': 'application/json',
-            'sec-ch-ua-platform': '"Windows"',
-            'authorization': f'Bearer {access_token}',
-            'sec-ch-ua': '"Not)A;Brand";v="8", "Chromium";v="138", "Google Chrome";v="138"',
-            'sec-ch-ua-mobile': '?0',
-            'origin': 'https://app.warp.dev',
-            'sec-fetch-site': 'same-origin',
-            'sec-fetch-mode': 'cors',
-            'sec-fetch-dest': 'empty',
-            'accept-language': 'zh-CN,zh;q=0.9',
-            'priority': 'u=1, i',
-            'Cookie': 'rl_anonymous_id=8b178568-76c6-4e9f-8c72-461bd1909c1c'
-        }
-        
-        payload = {
-            "operationName": "GetOrCreateUser",
-            "variables": {
-                "input": {},
-                "requestContext": {
-                    "osContext": {},
-                    "clientContext": {}
-                }
-            },
-            "query": "mutation GetOrCreateUser($input: GetOrCreateUserInput!, $requestContext: RequestContext!) {\n  getOrCreateUser(requestContext: $requestContext, input: $input) {\n    ... on GetOrCreateUserOutput {\n      uid\n      isOnboarded\n      anonymousUserInfo {\n        anonymousUserType\n        linkedAt\n        __typename\n      }\n      workspaces {\n        joinableTeams {\n          teamUid\n          numMembers\n          name\n          teamAcceptingInvites\n          __typename\n        }\n        __typename\n      }\n      deletedAnonymousUser\n      __typename\n    }\n    ... on UserFacingError {\n      error {\n        message\n        __typename\n      }\n      __typename\n    }\n    __typename\n  }\n}\n"
-        }
-        
-        try:
-            response = self.session.post(
-                'https://app.warp.dev/graphql/v2?op=GetOrCreateUser',
-                headers=headers,
-                json=payload,
-                timeout=30
-            )
-            
-            if response.status_code == 200:
-                response_data = response.json()
-                # 检查是否有错误
-                if 'errors' in response_data:
-                    logger.error(f"❌ 创建用户失败，GraphQL错误: {response_data['errors']}")
-                    return False
-                
-                # 检查响应数据
-                data = response_data.get('data', {})
-                get_or_create_user = data.get('getOrCreateUser', {})
-                
-                if 'uid' in get_or_create_user:
-                    logger.success(f"✅ 用户创建成功，UID: {get_or_create_user['uid']}")
-                    return True
-                elif 'error' in get_or_create_user:
-                    logger.error(f"❌ 创建用户失败: {get_or_create_user['error']}")
-                    return False
-                else:
-                    logger.success("✅ 用户创建请求已发送")
-                    return True
-                    
-            else:
-                logger.error(f"❌ 创建用户失败，状态码: {response.status_code}")
-                logger.debug(f"❌ 响应内容: {response.text}")
-                return False
-                
-        except Exception as e:
-            logger.error(f"❌ 创建用户时出错: {e}")
-            return False
-
-    def get_refresh_token(self, email: str, oob_code: str) -> Optional[str]:
-        """使用email和oob_code获取refresh token，并验证idToken"""
-        logger.info(f"🔄 开始获取refresh token for {email}")
-        
-        headers = {
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/137.0.0.0 Safari/537.36',
-            'Accept-Encoding': 'gzip, deflate, br, zstd',
-            'Content-Type': 'application/json',
-            'sec-ch-ua-platform': 'Windows',
-            'x-client-version': 'Chrome/JsCore/11.3.1/FirebaseCore-web',
-            'sec-ch-ua': '"Google Chrome";v="137", "Chromium";v="137", "Not/A)Brand";v="24"',
-            'sec-ch-ua-mobile': '?0',
-            'x-firebase-gmpid': '1:13153726198:web:1cc16bca7287752f8e06d7',
-            'origin': 'https://app.warp.dev',
-            'x-client-data': 'CK+1yQEIjLbJAQimtskBCKmdygEI74zLAQiVocsBCJKjywEIhaDNAQji8M0BCKXyzgEIk/bOAQ==',
-            'sec-fetch-site': 'cross-site',
-            'sec-fetch-mode': 'cors',
-            'sec-fetch-dest': 'empty',
-            'accept-language': 'zh-CN,zh;q=0.9',
-            'priority': 'u=1, i'
-        }
-
-        payload = {
-            'email': email,
-            'oobCode': oob_code
-        }
-
-        try:
-            response = self.session.post(Config.FIREBASE_LOGIN_URL, headers=headers, json=payload, timeout=30)
-            
-            if response.status_code == 200:
-                response_data = response.json()
-                refresh_token = response_data.get("refreshToken")
-                id_token = response_data.get("idToken")
-                
-                if not refresh_token:
-                    logger.error(f"❌ 响应中未找到refreshToken字段 for {email}")
-                    return None
-                
-                if not id_token:
-                    logger.error(f"❌ 响应中未找到idToken字段 for {email}")
-                    return None
-                
-                logger.success(f"✅ 获取到refresh token和idToken for {email}")
-                
-                # 验证idToken使refresh_token生效
-                if self.verify_id_token(id_token):
-                    logger.success(f"🎉 refresh token验证成功并已生效 for {email}")
-                    return refresh_token
-                else:
-                    logger.error(f"❌ refresh token验证失败，设置为无效 for {email}")
-                    return None
-                    
-            else:
-                logger.error(f"❌ 获取refresh token失败 for {email}: {response.status_code}")
-                logger.debug(f"❌ 响应内容: {response.text}")
-                return None
-                
-        except Exception as e:
-            logger.error(f"❌ 获取refresh token时出错 for {email}: {e}")
-            return None
-
-    def process_single_email_with_user_creation(self, email: str, login_url: str, token_manager=None) -> Tuple[str, Optional[str], str]:
-        """处理单个email，返回(email, refresh_token, status)，并在获取refresh_token后创建用户"""
-        logger.info(f"🚀 开始处理邮箱: {email}")
-        
-        try:
-            # 获取oob_code
-            oob_code = self.get_oob_code(login_url)
-            if not oob_code:
-                return email, None, "获取oob_code失败"
-            
-            # 获取refresh token并验证
-            refresh_token = self.get_refresh_token(email, oob_code)
-            if refresh_token:
-                logger.success(f"✅ 邮箱 {email} refresh_token获取成功")
-                
-                # 如果有token_manager，立即获取access_token并创建用户
-                if token_manager:
-                    logger.info(f"🔄 正在为 {email} 获取access_token并创建用户...")
-                    access_token = token_manager.get_access_token(refresh_token)
-                    if access_token:
-                        logger.info(f"✅ 为 {email} 获取到access_token")
-                        if self.create_user_with_access_token(access_token):
-                            logger.success(f"🎉 为 {email} 创建用户成功")
-                            return email, refresh_token, "成功并已创建用户"
-                        else:
-                            logger.warning(f"⚠️ 为 {email} 创建用户失败，但refresh_token有效")
-                            return email, refresh_token, "成功，但创建用户失败"
-                    else:
-                        logger.warning(f"⚠️ 为 {email} 获取access_token失败")
-                        return email, refresh_token, "成功，但获取access_token失败"
-                else:
-                    return email, refresh_token, "成功"
-            else:
-                logger.error(f"❌ 邮箱 {email} 处理失败")
-                return email, None, "获取refresh_token失败或验证失败"
-                
-        except Exception as e:
-            logger.error(f"❌ 处理邮箱 {email} 时异常: {e}")
-            return email, None, f"处理异常: {str(e)}"
-
-    def process_single_email(self, email: str, login_url: str) -> Tuple[str, Optional[str], str]:
-        """处理单个email，返回(email, refresh_token, status)"""
-        logger.info(f"🚀 开始处理邮箱: {email}")
-        
-        try:
-            # 获取oob_code
-            oob_code = self.get_oob_code(login_url)
-            if not oob_code:
-                return email, None, "获取oob_code失败"
-            
-            # 获取refresh token并验证
-            refresh_token = self.get_refresh_token(email, oob_code)
-            if refresh_token:
-                logger.success(f"✅ 邮箱 {email} 处理成功")
-                return email, refresh_token, "成功"
-            else:
-                logger.error(f"❌ 邮箱 {email} 处理失败")
-                return email, None, "获取refresh_token失败或验证失败"
-                
-        except Exception as e:
-            logger.error(f"❌ 处理邮箱 {email} 时异常: {e}")
-            return email, None, f"处理异常: {str(e)}"
-
-    def batch_process_emails(self, email_url_dict: Dict[str, str], max_workers: int = 5, token_manager=None) -> Dict[str, Dict]:
-        """批量处理emails，返回结果字典"""
-        logger.info(f"🚀 开始批量处理 {len(email_url_dict)} 个邮箱，并发数: {max_workers}")
-        results = {}
-        
-        with ThreadPoolExecutor(max_workers=max_workers) as executor:
-            # 提交所有任务
-            if token_manager:
-                future_to_email = {
-                    executor.submit(self.process_single_email_with_user_creation, email, url, token_manager): email 
-                    for email, url in email_url_dict.items()
-                }
-            else:
-                future_to_email = {
-                    executor.submit(self.process_single_email, email, url): email 
-                    for email, url in email_url_dict.items()
-                }
-            
-            # 收集结果
-            for future in future_to_email:
-                email = future_to_email[future]
-                try:
-                    email, refresh_token, status = future.result()
-                    results[email] = {
-                        'refresh_token': refresh_token,
-                        'status': status
-                    }
-                    
-                    if refresh_token:
-                        logger.success(f"✅ {email}: {status}")
-                    else:
-                        logger.error(f"❌ {email}: {status}")
-                        
-                except Exception as e:
-                    logger.error(f"❌ {email}: 处理异常: {str(e)}")
-                    results[email] = {
-                        'refresh_token': None,
-                        'status': f"处理异常: {str(e)}"
-                    }
-        
-        success_count = sum(1 for result in results.values() if result['refresh_token'])
-        logger.info(f"📊 批量处理完成: {success_count}/{len(email_url_dict)} 成功")
-        
-        return results