pgserve 2.1.3 → 2.2.1

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 (235) hide show
  1. package/CHANGELOG.md +96 -0
  2. package/README.md +105 -1
  3. package/bin/autopg-wrapper.cjs +16 -0
  4. package/bin/pgserve-wrapper.cjs +32 -6
  5. package/bin/postgres-server.js +56 -0
  6. package/console/README.md +131 -0
  7. package/console/api.js +173 -0
  8. package/console/app.jsx +483 -0
  9. package/console/colors_and_type.css +227 -0
  10. package/console/components.jsx +167 -0
  11. package/console/console.css +1666 -0
  12. package/console/data.jsx +350 -0
  13. package/console/index.html +31 -0
  14. package/console/screens/databases.jsx +5 -0
  15. package/console/screens/health.jsx +5 -0
  16. package/console/screens/ingress.jsx +5 -0
  17. package/console/screens/optimizer.jsx +5 -0
  18. package/console/screens/rlm-sim.jsx +5 -0
  19. package/console/screens/rlm-trace.jsx +5 -0
  20. package/console/screens/security.jsx +5 -0
  21. package/console/screens/settings.jsx +611 -0
  22. package/console/screens/sql.jsx +5 -0
  23. package/console/screens/sync.jsx +5 -0
  24. package/console/screens/tables.jsx +5 -0
  25. package/console/tweaks-panel.jsx +425 -0
  26. package/package.json +14 -2
  27. package/scripts/postinstall.cjs +60 -0
  28. package/src/cli-config.cjs +310 -0
  29. package/src/cli-install.cjs +112 -11
  30. package/src/cli-restart.cjs +228 -0
  31. package/src/cli-ui.cjs +580 -0
  32. package/src/cluster.js +43 -38
  33. package/src/postgres.js +141 -19
  34. package/src/settings-loader.cjs +235 -0
  35. package/src/settings-migrate.cjs +212 -0
  36. package/src/settings-pg-args.cjs +146 -0
  37. package/src/settings-schema.cjs +422 -0
  38. package/src/settings-validator.cjs +416 -0
  39. package/src/settings-writer.cjs +288 -0
  40. package/src/upgrade/index.js +65 -0
  41. package/src/upgrade/runner.js +23 -0
  42. package/src/upgrade/steps/binary-cache-flush.js +67 -0
  43. package/src/upgrade/steps/consumer-signal.js +40 -0
  44. package/src/upgrade/steps/env-refresh.js +89 -0
  45. package/src/upgrade/steps/health-validate.js +53 -0
  46. package/src/upgrade/steps/plpgsql-resolve.js +66 -0
  47. package/src/upgrade/steps/port-reconcile.js +52 -0
  48. package/.claude/context/windows-debug.md +0 -119
  49. package/.genie/AGENTS.md +0 -15
  50. package/.genie/agents/README.md +0 -110
  51. package/.genie/agents/analyze.md +0 -176
  52. package/.genie/agents/forge.md +0 -290
  53. package/.genie/agents/garbage-cleaner.md +0 -324
  54. package/.genie/agents/garbage-collector.md +0 -596
  55. package/.genie/agents/github-issue-gc.md +0 -618
  56. package/.genie/agents/review.md +0 -380
  57. package/.genie/agents/semantic-analyzer/find-duplicates.md +0 -90
  58. package/.genie/agents/semantic-analyzer/find-orphans.md +0 -99
  59. package/.genie/agents/semantic-analyzer.md +0 -101
  60. package/.genie/agents/update.md +0 -182
  61. package/.genie/agents/wish.md +0 -357
  62. package/.genie/brainstorms/pgserve-v2/DESIGN.md +0 -174
  63. package/.genie/code/AGENTS.md +0 -694
  64. package/.genie/code/agents/audit/risk.md +0 -173
  65. package/.genie/code/agents/audit/security.md +0 -189
  66. package/.genie/code/agents/audit.md +0 -145
  67. package/.genie/code/agents/challenge.md +0 -230
  68. package/.genie/code/agents/change-reviewer.md +0 -295
  69. package/.genie/code/agents/code-garbage-collector.md +0 -425
  70. package/.genie/code/agents/code-quality.md +0 -410
  71. package/.genie/code/agents/commit-suggester.md +0 -255
  72. package/.genie/code/agents/commit.md +0 -124
  73. package/.genie/code/agents/consensus.md +0 -204
  74. package/.genie/code/agents/daily-standup.md +0 -722
  75. package/.genie/code/agents/docgen.md +0 -48
  76. package/.genie/code/agents/explore.md +0 -79
  77. package/.genie/code/agents/fix.md +0 -100
  78. package/.genie/code/agents/git/commit-advisory.md +0 -219
  79. package/.genie/code/agents/git/workflows/issue.md +0 -244
  80. package/.genie/code/agents/git/workflows/pr.md +0 -179
  81. package/.genie/code/agents/git/workflows/release.md +0 -460
  82. package/.genie/code/agents/git/workflows/report.md +0 -342
  83. package/.genie/code/agents/git.md +0 -432
  84. package/.genie/code/agents/implementor.md +0 -161
  85. package/.genie/code/agents/install.md +0 -515
  86. package/.genie/code/agents/issue-creator.md +0 -344
  87. package/.genie/code/agents/polish.md +0 -116
  88. package/.genie/code/agents/qa.md +0 -653
  89. package/.genie/code/agents/refactor.md +0 -294
  90. package/.genie/code/agents/release.md +0 -1129
  91. package/.genie/code/agents/roadmap.md +0 -885
  92. package/.genie/code/agents/tests.md +0 -557
  93. package/.genie/code/agents/tracer.md +0 -50
  94. package/.genie/code/agents/update/upstream-update.md +0 -85
  95. package/.genie/code/agents/update/versions/generic-update.md +0 -305
  96. package/.genie/code/agents/vibe.md +0 -1317
  97. package/.genie/code/spells/agent-configuration.md +0 -58
  98. package/.genie/code/spells/automated-rc-publishing.md +0 -106
  99. package/.genie/code/spells/branch-tracker-guidance.md +0 -28
  100. package/.genie/code/spells/debug.md +0 -320
  101. package/.genie/code/spells/emoji-naming-convention.md +0 -303
  102. package/.genie/code/spells/evidence-storage.md +0 -26
  103. package/.genie/code/spells/file-naming-rules.md +0 -35
  104. package/.genie/code/spells/forge-code-blueprints.md +0 -195
  105. package/.genie/code/spells/genie-integration.md +0 -153
  106. package/.genie/code/spells/publishing-protocol.md +0 -61
  107. package/.genie/code/spells/team-consultation-protocol.md +0 -284
  108. package/.genie/code/spells/tool-requirements.md +0 -20
  109. package/.genie/code/spells/triad-maintenance-protocol.md +0 -154
  110. package/.genie/code/teams/tech-council/council.md +0 -328
  111. package/.genie/code/teams/tech-council/jt.md +0 -352
  112. package/.genie/code/teams/tech-council/nayr.md +0 -305
  113. package/.genie/code/teams/tech-council/oettam.md +0 -375
  114. package/.genie/neurons/README.md +0 -193
  115. package/.genie/neurons/forge.md +0 -106
  116. package/.genie/neurons/genie.md +0 -63
  117. package/.genie/neurons/review.md +0 -106
  118. package/.genie/neurons/wish.md +0 -104
  119. package/.genie/product/README.md +0 -20
  120. package/.genie/product/cli-automation.md +0 -359
  121. package/.genie/product/environment.md +0 -60
  122. package/.genie/product/mission.md +0 -60
  123. package/.genie/product/roadmap.md +0 -44
  124. package/.genie/product/tech-stack.md +0 -34
  125. package/.genie/product/templates/context-template.md +0 -218
  126. package/.genie/product/templates/qa-done-report-template.md +0 -68
  127. package/.genie/product/templates/review-report-template.md +0 -89
  128. package/.genie/product/templates/wish-template.md +0 -120
  129. package/.genie/scripts/helpers/analyze-commit.js +0 -195
  130. package/.genie/scripts/helpers/bullet-counter.js +0 -194
  131. package/.genie/scripts/helpers/bullet-find.js +0 -289
  132. package/.genie/scripts/helpers/bullet-id.js +0 -244
  133. package/.genie/scripts/helpers/check-secrets.js +0 -237
  134. package/.genie/scripts/helpers/count-tokens.js +0 -200
  135. package/.genie/scripts/helpers/create-frontmatter.js +0 -456
  136. package/.genie/scripts/helpers/detect-markers.js +0 -293
  137. package/.genie/scripts/helpers/detect-todos.js +0 -267
  138. package/.genie/scripts/helpers/detect-unlabeled-blocks.js +0 -135
  139. package/.genie/scripts/helpers/embeddings.js +0 -344
  140. package/.genie/scripts/helpers/find-empty-sections.js +0 -158
  141. package/.genie/scripts/helpers/index.js +0 -319
  142. package/.genie/scripts/helpers/validate-frontmatter.js +0 -578
  143. package/.genie/scripts/helpers/validate-links.js +0 -207
  144. package/.genie/scripts/helpers/validate-paths.js +0 -373
  145. package/.genie/spells/README.md +0 -9
  146. package/.genie/spells/ace-protocol.md +0 -118
  147. package/.genie/spells/ask-one-at-a-time.md +0 -175
  148. package/.genie/spells/backup-analyzer.md +0 -542
  149. package/.genie/spells/blocker.md +0 -12
  150. package/.genie/spells/break-things-move-fast.md +0 -56
  151. package/.genie/spells/context-candidates.md +0 -72
  152. package/.genie/spells/context-critic.md +0 -51
  153. package/.genie/spells/defer-to-expertise.md +0 -278
  154. package/.genie/spells/delegate-dont-do.md +0 -292
  155. package/.genie/spells/error-investigation-protocol.md +0 -328
  156. package/.genie/spells/evidence-based-completion.md +0 -273
  157. package/.genie/spells/experiment.md +0 -65
  158. package/.genie/spells/file-creation-protocol.md +0 -229
  159. package/.genie/spells/forge-integration.md +0 -281
  160. package/.genie/spells/forge-orchestration.md +0 -514
  161. package/.genie/spells/gather-context.md +0 -18
  162. package/.genie/spells/global-health-check.md +0 -34
  163. package/.genie/spells/global-noop-roundtrip.md +0 -25
  164. package/.genie/spells/install-genie.md +0 -1232
  165. package/.genie/spells/install.md +0 -82
  166. package/.genie/spells/investigate-before-commit.md +0 -112
  167. package/.genie/spells/know-yourself.md +0 -288
  168. package/.genie/spells/learn.md +0 -828
  169. package/.genie/spells/mcp-diagnostic-protocol.md +0 -246
  170. package/.genie/spells/mcp-first.md +0 -124
  171. package/.genie/spells/multi-step-execution.md +0 -67
  172. package/.genie/spells/orchestration-boundary-protocol.md +0 -256
  173. package/.genie/spells/orchestrator-not-implementor.md +0 -189
  174. package/.genie/spells/prompt.md +0 -746
  175. package/.genie/spells/reflect.md +0 -404
  176. package/.genie/spells/routing-decision-matrix.md +0 -368
  177. package/.genie/spells/run-in-parallel.md +0 -12
  178. package/.genie/spells/session-state-updater-example.md +0 -196
  179. package/.genie/spells/session-state-updater.md +0 -220
  180. package/.genie/spells/track-long-running-tasks.md +0 -133
  181. package/.genie/spells/troubleshoot-infrastructure.md +0 -176
  182. package/.genie/spells/upgrade-genie.md +0 -415
  183. package/.genie/spells/url-presentation-protocol.md +0 -301
  184. package/.genie/spells/wish-initiation.md +0 -158
  185. package/.genie/spells/wish-issue-linkage.md +0 -410
  186. package/.genie/spells/wish-lifecycle.md +0 -100
  187. package/.genie/state/provider-status.json +0 -3
  188. package/.genie/state/version.json +0 -16
  189. package/.genie/wishes/canonical-pgserve-pm2-supervision/WISH.md +0 -290
  190. package/.genie/wishes/pgserve-v2/BRIEF-from-genie-pgserve.md +0 -99
  191. package/.genie/wishes/pgserve-v2/WISH.md +0 -442
  192. package/.genie/wishes/release-system-genie-pattern/WISH.md +0 -268
  193. package/.genie/wishes/release-system-genie-pattern/validation.md +0 -205
  194. package/.gitguardian.yaml +0 -29
  195. package/.gitguardianignore +0 -16
  196. package/.github/workflows/ci.yml +0 -122
  197. package/.github/workflows/release.yml +0 -289
  198. package/.github/workflows/version.yml +0 -228
  199. package/.husky/pre-commit +0 -2
  200. package/AGENTS.md +0 -433
  201. package/CLAUDE.md +0 -1
  202. package/Makefile +0 -285
  203. package/assets/icon.ico +0 -0
  204. package/bun.lock +0 -435
  205. package/bunfig.toml +0 -28
  206. package/ecosystem.config.cjs +0 -23
  207. package/eslint.config.js +0 -63
  208. package/examples/multi-tenant-demo.js +0 -104
  209. package/install.sh +0 -123
  210. package/knip.json +0 -9
  211. package/tests/audit.test.js +0 -189
  212. package/tests/backpressure.test.js +0 -167
  213. package/tests/benchmarks/runner.js +0 -1197
  214. package/tests/benchmarks/vector-generator.js +0 -368
  215. package/tests/cli-install.test.js +0 -322
  216. package/tests/control-db.test.js +0 -285
  217. package/tests/daemon-args.test.js +0 -86
  218. package/tests/daemon-control.test.js +0 -171
  219. package/tests/daemon-fingerprint-integration.test.js +0 -111
  220. package/tests/daemon-pr24-regression.test.js +0 -198
  221. package/tests/fingerprint.test.js +0 -263
  222. package/tests/fixtures/240-orphan-seed.sql +0 -30
  223. package/tests/multi-tenant.test.js +0 -374
  224. package/tests/orphan-cleanup.test.js +0 -390
  225. package/tests/pg-version-regex.test.js +0 -129
  226. package/tests/quick-bench.js +0 -135
  227. package/tests/router-handshake-retry.test.js +0 -119
  228. package/tests/router-handshake-watchdog.test.js +0 -110
  229. package/tests/sdk.test.js +0 -71
  230. package/tests/stale-postmaster-pid.test.js +0 -85
  231. package/tests/stress-test.js +0 -439
  232. package/tests/sync-perf-test.js +0 -150
  233. package/tests/tcp-listen.test.js +0 -368
  234. package/tests/tenancy.test.js +0 -403
  235. package/tests/wrapper-supervision.test.js +0 -107
