using System.Text;
using System.Text.Json.Nodes;
using Microsoft.Extensions.Primitives;
using Npgsql;

namespace NpgsqlRest;

public class NpgsqlRestParameter : NpgsqlParameter
{
    public int Ordinal { get; private set; }
    public string ConvertedName { get; internal set; }
    public string ActualName { get; internal set; }
    /// <summary>
    /// The original PostgreSQL parameter name as defined in the function/source.
    /// Never changed by annotations. Used by RoutineSourceParameterFormatter for SQL generation.
    /// </summary>
    public string OriginalName { get; init; }
    /// <summary>
    /// Virtual parameters exist for HTTP matching and claim mapping but are NOT bound
    /// to the PostgreSQL command. Created by @define_param annotation on SQL file endpoints.
    /// </summary>
    public bool IsVirtual { get; init; }
    public TypeDescriptor TypeDescriptor { get; set; }

    /// <summary>
    /// For HTTP custom type composite parameters, the field names in order.
    /// Used to build the composite text value without depending on CompositeTypeCache.
    /// </summary>
    public string[]? CompositeFieldNames { get; set; }

    public ParamType ParamType { get; set; } = default!;
    public StringValues? QueryStringValues { get; set; } = null;
    public JsonNode? JsonBodyNode { get; set; } = null;
    public NpgsqlRestParameter? HashOf { get; set; } = null;

    /// <summary>
    /// The original string representation of the parameter value as received from the request
    /// (query string or JSON body). Used for cache key generation to ensure consistency.
    /// </summary>
    public string? OriginalStringValue { get; set; } = null;

    public bool IsUploadMetadata { get; set; } = false;

    /// <summary>
    /// Explicit default value set by annotation (e.g., @param my_param default null).
    /// null = no explicit default (use existing HasDefault/PostgreSQL behavior).
    /// DBNull.Value = default is SQL NULL.
    /// Any other value = the default value as a string to be bound.
    /// </summary>
    public object? DefaultValue { get; set; } = null;

    public bool IsIpAddress { get; set; } = false;
    public string? UserClaim { get; set; } = null;
    public bool IsUserClaims { get; set; } = false;
    public bool IsFromUserClaims => UserClaim is not null || IsUserClaims is true;

    public NpgsqlRestParameter(
        int ordinal, 
        string convertedName, 
        string actualName, 
        TypeDescriptor typeDescriptor)
    {
        Ordinal = ordinal;
        ConvertedName = convertedName;
        OriginalName = actualName;

        ActualName = actualName;
        TypeDescriptor = typeDescriptor;
        NpgsqlDbType = typeDescriptor.ActualDbType;

        if (actualName is not null && 
            Options.AuthenticationOptions.ParameterNameClaimsMapping.TryGetValue(actualName, out var claimName))
        {
            UserClaim = claimName;
        }

        if (actualName is not null && 
            string.Equals(Options.AuthenticationOptions.IpAddressParameterName, actualName, StringComparison.OrdinalIgnoreCase))
        {
            IsIpAddress = true;
        }

        if (actualName is not null && 
            string.Equals(Options.AuthenticationOptions.ClaimsJsonParameterName, actualName, StringComparison.OrdinalIgnoreCase))
        {
            IsUserClaims = true;
        }

        if (Options.UploadOptions.UseDefaultUploadMetadataParameter is true)
        {
            if (string.Equals(Options.UploadOptions.DefaultUploadMetadataParameterName, actualName, StringComparison.OrdinalIgnoreCase))
            {
                IsUploadMetadata = true;
            }
        }
    }

    private const char CacheKeySeparator = '\x1F'; // Unit Separator - non-printable ASCII character
    private const string CacheKeyNull = "\x00NULL\x00"; // Distinct marker for null/DBNull values

    internal string GetCacheStringValue()
    {
        if (Value is null || Value == DBNull.Value)
        {
            return CacheKeyNull;
        }
        // Prefer original string value from request for consistency
        if (OriginalStringValue is not null)
        {
            return OriginalStringValue;
        }
        // Fallback for internally-set values (user claims, IP address, etc.)
        if (TypeDescriptor.IsArray)
        {
            // Arrays can be stored as List<object?> (from query string parsing) or object[]
            IList<object?>? list = Value as IList<object?>;
            if (list is null || list.Count == 0)
            {
                return "[]";
            }
            var sb = new StringBuilder();
            sb.Append('[');
            for (int i = 0; i < list.Count; i++)
            {
                if (i > 0)
                {
                    sb.Append(CacheKeySeparator);
                }
                sb.Append(list[i]?.ToString() ?? CacheKeyNull);
            }
            sb.Append(']');
            return sb.ToString();
        }
        return Value.ToString() ?? CacheKeyNull;
    }

    internal static char GetCacheKeySeparator() => CacheKeySeparator;

#pragma warning disable CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider adding the 'required' modifier or declaring as nullable.
#pragma warning disable CS8603 // Possible null reference return.
    public NpgsqlRestParameter NpgsqlRestParameterMemberwiseClone() => MemberwiseClone() as NpgsqlRestParameter;
    private NpgsqlRestParameter() { }
#pragma warning restore CS8603 // Possible null reference return.
#pragma warning restore CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider adding the 'required' modifier or declaring as nullable.

    private static readonly NpgsqlRestParameter TextParam = new()
    {
        NpgsqlDbType = NpgsqlTypes.NpgsqlDbType.Text,
        Value = DBNull.Value,
    };

    public static NpgsqlParameter CreateParamWithType(NpgsqlTypes.NpgsqlDbType type)
    {
        var result = TextParam.NpgsqlRestParameterMemberwiseClone();
        result.NpgsqlDbType = type;
        return result;
    }

    public static NpgsqlParameter CreateTextParam(object? value)
    {
        var result = TextParam.NpgsqlRestParameterMemberwiseClone();
        if (value is null)
        {
            return result;
        }
        result.Value = value;
        return result;
    }
}