using System.Collections.Frozen;

using System.Data;

using System.IO.Pipelines;

using System.Net;

using System.Runtime.CompilerServices;

using System.Text;

using System.Text.Json;

using System.Text.Json.Nodes;

using Microsoft.AspNetCore.DataProtection;

using Microsoft.AspNetCore.Http.Extensions;

using Microsoft.Extensions.Primitives;

using Npgsql;

using NpgsqlRest.Auth;

using NpgsqlRest.HttpClientType;

using NpgsqlRest.UploadHandlers;

using NpgsqlRest.UploadHandlers.Handlers;

using NpgsqlTypes;

using static System.Net.Mime.MediaTypeNames;

using static NpgsqlRest.ParameterParser;

namespace NpgsqlRest;

public class NpgsqlRestEndpoint(

NpgsqlRestMetadataEntry entry,

FrozenDictionary<string, NpgsqlRestMetadataEntry> overloads)

{

public async Task InvokeAsync(HttpContext context, IServiceProvider serviceProvider, CancellationToken cancellationToken = default)

{

Routine routine = entry.Endpoint.Routine;

RoutineEndpoint endpoint = entry.Endpoint;

IRoutineSourceParameterFormatter formatter = entry.Formatter;

string? headers = null;

if (endpoint.RequestHeadersMode == RequestHeadersMode.Context)

{

SearializeHeader(Options, context, ref headers);

}

NpgsqlConnection? connection = null;

NpgsqlTransaction? transaction = null;

string? commandText = null;

bool shouldDispose = true;

bool shouldCommit = true;

IUploadHandler? uploadHandler = null;

var shouldLog = Options.LogCommands && Logger != null;

// Pooled StringBuilders - declared at top level to enable pooling in finally block

StringBuilder? cmdLog = null;

StringBuilder? cacheKeys = null;

StringBuilder? commandTextBuilder = null;

StringBuilder? rowBuilder = null;

StringBuilder? compositeFieldBuffer = null;

// proxy_out: capture function output into a buffer instead of writing to the real response

MemoryStream? proxyOutBuffer = null;

Stream? proxyOutOriginalBody = null;

if (endpoint.IsProxyOut && Options.ProxyOptions.Enabled)

{

proxyOutBuffer = new MemoryStream();

proxyOutOriginalBody = context.Response.Body;

context.Response.Body = proxyOutBuffer;

}

var writer = proxyOutBuffer is not null

? PipeWriter.Create(context.Response.Body, new StreamPipeWriterOptions(leaveOpen: true))

: PipeWriter.Create(context.Response.Body);

try

{

if (endpoint.ConnectionName is not null)

{

// First check if there's a data source for this connection name (for multi-host support)

if (Options.DataSources?.TryGetValue(endpoint.ConnectionName, out var namedDataSource) is true)

{

connection = namedDataSource.CreateConnection();

}

else if (Options.ConnectionStrings?.TryGetValue(endpoint.ConnectionName, out var connectionString) is true)

{

connection = new(connectionString);

}

else

{

await ReturnErrorAsync($"Connection name {endpoint.ConnectionName} could not be found in options DataSources or ConnectionStrings dictionaries.", true, context, cancellationToken);

return;

}

}

else if (Options.ServiceProviderMode != ServiceProviderObject.None)

{

if (serviceProvider is null)

{

await ReturnErrorAsync($"ServiceProvider must be provided when ServiceProviderMode is set to {Options.ServiceProviderMode}.", true, context, cancellationToken);

return;

}

if (Options.ServiceProviderMode == ServiceProviderObject.NpgsqlDataSource)

{

connection = serviceProvider.GetRequiredService<NpgsqlDataSource>().CreateConnection();

await connection.OpenRetryAsync(Options.ConnectionRetryOptions, cancellationToken);

}

else if (Options.ServiceProviderMode == ServiceProviderObject.NpgsqlConnection)

{

shouldDispose = false;

connection = serviceProvider.GetRequiredService<NpgsqlConnection>();

}

}

else

{

if (Options.DataSource is not null)

{

connection = Options.DataSource.CreateConnection();

}

else

{

connection = new(Options.ConnectionString);

}

}

if (connection is null)

{

await ReturnErrorAsync("Connection did not initialize!", log: true, context, cancellationToken);

return;

}

if ( (Options.LogConnectionNoticeEvents && Logger != null) || endpoint.SseEventsPath is not null)

{

var currentEndpoint = endpoint;

connection.Notice += (sender, args) =>

{

if (Options.LogConnectionNoticeEvents && Logger != null)

{

NpgsqlRestLogger.LogConnectionNotice(args.Notice, Options.LogConnectionNoticeEventsMode);

}

if (currentEndpoint.SseEventsPath is not null &&

currentEndpoint.SseEventNoticeLevel is not null &&

string.Equals(args.Notice.Severity, currentEndpoint.SseEventNoticeLevel.ToString(), StringComparison.OrdinalIgnoreCase))

{

NpgsqlRestSseEventSource

.Broadcaster

.Broadcast(new SseEvent(args.Notice, currentEndpoint, context.Request.Headers[Options.ExecutionIdHeaderName].FirstOrDefault()));

}

};

}

if (Options.AuthenticationOptions.BasicAuth.Enabled is true || endpoint.BasicAuth?.Enabled is true)

{

if (Options.AuthenticationOptions.BasicAuth.ChallengeCommand is not null || endpoint.BasicAuth?.ChallengeCommand is not null)

{

await OpenConnectionAsync(connection, context, endpoint, cancellationToken);

}

await BasicAuthHandler.HandleAsync(context, endpoint, connection, cancellationToken);

if (context.Response.HasStarted is true)

{

return;

}

}

await using var command = NpgsqlRestCommand.Create(connection);

if (shouldLog)

{

cmdLog = StringBuilderPool.Rent();

cmdLog.Append("-- ");

cmdLog.Append(context.Request.Method);

cmdLog.Append(' ');

if (Options.AuthenticationOptions.ObfuscateAuthParameterLogValues && endpoint.IsAuth)

{

cmdLog.Append(context.Request.Scheme);

cmdLog.Append("://");

cmdLog.Append(context.Request.Host);

cmdLog.Append(context.Request.Path);

}

else

{

cmdLog.Append(context.Request.GetDisplayUrl());

}

cmdLog.Append(Environment.NewLine);

}

if (formatter.IsFormattable is false)

{

// Use pooled StringBuilder for non-formattable commands to avoid string.Concat allocations

commandTextBuilder = StringBuilderPool.Rent(routine.Expression.Length + routine.ParamCount * 16);

commandTextBuilder.Append(routine.Expression);

}

// paramsList

bool hasNulls = false;

int paramIndex = 0;

JsonObject? jsonObj = null;

Dictionary<string, JsonNode?>? bodyDict = null;

string? body = null;

// Use IQueryCollection directly to avoid ToDictionary() allocation

IQueryCollection? queryCollection = null;

// Skip local upload handling if proxy will forward the upload content

var skipUploadForProxy = endpoint.Upload is true &&

endpoint.IsProxy &&

Options.ProxyOptions.Enabled &&

Options.ProxyOptions.ForwardUploadContent;

uploadHandler = (endpoint.Upload is true && !skipUploadForProxy) ? endpoint.CreateUploadHandler() : null;

int uploadMetaParamIndex = -1;

Dictionary<string, object?>? claimsDict = null;

List<string> customHttpTypes = Options.HttpClientOptions.Enabled ? new(routine.Parameters.Length) : null!;

if (endpoint.Cached is true)

{

cacheKeys = StringBuilderPool.Rent(routine.Expression.Length + (endpoint.CachedParams?.Count ?? 0) * 32);

cacheKeys.Append(routine.Expression);

}

if (endpoint.RequestParamType == RequestParamType.QueryString)

{

queryCollection = context.Request.Query;

}

if (endpoint.HasBodyParameter || endpoint.RequestParamType == RequestParamType.BodyJson)

{

context.Request.EnableBuffering();

context.Request.Body.Position = 0;

using (var reader = new StreamReader(context.Request.Body, Encoding.UTF8, detectEncodingFromByteOrderMarks: false, leaveOpen: true))

{

body = await reader.ReadToEndAsync(cancellationToken);

}

if (endpoint.RequestParamType == RequestParamType.BodyJson)

{

JsonNode? node = null;

try

{

node = JsonNode.Parse(string.IsNullOrWhiteSpace(body) ? "{}" : body);

}

catch (JsonException e)

{

Logger?.CouldNotParseJson(body, context.Request.Path, e.Message);

node = null;

}

if (node is not null)

{

try

{

jsonObj = node?.AsObject();

bodyDict = jsonObj?.ToDictionary();

}

catch (Exception e)

{

Logger?.CouldNotParseJson(body, context.Request.Path, e.Message);

bodyDict = null;

}

}

}

}

// start query string parameters

if (endpoint.RequestParamType == RequestParamType.QueryString)

{

if (queryCollection is null)

{

shouldCommit = false;

uploadHandler?.OnError(connection, context, null);

context.Response.StatusCode = StatusCodes.Status404NotFound;

await context.Response.CompleteAsync();

return;

}

// Account for path parameters when counting query string parameters

var pathParamCount = endpoint.PathParameters?.Length ?? 0;

if (queryCollection.Count + pathParamCount != routine.ParamCount && overloads.Count > 0)

{

if (overloads.TryGetValue(string.Concat(entry.Key, queryCollection.Count + pathParamCount), out var overload))

{

routine = overload.Endpoint.Routine;

endpoint = overload.Endpoint;

formatter = overload.Formatter;

if (formatter.IsFormattable is false)

{

// Reinitialize commandTextBuilder for the new routine

commandTextBuilder?.Clear();

if (commandTextBuilder is null)

{

commandTextBuilder = StringBuilderPool.Rent(routine.Expression.Length + routine.ParamCount * 16);

}

commandTextBuilder.Append(routine.Expression);

}

}

}

for (int i = 0; i < routine.Parameters.Length; i++)

{

var parameter = routine.Parameters[i].NpgsqlRestParameterMemberwiseClone();

// Resolved parameter: skip HTTP filling, set to DBNull.Value placeholder.

// The actual value will be resolved via SQL expression after the parameter loop.

if (endpoint.ResolvedParameterExpressions is not null &&

(endpoint.ResolvedParameterExpressions.ContainsKey(parameter.ActualName) ||

endpoint.ResolvedParameterExpressions.ContainsKey(parameter.ConvertedName)))

{

parameter.Value = DBNull.Value;

if (Options.HttpClientOptions.Enabled)

{

if (parameter.TypeDescriptor.CustomType is not null)

{

if (HttpClientTypes.Definitions.ContainsKey(parameter.TypeDescriptor.CustomType))

{

customHttpTypes.Add(parameter.TypeDescriptor.CustomType);

}

}

}

command.Parameters.Add(parameter);

hasNulls = true;

if (formatter.IsFormattable is false)

{

if (formatter.RefContext)

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex, context));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

else

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex));

}

}

paramIndex++;

if (shouldLog && Options.LogCommandParameters)

{

cmdLog!.Append("-- $").Append(paramIndex).Append(' ')

.Append(parameter.TypeDescriptor.OriginalType)

.Append(" = (resolved)").AppendLine();

}

continue;

}

