@effect/sql-pg 0.0.0-snapshot-189d4cae80e186661241002ad9d729628096520f

package/src/Migrator.ts

@@ -0,0 +1,113 @@
+/**
+ * @since 1.0.0
+ */
+import * as Command from "@effect/platform/Command"
+import type { CommandExecutor } from "@effect/platform/CommandExecutor"
+import { FileSystem } from "@effect/platform/FileSystem"
+import { Path } from "@effect/platform/Path"
+import type { SqlError } from "@effect/sql/Error"
+import * as Migrator from "@effect/sql/Migrator"
+import * as Effect from "effect/Effect"
+import * as Layer from "effect/Layer"
+import * as Secret from "effect/Secret"
+import * as Client from "./Client.js"
+
+/**
+ * @since 1.0.0
+ */
+export * from "@effect/sql/Migrator"
+
+/**
+ * @category constructor
+ * @since 1.0.0
+ */
+export const run: <R>(
+  options: Migrator.MigratorOptions<R>
+) => Effect.Effect<
+  ReadonlyArray<readonly [id: number, name: string]>,
+  SqlError | Migrator.MigrationError,
+  Client.PgClient | FileSystem | Path | CommandExecutor | R
+> = Migrator.make({
+  getClient: Client.PgClient,
+  ensureTable(sql, table) {
+    return Effect.catchAll(
+      sql`select ${table}::regclass`,
+      () =>
+        sql`
+        CREATE TABLE ${sql(table)} (
+          migration_id integer primary key,
+          created_at timestamp with time zone not null default now(),
+          name text not null
+        )
+      `
+    )
+  },
+  lockTable(sql, table) {
+    return sql`
+      LOCK TABLE ${sql(table)} IN ACCESS EXCLUSIVE MODE
+    `
+  },
+  dumpSchema(sql, path, table) {
+    const pgDump = (args: Array<string>) =>
+      Effect.gen(function*(_) {
+        const dump = yield* _(
+          Command.make("pg_dump", ...args, "--no-owner", "--no-privileges"),
+          Command.env({
+            PATH: (globalThis as any).process?.env.PATH,
+            PGHOST: sql.config.host,
+            PGPORT: sql.config.port?.toString(),
+            PGUSER: sql.config.username,
+            PGPASSWORD: sql.config.password
+              ? Secret.value(sql.config.password)
+              : undefined,
+            PGDATABASE: sql.config.database,
+            PGSSLMODE: sql.config.ssl ? "require" : "prefer"
+          }),
+          Command.string
+        )
+
+        return dump.replace(/^--.*$/gm, "")
+          .replace(/^SET .*$/gm, "")
+          .replace(/^SELECT pg_catalog\..*$/gm, "")
+          .replace(/\n{2,}/gm, "\n\n")
+          .trim()
+      }).pipe(
+        Effect.mapError((error) => new Migrator.MigrationError({ reason: "failed", message: error.message }))
+      )
+
+    const pgDumpSchema = pgDump(["--schema-only"])
+
+    const pgDumpMigrations = pgDump([
+      "--column-inserts",
+      "--data-only",
+      `--table=${table}`
+    ])
+
+    const pgDumpAll = Effect.map(
+      Effect.all([pgDumpSchema, pgDumpMigrations], { concurrency: 2 }),
+      ([schema, migrations]) => schema + "\n\n" + migrations
+    )
+
+    const pgDumpFile = (path: string) =>
+      Effect.gen(function*(_) {
+        const fs = yield* _(FileSystem)
+        const path_ = yield* _(Path)
+        const dump = yield* _(pgDumpAll)
+        yield* _(fs.makeDirectory(path_.dirname(path), { recursive: true }))
+        yield* _(fs.writeFileString(path, dump))
+      }).pipe(
+        Effect.mapError((error) => new Migrator.MigrationError({ reason: "failed", message: error.message }))
+      )
+
+    return pgDumpFile(path)
+  }
+})
+
+/**
+ * @category layers
+ * @since 1.0.0
+ */
+export const layer = (
+  options: Migrator.MigratorOptions
+): Layer.Layer<never, SqlError | Migrator.MigrationError, Client.PgClient | FileSystem | Path | CommandExecutor> =>
+  Layer.effectDiscard(run(options))

package/src/index.ts

@@ -0,0 +1,33 @@
+/**
+ * @since 1.0.0
+ */
+
+/**
+ * @since 1.0.0
+ */
+export * as client from "./Client.js"
+
+/**
+ * @since 1.0.0
+ */
+export * as error from "@effect/sql/Error"
+
+/**
+ * @since 1.0.0
+ */
+export * as migrator from "./Migrator.js"
+
+/**
+ * @since 1.0.0
+ */
+export * as resolver from "@effect/sql/Resolver"
+
+/**
+ * @since 1.0.0
+ */
+export * as schema from "@effect/sql/Schema"
+
+/**
+ * @since 1.0.0
+ */
+export * as statement from "@effect/sql/Statement"