// Primenza C# SDK v1
using System.Net.Http.Headers;
using System.Text;
using System.Text.Json;

namespace Primenza.Sdk;

public class PrimenzaClient : IDisposable
{
    private readonly HttpClient _client;
    private readonly string _baseUrl;
    private readonly string _apiKey;

    public PrimenzaClient(string apiKey, string baseUrl = "https://primenza.com/api/v1")
    {
        _apiKey = apiKey;
        _baseUrl = baseUrl.TrimEnd('/');
        _client = new HttpClient();
        _client.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", apiKey);
        _client.DefaultRequestHeaders.Add("X-API-Key", apiKey);
        _client.DefaultRequestHeaders.Accept.Add(new MediaTypeWithQualityHeaderValue("application/json"));
    }

    private JsonElement Unwrap(JsonDocument doc)
    {
        var root = doc.RootElement;
        if (root.TryGetProperty("errors", out var errors) && errors.GetArrayLength() > 0)
        {
            throw new InvalidOperationException(errors[0].GetProperty("message").GetString() ?? "API request failed");
        }

        return root.TryGetProperty("data", out var data) ? data : root;
    }

    public async Task<JsonElement> GetAsync(string path)
    {
        var response = await _client.GetAsync($"{_baseUrl}{path}");
        response.EnsureSuccessStatusCode();
        using var json = JsonDocument.Parse(await response.Content.ReadAsStringAsync());
        return Unwrap(json);
    }

    public async Task<JsonElement> PostAsync(string path, object? body = null)
    {
        var content = body != null
            ? new StringContent(JsonSerializer.Serialize(body), Encoding.UTF8, "application/json")
            : null;
        var response = await _client.PostAsync($"{_baseUrl}{path}", content);
        response.EnsureSuccessStatusCode();
        using var json = JsonDocument.Parse(await response.Content.ReadAsStringAsync());
        return Unwrap(json);
    }

    public async Task<JsonElement> PatchAsync(string path, object? body = null)
    {
        var content = body != null
            ? new StringContent(JsonSerializer.Serialize(body), Encoding.UTF8, "application/json")
            : null;
        var request = new HttpRequestMessage(HttpMethod.Patch, $"{_baseUrl}{path}") { Content = content };
        var response = await _client.SendAsync(request);
        response.EnsureSuccessStatusCode();
        using var json = JsonDocument.Parse(await response.Content.ReadAsStringAsync());
        return Unwrap(json);
    }

    public async Task<JsonElement> DeleteAsync(string path)
    {
        var response = await _client.DeleteAsync($"{_baseUrl}{path}");
        response.EnsureSuccessStatusCode();
        using var json = JsonDocument.Parse(await response.Content.ReadAsStringAsync());
        return Unwrap(json);
    }

    public async Task<JsonElement> PutAsync(string path, object? body = null)
    {
        var content = body != null
            ? new StringContent(JsonSerializer.Serialize(body), Encoding.UTF8, "application/json")
            : null;
        var response = await _client.PutAsync($"{_baseUrl}{path}", content);
        response.EnsureSuccessStatusCode();
        using var json = JsonDocument.Parse(await response.Content.ReadAsStringAsync());
        return Unwrap(json);
    }