if (parameter.HashOf is not null)

{

var hashValueQueryCollection = queryCollection.TryGetValue(parameter.HashOf.ConvertedName, out var hashQsValue) ? hashQsValue.ToString() : null;

if (string.IsNullOrEmpty(hashValueQueryCollection) is true)

{

parameter.Value = DBNull.Value;

}

else

{

parameter.Value = Options.AuthenticationOptions.PasswordHasher?.HashPassword(hashValueQueryCollection) as object ?? DBNull.Value;

}

}

if (endpoint.UseUserParameters is true)

{

if (parameter.IsFromUserClaims is true && claimsDict is null)

{

claimsDict = context.User.BuildClaimsDictionary(Options.AuthenticationOptions);

}

if (parameter.IsIpAddress is true)

{

parameter.Value = context.Request.GetClientIpAddressDbParam();

}

else if (context.User?.Identity?.IsAuthenticated is true && Options.AuthenticationOptions.ParameterNameClaimsMapping.TryGetValue(parameter.ActualName, out var claimType))

{

parameter.Value = claimsDict!.GetClaimDbParam(claimType);

}

else if (context.User?.Identity?.IsAuthenticated is true && parameter.IsUserClaims is true)

{

parameter.Value = claimsDict?.GetUserClaimsDbParam() ?? DBNull.Value;

}

}

