import os
import shutil
import sys
import subprocess
from pathlib import Path
import importlib.util

def install_package():
    print("正在安装/更新 cnhkmcp 包...")
    try:
        subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "cnhkmcp"])
    except subprocess.CalledProcessError as e:
        print(f"安装失败: {e}")
        sys.exit(1)

def get_package_path(package_name):
    # 重新加载 finder 以防刚安装完
    importlib.invalidate_caches()
    spec = importlib.util.find_spec(package_name)
    if spec and spec.origin:
        return Path(spec.origin).parent
    return None

def main():
    # 1. Install CNHKMCP
    install_package()

    # 2. Locate the package
    pkg_path = get_package_path("cnhkmcp")
    if not pkg_path:
        # Fallback: try site-packages directly if spec fails
        import site
        for site_pkg in site.getsitepackages():
             potential_path = Path(site_pkg) / "cnhkmcp"
             if potential_path.exists():
                 pkg_path = potential_path
                 break
    
    if not pkg_path:
        print("错误: 无法找到 cnhkmcp 包安装路径。")
        sys.exit(1)
        
    print(f"cnhkmcp 安装位置: {pkg_path}")

    # 3. Locate 'untracked' folder
    untracked_dir = pkg_path / "untracked"
    if not untracked_dir.exists():
        print(f"错误: 在 {pkg_path} 中未找到 'untracked' 文件夹")
        sys.exit(1)

    # 4. Copy Files
    files_to_copy = [
        "forum_functions.py",
        "platform_functions.py",
        "配置前运行我_安装必要依赖包.py",
        "brain-consultant.md"
    ]
    
    # Copy to current directory
    target_dir = Path.cwd()
    print(f"正在将文件复制到: {target_dir}")
    
    for filename in files_to_copy:
        src = untracked_dir / filename
        dst = target_dir / filename
        
        if src.exists():
            shutil.copy2(src, dst)
            print(f"✅ 已复制: {filename}")
        else:
            print(f"⚠️ 警告: 源文件未找到: {filename}")

    # 4.1 Copy entire 'untracked' folder to current directory for reference
    target_untracked_dir = target_dir / "untracked"
    print(f"\n正在将整个 untracked 文件夹复制到: {target_untracked_dir}")
    try:
        shutil.copytree(untracked_dir, target_untracked_dir, dirs_exist_ok=True)
        print(f"✅ untracked 文件夹已完整复制到当前目录！")
    except Exception as e:
        print(f"❌ 复制 untracked 文件夹失败: {e}")

    # 4.5 Copy 'skills' folder to user home .claude dir
    src_skills = untracked_dir / "skills"
    # Destination: ~/.claude/skills
    dst_skills = Path.home() / ".claude" / "skills"
    
    if src_skills.exists() and src_skills.is_dir():
        print(f"\n正在将 skills 文件夹复制到: {dst_skills}")
        try:
            # Ensure parent .claude dir exists
            if not dst_skills.parent.exists():
                dst_skills.parent.mkdir(parents=True, exist_ok=True)
            
            # Copy recursively
            shutil.copytree(src_skills, dst_skills, dirs_exist_ok=True)
            print(f"✅ Skills 文件夹已同步成功！")
        except Exception as e:
            print(f"❌ 复制 Skills 文件夹失败: {e}")
    else:
        print(f"\n(未在源包中找到 'skills' 文件夹，本次跳过复制)")

    # 4.6 Move 'settings.json' to .claude folder (Config for Claude Code)
    settings_file = target_dir / "settings.json"
    dst_claude_dir = Path.home() / ".claude"
    dst_settings = dst_claude_dir / "settings.json"

    if settings_file.exists():
        print(f"\n正在将 settings.json 移动(剪切)到: {dst_claude_dir}")
        try:
            if not dst_claude_dir.exists():
                dst_claude_dir.mkdir(parents=True, exist_ok=True)
            
            if dst_settings.exists():
                print(f"⚠️ 目标路径已存在同名文件，将被覆盖: {dst_settings}")
                dst_settings.unlink()
                
            shutil.move(str(settings_file), str(dst_settings))
            print(f"✅ settings.json 已成功移动到 .claude 目录")
        except Exception as e:
            print(f"❌ 移动 settings.json 失败: {e}")

    # 4.7 Move 'hooks' folder to .claude folder
    hooks_src = target_dir / "hooks"
    dst_hooks = dst_claude_dir / "hooks"

    if hooks_src.exists() and hooks_src.is_dir():
        print(f"\n正在将 hooks 文件夹移动(剪切)到: {dst_claude_dir}")
        try:
            if not dst_claude_dir.exists():
                dst_claude_dir.mkdir(parents=True, exist_ok=True)

            if dst_hooks.exists():
                print(f"⚠️ 目标路径已存在 hooks 文件夹，将被移除以进行覆盖...")
                shutil.rmtree(dst_hooks)
            
            shutil.move(str(hooks_src), str(dst_hooks))
            print(f"✅ hooks 文件夹已成功移动到 .claude 目录")
        except Exception as e:
            print(f"❌ 移动 hooks 文件夹失败: {e}")

    print("\n所有文件更新完成！")
    
    # 5. Execute the dependency installer script
    installer_script = "配置前运行我_安装必要依赖包.py"
    if (target_dir / installer_script).exists():
        print(f"\n正在运行依赖安装脚本: {installer_script} ...")
        sys.stdout.flush()
        try:
            # Use same python executable to run the installer script
            subprocess.check_call([sys.executable, str(target_dir / installer_script)])
            print("✅ 依赖安装脚本执行成功！")
        except subprocess.CalledProcessError as e:
            print(f"❌ 依赖安装脚本执行失败: {e}")
            sys.exit(1)
    else:
         print(f"⚠️ 未找到 {installer_script}，跳过依赖安装。")

if __name__ == "__main__":
    main()