@batadata/cli 0.2.5 → 0.2.6

package/dist/commands/powdb.js

@@ -1,4 +1,4 @@
-import { writeFileSync } from "node:fs";
+import { readFileSync, writeFileSync } from "node:fs";
 import * as path from "node:path";
 import { api, apiError } from "../api.js";
 import { requireToken, isJsonMode, loadConfig } from "../config.js";
@@ -683,12 +683,12 @@ async function powdbQuery(args) {
     const projectId = resolvePowdbProject(args);
     const res = await api.post(`/v1/powdb/${encodeURIComponent(projectId)}/query`, { query }, token);
     if (!res.ok) {
-        const code = res.status === 503 ? "RETRYABLE"
+        const code = res.status === 503 ? "COMPUTE_STARTING"
             : res.status === 401 || res.status === 403 ? "INVALID_KEY"
                 : res.status === 404 ? "NOT_FOUND"
                     : res.status >= 500 || res.status === 0 ? "API_UNAVAILABLE"
                         : "CLI_ERROR";
-        emitError(code, apiError(res, "PowQL query failed"), code === "RETRYABLE" ? "The database is starting — retry in a few seconds." : "");
+        emitError(code, apiError(res, "PowQL query failed"), code === "COMPUTE_STARTING" ? "The database is starting — retry in a few seconds." : "");
         return;
     }
     const data = res.data;
@@ -729,6 +729,91 @@ async function powdbQuery(args) {
     }
     log();
 }
+function truncateStatement(statement) {
+    const flat = statement.replace(/\s+/g, " ").trim();
+    return flat.length > 60 ? `${flat.slice(0, 57)}...` : flat;
+}
+async function powdbExec(args) {
+    const token = requireToken();
+    const jsonMode = isJsonMode();
+    const transactional = args.includes("--transactional");
+    const continueOnError = args.includes("--continue-on-error");
+    if (transactional && continueOnError) {
+        emitError("INVALID_ARG", "--transactional and --continue-on-error are mutually exclusive.", "Transactional scripts stop (and roll back) on the first failure by definition.");
+        return;
+    }
+    let filePath;
+    for (let i = 0; i < args.length; i++) {
+        if (args[i] === "--file" && args[i + 1])
+            filePath = args[i + 1];
+        if (args[i].startsWith("--file="))
+            filePath = args[i].slice("--file=".length);
+    }
+    let script;
+    if (filePath) {
+        try {
+            // `--file -` reads stdin (fd 0) — the pipe-friendly agent path.
+            script = readFileSync(filePath === "-" ? 0 : filePath, "utf8");
+        }
+        catch (err) {
+            emitError("FILE_ERROR", `Could not read ${filePath}: ${err instanceof Error ? err.message : String(err)}`, "");
+            return;
+        }
+    }
+    else {
+        script = args.find((a) => !a.startsWith("-") && a !== powdbProjectFlag(args) && a !== filePath);
+    }
+    if (!script || script.trim().length === 0) {
+        emitError("MISSING_ARG", 'Usage: bata powdb exec "<PowQL script>" | --file <path|-> [--transactional | --continue-on-error] [--project <id>]', "Statements are ;-separated (string/#-comment aware) and run pipelined on one connection.");
+        return;
+    }
+    const projectId = resolvePowdbProject(args);
+    const res = await api.post(`/v1/powdb/${encodeURIComponent(projectId)}/exec`, {
+        script,
+        ...(transactional ? { transactional: true } : {}),
+        ...(continueOnError ? { continue_on_error: true } : {}),
+    }, token);
+    if (!res.ok) {
+        const code = res.status === 503 ? "COMPUTE_STARTING"
+            : res.status === 401 || res.status === 403 ? "INVALID_KEY"
+                : res.status === 404 ? "NOT_FOUND"
+                    : res.status >= 500 || res.status === 0 ? "API_UNAVAILABLE"
+                        : "CLI_ERROR";
+        emitError(code, apiError(res, "PowQL script failed"), code === "COMPUTE_STARTING" ? "The database is starting — retry in a few seconds." : "");
+        return;
+    }
+    const data = res.data;
+    if (jsonMode) {
+        json(data);
+        if (("error" in data && data.error) || ("failed_count" in data && data.failed_count > 0)) {
+            process.exitCode = 1;
+        }
+        return;
+    }
+    if ("error" in data && data.error) {
+        const where = data.statement_index !== undefined
+            ? ` (statement ${data.statement_index + 1}${data.rolled_back ? ", rolled back — nothing persisted" : ""})`
+            : "";
+        emitError("QUERY_ERROR", `${data.error}${where}`, data.statement ?? "");
+        return;
+    }
+    log();
+    if ("outcomes" in data) {
+        for (const o of data.outcomes) {
+            if (o.ok)
+                log(`  ${colors.green(">")} ok  ${colors.dim(truncateStatement(o.statement))}`);
+            else
+                log(`  ${colors.red("x")} ${o.error}  ${colors.dim(truncateStatement(o.statement))}`);
+        }
+        log(`  ${colors.dim(`${data.statement_count} statement${data.statement_count === 1 ? "" : "s"}, ${data.failed_count} failed, ${data.duration_ms}ms`)}`);
+        if (data.failed_count > 0)
+            process.exitCode = 1;
+    }
+    else if ("results" in data) {
+        log(`  ${colors.green(">")} ${data.statement_count} statement${data.statement_count === 1 ? "" : "s"} executed${transactional ? " (transactional)" : ""} ${colors.dim(`${data.duration_ms}ms`)}`);
+    }
+    log();
+}
 async function powdbLifecycle(action, args) {
     const token = requireToken();
     const jsonMode = isJsonMode();
@@ -766,6 +851,9 @@ function powdbHelp() {
     log();
     log(`  ${colors.dim("Hosted PowDB projects (create one with: bata projects create --engine powdb):")}`);
     log(`    ${colors.cyan("query")}   Run a PowQL statement:  bata powdb query "Note" [--project <id>]`);
+    log(`    ${colors.cyan("exec")}    Run a multi-statement script, pipelined (bulk load / seed / migrate):`);
+    log(`    ${colors.dim('            bata powdb exec --file seed.powql [--transactional | --continue-on-error]')}`);
+    log(`    ${colors.dim('            bata powdb pull --out - | bata powdb exec --file -   (pipe-friendly)')}`);
     log(`    ${colors.cyan("status")}  Show the server state (running | parked)  [--project <id>]`);
     log(`    ${colors.cyan("park")}    Scale the server to zero (WAL is durable)  [--project <id>]`);
     log(`    ${colors.cyan("wake")}    Start a parked server (~tens of ms)  [--project <id>]`);
@@ -796,6 +884,8 @@ export async function handlePowdb(args) {
             return powdbPull(args.slice(1));
         case "query":
             return powdbQuery(args.slice(1));
+        case "exec":
+            return powdbExec(args.slice(1));
         case "status":
             return powdbLifecycle("status", args.slice(1));
         case "park":
@@ -806,6 +896,6 @@ export async function handlePowdb(args) {
             powdbHelp();
             return;
         default:
-            emitError("INVALID_FLAG", `Unknown subcommand: powdb ${sub}`, "Available: pull, query, status, park, wake. Run `bata powdb --help`.");
+            emitError("INVALID_FLAG", `Unknown subcommand: powdb ${sub}`, "Available: pull, query, exec, status, park, wake. Run `bata powdb --help`.");
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@batadata/cli",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "description": "CLI for BataDB — serverless Postgres platform",
   "bin": {
     "bata": "./dist/index.js"