Improve FAH create & delete

src/gcp/apphosting.ts

@@ -68,7 +68,7 @@
   | "error"
   | "image"
   | "sourceRef"
-  | "buildLogUri"
+  | "buildLogsUri"
   | "reconciling"
   | "uuid"
   | "etag"
@@ -197,10 +197,20 @@
   // end oneof trigger
   stages?: RolloutStage[];
   disabled?: boolean;
+
+  // TODO: This will be undefined if disabled is not true, right?
   disabledTime: string;
 }
 
-export type RolloutPolicyOutputOnlyFields = "disabledtime";
+export type RolloutPolicyOutputOnlyFields = "disabledTime";
+
+export type RolloutPolicyInput = Omit<RolloutPolicy, RolloutPolicyOutputOnlyFields | "stages"> & {
+  stages: Omit<RolloutStage, "startTime" | "endTime">[];
+};
+
+export type TrafficInput = Omit<Traffic, TrafficOutputOnlyFields | "rolloutPolicy"> & {
+  rolloutPolicy: RolloutPolicyInput;
+};
 
 export type RolloutProgression =
   | "PROGRESSION_UNSPECIFIED"
@@ -238,7 +248,7 @@
   done: boolean;
   // oneof result
   error?: Status;
   response?: any;
   // end oneof result
 }
 
@@ -304,8 +314,8 @@
   location: string,
   backendId: string,
 ): Promise<Operation> {
-  const name = `projects/${projectId}/locations/${location}/backends/${backendId}?force=true`;
-  const res = await client.delete<Operation>(name);
+  const name = `projects/${projectId}/locations/${location}/backends/${backendId}`;
+  const res = await client.delete<Operation>(name, { queryParams: { force: "true" } });
 
   return res.body;
 }
@@ -393,14 +403,21 @@
   projectId: string,
   location: string,
   backendId: string,
-  traffic: Omit<Traffic, TrafficOutputOnlyFields | "name">,
+  traffic: Omit<TrafficInput, "name">,
 ): Promise<Operation> {
-  const fieldMasks = proto.fieldMasks(traffic);
+  // BUG(b/322891558): setting deep fields on rolloutPolicy doesn't work for some
+  // reason. Create a copy without deep fields to force the updateMask to be
+  // correct.
+  const trafficCopy = { ...traffic };
+  if ("rolloutPolicy" in traffic) {
+    trafficCopy.rolloutPolicy = {} as any;
+  }
+  const fieldMasks = proto.fieldMasks(trafficCopy);
   const queryParams = {
     updateMask: fieldMasks.join(","),
   };
   const name = `projects/${projectId}/locations/${location}/backends/${backendId}/traffic`;
-  const res = await client.patch<Omit<Traffic, TrafficOutputOnlyFields>, Operation>(
+  const res = await client.patch<TrafficInput, Operation>(
     name,
     { ...traffic, name },
     {
@@ -439,7 +456,7 @@
 /**
  * Ensure that Frameworks API is enabled on the project.
  */
 export async function ensureApiEnabled(options: any): Promise<void> {
   const projectId = needProjectId(options);
   return await ensure(projectId, API_HOST, "frameworks", true);
 }

src/init/features/apphosting/index.ts

@@ -17,6 +17,7 @@
 import { promptOnce } from "../../../prompt";
 import { DEFAULT_REGION } from "./constants";
 import { ensure } from "../../../ensureApiEnabled";
+import * as deploymentTool from "../../../deploymentTool";
 
 const apphostingPollerOptions: Omit<poller.OperationPollerOptions, "operationResourceName"> = {
   apiOrigin: apphostingOrigin,
@@ -28,7 +29,7 @@
 /**
  * Set up a new App Hosting backend.
  */
 export async function doSetup(setup: any, projectId: string): Promise<void> {
   await Promise.all([
     ensure(projectId, "cloudbuild.googleapis.com", "apphosting", true),
     ensure(projectId, "secretmanager.googleapis.com", "apphosting", true),
@@ -38,10 +39,10 @@
 
   const allowedLocations = (await apphosting.listLocations(projectId)).map((loc) => loc.locationId);
 
   if (setup.location) {
     if (!allowedLocations.includes(setup.location)) {
       throw new FirebaseError(
         `Invalid location ${setup.location}. Valid choices are ${allowedLocations.join(", ")}`,
       );
     }
   }
@@ -76,6 +77,33 @@
 
   const backend = await onboardBackend(projectId, location, backendId);
 
+  // TODO: Once tag patterns are implemented, prompt which method the user
+  // prefers. We could reduce the nubmer of questions asked by letting people
+  // enter tag:<pattern>?
+  const branch = await promptOnce({
+    name: "branch",
+    type: "input",
+    default: "main",
+    message: "Pick a branch for continuous deployment",
+  });
+  const traffic: Omit<apphosting.TrafficInput, "name"> = {
+    rolloutPolicy: {
+      codebaseBranch: branch,
+      stages: [
+        {
+          progression: "IMMEDIATE",
+          targetPercent: 100,
+        },
+      ],
+    },
+  };
+  const op = await apphosting.updateTraffic(projectId, location, backendId, traffic);
+  await poller.pollOperation<apphosting.Traffic>({
+    ...apphostingPollerOptions,
+    pollerName: `updateTraffic-${projectId}-${location}-${backendId}`,
+    operationResourceName: op.name,
+  });
+
   const confirmRollout = await promptOnce({
     type: "confirm",
     name: "rollout",
@@ -89,13 +117,6 @@
     return;
   }
 
-  const branch = await promptOnce({
-    name: "branch",
-    type: "input",
-    default: "main",
-    message: "Which branch do you want to deploy?",
-  });
-
   const { build } = await onboardRollout(projectId, location, backendId, {
     source: {
       codebase: {
@@ -105,6 +126,12 @@
   });
 
   if (build.state !== "READY") {
+    if (!build.buildLogsUri) {
+      throw new FirebaseError(
+        "Failed to build your app, but failed to get build logs as well. " +
+          "This is an internal error and should be reported",
+      );
+    }
     throw new FirebaseError(
       `Failed to build your app. Please inspect the build logs at ${build.buildLogsUri}.`,
       { children: [build.error] },
@@ -137,7 +164,7 @@
       repository: `${cloudBuildConnRepo.name}`,
       rootDirectory: "/",
     },
-    labels: {},
+    labels: deploymentTool.labels(),
   };
 }
 

src/test/init/apphosting/index.spec.ts

@@ -7,6 +7,7 @@ import * as poller from "../../../operation-poller";
 import * as prompt from "../../../prompt";
 import { createBackend, onboardBackend } from "../../../init/features/apphosting/index";
 import { FirebaseError } from "../../../error";
+import * as deploymentTool from "../../../deploymentTool";
 
 describe("operationsConverter", () => {
   const sandbox: sinon.SinonSandbox = sinon.createSandbox();
@@ -66,7 +67,7 @@ describe("operationsConverter", () => {
         repository: cloudBuildConnRepo.name,
         rootDirectory: "/",
       },
-      labels: {},
+      labels: deploymentTool.labels(),
     };
 
     it("should createBackend", async () => {