using Gpulse.WCT.DataAnalyzer.Core.Infrastructure.Extensions;
using Gpulse.WCT.DataAnalyzer.Core.Infrastructure.LocalData;
using Microsoft.Data.Sqlite;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;

namespace Gpulse.WCT.DataAnalyzer.Tests;

/// <summary>
/// CLI 测试宿主：用内存 SQLite 搭建与 Program.Main 一致的服务容器。
/// 每个测试实例持有独立的「本地库」/「发布库」连接与一个临时目录，互不污染。
/// </summary>
public sealed class CliTestHost : IDisposable
{
    private readonly SqliteConnection _localConnection;
    private readonly SqliteConnection _releaseConnection;
    private readonly ServiceProvider _provider;
    private readonly IServiceScope _scope;

    /// <summary>测试作用域的服务提供者（与命令执行共用同一作用域）。</summary>
    public IServiceProvider Services => _scope.ServiceProvider;

    /// <summary>每个测试独立的临时目录，用于放置日志 fixture 文件。</summary>
    public string TempDir { get; }

    public WctMinerDbContext LocalDb => Services.GetRequiredService<WctMinerDbContext>();
    public ReleaseDbContext ReleaseDb => Services.GetRequiredService<ReleaseDbContext>();

    private CliTestHost(
        SqliteConnection localConnection,
        SqliteConnection releaseConnection,
        ServiceProvider provider,
        IServiceScope scope,
        string tempDir)
    {
        _localConnection = localConnection;
        _releaseConnection = releaseConnection;
        _provider = provider;
        _scope = scope;
        TempDir = tempDir;
    }

    /// <summary>
    /// 创建测试宿主：本地库/发布库各用一条独立的内存 SQLite 连接（镜像生产环境的两个库文件），
    /// 并做 EnsureCreated 与种子数据初始化。
    /// </summary>
    public static async Task<CliTestHost> CreateAsync()
    {
        var tempDir = Directory.CreateTempSubdirectory("wct-tests-").FullName;

        // 两个库各自独立连接，避免共享连接时 EnsureCreated 相互干扰
        var localConnection = new SqliteConnection("Data Source=:memory:");
        await localConnection.OpenAsync();

        var releaseConnection = new SqliteConnection("Data Source=:memory:");
        await releaseConnection.OpenAsync();

        var configuration = new ConfigurationBuilder()
            .AddInMemoryCollection(new Dictionary<string, string?>
            {
                ["Database:Type"] = "sqlite",
                ["Parser:BatchSize"] = "1000",
            })
            .Build();

        var services = new ServiceCollection();
        services.AddSingleton<IConfiguration>(configuration);
        services.AddLogging();
        services.AddDbContext<WctMinerDbContext>(o => o.UseSqlite(localConnection));
        services.AddDbContext<ReleaseDbContext>(o => o.UseSqlite(releaseConnection));
        services.AddApplicationServices();

        var provider = services.BuildServiceProvider();

        // 初始化表结构与种子数据（与 Program.Main 一致）
        using (var scope = provider.CreateScope())
        {
            var local = scope.ServiceProvider.GetRequiredService<WctMinerDbContext>();
            await local.Database.EnsureCreatedAsync();
            SeedData.Initialize(local);

            var release = scope.ServiceProvider.GetRequiredService<ReleaseDbContext>();
            await release.Database.EnsureCreatedAsync();
        }

        var hostScope = provider.CreateScope();
        return new CliTestHost(localConnection, releaseConnection, provider, hostScope, tempDir);
    }

    public void Dispose()
    {
        _scope.Dispose();
        _provider.Dispose();
        _localConnection.Dispose();
        _releaseConnection.Dispose();

        try { Directory.Delete(TempDir, recursive: true); }
        catch { /* 尽力清理临时目录 */ }
    }
}