neon 3.0.0 → 3.1.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 (207) hide show
  1. package/README.md +70 -5
  2. package/dist/_chunks/auth_selection-DGgq6ifc.js +83 -0
  3. package/dist/_chunks/cmd_pipeline-CUbBO9U_.js +2818 -0
  4. package/dist/_chunks/credentials-MYdHdKah.js +188 -0
  5. package/dist/_chunks/env-NbA61JR3.js +585 -0
  6. package/dist/_chunks/env_services-Tz9G4JeT.js +531 -0
  7. package/dist/_chunks/paths-DMq0Lt7a.js +151 -0
  8. package/dist/_chunks/profiles-Ir29rqns.js +217 -0
  9. package/dist/_chunks/psql-DWH-kc69.js +2169 -0
  10. package/dist/_chunks/rolldown-runtime-D7D4PA-g.js +13 -0
  11. package/dist/_chunks/secure_file-BucZj4yQ.js +39 -0
  12. package/dist/analytics.js +163 -207
  13. package/dist/api.js +815 -758
  14. package/dist/auth.js +121 -141
  15. package/dist/auth_context.js +39 -53
  16. package/dist/cli.js +4 -7
  17. package/dist/commands/api.js +220 -250
  18. package/dist/commands/api_keys.js +251 -314
  19. package/dist/commands/auth.js +283 -328
  20. package/dist/commands/bootstrap.js +372 -437
  21. package/dist/commands/branches.js +304 -455
  22. package/dist/commands/bucket.js +374 -514
  23. package/dist/commands/checkout.js +213 -298
  24. package/dist/commands/config.js +573 -690
  25. package/dist/commands/connection_string.js +137 -165
  26. package/dist/commands/data_api.js +238 -260
  27. package/dist/commands/databases.js +67 -76
  28. package/dist/commands/deploy.js +31 -25
  29. package/dist/commands/dev.js +639 -719
  30. package/dist/commands/diff.js +156 -200
  31. package/dist/commands/env.js +255 -305
  32. package/dist/commands/functions.js +275 -355
  33. package/dist/commands/index.js +70 -65
  34. package/dist/commands/init.js +84 -119
  35. package/dist/commands/inspect.js +55 -55
  36. package/dist/commands/ip_allow.js +88 -120
  37. package/dist/commands/link.js +874 -1019
  38. package/dist/commands/logs.js +291 -0
  39. package/dist/commands/neon_auth.js +725 -933
  40. package/dist/commands/operations.js +34 -25
  41. package/dist/commands/orgs.js +28 -18
  42. package/dist/commands/profile.js +615 -846
  43. package/dist/commands/projects.js +313 -373
  44. package/dist/commands/psql.js +60 -58
  45. package/dist/commands/roles.js +55 -58
  46. package/dist/commands/schema_diff.js +87 -131
  47. package/dist/commands/set_context.js +34 -26
  48. package/dist/commands/snapshots.js +288 -413
  49. package/dist/commands/status.js +41 -37
  50. package/dist/commands/user.js +21 -10
  51. package/dist/commands/vpc_endpoints.js +85 -113
  52. package/dist/config.js +7 -6
  53. package/dist/config_format.js +50 -66
  54. package/dist/config_template.js +128 -157
  55. package/dist/context.js +183 -235
  56. package/dist/current_branch_fast_path.js +40 -49
  57. package/dist/dev/env.js +2 -446
  58. package/dist/dev/functions.js +54 -68
  59. package/dist/dev/inputs.js +46 -58
  60. package/dist/dev/runtime.js +135 -164
  61. package/dist/dev/websocket.js +766 -959
  62. package/dist/env.js +27 -33
  63. package/dist/env_file.js +118 -132
  64. package/dist/env_services.js +2 -51
  65. package/dist/errors.js +57 -68
  66. package/dist/functions_api.js +45 -43
  67. package/dist/help.js +189 -140
  68. package/dist/index.js +182 -257
  69. package/dist/init/agents.js +137 -118
  70. package/dist/init/auth.js +58 -68
  71. package/dist/init/bootstrap.js +325 -396
  72. package/dist/init/build_config.js +4 -2
  73. package/dist/init/detect_agent.js +56 -101
  74. package/dist/init/editors.js +35 -52
  75. package/dist/init/enrich_output.js +51 -66
  76. package/dist/init/extension.js +134 -171
  77. package/dist/init/inspect.js +179 -266
  78. package/dist/init/interactive.js +510 -622
  79. package/dist/init/neonctl.js +117 -168
  80. package/dist/init/orchestrate.js +157 -173
  81. package/dist/init/phases/auth.js +188 -202
  82. package/dist/init/phases/cleanup.js +23 -23
  83. package/dist/init/phases/db.js +251 -277
  84. package/dist/init/phases/getting_started.js +213 -223
  85. package/dist/init/phases/mcp.js +174 -224
  86. package/dist/init/phases/migrations.js +247 -248
  87. package/dist/init/phases/neon_auth.js +114 -133
  88. package/dist/init/phases/setup.js +546 -703
  89. package/dist/init/phases/skills.js +75 -86
  90. package/dist/init/phases/status.js +72 -67
  91. package/dist/init/resolve_context.js +102 -99
  92. package/dist/init/route_command.js +91 -98
  93. package/dist/init/skills.js +174 -218
  94. package/dist/init/vsix.js +77 -99
  95. package/dist/log.js +17 -16
  96. package/dist/neon_services.js +104 -129
  97. package/dist/parameters.gen.js +481 -471
  98. package/dist/pkg.js +17 -19
  99. package/dist/profile_keys.js +44 -47
  100. package/dist/psql/cli.js +44 -47
  101. package/dist/psql/command/cmd_cond.js +231 -406
  102. package/dist/psql/command/cmd_connect.js +557 -764
  103. package/dist/psql/command/cmd_copy.js +728 -984
  104. package/dist/psql/command/cmd_describe.js +1499 -1688
  105. package/dist/psql/command/cmd_format.js +733 -905
  106. package/dist/psql/command/cmd_io.js +2 -2193
  107. package/dist/psql/command/cmd_lo.js +297 -359
  108. package/dist/psql/command/cmd_meta.js +727 -878
  109. package/dist/psql/command/cmd_misc.js +138 -172
  110. package/dist/psql/command/cmd_pipeline.js +2 -1148
  111. package/dist/psql/command/cmd_restrict.js +119 -155
  112. package/dist/psql/command/cmd_show.js +529 -688
  113. package/dist/psql/command/dispatch.js +260 -325
  114. package/dist/psql/command/inputQueue.js +35 -33
  115. package/dist/psql/command/shared.js +49 -63
  116. package/dist/psql/complete/filenames.js +90 -133
  117. package/dist/psql/complete/index.js +59 -97
  118. package/dist/psql/complete/matcher.js +236 -300
  119. package/dist/psql/complete/psqlVars.js +218 -223
  120. package/dist/psql/complete/queries.js +159 -177
  121. package/dist/psql/complete/rules.js +1493 -2299
  122. package/dist/psql/core/common.js +2 -1253
  123. package/dist/psql/core/help.js +456 -546
  124. package/dist/psql/core/mainloop.js +692 -1303
  125. package/dist/psql/core/prompt.js +391 -408
  126. package/dist/psql/core/settings.js +429 -644
  127. package/dist/psql/core/sqlHelp.js +480 -554
  128. package/dist/psql/core/startup.js +2 -846
  129. package/dist/psql/core/syncVars.js +67 -110
  130. package/dist/psql/core/variables.js +156 -278
  131. package/dist/psql/describe/formatters.js +884 -1285
  132. package/dist/psql/describe/processNamePattern.js +173 -260
  133. package/dist/psql/describe/queries.js +1368 -2403
  134. package/dist/psql/describe/versionGate.js +32 -41
  135. package/dist/psql/index.js +2 -2030
  136. package/dist/psql/io/history.js +232 -271
  137. package/dist/psql/io/input.js +103 -108
  138. package/dist/psql/io/lineEditor/buffer.js +238 -319
  139. package/dist/psql/io/lineEditor/complete.js +135 -213
  140. package/dist/psql/io/lineEditor/filename.js +139 -148
  141. package/dist/psql/io/lineEditor/index.js +653 -870
  142. package/dist/psql/io/lineEditor/keymap.js +544 -702
  143. package/dist/psql/io/lineEditor/vt100.js +294 -341
  144. package/dist/psql/io/pgpass.js +158 -187
  145. package/dist/psql/io/pgservice.js +146 -183
  146. package/dist/psql/io/psqlrc.js +328 -403
  147. package/dist/psql/print/aligned.js +1020 -1683
  148. package/dist/psql/print/asciidoc.js +180 -214
  149. package/dist/psql/print/crosstab.js +281 -442
  150. package/dist/psql/print/csv.js +48 -70
  151. package/dist/psql/print/html.js +195 -226
  152. package/dist/psql/print/json.js +75 -88
  153. package/dist/psql/print/latex.js +291 -364
  154. package/dist/psql/print/pager.js +171 -242
  155. package/dist/psql/print/troff.js +194 -226
  156. package/dist/psql/print/unaligned.js +69 -95
  157. package/dist/psql/print/units.js +167 -169
  158. package/dist/psql/scanner/slash.js +428 -483
  159. package/dist/psql/scanner/sql.js +445 -889
  160. package/dist/psql/scanner/stringutils.js +309 -379
  161. package/dist/psql/types/index.js +8 -7
  162. package/dist/psql/types/scanner.js +25 -22
  163. package/dist/psql/wire/connection.js +2042 -2803
  164. package/dist/psql/wire/copy.js +84 -100
  165. package/dist/psql/wire/notify.js +39 -59
  166. package/dist/psql/wire/pipeline.js +305 -518
  167. package/dist/psql/wire/protocol.js +349 -417
  168. package/dist/psql/wire/sasl.js +180 -265
  169. package/dist/psql/wire/tls.js +400 -561
  170. package/dist/storage_api.js +115 -129
  171. package/dist/test_utils/fixtures.js +94 -113
  172. package/dist/test_utils/oauth_server.js +10 -7
  173. package/dist/test_utils/project_dir.js +33 -0
  174. package/dist/utils/ai_gateway_notice.js +131 -162
  175. package/dist/utils/api_enums.js +21 -28
  176. package/dist/utils/auth.js +10 -4
  177. package/dist/utils/branch_notice.js +20 -19
  178. package/dist/utils/branch_picker.js +83 -89
  179. package/dist/utils/cli_name.js +15 -12
  180. package/dist/utils/compute_units.js +20 -27
  181. package/dist/utils/config_diff.js +127 -158
  182. package/dist/utils/enrichers.js +95 -148
  183. package/dist/utils/esbuild.js +130 -189
  184. package/dist/utils/flags.js +35 -47
  185. package/dist/utils/formats.js +8 -15
  186. package/dist/utils/git_diff.js +69 -80
  187. package/dist/utils/inspect_db.js +101 -143
  188. package/dist/utils/inspect_queries.js +179 -142
  189. package/dist/utils/middlewares.js +39 -45
  190. package/dist/utils/openapi.js +87 -99
  191. package/dist/utils/package_manager.js +312 -110
  192. package/dist/utils/point_in_time.js +49 -53
  193. package/dist/utils/psql.js +89 -106
  194. package/dist/utils/service_picker.js +55 -58
  195. package/dist/utils/string.js +5 -5
  196. package/dist/utils/ui.js +38 -55
  197. package/dist/utils/write_sync.js +26 -35
  198. package/dist/utils/zip.js +4 -3
  199. package/dist/writer.js +67 -87
  200. package/package.json +11 -6
  201. package/dist/_shared/auth_selection.js +0 -86
  202. package/dist/_shared/credentials.js +0 -209
  203. package/dist/_shared/env-core/env.js +0 -558
  204. package/dist/_shared/env-core/reuse-secrets.js +0 -223
  205. package/dist/_shared/paths.js +0 -148
  206. package/dist/_shared/profiles.js +0 -276
  207. package/dist/_shared/secure_file.js +0 -43
