import os
import sys
import shutil
import subprocess
import json
from typing import Dict

def check_and_install_packages(packages_with_versions: Dict[str, str]):
    """检查并安装指定版本的包"""
    print("正在检查并安装必要的依赖包...")
    
    for package, version in packages_with_versions.items():
        try:
            # 尝试导入包来检查是否已安装
            __import__(package.replace('-', '_').replace('.', '_'))
            print(f"✓ {package} 已安装")
        except ImportError:
            print(f"正在安装 {package}=={version}...")
            try:
                # 对于win11toast包特殊处理，忽略安装失败
                if package == 'win11toast':
                    try:
                        subprocess.run(
                            [sys.executable, "-m", "pip", "install", f"{package}=={version}"], 
                            check=False
                        )
                        print(f"✓ {package}=={version} 安装成功")
                    except:
                        print(f"! {package} 安装失败，但这不会影响程序核心功能，继续打包")
                else:
                    # 其他包正常安装
                    subprocess.run(
                        [sys.executable, "-m", "pip", "install", f"{package}=={version}"], 
                        check=True
                    )
                    print(f"✓ {package}=={version} 安装成功")
            except subprocess.CalledProcessError as e:
                print(f"! 无法安装 {package}=={version}: {str(e)}")
                print(f"  尝试安装不指定版本的 {package}...")
                try:
                    subprocess.run(
                        [sys.executable, "-m", "pip", "install", package], 
                        check=True
                    )
                    print(f"✓ {package} 安装成功")
                except subprocess.CalledProcessError as e2:
                    print(f"! 安装 {package} 失败: {str(e2)}")
                    if package == 'win11toast':
                        print(f"  {package} 安装失败，但这不会影响程序核心功能")
                    else:
                        return False
    return True

def create_notification_alternative():
    """创建替代win11toast的通知实现文件"""
    print("正在创建通知替代实现...")
    
    # 在dist目录下创建一个替代的通知模块
    notification_dir = os.path.join('dist', 'win11toast')
    if not os.path.exists(notification_dir):
        os.makedirs(notification_dir)
    
    # 创建__init__.py
    with open(os.path.join(notification_dir, '__init__.py'), 'w', encoding='utf-8') as f:
        f.write('''
# 替代win11toast的简易实现
import ctypes
import threading
import time

def toast(title, message, **kwargs):
    """显示一个Windows通知，使用MessageBox替代"""
    def show():
        ctypes.windll.user32.MessageBoxW(0, message, title, 0x40)
    threading.Thread(target=show, daemon=True).start()
    
def notify(title, message, **kwargs):
    """显示一个Windows通知，使用MessageBox替代"""
    toast(title, message, **kwargs)
''')
    
    # 创建空的__pycache__目录以避免警告
    pycache_dir = os.path.join(notification_dir, '__pycache__')
    if not os.path.exists(pycache_dir):
        os.makedirs(pycache_dir)
    
    print("通知替代实现创建完成")

def get_installed_packages() -> Dict[str, str]:
    """获取当前已安装的包版本信息"""
    result = {}
    try:
        output = subprocess.check_output([sys.executable, "-m", "pip", "freeze"]).decode('utf-8')
        for line in output.split('\n'):
            if '==' in line:
                package, version = line.strip().split('==', 1)
                result[package] = version
    except Exception as e:
        print(f"获取已安装包信息时出错: {str(e)}")
    return result

def write_requirements_file(packages_with_versions: Dict[str, str]):
    """生成requirements.txt文件"""
    print("正在生成requirements.txt文件...")
    with open('requirements.txt', 'w', encoding='utf-8') as f:
        for package, version in packages_with_versions.items():
            f.write(f"{package}=={version}\n")
    print("requirements.txt文件已生成")

def create_manifest_file():
    """创建应用程序清单文件，请求管理员权限"""
    print("正在创建应用程序清单文件...")
    manifest_content = '''<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
<assembly xmlns="urn:schemas-microsoft-com:asm.v1" manifestVersion="1.0">
    <trustInfo xmlns="urn:schemas-microsoft-com:asm.v3">
        <security>
            <requestedPrivileges>
                <requestedExecutionLevel level="requireAdministrator" uiAccess="false"/>
            </requestedPrivileges>
        </security>
    </trustInfo>
</assembly>'''
    
    with open('app.manifest', 'w', encoding='utf-8') as f:
        f.write(manifest_content)
    print("应用程序清单文件已创建")