    public Task<JsonElement> Dashboard(int? domainId = null)
        => GetAsync("/dashboard" + (domainId.HasValue ? $"?domain_id={domainId}" : ""));
    public Task<JsonElement> Domains() => GetAsync("/domains");
    public Task<JsonElement> Domain(int domainId) => GetAsync($"/domains/{domainId}");
    public Task<JsonElement> CreateDomain(string url, string? name = null)
        => PostAsync("/domains", new { url, name });
    public Task<JsonElement> UpdateDomain(int domainId, object fields) => PatchAsync($"/domains/{domainId}", fields);
    public Task<JsonElement> Crawl(int domainId) => PostAsync($"/domains/{domainId}/crawl");
    public Task<JsonElement> Audit(int domainId) => PostAsync($"/domains/{domainId}/audit");
    public Task<JsonElement> Seo(int domainId) => GetAsync($"/domains/{domainId}/seo");
    public Task<JsonElement> RunSeo(int domainId) => PostAsync($"/domains/{domainId}/seo/run");
    public Task<JsonElement> Geo(int domainId) => GetAsync($"/domains/{domainId}/geo");
    public Task<JsonElement> RunGeo(int domainId) => PostAsync($"/domains/{domainId}/geo/run");
    public Task<JsonElement> AiVisibility(int domainId) => GetAsync($"/domains/{domainId}/ai-visibility");
    public Task<JsonElement> RunAiVisibility(int domainId) => PostAsync($"/domains/{domainId}/ai-visibility/run");
    public Task<JsonElement> Performance(int domainId) => GetAsync($"/domains/{domainId}/performance");
    public Task<JsonElement> RunPerformance(int domainId) => PostAsync($"/domains/{domainId}/performance/run");
    public Task<JsonElement> Links(int domainId) => GetAsync($"/domains/{domainId}/links");
    public Task<JsonElement> RunLinks(int domainId) => PostAsync($"/domains/{domainId}/links/run");
    public Task<JsonElement> ContentGaps(int domainId) => GetAsync($"/domains/{domainId}/content-gaps");
    public Task<JsonElement> DiscoverContentGaps(int domainId) => PostAsync($"/domains/{domainId}/content-gaps/discover");
    public Task<JsonElement> DeleteDomain(int domainId) => DeleteAsync($"/domains/{domainId}");
    public Task<JsonElement> VerifyDomain(int domainId) => PostAsync($"/domains/{domainId}/verify");
    public Task<JsonElement> DetectTech(int domainId) => PostAsync($"/domains/{domainId}/detect-tech");
    public Task<JsonElement> Pages(int domainId, int page = 1) => GetAsync($"/domains/{domainId}/pages?page={page}");
    public Task<JsonElement> Audits(int domainId) => GetAsync($"/domains/{domainId}/audits");
    public Task<JsonElement> Schema(int domainId) => GetAsync($"/domains/{domainId}/schema");
    public Task<JsonElement> Robots(int domainId) => GetAsync($"/domains/{domainId}/robots");
    public Task<JsonElement> DeployRobots(int domainId) => PostAsync($"/domains/{domainId}/robots/deploy");
    public Task<JsonElement> DeployLlms(int domainId, string type = "llms") => PostAsync($"/domains/{domainId}/llms-txt/deploy", new { type });
    public Task<JsonElement> Competitors(int domainId) => GetAsync($"/domains/{domainId}/competitors");
    public Task<JsonElement> AddCompetitor(int domainId, string url) => PostAsync($"/domains/{domainId}/competitors", new { url });
    public Task<JsonElement> RefreshCompetitor(int domainId, int competitorId) => PostAsync($"/domains/{domainId}/competitors/{competitorId}/refresh");
    public Task<JsonElement> RemoveCompetitor(int domainId, int competitorId) => DeleteAsync($"/domains/{domainId}/competitors/{competitorId}");
    public Task<JsonElement> Tenant() => GetAsync("/tenant");
    public Task<JsonElement> Schedules() => GetAsync("/schedules");
    public Task<JsonElement> KnowledgeGraph(int domainId) => GetAsync($"/domains/{domainId}/knowledge-graph");
    public Task<JsonElement> BuildKnowledgeGraph(int domainId) => PostAsync($"/domains/{domainId}/knowledge-graph/build");
    public Task<JsonElement> Citation(int domainId) => GetAsync($"/domains/{domainId}/citation");
    public Task<JsonElement> DeployPreview(int domainId, string artifact = "robots") => GetAsync($"/domains/{domainId}/deploy/preview?artifact={artifact}");
    public Task<JsonElement> Agents(int domainId) => GetAsync($"/domains/{domainId}/agents");
    public Task<JsonElement> RegisterAgent(int domainId, string name, string type) => PostAsync($"/domains/{domainId}/agents", new { name, type });
    public Task<JsonElement> TestWebhook() => PostAsync("/webhooks/test");
    public Task<JsonElement> Monitoring(int domainId) => GetAsync($"/domains/{domainId}/monitoring");
    public Task<JsonElement> Reports(int domainId) => GetAsync($"/domains/{domainId}/reports");
    public Task<JsonElement> GraphQL(string query, object? variables = null)
        => PostAsync("/graphql", new { query, variables });

