use deadpool_postgres::{Config as PgConfig, Pool, Runtime};
use tokio_postgres::NoTls;
use std::env;

/// 应用配置
#[derive(Clone, Debug)]
pub struct AppConfig {
    pub host: String,
    pub port: u16,
    pub jwt_secret: String,
    pub admin_jwt_secret: String,
    pub db_host: String,
    pub db_port: u16,
    pub db_name: String,
    pub db_user: String,
    pub db_password: String,
    pub db_pool_size: usize,
    /// 内部服务调用密钥（PHP注册服务、上链服务使用）
    pub internal_api_key: String,
}

impl AppConfig {
    pub fn from_env() -> Result<Self, String> {
        Ok(AppConfig {
            host: env::var("WALLET_HOST").unwrap_or_else(|_| "127.0.0.1".to_string()),
            port: env::var("WALLET_PORT")
                .unwrap_or_else(|_| "8701".to_string())
                .parse()
                .map_err(|_| "WALLET_PORT必须是有效端口号")?,
            jwt_secret: env::var("JWT_SECRET")
                .map_err(|_| "JWT_SECRET环境变量未设置")?,
            admin_jwt_secret: env::var("ADMIN_JWT_SECRET")
                .map_err(|_| "ADMIN_JWT_SECRET环境变量未设置")?,
            db_host: env::var("PG_HOST").unwrap_or_else(|_| "127.0.0.1".to_string()),
            db_port: env::var("PG_PORT")
                .unwrap_or_else(|_| "5432".to_string())
                .parse()
                .map_err(|_| "PG_PORT必须是有效端口号")?,
            db_name: env::var("PG_DBNAME").unwrap_or_else(|_| "nac_wallet".to_string()),
            db_user: env::var("PG_USER").unwrap_or_else(|_| "nac_wallet_user".to_string()),
            db_password: env::var("PG_PASSWORD")
                .map_err(|_| "PG_PASSWORD环境变量未设置")?,
            db_pool_size: env::var("DB_POOL_SIZE")
                .unwrap_or_else(|_| "10".to_string())
                .parse()
                .unwrap_or(10),
            internal_api_key: env::var("INTERNAL_API_KEY")
                .map_err(|_| "INTERNAL_API_KEY环境变量未设置")?,
        })
    }
}

/// 创建PostgreSQL连接池
pub async fn create_db_pool(config: &AppConfig) -> Result<Pool, String> {
    let mut pg_config = PgConfig::new();
    pg_config.host = Some(config.db_host.clone());
    pg_config.port = Some(config.db_port);
    pg_config.dbname = Some(config.db_name.clone());
    pg_config.user = Some(config.db_user.clone());
    pg_config.password = Some(config.db_password.clone());

    let mut pool_config = deadpool_postgres::PoolConfig::new(config.db_pool_size);
    pool_config.timeouts.wait = Some(std::time::Duration::from_secs(5));
    pg_config.pool = Some(pool_config);

    pg_config
        .create_pool(Some(Runtime::Tokio1), NoTls)
        .map_err(|e| format!("数据库连接池创建失败: {}", e))
}