@agentunion/kite 1.3.2 → 1.4.0

package/kite_cli/core/downloader.py

@@ -0,0 +1,209 @@
+"""下载器 - 从各平台下载模块（改进版，带详细错误提示）"""
+import subprocess
+import tempfile
+import shutil
+import zipfile
+from pathlib import Path
+from typing import Optional, Tuple
+from kite_cli.core.tool_installer import ToolInstaller
+
+
+class DownloadError(Exception):
+    """下载错误"""
+    pass
+
+
+class Downloader:
+    """从各平台下载模块"""
+
+    @staticmethod
+    def download_pypi(name: str, dest: Path, version: str = None) -> Tuple[Optional[Path], Optional[str]]:
+        """
+        从 PyPI 下载包
+        Args:
+            name: 包名
+            dest: 目标目录
+            version: 版本号(可选),如 "1.2.0"
+        返回: (下载路径, 错误信息)
+        """
+        # 确保 pip 已安装
+        if not ToolInstaller.ensure_tool("pip"):
+            return None, "pip 未安装且自动安装失败，请手动安装 Python 和 pip"
+
+        # 构建包名(带版本)
+        package_spec = f"{name}=={version}" if version else name
+
+        try:
+            dest.mkdir(parents=True, exist_ok=True)
+            result = subprocess.run(
+                ["pip", "download", package_spec, "--dest", str(dest), "--no-deps"],
+                capture_output=True,
+                text=True,
+                timeout=60
+            )
+
+            if result.returncode != 0:
+                # pip 命令执行失败
+                error_msg = result.stderr.strip() if result.stderr else result.stdout.strip()
+                if "Could not find" in error_msg or "No matching distribution" in error_msg:
+                    return None, f"PyPI 上未找到包 '{name}'"
+                elif "Network" in error_msg or "timeout" in error_msg.lower():
+                    return None, f"网络错误: 无法连接到 PyPI"
+                else:
+                    return None, f"pip 下载失败: {error_msg[:200]}"
+
+            # 查找下载的文件
+            files = list(dest.glob("*"))
+            if not files:
+                return None, "下载成功但未找到文件"
+
+            # 解压（.whl 或 .tar.gz）
+            downloaded_file = files[0]
+            extract_dir = dest / "extracted"
+            extract_dir.mkdir(exist_ok=True)
+
+            try:
+                # .whl 文件是 zip 格式
+                if downloaded_file.suffix == '.whl':
+                    with zipfile.ZipFile(downloaded_file, 'r') as zip_ref:
+                        zip_ref.extractall(extract_dir)
+                else:
+                    shutil.unpack_archive(downloaded_file, extract_dir)
+                return extract_dir, None
+            except Exception as e:
+                return None, f"解压失败: {str(e)}"
+
+        except FileNotFoundError:
+            return None, "pip 命令未找到，请先安装 Python 和 pip"
+        except subprocess.TimeoutExpired:
+            return None, "下载超时（60秒），请检查网络连接"
+        except Exception as e:
+            return None, f"未知错误: {str(e)}"
+
+    @staticmethod
+    def download_npm(name: str, dest: Path, version: str = None) -> Tuple[Optional[Path], Optional[str]]:
+        """
+        从 npm 下载包
+        Args:
+            name: 包名
+            dest: 目标目录
+            version: 版本号(可选),如 "1.2.0"
+        返回: (下载路径, 错误信息)
+        """
+        # 确保 npm 已安装
+        if not ToolInstaller.ensure_tool("npm"):
+            return None, "npm 未安装且自动安装失败，请手动安装 Node.js 和 npm"
+
+        # 构建包名(带版本)
+        package_spec = f"{name}@{version}" if version else name
+
+        try:
+            dest.mkdir(parents=True, exist_ok=True)
+            result = subprocess.run(
+                ["npm", "pack", package_spec, "--pack-destination", str(dest)],
+                capture_output=True,
+                text=True,
+                timeout=60
+            )
+
+            if result.returncode != 0:
+                error_msg = result.stderr.strip() if result.stderr else result.stdout.strip()
+                if "404" in error_msg or "Not Found" in error_msg:
+                    return None, f"npm 上未找到包 '{name}'"
+                elif "network" in error_msg.lower() or "ETIMEDOUT" in error_msg:
+                    return None, "网络错误: 无法连接到 npm registry"
+                else:
+                    return None, f"npm 下载失败: {error_msg[:200]}"
+
+            # 查找 .tgz 文件
+            tgz_files = list(dest.glob("*.tgz"))
+            if not tgz_files:
+                return None, "下载成功但未找到 .tgz 文件"
+
+            tgz_file = tgz_files[0]
+            extract_dir = dest / "extracted"
+            extract_dir.mkdir(exist_ok=True)
+
+            try:
+                shutil.unpack_archive(tgz_file, extract_dir)
+                # npm pack 会创建 package/ 子目录
+                package_dir = extract_dir / "package"
+                if package_dir.exists():
+                    return package_dir, None
+                return extract_dir, None
+            except Exception as e:
+                return None, f"解压失败: {str(e)}"
+
+        except FileNotFoundError:
+            return None, "npm 命令未找到，请先安装 Node.js 和 npm"
+        except subprocess.TimeoutExpired:
+            return None, "下载超时（60秒），请检查网络连接"
+        except Exception as e:
+            return None, f"未知错误: {str(e)}"
+
+    @staticmethod
+    def download_git(url: str, dest: Path) -> Tuple[Optional[Path], Optional[str]]:
+        """
+        从 Git 克隆仓库
+        返回: (下载路径, 错误信息)
+        """
+        # 确保 git 已安装
+        if not ToolInstaller.ensure_tool("git"):
+            return None, "git 未安装且自动安装失败，请手动安装 Git"
+
+        try:
+            dest.mkdir(parents=True, exist_ok=True)
+            clone_dir = dest / "repo"
+            result = subprocess.run(
+                ["git", "clone", url, str(clone_dir)],
+                capture_output=True,
+                text=True,
+                timeout=120
+            )
+
+            if result.returncode != 0:
+                error_msg = result.stderr.strip() if result.stderr else result.stdout.strip()
+                if "not found" in error_msg.lower() or "does not exist" in error_msg.lower():
+                    return None, f"Git 仓库不存在: {url}"
+                elif "Authentication failed" in error_msg or "Permission denied" in error_msg:
+                    return None, "认证失败: 仓库可能是私有的或需要登录"
+                elif "network" in error_msg.lower() or "timeout" in error_msg.lower():
+                    return None, "网络错误: 无法连接到 Git 服务器"
+                else:
+                    return None, f"git clone 失败: {error_msg[:200]}"
+
+            if not clone_dir.exists():
+                return None, "克隆成功但目录不存在"
+
+            return clone_dir, None
+
+        except FileNotFoundError:
+            return None, "git 命令未找到，请先安装 Git"
+        except subprocess.TimeoutExpired:
+            return None, "克隆超时（120秒），仓库可能太大或网络太慢"
+        except Exception as e:
+            return None, f"未知错误: {str(e)}"
+
+    @staticmethod
+    def download_local(path: str, dest: Path) -> Tuple[Optional[Path], Optional[str]]:
+        """
+        从本地路径复制
+        返回: (下载路径, 错误信息)
+        """
+        try:
+            src = Path(path).resolve()
+            if not src.exists():
+                return None, f"本地路径不存在: {path}"
+
+            if not src.is_dir():
+                return None, f"路径不是目录: {path}"
+
+            dest.mkdir(parents=True, exist_ok=True)
+            copy_dir = dest / "local"
+            shutil.copytree(src, copy_dir)
+            return copy_dir, None
+
+        except PermissionError:
+            return None, f"权限不足: 无法读取 {path}"
+        except Exception as e:
+            return None, f"复制失败: {str(e)}"