    public Task<JsonElement> CreateSchedule(object data) => PostAsync("/schedules", data);
    public Task<JsonElement> UpdateSchedule(int id, object data) => PatchAsync($"/schedules/{id}", data);
    public Task<JsonElement> DeleteSchedule(int id) => DeleteAsync($"/schedules/{id}");
    public Task<JsonElement> Workspaces() => GetAsync("/workspaces");
    public Task<JsonElement> CreateWorkspace(object data) => PostAsync("/workspaces", data);
    public Task<JsonElement> UpdateWorkspace(int id, object data) => PatchAsync($"/workspaces/{id}", data);
    public Task<JsonElement> DeleteWorkspace(int id) => DeleteAsync($"/workspaces/{id}");
    public Task<JsonElement> Team() => GetAsync("/team");
    public Task<JsonElement> InviteTeamMember(object data) => PostAsync("/team/invitations", data);
    public Task<JsonElement> UpdateTeamRole(int userId, string role) => PatchAsync($"/team/{userId}", new { role });
    public Task<JsonElement> RemoveTeamMember(int userId) => DeleteAsync($"/team/{userId}");
    public Task<JsonElement> CancelInvitation(int invitationId) => DeleteAsync($"/team/invitations/{invitationId}");
    public Task<JsonElement> Settings() => GetAsync("/settings");
    public Task<JsonElement> UpdateSettings(object data) => PatchAsync("/settings", data);
    public Task<JsonElement> CreateApiKey(object data) => PostAsync("/api-keys", data);
    public Task<JsonElement> DeleteApiKey(int id) => DeleteAsync($"/api-keys/{id}");
    public Task<JsonElement> AuditLogs(string query = "") => GetAsync("/audit-logs" + (query != "" ? $"?{query}" : ""));
    public Task<JsonElement> Notifications() => GetAsync("/notifications");
    public Task<JsonElement> MarkNotificationRead(int id) => PostAsync($"/notifications/{id}/read");
    public Task<JsonElement> MarkAllNotificationsRead() => PostAsync("/notifications/read-all");
    public Task<JsonElement> RegenerateVerification(int domainId) => PostAsync($"/domains/{domainId}/verification/regenerate");
    public Task<JsonElement> UpdateVerificationMethod(int domainId, string method) => PatchAsync($"/domains/{domainId}/verification-method", new { method });
    public Task<JsonElement> ActivateDomain(int domainId) => PostAsync($"/domains/{domainId}/activate");
    public Task<JsonElement> GenerateContentGapOutline(int domainId, int gapId) => PostAsync($"/domains/{domainId}/content-gaps/{gapId}/outline");
    public Task<JsonElement> GenerateContentGapArticle(int domainId, int gapId) => PostAsync($"/domains/{domainId}/content-gaps/{gapId}/article");
    public Task<JsonElement> DismissContentGap(int domainId, int gapId) => PostAsync($"/domains/{domainId}/content-gaps/{gapId}/dismiss");
    public Task<JsonElement> ContentOptimizer(int domainId) => GetAsync($"/domains/{domainId}/content-optimizer");
    public Task<JsonElement> RefreshContentOptimizer(int domainId) => PostAsync($"/domains/{domainId}/content-optimizer/refresh");
    public Task<JsonElement> SimulateAiVisibility(int domainId, object? data = null) => PostAsync($"/domains/{domainId}/ai-visibility/simulate", data ?? new { });
    public Task<JsonElement> InstallAgent(int domainId, object data) => PostAsync($"/domains/{domainId}/agents/install", data);
    public Task<JsonElement> GenerateSchema(int domainId, object? data = null) => PostAsync($"/domains/{domainId}/schema/generate", data ?? new { });
    public Task<JsonElement> DeploySchema(int domainId, object? data = null) => PostAsync($"/domains/{domainId}/schema/deploy", data ?? new { });
    public Task<JsonElement> UpdateRobots(int domainId, string content) => PutAsync($"/domains/{domainId}/robots", new { content });
    public Task<JsonElement> GenerateLlms(int domainId, string type = "llms") => PostAsync($"/domains/{domainId}/llms-txt/generate", new { type });
    public Task<JsonElement> StoreSubdomain(int domainId, object data) => PostAsync($"/domains/{domainId}/subdomains", data);
    public Task<JsonElement> DestroySubdomain(int domainId, int subdomainId) => DeleteAsync($"/domains/{domainId}/subdomains/{subdomainId}");
    public Task<JsonElement> DownloadReport(int domainId) => GetAsync($"/domains/{domainId}/reports/download");
    public Task<JsonElement> DownloadReportPdf(int domainId) => GetAsync($"/domains/{domainId}/reports/download/pdf");
    public Task<JsonElement> SuggestCitation(int domainId, int pageId, object? data = null) => PostAsync($"/domains/{domainId}/citation/{pageId}/suggest", data ?? new { });
    public Task<JsonElement> ApproveAgentCommand(int domainId, int commandId) => PostAsync($"/domains/{domainId}/agents/commands/{commandId}/approve");

    public void Dispose() => _client.Dispose();
}

public class PrimenzaAgentClient : IDisposable
{
    private readonly HttpClient _client;
    private readonly string _baseUrl;
    private readonly string _agentToken;

    public PrimenzaAgentClient(string agentToken, string baseUrl = "https://primenza.com/api/v1")
    {
        _agentToken = agentToken;
        _baseUrl = baseUrl.TrimEnd('/');
        _client = new HttpClient();
        _client.DefaultRequestHeaders.Add("X-Agent-Token", agentToken);
        _client.DefaultRequestHeaders.Accept.Add(new MediaTypeWithQualityHeaderValue("application/json"));
    }

    private JsonElement Unwrap(JsonDocument doc)
    {
        var root = doc.RootElement;
        if (root.TryGetProperty("errors", out var errors) && errors.GetArrayLength() > 0)
        {
            throw new InvalidOperationException(errors[0].GetProperty("message").GetString() ?? "API request failed");
        }

        return root.TryGetProperty("data", out var data) ? data : root;
    }

    private async Task<JsonElement> PostAsync(string path, object? body = null)
    {
        var content = body != null
            ? new StringContent(JsonSerializer.Serialize(body), Encoding.UTF8, "application/json")
            : null;
        var response = await _client.PostAsync($"{_baseUrl}{path}", content);
        response.EnsureSuccessStatusCode();
        using var json = JsonDocument.Parse(await response.Content.ReadAsStringAsync());
        return Unwrap(json);
    }

    public Task<JsonElement> Poll() => PostAsync("/agent/poll");
    public Task<JsonElement> Report(int commandId, object result) => PostAsync("/agent/report", new { command_id = commandId, result });

    public void Dispose() => _client.Dispose();
}