添加项目源代码

services/cursor_service.py

@@ -0,0 +1,237 @@
+"""Cursor 服务管理类"""
+
+import os
+import json
+import hashlib
+from datetime import datetime, timedelta
+from typing import Dict, Optional, Tuple
+import urllib3
+import requests
+
+from cursor_token_refresher import CursorTokenRefresher
+from machine_resetter import MachineResetter
+from update_disabler import UpdateDisabler
+from logger import logger
+from common_utils import (
+    get_hardware_id, 
+    get_user_state,
+    check_user_state,
+    refresh_user_state,
+    update_user_state,
+    activate_device
+)
+
+class CursorService:
+    """Cursor 服务管理类"""
+    _instance = None
+    _initialized = False
+    _last_check_time = None
+    
+    def __new__(cls):
+        if cls._instance is None:
+            cls._instance = super().__new__(cls)
+        return cls._instance
+    
+    def __init__(self):
+        if not self._initialized:
+            # 基础组件初始化
+            self.token_refresher = CursorTokenRefresher()
+            self.machine_resetter = MachineResetter()
+            self.update_disabler = UpdateDisabler()
+            self.logger = logger.get_logger("CursorService")
+            
+            # 配置文件路径
+            self.config_dir = os.path.join(os.getenv('APPDATA'), 'TingquanAssistant')
+            self.config_file = os.path.join(self.config_dir, 'config.json')
+            
+            # 确保配置目录存在
+            os.makedirs(self.config_dir, exist_ok=True)
+            
+            # 初始化配置
+            self._init_config()
+            
+            # 首次初始化用户状态
+            self._lazy_init_user_state()
+            
+            self._initialized = True
+    
+    def _init_config(self):
+        """初始化配置文件"""
+        # 默认配置
+        self.default_config = {
+            "version": "4.0.0.1",
+            "last_check_update": None,
+            "auto_check_update": True,
+            "theme": "light"
+        }
+        
+        # 读取或创建配置文件
+        if not os.path.exists(self.config_file):
+            with open(self.config_file, 'w', encoding='utf-8') as f:
+                json.dump(self.default_config, f, indent=2, ensure_ascii=False)
+            self.config = self.default_config
+        else:
+            try:
+                with open(self.config_file, 'r', encoding='utf-8') as f:
+                    self.config = json.load(f)
+            except Exception as e:
+                self.logger.error(f"读取配置文件失败: {e}")
+                self.config = self.default_config
+    
+    def _lazy_init_user_state(self):
+        """
+        懒加载方式初始化用户状态
+        只在首次使用时初始化，避免频繁刷新
+        """
+        try:
+            if not self._last_check_time:
+                machine_id = get_hardware_id()
+                success, msg, state_data = refresh_user_state(machine_id)
+                if success:
+                    self._last_check_time = datetime.now()
+                    self.logger.info("用户状态初始化成功")
+                else:
+                    self.logger.warning(f"用户状态初始化失败: {msg}")
+        except Exception as e:
+            self.logger.error(f"初始化用户状态异常: {e}")
+    
+    def _should_refresh_state(self) -> bool:
+        """
+        检查是否需要刷新状态
+        每5分钟检测一次设备状态
+        
+        Returns:
+            bool: 是否需要刷新状态
+        """
+        try:
+            # 如果没有上次检查时间，需要刷新
+            if not self._last_check_time:
+                return True
+                
+            # 计算距离上次检查的时间
+            now = datetime.now()
+            time_diff = now - self._last_check_time
+            
+            # 如果超过5分钟，需要刷新
+            if time_diff.total_seconds() >= 300:  # 5分钟 = 300秒
+                self.logger.debug("距离上次状态检查已超过5分钟")
+                return True
+                
+            return False
+            
+        except Exception as e:
+            self.logger.error(f"检查状态刷新时间异常: {e}")
+            # 发生异常时返回False，避免频繁刷新
+            return False
+    
+    def check_activation_status(self) -> Dict:
+        """
+        检查当前激活状态
+        如果需要刷新则自动刷新状态
+        
+        Returns:
+            Dict: 状态信息字典
+        """
+        try:
+            # 检查是否需要刷新状态
+            if self._should_refresh_state():
+                machine_id = get_hardware_id()
+                success, msg, state_data = refresh_user_state(machine_id)
+                if success:
+                    self._last_check_time = datetime.now()
+                    self.logger.info("设备状态检查完成")
+                else:
+                    self.logger.warning(f"设备状态检查失败: {msg}")
+        except Exception as e:
+            self.logger.error(f"设备状态检查异常: {e}")
+            
+        # 无论是否刷新成功，都返回当前状态
+        return get_user_state()
+    
+    def activate_with_code(self, activation_code: str) -> Tuple[bool, str]:
+        """
+        使用激活码激活
+        
+        Args:
+            activation_code: 激活码
+            
+        Returns:
+            Tuple[bool, str]: (是否成功, 消息)
+        """
+        try:
+            self.logger.info(f"开始处理激活码: {activation_code}")
+            
+            # 获取机器ID
+            machine_id = get_hardware_id()
+            if not machine_id:
+                self.logger.error("获取机器ID失败")
+                return False, "获取机器ID失败"
+            
+            self.logger.info(f"获取到机器ID: {machine_id}")
+            
+            # 调用激活接口
+            self.logger.info("正在调用激活接口...")
+            success, msg, data = activate_device(machine_id, activation_code)
+            
+            if success:
+                self.logger.info(f"激活成功: {msg}")
+            else:
+                self.logger.warning(f"激活失败: {msg}")
+                
+            return success, msg
+            
+        except Exception as e:
+            error_msg = f"激活失败: {str(e)}"
+            self.logger.error(error_msg)
+            return False, error_msg
+    
+    def refresh_account(self, machine_id: str) -> Tuple[bool, str, Optional[Dict]]:
+        """刷新账号"""
+        # 检查是否已激活
+        if not check_user_state():
+            return False, "请先激活软件", None
+        return self.token_refresher.refresh_token(machine_id)
+    
+    def reset_machine(self) -> Tuple[bool, str, Dict]:
+        """重置机器码"""
+        # 检查是否已激活
+        if not check_user_state():
+            return False, "请先激活软件", {}
+            
+        try:
+            new_ids, new_guid = self.machine_resetter.run()
+            return True, "重置成功", {
+                "new_ids": new_ids,
+                "new_guid": new_guid
+            }
+        except Exception as e:
+            return False, f"重置失败: {str(e)}", {}
+    
+    def disable_update(self) -> Tuple[bool, str]:
+        """禁用更新"""
+        # 检查是否已激活
+        if not check_user_state():
+            return False, "请先激活软件"
+            
+        success = self.update_disabler.disable()
+        return success, "禁用成功" if success else "禁用失败"
+    
+    def save_config(self, new_config: Dict) -> bool:
+        """
+        保存配置
+        :param new_config: 新的配置信息
+        :return: 是否成功
+        """
+        try:
+            self.config.update(new_config)
+            with open(self.config_file, 'w', encoding='utf-8') as f:
+                json.dump(self.config, f, indent=2, ensure_ascii=False)
+            return True
+        except Exception as e:
+            self.logger.error(f"保存配置失败: {e}")
+            return False
+    
+    def get_config(self) -> Dict:
+        """获取当前配置"""
+        return self.config.copy()
+