using System.Text.Json;

using NpgsqlRest;

namespace NpgsqlRestClient;

/// <summary>

/// Captures endpoint metadata during NpgsqlRest initialization.

/// Used by the --endpoints CLI command to output endpoint information.

/// </summary>

public class EndpointCapture : IEndpointCreateHandler

{

public static RoutineEndpoint[] Endpoints { get; private set; } = [];

public void Cleanup(RoutineEndpoint[] endpoints)

{

Endpoints = endpoints;

}

public static void WriteEndpointsJson(Utf8JsonWriter writer)

{

writer.WriteStartArray();

foreach (var ep in Endpoints)

{

writer.WriteStartObject();

writer.WriteString("method", ep.Method.ToString());

writer.WriteString("path", ep.Path);

// Routine info

writer.WriteStartObject("routine");

writer.WriteString("schema", ep.Routine.Schema);

writer.WriteString("name", ep.Routine.Name);

writer.WriteString("type", ep.Routine.Type.ToString());

if (ep.Routine.Type is RoutineType.Table or RoutineType.View)

{

writer.WriteString("crudType", ep.Routine.CrudType.ToString());

}

writer.WriteBoolean("returnsSet", ep.Routine.ReturnsSet);

writer.WriteBoolean("isVoid", ep.Routine.IsVoid);

writer.WriteString("definition", ep.Routine.SimpleDefinition);

writer.WriteEndObject();

// Parameters

writer.WriteStartArray("parameters");

foreach (var param in ep.Routine.Parameters)

{

writer.WriteStartObject();

writer.WriteString("name", param.ConvertedName);

writer.WriteString("originalName", param.ActualName);

writer.WriteString("type", param.TypeDescriptor.OriginalType);

writer.WriteBoolean("hasDefault", param.TypeDescriptor.HasDefault);

if (ep.HasPathParameters && ep.PathParameters!.Contains(param.ConvertedName, StringComparer.OrdinalIgnoreCase))

{

writer.WriteBoolean("isPathParameter", true);

}

writer.WriteEndObject();

}

writer.WriteEndArray();

// Return columns

if (ep.Routine.ColumnCount > 0)

{

writer.WriteStartArray("returnColumns");

for (int i = 0; i < ep.Routine.ColumnCount; i++)

{

writer.WriteStartObject();

writer.WriteString("name", ep.Routine.ColumnNames[i]);

writer.WriteString("originalName", ep.Routine.OriginalColumnNames[i]);

if (i < ep.Routine.ColumnsTypeDescriptor.Length)

{

writer.WriteString("type", ep.Routine.ColumnsTypeDescriptor[i].OriginalType);

}

writer.WriteEndObject();

}

writer.WriteEndArray();

}

// Configuration

writer.WriteString("requestParamType", ep.RequestParamType.ToString());

writer.WriteBoolean("requiresAuthorization", ep.RequiresAuthorization);

if (ep.AuthorizeRoles is { Count: > 0 })

{

writer.WriteStartArray("authorizeRoles");

foreach (var role in ep.AuthorizeRoles) writer.WriteStringValue(role);

writer.WriteEndArray();

}

if (ep.Login) writer.WriteBoolean("login", true);

if (ep.Logout) writer.WriteBoolean("logout", true);

if (ep.SecuritySensitive) writer.WriteBoolean("securitySensitive", true);

if (ep.Raw) writer.WriteBoolean("raw", true);

if (ep.Cached) writer.WriteBoolean("cached", true);

if (ep.Upload) writer.WriteBoolean("upload", true);

if (ep.IsProxy) writer.WriteBoolean("isProxy", true);

if (ep.UserContext) writer.WriteBoolean("userContext", true);

if (ep.UseUserParameters) writer.WriteBoolean("userParameters", true);

if (ep.ConnectionName is not null) writer.WriteString("connectionName", ep.ConnectionName);

if (ep.SseEventsPath is not null) writer.WriteString("sseEventsPath", ep.SseEventsPath);

if (ep.RateLimiterPolicy is not null) writer.WriteString("rateLimiterPolicy", ep.RateLimiterPolicy);

if (ep.ErrorCodePolicy is not null) writer.WriteString("errorCodePolicy", ep.ErrorCodePolicy);

if (ep.CustomParameters is { Count: > 0 })

{

writer.WriteStartObject("customParameters");

foreach (var kvp in ep.CustomParameters)

{

writer.WriteString(kvp.Key, kvp.Value);

}

writer.WriteEndObject();

}

writer.WriteEndObject();

}

writer.WriteEndArray();

}

}