# -*- mode: python ; coding: utf-8 -*-

import os
from pathlib import Path
from PyInstaller.utils.hooks import collect_data_files, collect_submodules, collect_dynamic_libs

# 获取项目根目录（spec 文件所在目录）
import sys
_REPO_DIR = Path(os.getcwd()).resolve()

# 根目录执行：pyinstaller --clean main.spec
# 收集配置文件夹和数据文件
datas = [
    # 所有数据文件都不打包，从外部读取
    # ('.env', '.'),
    # ('data/schemas', 'data/schemas'),
    # ('data/embeddings', 'data/embeddings'),
    # ('data/experiences', 'data/experiences'),
]

# ChromaDB 在 Windows 单文件 exe 下如缺少 native 动态库，可能在持久化访问时直接“硬退出”
# 这里显式收集 chromadb 与 hnswlib（向量索引）的动态库/二进制依赖
binaries = []
try:
    binaries += collect_dynamic_libs("chromadb")
except Exception:
    pass
try:
    binaries += collect_dynamic_libs("hnswlib")
except Exception:
    pass

# 某些 chromadb 版本会依赖包内数据文件（如 migrations/默认配置等），一并收集更稳妥
try:
    datas += collect_data_files("chromadb")
except Exception:
    pass

# 收集所有需要的隐藏导入
hiddenimports = [
    # Web 框架
    'fastapi',
    'fastapi.middleware.cors',
    'fastapi.responses',
    'uvicorn',
    'uvicorn.lifespan.on',
    'uvicorn.config',
    'uvicorn.protocols.http.auto',
    'uvicorn.protocols.websockets.auto',
    'uvicorn.loops.auto',
    
    # Backend 模块（重要！）
    'main',  # backend/main.py
    'backend.main',
    'backend.nl_lite_store',
    'backend.agents',
    'backend.agents.orchestrator',
    'backend.agents.schema_linker',
    'backend.agents.sql_generator',
    'backend.agents.validator',
    'backend.config',
    'backend.config.prompts',
    'backend.config.settings',
    'backend.db',
    'backend.db.engine',
    'backend.db.dbhub_tools',
    'backend.db.dbhub_sql_parser',
    'backend.db.dbhub_allowed_keywords',
    'backend.llm',
    'backend.llm.deepseek_client',
    'backend.schema',
    'backend.schema.manager',
    'backend.schema.loader',
    'backend.schema.indexer',
    'backend.schema.models',
    'backend.utils',
    'backend.utils.dialog_classifier',
    'backend.utils.dialog_context',
    'backend.utils.embedding',
    'backend.utils.fewshot_selector',
    'backend.utils.question_locale',
    'backend.utils.sql_parser',
    'backend.utils.validators',
    
    # 数据库相关
    'pymssql',
    'sqlalchemy',
    'sqlalchemy.dialects.mssql.pymssql',
    'sqlalchemy.dialects.mysql',
    'sqlalchemy.dialects.postgresql',
    'sqlglot',
    'sqlglot.dialects',
    'sqlparse',
    
    # 向量数据库 - 使用 collect_submodules 自动收集所有子模块
    
    # Embedding 相关
    'sentencepiece',
    'accelerate',
    'safetensors',
    
    # LLM API
    'openai',
    
    # 数据处理
    'numpy',
    'jieba',
    
    # Pydantic
    'pydantic',
    'pydantic_settings',
    'dotenv',
    
    # 工具库
    'tenacity',
    'structlog',
    'tqdm',
    'camel_ai',
    
    # 标准库中可能被遗漏的
    'email.mime.text',
    'email.mime.multipart',
] + collect_submodules('chromadb')
# 排除不必要的模块以减小体积
excludes = [
    'tkinter',
    'matplotlib',
    'scipy',
    'pandas.tests',
    'numpy.testing',
    'jupyter',
    'notebook',
    'IPython',
    'pytest',
]

a = Analysis(
    ['api_server.py'],
    pathex=[str(_REPO_DIR)],
    binaries=binaries,
    datas=datas,
    hiddenimports=hiddenimports,
    hookspath=[],
    hooksconfig={},
    runtime_hooks=[],
    excludes=excludes,
    noarchive=False,
    optimize=0,
)
pyz = PYZ(a.pure)

exe = EXE(
    pyz,
    a.scripts,
    a.binaries,
    a.datas,  # 单文件模式：将数据文件打包进 exe
    [],
    name='Backman_Text2SQL_Server',
    debug=False,
    bootloader_ignore_signals=False,
    strip=False,
    upx=False,  # 禁用 UPX，避免兼容性问题
    upx_exclude=[],
    runtime_tmpdir=None,
    console=True,
    disable_windowed_traceback=False,
    argv_emulation=False,
    target_arch=None,
    codesign_identity=None,
    entitlements_file=None,
)