using NpgsqlTypes;

namespace NpgsqlRest;

public readonly struct TypeDescriptor
{
    public string OriginalType { get; }
    public string Type { get; }
    public bool IsArray { get; }
    public bool IsNumeric { get; }
    public bool IsJson { get; }
    public bool IsDate { get; }
    public bool IsDateTime { get; }
    public bool IsBoolean { get; }
    public bool IsText { get; }
    public NpgsqlDbType DbType { get; }
    public NpgsqlDbType BaseDbType { get; }
    public NpgsqlDbType ActualDbType { get; }
    public bool HasDefault { get; }
    public bool NeedsEscape { get; }
    public bool IsPk { get; }
    public bool IsIdentity { get; }

    public TypeDescriptor(string type, bool hasDefault = false, bool isPk = false, bool isIdentity = false)
    {
        OriginalType = type;
        HasDefault = hasDefault;
        IsPk = isPk;
        IsIdentity = isIdentity;
        IsArray = type.EndsWith("[]");
        Type = (IsArray ? type[..^2] : type).Trim('"');
        DbType = GetDbType();
        BaseDbType = DbType;

        ActualDbType = CastToText(BaseDbType) ? NpgsqlDbType.Text : BaseDbType;

        if (IsArray)
        {
            DbType |= NpgsqlDbType.Array;
            ActualDbType |= NpgsqlDbType.Array;
        }

        IsNumeric = BaseDbType switch
        {
            NpgsqlDbType.Smallint => true,
            NpgsqlDbType.Integer => true,
            NpgsqlDbType.Bigint => true,
            NpgsqlDbType.Numeric => true,
            NpgsqlDbType.Real => true,
            NpgsqlDbType.Double => true,
            NpgsqlDbType.Money => true,
            _ => false
        };

        IsJson = BaseDbType switch
        {
            NpgsqlDbType.Jsonb => true,
            NpgsqlDbType.Json => true,
            _ => false
        };
        
        IsDate = BaseDbType == NpgsqlDbType.Date;
        
        IsBoolean = BaseDbType == NpgsqlDbType.Boolean;

        IsDateTime = BaseDbType == NpgsqlDbType.Timestamp || BaseDbType == NpgsqlDbType.TimestampTz;

        IsText = BaseDbType == NpgsqlDbType.Text ||
            BaseDbType == NpgsqlDbType.Xml ||
            BaseDbType == NpgsqlDbType.Varchar ||
            BaseDbType == NpgsqlDbType.Char ||
            BaseDbType == NpgsqlDbType.Name ||
            BaseDbType == NpgsqlDbType.Jsonb ||
            BaseDbType == NpgsqlDbType.Json ||
            BaseDbType == NpgsqlDbType.JsonPath;

        NeedsEscape = BaseDbType == NpgsqlDbType.Xml ||
            BaseDbType == NpgsqlDbType.Varchar ||
            BaseDbType == NpgsqlDbType.Char ||
            BaseDbType == NpgsqlDbType.Name ||
            BaseDbType == NpgsqlDbType.JsonPath ||
            BaseDbType == NpgsqlDbType.Bytea ||
            BaseDbType == NpgsqlDbType.TsQuery ||
            BaseDbType == NpgsqlDbType.TsVector ||
            BaseDbType == NpgsqlDbType.Citext ||
            BaseDbType == NpgsqlDbType.LQuery ||
            BaseDbType == NpgsqlDbType.LTree ||
            BaseDbType == NpgsqlDbType.LTxtQuery ||
            BaseDbType == NpgsqlDbType.TsVector ||
            BaseDbType == NpgsqlDbType.Hstore;
    }

    public bool IsCastToText() => CastToText(BaseDbType);

    private static bool CastToText(NpgsqlDbType type) => type switch
    {
        NpgsqlDbType.Interval => true,
        NpgsqlDbType.Bit => true,
        NpgsqlDbType.Varbit => true,
        NpgsqlDbType.Bytea => true,
        NpgsqlDbType.Inet => true,
        NpgsqlDbType.MacAddr => true,
        NpgsqlDbType.Cidr => true,
        NpgsqlDbType.MacAddr8 => true,
        NpgsqlDbType.TsQuery => true,
        NpgsqlDbType.TsVector => true,
        NpgsqlDbType.Box => true,
        NpgsqlDbType.Circle => true,
        NpgsqlDbType.Line => true,
        NpgsqlDbType.LSeg => true,
        NpgsqlDbType.Path => true,
        NpgsqlDbType.Point => true,
        NpgsqlDbType.Polygon => true,
        NpgsqlDbType.Oid => true,
        NpgsqlDbType.Xid => true,
        NpgsqlDbType.Xid8 => true,
        NpgsqlDbType.Cid => true,
        NpgsqlDbType.Regtype => true,
        NpgsqlDbType.Regconfig => true,
        NpgsqlDbType.IntegerRange => true,
        NpgsqlDbType.BigIntRange => true,
        NpgsqlDbType.NumericRange => true,
        NpgsqlDbType.TimestampRange => true,
        NpgsqlDbType.TimestampTzRange => true,
        NpgsqlDbType.DateRange => true,
        NpgsqlDbType.IntegerMultirange => true,
        NpgsqlDbType.BigIntMultirange => true,
        NpgsqlDbType.NumericMultirange => true,
        NpgsqlDbType.TimestampMultirange => true,
        NpgsqlDbType.TimestampTzMultirange => true,
        NpgsqlDbType.DateMultirange => true,
        NpgsqlDbType.Int2Vector => true,
        NpgsqlDbType.Oidvector => true,
        NpgsqlDbType.PgLsn => true,
        NpgsqlDbType.Tid => true,
        NpgsqlDbType.Citext => true,
        NpgsqlDbType.LQuery => true,
        NpgsqlDbType.LTree => true,
        NpgsqlDbType.LTxtQuery => true,
        NpgsqlDbType.Hstore => true,
        NpgsqlDbType.Geometry => true,
        NpgsqlDbType.Geography => true,
        _ => false
    };

    private NpgsqlDbType GetDbType()
    {
        var result = Type switch
        {
            "smallint" => NpgsqlDbType.Smallint,
            "integer" => NpgsqlDbType.Integer,
            "bigint" => NpgsqlDbType.Bigint,
            "decimal" => NpgsqlDbType.Numeric,
            "numeric" => NpgsqlDbType.Numeric,
            "real" => NpgsqlDbType.Real,
            "double precision" => NpgsqlDbType.Double,
            "int2" => NpgsqlDbType.Smallint,
            "int4" => NpgsqlDbType.Integer,
            "int8" => NpgsqlDbType.Bigint,
            "float4" => NpgsqlDbType.Real,
            "float8" => NpgsqlDbType.Double,
            "money" => NpgsqlDbType.Money,
            "smallserial" => NpgsqlDbType.Smallint,
            "serial" => NpgsqlDbType.Integer,
            "bigserial" => NpgsqlDbType.Bigint,

            "text" => NpgsqlDbType.Text,
            "xml" => NpgsqlDbType.Xml,
            "varchar" => NpgsqlDbType.Varchar,
            "character varying" => NpgsqlDbType.Varchar,
            "bpchar" => NpgsqlDbType.Char,
            "character" => NpgsqlDbType.Char,
            "char" => NpgsqlDbType.Char,
            "name" => NpgsqlDbType.Name,
            "refcursor" => NpgsqlDbType.Refcursor,
            "jsonb" => NpgsqlDbType.Jsonb,
            "json" => NpgsqlDbType.Json,
            "jsonpath" => NpgsqlDbType.JsonPath,

            "timestamp" => NpgsqlDbType.Timestamp,
            "timestamptz" => NpgsqlDbType.TimestampTz,
            "timestamp without time zone" => NpgsqlDbType.Timestamp,
            "timestamp with time zone" => NpgsqlDbType.TimestampTz,
            "date" => NpgsqlDbType.Date,
            "time" => NpgsqlDbType.Time,
            "timetz" => NpgsqlDbType.TimeTz,
            "time without time zone" => NpgsqlDbType.Time,
            "time with time zone" => NpgsqlDbType.TimeTz,
            "interval" => NpgsqlDbType.Interval,

            "bool" => NpgsqlDbType.Boolean,
            "boolean" => NpgsqlDbType.Boolean,
            "bytea" => NpgsqlDbType.Bytea,
            "uuid" => NpgsqlDbType.Uuid,
            "bit varying" => NpgsqlDbType.Varbit,
            "varbit" => NpgsqlDbType.Varbit,
            "bit" => NpgsqlDbType.Bit,

            "cidr" => NpgsqlDbType.Cidr,
            "inet" => NpgsqlDbType.Inet,
            "macaddr" => NpgsqlDbType.MacAddr,
            "macaddr8" => NpgsqlDbType.MacAddr8,

            "tsquery" => NpgsqlDbType.TsQuery,
            "tsvector" => NpgsqlDbType.TsVector,

            "box" => NpgsqlDbType.Box,
            "circle" => NpgsqlDbType.Circle,
            "line" => NpgsqlDbType.Line,
            "lseg" => NpgsqlDbType.LSeg,
            "path" => NpgsqlDbType.Path,
            "point" => NpgsqlDbType.Point,
            "polygon" => NpgsqlDbType.Polygon,

            "oid" => NpgsqlDbType.Oid,
            "xid" => NpgsqlDbType.Xid,
            "xid8" => NpgsqlDbType.Xid8,
            "cid" => NpgsqlDbType.Cid,
            "regtype" => NpgsqlDbType.Regtype,
            "regconfig" => NpgsqlDbType.Regconfig,

            "int4range" => NpgsqlDbType.IntegerRange,
            "int8range" => NpgsqlDbType.BigIntRange,
            "numrange" => NpgsqlDbType.NumericRange,
            "tsrange" => NpgsqlDbType.TimestampRange,
            "tstzrange" => NpgsqlDbType.TimestampTzRange,
            "daterange" => NpgsqlDbType.DateRange,

            "int4multirange" => NpgsqlDbType.IntegerMultirange,
            "int8multirange" => NpgsqlDbType.BigIntMultirange,
            "nummultirange" => NpgsqlDbType.NumericMultirange,
            "tsmultirange" => NpgsqlDbType.TimestampMultirange,
            "tstzmultirange" => NpgsqlDbType.TimestampTzMultirange,
            "datemultirange" => NpgsqlDbType.DateMultirange,

            "int2vector" => NpgsqlDbType.Int2Vector,
            "oidvector" => NpgsqlDbType.Oidvector,
            "pg_lsn" => NpgsqlDbType.PgLsn,
            "tid" => NpgsqlDbType.Tid,

            "citext" => NpgsqlDbType.Citext,
            "lquery" => NpgsqlDbType.LQuery,
            "ltree" => NpgsqlDbType.LTree,
            "ltxtquery" => NpgsqlDbType.LTxtQuery,
            "hstore" => NpgsqlDbType.Hstore,
            "geometry" => NpgsqlDbType.Geometry,
            "geography" => NpgsqlDbType.Geography,

            _ => NpgsqlDbType.Unknown
        };
        return result;
    }
}