package/kite_cli/core/install_info.py

@@ -0,0 +1,40 @@
+"""安装信息记录"""
+import json
+from datetime import datetime
+from pathlib import Path
+from kite_cli import __version__
+
+
+def record_install_info(module_dir: Path, source_info: dict, module_data: dict, location: str):
+    """记录安装信息到 .kite_install.json"""
+    install_info = {
+        "name": module_data.get("name"),
+        "version": module_data.get("version", "unknown"),
+        "source": source_info,
+        "installed_at": datetime.utcnow().isoformat() + "Z",
+        "installed_by": f"kite-cli@{__version__}",
+        "location": location,
+        "dependencies": module_data.get("dependencies", {})
+    }
+
+    install_info_path = module_dir / ".kite_install.json"
+    try:
+        with open(install_info_path, "w", encoding="utf-8") as f:
+            json.dump(install_info, f, indent=2, ensure_ascii=False)
+        return True
+    except Exception as e:
+        print(f"[Warning] 记录安装信息失败: {e}")
+        return False
+
+
+def read_install_info(module_dir: Path) -> dict:
+    """读取安装信息"""
+    install_info_path = module_dir / ".kite_install.json"
+    if not install_info_path.exists():
+        return None
+
+    try:
+        with open(install_info_path, encoding="utf-8") as f:
+            return json.load(f)
+    except Exception:
+        return None

