@prisma-next/adapter-postgres 0.3.0-dev.11 → 0.3.0-dev.114

package/src/core/codec-ids.ts

@@ -0,0 +1,28 @@
+export {
+  SQL_CHAR_CODEC_ID,
+  SQL_FLOAT_CODEC_ID,
+  SQL_INT_CODEC_ID,
+  SQL_VARCHAR_CODEC_ID,
+} from '@prisma-next/sql-relational-core/ast';
+export const PG_TEXT_CODEC_ID = 'pg/text@1' as const;
+export const PG_ENUM_CODEC_ID = 'pg/enum@1' as const;
+export const PG_CHAR_CODEC_ID = 'pg/char@1' as const;
+export const PG_VARCHAR_CODEC_ID = 'pg/varchar@1' as const;
+export const PG_INT_CODEC_ID = 'pg/int@1' as const;
+export const PG_INT2_CODEC_ID = 'pg/int2@1' as const;
+export const PG_INT4_CODEC_ID = 'pg/int4@1' as const;
+export const PG_INT8_CODEC_ID = 'pg/int8@1' as const;
+export const PG_FLOAT_CODEC_ID = 'pg/float@1' as const;
+export const PG_FLOAT4_CODEC_ID = 'pg/float4@1' as const;
+export const PG_FLOAT8_CODEC_ID = 'pg/float8@1' as const;
+export const PG_NUMERIC_CODEC_ID = 'pg/numeric@1' as const;
+export const PG_BOOL_CODEC_ID = 'pg/bool@1' as const;
+export const PG_BIT_CODEC_ID = 'pg/bit@1' as const;
+export const PG_VARBIT_CODEC_ID = 'pg/varbit@1' as const;
+export const PG_TIMESTAMP_CODEC_ID = 'pg/timestamp@1' as const;
+export const PG_TIMESTAMPTZ_CODEC_ID = 'pg/timestamptz@1' as const;
+export const PG_TIME_CODEC_ID = 'pg/time@1' as const;
+export const PG_TIMETZ_CODEC_ID = 'pg/timetz@1' as const;
+export const PG_INTERVAL_CODEC_ID = 'pg/interval@1' as const;
+export const PG_JSON_CODEC_ID = 'pg/json@1' as const;
+export const PG_JSONB_CODEC_ID = 'pg/jsonb@1' as const;

package/src/core/codecs.ts

@@ -11,12 +11,93 @@
  * at compile time (to derive CodecTypes).
  */
 
