using System.Text.Json;
using Bit.Seeder;
using Bit.SeederApi.Services;

namespace Bit.SeederApi.Execution;

public class QueryExecutor(
    ILogger<QueryExecutor> logger,
    IServiceProvider serviceProvider) : IQueryExecutor
{

    public object Execute(string queryName, JsonElement? arguments)
    {
        try
        {
            var query = serviceProvider.GetKeyedService<IQuery>(queryName)
                ?? throw new QueryNotFoundException(queryName);

            var requestType = query.GetRequestType();
            var requestModel = DeserializeRequestModel(queryName, requestType, arguments);
            var result = query.Execute(requestModel);

            logger.LogInformation("Successfully executed query: {QueryName}", queryName);
            return result;
        }
        catch (Exception ex) when (ex is not QueryNotFoundException and not QueryExecutionException)
        {
            logger.LogError(ex, "Unexpected error executing query: {QueryName}", queryName);
            throw new QueryExecutionException(
                $"An unexpected error occurred while executing query '{queryName}'",
                ex.InnerException ?? ex);
        }
    }

    private object DeserializeRequestModel(string queryName, Type requestType, JsonElement? arguments)
    {
        if (arguments == null)
        {
            return CreateDefaultRequestModel(queryName, requestType);
        }

        try
        {
            var requestModel = JsonSerializer.Deserialize(arguments.Value.GetRawText(), requestType, JsonConfiguration.Options);
            if (requestModel == null)
            {
                throw new QueryExecutionException(
                    $"Failed to deserialize request model for query '{queryName}'");
            }
            return requestModel;
        }
        catch (JsonException ex)
        {
            throw new QueryExecutionException(
                $"Failed to deserialize request model for query '{queryName}': {ex.Message}", ex);
        }
    }

    private object CreateDefaultRequestModel(string queryName, Type requestType)
    {
        try
        {
            var requestModel = Activator.CreateInstance(requestType);
            if (requestModel == null)
            {
                throw new QueryExecutionException(
                    $"Arguments are required for query '{queryName}'");
            }
            return requestModel;
        }
        catch
        {
            throw new QueryExecutionException(
                $"Arguments are required for query '{queryName}'");
        }
    }
}