if (parameter.IsUploadMetadata is true)

{

//uploadMetaParamIndex = i;

uploadMetaParamIndex = command.Parameters.Count; // the last one added

parameter.Value = DBNull.Value;

}

// body parameter

if (parameter.Value is null && endpoint.HasBodyParameter &&

(

string.Equals(endpoint.BodyParameterName, parameter.ConvertedName, StringComparison.Ordinal) ||

string.Equals(endpoint.BodyParameterName, parameter.ActualName, StringComparison.Ordinal)

)

)

{

if (body is null)

{

parameter.ParamType = ParamType.BodyParam;

parameter.Value = DBNull.Value;

hasNulls = true;

if (Options.ValidateParameters is not null)

{

Options.ValidateParameters(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (Options.ValidateParametersAsync is not null)

{

await Options.ValidateParametersAsync(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (endpoint.Cached is true && endpoint.CachedParams is not null)

{

if (endpoint.CachedParams.Contains(parameter.ConvertedName) || endpoint.CachedParams.Contains(parameter.ActualName))

{

cacheKeys?.Append(NpgsqlRestParameter.GetCacheKeySeparator());

cacheKeys?.Append(parameter.GetCacheStringValue());

}

}

if (Options.HttpClientOptions.Enabled)

{

if (parameter.TypeDescriptor.CustomType is not null)

{

if (HttpClientTypes.Definitions.ContainsKey(parameter.TypeDescriptor.CustomType))

{

customHttpTypes.Add(parameter.TypeDescriptor.CustomType);

}

}

}

command.Parameters.Add(parameter);

if (hasNulls is false && parameter.Value == DBNull.Value)

{

hasNulls = true;

}

if (formatter.IsFormattable is false)

{

if (formatter.RefContext)

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex, context));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

else

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex));

}

}

paramIndex++;

if (shouldLog && Options.LogCommandParameters)

{

var p = Options.AuthenticationOptions.ObfuscateAuthParameterLogValues && endpoint.IsAuth ?

"***" :

FormatParameterForLog(parameter);

cmdLog!.Append("-- $").Append(paramIndex).Append(' ')

.Append(parameter.TypeDescriptor.OriginalType)

.Append(" = ").AppendLine(p);

}

}

