@prisma-next/adapter-postgres 0.3.0-dev.33 → 0.3.0-dev.36

package/src/core/codecs.ts

@@ -11,11 +11,89 @@
  * at compile time (to derive CodecTypes).
  */
 
-import { codec, defineCodecs } from '@prisma-next/sql-relational-core/ast';
+import type { Codec, CodecMeta } 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,
+  TWire,
+  TJs,
+  TParams = Record<string, unknown>,
+  THelper = unknown,
+>(
+  base: Codec<string, TWire, TJs, TParams, THelper>,
+  options: {
+    readonly typeId: Id;
+    readonly targetTypes: readonly string[];
+    readonly meta?: CodecMeta;
+  },
+): Codec<Id, 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),
+    decode: base.decode,
+  };
+}
+
+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'],
   encode: (value: string): string => value,
   decode: (wire: string): string => wire,
@@ -30,8 +108,64 @@ 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'],
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'integer',
+        },
+      },
+    },
+  },
+});
+
+const pgFloatCodec = aliasCodec(sqlFloatCodec, {
+  typeId: PG_FLOAT_CODEC_ID,
+  targetTypes: ['float8'],
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'double precision',
+        },
+      },
+    },
+  },
+});
+
+const pgInt4Codec = codec<typeof PG_INT4_CODEC_ID, number, number>({
+  typeId: PG_INT4_CODEC_ID,
   targetTypes: ['int4'],
   encode: (value) => value,
   decode: (wire) => wire,
@@ -46,8 +180,28 @@ const pgInt4Codec = codec<'pg/int4@1', number, number>({
   },
 });
 
-const pgInt2Codec = codec<'pg/int2@1', number, number>({
-  typeId: 'pg/int2@1',
+const pgNumericCodec = codec<typeof PG_NUMERIC_CODEC_ID, string, string>({
+  typeId: PG_NUMERIC_CODEC_ID,
+  targetTypes: ['numeric', 'decimal'],
+  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<typeof PG_INT2_CODEC_ID, number, number>({
+  typeId: PG_INT2_CODEC_ID,
   targetTypes: ['int2'],
   encode: (value) => value,
   decode: (wire) => wire,
@@ -62,8 +216,8 @@ const pgInt2Codec = codec<'pg/int2@1', number, number>({
   },
 });
 
-const pgInt8Codec = codec<'pg/int8@1', number, number>({
-  typeId: 'pg/int8@1',
+const pgInt8Codec = codec<typeof PG_INT8_CODEC_ID, number, number>({
+  typeId: PG_INT8_CODEC_ID,
   targetTypes: ['int8'],
   encode: (value) => value,
   decode: (wire) => wire,
@@ -78,8 +232,8 @@ const pgInt8Codec = codec<'pg/int8@1', number, number>({
   },
 });
 
-const pgFloat4Codec = codec<'pg/float4@1', number, number>({
-  typeId: 'pg/float4@1',
+const pgFloat4Codec = codec<typeof PG_FLOAT4_CODEC_ID, number, number>({
+  typeId: PG_FLOAT4_CODEC_ID,
   targetTypes: ['float4'],
   encode: (value) => value,
   decode: (wire) => wire,
@@ -94,8 +248,8 @@ const pgFloat4Codec = codec<'pg/float4@1', number, number>({
   },
 });
 
-const pgFloat8Codec = codec<'pg/float8@1', number, number>({
-  typeId: 'pg/float8@1',
+const pgFloat8Codec = codec<typeof PG_FLOAT8_CODEC_ID, number, number>({
+  typeId: PG_FLOAT8_CODEC_ID,
   targetTypes: ['float8'],
   encode: (value) => value,
   decode: (wire) => wire,
@@ -110,8 +264,8 @@ 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, string | Date, string>({
+  typeId: PG_TIMESTAMP_CODEC_ID,
   targetTypes: ['timestamp'],
   encode: (value: string | Date): string => {
     if (value instanceof Date) return value.toISOString();
@@ -123,6 +277,7 @@ const pgTimestampCodec = codec<'pg/timestamp@1', string | Date, string>({
     if (wire instanceof Date) return wire.toISOString();
     return String(wire);
   },
+  paramsSchema: precisionParamsSchema,
   meta: {
     db: {
       sql: {
@@ -134,8 +289,8 @@ 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, string | Date, string>({
+  typeId: PG_TIMESTAMPTZ_CODEC_ID,
   targetTypes: ['timestamptz'],
   encode: (value: string | Date): string => {
     if (value instanceof Date) return value.toISOString();
@@ -147,6 +302,7 @@ const pgTimestamptzCodec = codec<'pg/timestamptz@1', string | Date, string>({
     if (wire instanceof Date) return wire.toISOString();
     return String(wire);
   },
+  paramsSchema: precisionParamsSchema,
   meta: {
     db: {
       sql: {
@@ -158,8 +314,42 @@ 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, string, string>({
+  typeId: PG_TIME_CODEC_ID,
+  targetTypes: ['time'],
+  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, string, string>({
+  typeId: PG_TIMETZ_CODEC_ID,
+  targetTypes: ['timetz'],
+  encode: (value: string): string => value,
+  decode: (wire: string): string => wire,
+  paramsSchema: precisionParamsSchema,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'timetz',
+        },
+      },
+    },
+  },
+});
+
+const pgBoolCodec = codec<typeof PG_BOOL_CODEC_ID, boolean, boolean>({
+  typeId: PG_BOOL_CODEC_ID,
   targetTypes: ['bool'],
   encode: (value) => value,
   decode: (wire) => wire,
@@ -174,17 +364,124 @@ const pgBoolCodec = codec<'pg/bool@1', boolean, boolean>({
   },
 });
 
+const pgBitCodec = codec<typeof PG_BIT_CODEC_ID, string, string>({
+  typeId: PG_BIT_CODEC_ID,
+  targetTypes: ['bit'],
+  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, string, string>({
+  typeId: PG_VARBIT_CODEC_ID,
+  targetTypes: ['bit varying'],
+  encode: (value: string): string => value,
+  decode: (wire: string): string => wire,
+  paramsSchema: lengthParamsSchema,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'bit varying',
+        },
+      },
+    },
+  },
+});
+
+const pgEnumCodec = codec<typeof PG_ENUM_CODEC_ID, string, string>({
+  typeId: PG_ENUM_CODEC_ID,
+  targetTypes: ['enum'],
+  encode: (value) => value,
+  decode: (wire) => wire,
+});
+
+const pgIntervalCodec = codec<typeof PG_INTERVAL_CODEC_ID, string, string>({
+  typeId: PG_INTERVAL_CODEC_ID,
+  targetTypes: ['interval'],
+  encode: (value: string): string => value,
+  decode: (wire: string): string => wire,
+  paramsSchema: precisionParamsSchema,
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'interval',
+        },
+      },
+    },
+  },
+});
+
+const pgJsonCodec = codec<typeof PG_JSON_CODEC_ID, string | JsonValue, JsonValue>({
+  typeId: PG_JSON_CODEC_ID,
+  targetTypes: ['json'],
+  encode: (value) => JSON.stringify(value),
+  decode: (wire) => (typeof wire === 'string' ? JSON.parse(wire) : wire),
+  meta: {
+    db: {
+      sql: {
+        postgres: {
+          nativeType: 'json',
+        },
+      },
+    },
+  },
+});
+
+const pgJsonbCodec = codec<typeof PG_JSONB_CODEC_ID, string | JsonValue, JsonValue>({
+  typeId: PG_JSONB_CODEC_ID,
+  targetTypes: ['jsonb'],
+  encode: (value) => JSON.stringify(value),
+  decode: (wire) => (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;