@@ -1,439 +1,422 @@
1
- /**
2
- * psql Prompt renderer.
3
- *
4
- * TypeScript port of `get_prompt()` from `src/bin/psql/prompt.c`. Walks the
5
- * prompt template (PROMPT1/PROMPT2/PROMPT3) and expands `%`-escapes against
6
- * a `PromptContext` snapshot.
7
- *
8
- * Differences from upstream:
9
- *
10
- * - Upstream takes `promptStatus_t status, ConditionalStack cstack` and
11
- * pulls everything else off `pset`. We accept an explicit `PromptContext`
12
- * so the renderer is a pure function (easier to test, no module-level
13
- * state).
14
- * - `%s` ("service name") is provided from `settings.vars.SERVICE` per
15
- * upstream. We keep that behaviour. The WP-07 spec mentions a "current
16
- * timestamp" semantic for `%s`/`%S` — that is NOT what psql does today,
17
- * so we follow upstream rather than the spec note.
18
- * - `%S` is `search_path` from the connection's GUC report (upstream uses
19
- * `PQparameterStatus(pset.db, "search_path")`). With no Connection yet,
20
- * we fall back to `?`.
21
- * - `%#` checks the `IS_SUPERUSER` psql variable (set by upstream's
22
- * `is_superuser()` via the connect-time hook). Without a Connection we
23
- * default to non-superuser (`>`).
24
- * - `%P` outputs `off`/`on`/`abort` to match upstream's literal strings.
25
- * - `%i` outputs `primary` / `standby` / `?` (upstream reads the GUC
26
- * `in_hot_standby`); when no connection, `?`. The WP-07 spec asked for
27
- * "if-state from CondStack" — that conflicts with upstream `%i` and we
28
- * follow upstream. `%?` remains a no-op marker in upstream too.
29
- * - `%[`/`%]` are upstream's readline non-printing markers. We strip them
30
- * (emit nothing) for now; a future WP wiring readline integration can
31
- * re-introduce them.
32
- * - Backtick command interpolation (`%`\``cmd`\``) is executed synchronously
33
- * via `child_process.execSync` on `sh -c <cmd>`. Output is substituted
34
- * verbatim minus one trailing newline. Errors print to stderr and yield
35
- * an empty substitution — matching upstream's `get_prompt`, which re-runs
36
- * the command on every render rather than caching.
37
- * - Unknown `%X` escapes pass through as literal `X` (the upstream
38
- * `default:` branch). `%w` width is computed from PROMPT1 when present
39
- * and consumed by a subsequent PROMPT2 render via the optional
40
- * `lastPrompt1Width` field on the context.
41
- */
42
1
  import { execSync } from "node:child_process";
