Change how FDC determines whether creating a CSQL free trial is allowed. (#8065)

* Fix log statements.

* Catch backend quota error instead of querying for free trial metric.

Author: rosalyntan

src/dataconnect/freeTrial.ts

@@ -35,33 +35,40 @@ export async function getFreeTrialInstanceId(projectId: string): Promise<string
   return instances.find((i) => i.settings.userLabels?.["firebase-data-connect"] === "ft")?.name;
 }
 
+export async function isFreeTrialError(err: any, projectId: string): Promise<boolean> {
+  // checkFreeTrialInstanceUsed is also called to ensure the request didn't fail due to an unrelated quota issue.
+  return err.message.includes("Quota Exhausted") && (await checkFreeTrialInstanceUsed(projectId))
+    ? true
+    : false;
+}
+
 export function printFreeTrialUnavailable(
   projectId: string,
   configYamlPath: string,
   instanceId?: string,
 ): void {
   if (!instanceId) {
     utils.logLabeledError(
-      "data connect",
+      "dataconnect",
       "The CloudSQL free trial has already been used on this project.",
     );
     utils.logLabeledError(
-      "data connect",
+      "dataconnect",
       `You may create or use a paid CloudSQL instance by visiting https://console.cloud.google.com/sql/instances`,
     );
     return;
   }
   utils.logLabeledError(
-    "data connect",
+    "dataconnect",
     `Project '${projectId} already has a CloudSQL instance '${instanceId}' on the Firebase Data Connect no-cost trial.`,
   );
   const reuseHint =
     `To use a different database in the same instance, ${clc.bold(`change the ${clc.blue("instanceId")} to "${instanceId}"`)} and update ${clc.blue("location")} in ` +
     `${clc.green(configYamlPath)}.`;
 
-  utils.logLabeledError("data connect", reuseHint);
+  utils.logLabeledError("dataconnect", reuseHint);
   utils.logLabeledError(
-    "data connect",
+    "dataconnect",
     `Alternatively, you may create a new (paid) CloudSQL instance at https://console.cloud.google.com/sql/instances`,
   );
 }

src/dataconnect/provisionCloudSql.ts

@@ -11,7 +11,7 @@ import {
   getFreeTrialInstanceId,
   freeTrialTermsLink,
   printFreeTrialUnavailable,
-  checkFreeTrialInstanceUsed,
+  isFreeTrialError,
 } from "./freeTrial";
 import { FirebaseError } from "../error";
 
@@ -69,11 +69,6 @@ export async function provisionCloudSql(args: {
     if (err.status !== 404) {
       throw err;
     }
-    const freeTrialInstanceId = await getFreeTrialInstanceId(projectId);
-    if (await checkFreeTrialInstanceUsed(projectId)) {
-      printFreeTrialUnavailable(projectId, configYamlPath, freeTrialInstanceId);
-      throw new FirebaseError("No-cost Cloud SQL trial has already been used on this project.");
-    }
     const cta = dryRun ? "It will be created on your next deploy" : "Creating it now.";
     silent ||
       utils.logLabeledBullet(
@@ -84,27 +79,36 @@ export async function provisionCloudSql(args: {
           `\nMonitor the progress at ${cloudSqlAdminClient.instanceConsoleLink(projectId, instanceId)}`,
       );
     if (!dryRun) {
-      const newInstance = await promiseWithSpinner(
-        () =>
-          cloudSqlAdminClient.createInstance(
-            projectId,
-            locationId,
-            instanceId,
-            enableGoogleMlIntegration,
-            waitForCreation,
-          ),
-        "Creating your instance...",
-      );
-      if (newInstance) {
-        silent || utils.logLabeledBullet("dataconnect", "Instance created");
-        connectionName = newInstance?.connectionName || "";
-      } else {
-        silent ||
-          utils.logLabeledBullet(
-            "dataconnect",
-            "Cloud SQL instance creation started - it should be ready shortly. Database and users will be created on your next deploy.",
-          );
-        return connectionName;
+      try {
+        const newInstance = await promiseWithSpinner(
+          () =>
+            cloudSqlAdminClient.createInstance(
+              projectId,
+              locationId,
+              instanceId,
+              enableGoogleMlIntegration,
+              waitForCreation,
+            ),
+          "Creating your instance...",
+        );
+        if (newInstance) {
+          silent || utils.logLabeledBullet("dataconnect", "Instance created");
+          connectionName = newInstance?.connectionName || "";
+        } else {
+          silent ||
+            utils.logLabeledBullet(
+              "dataconnect",
+              "Cloud SQL instance creation started - it should be ready shortly. Database and users will be created on your next deploy.",
+            );
+          return connectionName;
+        }
+      } catch (err: any) {
+        if (await isFreeTrialError(err, projectId)) {
+          const freeTrialInstanceId = await getFreeTrialInstanceId(projectId);
+          printFreeTrialUnavailable(projectId, configYamlPath, freeTrialInstanceId);
+          throw new FirebaseError("No-cost Cloud SQL trial has already been used on this project.");
+        }
+        throw err;
       }
     }
   }