drizzle-orm 1.0.0-beta.8-dbc3565 → 1.0.0-beta.8-e601f80

package/pg-core/effect/session.d.cts

@@ -4,12 +4,14 @@ import type * as V1 from "../../_relations.cjs";
 import type { EffectCache } from "../../cache/core/cache-effect.cjs";
 import type { WithCacheConfig } from "../../cache/core/types.cjs";
 import { entityKind } from "../../entity.cjs";
-import { DrizzleQueryError } from "../../errors.cjs";
+import { DrizzleQueryError, TransactionRollbackError } from "../../errors.cjs";
+import type { MigrationConfig, MigrationMeta, MigratorInitFailResponse } from "../../migrator.cjs";
 import type { AnyRelations, EmptyRelations } from "../../relations.cjs";
-import type { Query } from "../../sql/index.cjs";
+import { type Query, type SQL } from "../../sql/sql.cjs";
 import type { PgDialect } from "../dialect.cjs";
 import type { SelectedFieldsOrdered } from "../query-builders/select.types.cjs";
-import { PgBasePreparedQuery, PgSession, type PreparedQueryConfig } from "../session.cjs";
+import { PgBasePreparedQuery, type PgQueryResultHKT, PgSession, type PgTransactionConfig, type PreparedQueryConfig } from "../session.cjs";
+import { PgEffectDatabase } from "./db.cjs";
 export declare abstract class PgEffectPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {
     private cache;
     private queryMetadata;
@@ -22,7 +24,7 @@ export declare abstract class PgEffectPreparedQuery<T extends PreparedQueryConfi
     protected queryWithCache<T>(queryString: string, params: any[], query: Effect.Effect<T, SqlError>): Effect.Effect<T, DrizzleQueryError>;
     abstract execute(placeholderValues?: Record<string, unknown>): Effect.Effect<T['execute'], DrizzleQueryError>;
 }
-export declare abstract class PgEffectSession<TFullSchema extends Record<string, unknown> = Record<string, never>, _TRelations extends AnyRelations = EmptyRelations, _TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgSession {
+export declare abstract class PgEffectSession<TQueryResult extends PgQueryResultHKT = PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TRelations extends AnyRelations = EmptyRelations, TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgSession {
     static readonly [entityKind]: string;
     constructor(dialect: PgDialect);
     abstract prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, isResponseInArrayMode: boolean, customResultMapper?: (rows: unknown[][], mapColumnValue?: (value: unknown) => unknown) => T['execute'], queryMetadata?: {
@@ -30,4 +32,26 @@ export declare abstract class PgEffectSession<TFullSchema extends Record<string,
         tables: string[];
     }, cacheConfig?: WithCacheConfig): PgEffectPreparedQuery<T>;
     abstract prepareRelationalQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, customResultMapper: (rows: Record<string, unknown>[], mapColumnValue?: (value: unknown) => unknown) => T['execute']): PgEffectPreparedQuery<T>;
+    execute<T>(query: SQL): Effect.Effect<T, DrizzleQueryError>;
+    all<T>(query: SQL): Effect.Effect<T[], DrizzleQueryError>;
+    abstract transaction<T>(transaction: (tx: PgEffectTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>): Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>;
 }
