import argparse
import os
import sys
from dotenv import load_dotenv
from functools import lru_cache
from pydantic import computed_field
from pydantic_settings import BaseSettings
from typing import Literal



class GetConfig:

    def __init__(self):
        self.parse_cli_args()

    @lru_cache()
    def get_llm_config(self):
        return LlmSettings()

    @lru_cache()
    def get_emb_config(self):
        return EmbSettings()

    @lru_cache()
    def get_mysql_database_config(self):
        """
        获取数据库配置
        """
        # 实例化数据库配置模型
        return MysqlDataBaseSettings()

    @lru_cache()
    def get_pgvector_database_config(self):
        """
        获取数据库配置
        """
        # 实例化数据库配置模型
        return PgvectorDataBaseSettings()

    @staticmethod
    def parse_cli_args():
        """
        解析命令行参数
        """
        if 'uvicorn' in sys.argv[0]:
            # 使用uvicorn启动时，命令行参数需要按照uvicorn的文档进行配置，无法自定义参数
            pass
        else:
            # 使用argparse定义命令行参数
            parser = argparse.ArgumentParser(description='命令行参数')
            parser.add_argument('--env', type=str, default='', help='运行环境')
            # 解析命令行参数
            args = parser.parse_args()
            # 设置环境变量，如果未设置命令行参数，默认APP_ENV为dev
            os.environ['APP_ENV'] = args.env if args.env else 'dev'
        # 读取运行环境
        run_env = os.environ.get('APP_ENV', '')
        # 运行环境未指定时默认加载.env.dev
        env_file = '.env.dev'
        # 运行环境不为空时按命令行参数加载对应.env文件
        if run_env != '':
            env_file = f'.env.{run_env}'
        # 加载配置
        load_dotenv(env_file)
        print(f'当前运行环境为：{run_env}')

# 实例化获取配置类
get_config = GetConfig()

class LlmSettings:
    """
    AI配置
    """
    base_url: str = os.getenv('llm_base_url', '')
    api_key: str = os.getenv('llm_api_key', '')
    model_name: str = os.getenv('llm_model', '')

class EmbSettings:
    """
    AI配置
    """
    base_url: str = os.getenv('emb_base_url', '')
    api_key: str = os.getenv('emb_api_key', '')
    model_name: str = os.getenv('emb_model', '')

class MysqlDataBaseSettings:
    """
    数据库配置
    """
    # 数据库配置
    db_host: str = os.getenv('MYSQL_DB_HOST', '')
    db_port: int = int(os.getenv('MYSQL_DB_PORT', 33306))
    db_username: str = os.getenv('MYSQL_DB_USERNAME', '')
    db_password: str = os.getenv('MYSQL_DB_PASSWORD', '')
    db_database: str = os.getenv('MYSQL_DB_DATABASE', '')

    db_echo: bool = True
    db_max_overflow: int = 10
    db_pool_size: int = 50
    db_pool_recycle: int = 3600
    db_pool_timeout: int = 30

class PgvectorDataBaseSettings:
    """
    数据库配置
    """
    # 数据库配置
    db_host: str = os.getenv('PG_DB_HOST', '')
    db_port: int = os.getenv('PG_DB_PORT', 5432)
    db_username: str = os.getenv('PG_DB_USERNAME', '')
    db_password: str = os.getenv('PG_DB_PASSWORD', '')
    db_database: str = os.getenv('PG_DB_DATABASE', '')

    db_echo: bool = True
    db_max_overflow: int = 10
    db_pool_size: int = 50
    db_pool_recycle: int = 3600
    db_pool_timeout: int = 30


# mysql数据库配置
MysqlDataBaseConfig = get_config.get_mysql_database_config()
# pgvector数据库配置
PgvectorDataBaseConfig = get_config.get_pgvector_database_config()

LlmBaseConfig = get_config.get_llm_config()
EmbBaseConfig = get_config.get_emb_config()