@prisma-next/sql-runtime 0.3.0-dev.4 → 0.3.0-dev.40

package/package.json

@@ -1,48 +1,57 @@
 {
   "name": "@prisma-next/sql-runtime",
-  "version": "0.3.0-dev.4",
+  "version": "0.3.0-dev.40",
   "type": "module",
   "sideEffects": false,
   "description": "SQL runtime implementation for Prisma Next",
   "dependencies": {
-    "@prisma-next/contract": "0.3.0-dev.4",
-    "@prisma-next/core-execution-plane": "0.3.0-dev.4",
-    "@prisma-next/operations": "0.3.0-dev.4",
-    "@prisma-next/runtime-executor": "0.3.0-dev.4",
-    "@prisma-next/sql-contract": "0.3.0-dev.4",
-    "@prisma-next/sql-operations": "0.3.0-dev.4",
-    "@prisma-next/sql-relational-core": "0.3.0-dev.4"
+    "arktype": "^2.1.26",
+    "@prisma-next/contract": "0.3.0-dev.40",
+    "@prisma-next/utils": "0.3.0-dev.40",
+    "@prisma-next/core-execution-plane": "0.3.0-dev.40",
+    "@prisma-next/ids": "0.3.0-dev.40",
+    "@prisma-next/operations": "0.3.0-dev.40",
+    "@prisma-next/runtime-executor": "0.3.0-dev.40",
+    "@prisma-next/sql-contract": "0.3.0-dev.40",
+    "@prisma-next/sql-operations": "0.3.0-dev.40",
+    "@prisma-next/sql-relational-core": "0.3.0-dev.40"
   },
   "devDependencies": {
     "@types/pg": "8.16.0",
-    "@vitest/coverage-v8": "4.0.16",
     "pg": "8.16.3",
-    "tsup": "8.5.1",
+    "tsdown": "0.18.4",
     "typescript": "5.9.3",
-    "vitest": "4.0.16",
-    "@prisma-next/test-utils": "0.0.1"
+    "vitest": "4.0.17",
+    "@prisma-next/test-utils": "0.0.1",
+    "@prisma-next/tsconfig": "0.0.0",
+    "@prisma-next/tsdown": "0.0.0"
   },
   "files": [
-    "dist"
+    "dist",
+    "src",
+    "test"
   ],
   "exports": {
-    ".": {
-      "types": "./dist/index.d.ts",
-      "import": "./dist/index.js"
-    },
-    "./test/utils": {
-      "types": "./dist/test/utils.d.ts",
-      "import": "./dist/test/utils.js"
-    }
+    ".": "./dist/index.mjs",
+    "./test/utils": "./dist/test/utils.mjs",
+    "./package.json": "./package.json"
+  },
+  "main": "./dist/index.mjs",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.mts",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/prisma/prisma-next.git",
+    "directory": "packages/2-sql/5-runtime"
   },
   "scripts": {
-    "build": "tsup --config tsup.config.ts",
+    "build": "tsdown",
     "test": "vitest run --passWithNoTests",
     "test:coverage": "vitest run --coverage --passWithNoTests",
     "typecheck": "tsc --project tsconfig.json --noEmit",
-    "lint": "biome check . --config-path ../../../biome.json --error-on-warnings",
-    "lint:fix": "biome check --write . --config-path ../../../biome.json",
-    "lint:fix:unsafe": "biome check --write --unsafe . --config-path ../../../biome.json",
-    "clean": "node ../../../scripts/clean.mjs"
+    "lint": "biome check . --error-on-warnings",
+    "lint:fix": "biome check --write .",
+    "lint:fix:unsafe": "biome check --write --unsafe .",
+    "clean": "rm -rf dist dist-tsc dist-tsc-prod coverage .tmp-output"
   }
 }

package/src/codecs/decoding.ts