package/kite_cli/core/tool_installer.py

@@ -0,0 +1,397 @@
+"""下载工具自动安装器 - 安装 pip/npm/git 等基础工具"""
+import subprocess
+import sys
+import platform
+from pathlib import Path
+
+
+class ToolInstaller:
+    """下载工具安装器"""
+
+    @staticmethod
+    def check_tool(tool_name: str) -> bool:
+        """检查工具是否已安装"""
+        try:
+            # Windows 上某些命令需要 shell=True (如 npm.cmd)
+            result = subprocess.run(
+                [tool_name, "--version"],
+                capture_output=True,
+                timeout=5,
+                shell=True  # Windows 兼容性
+            )
+            return result.returncode == 0
+        except (FileNotFoundError, subprocess.TimeoutExpired):
+            return False
+
+    @staticmethod
+    def install_python() -> bool:
+        """安装 Python（如果不存在）"""
+        if ToolInstaller.check_tool("python") or ToolInstaller.check_tool("python3"):
+            return True
+
+        print("[Install] Python 未安装，正在安装...")
+
+        system = platform.system()
+
+        if system == "Windows":
+            return ToolInstaller._install_python_windows()
+        elif system == "Darwin":  # macOS
+            return ToolInstaller._install_python_macos()
+        elif system == "Linux":
+            return ToolInstaller._install_python_linux()
+        else:
+            print(f"[Error] 不支持的操作系统: {system}")
+            return False
+
+    @staticmethod
+    def _install_python_windows() -> bool:
+        """Windows 上安装 Python"""
+        print("[Info] Windows 系统需要手动安装 Python")
+        print("[Info] 请访问: https://www.python.org/downloads/")
+        print("[Info] 或使用 winget: winget install Python.Python.3.12")
+        return False
+
+    @staticmethod
+    def _install_python_macos() -> bool:
+        """macOS 上安装 Python"""
+        # 检查是否有 brew
+        if not ToolInstaller.check_tool("brew"):
+            print("[Error] 需要先安装 Homebrew")
+            print("[Info] 访问: https://brew.sh/")
+            return False
+
+        try:
+            print("[Install] 使用 Homebrew 安装 Python...")
+            result = subprocess.run(
+                ["brew", "install", "python@3.12"],
+                capture_output=True,
+                text=True,
+                timeout=300
+            )
+            if result.returncode == 0:
+                print("[Done] Python 安装完成")
+                return True
+            else:
+                print(f"[Error] 安装失败: {result.stderr}")
+                return False
+        except Exception as e:
+            print(f"[Error] 安装失败: {e}")
+            return False
+
+    @staticmethod
+    def _install_python_linux() -> bool:
+        """Linux 上安装 Python"""
+        # 检测包管理器
+        if ToolInstaller.check_tool("apt"):
+            pkg_manager = "apt"
+            install_cmd = ["sudo", "apt", "update", "&&", "sudo", "apt", "install", "-y", "python3", "python3-pip"]
+        elif ToolInstaller.check_tool("yum"):
+            pkg_manager = "yum"
+            install_cmd = ["sudo", "yum", "install", "-y", "python3", "python3-pip"]
+        elif ToolInstaller.check_tool("dnf"):
+            pkg_manager = "dnf"
+            install_cmd = ["sudo", "dnf", "install", "-y", "python3", "python3-pip"]
+        else:
+            print("[Error] 未检测到支持的包管理器 (apt/yum/dnf)")
+            return False
+
+        try:
+            print(f"[Install] 使用 {pkg_manager} 安装 Python...")
+            result = subprocess.run(
+                " ".join(install_cmd),
+                shell=True,
+                capture_output=True,
+                text=True,
+                timeout=300
+            )
+            if result.returncode == 0:
+                print("[Done] Python 安装完成")
+                return True
+            else:
+                print(f"[Error] 安装失败: {result.stderr}")
+                return False
+        except Exception as e:
+            print(f"[Error] 安装失败: {e}")
+            return False
+
+    @staticmethod
+    def install_pip() -> bool:
+        """安装 pip（如果不存在）"""
+        if ToolInstaller.check_tool("pip") or ToolInstaller.check_tool("pip3"):
+            return True
+
+        print("[Install] pip 未安装，正在安装...")
+
+        # 尝试使用 ensurepip
+        try:
+            python_cmd = "python" if ToolInstaller.check_tool("python") else "python3"
+            result = subprocess.run(
+                [python_cmd, "-m", "ensurepip", "--upgrade"],
+                capture_output=True,
+                text=True,
+                timeout=60
+            )
+            if result.returncode == 0:
+                print("[Done] pip 安装完成")
+                return True
+        except Exception:
+            pass
+
+        # 尝试使用 get-pip.py
+        try:
+            import urllib.request
+            print("[Install] 下载 get-pip.py...")
+            url = "https://bootstrap.pypa.io/get-pip.py"
+            get_pip_path = Path.home() / "get-pip.py"
+
+            urllib.request.urlretrieve(url, get_pip_path)
+
+            python_cmd = "python" if ToolInstaller.check_tool("python") else "python3"
+            result = subprocess.run(
+                [python_cmd, str(get_pip_path)],
+                capture_output=True,
+                text=True,
+                timeout=120
+            )
+
+            get_pip_path.unlink()  # 删除临时文件
+
+            if result.returncode == 0:
+                print("[Done] pip 安装完成")
+                return True
+            else:
+                print(f"[Error] 安装失败: {result.stderr}")
+                return False
+
+        except Exception as e:
+            print(f"[Error] 安装失败: {e}")
+            return False
+
+    @staticmethod
+    def install_nodejs() -> bool:
+        """安装 Node.js 和 npm"""
+        if ToolInstaller.check_tool("node") and ToolInstaller.check_tool("npm"):
+            return True
+
+        print("[Install] Node.js/npm 未安装，正在安装...")
+
+        system = platform.system()
+
+        if system == "Windows":
+            return ToolInstaller._install_nodejs_windows()
+        elif system == "Darwin":
+            return ToolInstaller._install_nodejs_macos()
+        elif system == "Linux":
+            return ToolInstaller._install_nodejs_linux()
+        else:
+            print(f"[Error] 不支持的操作系统: {system}")
+            return False
+
+    @staticmethod
+    def _install_nodejs_windows() -> bool:
+        """Windows 上安装 Node.js"""
+        print("[Info] Windows 系统需要手动安装 Node.js")
+        print("[Info] 请访问: https://nodejs.org/")
+        print("[Info] 或使用 winget: winget install OpenJS.NodeJS")
+        return False
+
+    @staticmethod
+    def _install_nodejs_macos() -> bool:
+        """macOS 上安装 Node.js"""
+        if not ToolInstaller.check_tool("brew"):
+            print("[Error] 需要先安装 Homebrew")
+            print("[Info] 访问: https://brew.sh/")
+            return False
+
+        try:
+            print("[Install] 使用 Homebrew 安装 Node.js...")
+            result = subprocess.run(
+                ["brew", "install", "node"],
+                capture_output=True,
+                text=True,
+                timeout=300
+            )
+            if result.returncode == 0:
+                print("[Done] Node.js 安装完成")
+                return True
+            else:
+                print(f"[Error] 安装失败: {result.stderr}")
+                return False
+        except Exception as e:
+            print(f"[Error] 安装失败: {e}")
+            return False
+
+    @staticmethod
+    def _install_nodejs_linux() -> bool:
+        """Linux 上安装 Node.js"""
+        # 使用 NodeSource 仓库安装最新 LTS 版本
+        if ToolInstaller.check_tool("apt"):
+            try:
+                print("[Install] 添加 NodeSource 仓库...")
+                # 下载并执行 NodeSource 安装脚本
+                result = subprocess.run(
+                    "curl -fsSL https://deb.nodesource.com/setup_lts.x | sudo -E bash -",
+                    shell=True,
+                    capture_output=True,
+                    text=True,
+                    timeout=120
+                )
+                if result.returncode != 0:
+                    print(f"[Error] 添加仓库失败: {result.stderr}")
+                    return False
+
+                print("[Install] 安装 Node.js...")
+                result = subprocess.run(
+                    ["sudo", "apt", "install", "-y", "nodejs"],
+                    capture_output=True,
+                    text=True,
+                    timeout=300
+                )
+                if result.returncode == 0:
+                    print("[Done] Node.js 安装完成")
+                    return True
+                else:
+                    print(f"[Error] 安装失败: {result.stderr}")
+                    return False
+            except Exception as e:
+                print(f"[Error] 安装失败: {e}")
+                return False
+
+        elif ToolInstaller.check_tool("yum") or ToolInstaller.check_tool("dnf"):
+            pkg_manager = "dnf" if ToolInstaller.check_tool("dnf") else "yum"
+            try:
+                print(f"[Install] 使用 {pkg_manager} 安装 Node.js...")
+                result = subprocess.run(
+                    ["sudo", pkg_manager, "install", "-y", "nodejs", "npm"],
+                    capture_output=True,
+                    text=True,
+                    timeout=300
+                )
+                if result.returncode == 0:
+                    print("[Done] Node.js 安装完成")
+                    return True
+                else:
+                    print(f"[Error] 安装失败: {result.stderr}")
+                    return False
+            except Exception as e:
+                print(f"[Error] 安装失败: {e}")
+                return False
+        else:
+            print("[Error] 未检测到支持的包管理器")
+            return False
+
+    @staticmethod
+    def install_git() -> bool:
+        """安装 Git"""
+        if ToolInstaller.check_tool("git"):
+            return True
+
+        print("[Install] Git 未安装，正在安装...")
+
+        system = platform.system()
+
+        if system == "Windows":
+            return ToolInstaller._install_git_windows()
+        elif system == "Darwin":
+            return ToolInstaller._install_git_macos()
+        elif system == "Linux":
+            return ToolInstaller._install_git_linux()
+        else:
+            print(f"[Error] 不支持的操作系统: {system}")
+            return False
+
+    @staticmethod
+    def _install_git_windows() -> bool:
+        """Windows 上安装 Git"""
+        print("[Info] Windows 系统需要手动安装 Git")
+        print("[Info] 请访问: https://git-scm.com/download/win")
+        print("[Info] 或使用 winget: winget install Git.Git")
+        return False
+
+    @staticmethod
+    def _install_git_macos() -> bool:
+        """macOS 上安装 Git"""
+        if not ToolInstaller.check_tool("brew"):
+            print("[Error] 需要先安装 Homebrew")
+            print("[Info] 访问: https://brew.sh/")
+            return False
+
+        try:
+            print("[Install] 使用 Homebrew 安装 Git...")
+            result = subprocess.run(
+                ["brew", "install", "git"],
+                capture_output=True,
+                text=True,
+                timeout=300
+            )
+            if result.returncode == 0:
+                print("[Done] Git 安装完成")
+                return True
+            else:
+                print(f"[Error] 安装失败: {result.stderr}")
+                return False
+        except Exception as e:
+            print(f"[Error] 安装失败: {e}")
+            return False
+
+    @staticmethod
+    def _install_git_linux() -> bool:
+        """Linux 上安装 Git"""
+        if ToolInstaller.check_tool("apt"):
+            pkg_manager = "apt"
+            install_cmd = ["sudo", "apt", "update", "&&", "sudo", "apt", "install", "-y", "git"]
+        elif ToolInstaller.check_tool("yum"):
+            pkg_manager = "yum"
+            install_cmd = ["sudo", "yum", "install", "-y", "git"]
+        elif ToolInstaller.check_tool("dnf"):
+            pkg_manager = "dnf"
+            install_cmd = ["sudo", "dnf", "install", "-y", "git"]
+        else:
+            print("[Error] 未检测到支持的包管理器")
+            return False
+
+        try:
+            print(f"[Install] 使用 {pkg_manager} 安装 Git...")
+            result = subprocess.run(
+                " ".join(install_cmd),
+                shell=True,
+                capture_output=True,
+                text=True,
+                timeout=300
+            )
+            if result.returncode == 0:
+                print("[Done] Git 安装完成")
+                return True
+            else:
+                print(f"[Error] 安装失败: {result.stderr}")
+                return False
+        except Exception as e:
+            print(f"[Error] 安装失败: {e}")
+            return False
+
+    @classmethod
+    def ensure_tool(cls, tool: str) -> bool:
+        """确保工具已安装，如果没有则尝试安装"""
+        if tool in ["python", "python3"]:
+            if cls.check_tool("python") or cls.check_tool("python3"):
+                return True
+            return cls.install_python()
+
+        elif tool in ["pip", "pip3"]:
+            if cls.check_tool("pip") or cls.check_tool("pip3"):
+                return True
+            return cls.install_pip()
+
+        elif tool in ["node", "npm"]:
+            if cls.check_tool("node") and cls.check_tool("npm"):
+                return True
+            return cls.install_nodejs()
+
+        elif tool == "git":
+            if cls.check_tool("git"):
+                return True
+            return cls.install_git()
+
+        else:
+            print(f"[Error] 未知工具: {tool}")
+            return False