2
+ //#region src/psql/core/prompt.ts
43
3
  /**
44
- * Test seam for the prompt-level backtick executor. The default
45
- * implementation calls `execSync(cmd, { shell: '/bin/sh' })`; tests can
46
- * replace `.current` with a synchronous mock to avoid actually spawning
47
- * a child. Same pattern as `BACKTICK_EXECUTOR` in `scanner/slash.ts`.
48
- */
49
- export const PROMPT_BACKTICK_EXECUTOR = {
50
- current: (cmd) => execSync(cmd, {
51
- shell: "/bin/sh",
52
- encoding: "utf8",
53
- stdio: ["ignore", "pipe", "inherit"],
54
- // Keep prompt rendering responsive — a heavy backtick command should
55
- // not be able to fill arbitrary memory or hang the prompt forever.
56
- maxBuffer: 1 << 16,
57
- }),
58
- };
4
+ * psql Prompt renderer.
5
+ *
6
+ * TypeScript port of `get_prompt()` from `src/bin/psql/prompt.c`. Walks the
7
+ * prompt template (PROMPT1/PROMPT2/PROMPT3) and expands `%`-escapes against
8
+ * a `PromptContext` snapshot.
9
+ *
10
+ * Differences from upstream:
11
+ *
12
+ * - Upstream takes `promptStatus_t status, ConditionalStack cstack` and
13
+ * pulls everything else off `pset`. We accept an explicit `PromptContext`
14
+ * so the renderer is a pure function (easier to test, no module-level
15
+ * state).
16
+ * - `%s` ("service name") is provided from `settings.vars.SERVICE` per
17
+ * upstream. We keep that behaviour. The WP-07 spec mentions a "current
18
+ * timestamp" semantic for `%s`/`%S` — that is NOT what psql does today,
19
+ * so we follow upstream rather than the spec note.
20
+ * - `%S` is `search_path` from the connection's GUC report (upstream uses
21
+ * `PQparameterStatus(pset.db, "search_path")`). With no Connection yet,
22
+ * we fall back to `?`.
23
+ * - `%#` checks the `IS_SUPERUSER` psql variable (set by upstream's
24
+ * `is_superuser()` via the connect-time hook). Without a Connection we
25
+ * default to non-superuser (`>`).
26
+ * - `%P` outputs `off`/`on`/`abort` to match upstream's literal strings.
27
+ * - `%i` outputs `primary` / `standby` / `?` (upstream reads the GUC
28
+ * `in_hot_standby`); when no connection, `?`. The WP-07 spec asked for
29
+ * "if-state from CondStack" — that conflicts with upstream `%i` and we
30
+ * follow upstream. `%?` remains a no-op marker in upstream too.
31
+ * - `%[`/`%]` are upstream's readline non-printing markers. We strip them
32
+ * (emit nothing) for now; a future WP wiring readline integration can
33
+ * re-introduce them.
34
+ * - Backtick command interpolation (`%`\``cmd`\``) is executed synchronously
35
+ * via `child_process.execSync` on `sh -c <cmd>`. Output is substituted
36
+ * verbatim minus one trailing newline. Errors print to stderr and yield
37
+ * an empty substitution — matching upstream's `get_prompt`, which re-runs
38
+ * the command on every render rather than caching.
39
+ * - Unknown `%X` escapes pass through as literal `X` (the upstream
40
+ * `default:` branch). `%w` width is computed from PROMPT1 when present
41
+ * and consumed by a subsequent PROMPT2 render via the optional
42
+ * `lastPrompt1Width` field on the context.
43
+ */
44
+ /**
45
+ * Test seam for the prompt-level backtick executor. The default
46
+ * implementation calls `execSync(cmd, { shell: '/bin/sh' })`; tests can
47
+ * replace `.current` with a synchronous mock to avoid actually spawning
48
+ * a child. Same pattern as `BACKTICK_EXECUTOR` in `scanner/slash.ts`.
49
+ */
50
+ const PROMPT_BACKTICK_EXECUTOR = { current: (cmd) => execSync(cmd, {
51
+ shell: "/bin/sh",
52
+ encoding: "utf8",
53
+ stdio: [
54
+ "ignore",
55
+ "pipe",
56
+ "inherit"
57
+ ],
58
+ maxBuffer: 65536
59
+ }) };
59
60
  /**
60
- * Run a single backtick command and return its stdout, with one trailing
61
- * newline trimmed. On failure (non-zero exit / spawn error) print the
62
- * error to stderr and return the empty string so the prompt still renders.
63
- */
61
+ * Run a single backtick command and return its stdout, with one trailing
62
+ * newline trimmed. On failure (non-zero exit / spawn error) print the
63
+ * error to stderr and return the empty string so the prompt still renders.
64
+ */
64
65
  const runPromptBacktick = (cmd) => {
65
- if (cmd.length === 0)
66
- return "";
67
- try {
68
- const out = PROMPT_BACKTICK_EXECUTOR.current(cmd);
69
- return out.endsWith("\n") ? out.slice(0, -1) : out;
70
- }
71
- catch (err) {
72
- const msg = err instanceof Error ? err.message : String(err);
73
- process.stderr.write(`psql: error: \\!: ${cmd}: ${msg}\n`);
74
- return "";
75
- }
66
+ if (cmd.length === 0) return "";
67
+ try {
68
+ const out = PROMPT_BACKTICK_EXECUTOR.current(cmd);
69
+ return out.endsWith("\n") ? out.slice(0, -1) : out;
70
+ } catch (err) {
71
+ const msg = err instanceof Error ? err.message : String(err);
72
+ process.stderr.write(`psql: error: \\!: ${cmd}: ${msg}\n`);
73
+ return "";
74
+ }
76
75
  };
