@effect/sql-pg 4.0.0-beta.5 → 4.0.0-beta.51

package/src/PgMigrator.ts

@@ -1,11 +1,17 @@
 /**
  * @since 1.0.0
  */
-import type * as Effect from "effect/Effect"
+import * as Effect from "effect/Effect"
+import * as FileSystem from "effect/FileSystem"
 import * as Layer from "effect/Layer"
+import * as Path from "effect/Path"
+import * as Redacted from "effect/Redacted"
+import * as ChildProcess from "effect/unstable/process/ChildProcess"
+import * as ChildProcessSpawner from "effect/unstable/process/ChildProcessSpawner"
 import * as Migrator from "effect/unstable/sql/Migrator"
-import type * as Client from "effect/unstable/sql/SqlClient"
+import type { SqlClient } from "effect/unstable/sql/SqlClient"
 import type { SqlError } from "effect/unstable/sql/SqlError"
+import { PgClient } from "./PgClient.ts"
 
 /**
  * @since 1.0.0
@@ -21,64 +27,67 @@ export const run: <R2 = never>(
 ) => Effect.Effect<
   ReadonlyArray<readonly [id: number, name: string]>,
   Migrator.MigrationError | SqlError,
-  Client.SqlClient | R2
+  | SqlClient
+  | PgClient
+  | ChildProcessSpawner.ChildProcessSpawner
+  | FileSystem.FileSystem
+  | Path.Path
+  | R2
 > = Migrator.make({
-  // TODO: Wait for Command module
-  // dumpSchema(path, table) {
-  //   const pgDump = (args: Array<string>) =>
-  //     Effect.gen(function*() {
-  //       const sql = yield* PgClient
-  //       const dump = yield* pipe(
-  //         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
-  //             ? Redacted.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({ kind: "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({ kind: "Failed", message: error.message }))
-  //     )
-  //
-  //   return pgDumpFile(path)
-  // }
+  dumpSchema(path, table) {
+    const pgDump = (args: Array<string>) =>
+      Effect.gen(function*() {
+        const sql = yield* PgClient
+        const spawner = yield* ChildProcessSpawner.ChildProcessSpawner
+        const dump = yield* ChildProcess.make("pg_dump", [...args, "--no-owner", "--no-privileges"], {
+          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
+              ? Redacted.value(sql.config.password)
+              : undefined,
+            PGDATABASE: sql.config.database,
+            PGSSLMODE: sql.config.ssl ? "require" : "prefer"
+          }
+        }).pipe(spawner.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({ kind: "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.FileSystem
+        const path_ = yield* Path.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({ kind: "Failed", message: error.message }))
+      )
+
+    return pgDumpFile(path)
+  }
 })
 
 /**
@@ -90,5 +99,10 @@ export const layer = <R>(
 ): Layer.Layer<
   never,
   Migrator.MigrationError | SqlError,
-  Client.SqlClient | R
+  | SqlClient
+  | PgClient
+  | ChildProcessSpawner.ChildProcessSpawner
+  | FileSystem.FileSystem
+  | Path.Path
+  | R
 > => Layer.effectDiscard(run(options))