using KH.WMS.Core.Constants;
using KH.WMS.Core.Database.UnitOfWorks;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using SqlSugar;
using StackExchange.Profiling;

namespace KH.WMS.Core.Database.SqlSugar;

/// <summary>
/// SQL Sugar 配置
/// </summary>
public static class SqlSugarSetup
{
    /// <summary>
    /// 添加 SQL Sugar 服务
    /// </summary>
    public static IServiceCollection AddSqlSugar(this IServiceCollection services, IConfiguration configuration)
    {
        services.Configure<DatabaseOptions>(configuration.GetSection(AppSettingsConstants.DbConnection));

        Console.WriteLine("SqlSugar 正在初始化数据库连接...");

        services.AddSingleton<ISqlSugarClient>(provider =>
        {
            var options = provider.GetRequiredService<IOptions<DatabaseOptions>>().Value;
            var logger = provider.GetService<ILogger<SqlSugarClient>>();

            var config = new ConnectionConfig
            {
                ConnectionString = options.ConnectionString,
                DbType = GetDbType(options.DbType),
                IsAutoCloseConnection = true,
                InitKeyType = InitKeyType.Attribute,
                AopEvents = new AopEvents
                {
                    OnLogExecuting = (sql, pars) =>
                    {
                        var paramStr = string.Join(", ", pars.Select(p => $"{p.ParameterName}={p.Value}"));
                        Parallel.For(0, 1, e =>
                        {
                            MiniProfiler.Current.CustomTiming("SQL：", paramStr + "【SQL语句】：" + sql);
                        });

                        // 使用 Verbose 级别记录每一行 SQL 执行
                        if (options.EnableSqlLog && logger != null)
                        {
                            logger.LogTrace("[SQL执行] {Sql} | 参数: {Params}", sql, paramStr);
                        }

                        // 开发环境同时输出到控制台
                        if (options.EnableSqlLog)
                        {
                            Console.WriteLine($"SQL: {sql}\nPARAMS: {paramStr}");
                        }
                    }
                },
            };

            // 开发环境打印SQL
            if (options.EnableSqlLog)
            {
                config.ConfigureExternalServices = new ConfigureExternalServices
                {
                    //SqlFuncServices = new SqlFuncExternalService(),
                    EntityService = null,
                     //DataInfoCacheService =
                };
            }

            return new SqlSugarClient(config);
        });

        //services.AddScoped<IDbContext, SqlSugarDbContext>();
        //services.AddScoped<IUnitOfWork, UnitOfWork>();

        return services;
    }

    private static DbType GetDbType(string dbType)
    {
        return dbType.ToLower() switch
        {
            AppSettingsConstants.DbType_MySql => DbType.MySql,
            AppSettingsConstants.DbType_SqlServer => DbType.SqlServer,
            AppSettingsConstants.DbType_PostgreSql => DbType.PostgreSQL,
            AppSettingsConstants.DbType_Oracle => DbType.Oracle,
            AppSettingsConstants.DbType_Sqlite => DbType.Sqlite,
            _ => DbType.SqlServer
        };
    }
}

/// <summary>
/// 数据库配置选项
/// </summary>
public class DatabaseOptions
{
    /// <summary>
    /// 连接字符串
    /// </summary>
    public string ConnectionString { get; set; } = string.Empty;

    /// <summary>
    /// 数据库类型
    /// </summary>
    public string DbType { get; set; } = AppSettingsConstants.DbType_SqlServer;

    /// <summary>
    /// 是否启用SQL日志
    /// </summary>
    public bool EnableSqlLog { get; set; } = true;

    /// <summary>
    /// 命令超时时间（秒）
    /// </summary>
    public int CommandTimeout { get; set; } = 30;
}