@cedarjs/pg 0.1.0-alpha.1 → 0.2.0-alpha.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (121) hide show
  1. package/README.md +276 -25
  2. package/dist/cli.cjs +60 -14
  3. package/dist/cli.cjs.map +1 -1
  4. package/dist/cli.mjs +59 -13
  5. package/dist/cli.mjs.map +1 -1
  6. package/dist/{constants-CNZn5Xro.cjs → constants-Ct8myrEn.cjs} +2 -2
  7. package/dist/{constants-BY97wXjA.mjs.map → constants-Ct8myrEn.cjs.map} +1 -1
  8. package/dist/{constants-BY97wXjA.mjs → constants-NLR0U4NR.mjs} +2 -2
  9. package/dist/{constants-CNZn5Xro.cjs.map → constants-NLR0U4NR.mjs.map} +1 -1
  10. package/dist/dev-env.cjs +12 -0
  11. package/dist/dev-env.cjs.map +1 -0
  12. package/dist/dev-env.d.cts +1 -0
  13. package/dist/dev-env.d.mts +1 -0
  14. package/dist/dev-env.mjs +14 -0
  15. package/dist/dev-env.mjs.map +1 -0
  16. package/dist/index.cjs +50 -10
  17. package/dist/index.cjs.map +1 -0
  18. package/dist/index.d.cts +168 -82
  19. package/dist/index.d.mts +168 -82
  20. package/dist/index.mjs +32 -2
  21. package/dist/index.mjs.map +1 -0
  22. package/dist/jest-teardown.cjs +18 -0
  23. package/dist/jest-teardown.cjs.map +1 -0
  24. package/dist/jest-teardown.d.cts +13 -0
  25. package/dist/jest-teardown.d.mts +14 -0
  26. package/dist/jest-teardown.mjs +18 -0
  27. package/dist/jest-teardown.mjs.map +1 -0
  28. package/dist/jest-template.cjs +41 -0
  29. package/dist/jest-template.cjs.map +1 -0
  30. package/dist/jest-template.d.cts +28 -0
  31. package/dist/jest-template.d.mts +28 -0
  32. package/dist/jest-template.mjs +35 -0
  33. package/dist/jest-template.mjs.map +1 -0
  34. package/dist/jest.cjs +12 -15
  35. package/dist/jest.cjs.map +1 -1
  36. package/dist/jest.d.cts +10 -7
  37. package/dist/jest.d.mts +10 -6
  38. package/dist/jest.mjs +12 -10
  39. package/dist/jest.mjs.map +1 -1
  40. package/dist/lease-B3TuX92y.d.mts +24 -0
  41. package/dist/lease-BSBBpaR4.mjs +201 -0
  42. package/dist/lease-BSBBpaR4.mjs.map +1 -0
  43. package/dist/lease-Bq9wKwQa.cjs +272 -0
  44. package/dist/lease-Bq9wKwQa.cjs.map +1 -0
  45. package/dist/lease-t4I9JahV.d.cts +24 -0
  46. package/dist/lifecycle-BOo6xBjD.mjs +684 -0
  47. package/dist/lifecycle-BOo6xBjD.mjs.map +1 -0
  48. package/dist/lifecycle-DEJ4GgWV.cjs +856 -0
  49. package/dist/lifecycle-DEJ4GgWV.cjs.map +1 -0
  50. package/dist/load-dev-env-7UjMHhw7.mjs +10 -0
  51. package/dist/load-dev-env-7UjMHhw7.mjs.map +1 -0
  52. package/dist/load-dev-env-cII6N4ZP.cjs +15 -0
  53. package/dist/load-dev-env-cII6N4ZP.cjs.map +1 -0
  54. package/dist/load-mode-env-B0qO9DZx.cjs +39 -0
  55. package/dist/load-mode-env-B0qO9DZx.cjs.map +1 -0
  56. package/dist/load-mode-env-C8gy4v9V.mjs +28 -0
  57. package/dist/load-mode-env-C8gy4v9V.mjs.map +1 -0
  58. package/dist/load-test-env-C34wpkIX.mjs +18 -0
  59. package/dist/load-test-env-C34wpkIX.mjs.map +1 -0
  60. package/dist/load-test-env-CkdUjTpV.cjs +23 -0
  61. package/dist/load-test-env-CkdUjTpV.cjs.map +1 -0
  62. package/dist/naming-C2nGVxPk.d.cts +39 -0
  63. package/dist/naming-C2nGVxPk.d.mts +39 -0
  64. package/dist/nx.cjs +26 -17
  65. package/dist/nx.cjs.map +1 -1
  66. package/dist/nx.d.cts +9 -20
  67. package/dist/nx.d.mts +9 -20
  68. package/dist/nx.mjs +20 -18
  69. package/dist/nx.mjs.map +1 -1
  70. package/dist/tasks-Caud9yHr.cjs +67 -0
  71. package/dist/tasks-Caud9yHr.cjs.map +1 -0
  72. package/dist/tasks-CjaZRi_G.d.mts +19 -0
  73. package/dist/tasks-DdQoP9We.d.cts +19 -0
  74. package/dist/tasks-KherHLac.mjs +38 -0
  75. package/dist/tasks-KherHLac.mjs.map +1 -0
  76. package/dist/template-CGw3C1Ob.mjs +88 -0
  77. package/dist/template-CGw3C1Ob.mjs.map +1 -0
  78. package/dist/template-D0gJ_MS2.cjs +105 -0
  79. package/dist/template-D0gJ_MS2.cjs.map +1 -0
  80. package/dist/template-mode-BkEs9LnY.cjs +91 -0
  81. package/dist/template-mode-BkEs9LnY.cjs.map +1 -0
  82. package/dist/template-mode-CcxV_Iju.d.cts +28 -0
  83. package/dist/template-mode-CcxV_Iju.d.mts +28 -0
  84. package/dist/template-mode-IjlFOG4O.mjs +80 -0
  85. package/dist/template-mode-IjlFOG4O.mjs.map +1 -0
  86. package/dist/test-env.cjs +16 -0
  87. package/dist/test-env.cjs.map +1 -0
  88. package/dist/test-env.d.cts +1 -0
  89. package/dist/test-env.d.mts +1 -0
  90. package/dist/test-env.mjs +18 -0
  91. package/dist/test-env.mjs.map +1 -0
  92. package/dist/vite-plus.cjs +8 -33
  93. package/dist/vite-plus.cjs.map +1 -1
  94. package/dist/vite-plus.d.cts +4 -48
  95. package/dist/vite-plus.d.mts +4 -48
  96. package/dist/vite-plus.mjs +5 -30
  97. package/dist/vite-plus.mjs.map +1 -1
  98. package/dist/vitest-template.cjs +48 -0
  99. package/dist/vitest-template.cjs.map +1 -0
  100. package/dist/vitest-template.d.cts +31 -0
  101. package/dist/vitest-template.d.mts +31 -0
  102. package/dist/vitest-template.mjs +42 -0
  103. package/dist/vitest-template.mjs.map +1 -0
  104. package/dist/vitest.cjs +10 -10
  105. package/dist/vitest.cjs.map +1 -1
  106. package/dist/vitest.d.cts +7 -3
  107. package/dist/vitest.d.mts +7 -2
  108. package/dist/vitest.mjs +10 -5
  109. package/dist/vitest.mjs.map +1 -1
  110. package/package.json +36 -12
  111. package/scripts/autopg-version +1 -0
  112. package/scripts/ci-install-autopg.sh +73 -0
  113. package/scripts/postinstall.js +35 -9
  114. package/dist/lifecycle-BGYtVXtx.cjs +0 -743
  115. package/dist/lifecycle-BGYtVXtx.cjs.map +0 -1
  116. package/dist/lifecycle-CT_8AWPv.mjs +0 -577
  117. package/dist/lifecycle-CT_8AWPv.mjs.map +0 -1
  118. package/dist/tasks-13P6tMth.mjs +0 -16
  119. package/dist/tasks-13P6tMth.mjs.map +0 -1
  120. package/dist/tasks-B8ryV9xo.cjs +0 -21
  121. package/dist/tasks-B8ryV9xo.cjs.map +0 -1