else

{

StringValues bodyStringValues = body;

if (TryParseParameter(parameter, ref bodyStringValues, endpoint.QueryStringNullHandling))

{

parameter.ParamType = ParamType.BodyParam;

hasNulls = false;

if (Options.ValidateParameters is not null)

{

Options.ValidateParameters(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (Options.ValidateParametersAsync is not null)

{

await Options.ValidateParametersAsync(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (endpoint.Cached is true && endpoint.CachedParams is not null)

{

if (endpoint.CachedParams.Contains(parameter.ConvertedName) || endpoint.CachedParams.Contains(parameter.ActualName))

{

cacheKeys?.Append(NpgsqlRestParameter.GetCacheKeySeparator());

cacheKeys?.Append(parameter.GetCacheStringValue());

}

}

if (Options.HttpClientOptions.Enabled)

{

if (parameter.TypeDescriptor.CustomType is not null)

{

if (HttpClientTypes.Definitions.ContainsKey(parameter.TypeDescriptor.CustomType))

{

customHttpTypes.Add(parameter.TypeDescriptor.CustomType);

}

}

}

command.Parameters.Add(parameter);

if (hasNulls is false && parameter.Value == DBNull.Value)

{

hasNulls = true;

}

if (formatter.IsFormattable is false)

{

if (formatter.RefContext)

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex, context));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

else

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex));

}

}

paramIndex++;

if (shouldLog && Options.LogCommandParameters)

{

var p = Options.AuthenticationOptions.ObfuscateAuthParameterLogValues && endpoint.IsAuth ?

"***" :

FormatParameterForLog(parameter);

cmdLog!.Append("-- $").Append(paramIndex).Append(' ')

.Append(parameter.TypeDescriptor.OriginalType)

.Append(" = ").AppendLine(p);

}

}

}