@@ -0,0 +1,221 @@
+import type { ExecutionPlan } from '@prisma-next/contract/types';
+import type { Codec, CodecRegistry } from '@prisma-next/sql-relational-core/ast';
+import type { JsonSchemaValidatorRegistry } from '@prisma-next/sql-relational-core/query-lane-context';
+import { validateJsonValue } from './json-schema-validation';
+
+function resolveRowCodec(
+  alias: string,
+  plan: ExecutionPlan,
+  registry: CodecRegistry,
+): Codec | null {
+  const planCodecId = plan.meta.annotations?.codecs?.[alias] as string | undefined;
+  if (planCodecId) {
+    const codec = registry.get(planCodecId);
+    if (codec) {
+      return codec;
+    }
+  }
+
+  if (plan.meta.projectionTypes) {
+    const typeId = plan.meta.projectionTypes[alias];
+    if (typeId) {
+      const codec = registry.get(typeId);
+      if (codec) {
+        return codec;
+      }
+    }
+  }
+
+  return null;
+}
+
+type ColumnRefIndex = Map<string, { table: string; column: string }>;
+
+/**
+ * Builds a lookup index from column name → { table, column } ref.
+ * Called once per decodeRow invocation to avoid O(aliases × refs) linear scans.
+ */
+function buildColumnRefIndex(plan: ExecutionPlan): ColumnRefIndex | null {
+  const columns = plan.meta.refs?.columns;
+  if (!columns) return null;
+
+  const index: ColumnRefIndex = new Map();
+  for (const ref of columns) {
+    index.set(ref.column, ref);
+  }
+  return index;
+}
+
+function parseProjectionRef(value: string): { table: string; column: string } | null {
+  if (value.startsWith('include:') || value.startsWith('operation:')) {
+    return null;
+  }
+
+  const separatorIndex = value.indexOf('.');
+  if (separatorIndex <= 0 || separatorIndex === value.length - 1) {
+    return null;
+  }
+
+  return {
+    table: value.slice(0, separatorIndex),
+    column: value.slice(separatorIndex + 1),
+  };
+}
+
+function resolveColumnRefForAlias(
+  alias: string,
+  projection: ExecutionPlan['meta']['projection'],
+  fallbackColumnRefIndex: ColumnRefIndex | null,
+): { table: string; column: string } | undefined {
+  if (projection && !Array.isArray(projection)) {
+    const mappedRef = (projection as Record<string, string>)[alias];
+    if (typeof mappedRef !== 'string') {
+      return undefined;
+    }
+    return parseProjectionRef(mappedRef) ?? undefined;
+  }
+
+  return fallbackColumnRefIndex?.get(alias);
+}
+
+export function decodeRow(
+  row: Record<string, unknown>,
+  plan: ExecutionPlan,
+  registry: CodecRegistry,
+  jsonValidators?: JsonSchemaValidatorRegistry,
+): Record<string, unknown> {
+  const decoded: Record<string, unknown> = {};
+  const projection = plan.meta.projection;
+
+  // Fallback for plans that do not provide projection alias -> table.column mapping.
+  const fallbackColumnRefIndex =
+    jsonValidators && (!projection || Array.isArray(projection)) ? buildColumnRefIndex(plan) : null;
+
+  let aliases: readonly string[];
+  if (projection && !Array.isArray(projection)) {
+    aliases = Object.keys(projection);
+  } else if (projection && Array.isArray(projection)) {
+    aliases = projection;
+  } else {
+    aliases = Object.keys(row);
+  }
+
+  for (const alias of aliases) {
+    const wireValue = row[alias];
+
+    const projectionValue =
+      projection && typeof projection === 'object' && !Array.isArray(projection)
+        ? (projection as Record<string, string>)[alias]
+        : undefined;
+
+    if (typeof projectionValue === 'string' && projectionValue.startsWith('include:')) {
+      if (wireValue === null || wireValue === undefined) {
+        decoded[alias] = [];
+        continue;
+      }
+
+      try {
+        let parsed: unknown;
+        if (typeof wireValue === 'string') {
+          parsed = JSON.parse(wireValue);
+        } else if (Array.isArray(wireValue)) {
+          parsed = wireValue;
+        } else {
+          parsed = JSON.parse(String(wireValue));
+        }
+
+        if (!Array.isArray(parsed)) {
+          throw new Error(`Expected array for include alias '${alias}', got ${typeof parsed}`);
+        }
+
+        decoded[alias] = parsed;
+      } catch (error) {
+        const decodeError = new Error(
+          `Failed to parse JSON array for include alias '${alias}': ${error instanceof Error ? error.message : String(error)}`,
+        ) as Error & {
+          code: string;
+          category: string;
+          severity: string;
+          details?: Record<string, unknown>;
+        };
+        decodeError.code = 'RUNTIME.DECODE_FAILED';
+        decodeError.category = 'RUNTIME';
+        decodeError.severity = 'error';
+        decodeError.details = {
+          alias,
+          wirePreview:
+            typeof wireValue === 'string' && wireValue.length > 100
+              ? `${wireValue.substring(0, 100)}...`
+              : String(wireValue).substring(0, 100),
+        };
+        throw decodeError;
+      }
+      continue;
+    }
+
+    if (wireValue === null || wireValue === undefined) {
+      decoded[alias] = wireValue;
+      continue;
+    }
+
+    const codec = resolveRowCodec(alias, plan, registry);
+
+    if (!codec) {
+      decoded[alias] = wireValue;
+      continue;
+    }
+
+    try {
+      const decodedValue = codec.decode(wireValue);
+
+      // Validate decoded JSON value against schema
+      if (jsonValidators) {
+        const ref = resolveColumnRefForAlias(alias, projection, fallbackColumnRefIndex);
+        if (ref) {
+          validateJsonValue(
+            jsonValidators,
+            ref.table,
+            ref.column,
+            decodedValue,
+            'decode',
+            codec.id,
+          );
+        }
+      }
+
+      decoded[alias] = decodedValue;
+    } catch (error) {
+      // Re-throw JSON schema validation errors as-is
+      if (
+        error instanceof Error &&
+        'code' in error &&
+        (error as Error & { code: string }).code === 'RUNTIME.JSON_SCHEMA_VALIDATION_FAILED'
+      ) {
+        throw error;
+      }
+
+      const decodeError = new Error(
+        `Failed to decode row alias '${alias}' with codec '${codec.id}': ${error instanceof Error ? error.message : String(error)}`,
+      ) as Error & {
+        code: string;
+        category: string;
+        severity: string;
+        details?: Record<string, unknown>;
+      };
+      decodeError.code = 'RUNTIME.DECODE_FAILED';
+      decodeError.category = 'RUNTIME';
+      decodeError.severity = 'error';
+      decodeError.details = {
+        alias,
+        codec: codec.id,
+        wirePreview:
+          typeof wireValue === 'string' && wireValue.length > 100
+            ? `${wireValue.substring(0, 100)}...`
+            : String(wireValue).substring(0, 100),
+      };
+      throw decodeError;
+    }
+  }
+
+  return decoded;
+}

