neonctl 2.27.1 → 2.29.0

package/dist/psql/core/prompt.js

@@ -39,7 +39,7 @@
  *    and consumed by a subsequent PROMPT2 render via the optional
  *    `lastPrompt1Width` field on the context.
  */
-import { execSync } from 'node:child_process';
+import { execSync } from "node:child_process";
 /**
  * Test seam for the prompt-level backtick executor. The default
  * implementation calls `execSync(cmd, { shell: '/bin/sh' })`; tests can
@@ -48,9 +48,9 @@ import { execSync } from 'node:child_process';
  */
 export const PROMPT_BACKTICK_EXECUTOR = {
     current: (cmd) => execSync(cmd, {
-        shell: '/bin/sh',
-        encoding: 'utf8',
-        stdio: ['ignore', 'pipe', 'inherit'],
+        shell: "/bin/sh",
+        encoding: "utf8",
+        stdio: ["ignore", "pipe", "inherit"],
         // Keep prompt rendering responsive — a heavy backtick command should
         // not be able to fill arbitrary memory or hang the prompt forever.
         maxBuffer: 1 << 16,
@@ -63,15 +63,15 @@ export const PROMPT_BACKTICK_EXECUTOR = {
  */
 const runPromptBacktick = (cmd) => {
     if (cmd.length === 0)
-        return '';
+        return "";
     try {
         const out = PROMPT_BACKTICK_EXECUTOR.current(cmd);
-        return out.endsWith('\n') ? out.slice(0, -1) : out;
+        return out.endsWith("\n") ? out.slice(0, -1) : out;
     }
     catch (err) {
         const msg = err instanceof Error ? err.message : String(err);
         process.stderr.write(`psql: error: \\!: ${cmd}: ${msg}\n`);
-        return '';
+        return "";
     }
 };
 /**
@@ -79,12 +79,12 @@ const runPromptBacktick = (cmd) => {
  * `settings.prompt{1,2,3}` and delegates to the escape-walking core.
  */
 export const renderPromptByName = (name, ctx) => {
-    const template = name === 'PROMPT1'
+    const template = name === "PROMPT1"
         ? ctx.settings.prompt1
-        : name === 'PROMPT2'
+        : name === "PROMPT2"
             ? ctx.settings.prompt2
             : ctx.settings.prompt3;
-    const isPrompt1 = name === 'PROMPT1';
+    const isPrompt1 = name === "PROMPT1";
     const out = renderPrompt(template, ctx);
     if (isPrompt1) {
         // Stash the visible width on the context so a subsequent PROMPT2 render
@@ -97,11 +97,11 @@ export const renderPromptByName = (name, ctx) => {
  * Render an arbitrary prompt template. Pure function over `(template, ctx)`.
  */
 export const renderPrompt = (template, ctx) => {
-    let out = '';
+    let out = "";
     let i = 0;
     while (i < template.length) {
         const ch = template[i];
-        if (ch !== '%') {
+        if (ch !== "%") {
             out += ch;
             i += 1;
             continue;
@@ -126,150 +126,153 @@ start, ctx) => {
     const { settings } = ctx;
     const db = settings.db;
     switch (escape) {
-        case '%':
-            return { text: '%', consumed: 1 };
-        case '/':
+        case "%":
+            return { text: "%", consumed: 1 };
+        case "/":
             // Current database.
-            return { text: db ? currentDatabase(ctx) : '', consumed: 1 };
-        case '~': {
+            return { text: db ? currentDatabase(ctx) : "", consumed: 1 };
+        case "~": {
             // Like %/, but "~" when current_db matches the user's default db.
             // Upstream compares PQdb(pset.db) against PQuser(pset.db) and
             // against $PGDATABASE.
             if (!db)
-                return { text: '', consumed: 1 };
+                return { text: "", consumed: 1 };
             const cdb = currentDatabase(ctx);
             const user = currentUser(ctx);
             const pgdb = process.env.PGDATABASE;
             if (cdb === user || (pgdb !== undefined && pgdb === cdb)) {
-                return { text: '~', consumed: 1 };
+                return { text: "~", consumed: 1 };
             }
             return { text: cdb, consumed: 1 };
         }
-        case 'M':
-        case 'm':
+        case "M":
+        case "m":
             return {
-                text: db ? hostString(ctx, escape === 'm') : '',
+                text: db ? hostString(ctx, escape === "m") : "",
                 consumed: 1,
             };
-        case '>':
-            return { text: db ? portString(ctx) : '', consumed: 1 };
-        case 'n':
-            return { text: db ? currentUser(ctx) : '', consumed: 1 };
-        case 'p': {
+        case ">":
+            return { text: db ? portString(ctx) : "", consumed: 1 };
+        case "n":
+            return { text: db ? currentUser(ctx) : "", consumed: 1 };
+        case "p": {
             if (!db)
-                return { text: '', consumed: 1 };
+                return { text: "", consumed: 1 };
             const pid = backendPid(ctx);
-            return { text: pid !== null ? String(pid) : '', consumed: 1 };
+            return { text: pid !== null ? String(pid) : "", consumed: 1 };
         }
-        case 'P': {
+        case "P": {
             if (!db)
-                return { text: '', consumed: 1 };
+                return { text: "", consumed: 1 };
             // Upstream emits "off" / "on" / "abort" (no trailing 'ed').
-            const state = ctx.pipelineState ?? 'off';
-            const text = state === 'on' ? 'on' : state === 'aborted' ? 'abort' : 'off';
+            const state = ctx.pipelineState ?? "off";
+            const text = state === "on" ? "on" : state === "aborted" ? "abort" : "off";
             return { text, consumed: 1 };
         }
-        case 'i': {
+        case "i": {
             // Hot-standby indicator. Upstream reads in_hot_standby GUC.
             if (!db)
-                return { text: '', consumed: 1 };
-            const hs = parameterStatus(ctx, 'in_hot_standby');
+                return { text: "", consumed: 1 };
+            const hs = parameterStatus(ctx, "in_hot_standby");
             if (hs === undefined)
-                return { text: '?', consumed: 1 };
-            return { text: hs === 'on' ? 'standby' : 'primary', consumed: 1 };
+                return { text: "?", consumed: 1 };
+            return { text: hs === "on" ? "standby" : "primary", consumed: 1 };
         }
-        case 's': {
+        case "s": {
             // service name from psql var SERVICE (upstream prompt.c).
-            const svc = settings.vars.get('SERVICE');
-            return { text: svc ?? '', consumed: 1 };
+            const svc = settings.vars.get("SERVICE");
+            return { text: svc ?? "", consumed: 1 };
         }
-        case 'S': {
+        case "S": {
             // search_path; `?` when unavailable (older servers or no conn).
             if (!db)
-                return { text: '?', consumed: 1 };
-            const sp = parameterStatus(ctx, 'search_path');
-            return { text: sp ?? '?', consumed: 1 };
+                return { text: "?", consumed: 1 };
+            const sp = parameterStatus(ctx, "search_path");
+            return { text: sp ?? "?", consumed: 1 };
         }
-        case 'l':
+        case "l":
             return { text: String(ctx.lineNumber), consumed: 1 };
-        case 'w': {
+        case "w": {
             // Whitespace padding the width of the last PROMPT1 render.
             const width = Math.max(0, ctx.lastPrompt1Width ?? 0);
-            return { text: db ? ' '.repeat(width) : '', consumed: 1 };
+            return { text: db ? " ".repeat(width) : "", consumed: 1 };
         }
-        case 'a':
+        case "a":
             // Bell. Upstream falls into default, which is just literal 'a' (no
             // bell handling in get_prompt). The WP-07 spec explicitly asks for
             // ^G though, and emitting BEL is harmless. Honour the spec.
-            return { text: '\x07', consumed: 1 };
-        case 'R': {
+            return { text: "\x07", consumed: 1 };
+        case "R": {
             const text = expandR(ctx);
             return { text, consumed: 1 };
         }
-        case 'x': {
+        case "x": {
             if (!db)
-                return { text: '?', consumed: 1 };
-            const tx = ctx.inTransaction ?? 'idle';
-            if (tx === 'in-block')
-                return { text: '*', consumed: 1 };
-            if (tx === 'failed')
-                return { text: '!', consumed: 1 };
-            if (tx === 'unknown')
-                return { text: '?', consumed: 1 };
-            return { text: '', consumed: 1 };
+                return { text: "?", consumed: 1 };
+            const tx = ctx.inTransaction ?? "idle";
+            if (tx === "in-block")
+                return { text: "*", consumed: 1 };
+            if (tx === "failed")
+                return { text: "!", consumed: 1 };
+            if (tx === "unknown")
+                return { text: "?", consumed: 1 };
+            return { text: "", consumed: 1 };
         }
-        case '#': {
+        case "#": {
             // Superuser indicator. Upstream calls is_superuser() which checks
             // pset.vars["IS_SUPERUSER"] === "on".
-            const isSuper = settings.vars.get('IS_SUPERUSER') === 'on';
-            return { text: isSuper ? '#' : '>', consumed: 1 };
+            const isSuper = settings.vars.get("IS_SUPERUSER") === "on";
+            return { text: isSuper ? "#" : ">", consumed: 1 };
         }
-        case '?':
+        case "?":
             // Reserved by upstream ("not here yet"). Emit nothing.
-            return { text: '', consumed: 1 };
-        case '[':
-        case ']':
+            return { text: "", consumed: 1 };
+        case "[":
+        case "]":
             // Readline non-printing markers. Strip — no terminal info embedded.
-            return { text: '', consumed: 1 };
-        case '`': {
+            return { text: "", consumed: 1 };
+        case "`": {
             // Backtick command: read until the matching `` ` `` after the opening
             // backtick, then run the body through `sh -c` and substitute its
             // stdout. Upstream `get_prompt` re-runs the command on every render
             // rather than caching, so we do the same — callers who want caching
             // should stash the rendered prompt themselves.
-            const close = template.indexOf('`', start + 1);
+            const close = template.indexOf("`", start + 1);
             if (close === -1) {
                 // Unterminated — consume the rest defensively without spawning.
-                return { text: '', consumed: template.length - start };
+                return { text: "", consumed: template.length - start };
             }
             const body = template.slice(start + 1, close);
-            return { text: runPromptBacktick(body), consumed: close - start + 1 };
+            return {
+                text: runPromptBacktick(body),
+                consumed: close - start + 1,
+            };
         }
-        case ':': {
+        case ":": {
             // Variable interpolation %:name:
-            const close = template.indexOf(':', start + 1);
+            const close = template.indexOf(":", start + 1);
             if (close === -1) {
-                return { text: '', consumed: template.length - start };
+                return { text: "", consumed: template.length - start };
             }
             const name = template.slice(start + 1, close);
             const val = settings.vars.get(name);
-            return { text: val ?? '', consumed: close - start + 1 };
+            return { text: val ?? "", consumed: close - start + 1 };
         }
-        case '0':
-        case '1':
-        case '2':
-        case '3':
-        case '4':
-        case '5':
-        case '6':
-        case '7': {
+        case "0":
+        case "1":
+        case "2":
+        case "3":
+        case "4":
+        case "5":
+        case "6":
+        case "7": {
             // Octal byte value `%nnn`. Upstream uses strtol(p, &p, 8), which
             // greedily consumes 1–3 octal digits.
             let consumed = 1;
             let body = escape;
             while (consumed < 3 && start + consumed < template.length) {
                 const next = template[start + consumed];
-                if (next >= '0' && next <= '7') {
+                if (next >= "0" && next <= "7") {
                     body += next;
                     consumed += 1;
                 }
@@ -304,32 +307,32 @@ start, ctx) => {
 const expandR = (ctx) => {
     const { promptStatus, cond, settings } = ctx;
     switch (promptStatus) {
-        case 'ready': {
+        case "ready": {
             if (!isConditionalActive(cond))
-                return '@';
+                return "@";
             if (!settings.db)
-                return '!';
+                return "!";
             if (!settings.singleline)
-                return '=';
-            return '^';
+                return "=";
+            return "^";
         }
-        case 'continue':
-            return '-';
-        case 'continue-quote':
+        case "continue":
+            return "-";
+        case "continue-quote":
             return "'";
-        case 'continue-dquote':
+        case "continue-dquote":
             return '"';
-        case 'continue-dollar':
-            return '$';
-        case 'comment':
-            return '*';
-        case 'paren':
-            return '(';
-        case 'copy':
+        case "continue-dollar":
+            return "$";
+        case "comment":
+            return "*";
+        case "paren":
+            return "(";
+        case "copy":
             // PROMPT3 / COPY path — upstream emits nothing for %R here.
-            return '';
+            return "";
         default:
-            return '';
+            return "";
     }
 };
 /**
@@ -342,7 +345,7 @@ const isConditionalActive = (cond) => {
     const top = cond.top();
     if (!top)
         return true;
-    const inactive = ['false', 'else-false', 'ignored'];
+    const inactive = ["false", "else-false", "ignored"];
     return !inactive.includes(top.state);
 };
 /**
@@ -354,8 +357,8 @@ const visibleWidth = (s) => {
     // Strip CSI escapes (ESC `[` … letter). Conservative — only trims the
     // common ANSI color form psql emits. Newlines reset the count.
     // eslint-disable-next-line no-control-regex
-    const stripped = s.replace(/\x1b\[[0-9;]*[A-Za-z]/g, '');
-    const lastNl = stripped.lastIndexOf('\n');
+    const stripped = s.replace(/\x1b\[[0-9;]*[A-Za-z]/g, "");
+    const lastNl = stripped.lastIndexOf("\n");
     const tail = lastNl === -1 ? stripped : stripped.slice(lastNl + 1);
     // Code-unit count — good enough for ASCII prompts; CJK / surrogate-pair
     // width handling is a printer-WP concern.
@@ -364,39 +367,39 @@ const visibleWidth = (s) => {
 const currentDatabase = (ctx) => {
     const db = ctx.settings.db;
     if (!db)
-        return '';
+        return "";
     // Prefer a libpq-style accessor if the runtime Connection exposes one;
     // otherwise fall back to the application's GUC view.
     const meta = db;
-    if (typeof meta.database === 'string' && meta.database.length > 0) {
+    if (typeof meta.database === "string" && meta.database.length > 0) {
         return meta.database;
     }
-    return parameterStatus(ctx, 'database') ?? '';
+    return parameterStatus(ctx, "database") ?? "";
 };
 const currentUser = (ctx) => {
     const db = ctx.settings.db;
     if (!db)
-        return '';
+        return "";
     const meta = db;
-    if (typeof meta.user === 'string' && meta.user.length > 0) {
+    if (typeof meta.user === "string" && meta.user.length > 0) {
         return meta.user;
     }
-    return parameterStatus(ctx, 'session_authorization') ?? '';
+    return parameterStatus(ctx, "session_authorization") ?? "";
 };
 const hostString = (ctx, short) => {
     const db = ctx.settings.db;
     if (!db)
-        return '';
+        return "";
     const meta = db;
-    const host = typeof meta.host === 'string' ? meta.host : '';
-    if (host.length === 0 || host.startsWith('/')) {
+    const host = typeof meta.host === "string" ? meta.host : "";
+    if (host.length === 0 || host.startsWith("/")) {
         // UNIX socket / unknown — upstream emits "[local]" or "[local:path]".
         if (host.length === 0 || short)
-            return '[local]';
+            return "[local]";
         return `[local:${host}]`;
     }
     if (short) {
-        const dot = host.indexOf('.');
+        const dot = host.indexOf(".");
         return dot === -1 ? host : host.slice(0, dot);
     }
     return host;
@@ -404,22 +407,22 @@ const hostString = (ctx, short) => {
 const portString = (ctx) => {
     const db = ctx.settings.db;
     if (!db)
-        return '';
+        return "";
     const meta = db;
-    if (typeof meta.port === 'number' && Number.isFinite(meta.port)) {
+    if (typeof meta.port === "number" && Number.isFinite(meta.port)) {
         return String(meta.port);
     }
-    if (typeof meta.port === 'string' && meta.port.length > 0) {
+    if (typeof meta.port === "string" && meta.port.length > 0) {
         return meta.port;
     }
-    return '';
+    return "";
 };
 const backendPid = (ctx) => {
     const db = ctx.settings.db;
     if (!db)
         return null;
     const meta = db;
-    if (typeof meta.pid === 'number' && Number.isFinite(meta.pid)) {
+    if (typeof meta.pid === "number" && Number.isFinite(meta.pid)) {
         return meta.pid;
     }
     return null;