@agentunion/kite 1.3.2 → 1.5.0

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

package/kite_cli/core/validator.py

@@ -0,0 +1,78 @@
+"""验证器 - 验证 module.md"""
+import re
+from pathlib import Path
+from typing import Optional, Dict
+
+
+class Validator:
+    """验证模块的 module.md"""
+
+    @staticmethod
+    def find_module_md(directory: Path) -> Optional[Path]:
+        """在目录中查找 module.md"""
+        # 直接查找
+        module_md = directory / "module.md"
+        if module_md.exists():
+            return module_md
+
+        # 递归查找（最多 2 层）
+        for md_file in directory.rglob("module.md"):
+            if md_file.relative_to(directory).parts.__len__() <= 2:
+                return md_file
+
+        return None
+
+    @staticmethod
+    def parse_module_md(path: Path) -> Optional[Dict]:
+        """解析 module.md 的 YAML frontmatter"""
+        try:
+            content = path.read_text(encoding="utf-8")
+
+            # 提取 YAML frontmatter (--- ... ---)
+            match = re.search(r"^---\s*\n(.*?)\n---", content, re.DOTALL | re.MULTILINE)
+            if not match:
+                return None
+
+            yaml_content = match.group(1)
+
+            # 简单解析 YAML（只解析我们需要的字段）
+            data = {}
+            for line in yaml_content.split("\n"):
+                line = line.strip()
+                if not line or line.startswith("#"):
+                    continue
+                if ":" in line:
+                    key, value = line.split(":", 1)
+                    key = key.strip()
+                    value = value.strip().strip('"').strip("'")
+                    data[key] = value
+
+            return data
+        except Exception:
+            return None
+
+    @classmethod
+    def validate_module(cls, directory: Path, expected_name: str) -> tuple[bool, Optional[str], Optional[Dict]]:
+        """
+        验证模块
+        返回: (是否有效, 错误信息, module.md 数据)
+        """
+        # 查找 module.md
+        module_md = cls.find_module_md(directory)
+        if not module_md:
+            return False, "未找到 module.md", None
+
+        # 解析 module.md
+        data = cls.parse_module_md(module_md)
+        if not data:
+            return False, "无法解析 module.md", None
+
+        # 验证 name 字段
+        actual_name = data.get("name")
+        if not actual_name:
+            return False, "module.md 缺少 name 字段", None
+
+        if actual_name != expected_name:
+            return False, f"模块名不匹配: 期望 {expected_name}, 实际 {actual_name}", None
+
+        return True, None, data