continue;

}

// header parameter

if (parameter.Value is null &&

endpoint.RequestHeadersMode == RequestHeadersMode.Parameter &&

parameter.TypeDescriptor.HasDefault is true &&

(

string.Equals(endpoint.RequestHeadersParameterName, parameter.ConvertedName, StringComparison.Ordinal) ||

string.Equals(endpoint.RequestHeadersParameterName, parameter.ActualName, StringComparison.Ordinal)

)

)

{

if (queryCollection.ContainsKey(parameter.ConvertedName) is false)

{

if (headers is null)

{

SearializeHeader(Options, context, ref headers);

}

parameter.ParamType = ParamType.HeaderParam;

parameter.Value = headers;

if (Options.ValidateParameters is not null)

{

Options.ValidateParameters(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (Options.ValidateParametersAsync is not null)

{

await Options.ValidateParametersAsync(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (endpoint.Cached is true && endpoint.CachedParams is not null)

{

if (endpoint.CachedParams.Contains(parameter.ConvertedName) || endpoint.CachedParams.Contains(parameter.ActualName))

{

cacheKeys?.Append(NpgsqlRestParameter.GetCacheKeySeparator());

cacheKeys?.Append(parameter.GetCacheStringValue());

}

}

if (Options.HttpClientOptions.Enabled)

{

if (parameter.TypeDescriptor.CustomType is not null)

{

if (HttpClientTypes.Definitions.ContainsKey(parameter.TypeDescriptor.CustomType))

{

customHttpTypes.Add(parameter.TypeDescriptor.CustomType);

}

}

}

command.Parameters.Add(parameter);

if (hasNulls is false && parameter.Value == DBNull.Value)

{

hasNulls = true;

}

if (formatter.IsFormattable is false)

{

if (formatter.RefContext)

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex, context));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

else

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex));

}

}

paramIndex++;

if (shouldLog && Options.LogCommandParameters)

{

var p = Options.AuthenticationOptions.ObfuscateAuthParameterLogValues && endpoint.IsAuth ?

"***" :

FormatParameterForLog(parameter);

cmdLog!.Append("-- $").Append(paramIndex).Append(' ')

.Append(parameter.TypeDescriptor.OriginalType)

.Append(" = ").AppendLine(p);

}

continue;

}

}

// path parameter - extract from RouteValues

if (parameter.Value is null && endpoint.HasPathParameters)

{

// Use optimized O(1) lookup via HashSet

string? matchedPathParam = endpoint.FindMatchingPathParameter(parameter.ConvertedName, parameter.ActualName);

// Try to get route value using the path parameter name from the template

if (matchedPathParam is not null && context.Request.RouteValues.TryGetValue(matchedPathParam, out var routeValue))

{

StringValues pathStringValues = routeValue?.ToString() ?? "";

if (TryParseParameter(parameter, ref pathStringValues, endpoint.QueryStringNullHandling))

{

parameter.ParamType = ParamType.PathParam;

parameter.QueryStringValues = pathStringValues;

if (Options.ValidateParameters is not null)

{

Options.ValidateParameters(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (Options.ValidateParametersAsync is not null)

{

await Options.ValidateParametersAsync(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (endpoint.Cached is true && endpoint.CachedParams is not null)

{

if (endpoint.CachedParams.Contains(parameter.ConvertedName) || endpoint.CachedParams.Contains(parameter.ActualName))

{

cacheKeys?.Append(NpgsqlRestParameter.GetCacheKeySeparator());

cacheKeys?.Append(parameter.GetCacheStringValue());

}

}

if (Options.HttpClientOptions.Enabled)

{

if (parameter.TypeDescriptor.CustomType is not null)

{

if (HttpClientTypes.Definitions.ContainsKey(parameter.TypeDescriptor.CustomType))

{

customHttpTypes.Add(parameter.TypeDescriptor.CustomType);

}

}

}

command.Parameters.Add(parameter);

if (hasNulls is false && parameter.Value == DBNull.Value)

{

hasNulls = true;

}

if (formatter.IsFormattable is false)

{

if (formatter.RefContext)

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex, context));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

else

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex));

}

}

paramIndex++;

if (shouldLog && Options.LogCommandParameters)

{

var p = Options.AuthenticationOptions.ObfuscateAuthParameterLogValues && endpoint.IsAuth ?

"***" :

FormatParameterForLog(parameter);

cmdLog!.Append("-- $").Append(paramIndex).Append(' ')

.Append(parameter.TypeDescriptor.OriginalType)

.Append(" = ").AppendLine(p);

}

continue;

}

}

}

