Remove --location from apphosting:backends:delete

Author: annajowang

src/apphosting/backend.ts

@@ -429,6 +429,64 @@ export async function getBackendForLocation(
   }
 }
 
+/**
+ * Fetches backends of the given backendId and lets the user choose if more than one is found.
+ */
+export async function chooseBackends(
+  projectId: string,
+  backendId: string,
+  chooseBackendPrompt: string,
+  force?: boolean,
+): Promise<apphosting.Backend[]> {
+  let { unreachable, backends } = await apphosting.listBackends(projectId, "-");
+  if (unreachable && unreachable.length !== 0) {
+    logWarning(
+      `The following locations are currently unreachable: ${unreachable.join(",")}.\n` +
+        "If your backend is in one of these regions, please try again later.",
+    );
+  }
+  backends = backends.filter(
+    (backend) => apphosting.parseBackendName(backend.name).id === backendId,
+  );
+  if (backends.length === 0) {
+    throw new FirebaseError(`No backend named "${backendId}" found.`);
+  }
+  if (backends.length === 1) {
+    return backends;
+  }
+
+  if (force) {
+    throw new FirebaseError(
+      `Force cannot be used because multiple backends were found with ID ${backendId}`,
+    );
+  }
+  const backendsByDisplay = new Map<string, apphosting.Backend>();
+  backends.forEach((backend) => {
+    const { location, id } = apphosting.parseBackendName(backend.name);
+    backendsByDisplay.set(`${id}(${location})`, backend);
+  });
+  const chosenBackendDisplays = await promptOnce({
+    name: "backend",
+    type: "checkbox",
+    message: chooseBackendPrompt,
+    choices: Array.from(backendsByDisplay.keys(), (name) => {
+      return {
+        checked: false,
+        name: name,
+        value: name,
+      };
+    }),
+  });
+  const chosenBackends: apphosting.Backend[] = [];
+  chosenBackendDisplays.forEach((backendDisplay) => {
+    const backend = backendsByDisplay.get(backendDisplay);
+    if (backend !== undefined) {
+      chosenBackends.push(backend);
+    }
+  });
+  return chosenBackends;
+}
+
 /**
  * Fetches a backend from the server. If there are multiple backends with that name (ie multi-regional backends),
  * prompts the user to disambiguate. If the force option is specified and multiple backends have the same name,

src/commands/apphosting-backends-delete.ts

@@ -6,35 +6,25 @@ import { promptOnce } from "../prompt";
 import * as utils from "../utils";
 import * as apphosting from "../gcp/apphosting";
 import { printBackendsTable } from "./apphosting-backends-list";
-import {
-  deleteBackendAndPoll,
-  getBackendForAmbiguousLocation,
-  getBackendForLocation,
-} from "../apphosting/backend";
+import { deleteBackendAndPoll, chooseBackends } from "../apphosting/backend";
 import * as ora from "ora";
 
 export const command = new Command("apphosting:backends:delete <backend>")
   .description("delete a Firebase App Hosting backend")
-  .option("-l, --location <location>", "specify the location of the backend", "-")
   .withForce()
   .before(apphosting.ensureApiEnabled)
   .action(async (backendId: string, options: Options) => {
     const projectId = needProjectId(options);
-    let location = options.location as string;
-    let backend: apphosting.Backend;
-    if (location === "-" || location === "") {
-      backend = await getBackendForAmbiguousLocation(
-        projectId,
-        backendId,
-        "Please select the location of the backend you'd like to delete:",
-      );
-      location = apphosting.parseBackendName(backend.name).location;
-    } else {
-      backend = await getBackendForLocation(projectId, location, backendId);
-    }
 
-    utils.logWarning("You are about to permanently delete this backend:");
-    printBackendsTable([backend]);
+    const backends = await chooseBackends(
+      projectId,
+      backendId,
+      "Please select the backends you'd like to delete:",
+      options.force,
+    );
+
+    utils.logWarning("You are about to permanently delete these backend(s):");
+    printBackendsTable(backends);
 
     const confirmDeletion = await promptOnce(
       {
@@ -49,14 +39,20 @@ export const command = new Command("apphosting:backends:delete <backend>")
       return;
     }
 
-    const spinner = ora("Deleting backend...").start();
-    try {
-      await deleteBackendAndPoll(projectId, location, backendId);
-      spinner.succeed(`Successfully deleted the backend: ${backendId}`);
-    } catch (err: unknown) {
-      spinner.stop();
-      throw new FirebaseError(`Failed to delete backend: ${backendId}.`, {
-        original: getError(err),
-      });
+    for (const b of backends) {
+      const { location, id } = apphosting.parseBackendName(b.name);
+      const spinner = ora(`Deleting backend ${id}(${location})...`).start();
+      try {
+        await deleteBackendAndPoll(projectId, location, id);
+        spinner.succeed(`Successfully deleted the backend: ${id}(${location})`);
+      } catch (err: unknown) {
+        spinner.stop();
+        throw new FirebaseError(
+          `Failed to delete backend: ${id}(${location}). Please retry to delete remaining backends.`,
+          {
+            original: getError(err),
+          },
+        );
+      }
     }
   });