neonctl 2.27.1 → 2.29.0

package/dist/psql/command/cmd_lo.js

@@ -34,11 +34,11 @@
  * Variable side-effects: `\lo_import` sets `LASTOID` to the new OID
  * (mirrors `do_lo_import`'s `SetVariable(pset.vars, "LASTOID", oidbuf)`).
  */
-import { promises as fsPromises } from 'node:fs';
-import { Buffer } from 'node:buffer';
-import { alignedPrinter } from '../print/aligned.js';
-import { listLargeObjects } from '../describe/queries.js';
-import { writeErr, writeOut } from './shared.js';
+import { Buffer } from "node:buffer";
+import { promises as fsPromises } from "node:fs";
+import { listLargeObjects } from "../describe/queries.js";
+import { alignedPrinter } from "../print/aligned.js";
+import { writeErr, writeOut } from "./shared.js";
 // ---------------------------------------------------------------------------
 // Helpers shared by all four commands
 // ---------------------------------------------------------------------------
@@ -47,8 +47,8 @@ const conn = (ctx) => ctx.settings.db;
 /** Emit "no current connection" error in the psql style. */
 const noConn = (ctx) => {
     writeErr(`\\${ctx.cmdName}: no connection to the server\n`);
-    ctx.settings.lastErrorResult = { message: 'no connection to the server' };
-    return { status: 'error' };
+    ctx.settings.lastErrorResult = { message: "no connection to the server" };
+    return { status: "error" };
 };
 /** Pull the diagnostic-style error message off a thrown value. */
 const errMsg = (err) => err instanceof Error ? err.message : String(err);
@@ -57,7 +57,7 @@ const errMsg = (err) => err instanceof Error ? err.message : String(err);
  * Hex form is unambiguous and works regardless of `bytea_output` or
  * `standard_conforming_strings`.
  */
-const byteaText = (buf) => `\\x${buf.toString('hex')}`;
+const byteaText = (buf) => `\\x${buf.toString("hex")}`;
 /**
  * Parse a string argument as an unsigned 32-bit OID. Returns `null` on
  * malformed input (negative, non-integer, or out of range). Matches the
@@ -79,21 +79,21 @@ const parseOid = (raw) => {
  */
 const cellToString = (v) => {
     if (v === null || v === undefined)
-        return '';
-    if (typeof v === 'string')
+        return "";
+    if (typeof v === "string")
         return v;
     if (Buffer.isBuffer(v))
-        return v.toString('utf-8');
-    if (typeof v === 'number' ||
-        typeof v === 'boolean' ||
-        typeof v === 'bigint') {
+        return v.toString("utf-8");
+    if (typeof v === "number" ||
+        typeof v === "boolean" ||
+        typeof v === "bigint") {
         return String(v);
     }
     try {
         return JSON.stringify(v);
     }
     catch {
-        return '';
+        return "";
     }
 };
 // ---------------------------------------------------------------------------
@@ -126,24 +126,24 @@ const runLoList = async (ctx, verbose) => {
             },
         };
         await alignedPrinter.printQuery(coerced, opts, process.stdout);
-        return { status: 'ok' };
+        return { status: "ok" };
     }
     catch (err) {
         writeErr(`\\${ctx.cmdName}: ${errMsg(err)}\n`);
         ctx.settings.lastErrorResult = { message: errMsg(err) };
-        return { status: 'error' };
+        return { status: "error" };
     }
 };
 /** `\lo_list` — non-verbose listing. */
 export const cmdLoList = {
-    name: 'lo_list',
-    helpKey: 'lo_list',
+    name: "lo_list",
+    helpKey: "lo_list",
     run: (ctx) => runLoList(ctx, false),
 };
 /** `\lo_list+` — verbose listing (adds Access privileges column). */
 export const cmdLoListPlus = {
-    name: 'lo_list+',
-    helpKey: 'lo_list',
+    name: "lo_list+",
+    helpKey: "lo_list",
     run: (ctx) => runLoList(ctx, true),
 };
 // ---------------------------------------------------------------------------
