using System.Reflection;
using Api.Clients.Ai;
using Api.Clients.Ai.Contracts;
using Rag.Data;
using Rag.Data.Repositories;
using Rag.Data.Repositories.Contracts;
using Api.Services;
using Api.Services.Contracts;
using Microsoft.EntityFrameworkCore;
using Rag.Models.Settings;
using Serilog;
using Common.Settings;
using StartupHelpers;

StartupExtensions.LoadDotEnvFile();

const string ServiceName = "rag-api";
var appVersion = StartupExtensions.GetApplicationVersion(Assembly.GetExecutingAssembly());

try
{
    var builder = WebApplication.CreateBuilder(args);

    builder.ConfigureJsonSerilog(ServiceName, appVersion);
    Log.Information("Starting {Service} version {AppVersion}", ServiceName, appVersion);

    builder.AddAzureKeyVaultIfConfigured();

    builder.Services.Configure<DatabaseSettings>(builder.Configuration.GetSection("Database"));
    builder.Services.Configure<RagSettings>(builder.Configuration.GetSection("Rag"));
    builder.Services.Configure<Rag.Models.Settings.AiSettings>(builder.Configuration.GetSection("Ai"));
    builder.Services.Configure<InternalApiSettings>(builder.Configuration.GetSection("InternalApi"));

    builder.Services.AddDbContext<RagDbContext>(options =>
    {
        var configuration = builder.Configuration;
        var connectionString = builder.Services.GetConfiguredDbConnectionString(configuration);

        options.UseSqlServer(connectionString, sql =>
        {
            sql.MigrationsHistoryTable(RagDbContext.MigrationTableName, RagDbContext.SchemaName);
            sql.MigrationsAssembly("rag-data");
        });
    });

    builder.Services.AddHttpClient<RagAiClient>();
    builder.Services.AddScoped<IRagRepository, EfRagRepository>();
    builder.Services.AddScoped<IAiClient, CachedRagAiClient>();
    builder.Services.AddSingleton<ITextExtractor, TextExtractor>();
    builder.Services.AddSingleton<ITextChunker, TextChunker>();
    builder.Services.AddSingleton<IDocumentClassifier, DocumentClassifier>();
    builder.Services.AddScoped<IRagService, RagService>();

    builder.Services.AddControllers();
    builder.Services.AddSwaggerWithXmlComments(Assembly.GetExecutingAssembly(), ServiceName);

    var app = builder.Build();

    app.LogStartupDiagnostics(ServiceName);

    using (var scope = app.Services.CreateScope())
    {
        var repository = scope.ServiceProvider.GetRequiredService<IRagRepository>();
        await repository.InitializeAsync(CancellationToken.None);
    }

    app.UseDefaultSerilogRequestLogging();
    app.UseJsonExceptionHandler(ServiceName);
    app.UseInternalApiKeyProtection();
    app.UseSwaggerInDevelopment(ServiceName, ServiceName);

    app.MapControllers();

    Log.Information("Running EF Core migrations if any");
    using (var scope = app.Services.CreateScope())
    {
        var db = scope.ServiceProvider.GetRequiredService<RagDbContext>();
        db.Database.Migrate();
    }

    Log.Information("{Service} startup complete", ServiceName);
    app.Run();
}
catch (Exception ex)
{
    Log.Fatal(ex, "{Service} terminated unexpectedly", ServiceName);
}
finally
{
    Log.Information("Shutting down {Service}", ServiceName);
    Log.CloseAndFlush();
}
