@agentunion/kite 1.4.0 → 1.6.0

package/kite_cli/core/config_manager.py

@@ -0,0 +1,145 @@
+"""配置管理器"""
+import json
+from pathlib import Path
+from typing import Any, Optional
+
+
+# 默认配置
+DEFAULT_CONFIG = {
+    "default_location": "global",
+    "pypi_mirror": "",
+    "npm_registry": "",
+    "auto_update_check": False,
+    "confirm_install": True,
+    "confirm_uninstall": True,
+    "confirm_update": True,
+}
+
+# 配置项验证规则
+CONFIG_SCHEMA = {
+    "default_location": {
+        "type": "string",
+        "choices": ["dev", "local", "global"],
+        "description": "默认安装位置"
+    },
+    "pypi_mirror": {
+        "type": "string",
+        "description": "PyPI 镜像源 URL"
+    },
+    "npm_registry": {
+        "type": "string",
+        "description": "npm 镜像源 URL"
+    },
+    "auto_update_check": {
+        "type": "boolean",
+        "description": "自动检查更新"
+    },
+    "confirm_install": {
+        "type": "boolean",
+        "description": "安装前确认"
+    },
+    "confirm_uninstall": {
+        "type": "boolean",
+        "description": "卸载前确认"
+    },
+    "confirm_update": {
+        "type": "boolean",
+        "description": "更新前确认"
+    },
+}
+
+
+class ConfigManager:
+    """配置管理器"""
+
+    def __init__(self):
+        self.config_dir = Path.home() / '.kite' / 'data'
+        self.config_file = self.config_dir / 'cli_config.json'
+        self._config = None
+
+    def _ensure_config_dir(self):
+        """确保配置目录存在"""
+        self.config_dir.mkdir(parents=True, exist_ok=True)
+
+    def _load_config(self) -> dict:
+        """加载配置"""
+        if self._config is not None:
+            return self._config
+
+        self._ensure_config_dir()
+
+        if self.config_file.exists():
+            try:
+                with open(self.config_file, 'r', encoding='utf-8') as f:
+                    user_config = json.load(f)
+                # 合并默认配置和用户配置
+                self._config = {**DEFAULT_CONFIG, **user_config}
+            except Exception:
+                self._config = DEFAULT_CONFIG.copy()
+        else:
+            self._config = DEFAULT_CONFIG.copy()
+
+        return self._config
+
+    def _save_config(self):
+        """保存配置"""
+        self._ensure_config_dir()
+        with open(self.config_file, 'w', encoding='utf-8') as f:
+            json.dump(self._config, f, indent=2, ensure_ascii=False)
+
+    def get(self, key: str) -> Any:
+        """获取配置值"""
+        config = self._load_config()
+        return config.get(key)
+
+    def set(self, key: str, value: Any) -> bool:
+        """设置配置值"""
+        if key not in CONFIG_SCHEMA:
+            return False
+
+        # 验证值
+        schema = CONFIG_SCHEMA[key]
+        if schema["type"] == "boolean":
+            if not isinstance(value, bool):
+                # 尝试转换
+                if isinstance(value, str):
+                    value = value.lower() in ['true', '1', 'yes', 'on']
+                else:
+                    return False
+        elif schema["type"] == "string":
+            value = str(value)
+            if "choices" in schema and value not in schema["choices"]:
+                return False
+
+        config = self._load_config()
+        config[key] = value
+        self._config = config
+        self._save_config()
+        return True
+
+    def unset(self, key: str) -> bool:
+        """删除配置项（恢复为默认值）"""
+        if key not in CONFIG_SCHEMA:
+            return False
+
+        config = self._load_config()
+        if key in config:
+            config[key] = DEFAULT_CONFIG[key]
+            self._config = config
+            self._save_config()
+        return True
+
+    def list_all(self) -> dict:
+        """列出所有配置"""
+        return self._load_config()
+
+    def reset(self):
+        """重置为默认配置"""
+        self._config = DEFAULT_CONFIG.copy()
+        self._save_config()
+
+    def get_schema(self, key: Optional[str] = None):
+        """获取配置项的 schema"""
+        if key:
+            return CONFIG_SCHEMA.get(key)
+        return CONFIG_SCHEMA

package/kite_cli/core/downloader.py

@@ -33,10 +33,25 @@ class Downloader:
         # 构建包名(带版本)
         package_spec = f"{name}=={version}" if version else name
 
