"""
启动脚本 - 处理配置文件和目录初始化
"""

import os
import sys
import json
import shutil
from pathlib import Path


def get_exe_dir():
    """获取exe文件所在目录"""
    if getattr(sys, 'frozen', False):
        # 如果是打包后的exe文件
        return os.path.dirname(sys.executable)
    else:
        # 如果是开发环境
        return os.path.dirname(os.path.abspath(__file__))


def ensure_directories():
    """确保必要的目录存在"""
    exe_dir = get_exe_dir()
    
    # 创建日志目录
    logs_dir = os.path.join(exe_dir, 'logs')
    if not os.path.exists(logs_dir):
        os.makedirs(logs_dir)
        print(f"创建日志目录: {logs_dir}")
    
    return exe_dir


def ensure_config_file():
    """确保配置文件存在"""
    exe_dir = get_exe_dir()
    config_file = os.path.join(exe_dir, 'config.json')
    
    if not os.path.exists(config_file):
        # 如果config.json不存在，尝试复制生产配置模板
        production_config = os.path.join(exe_dir, 'config.production.json')
        example_config = os.path.join(exe_dir, 'config.example.json')
        
        if os.path.exists(production_config):
            shutil.copy2(production_config, config_file)
            print(f"复制生产配置文件: {production_config} -> {config_file}")
        elif os.path.exists(example_config):
            shutil.copy2(example_config, config_file)
            print(f"复制示例配置文件: {example_config} -> {config_file}")
        else:
            # 创建默认配置文件
            default_config = {
                "database": {
                    "url": "mysql+pymysql://root:password@localhost:3306/ecloud_dify"
                },
                "redis": {
                    "url": "redis://localhost:6379/0"
                },
                "ecloud": {
                    "base_url": "http://125.122.152.142:9899",
                    "authorization": "your_ecloud_authorization_token",
                    "w_id": "your_ecloud_w_id"
                },
                "dify": {
                    "base_url": "https://api.dify.ai/v1",
                    "api_key": "your_dify_api_key"
                },
                "server": {
                    "host": "0.0.0.0",
                    "port": 7979,
                    "debug": False
                },
                "logging": {
                    "level": "INFO",
                    "file": "logs/app.log"
                },
                "message_processing": {
                    "max_retry_count": 3,
                    "retry_delay": 5,
                    "queue_timeout": 300
                },
                "customer_service": {
                    "names": ["客服1", "客服2"]
                }
            }
            
            with open(config_file, 'w', encoding='utf-8') as f:
                json.dump(default_config, f, indent=2, ensure_ascii=False)
            print(f"创建默认配置文件: {config_file}")
    
    return config_file


def setup_environment():
    """设置运行环境"""
    exe_dir = ensure_directories()
    config_file = ensure_config_file()
    
    # 设置工作目录为exe所在目录
    os.chdir(exe_dir)
    print(f"设置工作目录: {exe_dir}")
    
    return exe_dir, config_file


if __name__ == "__main__":
    setup_environment()
    
    # 导入并启动主应用
    try:
        from main import app
        import uvicorn
        from config import settings
        from loguru import logger
        
        # 配置日志
        logger.add(
            settings.log_file,
            rotation="1 day",
            retention="7 days",
            level=settings.log_level,
            format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {name}:{function}:{line} - {message}",
        )
        
        logger.info("启动E云管家-DifyAI对接服务")
        
        # 启动服务
        uvicorn.run(
            app,
            host=settings.server_host,
            port=settings.server_port,
            log_level=settings.log_level.lower(),
        )
        
    except Exception as e:
        print(f"启动失败: {e}")
        input("按任意键退出...")
        sys.exit(1)