neonctl 2.27.1 → 2.29.0

package/dist/psql/io/psqlrc.js

@@ -29,19 +29,19 @@
  * primitive and delete this duplication. Tracked TODO at the bottom of this
  * file.
  */
-import { promises as fs } from 'node:fs';
-import * as path from 'node:path';
-import { initialScanState } from '../types/scanner.js';
-import { scanSql } from '../scanner/sql.js';
-import { scanSlashArgs } from '../scanner/slash.js';
-import { dispatchBackslash } from '../command/dispatch.js';
-import { attachCondStack, COND_COMMAND_NAMES } from '../command/cmd_cond.js';
-import { sendQuery } from '../core/common.js';
+import { promises as fs } from "node:fs";
+import * as path from "node:path";
+import { attachCondStack, COND_COMMAND_NAMES } from "../command/cmd_cond.js";
+import { dispatchBackslash } from "../command/dispatch.js";
+import { sendQuery } from "../core/common.js";
+import { scanSlashArgs } from "../scanner/slash.js";
+import { scanSql } from "../scanner/sql.js";
+import { initialScanState } from "../types/scanner.js";
 // ---------------------------------------------------------------------------
 // Path discovery.
 // ---------------------------------------------------------------------------
-const isWindows = process.platform === 'win32';
-const baseRcName = isWindows ? 'psqlrc.conf' : '.psqlrc';
+const isWindows = process.platform === "win32";
+const baseRcName = isWindows ? "psqlrc.conf" : ".psqlrc";
 /**
  * Return the default user-level psqlrc path. The path is *not* verified to
  * exist — callers stat it lazily. The Windows layout follows upstream:
@@ -51,11 +51,11 @@ export const defaultPsqlrcPath = (env = process.env) => {
     if (isWindows) {
         const appData = env.APPDATA;
         if (appData && appData.length > 0) {
-            return path.join(appData, 'postgresql', baseRcName);
+            return path.join(appData, "postgresql", baseRcName);
         }
-        return path.join(env.USERPROFILE ?? '', 'postgresql', baseRcName);
+        return path.join(env.USERPROFILE ?? "", "postgresql", baseRcName);
     }
-    const home = env.HOME ?? '';
+    const home = env.HOME ?? "";
     return path.join(home, baseRcName);
 };
 const versionSuffix = (serverVersion) => {
@@ -73,61 +73,61 @@ export const psqlrcCandidates = (env, serverVersion) => {
     // System-wide.
     const sysDir = env.PGSYSCONFDIR;
     if (sysDir && sysDir.length > 0) {
-        const sysBase = isWindows ? 'psqlrc.conf' : 'psqlrc';
+        const sysBase = isWindows ? "psqlrc.conf" : "psqlrc";
         if (suffix) {
             out.push({
                 path: path.join(sysDir, `${sysBase}-${suffix}`),
-                description: 'system psqlrc (versioned)',
+                description: "system psqlrc (versioned)",
             });
         }
         out.push({
             path: path.join(sysDir, sysBase),
-            description: 'system psqlrc',
+            description: "system psqlrc",
         });
     }
     // PSQLRC override suppresses HOME-based discovery (matches upstream).
     const envRc = env.PSQLRC;
-    if (envRc !== undefined && envRc !== '') {
-        out.push({ path: expandTilde(envRc, env), description: '$PSQLRC' });
+    if (envRc !== undefined && envRc !== "") {
+        out.push({ path: expandTilde(envRc, env), description: "$PSQLRC" });
         return out;
     }
     // HOME-based.
-    const home = env.HOME ?? '';
+    const home = env.HOME ?? "";
     if (isWindows) {
         const appData = env.APPDATA;
         const dir = appData && appData.length > 0
-            ? path.join(appData, 'postgresql')
-            : path.join(env.USERPROFILE ?? '', 'postgresql');
+            ? path.join(appData, "postgresql")
+            : path.join(env.USERPROFILE ?? "", "postgresql");
         if (suffix) {
             out.push({
                 path: path.join(dir, `${baseRcName}-${suffix}`),
-                description: 'user psqlrc (versioned)',
+                description: "user psqlrc (versioned)",
             });
         }
         out.push({
             path: path.join(dir, baseRcName),
-            description: 'user psqlrc',
+            description: "user psqlrc",
         });
     }
     else if (home.length > 0) {
         if (suffix) {
             out.push({
                 path: path.join(home, `${baseRcName}-${suffix}`),
-                description: 'user psqlrc (versioned)',
+                description: "user psqlrc (versioned)",
             });
         }
         out.push({
             path: path.join(home, baseRcName),
-            description: 'user psqlrc',
+            description: "user psqlrc",
         });
     }
     return out;
 };
 const expandTilde = (p, env) => {
-    if (p.startsWith('~/') && env.HOME) {
+    if (p.startsWith("~/") && env.HOME) {
         return path.join(env.HOME, p.slice(2));
     }
-    if (p === '~' && env.HOME) {
+    if (p === "~" && env.HOME) {
         return env.HOME;
     }
     return p;
@@ -137,11 +137,11 @@ const expandTilde = (p, env) => {
 // ---------------------------------------------------------------------------
 const readIfExists = async (file) => {
     try {
-        return await fs.readFile(file, 'utf8');
+        return await fs.readFile(file, "utf8");
     }
     catch (err) {
         const code = err.code;
-        if (code === 'ENOENT' || code === 'ENOTDIR')
+        if (code === "ENOENT" || code === "ENOTDIR")
             return null;
         // Permission / I/O errors propagate — they're worth surfacing because
         // they signal a likely misconfiguration. Mirror upstream which prints
@@ -166,7 +166,7 @@ const makeBackslashContext = (ctx, cmdName, rawArgs, queryBuf) => {
         cmdName,
         queryBuf,
         rawArgs,
-        nextArg(mode = 'normal') {
+        nextArg(mode = "normal") {
             const av = argsFor(mode);
             const idx = cursors.get(mode) ?? 0;
             if (idx >= av.length)
@@ -195,7 +195,7 @@ const makeBackslashContext = (ctx, cmdName, rawArgs, queryBuf) => {
  */
 export const executeInputString = async (input, ctx, opts = {}) => {
     let working = input;
-    let queryBuf = '';
+    let queryBuf = "";
     let scanState = initialScanState();
     let hadError = false;
     let stoppedOnError = false;
@@ -203,7 +203,7 @@ export const executeInputString = async (input, ctx, opts = {}) => {
     const print = opts.print ?? false;
     const noteConnectionLost = () => {
         if (ctx.settings.db?.isClosed()) {
-            ctx.stderr.write('psql: error: connection to server was lost\n');
+            ctx.stderr.write("psql: error: connection to server was lost\n");
             connectionLost = true;
             return true;
         }
@@ -217,11 +217,11 @@ export const executeInputString = async (input, ctx, opts = {}) => {
     while (working.length > 0) {
         const r = scanSql(working, scanState, sqlVarLookup);
         scanState = r.nextState;
-        if (r.kind === 'semicolon') {
+        if (r.kind === "semicolon") {
             // Use the substituted text rather than the raw slice so `:NAME`
             // expansions land in the executed SQL.
             const sqlText = queryBuf + r.sql;
-            queryBuf = '';
+            queryBuf = "";
             working = working.slice(r.consumed);
             scanState = initialScanState();
             if (!ctx.cond.isActive())
@@ -269,7 +269,7 @@ export const executeInputString = async (input, ctx, opts = {}) => {
             }
             continue;
         }
-        if (r.kind === 'backslash') {
+        if (r.kind === "backslash") {
             // Use `r.sql` — the scanner's substituted/processed text that
             // *preceded* the backslash boundary. The raw consumed slice would
             // also include the backslash command itself and any separators
@@ -301,26 +301,26 @@ export const executeInputString = async (input, ctx, opts = {}) => {
                 }
                 continue;
             }
-            if (res.status === 'exit')
+            if (res.status === "exit")
                 return { hadError, stoppedOnError, connectionLost };
-            if (res.status === 'reset-buf') {
-                queryBuf = res.newBuf ?? '';
+            if (res.status === "reset-buf") {
+                queryBuf = res.newBuf ?? "";
                 scanState = initialScanState();
             }
             // Mirror upstream `mainloop.c`: a backslash command that errored
             // drops the query buffer and resets the scanner. Without this the
             // residue would re-execute via the tail-dispatch at EOF (or get
             // glued onto the next dispatched statement), masking the failure.
-            if (res.status === 'error') {
-                queryBuf = '';
+            if (res.status === "error") {
+                queryBuf = "";
                 scanState = initialScanState();
             }
-            if (res.status === 'error')
+            if (res.status === "error")
                 hadError = true; // latch
             if (noteConnectionLost()) {
                 return { hadError, stoppedOnError, connectionLost };
             }
-            if (res.status === 'error' && ctx.settings.onErrorStop) {
+            if (res.status === "error" && ctx.settings.onErrorStop) {
                 stoppedOnError = true;
                 return { hadError, stoppedOnError, connectionLost };
             }
@@ -331,7 +331,7 @@ export const executeInputString = async (input, ctx, opts = {}) => {
         // substituted form. (Same caveat as the mainloop: cross-chunk colon
         // references fall back to the literal.)
         queryBuf += r.sql;
-        working = '';
+        working = "";
     }
     // Tail dispatch: if the file ended mid-statement (no trailing `;`), run
     // the residue. This matches `process_file` behaviour in upstream.
@@ -376,7 +376,7 @@ export const loadPsqlrc = async (ctx, opts = {}) => {
         if (content === null)
             return;
         const prevSource = ctx.settings.curCmdSource;
-        ctx.settings.curCmdSource = 'rcfile';
+        ctx.settings.curCmdSource = "rcfile";
         try {
             await executeInputString(content, ctx);
         }
@@ -394,14 +394,14 @@ export const loadPsqlrc = async (ctx, opts = {}) => {
     // clobber the versioned one.
     let systemDone = false;
     for (const c of candidates) {
-        const isSystem = c.description.startsWith('system');
+        const isSystem = c.description.startsWith("system");
         if (isSystem && systemDone)
             continue; // first system file already ran
         const content = await readIfExists(c.path);
         if (content === null)
             continue;
         const prevSource = ctx.settings.curCmdSource;
-        ctx.settings.curCmdSource = 'rcfile';
+        ctx.settings.curCmdSource = "rcfile";
         try {
             await executeInputString(content, ctx);
         }