77
76
  /**
78
- * Render a prompt by name. Selects the template from
79
- * `settings.prompt{1,2,3}` and delegates to the escape-walking core.
80
- */
81
- export const renderPromptByName = (name, ctx) => {
82
- const template = name === "PROMPT1"
83
- ? ctx.settings.prompt1
84
- : name === "PROMPT2"
85
- ? ctx.settings.prompt2
86
- : ctx.settings.prompt3;
87
- const isPrompt1 = name === "PROMPT1";
88
- const out = renderPrompt(template, ctx);
89
- if (isPrompt1) {
90
- // Stash the visible width on the context so a subsequent PROMPT2 render
91
- // sharing this context object can line up with `%w`.
92
- ctx.lastPrompt1Width = visibleWidth(out);
93
- }
94
- return out;
77
+ * Render a prompt by name. Selects the template from
78
+ * `settings.prompt{1,2,3}` and delegates to the escape-walking core.
79
+ */
80
+ const renderPromptByName = (name, ctx) => {
81
+ const template = name === "PROMPT1" ? ctx.settings.prompt1 : name === "PROMPT2" ? ctx.settings.prompt2 : ctx.settings.prompt3;
82
+ const isPrompt1 = name === "PROMPT1";
83
+ const out = renderPrompt(template, ctx);
84
+ if (isPrompt1) ctx.lastPrompt1Width = visibleWidth(out);
85
+ return out;
95
86
  };
96
87
  /**
97
- * Render an arbitrary prompt template. Pure function over `(template, ctx)`.
98
- */
99
- export const renderPrompt = (template, ctx) => {
100
- let out = "";
101
- let i = 0;
102
- while (i < template.length) {
103
- const ch = template[i];
104
- if (ch !== "%") {
105
- out += ch;
106
- i += 1;
107
- continue;
108
- }
109
- // We have `%X`; advance past the `%` and resolve.
110
- i += 1;
111
- if (i >= template.length) {
112
- // Trailing `%` — emit nothing, matches upstream (esc stays true and
113
- // the loop exits without strlcat).
114
- break;
115
- }
116
- const escape = template[i];
117
- const { text, consumed } = expandEscape(escape, template, i, ctx);
118
- out += text;
119
- i += consumed;
120
- }
121
- return out;
88
+ * Render an arbitrary prompt template. Pure function over `(template, ctx)`.
89
+ */
90
+ const renderPrompt = (template, ctx) => {
91
+ let out = "";
92
+ let i = 0;
93
+ while (i < template.length) {
94
+ const ch = template[i];
95
+ if (ch !== "%") {
96
+ out += ch;
97
+ i += 1;
98
+ continue;
99
+ }
100
+ i += 1;
101
+ if (i >= template.length) break;
102
+ const escape = template[i];
103
+ const { text, consumed } = expandEscape(escape, template, i, ctx);
104
+ out += text;
105
+ i += consumed;
106
+ }
107
+ return out;
122
108
  };
123
- const expandEscape = (escape, template,
124
- /** Index of `escape` within `template`. */
125
- start, ctx) => {
126
- const { settings } = ctx;
127
- const db = settings.db;
128
- switch (escape) {
129
- case "%":
130
- return { text: "%", consumed: 1 };
131
- case "/":
132
- // Current database.
133
- return { text: db ? currentDatabase(ctx) : "", consumed: 1 };
134
- case "~": {
135
- // Like %/, but "~" when current_db matches the user's default db.
136
- // Upstream compares PQdb(pset.db) against PQuser(pset.db) and
137
- // against $PGDATABASE.
138
- if (!db)
139
- return { text: "", consumed: 1 };
140
- const cdb = currentDatabase(ctx);
141
- const user = currentUser(ctx);
142
- const pgdb = process.env.PGDATABASE;
143
- if (cdb === user || (pgdb !== undefined && pgdb === cdb)) {
144
- return { text: "~", consumed: 1 };
145
- }
146
- return { text: cdb, consumed: 1 };
147
- }
148
- case "M":
149
- case "m":
150
- return {
151
- text: db ? hostString(ctx, escape === "m") : "",
152
- consumed: 1,
153
- };
154
- case ">":
155
- return { text: db ? portString(ctx) : "", consumed: 1 };
156
- case "n":
157
- return { text: db ? currentUser(ctx) : "", consumed: 1 };
158
- case "p": {
159
- if (!db)
160
- return { text: "", consumed: 1 };
161
- const pid = backendPid(ctx);
162
- return { text: pid !== null ? String(pid) : "", consumed: 1 };
163
- }
164
- case "P": {
165
- if (!db)
166
- return { text: "", consumed: 1 };
167
- // Upstream emits "off" / "on" / "abort" (no trailing 'ed').
168
- const state = ctx.pipelineState ?? "off";
169
- const text = state === "on" ? "on" : state === "aborted" ? "abort" : "off";
170
- return { text, consumed: 1 };
171
- }
172
- case "i": {
173
- // Hot-standby indicator. Upstream reads in_hot_standby GUC.
174
- if (!db)
175
- return { text: "", consumed: 1 };
176
- const hs = parameterStatus(ctx, "in_hot_standby");
177
- if (hs === undefined)
178
- return { text: "?", consumed: 1 };
179
- return { text: hs === "on" ? "standby" : "primary", consumed: 1 };
180
- }
181
- case "s": {
182
- // service name from psql var SERVICE (upstream prompt.c).
183
- const svc = settings.vars.get("SERVICE");
184
- return { text: svc ?? "", consumed: 1 };
185
- }
186
- case "S": {
187
- // search_path; `?` when unavailable (older servers or no conn).
188
- if (!db)
189
- return { text: "?", consumed: 1 };
190
- const sp = parameterStatus(ctx, "search_path");
191
- return { text: sp ?? "?", consumed: 1 };
192
- }
193
- case "l":
194
- return { text: String(ctx.lineNumber), consumed: 1 };
195
- case "w": {
196
- // Whitespace padding the width of the last PROMPT1 render.
197
- const width = Math.max(0, ctx.lastPrompt1Width ?? 0);
198
- return { text: db ? " ".repeat(width) : "", consumed: 1 };
199
- }
200
- case "a":
201
- // Bell. Upstream falls into default, which is just literal 'a' (no
202
- // bell handling in get_prompt). The WP-07 spec explicitly asks for
203
- // ^G though, and emitting BEL is harmless. Honour the spec.
204
- return { text: "\x07", consumed: 1 };
205
- case "R": {
206
- const text = expandR(ctx);
207
- return { text, consumed: 1 };
208
- }
209
- case "x": {
210
- if (!db)
211
- return { text: "?", consumed: 1 };
212
- const tx = ctx.inTransaction ?? "idle";
213
- if (tx === "in-block")
214
- return { text: "*", consumed: 1 };
215
- if (tx === "failed")
216
- return { text: "!", consumed: 1 };
217
- if (tx === "unknown")
218
- return { text: "?", consumed: 1 };
219
- return { text: "", consumed: 1 };
220
- }
221
- case "#": {
222
- // Superuser indicator. Upstream calls is_superuser() which checks
223
- // pset.vars["IS_SUPERUSER"] === "on".
224
- const isSuper = settings.vars.get("IS_SUPERUSER") === "on";
225
- return { text: isSuper ? "#" : ">", consumed: 1 };
226
- }
227
- case "?":
228
- // Reserved by upstream ("not here yet"). Emit nothing.
229
- return { text: "", consumed: 1 };
230
- case "[":
231
- case "]":
232
- // Readline non-printing markers. Strip — no terminal info embedded.
233
- return { text: "", consumed: 1 };
234
- case "`": {
235
- // Backtick command: read until the matching `` ` `` after the opening
236
- // backtick, then run the body through `sh -c` and substitute its
237
- // stdout. Upstream `get_prompt` re-runs the command on every render
238
- // rather than caching, so we do the same — callers who want caching
239
- // should stash the rendered prompt themselves.
240
- const close = template.indexOf("`", start + 1);
241
- if (close === -1) {
242
- // Unterminated — consume the rest defensively without spawning.
243
- return { text: "", consumed: template.length - start };
244
- }
245
- const body = template.slice(start + 1, close);
246
- return {
247
- text: runPromptBacktick(body),
248
- consumed: close - start + 1,
249
- };
250
- }
251
- case ":": {
252
- // Variable interpolation %:name:
253
- const close = template.indexOf(":", start + 1);
254
- if (close === -1) {
255
- return { text: "", consumed: template.length - start };
256
- }
257
- const name = template.slice(start + 1, close);
258
- const val = settings.vars.get(name);
259
- return { text: val ?? "", consumed: close - start + 1 };
260
- }
261
- case "0":
262
- case "1":
263
- case "2":
264
- case "3":
265
- case "4":
266
- case "5":
267
- case "6":
268
- case "7": {
269
- // Octal byte value `%nnn`. Upstream uses strtol(p, &p, 8), which
270
- // greedily consumes 1–3 octal digits.
271
- let consumed = 1;
272
- let body = escape;
273
- while (consumed < 3 && start + consumed < template.length) {
274
- const next = template[start + consumed];
275
- if (next >= "0" && next <= "7") {
276
- body += next;
277
- consumed += 1;
278
- }
279
- else {
280
- break;
281
- }
282
- }
283
- const byte = parseInt(body, 8);
284
- return { text: String.fromCharCode(byte), consumed };
285
- }
286
- default:
287
- // Unknown escape → literal character (matches upstream default:).
288
- return { text: escape, consumed: 1 };
289
- }
109
+ const expandEscape = (escape, template, start, ctx) => {
110
+ const { settings } = ctx;
111
+ const db = settings.db;
112
+ switch (escape) {
113
+ case "%": return {
114
+ text: "%",
115
+ consumed: 1
116
+ };
117
+ case "/": return {
118
+ text: db ? currentDatabase(ctx) : "",
119
+ consumed: 1
120
+ };
121
+ case "~": {
122
+ if (!db) return {
123
+ text: "",
124
+ consumed: 1
125
+ };
126
+ const cdb = currentDatabase(ctx);
127
+ const user = currentUser(ctx);
128
+ const pgdb = process.env.PGDATABASE;
129
+ if (cdb === user || pgdb !== void 0 && pgdb === cdb) return {
130
+ text: "~",
131
+ consumed: 1
132
+ };
133
+ return {
134
+ text: cdb,
135
+ consumed: 1
136
+ };
137
+ }
138
+ case "M":
139
+ case "m": return {
140
+ text: db ? hostString(ctx, escape === "m") : "",
141
+ consumed: 1
142
+ };
143
+ case ">": return {
144
+ text: db ? portString(ctx) : "",
145
+ consumed: 1
146
+ };
147
+ case "n": return {
148
+ text: db ? currentUser(ctx) : "",
149
+ consumed: 1
150
+ };
151
+ case "p": {
152
+ if (!db) return {
153
+ text: "",
154
+ consumed: 1
155
+ };
156
+ const pid = backendPid(ctx);
157
+ return {
158
+ text: pid !== null ? String(pid) : "",
159
+ consumed: 1
160
+ };
161
+ }
162
+ case "P": {
163
+ if (!db) return {
164
+ text: "",
165
+ consumed: 1
166
+ };
167
+ const state = ctx.pipelineState ?? "off";
168
+ return {
169
+ text: state === "on" ? "on" : state === "aborted" ? "abort" : "off",
170
+ consumed: 1
171
+ };
172
+ }
173
+ case "i": {
174
+ if (!db) return {
175
+ text: "",
176
+ consumed: 1
177
+ };
178
+ const hs = parameterStatus(ctx, "in_hot_standby");
179
+ if (hs === void 0) return {
180
+ text: "?",
181
+ consumed: 1
182
+ };
183
+ return {
184
+ text: hs === "on" ? "standby" : "primary",
185
+ consumed: 1
186
+ };
187
+ }
188
+ case "s": return {
189
+ text: settings.vars.get("SERVICE") ?? "",
190
+ consumed: 1
191
+ };
192
+ case "S":
193
+ if (!db) return {
194
+ text: "?",
195
+ consumed: 1
196
+ };
197
+ return {
198
+ text: parameterStatus(ctx, "search_path") ?? "?",
199
+ consumed: 1
200
+ };
201
+ case "l": return {
202
+ text: String(ctx.lineNumber),
203
+ consumed: 1
204
+ };
205
+ case "w": {
206
+ const width = Math.max(0, ctx.lastPrompt1Width ?? 0);
207
+ return {
208
+ text: db ? " ".repeat(width) : "",
209
+ consumed: 1
210
+ };
211
+ }
212
+ case "a": return {
213
+ text: "\x07",
214
+ consumed: 1
215
+ };
216
+ case "R": return {
217
+ text: expandR(ctx),
218
+ consumed: 1
219
+ };
220
+ case "x": {
221
+ if (!db) return {
222
+ text: "?",
223
+ consumed: 1
224
+ };
225
+ const tx = ctx.inTransaction ?? "idle";
226
+ if (tx === "in-block") return {
227
+ text: "*",
228
+ consumed: 1
229
+ };
230
+ if (tx === "failed") return {
231
+ text: "!",
232
+ consumed: 1
233
+ };
234
+ if (tx === "unknown") return {
235
+ text: "?",
236
+ consumed: 1
237
+ };
238
+ return {
239
+ text: "",
240
+ consumed: 1
241
+ };
242
+ }
243
+ case "#": return {
244
+ text: settings.vars.get("IS_SUPERUSER") === "on" ? "#" : ">",
245
+ consumed: 1
246
+ };
247
+ case "?": return {
248
+ text: "",
249
+ consumed: 1
250
+ };
251
+ case "[":
252
+ case "]": return {
253
+ text: "",
254
+ consumed: 1
255
+ };
256
+ case "`": {
257
+ const close = template.indexOf("`", start + 1);
258
+ if (close === -1) return {
259
+ text: "",
260
+ consumed: template.length - start
261
+ };
262
+ const body = template.slice(start + 1, close);
263
+ return {
264
+ text: runPromptBacktick(body),
265
+ consumed: close - start + 1
266
+ };
267
+ }
268
+ case ":": {
269
+ const close = template.indexOf(":", start + 1);
270
+ if (close === -1) return {
271
+ text: "",
272
+ consumed: template.length - start
273
+ };
274
+ const name = template.slice(start + 1, close);
275
+ return {
276
+ text: settings.vars.get(name) ?? "",
277
+ consumed: close - start + 1
278
+ };
279
+ }
280
+ case "0":
281
+ case "1":
282
+ case "2":
283
+ case "3":
284
+ case "4":
285
+ case "5":
286
+ case "6":
287
+ case "7": {
288
+ let consumed = 1;
289
+ let body = escape;
290
+ while (consumed < 3 && start + consumed < template.length) {
291
+ const next = template[start + consumed];
292
+ if (next >= "0" && next <= "7") {
293
+ body += next;
294
+ consumed += 1;
295
+ } else break;
296
+ }
297
+ const byte = parseInt(body, 8);
298
+ return {
299
+ text: String.fromCharCode(byte),
300
+ consumed
301
+ };
302
+ }
303
+ default: return {
304
+ text: escape,
305
+ consumed: 1
306
+ };
307
+ }
290
308
  };
291
309
  /**
292
- * Expand `%R`. PROMPT1 reflects the connection / cond-stack / singleline
293
- * state. PROMPT2 reflects the scanner's promptStatus — upstream renders a
294
- * mnemonic character for *why* the parser is still hanging on for more
295
- * input:
296
- *
297
- * - `-` plain continuation (default)
298
- * - `'` inside a single-quoted string
299
- * - `"` inside a double-quoted identifier
300
- * - `$` inside a dollar-quoted block
301
- * - `*` inside a block comment
302
- * - `(` inside unmatched parens
303
- *
304
- * PROMPT3 is the COPY indicator; upstream falls into the `default` branch
305
- * and emits nothing — we do the same.
306
- */
310
+ * Expand `%R`. PROMPT1 reflects the connection / cond-stack / singleline
311
+ * state. PROMPT2 reflects the scanner's promptStatus — upstream renders a
312
+ * mnemonic character for *why* the parser is still hanging on for more
313
+ * input:
314
+ *
315
+ * - `-` plain continuation (default)
316
+ * - `'` inside a single-quoted string
317
+ * - `"` inside a double-quoted identifier
318
+ * - `$` inside a dollar-quoted block
319
+ * - `*` inside a block comment
320
+ * - `(` inside unmatched parens
321
+ *
322
+ * PROMPT3 is the COPY indicator; upstream falls into the `default` branch
323
+ * and emits nothing — we do the same.
324
+ */
307
325
  const expandR = (ctx) => {
308
- const { promptStatus, cond, settings } = ctx;
309
- switch (promptStatus) {
310
- case "ready": {
311
- if (!isConditionalActive(cond))
312
- return "@";
313
- if (!settings.db)
314
- return "!";
315
- if (!settings.singleline)
316
- return "=";
317
- return "^";
318
- }
319
- case "continue":
320
- return "-";
321
- case "continue-quote":
322
- return "'";
323
- case "continue-dquote":
324
- return '"';
325
- case "continue-dollar":
326
- return "$";
327
- case "comment":
328
- return "*";
329
- case "paren":
330
- return "(";
331
- case "copy":
332
- // PROMPT3 / COPY path — upstream emits nothing for %R here.
333
- return "";
334
- default:
335
- return "";
336
- }
326
+ const { promptStatus, cond, settings } = ctx;
327
+ switch (promptStatus) {
328
+ case "ready":
329
+ if (!isConditionalActive(cond)) return "@";
330
+ if (!settings.db) return "!";
331
+ if (!settings.singleline) return "=";
332
+ return "^";
333
+ case "continue": return "-";
334
+ case "continue-quote": return "'";
335
+ case "continue-dquote": return "\"";
336
+ case "continue-dollar": return "$";
337
+ case "comment": return "*";
338
+ case "paren": return "(";
339
+ case "copy": return "";
340
+ default: return "";
341
+ }
337
342
  };
338
343
  /**
339
- * Return `false` if the current `\if` branch is non-active (we're inside a
340
- * skipped block). Mirrors upstream `conditional_active`. Inactive states are
341
- * `false`, `else-false`, and `ignored`; everything else (including the
342
- * sentinel `none` for "no \if active") is treated as active.
343
- */
344
+ * Return `false` if the current `\if` branch is non-active (we're inside a
345
+ * skipped block). Mirrors upstream `conditional_active`. Inactive states are
346
+ * `false`, `else-false`, and `ignored`; everything else (including the
347
+ * sentinel `none` for "no \if active") is treated as active.
348
+ */
344
349
  const isConditionalActive = (cond) => {
345
- const top = cond.top();
346
- if (!top)
347
- return true;
348
- const inactive = ["false", "else-false", "ignored"];
349
- return !inactive.includes(top.state);
350
+ const top = cond.top();
351
+ if (!top) return true;
352
+ return ![
353
+ "false",
354
+ "else-false",
355
+ "ignored"
356
+ ].includes(top.state);
350
357
  };
351
358
  /**
352
- * Visible width of a rendered prompt — used to seed `%w` on PROMPT2. We
353
- * subtract ANSI escape sequences and the unicode width is approximated by
354
- * code-point count (full Unicode width tables live in the printer WP).
355
- */
359
+ * Visible width of a rendered prompt — used to seed `%w` on PROMPT2. We
360
+ * subtract ANSI escape sequences and the unicode width is approximated by
361
+ * code-point count (full Unicode width tables live in the printer WP).
362
+ */
356
363
  const visibleWidth = (s) => {
357
- // Strip CSI escapes (ESC `[` … letter). Conservative — only trims the
358
- // common ANSI color form psql emits. Newlines reset the count.
359
- const stripped = s.replace(/\x1b\[[0-9;]*[A-Za-z]/g, "");
360
- const lastNl = stripped.lastIndexOf("\n");
361
- const tail = lastNl === -1 ? stripped : stripped.slice(lastNl + 1);
362
- // Code-unit count — good enough for ASCII prompts; CJK / surrogate-pair
363
- // width handling is a printer-WP concern.
364
- return tail.length;
364
+ const stripped = s.replace(/\x1b\[[0-9;]*[A-Za-z]/g, "");
365
+ const lastNl = stripped.lastIndexOf("\n");
366
+ return (lastNl === -1 ? stripped : stripped.slice(lastNl + 1)).length;
365
367
  };
366
368
  const currentDatabase = (ctx) => {
367
- const db = ctx.settings.db;
368
- if (!db)
369
- return "";
370
- // Prefer a libpq-style accessor if the runtime Connection exposes one;
371
- // otherwise fall back to the application's GUC view.
372
- const meta = db;
373
- if (typeof meta.database === "string" && meta.database.length > 0) {
374
- return meta.database;
375
- }
376
- return parameterStatus(ctx, "database") ?? "";
369
+ const db = ctx.settings.db;
370
+ if (!db) return "";
371
+ const meta = db;
372
+ if (typeof meta.database === "string" && meta.database.length > 0) return meta.database;
373
+ return parameterStatus(ctx, "database") ?? "";
377
374
  };
378
375
  const currentUser = (ctx) => {
379
- const db = ctx.settings.db;
380
- if (!db)
381
- return "";
382
- const meta = db;
383
- if (typeof meta.user === "string" && meta.user.length > 0) {
384
- return meta.user;
385
- }
386
- return parameterStatus(ctx, "session_authorization") ?? "";
376
+ const db = ctx.settings.db;
377
+ if (!db) return "";
378
+ const meta = db;
379
+ if (typeof meta.user === "string" && meta.user.length > 0) return meta.user;
380
+ return parameterStatus(ctx, "session_authorization") ?? "";
387
381
  };
388
382
  const hostString = (ctx, short) => {
389
- const db = ctx.settings.db;
390
- if (!db)
391
- return "";
392
- const meta = db;
393
- const host = typeof meta.host === "string" ? meta.host : "";
394
- if (host.length === 0 || host.startsWith("/")) {
395
- // UNIX socket / unknown — upstream emits "[local]" or "[local:path]".
396
- if (host.length === 0 || short)
397
- return "[local]";
398
- return `[local:${host}]`;
399
- }
400
- if (short) {
401
- const dot = host.indexOf(".");
402
- return dot === -1 ? host : host.slice(0, dot);
403
- }
404
- return host;
383
+ const db = ctx.settings.db;
384
+ if (!db) return "";
385
+ const meta = db;
386
+ const host = typeof meta.host === "string" ? meta.host : "";
387
+ if (host.length === 0 || host.startsWith("/")) {
388
+ if (host.length === 0 || short) return "[local]";
389
+ return `[local:${host}]`;
390
+ }
391
+ if (short) {
392
+ const dot = host.indexOf(".");
393
+ return dot === -1 ? host : host.slice(0, dot);
394
+ }
395
+ return host;
405
396
  };
406
397
  const portString = (ctx) => {
407
- const db = ctx.settings.db;
408
- if (!db)
409
- return "";
410
- const meta = db;
411
- if (typeof meta.port === "number" && Number.isFinite(meta.port)) {
412
- return String(meta.port);
413
- }
414
- if (typeof meta.port === "string" && meta.port.length > 0) {
415
- return meta.port;
416
- }
417
- return "";
398
+ const db = ctx.settings.db;
399
+ if (!db) return "";
400
+ const meta = db;
401
+ if (typeof meta.port === "number" && Number.isFinite(meta.port)) return String(meta.port);
402
+ if (typeof meta.port === "string" && meta.port.length > 0) return meta.port;
403
+ return "";
418
404
  };
419
405
  const backendPid = (ctx) => {
420
- const db = ctx.settings.db;
421
- if (!db)
422
- return null;
423
- const meta = db;
424
- if (typeof meta.pid === "number" && Number.isFinite(meta.pid)) {
425
- return meta.pid;
426
- }
427
- return null;
406
+ const db = ctx.settings.db;
407
+ if (!db) return null;
408
+ const meta = db;
409
+ if (typeof meta.pid === "number" && Number.isFinite(meta.pid)) return meta.pid;
410
+ return null;
428
411
  };
429
412
  const parameterStatus = (ctx, name) => {
430
- const db = ctx.settings.db;
431
- if (!db)
432
- return undefined;
433
- try {
434
- return db.parameterStatus(name);
435
- }
436
- catch {
437
- return undefined;
438
- }
413
+ const db = ctx.settings.db;
414
+ if (!db) return void 0;
415
+ try {
416
+ return db.parameterStatus(name);
417
+ } catch {
418
+ return;
419
+ }
439
420
  };
421
+ //#endregion
422
+ export { PROMPT_BACKTICK_EXECUTOR, renderPrompt, renderPromptByName };