rcoder 1.0.0__py3-none-any.whl

rcoder/server_installer.py

@@ -0,0 +1,373 @@
+#!/usr/bin/env python3
+"""
+Rcoder 服务器安装工具
+用于自动安装和配置中转服务器和目标服务器
+"""
+
+import os
+import time
+from typing import Optional, Dict, Any
+
+class ServerInstaller:
+    """
+    服务器安装器类
+    处理远程服务器的自动安装和配置
+    """
+    
+    def __init__(self, remote_host):
+        """
+        初始化服务器安装器
+        
+        Args:
+            remote_host: 远程主机实例
+        """
+        self.remote = remote_host
+        self.installation_steps = []
+        self.verbose = True
+    
+    def install_rcoder_server(self, server_type: str = 'target', 
+                             use_venv: bool = True, 
+                             venv_path: str = '~/.rcoder/venv') -> Dict[str, Any]:
+        """
+        在远程服务器上安装Rcoder服务器端
+        
+        Args:
+            server_type: 服务器类型 ('target' 或 'proxy')
+            use_venv: 是否使用虚拟环境
+            venv_path: 虚拟环境路径
+        
+        Returns:
+            安装结果
+        """
+        print(f"=== 安装 {server_type} 服务器 ===")
+        
+        results = {
+            'success': False,
+            'steps': [],
+            'errors': [],
+            'status': 'pending'
+        }
+        
+        try:
+            # 步骤1: 系统更新
+            self._log(f"1. 更新系统包")
+            if self._update_system():
+                results['steps'].append('系统更新成功')
+            else:
+                results['errors'].append('系统更新失败')
+            
+            # 步骤2: 安装必要依赖
+            self._log(f"2. 安装必要依赖")
+            if self._install_dependencies():
+                results['steps'].append('依赖安装成功')
+            else:
+                results['errors'].append('依赖安装失败')
+            
+            # 步骤3: 安装Python
+            self._log(f"3. 安装Python")
+            if self._install_python():
+                results['steps'].append('Python安装成功')
+            else:
+                results['errors'].append('Python安装失败')
+            
+            # 步骤4: 安装和配置虚拟环境
+            if use_venv:
+                self._log(f"4. 安装和配置虚拟环境")
+                if self._setup_venv(venv_path):
+                    results['steps'].append('虚拟环境配置成功')
+                else:
+                    results['errors'].append('虚拟环境配置失败')
+            else:
+                results['steps'].append('跳过虚拟环境配置')
+            
+            # 步骤5: 安装Rcoder服务器端
+            self._log(f"5. 安装Rcoder服务器端")
+            if self._install_rcoder_server(use_venv, venv_path):
+                results['steps'].append('Rcoder服务器安装成功')
+            else:
+                results['errors'].append('Rcoder服务器安装失败')
+            
+            # 步骤6: 配置服务器
+            self._log(f"6. 配置服务器")
+            config_result = self._configure_server(server_type, use_venv, venv_path)
+            if config_result:
+                results['steps'].append('服务器配置成功')
+                results['config'] = config_result
+            else:
+                results['errors'].append('服务器配置失败')
+            
+            # 步骤7: 启动服务
+            self._log(f"7. 启动服务")
+            if self._start_services(use_venv, venv_path):
+                results['steps'].append('服务启动成功')
+            else:
+                results['errors'].append('服务启动失败')
+            
+            # 步骤8: 验证安装
+            self._log(f"8. 验证安装")
+            if self._verify_installation():
+                results['steps'].append('安装验证成功')
+                results['success'] = True
+                results['status'] = 'completed'
+            else:
+                results['errors'].append('安装验证失败')
+                results['status'] = 'failed'
+            
+        except Exception as e:
+            error_msg = f"安装过程出错: {str(e)}"
+            self._log(error_msg)
+            results['errors'].append(error_msg)
+            results['status'] = 'failed'
+        
+        print()
+        if results['success']:
+            print(f"✅ {server_type} 服务器安装成功！")
+        else:
+            print(f"❌ {server_type} 服务器安装失败！")
+            print(f"错误: {results['errors']}")
+        
+        return results
+    
+    def _log(self, message: str):
+        """
+        记录日志
+        
+        Args:
+            message: 日志消息
+        """
+        if self.verbose:
+            print(f"   {message}")
+    
+    def _update_system(self) -> bool:
+        """
+        更新系统包
+        """
+        try:
+            # 检测系统类型
+            os_info = self.remote.run('cat /etc/os-release | grep -E "^ID=" | cut -d= -f2 | tr -d \"\'"')
+            os_info = os_info.strip().lower()
+            
+            if 'ubuntu' in os_info or 'debian' in os_info:
+                # Ubuntu/Debian
+                self.remote.run('sudo apt update -y')
+                self.remote.run('sudo apt upgrade -y')
+            elif 'centos' in os_info or 'rhel' in os_info or 'rocky' in os_info:
+                # CentOS/RHEL/Rocky
+                self.remote.run('sudo yum update -y')
+            elif 'alpine' in os_info:
+                # Alpine
+                self.remote.run('sudo apk update')
+                self.remote.run('sudo apk upgrade')
+            else:
+                # 尝试通用方法
+                self._log("未知系统类型，跳过系统更新")
+            
+            return True
+        except Exception as e:
+            self._log(f"系统更新出错: {e}")
+            return False
+    
+    def _install_dependencies(self) -> bool:
+        """
+        安装必要依赖
+        """
+        try:
+            # 检测系统类型
+            os_info = self.remote.run('cat /etc/os-release | grep -E "^ID=" | cut -d= -f2 | tr -d \"\'"')
+            os_info = os_info.strip().lower()
+            
+            if 'ubuntu' in os_info or 'debian' in os_info:
+                # Ubuntu/Debian
+                self.remote.run('sudo apt install -y build-essential curl wget git openssl libssl-dev')
+            elif 'centos' in os_info or 'rhel' in os_info or 'rocky' in os_info:
+                # CentOS/RHEL/Rocky
+                self.remote.run('sudo yum install -y gcc gcc-c++ curl wget git openssl openssl-devel')
+            elif 'alpine' in os_info:
+                # Alpine
+                self.remote.run('sudo apk add build-base curl wget git openssl openssl-dev')
+            
+            return True
+        except Exception as e:
+            self._log(f"依赖安装出错: {e}")
+            return False
+    
+    def _install_python(self) -> bool:
+        """
+        安装Python
+        """
+        try:
+            # 检查Python是否已安装
+            python_version = self.remote.run('python3 --version 2>/dev/null || python --version 2>/dev/null')
+            
+            if 'Python 3.' in python_version:
+                self._log(f"Python已安装: {python_version}")
+                return True
+            
+            # 检测系统类型并安装Python
+            os_info = self.remote.run('cat /etc/os-release | grep -E "^ID=" | cut -d= -f2 | tr -d \"\'"')
+            os_info = os_info.strip().lower()
+            
+            if 'ubuntu' in os_info or 'debian' in os_info:
+                # Ubuntu/Debian
+                self.remote.run('sudo apt install -y python3 python3-pip python3-venv')
+            elif 'centos' in os_info or 'rhel' in os_info or 'rocky' in os_info:
+                # CentOS/RHEL/Rocky
+                self.remote.run('sudo yum install -y python3 python3-pip python3-venv')
+            elif 'alpine' in os_info:
+                # Alpine
+                self.remote.run('sudo apk add python3 py3-pip python3-venv')
+            
+            # 验证安装
+            new_version = self.remote.run('python3 --version')
+            return 'Python 3.' in new_version
+        except Exception as e:
+            self._log(f"Python安装出错: {e}")
+            return False
+    
+    def _setup_venv(self, venv_path: str) -> bool:
+        """
+        安装和配置虚拟环境
+        """
+        try:
+            # 扩展路径
+            expanded_path = self.remote.run(f'echo {venv_path}')
+            expanded_path = expanded_path.strip()
+            
+            # 创建虚拟环境目录
+            self.remote.run(f'mkdir -p $(dirname {expanded_path})')
+            
+            # 删除现有虚拟环境
+            self.remote.run(f'rm -rf {expanded_path}')
+            
+            # 创建虚拟环境
+            self.remote.run(f'python3 -m venv {expanded_path}')
+            
+            # 激活虚拟环境并升级pip
+            activate_cmd = f'source {expanded_path}/bin/activate && pip install --upgrade pip'
+            self.remote.run(activate_cmd)
+            
+            # 验证虚拟环境
+            venv_python = self.remote.run(f'{expanded_path}/bin/python --version')
+            return 'Python 3.' in venv_python
+        except Exception as e:
+            self._log(f"虚拟环境配置出错: {e}")
+            return False
+    
+    def _install_rcoder_server(self, use_venv: bool, venv_path: str) -> bool:
+        """
+        安装Rcoder服务器端
+        """
+        try:
+            # 克隆Rcoder仓库
+            self.remote.run('git clone https://github.com/trae-ai/rcoder-server ~/.rcoder/server')
+            
+            # 安装依赖
+            if use_venv:
+                expanded_path = self.remote.run(f'echo {venv_path}')
+                expanded_path = expanded_path.strip()
+                install_cmd = f'source {expanded_path}/bin/activate && pip install -r ~/.rcoder/server/requirements.txt'
+            else:
+                install_cmd = 'pip install -r ~/.rcoder/server/requirements.txt'
+            
+            self.remote.run(install_cmd)
+            
+            return True
+        except Exception as e:
+            self._log(f"Rcoder服务器安装出错: {e}")
+            return False
+    
+    def _configure_server(self, server_type: str, use_venv: bool, venv_path: str) -> Dict[str, Any]:
+        """
+        配置服务器
+        """
+        try:
+            # 创建配置文件
+            config = {
+                'server_type': server_type,
+                'port': 443,
+                'use_https': True,
+                'enable_compression': True,
+                'enable_connection_pool': True
+            }
+            
+            # 写入配置文件
+            config_json = str(config).replace("'", '"')
+            self.remote.run(f'echo "{config_json}" > ~/.rcoder/server/config.json')
+            
+            # 创建启动脚本
+            if use_venv:
+                expanded_path = self.remote.run(f'echo {venv_path}')
+                expanded_path = expanded_path.strip()
+                start_script = f"#!/bin/bash\nsource {expanded_path}/bin/activate\npython ~/.rcoder/server/server.py"
+            else:
+                start_script = f"#!/bin/bash\npython ~/.rcoder/server/server.py"
+            
+            self.remote.run(f'echo "{start_script}" > ~/.rcoder/server/start.sh')
+            self.remote.run('chmod +x ~/.rcoder/server/start.sh')
+            
+            # 创建systemd服务文件
+            service_file = f"[Unit]\nDescription=Rcoder Server\nAfter=network.target\n\n[Service]\nType=simple\nUser=$(whoami)\nExecStart=~/.rcoder/server/start.sh\nRestart=always\n\n[Install]\nWantedBy=multi-user.target"
+            self.remote.run(f'echo "{service_file}" | sudo tee /etc/systemd/system/rcoder.service')
+            
+            return config
+        except Exception as e:
+            self._log(f"服务器配置出错: {e}")
+            return {}
+    
+    def _start_services(self, use_venv: bool, venv_path: str) -> bool:
+        """
+        启动服务
+        """
+        try:
+            # 重新加载systemd
+            self.remote.run('sudo systemctl daemon-reload')
+            
+            # 启用并启动服务
+            self.remote.run('sudo systemctl enable rcoder.service')
+            self.remote.run('sudo systemctl start rcoder.service')
+            
+            # 等待服务启动
+            time.sleep(5)
+            
+            # 检查服务状态
+            status = self.remote.run('sudo systemctl status rcoder.service | grep -E "Active:"')
+            return 'active (running)' in status
+        except Exception as e:
+            self._log(f"服务启动出错: {e}")
+            return False
+    
+    def _verify_installation(self) -> bool:
+        """
+        验证安装
+        """
+        try:
+            # 检查服务状态
+            status = self.remote.run('sudo systemctl status rcoder.service | grep -E "Active:"')
+            if 'active (running)' not in status:
+                return False
+            
+            # 测试连接
+            test_result = self.remote.run('curl -s http://localhost:443/health')
+            return 'ok' in test_result.lower()
+        except Exception as e:
+            self._log(f"安装验证出错: {e}")
+            return False
+
+
+def install_server(remote_host, server_type: str = 'target', 
+                  use_venv: bool = True, 
+                  venv_path: str = '~/.rcoder/venv') -> Dict[str, Any]:
+    """
+    安装服务器的便捷函数
+    
+    Args:
+        remote_host: 远程主机实例
+        server_type: 服务器类型
+        use_venv: 是否使用虚拟环境
+        venv_path: 虚拟环境路径
+    
+    Returns:
+        安装结果
+    """
+    installer = ServerInstaller(remote_host)
+    return installer.install_rcoder_server(server_type, use_venv, venv_path)