+export declare abstract class PgEffectTransaction<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TRelations extends AnyRelations = EmptyRelations, TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgEffectDatabase<TQueryResult, TFullSchema, TRelations, TSchema> {
+    protected relations: TRelations;
+    protected schema: {
+        fullSchema: Record<string, unknown>;
+        schema: TSchema;
+        tableNamesMap: Record<string, string>;
+    } | undefined;
+    protected readonly nestedIndex: number;
+    static readonly [entityKind]: string;
+    constructor(dialect: PgDialect, session: PgEffectSession<any, any, any, any>, relations: TRelations, schema: {
+        fullSchema: Record<string, unknown>;
+        schema: TSchema;
+        tableNamesMap: Record<string, string>;
+    } | undefined, nestedIndex?: number, parseRqbJson?: boolean);
+    rollback: () => Generator<import("effect/Utils").YieldWrap<Effect.Effect<never, TransactionRollbackError, never>>, void, any>;
+    setTransaction(config: PgTransactionConfig): Effect.Effect<void, DrizzleQueryError>;
+    abstract transaction<T>(transaction: (tx: PgEffectTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>): Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>;
+}
+export declare function migrate(migrations: MigrationMeta[], session: PgEffectSession, config: string | MigrationConfig): Effect.Effect<void | MigratorInitFailResponse, DrizzleQueryError, never>;

package/pg-core/effect/session.d.ts

@@ -4,12 +4,14 @@ import type * as V1 from "../../_relations.js";
 import type { EffectCache } from "../../cache/core/cache-effect.js";
 import type { WithCacheConfig } from "../../cache/core/types.js";
 import { entityKind } from "../../entity.js";
-import { DrizzleQueryError } from "../../errors.js";
+import { DrizzleQueryError, TransactionRollbackError } from "../../errors.js";
+import type { MigrationConfig, MigrationMeta, MigratorInitFailResponse } from "../../migrator.js";
 import type { AnyRelations, EmptyRelations } from "../../relations.js";
-import type { Query } from "../../sql/index.js";
+import { type Query, type SQL } from "../../sql/sql.js";
 import type { PgDialect } from "../dialect.js";
 import type { SelectedFieldsOrdered } from "../query-builders/select.types.js";
-import { PgBasePreparedQuery, PgSession, type PreparedQueryConfig } from "../session.js";
+import { PgBasePreparedQuery, type PgQueryResultHKT, PgSession, type PgTransactionConfig, type PreparedQueryConfig } from "../session.js";
+import { PgEffectDatabase } from "./db.js";
 export declare abstract class PgEffectPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {
     private cache;
     private queryMetadata;
@@ -22,7 +24,7 @@ export declare abstract class PgEffectPreparedQuery<T extends PreparedQueryConfi
     protected queryWithCache<T>(queryString: string, params: any[], query: Effect.Effect<T, SqlError>): Effect.Effect<T, DrizzleQueryError>;
     abstract execute(placeholderValues?: Record<string, unknown>): Effect.Effect<T['execute'], DrizzleQueryError>;
 }
-export declare abstract class PgEffectSession<TFullSchema extends Record<string, unknown> = Record<string, never>, _TRelations extends AnyRelations = EmptyRelations, _TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgSession {
+export declare abstract class PgEffectSession<TQueryResult extends PgQueryResultHKT = PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TRelations extends AnyRelations = EmptyRelations, TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgSession {
     static readonly [entityKind]: string;
     constructor(dialect: PgDialect);
     abstract prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, isResponseInArrayMode: boolean, customResultMapper?: (rows: unknown[][], mapColumnValue?: (value: unknown) => unknown) => T['execute'], queryMetadata?: {
@@ -30,4 +32,26 @@ export declare abstract class PgEffectSession<TFullSchema extends Record<string,
         tables: string[];
     }, cacheConfig?: WithCacheConfig): PgEffectPreparedQuery<T>;
     abstract prepareRelationalQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, customResultMapper: (rows: Record<string, unknown>[], mapColumnValue?: (value: unknown) => unknown) => T['execute']): PgEffectPreparedQuery<T>;
+    execute<T>(query: SQL): Effect.Effect<T, DrizzleQueryError>;
+    all<T>(query: SQL): Effect.Effect<T[], DrizzleQueryError>;
+    abstract transaction<T>(transaction: (tx: PgEffectTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>): Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>;
 }
+export declare abstract class PgEffectTransaction<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TRelations extends AnyRelations = EmptyRelations, TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgEffectDatabase<TQueryResult, TFullSchema, TRelations, TSchema> {
+    protected relations: TRelations;
+    protected schema: {
+        fullSchema: Record<string, unknown>;
+        schema: TSchema;
+        tableNamesMap: Record<string, string>;
+    } | undefined;
+    protected readonly nestedIndex: number;
+    static readonly [entityKind]: string;
+    constructor(dialect: PgDialect, session: PgEffectSession<any, any, any, any>, relations: TRelations, schema: {
+        fullSchema: Record<string, unknown>;
+        schema: TSchema;
+        tableNamesMap: Record<string, string>;
+    } | undefined, nestedIndex?: number, parseRqbJson?: boolean);
+    rollback: () => Generator<import("effect/Utils").YieldWrap<Effect.Effect<never, TransactionRollbackError, never>>, void, any>;
+    setTransaction(config: PgTransactionConfig): Effect.Effect<void, DrizzleQueryError>;
+    abstract transaction<T>(transaction: (tx: PgEffectTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>): Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>;
+}
+export declare function migrate(migrations: MigrationMeta[], session: PgEffectSession, config: string | MigrationConfig): Effect.Effect<void | MigratorInitFailResponse, DrizzleQueryError, never>;

package/pg-core/effect/session.js

@@ -1,9 +1,14 @@
 import { Effect } from "effect";
 import { strategyFor } from "../../cache/core/cache.js";
 import { entityKind } from "../../entity.js";
-import { DrizzleQueryError } from "../../errors.js";
+import { DrizzleQueryError, TransactionRollbackError } from "../../errors.js";
+import { sql } from "../../sql/sql.js";
 import { assertUnreachable } from "../../utils.js";
-import { PgBasePreparedQuery, PgSession } from "../session.js";
+import {
+  PgBasePreparedQuery,
+  PgSession
+} from "../session.js";
+import { PgEffectDatabase } from "./db.js";
 class PgEffectPreparedQuery extends PgBasePreparedQuery {
   constructor(query, cache, queryMetadata, cacheConfig) {
     super(query);
@@ -62,9 +67,96 @@ class PgEffectSession extends PgSession {
   constructor(dialect) {
     super(dialect);
   }
+  execute(query) {
+    const { sql: sql2, params } = this.dialect.sqlToQuery(query);
+    return this.prepareQuery({ sql: sql2, params }, void 0, void 0, false).execute();
+  }
+  all(query) {
+    const { sql: sql2, params } = this.dialect.sqlToQuery(query);
+    return this.prepareQuery({ sql: sql2, params }, void 0, void 0, false).all();
+  }
+}
+class PgEffectTransaction extends PgEffectDatabase {
+  constructor(dialect, session, relations, schema, nestedIndex = 0, parseRqbJson) {
+    super(dialect, session, relations, schema, parseRqbJson);
+    this.relations = relations;
+    this.schema = schema;
+    this.nestedIndex = nestedIndex;
+  }
+  static [entityKind] = "PgEffectTransaction";
+  rollback = function* () {
+    yield* Effect.fail(new TransactionRollbackError());
+  };
+  /** @internal */
+  getTransactionConfigSQL(config) {
+    const chunks = [];
+    if (config.isolationLevel) {
+      chunks.push(`isolation level ${config.isolationLevel}`);
+    }
+    if (config.accessMode) {
+      chunks.push(config.accessMode);
+    }
+    if (typeof config.deferrable === "boolean") {
+      chunks.push(config.deferrable ? "deferrable" : "not deferrable");
+    }
+    return sql.raw(chunks.join(" "));
+  }
+  setTransaction(config) {
+    return this.session.execute(sql`set transaction ${this.getTransactionConfigSQL(config)}`);
+  }
+}
+function migrate(migrations, session, config) {
+  return Effect.gen(function* () {
+    const migrationsTable = typeof config === "string" ? "__drizzle_migrations" : config.migrationsTable ?? "__drizzle_migrations";
+    const migrationsSchema = typeof config === "string" ? "drizzle" : config.migrationsSchema ?? "drizzle";
+    const migrationTableCreate = sql`
+			CREATE TABLE IF NOT EXISTS ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)} (
+				id SERIAL PRIMARY KEY,
+				hash text NOT NULL,
+				created_at bigint
+			)
+		`;
+    yield* session.execute(sql`CREATE SCHEMA IF NOT EXISTS ${sql.identifier(migrationsSchema)}`);
+    yield* session.execute(migrationTableCreate);
+    const dbMigrations = yield* session.all(
+      sql`select id, hash, created_at from ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)} order by created_at desc limit 1`
+    );
+    if (typeof config === "object" && config.init) {
+      if (dbMigrations.length) {
+        return { exitCode: "databaseMigrations" };
+      }
+      if (migrations.length > 1) {
+        return { exitCode: "localMigrations" };
+      }
+      const [migration] = migrations;
+      if (!migration) return;
+      yield* session.execute(
+        sql`insert into ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)} ("hash", "created_at") values(${migration.hash}, ${migration.folderMillis})`
+      );
+      return;
+    }
+    const lastDbMigration = dbMigrations[0];
+    yield* session.transaction(
+      (tx) => Effect.gen(function* () {
+        for (const migration of migrations) {
+          if (!lastDbMigration || Number(lastDbMigration.created_at) < migration.folderMillis) {
+            for (const stmt of migration.sql) {
+              yield* tx.execute(sql.raw(stmt));
+            }
+            yield* tx.execute(
+              sql`insert into ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)} ("hash", "created_at") values(${migration.hash}, ${migration.folderMillis})`
+            );
+          }
+        }
+      })
+    );
+    return;
+  });
 }
 export {
   PgEffectPreparedQuery,
-  PgEffectSession
+  PgEffectSession,
+  PgEffectTransaction,
+  migrate
 };
 //# sourceMappingURL=session.js.map

package/pg-core/effect/session.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/pg-core/effect/session.ts"],"sourcesContent":["import type { SqlError } from '@effect/sql/SqlError';\nimport { Effect } from 'effect';\nimport type * as V1 from '~/_relations.ts';\nimport type { EffectCache } from '~/cache/core/cache-effect.ts';\nimport { strategyFor } from '~/cache/core/cache.ts';\nimport type { WithCacheConfig } from '~/cache/core/types.ts';\nimport { entityKind } from '~/entity.ts';\nimport { DrizzleQueryError } from '~/errors.ts';\nimport type { AnyRelations, EmptyRelations } from '~/relations.ts';\nimport type { Query } from '~/sql/index.ts';\nimport { assertUnreachable } from '~/utils.ts';\nimport type { PgDialect } from '../dialect.ts';\nimport type { SelectedFieldsOrdered } from '../query-builders/select.types.ts';\nimport { PgBasePreparedQuery, PgSession, type PreparedQueryConfig } from '../session.ts';\n\nexport abstract class PgEffectPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {\n\tstatic override readonly [entityKind]: string = 'PgEffectPreparedQuery';\n\n\tconstructor(\n\t\tquery: Query,\n\t\tprivate cache: EffectCache | undefined,\n\t\tprivate queryMetadata: {\n\t\t\ttype: 'select' | 'update' | 'delete' | 'insert';\n\t\t\ttables: string[];\n\t\t} | undefined,\n\t\tprivate cacheConfig?: WithCacheConfig,\n\t) {\n\t\tsuper(query);\n\n\t\tif (cache && cache.strategy() === 'all' && cacheConfig === undefined) {\n\t\t\tthis.cacheConfig = { enabled: true, autoInvalidate: true };\n\t\t}\n\t\tif (!this.cacheConfig?.enabled) {\n\t\t\tthis.cacheConfig = undefined;\n\t\t}\n\t}\n\n\tprotected override queryWithCache<T>(\n\t\tqueryString: string,\n\t\tparams: any[],\n\t\tquery: Effect.Effect<T, SqlError>,\n\t): Effect.Effect<T, DrizzleQueryError> {\n\t\tconst { cache, cacheConfig, queryMetadata } = this;\n\t\treturn Effect.gen(function*() {\n\t\t\tconst cacheStrat: Awaited<ReturnType<typeof strategyFor>> = cache\n\t\t\t\t? yield* Effect.tryPromise(\n\t\t\t\t\t() => strategyFor(queryString, params, queryMetadata, cacheConfig),\n\t\t\t\t)\n\t\t\t\t: { type: 'skip' as const };\n\n\t\t\tif (cacheStrat.type === 'skip') {\n\t\t\t\treturn yield* query;\n\t\t\t}\n\n\t\t\t// For mutate queries, we should query the database, wait for a response, and then perform invalidation\n\t\t\tif (cacheStrat.type === 'invalidate') {\n\t\t\t\tconst result = yield* query;\n\t\t\t\tyield* cache!.onMutate({ tables: cacheStrat.tables });\n\n\t\t\t\treturn result;\n\t\t\t}\n\n\t\t\tif (cacheStrat.type === 'try') {\n\t\t\t\tconst { tables, key, isTag, autoInvalidate, config } = cacheStrat;\n\t\t\t\tconst fromCache: any[] | undefined = yield* cache!.get(\n\t\t\t\t\tkey,\n\t\t\t\t\ttables,\n\t\t\t\t\tisTag,\n\t\t\t\t\tautoInvalidate,\n\t\t\t\t);\n\n\t\t\t\tif (typeof fromCache !== 'undefined') return fromCache as unknown as T;\n\n\t\t\t\tconst result = yield* query;\n\n\t\t\t\tyield* cache!.put(\n\t\t\t\t\tkey,\n\t\t\t\t\tresult,\n\t\t\t\t\tautoInvalidate ? tables : [],\n\t\t\t\t\tisTag,\n\t\t\t\t\tconfig,\n\t\t\t\t);\n\n\t\t\t\treturn result;\n\t\t\t}\n\n\t\t\tassertUnreachable(cacheStrat);\n\t\t}).pipe(Effect.catchAll((e) => {\n\t\t\t// eslint-disable-next-line @drizzle-internal/no-instanceof\n\t\t\treturn Effect.fail(new DrizzleQueryError(queryString, params, e instanceof Error ? e : undefined));\n\t\t}));\n\t}\n\n\tabstract override execute(\n\t\tplaceholderValues?: Record<string, unknown>,\n\t): Effect.Effect<T['execute'], DrizzleQueryError>;\n\n\t/** @internal */\n\tabstract override all(placeholderValues?: Record<string, unknown>): Effect.Effect<T['all'], DrizzleQueryError>;\n}\n\nexport abstract class PgEffectSession<\n\tTFullSchema extends Record<string, unknown> = Record<string, never>,\n\t_TRelations extends AnyRelations = EmptyRelations,\n\t_TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>,\n> extends PgSession {\n\tstatic override readonly [entityKind]: string = 'PgEffectSession';\n\n\tconstructor(dialect: PgDialect) {\n\t\tsuper(dialect);\n\t}\n\n\tabstract override prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tisResponseInArrayMode: boolean,\n\t\tcustomResultMapper?: (rows: unknown[][], mapColumnValue?: (value: unknown) => unknown) => T['execute'],\n\t\tqueryMetadata?: {\n\t\t\ttype: 'select' | 'update' | 'delete' | 'insert';\n\t\t\ttables: string[];\n\t\t},\n\t\tcacheConfig?: WithCacheConfig,\n\t): PgEffectPreparedQuery<T>;\n\n\tabstract override prepareRelationalQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tcustomResultMapper: (\n\t\t\trows: Record<string, unknown>[],\n\t\t\tmapColumnValue?: (value: unknown) => unknown,\n\t\t) => T['execute'],\n\t): PgEffectPreparedQuery<T>;\n}\n"],"mappings":"AACA,SAAS,cAAc;AAGvB,SAAS,mBAAmB;AAE5B,SAAS,kBAAkB;AAC3B,SAAS,yBAAyB;AAGlC,SAAS,yBAAyB;AAGlC,SAAS,qBAAqB,iBAA2C;AAElE,MAAe,8BAA6D,oBAAoB;AAAA,EAGtG,YACC,OACQ,OACA,eAIA,aACP;AACD,UAAM,KAAK;AAPH;AACA;AAIA;AAIR,QAAI,SAAS,MAAM,SAAS,MAAM,SAAS,gBAAgB,QAAW;AACrE,WAAK,cAAc,EAAE,SAAS,MAAM,gBAAgB,KAAK;AAAA,IAC1D;AACA,QAAI,CAAC,KAAK,aAAa,SAAS;AAC/B,WAAK,cAAc;AAAA,IACpB;AAAA,EACD;AAAA,EAnBA,QAA0B,UAAU,IAAY;AAAA,EAqB7B,eAClB,aACA,QACA,OACsC;AACtC,UAAM,EAAE,OAAO,aAAa,cAAc,IAAI;AAC9C,WAAO,OAAO,IAAI,aAAY;AAC7B,YAAM,aAAsD,QACzD,OAAO,OAAO;AAAA,QACf,MAAM,YAAY,aAAa,QAAQ,eAAe,WAAW;AAAA,MAClE,IACE,EAAE,MAAM,OAAgB;AAE3B,UAAI,WAAW,SAAS,QAAQ;AAC/B,eAAO,OAAO;AAAA,MACf;AAGA,UAAI,WAAW,SAAS,cAAc;AACrC,cAAM,SAAS,OAAO;AACtB,eAAO,MAAO,SAAS,EAAE,QAAQ,WAAW,OAAO,CAAC;AAEpD,eAAO;AAAA,MACR;AAEA,UAAI,WAAW,SAAS,OAAO;AAC9B,cAAM,EAAE,QAAQ,KAAK,OAAO,gBAAgB,OAAO,IAAI;AACvD,cAAM,YAA+B,OAAO,MAAO;AAAA,UAClD;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACD;AAEA,YAAI,OAAO,cAAc,YAAa,QAAO;AAE7C,cAAM,SAAS,OAAO;AAEtB,eAAO,MAAO;AAAA,UACb;AAAA,UACA;AAAA,UACA,iBAAiB,SAAS,CAAC;AAAA,UAC3B;AAAA,UACA;AAAA,QACD;AAEA,eAAO;AAAA,MACR;AAEA,wBAAkB,UAAU;AAAA,IAC7B,CAAC,EAAE,KAAK,OAAO,SAAS,CAAC,MAAM;AAE9B,aAAO,OAAO,KAAK,IAAI,kBAAkB,aAAa,QAAQ,aAAa,QAAQ,IAAI,MAAS,CAAC;AAAA,IAClG,CAAC,CAAC;AAAA,EACH;AAQD;AAEO,MAAe,wBAIZ,UAAU;AAAA,EACnB,QAA0B,UAAU,IAAY;AAAA,EAEhD,YAAY,SAAoB;AAC/B,UAAM,OAAO;AAAA,EACd;AAwBD;","names":[]}
+{"version":3,"sources":["../../../src/pg-core/effect/session.ts"],"sourcesContent":["import type { SqlError } from '@effect/sql/SqlError';\nimport { Effect } from 'effect';\nimport type * as V1 from '~/_relations.ts';\nimport type { EffectCache } from '~/cache/core/cache-effect.ts';\nimport { strategyFor } from '~/cache/core/cache.ts';\nimport type { WithCacheConfig } from '~/cache/core/types.ts';\nimport { entityKind } from '~/entity.ts';\nimport { DrizzleQueryError, TransactionRollbackError } from '~/errors.ts';\nimport type { MigrationConfig, MigrationMeta, MigratorInitFailResponse } from '~/migrator.ts';\nimport type { AnyRelations, EmptyRelations } from '~/relations.ts';\nimport { type Query, type SQL, sql } from '~/sql/sql.ts';\nimport { assertUnreachable } from '~/utils.ts';\nimport type { PgDialect } from '../dialect.ts';\nimport type { SelectedFieldsOrdered } from '../query-builders/select.types.ts';\nimport {\n\tPgBasePreparedQuery,\n\ttype PgQueryResultHKT,\n\tPgSession,\n\ttype PgTransactionConfig,\n\ttype PreparedQueryConfig,\n} from '../session.ts';\nimport { PgEffectDatabase } from './db.ts';\n\nexport abstract class PgEffectPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {\n\tstatic override readonly [entityKind]: string = 'PgEffectPreparedQuery';\n\n\tconstructor(\n\t\tquery: Query,\n\t\tprivate cache: EffectCache | undefined,\n\t\tprivate queryMetadata: {\n\t\t\ttype: 'select' | 'update' | 'delete' | 'insert';\n\t\t\ttables: string[];\n\t\t} | undefined,\n\t\tprivate cacheConfig?: WithCacheConfig,\n\t) {\n\t\tsuper(query);\n\n\t\tif (cache && cache.strategy() === 'all' && cacheConfig === undefined) {\n\t\t\tthis.cacheConfig = { enabled: true, autoInvalidate: true };\n\t\t}\n\t\tif (!this.cacheConfig?.enabled) {\n\t\t\tthis.cacheConfig = undefined;\n\t\t}\n\t}\n\n\tprotected override queryWithCache<T>(\n\t\tqueryString: string,\n\t\tparams: any[],\n\t\tquery: Effect.Effect<T, SqlError>,\n\t): Effect.Effect<T, DrizzleQueryError> {\n\t\tconst { cache, cacheConfig, queryMetadata } = this;\n\t\treturn Effect.gen(function*() {\n\t\t\tconst cacheStrat: Awaited<ReturnType<typeof strategyFor>> = cache\n\t\t\t\t? yield* Effect.tryPromise(\n\t\t\t\t\t() => strategyFor(queryString, params, queryMetadata, cacheConfig),\n\t\t\t\t)\n\t\t\t\t: { type: 'skip' as const };\n\n\t\t\tif (cacheStrat.type === 'skip') {\n\t\t\t\treturn yield* query;\n\t\t\t}\n\n\t\t\t// For mutate queries, we should query the database, wait for a response, and then perform invalidation\n\t\t\tif (cacheStrat.type === 'invalidate') {\n\t\t\t\tconst result = yield* query;\n\t\t\t\tyield* cache!.onMutate({ tables: cacheStrat.tables });\n\n\t\t\t\treturn result;\n\t\t\t}\n\n\t\t\tif (cacheStrat.type === 'try') {\n\t\t\t\tconst { tables, key, isTag, autoInvalidate, config } = cacheStrat;\n\t\t\t\tconst fromCache: any[] | undefined = yield* cache!.get(\n\t\t\t\t\tkey,\n\t\t\t\t\ttables,\n\t\t\t\t\tisTag,\n\t\t\t\t\tautoInvalidate,\n\t\t\t\t);\n\n\t\t\t\tif (typeof fromCache !== 'undefined') return fromCache as unknown as T;\n\n\t\t\t\tconst result = yield* query;\n\n\t\t\t\tyield* cache!.put(\n\t\t\t\t\tkey,\n\t\t\t\t\tresult,\n\t\t\t\t\tautoInvalidate ? tables : [],\n\t\t\t\t\tisTag,\n\t\t\t\t\tconfig,\n\t\t\t\t);\n\n\t\t\t\treturn result;\n\t\t\t}\n\n\t\t\tassertUnreachable(cacheStrat);\n\t\t}).pipe(Effect.catchAll((e) => {\n\t\t\t// eslint-disable-next-line @drizzle-internal/no-instanceof\n\t\t\treturn Effect.fail(new DrizzleQueryError(queryString, params, e instanceof Error ? e : undefined));\n\t\t}));\n\t}\n\n\tabstract override execute(\n\t\tplaceholderValues?: Record<string, unknown>,\n\t): Effect.Effect<T['execute'], DrizzleQueryError>;\n\n\t/** @internal */\n\tabstract override all(placeholderValues?: Record<string, unknown>): Effect.Effect<T['all'], DrizzleQueryError>;\n}\n\nexport abstract class PgEffectSession<\n\tTQueryResult extends PgQueryResultHKT = PgQueryResultHKT,\n\tTFullSchema extends Record<string, unknown> = Record<string, never>,\n\tTRelations extends AnyRelations = EmptyRelations,\n\tTSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>,\n> extends PgSession {\n\tstatic override readonly [entityKind]: string = 'PgEffectSession';\n\n\tconstructor(dialect: PgDialect) {\n\t\tsuper(dialect);\n\t}\n\n\tabstract override prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tisResponseInArrayMode: boolean,\n\t\tcustomResultMapper?: (rows: unknown[][], mapColumnValue?: (value: unknown) => unknown) => T['execute'],\n\t\tqueryMetadata?: {\n\t\t\ttype: 'select' | 'update' | 'delete' | 'insert';\n\t\t\ttables: string[];\n\t\t},\n\t\tcacheConfig?: WithCacheConfig,\n\t): PgEffectPreparedQuery<T>;\n\n\tabstract override prepareRelationalQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tcustomResultMapper: (\n\t\t\trows: Record<string, unknown>[],\n\t\t\tmapColumnValue?: (value: unknown) => unknown,\n\t\t) => T['execute'],\n\t): PgEffectPreparedQuery<T>;\n\n\toverride execute<T>(query: SQL): Effect.Effect<T, DrizzleQueryError> {\n\t\tconst { sql, params } = this.dialect.sqlToQuery(query);\n\t\treturn this.prepareQuery<PreparedQueryConfig & { execute: T }>({ sql, params }, undefined, undefined, false)\n\t\t\t.execute();\n\t}\n\n\toverride all<T>(query: SQL): Effect.Effect<T[], DrizzleQueryError> {\n\t\tconst { sql, params } = this.dialect.sqlToQuery(query);\n\t\treturn this.prepareQuery<PreparedQueryConfig & { all: T[] }>({ sql, params }, undefined, undefined, false)\n\t\t\t.all();\n\t}\n\n\tabstract transaction<T>(\n\t\ttransaction: (\n\t\t\ttx: PgEffectTransaction<TQueryResult, TFullSchema, TRelations, TSchema>,\n\t\t) => Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>,\n\t): Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>;\n}\n\nexport abstract class PgEffectTransaction<\n\tTQueryResult extends PgQueryResultHKT,\n\tTFullSchema extends Record<string, unknown> = Record<string, never>,\n\tTRelations extends AnyRelations = EmptyRelations,\n\tTSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>,\n> extends PgEffectDatabase<TQueryResult, TFullSchema, TRelations, TSchema> {\n\tstatic override readonly [entityKind]: string = 'PgEffectTransaction';\n\n\tconstructor(\n\t\tdialect: PgDialect,\n\t\tsession: PgEffectSession<any, any, any, any>,\n\t\tprotected relations: TRelations,\n\t\tprotected schema: {\n\t\t\tfullSchema: Record<string, unknown>;\n\t\t\tschema: TSchema;\n\t\t\ttableNamesMap: Record<string, string>;\n\t\t} | undefined,\n\t\tprotected readonly nestedIndex = 0,\n\t\tparseRqbJson?: boolean,\n\t) {\n\t\tsuper(dialect, session, relations, schema, parseRqbJson);\n\t}\n\n\trollback = function*() {\n\t\tyield* Effect.fail(new TransactionRollbackError());\n\t};\n\n\t/** @internal */\n\tgetTransactionConfigSQL(config: PgTransactionConfig): SQL {\n\t\tconst chunks: string[] = [];\n\t\tif (config.isolationLevel) {\n\t\t\tchunks.push(`isolation level ${config.isolationLevel}`);\n\t\t}\n\t\tif (config.accessMode) {\n\t\t\tchunks.push(config.accessMode);\n\t\t}\n\t\tif (typeof config.deferrable === 'boolean') {\n\t\t\tchunks.push(config.deferrable ? 'deferrable' : 'not deferrable');\n\t\t}\n\t\treturn sql.raw(chunks.join(' '));\n\t}\n\n\tsetTransaction(config: PgTransactionConfig): Effect.Effect<void, DrizzleQueryError> {\n\t\treturn this.session.execute(sql`set transaction ${this.getTransactionConfigSQL(config)}`);\n\t}\n\n\tabstract override transaction<T>(\n\t\ttransaction: (\n\t\t\ttx: PgEffectTransaction<TQueryResult, TFullSchema, TRelations, TSchema>,\n\t\t) => Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>,\n\t): Effect.Effect<T, DrizzleQueryError | TransactionRollbackError>;\n}\n\nexport function migrate(\n\tmigrations: MigrationMeta[],\n\tsession: PgEffectSession,\n\tconfig: string | MigrationConfig,\n): Effect.Effect<void | MigratorInitFailResponse, DrizzleQueryError, never> {\n\treturn Effect.gen(function*() {\n\t\tconst migrationsTable = typeof config === 'string'\n\t\t\t? '__drizzle_migrations'\n\t\t\t: config.migrationsTable ?? '__drizzle_migrations';\n\t\tconst migrationsSchema = typeof config === 'string' ? 'drizzle' : config.migrationsSchema ?? 'drizzle';\n\t\tconst migrationTableCreate = sql`\n\t\t\tCREATE TABLE IF NOT EXISTS ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)} (\n\t\t\t\tid SERIAL PRIMARY KEY,\n\t\t\t\thash text NOT NULL,\n\t\t\t\tcreated_at bigint\n\t\t\t)\n\t\t`;\n\t\tyield* session.execute(sql`CREATE SCHEMA IF NOT EXISTS ${sql.identifier(migrationsSchema)}`);\n\t\tyield* session.execute(migrationTableCreate);\n\n\t\tconst dbMigrations = yield* session.all<{ id: number; hash: string; created_at: string }>(\n\t\t\tsql`select id, hash, created_at from ${sql.identifier(migrationsSchema)}.${\n\t\t\t\tsql.identifier(migrationsTable)\n\t\t\t} order by created_at desc limit 1`,\n\t\t);\n\n\t\tif (typeof config === 'object' && config.init) {\n\t\t\tif (dbMigrations.length) {\n\t\t\t\treturn { exitCode: 'databaseMigrations' as const };\n\t\t\t}\n\n\t\t\tif (migrations.length > 1) {\n\t\t\t\treturn { exitCode: 'localMigrations' as const };\n\t\t\t}\n\n\t\t\tconst [migration] = migrations;\n\n\t\t\tif (!migration) return;\n\n\t\t\tyield* session.execute(\n\t\t\t\tsql`insert into ${sql.identifier(migrationsSchema)}.${\n\t\t\t\t\tsql.identifier(migrationsTable)\n\t\t\t\t} (\"hash\", \"created_at\") values(${migration.hash}, ${migration.folderMillis})`,\n\t\t\t);\n\n\t\t\treturn;\n\t\t}\n\n\t\tconst lastDbMigration = dbMigrations[0];\n\t\tyield* session.transaction((tx) =>\n\t\t\tEffect.gen(function*() {\n\t\t\t\tfor (const migration of migrations) {\n\t\t\t\t\tif (\n\t\t\t\t\t\t!lastDbMigration\n\t\t\t\t\t\t|| Number(lastDbMigration.created_at) < migration.folderMillis\n\t\t\t\t\t) {\n\t\t\t\t\t\tfor (const stmt of migration.sql) {\n\t\t\t\t\t\t\tyield* tx.execute(sql.raw(stmt));\n\t\t\t\t\t\t}\n\t\t\t\t\t\tyield* tx.execute(\n\t\t\t\t\t\t\tsql`insert into ${sql.identifier(migrationsSchema)}.${\n\t\t\t\t\t\t\t\tsql.identifier(migrationsTable)\n\t\t\t\t\t\t\t} (\"hash\", \"created_at\") values(${migration.hash}, ${migration.folderMillis})`,\n\t\t\t\t\t\t);\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t})\n\t\t);\n\n\t\treturn;\n\t}) as Effect.Effect<void | MigratorInitFailResponse, DrizzleQueryError, never>;\n}\n"],"mappings":"AACA,SAAS,cAAc;AAGvB,SAAS,mBAAmB;AAE5B,SAAS,kBAAkB;AAC3B,SAAS,mBAAmB,gCAAgC;AAG5D,SAA+B,WAAW;AAC1C,SAAS,yBAAyB;AAGlC;AAAA,EACC;AAAA,EAEA;AAAA,OAGM;AACP,SAAS,wBAAwB;AAE1B,MAAe,8BAA6D,oBAAoB;AAAA,EAGtG,YACC,OACQ,OACA,eAIA,aACP;AACD,UAAM,KAAK;AAPH;AACA;AAIA;AAIR,QAAI,SAAS,MAAM,SAAS,MAAM,SAAS,gBAAgB,QAAW;AACrE,WAAK,cAAc,EAAE,SAAS,MAAM,gBAAgB,KAAK;AAAA,IAC1D;AACA,QAAI,CAAC,KAAK,aAAa,SAAS;AAC/B,WAAK,cAAc;AAAA,IACpB;AAAA,EACD;AAAA,EAnBA,QAA0B,UAAU,IAAY;AAAA,EAqB7B,eAClB,aACA,QACA,OACsC;AACtC,UAAM,EAAE,OAAO,aAAa,cAAc,IAAI;AAC9C,WAAO,OAAO,IAAI,aAAY;AAC7B,YAAM,aAAsD,QACzD,OAAO,OAAO;AAAA,QACf,MAAM,YAAY,aAAa,QAAQ,eAAe,WAAW;AAAA,MAClE,IACE,EAAE,MAAM,OAAgB;AAE3B,UAAI,WAAW,SAAS,QAAQ;AAC/B,eAAO,OAAO;AAAA,MACf;AAGA,UAAI,WAAW,SAAS,cAAc;AACrC,cAAM,SAAS,OAAO;AACtB,eAAO,MAAO,SAAS,EAAE,QAAQ,WAAW,OAAO,CAAC;AAEpD,eAAO;AAAA,MACR;AAEA,UAAI,WAAW,SAAS,OAAO;AAC9B,cAAM,EAAE,QAAQ,KAAK,OAAO,gBAAgB,OAAO,IAAI;AACvD,cAAM,YAA+B,OAAO,MAAO;AAAA,UAClD;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACD;AAEA,YAAI,OAAO,cAAc,YAAa,QAAO;AAE7C,cAAM,SAAS,OAAO;AAEtB,eAAO,MAAO;AAAA,UACb;AAAA,UACA;AAAA,UACA,iBAAiB,SAAS,CAAC;AAAA,UAC3B;AAAA,UACA;AAAA,QACD;AAEA,eAAO;AAAA,MACR;AAEA,wBAAkB,UAAU;AAAA,IAC7B,CAAC,EAAE,KAAK,OAAO,SAAS,CAAC,MAAM;AAE9B,aAAO,OAAO,KAAK,IAAI,kBAAkB,aAAa,QAAQ,aAAa,QAAQ,IAAI,MAAS,CAAC;AAAA,IAClG,CAAC,CAAC;AAAA,EACH;AAQD;AAEO,MAAe,wBAKZ,UAAU;AAAA,EACnB,QAA0B,UAAU,IAAY;AAAA,EAEhD,YAAY,SAAoB;AAC/B,UAAM,OAAO;AAAA,EACd;AAAA,EAyBS,QAAW,OAAiD;AACpE,UAAM,EAAE,KAAAA,MAAK,OAAO,IAAI,KAAK,QAAQ,WAAW,KAAK;AACrD,WAAO,KAAK,aAAmD,EAAE,KAAAA,MAAK,OAAO,GAAG,QAAW,QAAW,KAAK,EACzG,QAAQ;AAAA,EACX;AAAA,EAES,IAAO,OAAmD;AAClE,UAAM,EAAE,KAAAA,MAAK,OAAO,IAAI,KAAK,QAAQ,WAAW,KAAK;AACrD,WAAO,KAAK,aAAiD,EAAE,KAAAA,MAAK,OAAO,GAAG,QAAW,QAAW,KAAK,EACvG,IAAI;AAAA,EACP;AAOD;AAEO,MAAe,4BAKZ,iBAAiE;AAAA,EAG1E,YACC,SACA,SACU,WACA,QAKS,cAAc,GACjC,cACC;AACD,UAAM,SAAS,SAAS,WAAW,QAAQ,YAAY;AAT7C;AACA;AAKS;AAAA,EAIpB;AAAA,EAfA,QAA0B,UAAU,IAAY;AAAA,EAiBhD,WAAW,aAAY;AACtB,WAAO,OAAO,KAAK,IAAI,yBAAyB,CAAC;AAAA,EAClD;AAAA;AAAA,EAGA,wBAAwB,QAAkC;AACzD,UAAM,SAAmB,CAAC;AAC1B,QAAI,OAAO,gBAAgB;AAC1B,aAAO,KAAK,mBAAmB,OAAO,cAAc,EAAE;AAAA,IACvD;AACA,QAAI,OAAO,YAAY;AACtB,aAAO,KAAK,OAAO,UAAU;AAAA,IAC9B;AACA,QAAI,OAAO,OAAO,eAAe,WAAW;AAC3C,aAAO,KAAK,OAAO,aAAa,eAAe,gBAAgB;AAAA,IAChE;AACA,WAAO,IAAI,IAAI,OAAO,KAAK,GAAG,CAAC;AAAA,EAChC;AAAA,EAEA,eAAe,QAAqE;AACnF,WAAO,KAAK,QAAQ,QAAQ,sBAAsB,KAAK,wBAAwB,MAAM,CAAC,EAAE;AAAA,EACzF;AAOD;AAEO,SAAS,QACf,YACA,SACA,QAC2E;AAC3E,SAAO,OAAO,IAAI,aAAY;AAC7B,UAAM,kBAAkB,OAAO,WAAW,WACvC,yBACA,OAAO,mBAAmB;AAC7B,UAAM,mBAAmB,OAAO,WAAW,WAAW,YAAY,OAAO,oBAAoB;AAC7F,UAAM,uBAAuB;AAAA,gCACC,IAAI,WAAW,gBAAgB,CAAC,IAAI,IAAI,WAAW,eAAe,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAMjG,WAAO,QAAQ,QAAQ,kCAAkC,IAAI,WAAW,gBAAgB,CAAC,EAAE;AAC3F,WAAO,QAAQ,QAAQ,oBAAoB;AAE3C,UAAM,eAAe,OAAO,QAAQ;AAAA,MACnC,uCAAuC,IAAI,WAAW,gBAAgB,CAAC,IACtE,IAAI,WAAW,eAAe,CAC/B;AAAA,IACD;AAEA,QAAI,OAAO,WAAW,YAAY,OAAO,MAAM;AAC9C,UAAI,aAAa,QAAQ;AACxB,eAAO,EAAE,UAAU,qBAA8B;AAAA,MAClD;AAEA,UAAI,WAAW,SAAS,GAAG;AAC1B,eAAO,EAAE,UAAU,kBAA2B;AAAA,MAC/C;AAEA,YAAM,CAAC,SAAS,IAAI;AAEpB,UAAI,CAAC,UAAW;AAEhB,aAAO,QAAQ;AAAA,QACd,kBAAkB,IAAI,WAAW,gBAAgB,CAAC,IACjD,IAAI,WAAW,eAAe,CAC/B,kCAAkC,UAAU,IAAI,KAAK,UAAU,YAAY;AAAA,MAC5E;AAEA;AAAA,IACD;AAEA,UAAM,kBAAkB,aAAa,CAAC;AACtC,WAAO,QAAQ;AAAA,MAAY,CAAC,OAC3B,OAAO,IAAI,aAAY;AACtB,mBAAW,aAAa,YAAY;AACnC,cACC,CAAC,mBACE,OAAO,gBAAgB,UAAU,IAAI,UAAU,cACjD;AACD,uBAAW,QAAQ,UAAU,KAAK;AACjC,qBAAO,GAAG,QAAQ,IAAI,IAAI,IAAI,CAAC;AAAA,YAChC;AACA,mBAAO,GAAG;AAAA,cACT,kBAAkB,IAAI,WAAW,gBAAgB,CAAC,IACjD,IAAI,WAAW,eAAe,CAC/B,kCAAkC,UAAU,IAAI,KAAK,UAAU,YAAY;AAAA,YAC5E;AAAA,UACD;AAAA,QACD;AAAA,MACD,CAAC;AAAA,IACF;AAEA;AAAA,EACD,CAAC;AACF;","names":["sql"]}