@@ -0,0 +1,684 @@
1
+ import { n as PASSWORD_SALT_PREFIX } from "./constants-NLR0U4NR.mjs";
2
+ import { a as leaseDir, f as resolveWorktreeIdentity, i as isOrphanLease, l as readLease, n as envPath, r as forgetLease, s as listRegistryLeases, u as writeLease } from "./lease-BSBBpaR4.mjs";
3
+ import { existsSync, mkdirSync, writeFileSync } from "node:fs";
4
+ import { homedir, tmpdir } from "node:os";
5
+ import { join } from "node:path";
6
+ import { createHash } from "node:crypto";
7
+ import { execFileSync, spawn, spawnSync } from "node:child_process";
8
+ import pg from "pg";
9
+ import { createConnection } from "node:net";
10
+ //#region src/core/naming.ts
11
+ const PG_MAX = 63;
12
+ const PREFIX = "cpg_";
13
+ /**
14
+ * Build an observable Postgres database name:
15
+ * cpg_<repoSlug>_<worktreeSlug>_<mode>_<pathHash8>
16
+ *
17
+ * Never drops `mode` or `pathHash`. Truncates slugs to fit ≤63 chars.
18
+ */
19
+ function buildDatabaseName(identity, mode) {
20
+ const modePart = mode;
21
+ const hashPart = identity.pathHash;
22
+ const fixed = 6 + modePart.length + 1 + hashPart.length;
23
+ let budget = PG_MAX - fixed;
24
+ if (budget < 2) return `${PREFIX}${modePart}_${hashPart}`.slice(0, PG_MAX);
25
+ let repo = identity.repoSlug;
26
+ let wt = identity.worktreeSlug;
27
+ while (repo.length + wt.length > budget) if (repo.length >= wt.length && repo.length > 1) repo = repo.slice(0, -1);
28
+ else if (wt.length > 1) wt = wt.slice(0, -1);
29
+ else break;
30
+ const name = `${PREFIX}${repo}_${wt}_${modePart}_${hashPart}`;
31
+ if (name.length > PG_MAX) return name.slice(0, PG_MAX);
32
+ return name;
33
+ }
34
+ function buildRoleName(databaseName) {
35
+ const suffix = "_role";
36
+ const maxBase = PG_MAX - 5;
37
+ return `${databaseName.slice(0, maxBase)}${suffix}`;
38
+ }
39
+ /**
40
+ * Build a worker clone datname from a TEMPLATE database name.
41
+ * Layout: `<template>_c_<suffix>` truncated to ≤63 chars (keeps suffix).
42
+ */
43
+ function buildCloneDatabaseName(templateName, suffix) {
44
+ const safe = suffix.toLowerCase().replace(/[^a-z0-9_]+/g, "_").replace(/^_+|_+$/g, "").slice(0, 24);
45
+ const tag = safe.length > 0 ? safe : "w";
46
+ const sep = "_c_";
47
+ const maxTemplate = PG_MAX - 3 - tag.length;
48
+ return `${templateName.slice(0, Math.max(1, maxTemplate))}${sep}${tag}`.slice(0, PG_MAX);
49
+ }
50
+ //#endregion
51
+ //#region src/core/policy.ts
52
+ /**
53
+ * Inject DATABASE_URL (and TEST_DATABASE_URL for test mode) from a resolved URL.
54
+ * Single env path for acquire, clone, and external-url skip.
55
+ */
56
+ function applyDatabaseUrlEnv(databaseUrl, options = {}) {
57
+ process.env.DATABASE_URL = databaseUrl;
58
+ if ((options.mode ?? "test") === "test") process.env.TEST_DATABASE_URL = databaseUrl;
59
+ }
60
+ /**
61
+ * Shared host skip+env gate for `acquireIfNeeded` / `cloneFromTemplateIfNeeded`.
62
+ * On external-url skip, applies env when `setEnv` is not false.
63
+ */
64
+ async function runIfNeeded(options, run) {
65
+ const skip = resolveAcquireSkip({
66
+ url: options.url,
67
+ force: options.force,
68
+ disabled: options.disabled
69
+ });
70
+ if (skip.skip) {
71
+ if (skip.reason === "external-url") {
72
+ if (options.setEnv !== false) applyDatabaseUrlEnv(skip.databaseUrl, { mode: options.mode });
73
+ return {
74
+ status: "skipped",
75
+ reason: "external-url",
76
+ databaseUrl: skip.databaseUrl
77
+ };
78
+ }
79
+ return {
80
+ status: "skipped",
81
+ reason: "disabled"
82
+ };
83
+ }
84
+ return {
85
+ status: "ran",
86
+ value: await run()
87
+ };
88
+ }
89
+ /**
90
+ * True when the URL looks like a cedarpg provisioned database (`cpg_*` name/role).
91
+ * These must never be treated as an external escape hatch; always re-acquire so
92
+ * disposed/stale shell env cannot skip provisioning.
93
+ */
94
+ function isCedarPgManagedUrl(url) {
95
+ if (!url || url.startsWith("file:")) return false;
96
+ try {
97
+ const parsed = new URL(url);
98
+ const databaseName = decodeURIComponent((parsed.pathname.replace(/^\//, "").split("?")[0] ?? "").trim());
99
+ const user = decodeURIComponent(parsed.username);
100
+ return databaseName.startsWith("cpg_") || user.startsWith("cpg_");
101
+ } catch {
102
+ return false;
103
+ }
104
+ }
105
+ /**
106
+ * True when `url` still looks like an unset dotenv/template value
107
+ * (e.g. `postgresql://{yourMachine}@localhost:5432/app_test` or
108
+ * `postgresql://<user>@localhost/db`), not a real external database.
109
+ */
110
+ function isUnsetTemplateUrl(url) {
111
+ if (/\{[^}]+\}/.test(url)) return true;
112
+ if (/<[^>]+>/.test(url)) return true;
113
+ return false;
114
+ }
115
+ /**
116
+ * True when `url` is a real external database and acquire should be skipped.
117
+ * Sqlite `file:` URLs, cedarpg `cpg_*` URLs, and unset template placeholders
118
+ * are not external.
119
+ */
120
+ function isExternalDatabaseEscapeHatch(url) {
121
+ if (!url) return false;
122
+ if (url.startsWith("file:")) return false;
123
+ if (isCedarPgManagedUrl(url)) return false;
124
+ if (isUnsetTemplateUrl(url)) return false;
125
+ return true;
126
+ }
127
+ /**
128
+ * Decide whether acquire should run.
129
+ * Adapters may call with no args (env defaults). Cedar opt-in should pass `disabled: false`
130
+ * and the relevant `url` (`DATABASE_URL` or `TEST_DATABASE_URL`).
131
+ */
132
+ function resolveAcquireSkip(input = {}) {
133
+ if (input.disabled ?? (process.env.CEDAR_PG === "0" || process.env.CEDAR_PG === "false")) return {
134
+ skip: true,
135
+ reason: "disabled"
136
+ };
137
+ if (input.force ?? process.env.CEDAR_PG_FORCE === "1") return { skip: false };
138
+ const url = input.url !== void 0 ? input.url : process.env.TEST_DATABASE_URL;
139
+ if (url && isExternalDatabaseEscapeHatch(url)) return {
140
+ skip: true,
141
+ reason: "external-url",
142
+ databaseUrl: url
143
+ };
144
+ return { skip: false };
145
+ }
146
+ //#endregion
147
+ //#region src/providers/autopg.ts
148
+ const INSTALL_HINT = "autopg is required. Install with:\n curl -fsSL https://raw.githubusercontent.com/automagik-dev/autopg/main/install.sh | bash\nThen ensure ~/.local/bin is on PATH, or set AUTOPG_BIN.";
149
+ /** Password scheme v2: sha256(PASSWORD_SALT_PREFIX + "\\0" + roleName) hex[:32]. Frozen for URL rebuild. */
150
+ const ROLE_PASSWORD_SCHEME = "v2";
151
+ function candidateBins() {
152
+ const out = [];
153
+ if (process.env.AUTOPG_BIN) out.push(process.env.AUTOPG_BIN);
154
+ out.push("autopg");
155
+ out.push(join(homedir(), ".local", "bin", "autopg"));
156
+ return out;
157
+ }
158
+ function resolveAutopgBin() {
159
+ for (const bin of candidateBins()) {
160
+ if (bin === "autopg") {
161
+ const which = spawnSync("which", ["autopg"], { encoding: "utf8" });
162
+ if (which.status === 0 && which.stdout.trim()) return which.stdout.trim();
163
+ continue;
164
+ }
165
+ if (existsSync(bin)) return bin;
166
+ }
167
+ return null;
168
+ }
169
+ function requireAutopgBin() {
170
+ const bin = resolveAutopgBin();
171
+ if (!bin) throw new Error(INSTALL_HINT);
172
+ return bin;
173
+ }
174
+ /**
175
+ * Parse `autopg status --json` output. Requires a numeric port and running !== false.
176
+ */
177
+ function parseHostStatus(json) {
178
+ let parsed;
179
+ try {
180
+ parsed = JSON.parse(json);
181
+ } catch {
182
+ throw new Error(`autopg status --json returned invalid JSON.\n${INSTALL_HINT}`);
183
+ }
184
+ if (typeof parsed.port !== "number") throw new Error(`autopg status --json missing numeric port.\n${INSTALL_HINT}`);
185
+ if (parsed.running === false) throw new Error(`autopg host is not running.\n${INSTALL_HINT}`);
186
+ return { port: parsed.port };
187
+ }
188
+ /**
189
+ * Admin URL for an autopg host on `port`.
190
+ *
191
+ * Credentials follow autopg's own defaults / env chain (not "any local Postgres"):
192
+ * - user: `AUTOPG_PG_USER` / `PGSERVE_PG_USER` / `postgres`
193
+ * - password: `AUTOPG_PG_PASSWORD` / `PGSERVE_PG_PASSWORD` / `postgres`
194
+ *
195
+ * Port is never scanned: callers pass the port from `autopg status` (attach) or the
196
+ * ephemeral recipe (`55432`). That keeps us off unrelated local servers (e.g. brew on 5432).
197
+ */
198
+ function adminUrlFor(port, env = process.env) {
199
+ return `postgresql://${encodeURIComponent(env.AUTOPG_PG_USER || env.PGSERVE_PG_USER || "postgres")}:${encodeURIComponent(env.AUTOPG_PG_PASSWORD || env.PGSERVE_PG_PASSWORD || "postgres")}@127.0.0.1:${port}/postgres`;
200
+ }
201
+ /**
202
+ * Discover a live autopg host via `autopg status --json`. Throws if the host is not proven live.
203
+ */
204
+ function discoverHost(bin = requireAutopgBin()) {
205
+ let status;
206
+ try {
207
+ status = execFileSync(bin, ["status", "--json"], {
208
+ encoding: "utf8",
209
+ stdio: [
210
+ "ignore",
211
+ "pipe",
212
+ "pipe"
213
+ ]
214
+ });
215
+ } catch (err) {
216
+ const detail = err instanceof Error ? err.message : String(err);
217
+ throw new Error(`Failed to query autopg status.\n${detail}\n${INSTALL_HINT}`);
218
+ }
219
+ const { port } = parseHostStatus(status);
220
+ return {
221
+ port,
222
+ adminUrl: adminUrlFor(port),
223
+ bin
224
+ };
225
+ }
226
+ function quoteIdent(name) {
227
+ return `"${name.replace(/"/g, "\"\"")}"`;
228
+ }
229
+ function quoteLiteral(value) {
230
+ return `'${value.replace(/'/g, "''")}'`;
231
+ }
232
+ async function withAdminClient(adminUrl, fn) {
233
+ const client = new pg.Client({ connectionString: adminUrl });
234
+ await client.connect();
235
+ try {
236
+ return await fn(client);
237
+ } finally {
238
+ await client.end();
239
+ }
240
+ }
241
+ /**
242
+ * Deterministic local-only password for an app role (Prisma/TCP need it;
243
+ * autopg hba uses `password` for 127.0.0.1).
244
+ *
245
+ * Keyed by `roleName` (not databaseName) so TEMPLATE clones that reuse the
246
+ * same role keep working when `buildDatabaseUrl` is called with a new database.
247
+ */
248
+ function rolePasswordFor(roleName) {
249
+ return createHash("sha256").update(`${PASSWORD_SALT_PREFIX}\0${roleName}`).digest("hex").slice(0, 32);
250
+ }
251
+ /**
252
+ * Idempotently CREATE ROLE + CREATE DATABASE with cedar-pg owned names.
253
+ */
254
+ async function ensureDatabase(opts) {
255
+ const password = opts.password ?? rolePasswordFor(opts.roleName);
256
+ await withAdminClient(opts.adminUrl, async (client) => {
257
+ if ((await client.query("SELECT 1 FROM pg_roles WHERE rolname = $1", [opts.roleName])).rowCount === 0) await client.query(`CREATE ROLE ${quoteIdent(opts.roleName)} WITH LOGIN PASSWORD ${quoteLiteral(password)}`);
258
+ else await client.query(`ALTER ROLE ${quoteIdent(opts.roleName)} WITH LOGIN PASSWORD ${quoteLiteral(password)}`);
259
+ if ((await client.query("SELECT 1 FROM pg_database WHERE datname = $1", [opts.databaseName])).rowCount === 0) await client.query(`CREATE DATABASE ${quoteIdent(opts.databaseName)} OWNER ${quoteIdent(opts.roleName)}`);
260
+ else await client.query(`ALTER DATABASE ${quoteIdent(opts.databaseName)} OWNER TO ${quoteIdent(opts.roleName)}`);
261
+ });
262
+ }
263
+ /**
264
+ * Mark (or unmark) a database as a PostgreSQL template (`IS_TEMPLATE`).
265
+ * Template DBs cannot be dropped until unset.
266
+ */
267
+ async function setDatabaseIsTemplate(opts) {
268
+ const flag = opts.isTemplate ? "true" : "false";
269
+ await withAdminClient(opts.adminUrl, async (client) => {
270
+ await client.query(`ALTER DATABASE ${quoteIdent(opts.databaseName)} WITH IS_TEMPLATE ${flag}`);
271
+ });
272
+ }
273
+ /**
274
+ * CREATE DATABASE … TEMPLATE … OWNER via admin connection.
275
+ * Test roles are LOGIN-only; workers cannot CREATE DATABASE themselves.
276
+ */
277
+ async function cloneDatabaseFromTemplate(opts) {
278
+ await withAdminClient(opts.adminUrl, async (client) => {
279
+ const tmpl = await client.query(`SELECT datistemplate FROM pg_database WHERE datname = $1`, [opts.templateName]);
280
+ if (!tmpl.rowCount) throw new Error(`template database not found: ${opts.templateName}`);
281
+ if (!tmpl.rows[0]?.datistemplate) throw new Error(`database is not a TEMPLATE; run markTemplate first: ${opts.templateName}`);
282
+ const exists = await client.query("SELECT 1 FROM pg_database WHERE datname = $1", [opts.databaseName]);
283
+ if (exists.rowCount && exists.rowCount > 0) throw new Error(`database already exists: ${opts.databaseName}`);
284
+ await client.query(`CREATE DATABASE ${quoteIdent(opts.databaseName)} WITH TEMPLATE ${quoteIdent(opts.templateName)} OWNER ${quoteIdent(opts.roleName)}`);
285
+ });
286
+ }
287
+ async function listOwnedDatnames(client, roleName) {
288
+ return (await client.query(`SELECT datname FROM pg_database
289
+ WHERE datdba = (SELECT oid FROM pg_roles WHERE rolname = $1)
290
+ ORDER BY datname`, [roleName])).rows.map((r) => r.datname);
291
+ }
292
+ /** Unset IS_TEMPLATE if needed, terminate backends, DROP DATABASE (no-op if missing). */
293
+ async function dropOneDatabase(client, databaseName) {
294
+ const db = await client.query(`SELECT datistemplate FROM pg_database WHERE datname = $1`, [databaseName]);
295
+ if (!db.rowCount || db.rowCount === 0) return;
296
+ if (db.rows[0]?.datistemplate) await client.query(`ALTER DATABASE ${quoteIdent(databaseName)} WITH IS_TEMPLATE false`);
297
+ await client.query(`
298
+ SELECT pg_terminate_backend(pid)
299
+ FROM pg_stat_activity
300
+ WHERE datname = $1 AND pid <> pg_backend_pid()
301
+ `, [databaseName]);
302
+ await client.query(`DROP DATABASE IF EXISTS ${quoteIdent(databaseName)}`);
303
+ }
304
+ /**
305
+ * DROP every database owned by `roleName` on one admin connection, then DROP ROLE.
306
+ * When `preferLast` is owned, it is dropped after the other owned datnames (TEMPLATE after clones).
307
+ * Never invents a DROP target beyond role ownership.
308
+ */
309
+ async function dropDatabasesOwnedByRole(opts) {
310
+ return withAdminClient(opts.adminUrl, async (client) => {
311
+ const owned = await listOwnedDatnames(client, opts.roleName);
312
+ const ordered = [...owned.filter((name) => name !== opts.preferLast), ...owned.filter((name) => name === opts.preferLast)];
313
+ const dropped = [];
314
+ for (const databaseName of ordered) {
315
+ await dropOneDatabase(client, databaseName);
316
+ dropped.push(databaseName);
317
+ }
318
+ await client.query(`DROP ROLE IF EXISTS ${quoteIdent(opts.roleName)}`);
319
+ return dropped;
320
+ });
321
+ }
322
+ /**
323
+ * DROP DATABASE (unset IS_TEMPLATE, force terminate backends) + DROP ROLE
324
+ * when the role owns no remaining databases.
325
+ */
326
+ async function dropDatabase(opts) {
327
+ await withAdminClient(opts.adminUrl, async (client) => {
328
+ await dropOneDatabase(client, opts.databaseName);
329
+ if ((await client.query(`SELECT 1 FROM pg_database WHERE datdba = (SELECT oid FROM pg_roles WHERE rolname = $1) LIMIT 1`, [opts.roleName])).rowCount === 0) await client.query(`DROP ROLE IF EXISTS ${quoteIdent(opts.roleName)}`);
330
+ });
331
+ }
332
+ function buildDatabaseUrl(opts) {
333
+ const password = opts.password ?? rolePasswordFor(opts.roleName);
334
+ return `postgresql://${encodeURIComponent(opts.roleName)}:${encodeURIComponent(password)}@127.0.0.1:${opts.port}/${opts.databaseName}`;
335
+ }
336
+ //#endregion
337
+ //#region src/providers/host.ts
338
+ /** Fixed CI TCP port so install registration and postmaster listen on the same port. */
339
+ const EPHEMERAL_PORT = 55432;
340
+ const OWNED_POSTMASTER_READY_MS = 3e4;
341
+ const OWNED_POSTMASTER_POLL_MS = 200;
342
+ /**
343
+ * Resolve whether to start an owned ephemeral postmaster or use local pm2 install.
344
+ *
345
+ * - `CEDAR_PG_EPHEMERAL_HOST=1` → ephemeral
346
+ * - `CEDAR_PG_EPHEMERAL_HOST=0` → local (even when `CI=true`)
347
+ * - unset + `CI=true` → ephemeral
348
+ * - otherwise → local
349
+ */
350
+ function resolveEphemeralHostPolicy(env = process.env) {
351
+ const force = env.CEDAR_PG_EPHEMERAL_HOST;
352
+ if (force === "1") return "ephemeral";
353
+ if (force === "0") return "local";
354
+ if (env.CI === "true") return "ephemeral";
355
+ return "local";
356
+ }
357
+ /**
358
+ * Fixed CI recipe: shared `--port` on install + postmaster, `--no-pm2 --no-ui`,
359
+ * and `--ram` + `/dev/shm/cedar-pg-<uid>` on Linux when shm exists.
360
+ */
361
+ function ephemeralHostRecipe(ctx = {}) {
362
+ const platform = ctx.platform ?? process.platform;
363
+ const shmAvailable = ctx.shmAvailable ?? (platform === "linux" && existsSync("/dev/shm"));
364
+ const uid = ctx.uid ?? process.getuid?.() ?? 0;
365
+ const port = ctx.port ?? EPHEMERAL_PORT;
366
+ const useRam = platform === "linux" && shmAvailable;
367
+ const dataDir = useRam ? `/dev/shm/cedar-pg-${uid}` : join(ctx.tmpDir ?? tmpdir(), "cedar-pg-host");
368
+ return {
369
+ dataDir,
370
+ port,
371
+ installArgs: [
372
+ "install",
373
+ "--no-pm2",
374
+ "--no-ui",
375
+ "--port",
376
+ String(port),
377
+ "--data",
378
+ dataDir
379
+ ],
380
+ postmasterArgs: [
381
+ "postmaster",
382
+ ...useRam ? ["--ram"] : [],
383
+ "--port",
384
+ String(port),
385
+ "--socket-dir",
386
+ dataDir,
387
+ "--data",
388
+ dataDir
389
+ ]
390
+ };
391
+ }
392
+ /** Discovery from the recipe port — TCP readiness is the authority, not `autopg status`. */
393
+ function discoveryFromRecipe(bin, recipe) {
394
+ return {
395
+ port: recipe.port,
396
+ adminUrl: adminUrlFor(recipe.port),
397
+ bin
398
+ };
399
+ }
400
+ function sleep(ms) {
401
+ return new Promise((resolve) => setTimeout(resolve, ms));
402
+ }
403
+ function runInstall(bin, args) {
404
+ const install = spawnSync(bin, args, {
405
+ encoding: "utf8",
406
+ stdio: [
407
+ "ignore",
408
+ "pipe",
409
+ "pipe"
410
+ ]
411
+ });
412
+ if (install.status !== 0) throw new Error(`Failed to start autopg host (exit ${install.status}).\n${install.stderr || install.stdout || ""}\n${INSTALL_HINT}`);
413
+ }
414
+ /** True when something accepts TCP on 127.0.0.1:port (postmaster live, not just admin.json). */
415
+ function canConnect(port) {
416
+ return new Promise((resolve) => {
417
+ const socket = createConnection({
418
+ host: "127.0.0.1",
419
+ port
420
+ }, () => {
421
+ socket.end();
422
+ resolve(true);
423
+ });
424
+ socket.on("error", () => {
425
+ resolve(false);
426
+ });
427
+ });
428
+ }
429
+ /** Kill a detached owned postmaster (process group when possible) and drop our handle. */
430
+ function killOwnedChild(child) {
431
+ const pid = child.pid;
432
+ if (pid != null) try {
433
+ process.kill(-pid, "SIGTERM");
434
+ } catch {
435
+ try {
436
+ child.kill("SIGTERM");
437
+ } catch {}
438
+ }
439
+ child.unref();
440
+ }
441
+ /**
442
+ * Poll until TCP accepts on port, or throw on timeout / {@link WaitForOwnedPostmasterOptions.failure}.
443
+ * Exported for tests — the readiness seam for owned ephemeral start.
444
+ */
445
+ async function waitForOwnedPostmaster(opts) {
446
+ const probe = opts.canConnect ?? canConnect;
447
+ const now = opts.now ?? Date.now;
448
+ const pause = opts.sleep ?? sleep;
449
+ const readyMs = opts.readyMs ?? OWNED_POSTMASTER_READY_MS;
450
+ const pollMs = opts.pollMs ?? OWNED_POSTMASTER_POLL_MS;
451
+ const deadline = now() + readyMs;
452
+ let lastDetail = "TCP not accepting";
453
+ while (now() < deadline) {
454
+ const fail = opts.failure?.();
455
+ if (fail) throw fail;
456
+ if (await probe(opts.port)) return;
457
+ lastDetail = `127.0.0.1:${opts.port} not accepting connections`;
458
+ await pause(pollMs);
459
+ }
460
+ throw new Error(`Timed out waiting for autopg host after ${readyMs}ms.\n${lastDetail}\n${INSTALL_HINT}`);
461
+ }
462
+ /**
463
+ * Install + detached postmaster with fully ignored stdio so the CI job owns lifetime
464
+ * (caller exit must not close pipes under the daemon).
465
+ *
466
+ * Readiness is TCP accept on the recipe port — not `autopg status`, which succeeds
467
+ * after `install --no-pm2` before postmaster is listening. On success, discovery is
468
+ * recipe-derived (same port). On failure, the spawned child is killed.
469
+ */
470
+ async function startEphemeralHost(bin) {
471
+ const recipe = ephemeralHostRecipe();
472
+ mkdirSync(recipe.dataDir, { recursive: true });
473
+ runInstall(bin, recipe.installArgs);
474
+ const state = {
475
+ exit: null,
476
+ spawnError: null
477
+ };
478
+ const child = spawn(bin, recipe.postmasterArgs, {
479
+ detached: true,
480
+ stdio: "ignore"
481
+ });
482
+ child.on("error", (err) => {
483
+ state.spawnError = err;
484
+ });
485
+ child.on("exit", (code, signal) => {
486
+ state.exit = {
487
+ code,
488
+ signal
489
+ };
490
+ });
491
+ try {
492
+ await waitForOwnedPostmaster({
493
+ port: recipe.port,
494
+ failure: () => {
495
+ if (state.spawnError) return /* @__PURE__ */ new Error(`Failed to spawn autopg postmaster.\n${state.spawnError.message}\n${INSTALL_HINT}`);
496
+ if (state.exit) return /* @__PURE__ */ new Error(`autopg postmaster exited before ready (code=${state.exit.code}, signal=${state.exit.signal}).\nautopg is required. Install with:
497
+ curl -fsSL https://raw.githubusercontent.com/automagik-dev/autopg/main/install.sh | bash
498
+ Then ensure ~/.local/bin is on PATH, or set AUTOPG_BIN.`);
499
+ return null;
500
+ }
501
+ });
502
+ } catch (err) {
503
+ killOwnedChild(child);
504
+ throw err;
505
+ }
506
+ child.unref();
507
+ return discoveryFromRecipe(bin, recipe);
508
+ }
509
+ /**
510
+ * Ensure the host postmaster is up.
511
+ *
512
+ * - Attaches when `autopg status --json` shows a live host.
513
+ * - Otherwise starts from {@link resolveEphemeralHostPolicy}: local pm2 `install`,
514
+ * or opinionated ephemeral (`--no-pm2 --no-ui` + detached postmaster).
515
+ * - CI job owns ephemeral postmaster lifetime on success (no cedar-pg host dispose).
516
+ * - Failed ephemeral start kills the spawned child so ensure is atomic for the caller.
517
+ */
518
+ async function ensureHostRunning(bin = requireAutopgBin()) {
519
+ try {
520
+ return discoverHost(bin);
521
+ } catch {}
522
+ if (resolveEphemeralHostPolicy() === "ephemeral") return startEphemeralHost(bin);
523
+ runInstall(bin, ["install"]);
524
+ return discoverHost(bin);
525
+ }
526
+ //#endregion
527
+ //#region src/core/lifecycle.ts
528
+ function writeEnvFile(root, mode, databaseUrl) {
529
+ mkdirSync(leaseDir(root), {
530
+ recursive: true,
531
+ mode: 448
532
+ });
533
+ let body = `DATABASE_URL=${databaseUrl}\n`;
534
+ if (mode === "test") body += `TEST_DATABASE_URL=${databaseUrl}\n`;
535
+ writeFileSync(envPath(root, mode), body, { mode: 384 });
536
+ }
537
+ function urlFromLease(lease) {
538
+ return buildDatabaseUrl({
539
+ port: lease.port,
540
+ databaseName: lease.databaseName,
541
+ roleName: lease.roleName
542
+ });
543
+ }
544
+ /**
545
+ * DROP every DB owned by the lease role (TEMPLATE + clones), then forget lease.
546
+ * Provider owns ordering (clones before leased datname) on one admin connection.
547
+ */
548
+ async function dropThenForget(lease, adminUrl) {
549
+ const dropped = await dropDatabasesOwnedByRole({
550
+ adminUrl,
551
+ roleName: lease.roleName,
552
+ preferLast: lease.databaseName
553
+ });
554
+ forgetLease(lease);
555
+ return dropped;
556
+ }
557
+ /**
558
+ * Acquire a worktree-scoped database and return connection info.
559
+ *
560
+ * - `dev`: keep DB across restarts.
561
+ * - `test`: DROP when `dispose()` is awaited (callers / test runners own teardown).
562
+ */
563
+ async function acquire(options) {
564
+ const identity = resolveWorktreeIdentity(options.root);
565
+ const mode = options.mode;
566
+ const databaseName = buildDatabaseName(identity, mode);
567
+ const roleName = buildRoleName(databaseName);
568
+ const host = await ensureHostRunning();
569
+ await ensureDatabase({
570
+ adminUrl: host.adminUrl,
571
+ databaseName,
572
+ roleName
573
+ });
574
+ const databaseUrl = buildDatabaseUrl({
575
+ port: host.port,
576
+ databaseName,
577
+ roleName
578
+ });
579
+ writeLease({
580
+ schemaVersion: 1,
581
+ mode,
582
+ root: identity.root,
583
+ repoSlug: identity.repoSlug,
584
+ worktreeSlug: identity.worktreeSlug,
585
+ pathHash: identity.pathHash,
586
+ databaseName,
587
+ roleName,
588
+ port: host.port,
589
+ pid: process.pid,
590
+ createdAt: (/* @__PURE__ */ new Date()).toISOString()
591
+ });
592
+ writeEnvFile(identity.root, mode, databaseUrl);
593
+ if (options.setEnv !== false) applyDatabaseUrlEnv(databaseUrl, { mode });
594
+ const disposeFn = async () => {
595
+ await dispose({
596
+ root: identity.root,
597
+ mode
598
+ });
599
+ };
600
+ return {
601
+ databaseUrl,
602
+ adminUrl: host.adminUrl,
603
+ databaseName,
604
+ roleName,
605
+ repoSlug: identity.repoSlug,
606
+ worktreeSlug: identity.worktreeSlug,
607
+ pathHash: identity.pathHash,
608
+ root: identity.root,
609
+ mode,
610
+ port: host.port,
611
+ dispose: disposeFn
612
+ };
613
+ }
614
+ /**
615
+ * Resolve skip policy then acquire. Single entry for hosts (Cedar CLI, Jest, Vitest).
616
+ * On external-url skip, applies DATABASE_URL / TEST_DATABASE_URL when `setEnv` is not false.
617
+ */
618
+ async function acquireIfNeeded(options) {
619
+ const outcome = await runIfNeeded(options, () => acquire({
620
+ root: options.root,
621
+ mode: options.mode,
622
+ setEnv: options.setEnv
623
+ }));
624
+ if (outcome.status === "skipped") return outcome;
625
+ return {
626
+ status: "acquired",
627
+ ...outcome.value
628
+ };
629
+ }
630
+ /**
631
+ * Role-scoped suite teardown: DROP every database owned by the lease role
632
+ * (TEMPLATE + clones), then DROP ROLE and forget the lease. Unsets `IS_TEMPLATE`
633
+ * as needed. This is not per-clone cleanup — use `CloneResult.dropClone` for that.
634
+ * No-ops without a valid lease; never invents a DROP target beyond role ownership.
635
+ * If the host is unavailable, leaves the lease so dispose/gc can retry.
636
+ */
637
+ async function dispose(options = {}) {
638
+ const identity = resolveWorktreeIdentity(options.root);
639
+ const mode = options.mode ?? "test";
640
+ const lease = readLease(identity.root, mode);
641
+ if (!lease) return {
642
+ dropped: false,
643
+ reason: "no-lease"
644
+ };
645
+ let host;
646
+ try {
647
+ host = await ensureHostRunning();
648
+ } catch {
649
+ return {
650
+ dropped: false,
651
+ reason: "host-unavailable"
652
+ };
653
+ }
654
+ const droppedDatabases = await dropThenForget(lease, host.adminUrl);
655
+ return {
656
+ dropped: true,
657
+ databaseName: lease.databaseName,
658
+ droppedDatabases
659
+ };
660
+ }
661
+ /**
662
+ * Drop databases whose registered worktree root no longer exists on disk.
663
+ * Registry entries are removed only after a successful DROP (owned DBs + lease DB).
664
+ */
665
+ async function gc() {
666
+ const dropped = [];
667
+ const orphans = listRegistryLeases().filter(isOrphanLease);
668
+ if (orphans.length === 0) return { dropped };
669
+ let host;
670
+ try {
671
+ host = await ensureHostRunning();
672
+ } catch {
673
+ return { dropped };
674
+ }
675
+ for (const lease of orphans) try {
676
+ const names = await dropThenForget(lease, host.adminUrl);
677
+ dropped.push(...names);
678
+ } catch {}
679
+ return { dropped };
680
+ }
681
+ //#endregion
682
+ export { buildCloneDatabaseName as C, runIfNeeded as S, buildRoleName as T, setDatabaseIsTemplate as _, urlFromLease as a, isExternalDatabaseEscapeHatch as b, ROLE_PASSWORD_SCHEME as c, discoverHost as d, dropDatabase as f, rolePasswordFor as g, resolveAutopgBin as h, gc as i, buildDatabaseUrl as l, requireAutopgBin as m, acquireIfNeeded as n, ensureHostRunning as o, parseHostStatus as p, dispose as r, INSTALL_HINT as s, acquire as t, cloneDatabaseFromTemplate as u, applyDatabaseUrlEnv as v, buildDatabaseName as w, resolveAcquireSkip as x, isCedarPgManagedUrl as y };
683
+
684
+ //# sourceMappingURL=lifecycle-BOo6xBjD.mjs.map