using Microsoft.Data.SqlClient;

namespace ZB.MOM.WW.Auth.ApiKeys.SqlServer;

/// <summary>
/// Opens connections to the shared SQL Server API-key database and carries the settings every
/// command needs (schema name, command timeout).
/// </summary>
/// <param name="options">The SQL Server API-key store options; validated in the constructor.</param>
/// <exception cref="InvalidOperationException"><paramref name="options"/> is not usable.</exception>
public sealed class AuthSqlServerConnectionFactory(SqlServerApiKeyOptions options)
{
    private readonly SqlServerApiKeyOptions _options = Validate(options);

    private static SqlServerApiKeyOptions Validate(SqlServerApiKeyOptions options)
    {
        ArgumentNullException.ThrowIfNull(options);
        options.Validate();
        return options;
    }

    /// <summary>The schema that owns the three API-key tables.</summary>
    public string Schema => _options.Schema;

    /// <summary>Command timeout, in whole seconds, for every command issued against this store.</summary>
    public int CommandTimeoutSeconds => Math.Max(1, (int)_options.CommandTimeout.TotalSeconds);

    /// <summary>Opens a connection to the shared API-key database.</summary>
    /// <param name="cancellationToken">A token to cancel the connect.</param>
    /// <returns>An opened connection. The caller owns disposal.</returns>
    public async Task<SqlConnection> OpenConnectionAsync(CancellationToken cancellationToken)
    {
        var connection = new SqlConnection(_options.ConnectionString);
        try
        {
            await connection.OpenAsync(cancellationToken).ConfigureAwait(false);
            return connection;
        }
        catch
        {
            await connection.DisposeAsync().ConfigureAwait(false);
            throw;
        }
    }

    /// <summary>
    /// Creates a command on <paramref name="connection"/> with the configured timeout applied.
    /// </summary>
    /// <param name="connection">The open connection.</param>
    /// <param name="commandText">The command text.</param>
    /// <param name="transaction">The ambient transaction, if any.</param>
    /// <returns>The configured command. The caller owns disposal.</returns>
    public SqlCommand CreateCommand(
        SqlConnection connection, string commandText, SqlTransaction? transaction = null)
    {
        ArgumentNullException.ThrowIfNull(connection);

        SqlCommand command = connection.CreateCommand();
        command.CommandText = commandText;
        command.CommandTimeout = CommandTimeoutSeconds;
        command.Transaction = transaction;
        return command;
    }
}