using System.Text.Json;
using System.Threading.Tasks;
using Gott.Models;
using Gott.Services;
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.Routing;

namespace Gott.Endpoints;

public static class AgentEndpoint
{
    public static void MapAgentEndpoints(this IEndpointRouteBuilder app)
    {
        app.MapGet("/api/agents", HandleList);
        app.MapGet("/api/agents/{agentId}", HandleGet);
        app.MapPost("/api/agents", HandleCreate);
        app.MapPut("/api/agents/{agentId}", HandleUpdate);
        app.MapDelete("/api/agents/{agentId}", HandleDelete);
    }

    private static async Task HandleList(HttpContext ctx, DatabaseService db)
    {
        var agents = await db.GetAgents();
        await ctx.Response.WriteAsJsonAsync(agents);
    }

    private static async Task HandleGet(HttpContext ctx, DatabaseService db)
    {
        var agentId = ctx.Request.RouteValues["agentId"]?.ToString();
        if (string.IsNullOrEmpty(agentId)) { ctx.Response.StatusCode = 400; return; }

        var agent = await db.GetAgent(agentId);
        if (agent == null) { ctx.Response.StatusCode = 404; return; }

        await ctx.Response.WriteAsJsonAsync(agent);
    }

    private static async Task HandleCreate(HttpContext ctx, DatabaseService db, ImageService imageService, MemoryService memoryService)
    {
        var body = await JsonSerializer.DeserializeAsync<JsonElement>(ctx.Request.Body);

        var avatarRaw = body.TryGetProperty("avatar", out var av) ? av.GetString() : null;

        string customCommands = null;
        if (body.TryGetProperty("customCommands", out var ccEl))
            customCommands = ccEl.ValueKind == JsonValueKind.Null ? null : ccEl.GetRawText();

        var agent = new AgentRecord
        {
            Name = body.TryGetProperty("name", out var n) ? n.GetString() ?? "Neuer Agent" : "Neuer Agent",
            Avatar = imageService.ResizeDataUrl(avatarRaw),
            Description = body.TryGetProperty("description", out var d) ? d.GetString() : null,
            Model = body.TryGetProperty("model", out var m) ? m.GetString() ?? "sonnet" : "sonnet",
            ToolChoice = body.TryGetProperty("toolChoice", out var tc) ? tc.GetString() ?? "any" : "any",
            CustomCommands = customCommands,
            AutoCommit = body.TryGetProperty("autoCommit", out var ac) ? ac.GetString() ?? "" : "",
            UseSoul = !body.TryGetProperty("useSoul", out var us) || us.GetBoolean(),
            UseIdentity = !body.TryGetProperty("useIdentity", out var ui) || ui.GetBoolean(),
            UseUserInfo = !body.TryGetProperty("useUserInfo", out var uu) || uu.GetBoolean(),
            UseMemory = !body.TryGetProperty("useMemory", out var um) || um.GetBoolean()
        };

        agent = await db.CreateAgent(agent);
        memoryService.InitializeAgentFiles(agent.AgentId, agent.Name);
        await ctx.Response.WriteAsJsonAsync(agent);
    }

    private static async Task HandleUpdate(HttpContext ctx, DatabaseService db, ImageService imageService, MemoryService memoryService)
    {
        var agentId = ctx.Request.RouteValues["agentId"]?.ToString();
        if (string.IsNullOrEmpty(agentId)) { ctx.Response.StatusCode = 400; return; }

        var agent = await db.GetAgent(agentId);
        if (agent == null) { ctx.Response.StatusCode = 404; return; }

        var body = await JsonSerializer.DeserializeAsync<JsonElement>(ctx.Request.Body);

        if (body.TryGetProperty("name", out var n)) agent.Name = n.GetString();
        if (body.TryGetProperty("avatar", out var av)) agent.Avatar = imageService.ResizeDataUrl(av.GetString());
        if (body.TryGetProperty("description", out var d)) agent.Description = d.GetString();
        if (body.TryGetProperty("model", out var m)) agent.Model = m.GetString();
        if (body.TryGetProperty("toolChoice", out var tc)) agent.ToolChoice = tc.GetString();
        if (body.TryGetProperty("customCommands", out var cc))
            agent.CustomCommands = cc.ValueKind == JsonValueKind.Null ? null : cc.GetRawText();
        if (body.TryGetProperty("autoCommit", out var ac)) agent.AutoCommit = ac.GetString() ?? "";
        if (body.TryGetProperty("useSoul", out var us)) agent.UseSoul = us.GetBoolean();
        if (body.TryGetProperty("useIdentity", out var ui)) agent.UseIdentity = ui.GetBoolean();
        if (body.TryGetProperty("useUserInfo", out var uu)) agent.UseUserInfo = uu.GetBoolean();
        if (body.TryGetProperty("useMemory", out var um)) agent.UseMemory = um.GetBoolean();

        await db.UpdateAgent(agent);
        memoryService.InitializeAgentFiles(agentId, agent.Name);
        agent = await db.GetAgent(agentId);
        await ctx.Response.WriteAsJsonAsync(agent);
    }

    private static async Task HandleDelete(HttpContext ctx, DatabaseService db)
    {
        var agentId = ctx.Request.RouteValues["agentId"]?.ToString();
        if (string.IsNullOrEmpty(agentId)) { ctx.Response.StatusCode = 400; return; }

        await db.DeleteAgent(agentId);
        ctx.Response.StatusCode = 204;
    }
}