rcoder/utils.py

@@ -0,0 +1,213 @@
+#!/usr/bin/env python3
+"""
+Rcoder 工具模块
+提供简化使用的工具函数和辅助功能，降低使用门槛
+"""
+
+import os
+import json
+import platform
+from typing import Optional, Dict, Any
+
+class ConfigManager:
+    """配置管理类，简化配置过程"""
+    
+    def __init__(self, config_file: str = '~/.rcoder/config.json'):
+        """
+        初始化配置管理器
+        
+        Args:
+            config_file: 配置文件路径
+        """
+        self.config_file = os.path.expanduser(config_file)
+        self.config = self._load_config()
+    
+    def _load_config(self) -> Dict[str, Any]:
+        """加载配置文件"""
+        try:
+            if os.path.exists(self.config_file):
+                with open(self.config_file, 'r', encoding='utf-8') as f:
+                    return json.load(f)
+        except Exception as e:
+            print(f"警告: 加载配置文件失败: {e}")
+        return self._get_default_config()
+    
+    def _get_default_config(self) -> Dict[str, Any]:
+        """获取默认配置"""
+        return {
+            'servers': {
+                'local': {
+                    'host': '127.0.0.1',
+                    'port': 443,
+                    'use_https_disguise': True,
+                    'proxy_server': None
+                }
+            },
+            'default_server': 'local',
+            'timeout': 60,
+            'restart_max_wait': 60,
+            'monitoring_interval': 30
+        }
+    
+    def save_config(self):
+        """保存配置文件"""
+        try:
+            os.makedirs(os.path.dirname(self.config_file), exist_ok=True)
+            with open(self.config_file, 'w', encoding='utf-8') as f:
+                json.dump(self.config, f, indent=2, ensure_ascii=False)
+            print(f"✅ 配置已保存到: {self.config_file}")
+        except Exception as e:
+            print(f"❌ 保存配置文件失败: {e}")
+    
+    def add_server(self, name: str, host: str, port: int = 443, 
+                   use_https_disguise: bool = True, 
+                   proxy_server: Optional[tuple] = None):
+        """添加服务器配置"""
+        self.config['servers'][name] = {
+            'host': host,
+            'port': port,
+            'use_https_disguise': use_https_disguise,
+            'proxy_server': proxy_server
+        }
+        self.save_config()
+    
+    def get_server(self, name: Optional[str] = None) -> Dict[str, Any]:
+        """获取服务器配置"""
+        server_name = name or self.config.get('default_server', 'local')
+        return self.config['servers'].get(server_name, self.config['servers']['local'])
+
+
+def quick_setup():
+    """快速设置向导，引导用户完成初始配置"""
+    print("=== Rcoder 快速设置向导 ===")
+    print("帮助您快速配置 rcoder，降低使用门槛")
+    print()
+    
+    config = ConfigManager()
+    
+    # 询问是否添加新服务器
+    add_server = input("是否添加新服务器? (y/n): ").lower() == 'y'
+    
+    if add_server:
+        server_name = input("服务器名称 (例如: my-server): ")
+        server_host = input("服务器地址 (IP或域名): ")
+        server_port = input("端口号 (默认443): ")
+        server_port = int(server_port) if server_port else 443
+        use_https = input("启用 HTTPS 伪装? (y/n, 默认y): ").lower() != 'n'
+        
+        # 询问是否使用中转服务器
+        use_proxy = input("使用中转服务器? (y/n): ").lower() == 'y'
+        proxy_server = None
+        if use_proxy:
+            proxy_host = input("中转服务器地址: ")
+            proxy_port = input("中转服务器端口: ")
+            proxy_server = (proxy_host, int(proxy_port))
+        
+        # 添加服务器配置
+        config.add_server(
+            name=server_name,
+            host=server_host,
+            port=server_port,
+            use_https_disguise=use_https,
+            proxy_server=proxy_server
+        )
+        
+        # 设置为默认服务器
+        config.config['default_server'] = server_name
+        config.save_config()
+        
+        print(f"✅ 服务器 '{server_name}' 已添加并设置为默认服务器")
+    
+    print()
+    print("=== 设置完成 ===")
+    print("现在您可以使用以下命令快速开始:")
+    print()
+    print("1. 导入并获取远程主机:")
+    print("   from rcoder.utils import get_default_remote")
+    print("   remote = get_default_remote()")
+    print()
+    print("2. 执行命令:")
+    print("   result = remote.run('ls -la')")
+    print()
+
+
+def get_default_remote():
+    """获取默认远程主机实例，无需手动配置"""
+    from rcoder.core import get_remote_host
+    
+    config = ConfigManager()
+    server_config = config.get_server()
+    
+    return get_remote_host(
+        host=server_config['host'],
+        port=server_config['port'],
+        use_https_disguise=server_config['use_https_disguise'],
+        proxy_server=server_config['proxy_server']
+    )
+
+
+def create_alias():
+    """创建命令行别名，方便快速访问"""
+    shell = platform.system().lower()
+    
+    if shell == 'windows':
+        # Windows 系统
+        alias_cmd = 'doskey rcoder=python -m rcoder.cli $*'
+        print("在 Windows 中创建别名:")
+        print(f"请在命令提示符中执行: {alias_cmd}")
+        print("或添加到注册表以永久生效")
+    else:
+        # Linux/macOS 系统
+        shell_config = os.path.expanduser('~/.bashrc')
+        if os.path.exists(os.path.expanduser('~/.zshrc')):
+            shell_config = os.path.expanduser('~/.zshrc')
+        
+        alias_cmd = 'alias rcoder="python -m rcoder.cli"'
+        print(f"在 {shell_config} 中添加以下行:")
+        print(alias_cmd)
+        print("然后执行: source", shell_config)
+    
+    print()
+    print("创建别名后，您可以直接在终端中使用 'rcoder' 命令")
+
+
+def validate_config():
+    """验证配置文件的有效性"""
+    config = ConfigManager()
+    print("=== 配置验证 ===")
+    print(f"配置文件: {config.config_file}")
+    print(f"默认服务器: {config.config.get('default_server', 'local')}")
+    print(f"服务器数量: {len(config.config.get('servers', {}))}")
+    print()
+    print("服务器配置:")
+    for name, server in config.config.get('servers', {}).items():
+        print(f"  - {name}:")
+        print(f"    地址: {server.get('host')}:{server.get('port')}")
+        print(f"    HTTPS伪装: {'启用' if server.get('use_https_disguise') else '禁用'}")
+        if server.get('proxy_server'):
+            print(f"    中转服务器: {server.get('proxy_server')[0]}:{server.get('proxy_server')[1]}")
+    print()
+    print("✅ 配置验证完成")
+
+
+def export_config():
+    """导出配置文件，方便在不同设备间共享"""
+    config = ConfigManager()
+    export_path = os.path.expanduser('~/.rcoder/config_export.json')
+    config.save_config()
+    print(f"✅ 配置已导出到: {export_path}")
+    print("您可以将此文件复制到其他设备，然后使用 import_config() 导入")
+
+
+def import_config(config_file: str):
+    """导入配置文件"""
+    try:
+        with open(config_file, 'r', encoding='utf-8') as f:
+            imported_config = json.load(f)
+        
+        config = ConfigManager()
+        config.config = imported_config
+        config.save_config()
+        print(f"✅ 配置已从 {config_file} 导入")
+    except Exception as e:
+        print(f"❌ 导入配置失败: {e}")