stigmergy 1.0.57

package/src/adapters/iflow/install_iflow_integration.py

@@ -0,0 +1,271 @@
+"""
+iFlow CLI Hook集成安装脚本
+为iFlow CLI安装跨CLI协作感知能力
+
+使用方法：
+python install_iflow_integration.py [--verify|--uninstall]
+"""
+
+import os
+import sys
+import json
+import shutil
+import argparse
+from pathlib import Path
+from datetime import datetime
+
+# 获取当前文件目录
+current_dir = Path(__file__).parent
+project_root = current_dir.parent.parent.parent
+
+# iFlow CLI配置路径
+IFLOW_CONFIG_DIR = os.path.expanduser("~/.config/iflow")
+IFLOW_HOOKS_FILE = os.path.join(IFLOW_CONFIG_DIR, "hooks.yml")
+
+def create_iflow_config_directory():
+    """创建iFlow配置目录"""
+    os.makedirs(IFLOW_CONFIG_DIR, exist_ok=True)
+    print(f"[OK] 创建iFlow配置目录: {IFLOW_CONFIG_DIR}")
+
+def install_iflow_hooks():
+    """安装iFlow Hook配置"""
+    # 读取现有hooks配置
+    existing_hooks = {}
+    if os.path.exists(IFLOW_HOOKS_FILE):
+        try:
+            import yaml
+            with open(IFLOW_HOOKS_FILE, 'r', encoding='utf-8') as f:
+                existing_hooks = yaml.safe_load(f) or {}
+        except Exception as e:
+            print(f"⚠️ 读取现有hooks配置失败: {e}")
+            existing_hooks = {}
+
+    # 定义跨CLI协作的Hook配置
+    cross_cli_hooks = {
+        "cross_cli_hook_adapter": {
+            "name": "CrossCLIHookAdapter",
+            "module": "src.adapters.iflow.hook_adapter",
+            "class": "IFlowHookAdapter",
+            "enabled": True,
+            "priority": 100,
+            "hooks": [
+                "on_command_start",
+                "on_command_end",
+                "on_user_input",
+                "on_workflow_stage",
+                "on_pipeline_execute",
+                "on_output_render",
+                "on_error"
+            ],
+            "config": {
+                "cross_cli_enabled": True,
+                "supported_clis": ["claude", "gemini", "qwencode", "qoder", "codebuddy", "copilot"],
+                "auto_detect": True,
+                "timeout": 30,
+                "error_handling": "continue",
+                "collaboration_mode": "active"
+            }
+        }
+    }
+
+    # 合并配置（保留现有hooks，添加协作功能）
+    merged_hooks = existing_hooks.copy()
+    if 'plugins' not in merged_hooks:
+        merged_hooks['plugins'] = []
+
+    # 检查是否已存在跨CLI Hook
+    existing_plugin_names = [plugin.get('name') for plugin in merged_hooks.get('plugins', [])]
+    cross_cli_hook_exists = any(plugin_name == 'CrossCLIHookAdapter' for plugin_name in existing_plugin_names)
+
+    if not cross_cli_hook_exists:
+        merged_hooks['plugins'].append(cross_cli_hooks['cross_cli_hook_adapter'])
+
+    # 写入hooks配置文件
+    try:
+        import yaml
+        with open(IFLOW_HOOKS_FILE, 'w', encoding='utf-8') as f:
+            yaml.dump(merged_hooks, f, default_flow_style=False, allow_unicode=True)
+
+        print(f"[OK] iFlow Hook配置已安装: {IFLOW_HOOKS_FILE}")
+        print("🔗 已安装的Hook:")
+        for plugin in merged_hooks.get('plugins', []):
+            if plugin.get('name') == 'CrossCLIHookAdapter':
+                print(f"   - {plugin['name']}: [OK] 跨CLI协作感知")
+                print(f"     支持的CLI: {', '.join(plugin['config'].get('supported_clis', []))}")
+
+        return True
+    except Exception as e:
+        print(f"❌ 安装iFlow Hook配置失败: {e}")
+        return False
+
+def copy_adapter_file():
+    """复制适配器文件到iFlow配置目录"""
+    try:
+        # 创建适配器目录
+        adapter_dir = os.path.join(IFLOW_CONFIG_DIR, "adapters")
+        os.makedirs(adapter_dir, exist_ok=True)
+
+        # 复制适配器文件
+        adapter_files = [
+            "hook_adapter.py",
+            "standalone_iflow_adapter.py"
+        ]
+
+        for file_name in adapter_files:
+            src_file = current_dir / file_name
+            dst_file = os.path.join(adapter_dir, file_name)
+
+            if src_file.exists():
+                shutil.copy2(src_file, dst_file)
+                print(f"[OK] 复制适配器文件: {file_name}")
+            else:
+                print(f"⚠️ 适配器文件不存在: {file_name}")
+
+        return True
+    except Exception as e:
+        print(f"❌ 复制适配器文件失败: {e}")
+        return False
+
+def verify_installation():
+    """验证安装是否成功"""
+    print("\n🔍 验证iFlow CLI集成安装...")
+
+    # 检查配置目录
+    if not os.path.exists(IFLOW_CONFIG_DIR):
+        print(f"❌ 配置目录不存在: {IFLOW_CONFIG_DIR}")
+        return False
+
+    # 检查hooks文件
+    if not os.path.exists(IFLOW_HOOKS_FILE):
+        print(f"❌ Hooks配置文件不存在: {IFLOW_HOOKS_FILE}")
+        return False
+
+    # 检查适配器目录
+    adapter_dir = os.path.join(IFLOW_CONFIG_DIR, "adapters")
+    if not os.path.exists(adapter_dir):
+        print(f"❌ 适配器目录不存在: {adapter_dir}")
+        return False
+
+    # 读取并验证hooks配置
+    try:
+        import yaml
+        with open(IFLOW_HOOKS_FILE, 'r', encoding='utf-8') as f:
+            hooks_config = yaml.safe_load(f) or {}
+
+        plugins = hooks_config.get('plugins', [])
+        cross_cli_plugin = None
+
+        for plugin in plugins:
+            if plugin.get('name') == 'CrossCLIHookAdapter':
+                cross_cli_plugin = plugin
+                print(f"[OK] 跨CLI协作Hook: 已启用")
+                print(f"[OK] 支持的CLI工具: {', '.join(plugin.get('config', {}).get('supported_clis', []))}")
+                break
+
+        if not cross_cli_plugin:
+            print("❌ 跨CLI协作Hook: 未找到")
+            return False
+
+    except Exception as e:
+        print(f"❌ 读取hooks配置失败: {e}")
+        return False
+
+def uninstall_iflow_integration():
+    """卸载iFlow集成"""
+    try:
+        # 备份现有配置
+        if os.path.exists(IFLOW_HOOKS_FILE):
+            backup_file = f"{IFLOW_HOOKS_FILE}.backup_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
+            shutil.copy2(IFLOW_HOOKS_FILE, backup_file)
+            print(f"📦 已备份现有配置: {backup_file}")
+
+        # 移除跨CLI Hook
+        if os.path.exists(IFLOW_HOOKS_FILE):
+            import yaml
+            with open(IFLOW_HOOKS_FILE, 'r', encoding='utf-8') as f:
+                hooks_config = yaml.safe_load(f) or {}
+
+            if 'plugins' in hooks_config:
+                plugins = hooks_config['plugins']
+                # 移除跨CLI Hook
+                plugins = [p for p in plugins if p.get('name') != 'CrossCLIHookAdapter']
+
+                hooks_config['plugins'] = plugins
+
+                # 写回配置
+                with open(IFLOW_HOOKS_FILE, 'w', encoding='utf-8') as f:
+                    yaml.dump(hooks_config, f, default_flow_style=False, allow_unicode=True)
+
+        print("[OK] iFlow跨CLI协作集成已卸载")
+        return True
+    except Exception as e:
+        print(f"❌ 卸载失败: {e}")
+        return False
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="iFlow CLI跨CLI协作集成安装脚本",
+        formatter_class=argparse.RawDescriptionHelpFormatter
+    )
+
+    parser.add_argument(
+        "--install",
+        action="store_true",
+        help="安装iFlow CLI跨CLI协作集成"
+    )
+
+    parser.add_argument(
+        "--verify",
+        action="store_true",
+        help="验证iFlow CLI集成安装"
+    )
+
+    parser.add_argument(
+        "--uninstall",
+        action="store_true",
+        help="卸载iFlow CLI跨CLI协作集成"
+    )
+
+    args = parser.parse_args()
+
+    print("[INSTALL] iFlow CLI跨CLI协作集成安装器")
+    print("=" * 60)
+
+    if args.uninstall:
+        print("[UNINSTALL] 卸载模式...")
+        success = uninstall_iflow_integration()
+    elif args.verify:
+        print("[VERIFY] 验证模式...")
+        success = verify_installation()
+    elif args.install or len(sys.argv) == 1:
+        print("[INSTALL] 安装模式...")
+
+        # 1. 创建配置目录
+        print("Step 1. 创建配置目录...")
+        create_iflow_config_directory()
+
+        # 2. 安装hooks配置
+        print("Step 2. 安装hooks配置...")
+        hooks_success = install_iflow_hooks()
+
+        # 3. 复制适配器文件
+        print("Step 3. 复制适配器文件...")
+        adapter_success = copy_adapter_file()
+
+        success = hooks_success and adapter_success
+
+        if success:
+            print("\n🎉 iFlow CLI集成安装成功！")
+            print("\n[INFO] 安装摘要:")
+            print(f"   [OK] 配置目录: {IFLOW_CONFIG_DIR}")
+            print(f"   [OK] Hooks文件: {IFLOW_HOOKS_FILE}")
+            print(f"   [OK] 适配器目录: {os.path.join(IFLOW_CONFIG_DIR, 'adapters')}")
+            print(f"   [OK] 跨CLI协作: 已启用")
+            print(f"   [OK] 支持的CLI: claude, gemini, qwencode, qoder, codebuddy, copilot")
+        else:
+            print("\n❌ iFlow CLI集成安装失败！")
+    else:
+        parser.print_help()
+
+if __name__ == "__main__":
+    main()