def create_spec_file(sv_ttk_path: str):
    """创建PyInstaller spec文件"""
    print("正在创建PyInstaller spec文件...")
    
    # 这里列出所有需要的隐藏导入
    hidden_imports = [
        'sv_ttk',
        'keyboard',
        'mouse',
        'win32gui',
        'win32process',
        'win32con',
        'win32api',
        'win32com.client',
        'json',
        'requests',
        'math',
        'ctypes',
        'threading',
        'time',
        'webbrowser',
        're',
        'traceback',
        'wmi',
        'pythoncom',
        'concurrent.futures',
        'winreg',
        'win11toast'  # 总是包含win11toast，即使安装失败也不影响
    ]
    
    # 创建spec文件内容
    spec_content = f'''# -*- mode: python ; coding: utf-8 -*-

a = Analysis(
    ['chrome_manager.py'],
    pathex=[],
    binaries=[],
    datas=[
        ('app.ico', '.'),
        (r'{sv_ttk_path}', 'sv_ttk'),
        ('README.md', '.'),
        ('settings.json', '.') if os.path.exists('settings.json') else None,
    ],
    hiddenimports={hidden_imports},
    hookspath=[],
    hooksconfig={{}},
    runtime_hooks=[],
    excludes=[],
    noarchive=False,
)

# 移除None值
a.datas = [x for x in a.datas if x is not None]

pyz = PYZ(a.pure)

exe = EXE(
    pyz,
    a.scripts,
    a.binaries,
    a.zipfiles,
    a.datas,
    [('app.manifest', 'app.manifest', 'DATA')],
    name='chrome_manager',
    debug=False,
    bootloader_ignore_signals=False,
    strip=False,
    upx=False,  # 禁用UPX压缩，防止被杀毒软件误报
    console=False,
    icon=['app.ico'],
    manifest="app.manifest",
    uac_admin=True,  # 添加UAC管理员请求
    uac_uiaccess=False,
    disable_windowed_traceback=False,
)
'''
    
    with open('chrome_manager.spec', 'w', encoding='utf-8') as f:
        f.write(spec_content)
    print("PyInstaller spec文件已创建")

def find_sv_ttk_path():
    try:
        import sv_ttk
        return os.path.dirname(sv_ttk.__file__)
    except ImportError:
        print("未找到sv_ttk模块，请先安装")
        return None

def ensure_icon_exists():
    if not os.path.exists('app.ico'):
        print("警告: 未找到app.ico文件，将使用默认图标")
        # 可以考虑生成一个简单的图标或从网络下载一个
        try:
            # 尝试从Windows系统中复制一个默认图标
            shutil.copy(os.path.join(os.environ['SystemRoot'], 'System32', 'shell32.dll'), 'temp_icon.dll')
            subprocess.run(['powershell', '-Command', 
                           "(New-Object -ComObject Shell.Application).NameSpace(0).ParseName('temp_icon.dll').GetLink.GetIconLocation() | Out-File -FilePath 'app.ico'"],
                           check=True)
            os.remove('temp_icon.dll')
        except Exception as e:
            print(f"生成默认图标失败: {str(e)}")
            print("将使用PyInstaller默认图标")

def ensure_settings_exists():
    """确保settings.json文件存在"""
    if not os.path.exists('settings.json'):
        print("正在创建默认settings.json文件...")
        default_settings = {
            "shortcut_path": "",
            "cache_dir": "",
            "icon_dir": "",
            "screen_selection": "",
            "sync_shortcut": None,
            "window_position": {"x": 100, "y": 100}
        }
        with open('settings.json', 'w', encoding='utf-8') as f:
            json.dump(default_settings, f, ensure_ascii=False, indent=4)
        print("默认settings.json文件已创建")

def modify_chrome_manager_for_win11toast():
    """修改chrome_manager.py中的通知相关代码，添加简单的try-except处理"""
    print("检查chrome_manager.py是否需要修改通知实现...")
    
    try:
        with open('chrome_manager.py', 'r', encoding='utf-8') as f:
            content = f.read()
            
        # 如果已经有错误处理，则不需要修改
        if "try:" in content and "from win11toast import notify, toast" in content and "except ImportError:" in content:
            print("chrome_manager.py已包含通知错误处理")
            return
            
        # 查找win11toast导入行
        if "from win11toast import notify, toast" in content:
            # 替换成带错误处理的版本
            original = "from win11toast import notify, toast"
            replacement = '''# 添加通知错误处理
try:
    from win11toast import notify, toast
except ImportError:
    # 简单的空函数替代
    def toast(title, message, **kwargs):
        pass
    def notify(title, message, **kwargs):
        pass'''
            
            modified_content = content.replace(original, replacement)
            
            with open('chrome_manager.py', 'w', encoding='utf-8') as f:
                f.write(modified_content)
                
            print("成功添加通知错误处理到chrome_manager.py")
        else:
            print("未找到win11toast导入行，跳过修改")
    except Exception as e:
        print(f"修改chrome_manager.py失败: {str(e)}")
        print("继续打包过程...")