package/src/codecs/encoding.ts

@@ -0,0 +1,89 @@
+import type { ExecutionPlan, ParamDescriptor } from '@prisma-next/contract/types';
+import type { Codec, CodecRegistry } from '@prisma-next/sql-relational-core/ast';
+import type { JsonSchemaValidatorRegistry } from '@prisma-next/sql-relational-core/query-lane-context';
+import { validateJsonValue } from './json-schema-validation';
+
+function resolveParamCodec(
+  paramDescriptor: ParamDescriptor,
+  plan: ExecutionPlan,
+  registry: CodecRegistry,
+): Codec | null {
+  const paramName = paramDescriptor.name ?? `param_${paramDescriptor.index ?? 0}`;
+
+  const planCodecId = plan.meta.annotations?.codecs?.[paramName] as string | undefined;
+  if (planCodecId) {
+    const codec = registry.get(planCodecId);
+    if (codec) {
+      return codec;
+    }
+  }
+
+  if (paramDescriptor.codecId) {
+    const codec = registry.get(paramDescriptor.codecId);
+    if (codec) {
+      return codec;
+    }
+  }
+
+  return null;
+}
+
+export function encodeParam(
+  value: unknown,
+  paramDescriptor: ParamDescriptor,
+  plan: ExecutionPlan,
+  registry: CodecRegistry,
+  jsonValidators?: JsonSchemaValidatorRegistry,
+): unknown {
+  if (value === null || value === undefined) {
+    return null;
+  }
+
+  // Validate JSON value against schema before encoding
+  if (jsonValidators && paramDescriptor.refs) {
+    const { table, column } = paramDescriptor.refs;
+    validateJsonValue(jsonValidators, table, column, value, 'encode', paramDescriptor.codecId);
+  }
+
+  const codec = resolveParamCodec(paramDescriptor, plan, registry);
+  if (!codec) {
+    return value;
+  }
+
+  if (codec.encode) {
+    try {
+      return codec.encode(value);
+    } catch (error) {
+      throw new Error(
+        `Failed to encode parameter ${paramDescriptor.name ?? paramDescriptor.index}: ${error instanceof Error ? error.message : String(error)}`,
+      );
+    }
+  }
+
+  return value;
+}
+
+export function encodeParams(
+  plan: ExecutionPlan,
+  registry: CodecRegistry,
+  jsonValidators?: JsonSchemaValidatorRegistry,
+): readonly unknown[] {
+  if (plan.params.length === 0) {
+    return plan.params;
+  }
+
+  const encoded: unknown[] = [];
+
+  for (let i = 0; i < plan.params.length; i++) {
+    const paramValue = plan.params[i];
+    const paramDescriptor = plan.meta.paramDescriptors[i];
+
+    if (paramDescriptor) {
+      encoded.push(encodeParam(paramValue, paramDescriptor, plan, registry, jsonValidators));
+    } else {
+      encoded.push(paramValue);
+    }
+  }
+
+  return Object.freeze(encoded);
+}