-import { codec, defineCodecs } from '@prisma-next/sql-relational-core/ast';
+import type { Codec, CodecMeta, CodecTrait } from '@prisma-next/sql-relational-core/ast';
+import { codec, defineCodecs, sqlCodecDefinitions } from '@prisma-next/sql-relational-core/ast';
+import { ifDefined } from '@prisma-next/utils/defined';
+import { type as arktype } from 'arktype';
+import {
+  PG_BIT_CODEC_ID,
+  PG_BOOL_CODEC_ID,
+  PG_CHAR_CODEC_ID,
+  PG_ENUM_CODEC_ID,
+  PG_FLOAT_CODEC_ID,
+  PG_FLOAT4_CODEC_ID,
+  PG_FLOAT8_CODEC_ID,
+  PG_INT_CODEC_ID,
+  PG_INT2_CODEC_ID,
+  PG_INT4_CODEC_ID,
+  PG_INT8_CODEC_ID,
+  PG_INTERVAL_CODEC_ID,
+  PG_JSON_CODEC_ID,
+  PG_JSONB_CODEC_ID,
+  PG_NUMERIC_CODEC_ID,
+  PG_TEXT_CODEC_ID,
+  PG_TIME_CODEC_ID,
+  PG_TIMESTAMP_CODEC_ID,
+  PG_TIMESTAMPTZ_CODEC_ID,
+  PG_TIMETZ_CODEC_ID,
+  PG_VARBIT_CODEC_ID,
+  PG_VARCHAR_CODEC_ID,
+} from './codec-ids';
+
+const lengthParamsSchema = arktype({
+  length: 'number.integer > 0',
+});
+
+const numericParamsSchema = arktype({
+  precision: 'number.integer > 0 & number.integer <= 1000',
+  'scale?': 'number.integer >= 0',
+});
+
+const precisionParamsSchema = arktype({
+  'precision?': 'number.integer >= 0 & number.integer <= 6',
+});
+
+function aliasCodec<
+  Id extends string,
+  TTraits extends readonly CodecTrait[],
+  TWire,
+  TJs,
+  TParams,
+  THelper,
+>(
+  base: Codec<string, TTraits, TWire, TJs, TParams, THelper>,
+  options: {
+    readonly typeId: Id;
+    readonly targetTypes: readonly string[];
+    readonly meta?: CodecMeta;
+  },
+): Codec<Id, TTraits, TWire, TJs, TParams, THelper> {
+  return {
+    id: options.typeId,
+    targetTypes: options.targetTypes,
+    ...ifDefined('meta', options.meta),
+    ...ifDefined('paramsSchema', base.paramsSchema),
+    ...ifDefined('init', base.init),
+    ...ifDefined('encode', base.encode),
+    ...ifDefined('traits', base.traits),
+    decode: base.decode,
+  } as Codec<Id, TTraits, TWire, TJs, TParams, THelper>;
+}
+
+const sqlCharCodec = sqlCodecDefinitions.char.codec;
+const sqlVarcharCodec = sqlCodecDefinitions.varchar.codec;
+const sqlIntCodec = sqlCodecDefinitions.int.codec;
+const sqlFloatCodec = sqlCodecDefinitions.float.codec;
+
+export type JsonValue =
+  | string
+  | number
+  | boolean
+  | null
+  | { readonly [key: string]: JsonValue }
+  | readonly JsonValue[];
 
 // Create individual codec instances
 const pgTextCodec = codec({
-  typeId: 'pg/text@1',
+  typeId: PG_TEXT_CODEC_ID,
   targetTypes: ['text'],
+  traits: ['equality', 'order', 'textual'],
   encode: (value: string): string => value,
   decode: (wire: string): string => wire,
   meta: {
@@ -30,11 +111,37 @@ const pgTextCodec = codec({
   },
 });
 
-const pgInt4Codec = codec<'pg/int4@1', number, number>({
-  typeId: 'pg/int4@1',
+const pgCharCodec = aliasCodec(sqlCharCodec, {
+  typeId: PG_CHAR_CODEC_ID,
+  targetTypes: ['character'],
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'character',
+        },
+      },
+    },
+  },
+});
+
+const pgVarcharCodec = aliasCodec(sqlVarcharCodec, {
+  typeId: PG_VARCHAR_CODEC_ID,
+  targetTypes: ['character varying'],
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'character varying',
+        },
+      },
+    },
+  },
+});
+
+const pgIntCodec = aliasCodec(sqlIntCodec, {
+  typeId: PG_INT_CODEC_ID,
   targetTypes: ['int4'],
-  encode: (value) => value,
-  decode: (wire) => wire,
   meta: {
     db: {
       sql: {
@@ -46,11 +153,69 @@ const pgInt4Codec = codec<'pg/int4@1', number, number>({
   },
 });
 
-const pgInt2Codec = codec<'pg/int2@1', number, number>({
-  typeId: 'pg/int2@1',
+const pgFloatCodec = aliasCodec(sqlFloatCodec, {
+  typeId: PG_FLOAT_CODEC_ID,
+  targetTypes: ['float8'],
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'double precision',
+        },
+      },
+    },
+  },
+});
+
+const pgInt4Codec = codec({
+  typeId: PG_INT4_CODEC_ID,
+  targetTypes: ['int4'],
+  traits: ['equality', 'order', 'numeric'],
+  encode: (value: number): number => value,
+  decode: (wire: number): number => wire,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'integer',
+        },
+      },
+    },
+  },
+});
+
+const pgNumericCodec = codec<
+  typeof PG_NUMERIC_CODEC_ID,
+  readonly ['equality', 'order', 'numeric'],
+  string,
+  string
+>({
+  typeId: PG_NUMERIC_CODEC_ID,
+  targetTypes: ['numeric', 'decimal'],
+  traits: ['equality', 'order', 'numeric'],
+  encode: (value: string): string => value,
+  decode: (wire: string | number): string => {
+    if (typeof wire === 'number') return String(wire);
+    return wire;
+  },
+  paramsSchema: numericParamsSchema,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'numeric',
+        },
+      },
+    },
+  },
+});
+
+const pgInt2Codec = codec({
+  typeId: PG_INT2_CODEC_ID,
   targetTypes: ['int2'],
-  encode: (value) => value,
-  decode: (wire) => wire,
+  traits: ['equality', 'order', 'numeric'],
+  encode: (value: number): number => value,
+  decode: (wire: number): number => wire,
   meta: {
     db: {
       sql: {
@@ -62,11 +227,12 @@ const pgInt2Codec = codec<'pg/int2@1', number, number>({
   },
 });
 
-const pgInt8Codec = codec<'pg/int8@1', number, number>({
-  typeId: 'pg/int8@1',
+const pgInt8Codec = codec({
+  typeId: PG_INT8_CODEC_ID,
   targetTypes: ['int8'],
-  encode: (value) => value,
-  decode: (wire) => wire,
+  traits: ['equality', 'order', 'numeric'],
+  encode: (value: number): number => value,
+  decode: (wire: number): number => wire,
   meta: {
     db: {
       sql: {
@@ -78,11 +244,12 @@ const pgInt8Codec = codec<'pg/int8@1', number, number>({
   },
 });
 
-const pgFloat4Codec = codec<'pg/float4@1', number, number>({
-  typeId: 'pg/float4@1',
+const pgFloat4Codec = codec({
+  typeId: PG_FLOAT4_CODEC_ID,
   targetTypes: ['float4'],
-  encode: (value) => value,
-  decode: (wire) => wire,
+  traits: ['equality', 'order', 'numeric'],
+  encode: (value: number): number => value,
+  decode: (wire: number): number => wire,
   meta: {
     db: {
       sql: {
@@ -94,11 +261,12 @@ const pgFloat4Codec = codec<'pg/float4@1', number, number>({
   },
 });
 
-const pgFloat8Codec = codec<'pg/float8@1', number, number>({
-  typeId: 'pg/float8@1',
+const pgFloat8Codec = codec({
+  typeId: PG_FLOAT8_CODEC_ID,
   targetTypes: ['float8'],
-  encode: (value) => value,
-  decode: (wire) => wire,
+  traits: ['equality', 'order', 'numeric'],
+  encode: (value: number): number => value,
+  decode: (wire: number): number => wire,
   meta: {
     db: {
       sql: {
@@ -110,19 +278,25 @@ const pgFloat8Codec = codec<'pg/float8@1', number, number>({
   },
 });
 
-const pgTimestampCodec = codec<'pg/timestamp@1', string | Date, string>({
-  typeId: 'pg/timestamp@1',
+const pgTimestampCodec = codec<
+  typeof PG_TIMESTAMP_CODEC_ID,
+  readonly ['equality', 'order'],
+  string | Date,
+  string
+>({
+  typeId: PG_TIMESTAMP_CODEC_ID,
   targetTypes: ['timestamp'],
+  traits: ['equality', 'order'],
   encode: (value: string | Date): string => {
     if (value instanceof Date) return value.toISOString();
     if (typeof value === 'string') return value;
     return String(value);
   },
   decode: (wire: string | Date): string => {
-    if (typeof wire === 'string') return wire;
     if (wire instanceof Date) return wire.toISOString();
-    return String(wire);
+    return wire;
   },
+  paramsSchema: precisionParamsSchema,
   meta: {
     db: {
       sql: {
@@ -134,19 +308,25 @@ const pgTimestampCodec = codec<'pg/timestamp@1', string | Date, string>({
   },
 });
 
-const pgTimestamptzCodec = codec<'pg/timestamptz@1', string | Date, string>({
-  typeId: 'pg/timestamptz@1',
+const pgTimestamptzCodec = codec<
+  typeof PG_TIMESTAMPTZ_CODEC_ID,
+  readonly ['equality', 'order'],
+  string | Date,
+  string
+>({
+  typeId: PG_TIMESTAMPTZ_CODEC_ID,
   targetTypes: ['timestamptz'],
+  traits: ['equality', 'order'],
   encode: (value: string | Date): string => {
     if (value instanceof Date) return value.toISOString();
     if (typeof value === 'string') return value;
     return String(value);
   },
   decode: (wire: string | Date): string => {
-    if (typeof wire === 'string') return wire;
     if (wire instanceof Date) return wire.toISOString();
-    return String(wire);
+    return wire;
   },
+  paramsSchema: precisionParamsSchema,
   meta: {
     db: {
       sql: {
@@ -158,11 +338,53 @@ const pgTimestamptzCodec = codec<'pg/timestamptz@1', string | Date, string>({
   },
 });
 
-const pgBoolCodec = codec<'pg/bool@1', boolean, boolean>({
-  typeId: 'pg/bool@1',
+const pgTimeCodec = codec<typeof PG_TIME_CODEC_ID, readonly ['equality', 'order'], string, string>({
+  typeId: PG_TIME_CODEC_ID,
+  targetTypes: ['time'],
+  traits: ['equality', 'order'],
+  encode: (value: string): string => value,
+  decode: (wire: string): string => wire,
+  paramsSchema: precisionParamsSchema,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'time',
+        },
+      },
+    },
+  },
+});
+
+const pgTimetzCodec = codec<
+  typeof PG_TIMETZ_CODEC_ID,
+  readonly ['equality', 'order'],
+  string,
+  string
+>({
+  typeId: PG_TIMETZ_CODEC_ID,
+  targetTypes: ['timetz'],
+  traits: ['equality', 'order'],
+  encode: (value: string): string => value,
+  decode: (wire: string): string => wire,
+  paramsSchema: precisionParamsSchema,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'timetz',
+        },
+      },
+    },
+  },
+});
+
+const pgBoolCodec = codec({
+  typeId: PG_BOOL_CODEC_ID,
   targetTypes: ['bool'],
-  encode: (value) => value,
-  decode: (wire) => wire,
+  traits: ['equality', 'boolean'],
+  encode: (value: boolean): boolean => value,
+  decode: (wire: boolean): boolean => wire,
   meta: {
     db: {
       sql: {
@@ -174,21 +396,148 @@ const pgBoolCodec = codec<'pg/bool@1', boolean, boolean>({
   },
 });
 
+const pgBitCodec = codec<typeof PG_BIT_CODEC_ID, readonly ['equality', 'order'], string, string>({
+  typeId: PG_BIT_CODEC_ID,
+  targetTypes: ['bit'],
+  traits: ['equality', 'order'],
+  encode: (value: string): string => value,
+  decode: (wire: string): string => wire,
+  paramsSchema: lengthParamsSchema,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'bit',
+        },
+      },
+    },
+  },
+});
+
+const pgVarbitCodec = codec<
+  typeof PG_VARBIT_CODEC_ID,
+  readonly ['equality', 'order'],
+  string,
+  string
+>({
+  typeId: PG_VARBIT_CODEC_ID,
+  targetTypes: ['bit varying'],
+  traits: ['equality', 'order'],
+  encode: (value: string): string => value,
+  decode: (wire: string): string => wire,
+  paramsSchema: lengthParamsSchema,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'bit varying',
+        },
+      },
+    },
+  },
+});
+
+const pgEnumCodec = codec({
+  typeId: PG_ENUM_CODEC_ID,
+  targetTypes: ['enum'],
+  traits: ['equality', 'order'],
+  encode: (value: string): string => value,
+  decode: (wire: string): string => wire,
+});
+
+const pgIntervalCodec = codec<
+  typeof PG_INTERVAL_CODEC_ID,
+  readonly ['equality', 'order'],
+  string | Record<string, unknown>,
+  string
+>({
+  typeId: PG_INTERVAL_CODEC_ID,
+  targetTypes: ['interval'],
+  traits: ['equality', 'order'],
+  encode: (value: string): string => value,
+  decode: (wire: string | Record<string, unknown>): string => {
+    if (typeof wire === 'string') return wire;
+    return JSON.stringify(wire);
+  },
+  paramsSchema: precisionParamsSchema,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'interval',
+        },
+      },
+    },
+  },
+});
+
+const pgJsonCodec = codec({
+  typeId: PG_JSON_CODEC_ID,
+  targetTypes: ['json'],
+  traits: [],
+  encode: (value: string | JsonValue): string => JSON.stringify(value),
+  decode: (wire: string | JsonValue): JsonValue =>
+    typeof wire === 'string' ? JSON.parse(wire) : wire,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'json',
+        },
+      },
+    },
+  },
+});
+
+const pgJsonbCodec = codec({
+  typeId: PG_JSONB_CODEC_ID,
+  targetTypes: ['jsonb'],
+  traits: ['equality'],
+  encode: (value: string | JsonValue): string => JSON.stringify(value),
+  decode: (wire: string | JsonValue): JsonValue =>
+    typeof wire === 'string' ? JSON.parse(wire) : wire,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'jsonb',
+        },
+      },
+    },
+  },
+});
+
 // Build codec definitions using the builder DSL
 const codecs = defineCodecs()
+  .add('char', sqlCharCodec)
+  .add('varchar', sqlVarcharCodec)
+  .add('int', sqlIntCodec)
+  .add('float', sqlFloatCodec)
   .add('text', pgTextCodec)
+  .add('character', pgCharCodec)
+  .add('character varying', pgVarcharCodec)
+  .add('integer', pgIntCodec)
+  .add('double precision', pgFloatCodec)
   .add('int4', pgInt4Codec)
   .add('int2', pgInt2Codec)
   .add('int8', pgInt8Codec)
   .add('float4', pgFloat4Codec)
   .add('float8', pgFloat8Codec)
+  .add('numeric', pgNumericCodec)
   .add('timestamp', pgTimestampCodec)
   .add('timestamptz', pgTimestamptzCodec)
-  .add('bool', pgBoolCodec);
+  .add('time', pgTimeCodec)
+  .add('timetz', pgTimetzCodec)
+  .add('bool', pgBoolCodec)
+  .add('bit', pgBitCodec)
+  .add('bit varying', pgVarbitCodec)
+  .add('interval', pgIntervalCodec)
+  .add('enum', pgEnumCodec)
+  .add('json', pgJsonCodec)
+  .add('jsonb', pgJsonbCodec);
 
 // Export derived structures directly from codecs builder
 export const codecDefinitions = codecs.codecDefinitions;
 export const dataTypes = codecs.dataTypes;
 
-// Export types derived from codecs builder
 export type CodecTypes = typeof codecs.CodecTypes;