+        # 读取镜像源配置
+        try:
+            from kite_cli.core.config_manager import ConfigManager
+            mgr = ConfigManager()
+            pypi_mirror = mgr.get("pypi_mirror")
+        except Exception:
+            pypi_mirror = None
+
         try:
             dest.mkdir(parents=True, exist_ok=True)
+
+            cmd = ["pip", "download", package_spec, "--dest", str(dest), "--no-deps"]
+
+            # 如果配置了镜像源，添加 --index-url
+            if pypi_mirror:
+                cmd.extend(["--index-url", pypi_mirror])
+
             result = subprocess.run(
-                ["pip", "download", package_spec, "--dest", str(dest), "--no-deps"],
+                cmd,
                 capture_output=True,
                 text=True,
                 timeout=60
@@ -97,10 +112,25 @@ class Downloader:
         # 构建包名(带版本)
         package_spec = f"{name}@{version}" if version else name
 
+        # 读取 npm 镜像源配置
+        try:
+            from kite_cli.core.config_manager import ConfigManager
+            mgr = ConfigManager()
+            npm_registry = mgr.get("npm_registry")
+        except Exception:
+            npm_registry = None
+
         try:
             dest.mkdir(parents=True, exist_ok=True)
+
+            cmd = ["npm", "pack", package_spec, "--pack-destination", str(dest)]
+
+            # 如果配置了镜像源，添加 --registry
+            if npm_registry:
+                cmd.extend(["--registry", npm_registry])
+
             result = subprocess.run(
-                ["npm", "pack", package_spec, "--pack-destination", str(dest)],
+                cmd,
                 capture_output=True,
                 text=True,
                 timeout=60

package/kite_cli/main.py

@@ -6,7 +6,21 @@ from kite_cli import __version__
 from kite_cli.utils.i18n import t
 
 
-KNOWN_COMMANDS = ["install", "uninstall", "update", "list", "search", "info", "log", "rollback", "clean", "doctor", "history"]
+KNOWN_COMMANDS = ["install", "uninstall", "update", "list", "search", "info", "log", "rollback", "clean", "doctor", "history", "prepare", "env-check", "venv-setup", "deps-install", "outdated", "enable", "disable", "export", "import", "config", "deps", "why", "check", "build", "install-skill"]
+
+# 命令别名映射
+COMMAND_ALIASES = {
+    "i": "install",
+    "add": "install",
+    "rm": "uninstall",
+    "remove": "uninstall",
+    "up": "update",
+    "upgrade": "update",
+    "ls": "list",
+    "show": "list",
+    "b": "build",
+    "compile": "build",
+}
 
 
 class SilentArgumentParser(argparse.ArgumentParser):
@@ -35,17 +49,32 @@ Kite CLI v{__version__} — 模块安装管理工具
   kite <命令> [参数] [选项]
 
 命令:
-  install   <source>    安装模块
-  uninstall <name>      卸载模块
-  update    <name>      更新模块（使用安装时记录的来源）
-  list                  列出已安装的模块
+  install   <source>    安装模块 (别名: i, add)
+  uninstall <name>      卸载模块 (别名: rm, remove)
+  update    <name>      更新模块（使用安装时记录的来源）(别名: up, upgrade)
+  list                  列出已安装的模块 (别名: ls, show)
   search    <keyword>   搜索可用模块
   info      <name>      显示模块详细信息
+  build     [name]      编译模块 (别名: b, compile)
   log                   查看操作日志
   history               查看最近安装的模块
   rollback              回滚最后一次操作
   clean                 清理临时文件和缓存
   doctor                检查下载工具状态
+  prepare               打包前准备（扫描依赖，生成锁定文件）
+  env-check             环境检查（虚拟环境和依赖库）
+  venv-setup            创建虚拟环境
+  deps-install          安装依赖库
+  outdated              列出可更新的模块
+  enable    <name>      启用模块
+  disable   <name>      禁用模块
+  export                导出已安装模块列表
+  import    <file>      从文件批量安装模块
+  config    <子命令>    配置管理 (list/get/set/unset/reset)
+  deps      <name>      显示模块依赖树
+  why       <name>      显示为什么安装了某个模块
+  check                 检查依赖完整性
+  install-skill         安装 Kite skill 到 Claude Code
 
 全局选项:
   -h, -H, --help        显示此帮助信息
@@ -146,6 +175,10 @@ def main():
         print(f"kite-cli {__version__}")
         return 0
 
+    # 处理命令别名（在 argparse 解析之前）
+    if len(sys.argv) > 1 and sys.argv[1] in COMMAND_ALIASES:
+        sys.argv[1] = COMMAND_ALIASES[sys.argv[1]]
+
     parser = SilentArgumentParser(
         prog="kite",
         description="Kite 模块安装管理工具",
@@ -177,6 +210,9 @@ def main():
     list_parser.add_argument("-l", "--location", choices=["dev", "local", "global"], help="筛选位置")
     list_parser.add_argument("--all", action="store_true", help="所有位置")
     list_parser.add_argument("-v", "--verbose", action="store_true", help="详细信息")
+    list_parser.add_argument("--outdated", action="store_true", help="只显示可更新的模块")
+    list_parser.add_argument("--disabled", action="store_true", help="只显示禁用的模块")
+    list_parser.add_argument("--enabled", action="store_true", help="只显示启用的模块")
 
     # search 命令
     search_parser = subparsers.add_parser("search", help="搜索可用模块", add_help=True)
@@ -218,8 +254,101 @@ def main():
     # doctor 命令
     doctor_parser = subparsers.add_parser("doctor", help="检查下载工具状态", add_help=True)
 
+    # prepare 命令
+    prepare_parser = subparsers.add_parser("prepare", help="打包前准备（扫描依赖）", add_help=True)
+
+    # env-check 命令
+    env_check_parser = subparsers.add_parser("env-check", help="环境检查（虚拟环境和依赖库）", add_help=True)
+
+    # venv-setup 命令
+    venv_setup_parser = subparsers.add_parser("venv-setup", help="创建虚拟环境", add_help=True)
+
+    # deps-install 命令
+    deps_install_parser = subparsers.add_parser("deps-install", help="安装依赖库", add_help=True)
+
+    # outdated 命令
+    outdated_parser = subparsers.add_parser("outdated", help="列出可更新的模块", add_help=True)
+    outdated_parser.add_argument("-l", "--location", choices=["dev", "local", "global"], help="只检查指定位置")
+
+    # enable 命令
+    enable_parser = subparsers.add_parser("enable", help="启用模块", add_help=True)
+    enable_parser.add_argument("name", help="模块名")
+    enable_parser.add_argument("-l", "--location", choices=["dev", "local", "global"], help="指定位置")
+    enable_parser.add_argument("-y", "--yes", action="store_true", help="跳过确认")
+
+    # disable 命令
+    disable_parser = subparsers.add_parser("disable", help="禁用模块", add_help=True)
+    disable_parser.add_argument("name", help="模块名")
+    disable_parser.add_argument("-l", "--location", choices=["dev", "local", "global"], help="指定位置")
+    disable_parser.add_argument("-y", "--yes", action="store_true", help="跳过确认")
+
+    # export 命令
+    export_parser = subparsers.add_parser("export", help="导出已安装模块列表", add_help=True)
+    export_parser.add_argument("-l", "--location", choices=["dev", "local", "global"], help="只导出指定位置")
+    export_parser.add_argument("--format", choices=["text", "json"], default="text", help="输出格式")
+
+    # import 命令
+    import_parser = subparsers.add_parser("import", help="从文件批量安装模块", add_help=True)
+    import_parser.add_argument("file", help="模块列表文件（使用 - 表示从 stdin 读取）")
+    import_parser.add_argument("-y", "--yes", action="store_true", help="跳过确认")
+
+    # deps 命令
+    deps_parser = subparsers.add_parser("deps", help="显示模块依赖树", add_help=True)
+    deps_parser.add_argument("name", help="模块名")
+    deps_parser.add_argument("-l", "--location", choices=["dev", "local", "global"], help="指定位置")
+    deps_parser.add_argument("--depth", type=int, default=-1, help="最大深度（-1 表示无限制）")
+
+    # why 命令
+    why_parser = subparsers.add_parser("why", help="显示为什么安装了某个模块", add_help=True)
+    why_parser.add_argument("name", help="模块名")
+    why_parser.add_argument("-l", "--location", choices=["dev", "local", "global"], help="指定位置")
+
+    # check 命令
+    check_parser = subparsers.add_parser("check", help="检查依赖完整性", add_help=True)
+    check_parser.add_argument("-l", "--location", choices=["dev", "local", "global"], help="指定位置")
+    check_parser.add_argument("--fix", action="store_true", help="尝试修复问题")
+
+    # build 命令
+    build_parser = subparsers.add_parser("build", help="编译模块", add_help=True)
+    build_parser.add_argument("name", nargs="?", help="模块名（省略则编译当前目录）")
+    build_parser.add_argument("-l", "--location", choices=["dev", "local", "global"], help="指定位置")
+    build_parser.add_argument("-t", "--type", help="强制指定构建类型")
+    build_parser.add_argument("--clean", action="store_true", help="清理后编译")
+    build_parser.add_argument("--dry-run", action="store_true", help="显示命令但不执行")
+    build_parser.add_argument("--env", default="default", help="指定环境（dev/prod）")
+    build_parser.add_argument("-v", "--verbose", action="store_true", help="详细输出")
+    build_parser.add_argument("-y", "--yes", action="store_true", help="跳过确认")
+
+    # install-skill 命令
+    install_skill_parser = subparsers.add_parser("install-skill", help="安装 Kite skill 到 Claude Code", add_help=True)
+    install_skill_parser.add_argument("--user", action="store_true", help="安装到用户级 (~/.claude/skills/kite/)")
+    install_skill_parser.add_argument("--project", action="store_true", help="安装到项目级 (./.claude/skills/kite/)")
+
+    # config 命令
+    config_parser = subparsers.add_parser("config", help="配置管理", add_help=True)
+    config_subs = config_parser.add_subparsers(dest="subcmd")
+
+    config_subs.add_parser("list", help="列出所有配置")
+
+    config_get = config_subs.add_parser("get", help="获取配置值")
+    config_get.add_argument("key", help="配置项名称")
+
+    config_set = config_subs.add_parser("set", help="设置配置值")
+    config_set.add_argument("key", help="配置项名称")
+    config_set.add_argument("value", help="配置值")
+
+    config_unset = config_subs.add_parser("unset", help="删除配置（恢复默认）")
+    config_unset.add_argument("key", help="配置项名称")
+
+    config_reset = config_subs.add_parser("reset", help="重置所有配置为默认值")
+    config_reset.add_argument("-y", "--yes", action="store_true", help="跳过确认")
+
     args, unknown = parser.parse_known_args()
 
+    # 处理命令别名
+    if args.command and args.command in COMMAND_ALIASES:
+        args.command = COMMAND_ALIASES[args.command]
+
     # 处理全局 --help / --version
     if args.help:
         print_help()
@@ -280,6 +409,51 @@ def main():
     elif args.command == "doctor":
         from kite_cli.commands.doctor import run_doctor
         return run_doctor(args)
+    elif args.command == "prepare":
+        from kite_cli.commands.prepare import run_prepare
+        return run_prepare(args)
+    elif args.command == "env-check":
+        from kite_cli.commands.env_check import run_env_check
+        return run_env_check(args)
+    elif args.command == "venv-setup":
+        from kite_cli.commands.venv_setup import run_venv_setup
+        return run_venv_setup(args)
+    elif args.command == "deps-install":
+        from kite_cli.commands.deps_install import run_deps_install
+        return run_deps_install(args)
+    elif args.command == "outdated":
+        from kite_cli.commands.outdated import run_outdated
+        return run_outdated(args)
+    elif args.command == "enable":
+        from kite_cli.commands.enable import run_enable
+        return run_enable(args)
+    elif args.command == "disable":
+        from kite_cli.commands.disable import run_disable
+        return run_disable(args)
+    elif args.command == "export":
+        from kite_cli.commands.export import run_export
+        return run_export(args)
+    elif args.command == "import":
+        from kite_cli.commands.import_cmd import run_import
+        return run_import(args)
+    elif args.command == "deps":
+        from kite_cli.commands.deps import run_deps
+        return run_deps(args)
+    elif args.command == "why":
+        from kite_cli.commands.why import run_why
+        return run_why(args)
+    elif args.command == "check":
+        from kite_cli.commands.check import run_check
+        return run_check(args)
+    elif args.command == "build":
+        from kite_cli.commands.build import run_build
+        return run_build(args)
+    elif args.command == "install-skill":
+        from kite_cli.commands.install_skill import main
+        return main()
+    elif args.command == "config":
+        from kite_cli.commands.config import run_config
+        return run_config(args)
     else:
         print(f"未知命令: {args.command}")
         return 1

package/kite_cli/utils/colors.py

@@ -0,0 +1,153 @@
+"""彩色输出工具"""
+import sys
+import os
+
+
+# ANSI 颜色代码
+class Colors:
+    """ANSI 颜色代码"""
+    # 基础颜色
+    BLACK = '\033[30m'
+    RED = '\033[31m'
+    GREEN = '\033[32m'
+    YELLOW = '\033[33m'
+    BLUE = '\033[34m'
+    MAGENTA = '\033[35m'
+    CYAN = '\033[36m'
+    WHITE = '\033[37m'
+
+    # 亮色
+    BRIGHT_BLACK = '\033[90m'
+    BRIGHT_RED = '\033[91m'
+    BRIGHT_GREEN = '\033[92m'
+    BRIGHT_YELLOW = '\033[93m'
+    BRIGHT_BLUE = '\033[94m'
+    BRIGHT_MAGENTA = '\033[95m'
+    BRIGHT_CYAN = '\033[96m'
+    BRIGHT_WHITE = '\033[97m'
+
+    # 样式
+    BOLD = '\033[1m'
+    DIM = '\033[2m'
+    ITALIC = '\033[3m'
+    UNDERLINE = '\033[4m'
+
+    # 重置
+    RESET = '\033[0m'
+
+
+def supports_color():
+    """检查终端是否支持彩色输出"""
+    # Windows 10+ 支持 ANSI 颜色
+    if sys.platform == 'win32':
+        # 尝试启用 Windows 10 的 ANSI 支持
+        try:
+            import ctypes
+            kernel32 = ctypes.windll.kernel32
+            kernel32.SetConsoleMode(kernel32.GetStdHandle(-11), 7)
+            return True
+        except Exception:
+            return False
+
+    # Unix/Linux/Mac 检查 TERM 环境变量
+    if not hasattr(sys.stdout, 'isatty'):
+        return False
+    if not sys.stdout.isatty():
+        return False
+
+    term = os.environ.get('TERM', '')
+    if term == 'dumb':
+        return False
+
+    return True
+
+
+# 全局标志
+_COLOR_ENABLED = supports_color()
+
+
+def enable_color(enabled=True):
+    """启用或禁用彩色输出"""
+    global _COLOR_ENABLED
+    _COLOR_ENABLED = enabled
+
+
+def is_color_enabled():
+    """检查彩色输出是否启用"""
+    return _COLOR_ENABLED
+
+
+def colorize(text, color, bold=False):
+    """给文本添加颜色"""
+    if not _COLOR_ENABLED:
+        return text
+
+    prefix = color
+    if bold:
+        prefix = Colors.BOLD + color
+
+    return f"{prefix}{text}{Colors.RESET}"
+
+
+# 便捷函数
+def success(text, bold=False):
+    """成功消息（绿色）"""
+    return colorize(text, Colors.GREEN, bold)
+
+
+def error(text, bold=False):
+    """错误消息（红色）"""
+    return colorize(text, Colors.RED, bold)
+
+
+def warning(text, bold=False):
+    """警告消息（黄色）"""
+    return colorize(text, Colors.YELLOW, bold)
+
+
+def info(text, bold=False):
+    """信息消息（蓝色）"""
+    return colorize(text, Colors.BLUE, bold)
+
+
+def dim(text):
+    """暗淡文本（灰色）"""
+    return colorize(text, Colors.BRIGHT_BLACK, False)
+
+
+def highlight(text):
+    """高亮文本（青色加粗）"""
+    return colorize(text, Colors.CYAN, True)
+
+
+def bold(text):
+    """加粗文本"""
+    if not _COLOR_ENABLED:
+        return text
+    return f"{Colors.BOLD}{text}{Colors.RESET}"
+
+
+# 带标签的输出函数
+def print_success(message):
+    """打印成功消息"""
+    print(f"{success('[Success]', True)} {message}")
+
+
+def print_error(message):
+    """打印错误消息"""
+    print(f"{error('[Error]', True)} {message}")
+
+
+def print_warning(message):
+    """打印警告消息"""
+    print(f"{warning('[Warning]', True)} {message}")
+
+
+def print_info(message):
+    """打印信息消息"""
+    print(f"{info('[Info]', True)} {message}")
+
+
+def print_cancelled(message="已取消"):
+    """打印取消消息"""
+    print(f"{dim('[Cancelled]')} {message}")