package/src/codecs/json-schema-validation.ts

@@ -0,0 +1,61 @@
+import { runtimeError } from '@prisma-next/runtime-executor';
+import type {
+  JsonSchemaValidationError,
+  JsonSchemaValidatorRegistry,
+} from '@prisma-next/sql-relational-core/query-lane-context';
+
+/**
+ * Validates a JSON value against its column's JSON Schema, if a validator exists.
+ *
+ * Throws `RUNTIME.JSON_SCHEMA_VALIDATION_FAILED` on validation failure.
+ * No-ops if no validator is registered for the column.
+ */
+export function validateJsonValue(
+  registry: JsonSchemaValidatorRegistry,
+  table: string,
+  column: string,
+  value: unknown,
+  direction: 'encode' | 'decode',
+  codecId?: string,
+): void {
+  const key = `${table}.${column}`;
+  const validate = registry.get(key);
+  if (!validate) return;
+
+  const result = validate(value);
+  if (result.valid) return;
+
+  throw createJsonSchemaValidationError(table, column, direction, result.errors, codecId);
+}
+
+function createJsonSchemaValidationError(
+  table: string,
+  column: string,
+  direction: 'encode' | 'decode',
+  errors: ReadonlyArray<JsonSchemaValidationError>,
+  codecId?: string,
+): Error {
+  const summary = formatErrorSummary(errors);
+  return runtimeError(
+    'RUNTIME.JSON_SCHEMA_VALIDATION_FAILED',
+    `JSON schema validation failed for column '${table}.${column}' (${direction}): ${summary}`,
+    {
+      table,
+      column,
+      codecId,
+      direction,
+      errors: [...errors],
+    },
+  );
+}
+
+function formatErrorSummary(errors: ReadonlyArray<JsonSchemaValidationError>): string {
+  if (errors.length === 0) return 'unknown validation error';
+  if (errors.length === 1) {
+    const err = errors[0] as JsonSchemaValidationError;
+    return err.path === '/' ? err.message : `${err.path}: ${err.message}`;
+  }
+  return errors
+    .map((err) => (err.path === '/' ? err.message : `${err.path}: ${err.message}`))
+    .join('; ');
+}

package/src/codecs/validation.ts

@@ -0,0 +1,67 @@
+import { runtimeError } from '@prisma-next/runtime-executor';
+import type { SqlContract, SqlStorage } from '@prisma-next/sql-contract/types';
+import type { CodecRegistry } from '@prisma-next/sql-relational-core/ast';
+
+export function extractCodecIds(contract: SqlContract<SqlStorage>): Set<string> {
+  const codecIds = new Set<string>();
+
+  for (const table of Object.values(contract.storage.tables)) {
+    for (const column of Object.values(table.columns)) {
+      const codecId = column.codecId;
+      codecIds.add(codecId);
+    }
+  }
+
+  return codecIds;
+}
+
+function extractCodecIdsFromColumns(contract: SqlContract<SqlStorage>): Map<string, string> {
+  const codecIds = new Map<string, string>();
+
+  for (const [tableName, table] of Object.entries(contract.storage.tables)) {
+    for (const [columnName, column] of Object.entries(table.columns)) {
+      const codecId = column.codecId;
+      const key = `${tableName}.${columnName}`;
+      codecIds.set(key, codecId);
+    }
+  }
+
+  return codecIds;
+}
+
+export function validateContractCodecMappings(
+  registry: CodecRegistry,
+  contract: SqlContract<SqlStorage>,
+): void {
+  const codecIds = extractCodecIdsFromColumns(contract);
+  const invalidCodecs: Array<{ table: string; column: string; codecId: string }> = [];
+
+  for (const [key, codecId] of codecIds.entries()) {
+    if (!registry.has(codecId)) {
+      const parts = key.split('.');
+      const table = parts[0] ?? '';
+      const column = parts[1] ?? '';
+      invalidCodecs.push({ table, column, codecId });
+    }
+  }
+
+  if (invalidCodecs.length > 0) {
+    const details: Record<string, unknown> = {
+      contractTarget: contract.target,
+      invalidCodecs,
+    };
+
+    throw runtimeError(
+      'RUNTIME.CODEC_MISSING',
+      `Missing codec implementations for column codecIds: ${invalidCodecs.map((c) => `${c.table}.${c.column} (${c.codecId})`).join(', ')}`,
+      details,
+    );
+  }
+}
+
+export function validateCodecRegistryCompleteness(
+  registry: CodecRegistry,
+  contract: SqlContract<SqlStorage>,
+): void {
+  validateContractCodecMappings(registry, contract);
+}

