using DotNet.Testcontainers.Builders;
using DotNet.Testcontainers.Configurations;
using DotNet.Testcontainers.Containers;
using Microsoft.EntityFrameworkCore;
using Testcontainers.MsSql;
using Xunit;

namespace Tests;

public abstract class MsSqlTestBase : IAsyncLifetime
{
    private static bool IsCiSystem => !string.IsNullOrWhiteSpace(Environment.GetEnvironmentVariable("TF_BUILD"));

    protected MsSqlContainer MsSqlContainer { get; } =
        new MsSqlBuilder()
            .WithAutoRemove(true)
            .WithCleanUp(!IsCiSystem)
            .WithWaitStrategy(
                Wait.ForUnixContainer()
                    .UntilMessageIsLogged("SQL Server is now ready for client connections")
                    .AddCustomWaitStrategy(new WaitSqlCmdSuccess())
            )
            .Build();

    public Task InitializeAsync() => MsSqlContainer.StartAsync();

    public virtual async Task DisposeAsync()
    {
        try
        {
            await MsSqlContainer.StopAsync();
            await MsSqlContainer.DisposeAsync();
        }
        catch
        {
            // Ignore dispose errors
        }
    }

    public DbContextOptions<TContext> CreateDbContextOptions<TContext>()
        where TContext : DbContext =>
        new DbContextOptionsBuilder<TContext>().UseSqlServer(MsSqlContainer.GetConnectionString()).Options;

    /// <remarks>
    /// Uses the <c>sqlcmd</c> utility scripting variables to detect readiness of the MsSql container:
    /// <see href="https://learn.microsoft.com/en-us/sql/tools/sqlcmd/sqlcmd-utility?view=sql-server-linux-ver15#sqlcmd-scripting-variables"/>.
    /// </remarks>
    private sealed class WaitSqlCmdSuccess : IWaitUntil
    {
        /// <inheritdoc />
        public Task<bool> UntilAsync(IContainer container)
        {
            return UntilAsync((MsSqlContainer)container);
        }

        /// <inheritdoc cref="IWaitUntil.UntilAsync" />
        private static async Task<bool> UntilAsync(MsSqlContainer container)
        {
            var sqlCmdFilePath = await container.GetSqlCmdFilePathAsync().ConfigureAwait(false);

            var execResult = await container
                .ExecAsync(new[] { sqlCmdFilePath, "-C", "-Q", "SELECT 1;" })
                .ConfigureAwait(false);

            return 0L.Equals(execResult.ExitCode);
        }
    }
}