using EAIAdminDAL.DTO;
using EAIAdminDAL.Query;
using GB5Shared.DTO.Framework.Login;
using GB5Shared.GenerateAutoNumber;
using GB5Shared.QueryExecutor;
using Microsoft.Extensions.Logging;
using System;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
using static GB5Shared.GB5Constant.Constant;

namespace EAIAdminDAL.CustomCode
{
    public class EAIModelConfigAdminDAL : IEAIModelConfigAdminDAL
    {
        private readonly IQueryExecutor _qe;
        private readonly AutoNumber _autoNumber;
        private readonly ILogger<EAIModelConfigAdminDAL> _logger;

        public EAIModelConfigAdminDAL(IQueryExecutor qe, AutoNumber autoNumber, ILogger<EAIModelConfigAdminDAL> logger)
        {
            _qe         = qe         ?? throw new ArgumentNullException(nameof(qe));
            _autoNumber = autoNumber ?? throw new ArgumentNullException(nameof(autoNumber));
            _logger     = logger     ?? throw new ArgumentNullException(nameof(logger));
        }

        public async Task<IEnumerable<EAIModelConfigAdminDTO>> GetModelConfigListAsync(
            string? search, LoginDTO login, CancellationToken ct = default)
        {
            ct.ThrowIfCancellationRequested();

            try
            {
                var searchPattern = string.IsNullOrWhiteSpace(search) ? null : $"%{search}%";

                var result = await _qe.QueryAsync<EAIModelConfigAdminDTO>(
                    login,
                    EAIModelConfigAdminQB.GET_MODELCONFIG_LIST,
                    new { TenantId = login.ClientId, SearchPattern = searchPattern },
                    cancellationToken: ct).ConfigureAwait(false);

                return result ?? [];
            }
            catch (OperationCanceledException)
            {
                _logger.LogWarning("DAL | GetModelConfigList | Cancelled | Tenant={TenantId}", login.ClientId);
                throw;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "DAL | GetModelConfigList | Error | Tenant={TenantId}", login.ClientId);
                throw;
            }
        }

        public async Task<EAIModelConfigAdminDTO?> GetModelConfigAsync(
            int modelId, LoginDTO login, CancellationToken ct = default)
        {
            ct.ThrowIfCancellationRequested();

            try
            {
                return await _qe.QuerySingleAsync<EAIModelConfigAdminDTO>(
                    login,
                    EAIModelConfigAdminQB.GET_MODELCONFIG,
                    new { ModelId = modelId, TenantId = login.ClientId },
                    cancellationToken: ct).ConfigureAwait(false);
            }
            catch (OperationCanceledException)
            {
                _logger.LogWarning("DAL | GetModelConfig | Cancelled | ModelId={Id}", modelId);
                throw;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "DAL | GetModelConfig | Error | ModelId={Id}", modelId);
                throw;
            }
        }

        public async Task<int> InsertModelConfigAsync(
            EAIModelConfigAdminDTO dto, LoginDTO login, CancellationToken ct = default)
        {
            ct.ThrowIfCancellationRequested();

            try
            {
                // AIMODELID is a plain INT PK (not IDENTITY) — must be generated here.
                var auto = await _autoNumber.GetNumberAsync(1, AUTONUMBERCONSTANT.EAIMODELCONFIG, login)
                    .ConfigureAwait(false);
                var newId = auto.StartNumber;
                var now = DateTime.UtcNow;

                await _qe.ExecuteAsync(
                    login,
                    EAIModelConfigAdminQB.INSERT_MODELCONFIG,
                    new
                    {
                        ModelId        = newId,
                        TenantId       = login.ClientId,
                        dto.ModelCode,
                        dto.ModelName,
                        dto.Provider,
                        dto.Endpoint,
                        dto.ApiKeyVaultKey,
                        dto.IsActive,
                        dto.EAIRemoteId,
                        dto.Version,
                        CreatedById    = login.UserId,
                        CreatedOn      = now
                    },
                    cancellationToken: ct).ConfigureAwait(false);

                _logger.LogInformation("DAL | InsertModelConfig | Completed | NewId={NewId}", newId);
                return newId;
            }
            catch (OperationCanceledException)
            {
                _logger.LogWarning("DAL | InsertModelConfig | Cancelled | ModelCode={Code}", dto.ModelCode);
                throw;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "DAL | InsertModelConfig | Error | ModelCode={Code}", dto.ModelCode);
                throw;
            }
        }

        public async Task UpdateModelConfigAsync(
            EAIModelConfigAdminDTO dto, LoginDTO login, CancellationToken ct = default)
        {
            ct.ThrowIfCancellationRequested();

            try
            {
                await _qe.ExecuteAsync(
                    login,
                    EAIModelConfigAdminQB.UPDATE_MODELCONFIG,
                    new
                    {
                        dto.ModelId,
                        TenantId     = login.ClientId,
                        dto.ModelName,
                        dto.Provider,
                        dto.Endpoint,
                        dto.IsActive,
                        dto.EAIRemoteId,
                        dto.Version,
                        ModifiedById = login.UserId,
                        ModifiedOn   = DateTime.UtcNow
                    },
                    cancellationToken: ct).ConfigureAwait(false);
            }
            catch (OperationCanceledException)
            {
                _logger.LogWarning("DAL | UpdateModelConfig | Cancelled | ModelId={Id}", dto.ModelId);
                throw;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "DAL | UpdateModelConfig | Error | ModelId={Id}", dto.ModelId);
                throw;
            }
        }

        public async Task UpdateEAIRemoteIdAsync(
            int modelId, long eaiRemoteId, LoginDTO login, CancellationToken ct = default)
        {
            ct.ThrowIfCancellationRequested();

            await _qe.ExecuteAsync(
                login,
                EAIModelConfigAdminQB.UPDATE_EAIREMOTEID,
                new { ModelId = modelId, TenantId = login.ClientId, EAIRemoteId = eaiRemoteId },
                cancellationToken: ct).ConfigureAwait(false);
        }

        public async Task DeleteModelConfigAsync(
            int modelId, LoginDTO login, CancellationToken ct = default)
        {
            ct.ThrowIfCancellationRequested();

            try
            {
                await _qe.ExecuteAsync(
                    login,
                    EAIModelConfigAdminQB.DELETE_MODELCONFIG,
                    new
                    {
                        ModelId      = modelId,
                        TenantId     = login.ClientId,
                        ModifiedById = login.UserId,
                        ModifiedOn   = DateTime.UtcNow
                    },
                    cancellationToken: ct).ConfigureAwait(false);
            }
            catch (OperationCanceledException)
            {
                _logger.LogWarning("DAL | DeleteModelConfig | Cancelled | ModelId={Id}", modelId);
                throw;
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "DAL | DeleteModelConfig | Error | ModelId={Id}", modelId);
                throw;
            }
        }

        public async Task<bool> ModelCodeExistsAsync(
            string modelCode, int excludeModelId, LoginDTO login, CancellationToken ct = default)
        {
            ct.ThrowIfCancellationRequested();

            var count = await _qe.ExecuteScalarAsync<int>(
                login,
                EAIModelConfigAdminQB.EXISTS_MODELCODE,
                new { TenantId = login.ClientId, ModelCode = modelCode, ModelId = excludeModelId },
                cancellationToken: ct).ConfigureAwait(false);

            return count == 1;
        }
    }
}