package/src/exports/index.ts

@@ -0,0 +1,45 @@
+export type {
+  AfterExecuteResult,
+  BudgetsOptions,
+  LintsOptions,
+  Log,
+  Plugin,
+  PluginContext,
+} from '@prisma-next/runtime-executor';
+export { budgets, lints } from '@prisma-next/runtime-executor';
+export {
+  extractCodecIds,
+  validateCodecRegistryCompleteness,
+  validateContractCodecMappings,
+} from '../codecs/validation';
+export { lowerSqlPlan } from '../lower-sql-plan';
+export type {
+  ExecutionContext,
+  RuntimeParameterizedCodecDescriptor,
+  SqlExecutionStack,
+  SqlExecutionStackWithDriver,
+  SqlRuntimeAdapterDescriptor,
+  SqlRuntimeAdapterInstance,
+  SqlRuntimeDriverInstance,
+  SqlRuntimeExtensionDescriptor,
+  SqlRuntimeExtensionInstance,
+  SqlRuntimeTargetDescriptor,
+  SqlStaticContributions,
+  TypeHelperRegistry,
+} from '../sql-context';
+export { createExecutionContext, createSqlExecutionStack } from '../sql-context';
+export type { SqlStatement } from '../sql-marker';
+export {
+  ensureSchemaStatement,
+  ensureTableStatement,
+  readContractMarker,
+  writeContractMarker,
+} from '../sql-marker';
+export type {
+  CreateRuntimeOptions,
+  Runtime,
+  RuntimeTelemetryEvent,
+  RuntimeVerifyOptions,
+  TelemetryOutcome,
+} from '../sql-runtime';
+export { createRuntime } from '../sql-runtime';

package/src/lower-sql-plan.ts

@@ -0,0 +1,32 @@
+import type { ExecutionPlan } from '@prisma-next/contract/types';
+import type { SqlContract, SqlStorage } from '@prisma-next/sql-contract/types';
+import type { Adapter, LoweredStatement, QueryAst } from '@prisma-next/sql-relational-core/ast';
+import type { SqlQueryPlan } from '@prisma-next/sql-relational-core/plan';
+
+/**
+ * Lowers a SQL query plan to an executable Plan by calling the adapter's lower method.
+ *
+ * @param adapter - Adapter to lower AST to SQL
+ * @param contract - Contract for lowering context
+ * @param queryPlan - SQL query plan from a lane (contains AST, params, meta, but no SQL)
+ * @returns Fully executable Plan with SQL string
+ */
+export function lowerSqlPlan<Row>(
+  adapter: Adapter<QueryAst, SqlContract<SqlStorage>, LoweredStatement>,
+  contract: SqlContract<SqlStorage>,
+  queryPlan: SqlQueryPlan<Row>,
+): ExecutionPlan<Row> {
+  const lowered = adapter.lower(queryPlan.ast, {
+    contract,
+    params: queryPlan.params,
+  });
+
+  const body = lowered.body;
+
+  return Object.freeze({
+    sql: body.sql,
+    params: body.params ?? queryPlan.params,
+    ast: queryPlan.ast,
+    meta: queryPlan.meta,
+  });
+}