if (queryCollection.TryGetValue(parameter.ConvertedName, out var qsValue) is false)

{

if (parameter.Value is null)

{

if (Options.ValidateParameters is not null)

{

Options.ValidateParameters(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (Options.ValidateParametersAsync is not null)

{

await Options.ValidateParametersAsync(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (parameter.Value is null)

{

if (parameter.TypeDescriptor.CustomType is not null)

{

parameter.Value = DBNull.Value;

}

else if (IsProxyResponseParameter(endpoint, parameter))

{

// Proxy response parameter - set placeholder value and fall through to add it

parameter.Value = DBNull.Value;

parameter.ParamType = ParamType.QueryString;

}

else if (parameter.TypeDescriptor.HasDefault is false)

{

shouldCommit = false;

uploadHandler?.OnError(connection, context, null);

context.Response.StatusCode = StatusCodes.Status404NotFound;

await context.Response.CompleteAsync();

return;

}

else

{

continue;

}

}

}

}

if (parameter.Value is null && TryParseParameter(parameter, ref qsValue, endpoint.QueryStringNullHandling) is false)

{

// Check if this is a proxy response parameter - it will be filled in later

if (IsProxyResponseParameter(endpoint, parameter))

{

parameter.Value = DBNull.Value;

parameter.ParamType = ParamType.QueryString;

// Don't skip - fall through to add the parameter

}

else if (parameter.TypeDescriptor.HasDefault is false)

{

shouldCommit = false;

uploadHandler?.OnError(connection, context, null);

context.Response.StatusCode = StatusCodes.Status404NotFound;

await context.Response.CompleteAsync();

return;

}

else

{

continue;

}

}

parameter.ParamType = ParamType.QueryString;

parameter.QueryStringValues = qsValue;

if (Options.ValidateParameters is not null)

{

Options.ValidateParameters(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (Options.ValidateParametersAsync is not null)

{

await Options.ValidateParametersAsync(new ParameterValidationValues(

context,

routine,

parameter));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

if (endpoint.Cached is true && endpoint.CachedParams is not null)

{

if (endpoint.CachedParams.Contains(parameter.ConvertedName) || endpoint.CachedParams.Contains(parameter.ActualName))

{

cacheKeys?.Append(NpgsqlRestParameter.GetCacheKeySeparator());

cacheKeys?.Append(parameter.GetCacheStringValue());

}

}

if (Options.HttpClientOptions.Enabled)

{

if (parameter.TypeDescriptor.CustomType is not null)

{

if (HttpClientTypes.Definitions.ContainsKey(parameter.TypeDescriptor.CustomType))

{

customHttpTypes.Add(parameter.TypeDescriptor.CustomType);

}

}

}

command.Parameters.Add(parameter);

if (hasNulls is false && parameter.Value == DBNull.Value)

{

hasNulls = true;

}

if (formatter.IsFormattable is false)

{

if (formatter.RefContext)

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex, context));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

else

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex));

}

}

paramIndex++;

if (shouldLog && Options.LogCommandParameters)

{

var p = Options.AuthenticationOptions.ObfuscateAuthParameterLogValues && endpoint.IsAuth ?

"***" :

FormatParameterForLog(parameter);

cmdLog!.Append("-- $").Append(paramIndex).Append(' ')

.Append(parameter.TypeDescriptor.OriginalType)

.Append(" = ").AppendLine(p);

}

}

// Skip query string validation for passthrough proxy endpoints - query will be forwarded as-is

if (!(endpoint.IsProxy && Options.ProxyOptions.Enabled && !endpoint.HasProxyResponseParameters))

{

foreach (var queryKey in queryCollection.Keys)

{

if (routine.ParamsHash.Contains(queryKey) is false)

{

shouldCommit = false;

uploadHandler?.OnError(connection, context, null);

context.Response.StatusCode = StatusCodes.Status404NotFound;

await context.Response.CompleteAsync();

return;

}

}

}

} // end of query string parameters