def show_success_message():
    print("\n")
    print("─────────────────────────────────────────────────────")
    print("                                                     ")
    print("       ✨  Chrome多窗口管理器 V2.0 打包成功  ✨       ")
    print("                                                     ")
    print("  📦 可执行文件已生成到dist文件夹                      ")
    print("  🚀 双击chrome_manager.exe即可运行                   ")
    print("  🔑 首次打开程序时间会稍长，请耐心等待                 ")
    print("  🌐 使用中遇到问题，请查阅Github中的说明               ")
    print("                                                     ")
    print("─────────────────────────────────────────────────────")
    print("\n")

def show_failure_message(error_msg="未知错误"):
    print("\n")
    print("─────────────────────────────────────────────────────")
    print("                                                     ")
    print("       ❌  Chrome多窗口管理器 V2.0 打包失败  ❌       ")
    print("                                                     ")
    print("  ❗ 打包过程中出现错误                              ")
    print(f"  📋 错误信息: {error_msg[:35]}{'...' if len(error_msg) > 35 else ''}")
    print("  🔄 请查阅Github中的说明                              ")
    print("  💡 然后尝试重新运行打包程序                             ")
    print("                                                     ")
    print("─────────────────────────────────────────────────────")
    print("\n")

def build():
    """打包程序"""
    print("\n===== 开始打包Chrome多窗口管理器 V2.0 =====\n")
    
    # 修改chrome_manager.py添加简单的错误处理
    modify_chrome_manager_for_win11toast()
    
    # 需要的包和版本列表
    required_packages = {
        'pyinstaller': '6.12.0',
        'sv-ttk': '2.6.0',
        'keyboard': '0.13.5',
        'mouse': '0.7.1',
        'pywin32': '309',
        'wmi': '1.5.1',
        'requests': '2.32.3',
        'pillow': '11.1.0',
        'win11toast': '0.32',  # 包含win11toast但允许安装失败
    }
    
    # 获取当前已安装的包
    installed_packages = get_installed_packages()
    
    # 更新为实际安装的版本
    for package in required_packages:
        if package in installed_packages:
            required_packages[package] = installed_packages[package]
    
    # 检查并安装必要的包
    if not check_and_install_packages(required_packages):
        print("安装必要的包失败，尝试继续打包...")
    
    # 创建requirements.txt文件
    write_requirements_file(required_packages)
    
    # 确保其他必要文件存在
    ensure_icon_exists()
    ensure_settings_exists()
    
    # 创建清单文件
    create_manifest_file()
    
    # 查找sv_ttk路径
    sv_ttk_path = find_sv_ttk_path()
    if not sv_ttk_path:
        print("无法找到sv_ttk模块，打包终止")
        show_failure_message("未找到sv_ttk模块")
        return False
    
    # 创建spec文件
    create_spec_file(sv_ttk_path)
    
    # 清理旧的构建文件
    print("正在清理旧的构建文件...")
    if os.path.exists('build'):
        shutil.rmtree('build')
    if os.path.exists('dist'):
        shutil.rmtree('dist')
    
    # 运行PyInstaller
    print("\n正在打包程序...")
    try:
        # 当使用 .spec 文件时，不应在命令行传递 --clean 或 --noupx 等选项
        # 这些选项应在 spec 文件内配置，或由脚本本身处理（如清理目录）
        subprocess.run(['pyinstaller', 'chrome_manager.spec'], check=True)
        print("\n打包完成！程序文件在dist文件夹中。")
        
        # 复制额外文件到dist目录
        if not os.path.exists(os.path.join('dist', 'settings.json')) and os.path.exists('settings.json'):
            shutil.copy('settings.json', os.path.join('dist', 'settings.json'))
        
        show_success_message()
        return True
    except subprocess.CalledProcessError as e:
        error_msg = str(e)
        show_failure_message(error_msg)
        return False

if __name__ == "__main__":
    try:
        success = build()
    except Exception as e:
        show_failure_message(str(e))
    finally:
        input("\n按回车键退出...")