@@ -1,285 +0,0 @@
1
- /**
2
- * Tests for src/control-db.js — pgserve_meta schema + accessors.
3
- *
4
- * Boots an ephemeral pgserve router (memory mode), connects via node-pg
5
- * to the default `postgres` database, and exercises every exported function.
6
- */
7
-
8
- import { test, expect, beforeAll, afterAll } from 'bun:test';
9
- import fs from 'fs';
10
- import pg from 'pg';
11
- import { startMultiTenantServer } from '../src/index.js';
12
- import {
13
- ensureMetaSchema,
14
- recordDbCreated,
15
- touchLastConnection,
16
- markPersist,
17
- forEachReapable,
18
- deleteMetaRow,
19
- addAllowedToken,
20
- revokeAllowedToken,
21
- verifyToken,
22
- findRowByFingerprint,
23
- } from '../src/control-db.js';
24
-
25
- const { Client } = pg;
26
-
27
- const TEST_DATA_DIR = './test-data-control-db';
28
- const PORT = 15561;
29
-
30
- let router;
31
- let client;
32
-
33
- function cleanupDataDir() {
34
- if (fs.existsSync(TEST_DATA_DIR)) {
35
- fs.rmSync(TEST_DATA_DIR, { recursive: true, force: true });
36
- }
37
- }
38
-
39
- beforeAll(async () => {
40
- cleanupDataDir();
41
- router = await startMultiTenantServer({
42
- port: PORT,
43
- baseDir: TEST_DATA_DIR,
44
- logLevel: 'warn',
45
- });
46
-
47
- client = new Client({
48
- host: '127.0.0.1',
49
- port: PORT,
50
- database: 'postgres',
51
- user: 'postgres',
52
- password: 'postgres',
53
- });
54
- await client.connect();
55
- await client.query('DROP TABLE IF EXISTS pgserve_meta');
56
- });
57
-
58
- afterAll(async () => {
59
- try { await client.end(); } catch { /* noop */ }
60
- try { await router.stop(); } catch { /* noop */ }
61
- cleanupDataDir();
62
- });
63
-
64
- test('ensureMetaSchema creates table on first call', async () => {
65
- await ensureMetaSchema(client);
66
- const r = await client.query(`
67
- SELECT column_name FROM information_schema.columns
68
- WHERE table_name = 'pgserve_meta'
69
- ORDER BY ordinal_position
70
- `);
71
- const columns = r.rows.map(row => row.column_name);
72
- expect(columns).toEqual([
73
- 'database_name',
74
- 'fingerprint',
75
- 'peer_uid',
76
- 'package_realpath',
77
- 'created_at',
78
- 'last_connection_at',
79
- 'liveness_pid',
80
- 'persist',
81
- 'allowed_tokens',
82
- ]);
83
- });
84
-
85
- test('ensureMetaSchema is idempotent', async () => {
86
- await ensureMetaSchema(client);
87
- await ensureMetaSchema(client);
88
- // No throw — schema unchanged.
89
- const r = await client.query(`SELECT count(*)::int AS n FROM pgserve_meta`);
90
- expect(r.rows[0].n).toBe(0);
91
- });
92
-
93
- test('recordDbCreated inserts a row + select round-trip', async () => {
94
- await client.query('TRUNCATE pgserve_meta');
95
- await recordDbCreated(client, {
96
- databaseName: 'app_demo_abc123def456',
97
- fingerprint: 'abc123def456',
98
- peerUid: 1000,
99
- packageRealpath: '/home/me/proj/package.json',
100
- livenessPid: 4242,
101
- persist: false,
102
- });
103
- const r = await client.query(`SELECT * FROM pgserve_meta WHERE database_name = $1`, [
104
- 'app_demo_abc123def456',
105
- ]);
106
- expect(r.rows.length).toBe(1);
107
- const row = r.rows[0];
108
- expect(row.fingerprint).toBe('abc123def456');
109
- expect(row.peer_uid).toBe(1000);
110
- expect(row.package_realpath).toBe('/home/me/proj/package.json');
111
- expect(row.liveness_pid).toBe(4242);
112
- expect(row.persist).toBe(false);
113
- expect(row.created_at).toBeInstanceOf(Date);
114
- expect(row.last_connection_at).toBeInstanceOf(Date);
115
- });
116
-
117
- test('recordDbCreated upserts on conflict (database_name PK)', async () => {
118
- await client.query('TRUNCATE pgserve_meta');
119
- await recordDbCreated(client, {
120
- databaseName: 'app_demo_abc123def456',
121
- fingerprint: 'abc123def456',
122
- peerUid: 1000,
123
- packageRealpath: '/home/me/proj/package.json',
124
- livenessPid: 4242,
125
- });
126
- // Re-insert with new peerUid + livenessPid → must upsert.
127
- await recordDbCreated(client, {
128
- databaseName: 'app_demo_abc123def456',
129
- fingerprint: 'abc123def456',
130
- peerUid: 1001,
131
- packageRealpath: '/home/me/proj/package.json',
132
- livenessPid: 9999,
133
- persist: true,
134
- });
135
- const r = await client.query(`SELECT peer_uid, liveness_pid, persist FROM pgserve_meta`);
136
- expect(r.rows.length).toBe(1);
137
- expect(r.rows[0].peer_uid).toBe(1001);
138
- expect(r.rows[0].liveness_pid).toBe(9999);
139
- expect(r.rows[0].persist).toBe(true);
140
- });
141
-
142
- test('touchLastConnection bumps last_connection_at and liveness_pid', async () => {
143
- await client.query('TRUNCATE pgserve_meta');
144
- await recordDbCreated(client, {
145
- databaseName: 'app_x_111111111111',
146
- fingerprint: '111111111111',
147
- peerUid: 1000,
148
- livenessPid: 100,
149
- });
150
- const before = await client.query(
151
- `SELECT last_connection_at, liveness_pid FROM pgserve_meta WHERE database_name = $1`,
152
- ['app_x_111111111111'],
153
- );
154
- // Sleep briefly so now() advances visibly.
155
- await new Promise(r => setTimeout(r, 50));
156
-
157
- await touchLastConnection(client, {
158
- databaseName: 'app_x_111111111111',
159
- livenessPid: 200,
160
- });
161
- const after = await client.query(
162
- `SELECT last_connection_at, liveness_pid FROM pgserve_meta WHERE database_name = $1`,
163
- ['app_x_111111111111'],
164
- );
165
- expect(after.rows[0].liveness_pid).toBe(200);
166
- expect(after.rows[0].last_connection_at.getTime()).toBeGreaterThan(
167
- before.rows[0].last_connection_at.getTime(),
168
- );
169
- });
170
-
171
- test('markPersist toggles persist flag', async () => {
172
- await client.query('TRUNCATE pgserve_meta');
173
- await recordDbCreated(client, {
174
- databaseName: 'app_p_222222222222',
175
- fingerprint: '222222222222',
176
- peerUid: 1000,
177
- });
178
- await markPersist(client, 'app_p_222222222222', true);
179
- let r = await client.query(`SELECT persist FROM pgserve_meta WHERE database_name = $1`, [
180
- 'app_p_222222222222',
181
- ]);
182
- expect(r.rows[0].persist).toBe(true);
183
-
184
- await markPersist(client, 'app_p_222222222222', false);
185
- r = await client.query(`SELECT persist FROM pgserve_meta WHERE database_name = $1`, [
186
- 'app_p_222222222222',
187
- ]);
188
- expect(r.rows[0].persist).toBe(false);
189
- });
190
-
191
- test('forEachReapable yields only persist=false rows in last_connection_at order', async () => {
192
- await client.query('TRUNCATE pgserve_meta');
193
- // Older row first, newer row second; persistent row separately.
194
- await client.query(
195
- `INSERT INTO pgserve_meta (database_name, fingerprint, peer_uid, last_connection_at, persist)
196
- VALUES
197
- ('app_a_aaaaaaaaaaaa', 'aaaaaaaaaaaa', 1000, now() - interval '2 hours', false),
198
- ('app_b_bbbbbbbbbbbb', 'bbbbbbbbbbbb', 1000, now() - interval '1 hour', false),
199
- ('app_c_cccccccccccc', 'cccccccccccc', 1000, now(), true)`,
200
- );
201
-
202
- const seen = [];
203
- for await (const row of forEachReapable(client, { now: new Date() })) {
204
- seen.push(row.databaseName);
205
- }
206
- expect(seen).toEqual(['app_a_aaaaaaaaaaaa', 'app_b_bbbbbbbbbbbb']);
207
- });
208
-
209
- test('deleteMetaRow removes the row', async () => {
210
- await client.query('TRUNCATE pgserve_meta');
211
- await recordDbCreated(client, {
212
- databaseName: 'app_del_333333333333',
213
- fingerprint: '333333333333',
214
- peerUid: 1000,
215
- });
216
- await deleteMetaRow(client, 'app_del_333333333333');
217
- const r = await client.query(`SELECT count(*)::int AS n FROM pgserve_meta`);
218
- expect(r.rows[0].n).toBe(0);
219
- });
220
-
221
- test('recordDbCreated rejects bad input', async () => {
222
- await expect(recordDbCreated(client, { fingerprint: 'x', peerUid: 1 })).rejects.toThrow(
223
- /databaseName required/,
224
- );
225
- await expect(recordDbCreated(client, { databaseName: 'd', peerUid: 1 })).rejects.toThrow(
226
- /fingerprint required/,
227
- );
228
- await expect(
229
- recordDbCreated(client, { databaseName: 'd', fingerprint: 'f', peerUid: 'nope' }),
230
- ).rejects.toThrow(/peerUid must be number/);
231
- });
232
-
233
- test('addAllowedToken refuses unknown fingerprint', async () => {
234
- await client.query('TRUNCATE pgserve_meta');
235
- await expect(
236
- addAllowedToken(client, { fingerprint: 'deadbeef0000', tokenId: 'tk1', tokenHash: 'h1' }),
237
- ).rejects.toThrow(/no pgserve_meta row/);
238
- });
239
-
240
- test('addAllowedToken appends, verifyToken finds it, revokeAllowedToken removes it', async () => {
241
- await client.query('TRUNCATE pgserve_meta');
242
- await recordDbCreated(client, {
243
- databaseName: 'app_demo_4444aabbccdd',
244
- fingerprint: '4444aabbccdd',
245
- peerUid: 1000,
246
- });
247
- await addAllowedToken(client, {
248
- fingerprint: '4444aabbccdd',
249
- tokenId: 'aaaa1111',
250
- tokenHash: 'hash-1',
251
- });
252
- await addAllowedToken(client, {
253
- fingerprint: '4444aabbccdd',
254
- tokenId: 'bbbb2222',
255
- tokenHash: 'hash-2',
256
- });
257
-
258
- const row = await findRowByFingerprint(client, '4444aabbccdd');
259
- expect(row).not.toBeNull();
260
- expect(row.allowedTokens.length).toBe(2);
261
- expect(row.allowedTokens.map(t => t.id).sort()).toEqual(['aaaa1111', 'bbbb2222']);
262
-
263
- const ok = await verifyToken(client, { fingerprint: '4444aabbccdd', tokenHash: 'hash-2' });
264
- expect(ok).toEqual({ tokenId: 'bbbb2222', databaseName: 'app_demo_4444aabbccdd' });
265
-
266
- const miss = await verifyToken(client, { fingerprint: '4444aabbccdd', tokenHash: 'no-such' });
267
- expect(miss).toBeNull();
268
-
269
- const affected = await revokeAllowedToken(client, 'aaaa1111');
270
- expect(affected).toBe(1);
271
-
272
- const after = await findRowByFingerprint(client, '4444aabbccdd');
273
- expect(after.allowedTokens.map(t => t.id)).toEqual(['bbbb2222']);
274
- });
275
-
276
- test('revokeAllowedToken returns 0 for unknown id', async () => {
277
- await client.query('TRUNCATE pgserve_meta');
278
- await recordDbCreated(client, {
279
- databaseName: 'app_x_5555aabbccdd',
280
- fingerprint: '5555aabbccdd',
281
- peerUid: 1000,
282
- });
283
- const affected = await revokeAllowedToken(client, 'nonexistent');
284
- expect(affected).toBe(0);
285
- });
@@ -1,86 +0,0 @@
1
- /**
2
- * Integration tests for `pgserve daemon` argv parsing.
3
- *
4
- * `parseDaemonArgs` lives inside `bin/postgres-server.js` (the script
5
- * entry point) and isn't exported, so we exercise it via subprocess
6
- * invocations of the wrapper. Each test runs in <100ms — they only ask
7
- * the daemon to print help or reject an invalid argument; no real
8
- * postgres backend is started.
9
- *
10
- * Background: every recent CLI-flag mismatch between callers and
11
- * `pgserve daemon` exited the daemon child with code 1 immediately,
12
- * surfacing upstream as the unhelpful "pgserve v2 daemon exited before
13
- * binding …" error. These tests pin the daemon's accepted flag set
14
- * explicitly so the next mismatch fails CI here, not at runtime.
15
- */
16
-
17
- import { spawnSync } from 'node:child_process';
18
- import { join } from 'node:path';
19
- import { fileURLToPath } from 'node:url';
20
- import { describe, expect, test } from 'bun:test';
21
-
22
- const REPO_ROOT = join(fileURLToPath(import.meta.url), '..', '..');
23
- const PGSERVE_BIN = join(REPO_ROOT, 'bin', 'pgserve-wrapper.cjs');
24
-
25
- function runDaemon(args, timeoutMs = 3000) {
26
- return spawnSync('node', [PGSERVE_BIN, 'daemon', ...args], {
27
- encoding: 'utf-8',
28
- timeout: timeoutMs,
29
- stdio: ['ignore', 'pipe', 'pipe'],
30
- });
31
- }
32
-
33
- describe('pgserve daemon — argv parser', () => {
34
- test('--help lists every flag the daemon accepts', () => {
35
- const result = runDaemon(['--help']);
36
- expect(result.status).toBe(0);
37
- const help = result.stdout;
38
- // Every flag the parser accepts must appear in --help so callers
39
- // (and the next operator running `pgserve daemon --help`) discover them.
40
- expect(help).toContain('--data');
41
- expect(help).toContain('--ram');
42
- expect(help).toContain('--log');
43
- expect(help).toContain('--no-provision');
44
- expect(help).toContain('--listen');
45
- expect(help).toContain('--pgvector');
46
- expect(help).toContain('--max-connections');
47
- expect(help).toContain('--help');
48
- });
49
-
50
- test('--max-connections accepts a positive integer (no "Unknown option" error)', () => {
51
- // Use a bogus --data path so the daemon never actually starts postgres
52
- // — the parser runs, accepts --max-connections, then PgserveDaemon
53
- // tries to start and fails on the missing/invalid data dir. We only
54
- // care that the parser doesn't reject the flag.
55
- const result = runDaemon(['--data', '/nonexistent/pgserve-test-dir', '--max-connections', '5000', '--log', 'error']);
56
- // The daemon may exit non-zero because the data dir is invalid, but
57
- // it MUST NOT exit with "Unknown daemon option" — that's the
58
- // pre-fix behavior we're guarding against.
59
- const stderr = result.stderr ?? '';
60
- expect(stderr).not.toContain('Unknown daemon option: --max-connections');
61
- });
62
-
63
- test('--max-connections rejects non-numeric values with a clear error', () => {
64
- const result = runDaemon(['--max-connections', 'abc']);
65
- expect(result.status).toBe(1);
66
- expect(result.stderr).toContain('--max-connections: expected a positive integer');
67
- });
68
-
69
- test('--max-connections rejects zero / negative values', () => {
70
- const zero = runDaemon(['--max-connections', '0']);
71
- expect(zero.status).toBe(1);
72
- expect(zero.stderr).toContain('--max-connections: expected a positive integer');
73
-
74
- const negative = runDaemon(['--max-connections', '-50']);
75
- expect(negative.status).toBe(1);
76
- expect(negative.stderr).toContain('--max-connections: expected a positive integer');
77
- });
78
-
79
- test('unknown flags still exit 1 with the documented "Unknown daemon option" error', () => {
80
- // Sanity: the parser hasn't become permissive. Genuinely unknown
81
- // flags must still error out so callers learn about the mismatch.
82
- const result = runDaemon(['--definitely-not-a-flag']);
83
- expect(result.status).toBe(1);
84
- expect(result.stderr).toContain('Unknown daemon option: --definitely-not-a-flag');
85
- });
86
- });
@@ -1,171 +0,0 @@
1
- import { describe, expect, test } from 'bun:test';
2
- import fs from 'fs';
3
- import net from 'net';
4
- import path from 'path';
5
-
6
- import {
7
- PgserveDaemon,
8
- resolveControlSocketPath,
9
- resolvePidLockPath,
10
- } from '../src/daemon.js';
11
- import { createLogger } from '../src/logger.js';
12
-
13
- const SSL_REQUEST_CODE = 80877103;
14
- const PROTOCOL_VERSION_3 = 196608;
15
-
16
- function silentLogger() {
17
- return createLogger({ level: process.env.PGSERVE_TEST_LOG || 'warn' });
18
- }
19
-
20
- function makeIsolated(tag) {
21
- const dir = path.join('/tmp', `pgs-${tag}-${process.pid}-${Date.now()}`);
22
- fs.mkdirSync(dir, { recursive: true, mode: 0o700 });
23
- return dir;
24
- }
25
-
26
- function freeTcpPort() {
27
- return new Promise((resolve, reject) => {
28
- const srv = net.createServer();
29
- srv.unref();
30
- srv.on('error', reject);
31
- srv.listen(0, '127.0.0.1', () => {
32
- const { port } = srv.address();
33
- srv.close(() => resolve(port));
34
- });
35
- });
36
- }
37
-
38
- function sslRequest() {
39
- const buf = Buffer.alloc(8);
40
- buf.writeUInt32BE(8, 0);
41
- buf.writeUInt32BE(SSL_REQUEST_CODE, 4);
42
- return buf;
43
- }
44
-
45
- function startupMessage({ user = 'postgres', database = 'postgres' } = {}) {
46
- const params = Buffer.from(`user\0${user}\0database\0${database}\0client_encoding\0UTF8\0\0`);
47
- const buf = Buffer.alloc(8 + params.length);
48
- buf.writeUInt32BE(buf.length, 0);
49
- buf.writeUInt32BE(PROTOCOL_VERSION_3, 4);
50
- params.copy(buf, 8);
51
- return buf;
52
- }
53
-
54
- function passwordMessage(password = 'postgres') {
55
- const body = Buffer.from(`${password}\0`);
56
- const buf = Buffer.alloc(1 + 4 + body.length);
57
- buf.write('p', 0);
58
- buf.writeUInt32BE(4 + body.length, 1);
59
- body.copy(buf, 5);
60
- return buf;
61
- }
62
-
63
- async function connectWithCoalescedStartup(socketPath) {
64
- return new Promise((resolve, reject) => {
65
- const socket = net.createConnection(socketPath);
66
- let buffer = Buffer.alloc(0);
67
- let sawSslReject = false;
68
- let sawAuthOk = false;
69
-
70
- const timer = setTimeout(() => {
71
- socket.destroy();
72
- reject(new Error('timed out waiting for ReadyForQuery after coalesced startup'));
73
- }, 5000);
74
- timer.unref();
75
-
76
- const done = (err, result) => {
77
- clearTimeout(timer);
78
- socket.destroy();
79
- if (err) reject(err);
80
- else resolve(result);
81
- };
82
-
83
- const pump = () => {
84
- if (!sawSslReject) {
85
- if (buffer.length < 1) return;
86
- if (buffer[0] !== 78) {
87
- done(new Error(`expected SSL reject byte N, got ${buffer[0]}`));
88
- return;
89
- }
90
- sawSslReject = true;
91
- buffer = buffer.subarray(1);
92
- }
93
-
94
- while (buffer.length >= 5) {
95
- const type = String.fromCharCode(buffer[0]);
96
- const length = buffer.readUInt32BE(1);
97
- if (buffer.length < 1 + length) return;
98
-
99
- const payload = buffer.subarray(5, 1 + length);
100
- buffer = buffer.subarray(1 + length);
101
-
102
- if (type === 'R') {
103
- const authCode = payload.readUInt32BE(0);
104
- if (authCode === 3) socket.write(passwordMessage());
105
- if (authCode === 0) sawAuthOk = true;
106
- } else if (type === 'E') {
107
- done(new Error(`postgres error response: ${payload.toString('utf8')}`));
108
- return;
109
- } else if (type === 'Z') {
110
- done(null, { sawSslReject, sawAuthOk });
111
- return;
112
- }
113
- }
114
- };
115
-
116
- socket.on('connect', () => {
117
- socket.write(Buffer.concat([sslRequest(), startupMessage()]));
118
- });
119
- socket.on('data', (chunk) => {
120
- buffer = Buffer.concat([buffer, chunk]);
121
- pump();
122
- });
123
- socket.on('error', done);
124
- });
125
- }
126
-
127
- describe('daemon Unix control protocol', () => {
128
- test('processes startup already buffered behind SSLRequest', async () => {
129
- const dir = makeIsolated('coalesced');
130
- const daemon = new PgserveDaemon({
131
- controlSocketDir: dir,
132
- controlSocketPath: resolveControlSocketPath(dir),
133
- pidLockPath: resolvePidLockPath(dir),
134
- pgPort: await freeTcpPort(),
135
- logger: silentLogger(),
136
- });
137
-
138
- await daemon.start();
139
- try {
140
- const result = await connectWithCoalescedStartup(resolveControlSocketPath(dir));
141
- expect(result).toEqual({ sawSslReject: true, sawAuthOk: true });
142
- } finally {
143
- await daemon.stop();
144
- fs.rmSync(dir, { recursive: true, force: true });
145
- }
146
- });
147
-
148
- test('processes startup after the admin client idles out', async () => {
149
- const dir = makeIsolated('admin-idle');
150
- const daemon = new PgserveDaemon({
151
- controlSocketDir: dir,
152
- controlSocketPath: resolveControlSocketPath(dir),
153
- pidLockPath: resolvePidLockPath(dir),
154
- pgPort: await freeTcpPort(),
155
- adminIdleTimeout: 1,
156
- adminLookupTimeoutMs: 1000,
157
- logger: silentLogger(),
158
- });
159
-
160
- await daemon.start();
161
- try {
162
- await connectWithCoalescedStartup(resolveControlSocketPath(dir));
163
- await Bun.sleep(1500);
164
- const result = await connectWithCoalescedStartup(resolveControlSocketPath(dir));
165
- expect(result).toEqual({ sawSslReject: true, sawAuthOk: true });
166
- } finally {
167
- await daemon.stop();
168
- fs.rmSync(dir, { recursive: true, force: true });
169
- }
170
- });
171
- });
@@ -1,111 +0,0 @@
1
- /**
2
- * Daemon × fingerprint integration test (Group 3, deliverable 2).
3
- *
4
- * Verifies that PgserveDaemon.handleSocketOpen calls handleControlAccept on
5
- * every accept, producing a `connection_routed` audit entry whose fingerprint
6
- * is the documented 12-hex blob.
7
- *
8
- * Boots a real daemon (with isolated controlSocketDir + auditLogFile), dials
9
- * the control socket via Bun.connect, and tails the audit log.
10
- */
11
-
12
- import { describe, test, expect } from 'bun:test';
13
- import fs from 'fs';
14
- import os from 'os';
15
- import path from 'path';
16
-
17
- import {
18
- PgserveDaemon,
19
- resolveControlSocketPath,
20
- resolvePidLockPath,
21
- } from '../src/daemon.js';
22
- import { createLogger } from '../src/logger.js';
23
- import { AUDIT_EVENTS, configureAudit } from '../src/audit.js';
24
-
25
- function silentLogger() {
26
- return createLogger({ level: 'warn' });
27
- }
28
-
29
- function makeIsolated(tag) {
30
- return fs.mkdtempSync(path.join('/tmp', `pgs-fp-${tag}-`));
31
- }
32
-
33
- function readAuditLines(logFile) {
34
- if (!fs.existsSync(logFile)) return [];
35
- return fs.readFileSync(logFile, 'utf8')
36
- .split('\n')
37
- .filter(Boolean)
38
- .map((l) => JSON.parse(l));
39
- }
40
-
41
- describe('Group 3 — daemon emits connection_routed on accept', () => {
42
- test('handleSocketOpen derives fingerprint and audits connection_routed', async () => {
43
- const dir = makeIsolated('routed');
44
- const auditLogFile = path.join(dir, 'audit.log');
45
-
46
- const daemon = new PgserveDaemon({
47
- controlSocketDir: dir,
48
- controlSocketPath: resolveControlSocketPath(dir),
49
- pidLockPath: resolvePidLockPath(dir),
50
- pgPort: 16100,
51
- auditLogFile,
52
- auditTarget: 'file',
53
- logger: silentLogger(),
54
- _fingerprintAcceptOpts: () => ({
55
- cwdOverride: dir,
56
- cmdlineOverride: [process.execPath, import.meta.url],
57
- }),
58
- });
59
- await daemon.start();
60
-
61
- try {
62
- // Dial the control socket. We don't need to push a real PG startup
63
- // message — the accept hook fires the moment the connection opens,
64
- // before any handshake bytes are needed.
65
- const acceptedFingerprint = await new Promise((resolve, reject) => {
66
- const timer = setTimeout(() => reject(new Error('timeout waiting for accept')), 2000);
67
- daemon.once('accept', ({ fingerprint }) => {
68
- clearTimeout(timer);
69
- resolve(fingerprint);
70
- });
71
- Bun.connect({
72
- unix: daemon.controlSocketPath,
73
- socket: {
74
- open(s) { s.end(); },
75
- data() {},
76
- close() {},
77
- error(_s, err) { clearTimeout(timer); reject(err); },
78
- },
79
- }).catch((err) => { clearTimeout(timer); reject(err); });
80
- });
81
-
82
- expect(acceptedFingerprint).toBeDefined();
83
- expect(acceptedFingerprint.fingerprint).toMatch(/^[0-9a-f]{12}$/);
84
-
85
- // Allow the audit appendFileSync to flush. Poll briefly.
86
- const deadline = Date.now() + 1000;
87
- let entries = [];
88
- while (Date.now() < deadline) {
89
- entries = readAuditLines(auditLogFile);
90
- if (entries.length > 0) break;
91
- await new Promise((r) => setTimeout(r, 25));
92
- }
93
- expect(entries.length).toBeGreaterThan(0);
94
- const routed = entries.find((e) => e.event === AUDIT_EVENTS.CONNECTION_ROUTED);
95
- expect(routed).toBeDefined();
96
- expect(routed.fingerprint).toMatch(/^[0-9a-f]{12}$/);
97
- expect(routed.fingerprint).toBe(acceptedFingerprint.fingerprint);
98
- expect(routed.peer_uid).toBe(process.getuid());
99
- expect(typeof routed.peer_pid).toBe('number');
100
- expect(['package', 'script']).toContain(routed.mode);
101
- } finally {
102
- await daemon.stop();
103
- // Reset audit module's mutable defaults so other tests aren't affected.
104
- configureAudit({
105
- logFile: path.join(os.homedir(), '.pgserve', 'audit.log'),
106
- target: process.env.PGSERVE_AUDIT_TARGET || 'file',
107
- });
108
- fs.rmSync(dir, { recursive: true, force: true });
109
- }
110
- });
111
- });