// start json body parameters

else if (endpoint.RequestParamType == RequestParamType.BodyJson)

{

if (bodyDict is null)

{

shouldCommit = false;

uploadHandler?.OnError(connection, context, null);

context.Response.StatusCode = StatusCodes.Status404NotFound;

await context.Response.CompleteAsync();

return;

}

// Account for path parameters when counting body parameters

var pathParamCount = endpoint.PathParameters?.Length ?? 0;

var expectedBodyParamCount = routine.ParamCount - pathParamCount;

if (bodyDict.Count != expectedBodyParamCount && overloads.Count > 0)

{

if (overloads.TryGetValue(string.Concat(entry.Key, bodyDict.Count + pathParamCount), out var overload))

{

routine = overload.Endpoint.Routine;

endpoint = overload.Endpoint;

formatter = overload.Formatter;

if (formatter.IsFormattable is false)

{

// Reinitialize commandTextBuilder for the new routine

commandTextBuilder?.Clear();

if (commandTextBuilder is null)

{

commandTextBuilder = StringBuilderPool.Rent(routine.Expression.Length + routine.ParamCount * 16);

}

commandTextBuilder.Append(routine.Expression);

}

}

}

for (int i = 0; i < routine.Parameters.Length; i++)

{

var parameter = routine.Parameters[i].NpgsqlRestParameterMemberwiseClone();

// Resolved parameter: skip HTTP filling, set to DBNull.Value placeholder.

// The actual value will be resolved via SQL expression after the parameter loop.

if (endpoint.ResolvedParameterExpressions is not null &&

(endpoint.ResolvedParameterExpressions.ContainsKey(parameter.ActualName) ||

endpoint.ResolvedParameterExpressions.ContainsKey(parameter.ConvertedName)))

{

parameter.Value = DBNull.Value;

if (Options.HttpClientOptions.Enabled)

{

if (parameter.TypeDescriptor.CustomType is not null)

{

if (HttpClientTypes.Definitions.ContainsKey(parameter.TypeDescriptor.CustomType))

{

customHttpTypes.Add(parameter.TypeDescriptor.CustomType);

}

}

}

command.Parameters.Add(parameter);

hasNulls = true;

if (formatter.IsFormattable is false)

{

if (formatter.RefContext)

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex, context));

if (context.Response.HasStarted || context.Response.StatusCode != (int)HttpStatusCode.OK)

{

return;

}

}

else

{

commandTextBuilder!.Append(formatter.AppendCommandParameter(parameter, paramIndex));

}

}

paramIndex++;

if (shouldLog && Options.LogCommandParameters)

{

cmdLog!.Append("-- $").Append(paramIndex).Append(' ')

.Append(parameter.TypeDescriptor.OriginalType)

.Append(" = (resolved)").AppendLine();

}

continue;

}

if (parameter.HashOf is not null)

{

var hashValueBodyDict = bodyDict.GetValueOrDefault(parameter.HashOf.ConvertedName)?.ToString();

if (string.IsNullOrEmpty(hashValueBodyDict) is true)

{

parameter.Value = DBNull.Value;

}

else

{

parameter.Value = Options.AuthenticationOptions.PasswordHasher?.HashPassword(hashValueBodyDict) as object ?? DBNull.Value;

}

}

if (endpoint.UseUserParameters is true)

{