ardent-cli 0.0.28 → 0.0.29

package/dist/index.js

@@ -343,6 +343,15 @@ var ApiClient = class {
     });
     return this.handleResponse(response);
   }
+  async put(path, body) {
+    const url = `${getApiUrl()}${path}`;
+    const response = await fetch(url, {
+      method: "PUT",
+      headers: this.getHeaders(),
+      body: JSON.stringify(body)
+    });
+    return this.handleResponse(response);
+  }
 };
 var api = new ApiClient();
 function isNetworkError(err) {
@@ -987,6 +996,52 @@ Example:
   }
 }
 
+// src/lib/prompt.ts
+import { createInterface } from "readline/promises";
+function isTtyInteractive() {
+  return Boolean(process.stdin.isTTY && process.stdout.isTTY);
+}
+async function readOneLine(question) {
+  const rl = createInterface({
+    input: process.stdin,
+    output: process.stdout
+  });
+  try {
+    const answer = await rl.question(question);
+    return answer.trim();
+  } finally {
+    rl.close();
+  }
+}
+async function confirm(question, options = {}) {
+  if (!isTtyInteractive()) {
+    return false;
+  }
+  const defaultYes = options.defaultYes ?? true;
+  const suffix = defaultYes ? "[Y/n] " : "[y/N] ";
+  const answer = (await readOneLine(`${question} ${suffix}`)).toLowerCase();
+  if (answer === "") return defaultYes;
+  if (answer === "y" || answer === "yes") return true;
+  if (answer === "n" || answer === "no") return false;
+  const retry = (await readOneLine(`Please answer yes or no. ${suffix}`)).toLowerCase();
+  if (retry === "") return defaultYes;
+  if (retry === "y" || retry === "yes") return true;
+  if (retry === "n" || retry === "no") return false;
+  console.log('Unrecognised input; treating as "no".');
+  return false;
+}
+async function multiSelect(items, promptForItem, options = {}) {
+  if (!isTtyInteractive()) {
+    return [];
+  }
+  const accepted = [];
+  for (const item of items) {
+    const yes = await confirm(promptForItem(item), { defaultYes: options.defaultYes });
+    if (yes) accepted.push(item);
+  }
+  return accepted;
+}
+
 // src/commands/connector/create.ts
 function parsePostgresUrl(url) {
   const atIndex = url.lastIndexOf("@");
@@ -1007,6 +1062,57 @@ function printEngineSetupRecoveryHint(connectorName) {
   console.error("  Inspect: ardent connector list");
   console.error(`  Retry:   ardent connector retry-setup ${connectorName}`);
 }
+async function promptForUnsupportedExtensions(connectorId, unsupported, alreadyPersisted) {
+  function mergeAllowlist(newlyAccepted) {
+    return Array.from(/* @__PURE__ */ new Set([...alreadyPersisted, ...newlyAccepted])).sort();
+  }
+  console.log("");
+  if (unsupported.length === 1) {
+    const ext = unsupported[0];
+    console.log(
+      `Found 1 extension on your source database that isn't supported on branches:`
+    );
+    console.log(`  \u2022 ${ext}`);
+    console.log("");
+    console.log(
+      "Branches from this connector will fail unless this extension is excluded."
+    );
+    const yes = await confirm(`Branch without ${ext}?`, { defaultYes: true });
+    if (!yes) return "aborted";
+    const merged2 = mergeAllowlist([ext]);
+    await api.put(`/v1/connectors/${connectorId}`, {
+      drop_extensions: merged2
+    });
+    console.log(`\u2713 Saved drop selection: ${merged2.join(", ")}`);
+    return "applied";
+  }
+  console.log(
+    `Found ${unsupported.length} extensions on your source database that aren't supported on branches:`
+  );
+  for (const ext of unsupported) {
+    console.log(`  \u2022 ${ext}`);
+  }
+  console.log("");
+  console.log(
+    "Pick which extensions to exclude from branches. Branches from this connector"
+  );
+  console.log(
+    "will fail unless every unsupported extension you keep is configured by support."
+  );
+  console.log("");
+  const accepted = await multiSelect(
+    unsupported,
+    (ext) => `Branch without ${ext}?`,
+    { defaultYes: true }
+  );
+  if (accepted.length === 0) return "aborted";
+  const merged = mergeAllowlist(accepted);
+  await api.put(`/v1/connectors/${connectorId}`, {
+    drop_extensions: merged
+  });
+  console.log(`\u2713 Saved drop selection: ${merged.join(", ")}`);
+  return "applied";
+}
 async function createAction2(type, url, options) {
   const supportedTypes = ["postgresql"];
   if (!supportedTypes.includes(type.toLowerCase())) {
@@ -1145,6 +1251,36 @@ async function createAction2(type, url, options) {
         selected_paths: ["*"]
       });
       const connector = await api.get(`/v1/connectors/${connectorId}`);
+      const preview = connector.unsupported_extensions_preview ?? null;
+      const unsupportedExtensions = preview?.unsupported ?? [];
+      const dropExtensionsPersisted = preview?.drop_extensions_persisted ?? [];
+      if (unsupportedExtensions.length > 0) {
+        const promptOutcome = await promptForUnsupportedExtensions(
+          connectorId,
+          unsupportedExtensions,
+          dropExtensionsPersisted
+        );
+        if (promptOutcome === "aborted") {
+          trackEvent("CLI: connector create aborted", {
+            reason: "declined_extension_drops",
+            unsupported_count: unsupportedExtensions.length
+          });
+          console.error(
+            "\u2717 Connector setup aborted because no extension drop selection was made."
+          );
+          console.error(
+            "  Branches cannot be created until the connector has a drop selection."
+          );
+          console.error("  Run this when ready:");
+          console.error(
+            `    ardent connector update ${connectorName} --drop-extensions <ext,...>`
+          );
+          console.error(
+            `    ardent connector retry-setup ${connectorName}`
+          );
+          process.exit(1);
+        }
+      }
       if (connector.branching_engine_status === "configuration_verified") {
         console.log("Setting up branching engine...");
         try {
@@ -1535,6 +1671,86 @@ async function switchAction2(name) {
   trackEvent("CLI: connector switch");
 }
 
+// src/lib/drop_extensions.ts
+function parseDropExtensions(raw) {
+  if (raw === void 0) return null;
+  if (raw.trim() === "") return [];
+  return raw.split(",").map((entry) => entry.trim()).filter((entry) => entry.length > 0);
+}
+
+// src/commands/connector/update.ts
+async function resolveConnectorByName(name) {
+  const cached = getCacheEntry("connectors");
+  let connector = cached?.data.find((c) => c.name === name);
+  if (!connector) {
+    try {
+      const result = await api.get(
+        "/v1/cli/connectors"
+      );
+      if (result.connectors) {
+        setCacheEntry("connectors", result.connectors);
+        connector = result.connectors.find((c) => c.name === name);
+      }
+    } catch (err) {
+      if (isNetworkError(err)) {
+        console.error("\u2717 Connector not found in cache and offline");
+        process.exit(1);
+      }
+      throw err;
+    }
+  }
+  if (!connector) {
+    console.error(`\u2717 Connector "${name}" not found`);
+    console.log("  Run: ardent connector list");
+    process.exit(1);
+  }
+  return connector;
+}
+async function updateAction(name, options = {}) {
+  const dropExtensions = parseDropExtensions(options.dropExtensions);
+  if (dropExtensions === null) {
+    console.error("\u2717 Nothing to update.");
+    console.error("  Try: ardent connector update <name> --drop-extensions <ext,...>");
+    process.exit(1);
+  }
+  const connector = await resolveConnectorByName(name);
+  try {
+    const payload = { drop_extensions: dropExtensions };
+    await api.put(`/v1/connectors/${connector.id}`, payload);
+    trackEvent("CLI: connector update succeeded", {
+      drop_extensions_count: dropExtensions.length
+    });
+    if (dropExtensions.length === 0) {
+      console.log(`\u2713 Cleared extension drop allowlist on '${name}'`);
+    } else {
+      console.log(
+        `\u2713 Updated extension drop allowlist on '${name}': ${dropExtensions.join(", ")}`
+      );
+      console.log(
+        "  These extensions will be omitted from future branches created from this connector."
+      );
+    }
+  } catch (err) {
+    if (isPermissionError(err)) {
+      trackEvent("CLI: connector update failed", { reason: "permission_denied" });
+      console.error("\u2717 You don't have permission to update this connector.");
+      console.error("");
+      console.error("  Ask your organization admin to either:");
+      console.error("  \u2022 Update the connector for you, or");
+      console.error("  \u2022 Upgrade your role to Admin");
+      process.exit(1);
+    }
+    if (isNetworkError(err)) {
+      trackEvent("CLI: connector update failed", { reason: "offline" });
+      console.error("\u2717 Cannot update connector while offline");
+      process.exit(1);
+    }
+    trackEvent("CLI: connector update failed", { reason: "api_error" });
+    console.error("\u2717 Failed:", err instanceof Error ? err.message : err);
+    process.exit(1);
+  }
+}
+
 // src/commands/connector/index.ts
 var connectorCommand = new Command2("connector").description("Manage database connectors");
 connectorCommand.command("create <type> [url]").description("Create a new connector").option("-n, --name <name>", "Connector name").option("--byoc <provider>", "Bring your own Neon project (e.g. neon)").option("--api-key <key>", "Neon API key (required with --byoc)").option("--project-id <id>", "Neon project ID (required with --byoc)").option(
@@ -1558,6 +1774,10 @@ connectorCommand.command("switch <name>").description("Switch to a different con
 connectorCommand.command("retry-setup <name>").description(
   "Retry branching engine setup for a connector that didn't finish (status: configuration_verified or validating)"
 ).action(retrySetupAction);
+connectorCommand.command("update <name>").description("Update a connector's configuration").option(
+  "--drop-extensions <list>",
+  "Comma-separated list of source extensions to drop on branches (e.g. pgmq,supabase_vault). Pass an empty string to clear the allowlist."
+).action(updateAction);
 connectorCommand.command("delete <name>").description("Delete a connector by name").option(
   "--force",
   "Skip the in-flight WAL/Kafka drain wait. Any un-replicated changes are abandoned and recorded for operator triage."

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ardent-cli",
-  "version": "0.0.28",
+  "version": "0.0.29",
   "description": "Git for Data infrastructure",
   "type": "module",
   "bin": {