@@ -164,24 +164,26 @@ export const cmdLoListPlus = {
  * Errors fall through to the standard `\lo_import: <msg>` diagnostic.
  */
 export const cmdLoImport = {
-    name: 'lo_import',
-    helpKey: 'lo_import',
+    name: "lo_import",
+    helpKey: "lo_import",
     run: async (ctx) => {
         const c = conn(ctx);
         if (!c)
             return noConn(ctx);
-        const file = ctx.nextArg('normal');
+        const file = ctx.nextArg("normal");
         if (file === null || file.length === 0) {
-            writeErr('\\lo_import: missing required argument\n');
-            ctx.settings.lastErrorResult = { message: 'missing required argument' };
-            return { status: 'error' };
+            writeErr("\\lo_import: missing required argument\n");
+            ctx.settings.lastErrorResult = {
+                message: "missing required argument",
+            };
+            return { status: "error" };
         }
         // Comment is the next lexed slash-arg token, mirroring upstream
         // `do_lo_import`'s second `psql_scan_slash_option(OT_NORMAL)`. Using the
         // lexer (not the raw line) means it picks up the token AFTER the file —
         // `restOfLine()` ignored the read cursor and re-included the filename
         // itself in the comment.
-        const commentRaw = ctx.nextArg('normal');
+        const commentRaw = ctx.nextArg("normal");
         const comment = commentRaw !== null && commentRaw.length > 0 ? commentRaw : null;
         let bytes;
         try {
@@ -190,13 +192,13 @@ export const cmdLoImport = {
         catch (err) {
             writeErr(`\\lo_import: ${errMsg(err)}\n`);
             ctx.settings.lastErrorResult = { message: errMsg(err) };
-            return { status: 'error' };
+            return { status: "error" };
         }
         let oidStr;
         try {
-            const rs = await c.query('SELECT pg_catalog.lo_from_bytea(0, $1::bytea)', [byteaText(bytes)]);
+            const rs = await c.query("SELECT pg_catalog.lo_from_bytea(0, $1::bytea)", [byteaText(bytes)]);
             if (rs.rows.length === 0) {
-                throw new Error('lo_from_bytea returned no rows');
+                throw new Error("lo_from_bytea returned no rows");
             }
             oidStr = cellToString(rs.rows[0][0]);
             if (!/^\d+$/.test(oidStr)) {
@@ -206,7 +208,7 @@ export const cmdLoImport = {
         catch (err) {
             writeErr(`\\lo_import: ${errMsg(err)}\n`);
             ctx.settings.lastErrorResult = { message: errMsg(err) };
-            return { status: 'error' };
+            return { status: "error" };
         }
         if (comment !== null) {
             try {
@@ -215,13 +217,13 @@ export const cmdLoImport = {
             catch (err) {
                 writeErr(`\\lo_import: ${errMsg(err)}\n`);
                 ctx.settings.lastErrorResult = { message: errMsg(err) };
-                return { status: 'error' };
+                return { status: "error" };
             }
         }
         // Side effect: set LASTOID (matches upstream `do_lo_import`).
-        ctx.settings.vars.set('LASTOID', oidStr);
+        ctx.settings.vars.set("LASTOID", oidStr);
         writeOut(`lo_import ${oidStr}\n`);
-        return { status: 'ok' };
+        return { status: "ok" };
     },
 };
 // ---------------------------------------------------------------------------
@@ -239,30 +241,34 @@ export const cmdLoImport = {
  * Print `lo_export\n` on success — matches upstream `do_lo_export`.
  */
 export const cmdLoExport = {
-    name: 'lo_export',
-    helpKey: 'lo_export',
+    name: "lo_export",
+    helpKey: "lo_export",
     run: async (ctx) => {
         const c = conn(ctx);
         if (!c)
             return noConn(ctx);
-        const oidArg = ctx.nextArg('normal');
-        const file = ctx.nextArg('normal');
+        const oidArg = ctx.nextArg("normal");
+        const file = ctx.nextArg("normal");
         if (oidArg === null || file === null || file.length === 0) {
-            writeErr('\\lo_export: missing required argument\n');
-            ctx.settings.lastErrorResult = { message: 'missing required argument' };
-            return { status: 'error' };
+            writeErr("\\lo_export: missing required argument\n");
+            ctx.settings.lastErrorResult = {
+                message: "missing required argument",
+            };
+            return { status: "error" };
         }
         const oid = parseOid(oidArg);
         if (oid === null) {
             writeErr(`\\lo_export: "${oidArg}" is not a valid large object OID\n`);
-            ctx.settings.lastErrorResult = { message: 'invalid OID' };
-            return { status: 'error' };
+            ctx.settings.lastErrorResult = { message: "invalid OID" };
+            return { status: "error" };
         }
         let bytes;
         try {
-            const rs = await c.query('SELECT pg_catalog.lo_get($1::oid)', [oid]);
+            const rs = await c.query("SELECT pg_catalog.lo_get($1::oid)", [
+                oid,
+            ]);
             if (rs.rows.length === 0) {
-                throw new Error('lo_get returned no rows');
+                throw new Error("lo_get returned no rows");
             }
             const cell = rs.rows[0][0];
             bytes = coerceBytea(cell);
@@ -270,7 +276,7 @@ export const cmdLoExport = {
         catch (err) {
             writeErr(`\\lo_export: ${errMsg(err)}\n`);
             ctx.settings.lastErrorResult = { message: errMsg(err) };
-            return { status: 'error' };
+            return { status: "error" };
         }
         try {
             await fsPromises.writeFile(file, bytes);
@@ -278,10 +284,10 @@ export const cmdLoExport = {
         catch (err) {
             writeErr(`\\lo_export: ${errMsg(err)}\n`);
             ctx.settings.lastErrorResult = { message: errMsg(err) };
-            return { status: 'error' };
+            return { status: "error" };
         }
-        writeOut('lo_export\n');
-        return { status: 'ok' };
+        writeOut("lo_export\n");
+        return { status: "ok" };
     },
 };
 /**
@@ -295,19 +301,19 @@ export const cmdLoExport = {
 const coerceBytea = (cell) => {
     if (Buffer.isBuffer(cell))
         return cell;
-    if (typeof cell !== 'string') {
+    if (typeof cell !== "string") {
         throw new Error(`lo_get returned unexpected cell type: ${typeof cell}`);
     }
-    if (cell.startsWith('\\x')) {
-        return Buffer.from(cell.slice(2), 'hex');
+    if (cell.startsWith("\\x")) {
+        return Buffer.from(cell.slice(2), "hex");
     }
     // Legacy octal-escape decode (`\\\\NNN` → byte, `\\\\` → `\\`, others
     // pass through). Upstream `PQunescapeBytea` does the same.
     const out = [];
     let i = 0;
     while (i < cell.length) {
-        if (cell[i] === '\\') {
-            if (cell[i + 1] === '\\') {
+        if (cell[i] === "\\") {
+            if (cell[i + 1] === "\\") {
                 out.push(0x5c);
                 i += 2;
                 continue;
@@ -335,34 +341,36 @@ const coerceBytea = (cell) => {
  * Print `lo_unlink <oid>\n` on success.
  */
 export const cmdLoUnlink = {
-    name: 'lo_unlink',
-    helpKey: 'lo_unlink',
+    name: "lo_unlink",
+    helpKey: "lo_unlink",
     run: async (ctx) => {
         const c = conn(ctx);
         if (!c)
             return noConn(ctx);
-        const oidArg = ctx.nextArg('normal');
+        const oidArg = ctx.nextArg("normal");
         if (oidArg === null) {
-            writeErr('\\lo_unlink: missing required argument\n');
-            ctx.settings.lastErrorResult = { message: 'missing required argument' };
-            return { status: 'error' };
+            writeErr("\\lo_unlink: missing required argument\n");
+            ctx.settings.lastErrorResult = {
+                message: "missing required argument",
+            };
+            return { status: "error" };
         }
         const oid = parseOid(oidArg);
         if (oid === null) {
             writeErr(`\\lo_unlink: "${oidArg}" is not a valid large object OID\n`);
-            ctx.settings.lastErrorResult = { message: 'invalid OID' };
-            return { status: 'error' };
+            ctx.settings.lastErrorResult = { message: "invalid OID" };
+            return { status: "error" };
         }
         try {
-            await c.query('SELECT pg_catalog.lo_unlink($1::oid)', [oid]);
+            await c.query("SELECT pg_catalog.lo_unlink($1::oid)", [oid]);
         }
         catch (err) {
             writeErr(`\\lo_unlink: ${errMsg(err)}\n`);
             ctx.settings.lastErrorResult = { message: errMsg(err) };
-            return { status: 'error' };
+            return { status: "error" };
         }
         writeOut(`lo_unlink ${String(oid)}\n`);
-        return { status: 'ok' };
+        return { status: "ok" };
     },
 };
 // ---------------------------------------------------------------------------