pgserve 2.1.3 → 2.2.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 (228) hide show
  1. package/CHANGELOG.md +86 -0
  2. package/README.md +105 -1
  3. package/bin/autopg-wrapper.cjs +16 -0
  4. package/bin/pgserve-wrapper.cjs +31 -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 +11 -1
  27. package/src/cli-config.cjs +310 -0
  28. package/src/cli-install.cjs +98 -11
  29. package/src/cli-restart.cjs +228 -0
  30. package/src/cli-ui.cjs +580 -0
  31. package/src/cluster.js +43 -38
  32. package/src/postgres.js +141 -19
  33. package/src/settings-loader.cjs +235 -0
  34. package/src/settings-migrate.cjs +212 -0
  35. package/src/settings-pg-args.cjs +146 -0
  36. package/src/settings-schema.cjs +422 -0
  37. package/src/settings-validator.cjs +416 -0
  38. package/src/settings-writer.cjs +288 -0
  39. package/.claude/context/windows-debug.md +0 -119
  40. package/.genie/AGENTS.md +0 -15
  41. package/.genie/agents/README.md +0 -110
  42. package/.genie/agents/analyze.md +0 -176
  43. package/.genie/agents/forge.md +0 -290
  44. package/.genie/agents/garbage-cleaner.md +0 -324
  45. package/.genie/agents/garbage-collector.md +0 -596
  46. package/.genie/agents/github-issue-gc.md +0 -618
  47. package/.genie/agents/review.md +0 -380
  48. package/.genie/agents/semantic-analyzer/find-duplicates.md +0 -90
  49. package/.genie/agents/semantic-analyzer/find-orphans.md +0 -99
  50. package/.genie/agents/semantic-analyzer.md +0 -101
  51. package/.genie/agents/update.md +0 -182
  52. package/.genie/agents/wish.md +0 -357
  53. package/.genie/brainstorms/pgserve-v2/DESIGN.md +0 -174
  54. package/.genie/code/AGENTS.md +0 -694
  55. package/.genie/code/agents/audit/risk.md +0 -173
  56. package/.genie/code/agents/audit/security.md +0 -189
  57. package/.genie/code/agents/audit.md +0 -145
  58. package/.genie/code/agents/challenge.md +0 -230
  59. package/.genie/code/agents/change-reviewer.md +0 -295
  60. package/.genie/code/agents/code-garbage-collector.md +0 -425
  61. package/.genie/code/agents/code-quality.md +0 -410
  62. package/.genie/code/agents/commit-suggester.md +0 -255
  63. package/.genie/code/agents/commit.md +0 -124
  64. package/.genie/code/agents/consensus.md +0 -204
  65. package/.genie/code/agents/daily-standup.md +0 -722
  66. package/.genie/code/agents/docgen.md +0 -48
  67. package/.genie/code/agents/explore.md +0 -79
  68. package/.genie/code/agents/fix.md +0 -100
  69. package/.genie/code/agents/git/commit-advisory.md +0 -219
  70. package/.genie/code/agents/git/workflows/issue.md +0 -244
  71. package/.genie/code/agents/git/workflows/pr.md +0 -179
  72. package/.genie/code/agents/git/workflows/release.md +0 -460
  73. package/.genie/code/agents/git/workflows/report.md +0 -342
  74. package/.genie/code/agents/git.md +0 -432
  75. package/.genie/code/agents/implementor.md +0 -161
  76. package/.genie/code/agents/install.md +0 -515
  77. package/.genie/code/agents/issue-creator.md +0 -344
  78. package/.genie/code/agents/polish.md +0 -116
  79. package/.genie/code/agents/qa.md +0 -653
  80. package/.genie/code/agents/refactor.md +0 -294
  81. package/.genie/code/agents/release.md +0 -1129
  82. package/.genie/code/agents/roadmap.md +0 -885
  83. package/.genie/code/agents/tests.md +0 -557
  84. package/.genie/code/agents/tracer.md +0 -50
  85. package/.genie/code/agents/update/upstream-update.md +0 -85
  86. package/.genie/code/agents/update/versions/generic-update.md +0 -305
  87. package/.genie/code/agents/vibe.md +0 -1317
  88. package/.genie/code/spells/agent-configuration.md +0 -58
  89. package/.genie/code/spells/automated-rc-publishing.md +0 -106
  90. package/.genie/code/spells/branch-tracker-guidance.md +0 -28
  91. package/.genie/code/spells/debug.md +0 -320
  92. package/.genie/code/spells/emoji-naming-convention.md +0 -303
  93. package/.genie/code/spells/evidence-storage.md +0 -26
  94. package/.genie/code/spells/file-naming-rules.md +0 -35
  95. package/.genie/code/spells/forge-code-blueprints.md +0 -195
  96. package/.genie/code/spells/genie-integration.md +0 -153
  97. package/.genie/code/spells/publishing-protocol.md +0 -61
  98. package/.genie/code/spells/team-consultation-protocol.md +0 -284
  99. package/.genie/code/spells/tool-requirements.md +0 -20
  100. package/.genie/code/spells/triad-maintenance-protocol.md +0 -154
  101. package/.genie/code/teams/tech-council/council.md +0 -328
  102. package/.genie/code/teams/tech-council/jt.md +0 -352
  103. package/.genie/code/teams/tech-council/nayr.md +0 -305
  104. package/.genie/code/teams/tech-council/oettam.md +0 -375
  105. package/.genie/neurons/README.md +0 -193
  106. package/.genie/neurons/forge.md +0 -106
  107. package/.genie/neurons/genie.md +0 -63
  108. package/.genie/neurons/review.md +0 -106
  109. package/.genie/neurons/wish.md +0 -104
  110. package/.genie/product/README.md +0 -20
  111. package/.genie/product/cli-automation.md +0 -359
  112. package/.genie/product/environment.md +0 -60
  113. package/.genie/product/mission.md +0 -60
  114. package/.genie/product/roadmap.md +0 -44
  115. package/.genie/product/tech-stack.md +0 -34
  116. package/.genie/product/templates/context-template.md +0 -218
  117. package/.genie/product/templates/qa-done-report-template.md +0 -68
  118. package/.genie/product/templates/review-report-template.md +0 -89
  119. package/.genie/product/templates/wish-template.md +0 -120
  120. package/.genie/scripts/helpers/analyze-commit.js +0 -195
  121. package/.genie/scripts/helpers/bullet-counter.js +0 -194
  122. package/.genie/scripts/helpers/bullet-find.js +0 -289
  123. package/.genie/scripts/helpers/bullet-id.js +0 -244
  124. package/.genie/scripts/helpers/check-secrets.js +0 -237
  125. package/.genie/scripts/helpers/count-tokens.js +0 -200
  126. package/.genie/scripts/helpers/create-frontmatter.js +0 -456
  127. package/.genie/scripts/helpers/detect-markers.js +0 -293
  128. package/.genie/scripts/helpers/detect-todos.js +0 -267
  129. package/.genie/scripts/helpers/detect-unlabeled-blocks.js +0 -135
  130. package/.genie/scripts/helpers/embeddings.js +0 -344
  131. package/.genie/scripts/helpers/find-empty-sections.js +0 -158
  132. package/.genie/scripts/helpers/index.js +0 -319
  133. package/.genie/scripts/helpers/validate-frontmatter.js +0 -578
  134. package/.genie/scripts/helpers/validate-links.js +0 -207
  135. package/.genie/scripts/helpers/validate-paths.js +0 -373
  136. package/.genie/spells/README.md +0 -9
  137. package/.genie/spells/ace-protocol.md +0 -118
  138. package/.genie/spells/ask-one-at-a-time.md +0 -175
  139. package/.genie/spells/backup-analyzer.md +0 -542
  140. package/.genie/spells/blocker.md +0 -12
  141. package/.genie/spells/break-things-move-fast.md +0 -56
  142. package/.genie/spells/context-candidates.md +0 -72
  143. package/.genie/spells/context-critic.md +0 -51
  144. package/.genie/spells/defer-to-expertise.md +0 -278
  145. package/.genie/spells/delegate-dont-do.md +0 -292
  146. package/.genie/spells/error-investigation-protocol.md +0 -328
  147. package/.genie/spells/evidence-based-completion.md +0 -273
  148. package/.genie/spells/experiment.md +0 -65
  149. package/.genie/spells/file-creation-protocol.md +0 -229
  150. package/.genie/spells/forge-integration.md +0 -281
  151. package/.genie/spells/forge-orchestration.md +0 -514
  152. package/.genie/spells/gather-context.md +0 -18
  153. package/.genie/spells/global-health-check.md +0 -34
  154. package/.genie/spells/global-noop-roundtrip.md +0 -25
  155. package/.genie/spells/install-genie.md +0 -1232
  156. package/.genie/spells/install.md +0 -82
  157. package/.genie/spells/investigate-before-commit.md +0 -112
  158. package/.genie/spells/know-yourself.md +0 -288
  159. package/.genie/spells/learn.md +0 -828
  160. package/.genie/spells/mcp-diagnostic-protocol.md +0 -246
  161. package/.genie/spells/mcp-first.md +0 -124
  162. package/.genie/spells/multi-step-execution.md +0 -67
  163. package/.genie/spells/orchestration-boundary-protocol.md +0 -256
  164. package/.genie/spells/orchestrator-not-implementor.md +0 -189
  165. package/.genie/spells/prompt.md +0 -746
  166. package/.genie/spells/reflect.md +0 -404
  167. package/.genie/spells/routing-decision-matrix.md +0 -368
  168. package/.genie/spells/run-in-parallel.md +0 -12
  169. package/.genie/spells/session-state-updater-example.md +0 -196
  170. package/.genie/spells/session-state-updater.md +0 -220
  171. package/.genie/spells/track-long-running-tasks.md +0 -133
  172. package/.genie/spells/troubleshoot-infrastructure.md +0 -176
  173. package/.genie/spells/upgrade-genie.md +0 -415
  174. package/.genie/spells/url-presentation-protocol.md +0 -301
  175. package/.genie/spells/wish-initiation.md +0 -158
  176. package/.genie/spells/wish-issue-linkage.md +0 -410
  177. package/.genie/spells/wish-lifecycle.md +0 -100
  178. package/.genie/state/provider-status.json +0 -3
  179. package/.genie/state/version.json +0 -16
  180. package/.genie/wishes/canonical-pgserve-pm2-supervision/WISH.md +0 -290
  181. package/.genie/wishes/pgserve-v2/BRIEF-from-genie-pgserve.md +0 -99
  182. package/.genie/wishes/pgserve-v2/WISH.md +0 -442
  183. package/.genie/wishes/release-system-genie-pattern/WISH.md +0 -268
  184. package/.genie/wishes/release-system-genie-pattern/validation.md +0 -205
  185. package/.gitguardian.yaml +0 -29
  186. package/.gitguardianignore +0 -16
  187. package/.github/workflows/ci.yml +0 -122
  188. package/.github/workflows/release.yml +0 -289
  189. package/.github/workflows/version.yml +0 -228
  190. package/.husky/pre-commit +0 -2
  191. package/AGENTS.md +0 -433
  192. package/CLAUDE.md +0 -1
  193. package/Makefile +0 -285
  194. package/assets/icon.ico +0 -0
  195. package/bun.lock +0 -435
  196. package/bunfig.toml +0 -28
  197. package/ecosystem.config.cjs +0 -23
  198. package/eslint.config.js +0 -63
  199. package/examples/multi-tenant-demo.js +0 -104
  200. package/install.sh +0 -123
  201. package/knip.json +0 -9
  202. package/scripts/test-bun-self-heal.sh +0 -163
  203. package/scripts/test-npx.sh +0 -60
  204. package/tests/audit.test.js +0 -189
  205. package/tests/backpressure.test.js +0 -167
  206. package/tests/benchmarks/runner.js +0 -1197
  207. package/tests/benchmarks/vector-generator.js +0 -368
  208. package/tests/cli-install.test.js +0 -322
  209. package/tests/control-db.test.js +0 -285
  210. package/tests/daemon-args.test.js +0 -86
  211. package/tests/daemon-control.test.js +0 -171
  212. package/tests/daemon-fingerprint-integration.test.js +0 -111
  213. package/tests/daemon-pr24-regression.test.js +0 -198
  214. package/tests/fingerprint.test.js +0 -263
  215. package/tests/fixtures/240-orphan-seed.sql +0 -30
  216. package/tests/multi-tenant.test.js +0 -374
  217. package/tests/orphan-cleanup.test.js +0 -390
  218. package/tests/pg-version-regex.test.js +0 -129
  219. package/tests/quick-bench.js +0 -135
  220. package/tests/router-handshake-retry.test.js +0 -119
  221. package/tests/router-handshake-watchdog.test.js +0 -110
  222. package/tests/sdk.test.js +0 -71
  223. package/tests/stale-postmaster-pid.test.js +0 -85
  224. package/tests/stress-test.js +0 -439
  225. package/tests/sync-perf-test.js +0 -150
  226. package/tests/tcp-listen.test.js +0 -368
  227. package/tests/tenancy.test.js +0 -403
  228. 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
- });