"""
数据库连接组件

提供数据库连接和查询接口，支持 SQL Server 和 MySQL
"""

from typing import List, Dict, Any, Optional
import sys
import os

# 添加项目根目录到 sys.path
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
if project_root not in sys.path:
    sys.path.insert(0, project_root)

from config.schema import DatabaseType
from db.connection_factory import ConnectionFactory
from db.base_connection import BaseDatabaseConnection


def get_connection(config=None) -> BaseDatabaseConnection:
    """
    获取数据库连接实例

    Args:
        config: 可选的数据库配置对象，默认从用户配置文件加载

    Returns:
        BaseDatabaseConnection: 数据库连接对象
    """
    if config is not None:
        # 使用提供的配置
        database_config = config
    else:
        # 从用户配置文件加载
        from config.loader import ConfigLoader
        app_config = ConfigLoader.load()
        database_config = app_config.database

    return ConnectionFactory.create_from_config(database_config)


def query_production_orders(总排号_list: List[str]) -> List[Dict[str, Any]]:
    """
    根据总排号列表查询生产订单号

    支持两种数据库格式：
    - SQL Server: [productionContractData].[26年压力表合同数据]
    - MySQL: productionContractData_26年压力表合同数据

    Args:
        总排号_list: 总排号列表

    Returns:
        List[Dict[str, Any]]: 查询结果
    """
    from db.table_name_converter import TableNameConverter
    from config.loader import ConfigLoader

    # 获取当前数据库类型
    app_config = ConfigLoader.load()
    db_type = app_config.database.db_type

    with get_connection() as db:
        # 获取正确的占位符
        placeholder = db.get_placeholder()

        # 构建占位符字符串
        placeholders = ",".join([placeholder for _ in 总排号_list])

        # 根据数据库类型选择表名格式
        if db_type == DatabaseType.MYSQL:
            table_name = "productionContractData_26年压力表合同数据"
            sql = f"""
            SELECT 总排号, 生产订单号, 序号, 订单号, 客户名称, 产品型号
            FROM {table_name}
            WHERE 总排号 IN ({placeholders})
            ORDER BY 序号
            """
        else:
            table_name = "[productionContractData].[26年压力表合同数据]"
            sql = f"""
            SELECT [总排号], [生产订单号], [序号], [订单号], [客户名称], [产品型号]
            FROM {table_name}
            WHERE [总排号] IN ({placeholders})
            ORDER BY [序号]
            """

        results = db.execute_query(sql, tuple(总排号_list))
        return results