AiDOPWarehouse/server/Plugins/Admin.NET.Plugin.AiDOP/DataPlatform/MdpSourceScopeFactory.cs

using Admin.NET.Core;
using Admin.NET.Plugin.AiDOP.Entity.DataPlatform;
using SqlSugar;

namespace Admin.NET.Plugin.AiDOP.DataPlatform;

/// <summary>
/// 按 mdp_source 动态获取只读 SqlSugar 连接域。
/// ConfigId 约定：mdp-src-{sourceCode}。连接只看 conn_mode 与 db_*，不按来源编码分支。
/// </summary>
public sealed class MdpSourceScopeFactory : ITransient
{
    private const string ConfigIdPrefix = "mdp-src-";
    private readonly ISqlSugarClient _db;

    public MdpSourceScopeFactory(ISqlSugarClient db)
    {
        _db = db;
    }

    /// <summary>按 sourceCode 获取连接域；优先复用已注册 ConfigId。</summary>
    public async Task<ISqlSugarClient> GetScopeAsync(string sourceCode, CancellationToken cancellationToken = default)
    {
        if (string.IsNullOrWhiteSpace(sourceCode))
            throw new ArgumentException("sourceCode 不能为空", nameof(sourceCode));

        var code = sourceCode.Trim();
        var source = await _db.Queryable<MdpSource>()
            .Where(x => x.SourceCode == code && x.Status == 1)
            .FirstAsync(cancellationToken)
            ?? throw new InvalidOperationException($"mdp_source 未找到启用源：{code}");

        return OpenScope(source);
    }

    /// <summary>
    /// 按 system_code 取唯一一条启用的 DB 源。0 条或多条都抛错，禁止取第一条。
    /// </summary>
    public async Task<ISqlSugarClient> GetScopeBySystemCodeAsync(string systemCode, CancellationToken cancellationToken = default)
    {
        if (string.IsNullOrWhiteSpace(systemCode))
            throw new ArgumentException("systemCode 不能为空", nameof(systemCode));

        var code = systemCode.Trim();
        var all = await _db.Queryable<MdpSource>()
            .Where(x => x.Status == 1 && x.SourceType == "DB")
            .ToListAsync(cancellationToken);
        var rows = all
            .Where(x => string.Equals(
                string.IsNullOrWhiteSpace(x.SystemCode) ? x.SourceCode : x.SystemCode,
                code,
                StringComparison.OrdinalIgnoreCase))
            .ToList();

        if (rows.Count == 0)
            throw new InvalidOperationException($"mdp_source 未找到启用的 DB 源：system_code={code}");
        if (rows.Count > 1)
            throw new InvalidOperationException(
                $"system_code={code} 命中多条 DB 源：{string.Join(",", rows.Select(x => x.SourceCode))}。请改用 GetScopeAsync(sourceCode)。");

        return OpenScope(rows[0]);
    }

    private ISqlSugarClient OpenScope(MdpSource source)
    {
        if (!string.Equals(source.SourceType, "DB", StringComparison.OrdinalIgnoreCase))
            throw new InvalidOperationException($"源 {source.SourceCode} 的 source_type={source.SourceType}，不是 DB，无法建 SqlSugar 连接");

        var mode = (source.ConnMode ?? "").Trim().ToUpperInvariant();
        if (mode == "SELF")
            return _db;
        if (mode != "EXTERNAL")
            throw new InvalidOperationException($"源 {source.SourceCode} 的 conn_mode='{source.ConnMode}' 无法解析。只接受 SELF / EXTERNAL。");

        if (string.IsNullOrWhiteSpace(source.DbHost) || string.IsNullOrWhiteSpace(source.DbName) || string.IsNullOrWhiteSpace(source.DbUser))
            throw new InvalidOperationException($"源 {source.SourceCode} 是 EXTERNAL，但缺少 db_host / db_name / db_user。请在数据源页面补齐，禁止回退主库。");

        var configId = ConfigIdPrefix + source.SourceCode;
        if (_db.AsTenant().IsAnyConnection(configId))
            return _db.AsTenant().GetConnectionScope(configId);

        var connStr = BuildConnectionString(source);
        var dbType = MapDbType(source.DbType);

        _db.AsTenant().AddConnection(new ConnectionConfig
        {
            ConfigId = configId,
            DbType = dbType,
            ConnectionString = connStr,
            InitKeyType = InitKeyType.Attribute,
            IsAutoCloseConnection = true,
            MoreSettings = new ConnMoreSettings
            {
                IsAutoRemoveDataCache = true
            }
        });

        return _db.AsTenant().GetConnectionScope(configId);
    }

    public ISqlSugarClient GetScope(string sourceCode) =>
        GetScopeAsync(sourceCode).GetAwaiter().GetResult();

    private static DbType MapDbType(string? dbType)
    {
        if (string.IsNullOrWhiteSpace(dbType)) return DbType.MySql;
        return dbType.Trim().ToUpperInvariant() switch
        {
            "MYSQL" => DbType.MySql,
            "SQLSERVER" or "MSSQL" => DbType.SqlServer,
            "ORACLE" => DbType.Oracle,
            "POSTGRES" or "POSTGRESQL" => DbType.PostgreSQL,
            _ => throw new NotSupportedException($"不支持的 db_type：{dbType}")
        };
    }

    private static string BuildConnectionString(MdpSource source)
    {
        if (string.IsNullOrWhiteSpace(source.DbHost) || string.IsNullOrWhiteSpace(source.DbName))
            throw new InvalidOperationException($"源 {source.SourceCode} 缺少 db_host/db_name");

        var password = DecryptPassword(source.DbPasswordEnc);
        var port = source.DbPort;
        var user = source.DbUser ?? "";
        var extra = string.IsNullOrWhiteSpace(source.DbExtraParams) ? "" : ";" + source.DbExtraParams.Trim().TrimStart(';');

        return MapDbType(source.DbType) switch
        {
            DbType.SqlServer =>
                $"Server={source.DbHost}{(port is > 0 ? $",{port}" : "")};Database={source.DbName};User Id={user};Password={password};TrustServerCertificate=true;Encrypt=false{extra}",
            DbType.MySql =>
                $"Server={source.DbHost};Port={(port is > 0 ? port : 3306)};Database={source.DbName};Uid={user};Pwd={password};CharSet=utf8mb4;AllowLoadLocalInfile=true{extra}",
            DbType.PostgreSQL =>
                $"Host={source.DbHost};Port={(port is > 0 ? port : 5432)};Database={source.DbName};Username={user};Password={password}{extra}",
            DbType.Oracle =>
                $"Data Source={source.DbHost}:{(port is > 0 ? port : 1521)}/{source.DbName};User Id={user};Password={password}{extra}",
            _ => throw new NotSupportedException($"不支持的 db_type：{source.DbType}")
        };
    }

    private static string DecryptPassword(string? enc)
    {
        if (string.IsNullOrEmpty(enc)) return "";
        try
        {
            var plain = CryptogramUtil.Decrypt(enc);
            return string.IsNullOrEmpty(plain) ? enc : plain;
        }
        catch
        {
            // 明文或非本系统密文时原样使用（与 Database.json EnableConnEncrypt=false 一致）
            return enc;
        }
    }
}