"""
数据库连接工厂

根据配置创建对应数据库类型的连接实例
"""

from typing import Dict, Any, Optional
from config.schema import DatabaseType
from db.base_connection import BaseDatabaseConnection
from db.sqlserver_connection import SQLServerConnection
from db.mysql_connection import MySQLConnection


class ConnectionFactory:
    """数据库连接工厂类"""

    @staticmethod
    def create_connection(
        db_type: DatabaseType, config: Optional[Dict[str, Any]] = None
    ) -> BaseDatabaseConnection:
        """
        根据数据库类型创建对应的连接实例

        Args:
            db_type: 数据库类型（SQLSERVER 或 MYSQL）
            config: 数据库配置字典

        Returns:
            对应数据库的连接实例

        Raises:
            ValueError: 不支持的数据库类型
        """
        if db_type == DatabaseType.SQLSERVER:
            return SQLServerConnection(config)
        elif db_type == DatabaseType.MYSQL:
            return MySQLConnection(config)
        else:
            raise ValueError(f"不支持的数据库类型: {db_type}")

    @staticmethod
    def create_from_config(database_config) -> BaseDatabaseConnection:
        """
        从 DatabaseConfig 配置对象创建连接

        Args:
            database_config: DatabaseConfig 配置对象

        Returns:
            对应数据库的连接实例

        Raises:
            ValueError: 不支持的数据库类型
        """
        db_type = database_config.db_type

        if db_type == DatabaseType.SQLSERVER:
            # 构建 SQL Server 配置字典
            config = {
                "server": database_config.server,
                "database": database_config.database,
                "username": database_config.username,
                "password": database_config.password,
            }
            if database_config.sqlserver:
                config["driver"] = database_config.sqlserver.driver
                config["trust_server_certificate"] = (
                    database_config.sqlserver.trust_server_certificate
                )
            return SQLServerConnection(config)

        elif db_type == DatabaseType.MYSQL:
            # 构建 MySQL 配置字典
            config = {
                "database": database_config.database,
                "username": database_config.username,
                "password": database_config.password,
            }
            if database_config.mysql:
                config["host"] = database_config.mysql.host
                config["port"] = database_config.mysql.port
                config["charset"] = database_config.mysql.charset
            else:
                # 回退到 server 字段（兼容旧配置）
                config["host"] = database_config.server
            return MySQLConnection(config)

        else:
            raise ValueError(f"不支持的数据库类型: {db_type}")