firebase-tools 14.2.2 → 14.3.0

package/lib/functions/secrets.js

@@ -42,13 +42,13 @@ async function ensureValidKey(key, options) {
             throw new error_1.FirebaseError("Secret key must be in UPPER_SNAKE_CASE.");
         }
         (0, utils_1.logWarning)(`By convention, secret key must be in UPPER_SNAKE_CASE.`);
-        const confirm = await (0, prompt_1.promptOnce)({
-            name: "updateKey",
-            type: "confirm",
+        const useTransformed = await (0, prompt_1.confirm)({
             default: true,
             message: `Would you like to use ${transformedKey} as key instead?`,
-        }, options);
-        if (!confirm) {
+            nonInteractive: options.nonInteractive,
+            force: options.force,
+        });
+        if (!useTransformed) {
             throw new error_1.FirebaseError("Secret key must be in UPPER_SNAKE_CASE.");
         }
     }
@@ -66,12 +66,11 @@ async function ensureSecret(projectId, name, options) {
         const secret = await (0, secretManager_1.getSecret)(projectId, name);
         if ((0, secretManager_1.isAppHostingManaged)(secret)) {
             (0, utils_1.logWarning)("Your secret is managed by Firebase App Hosting. Continuing will disable automatic deletion of old versions.");
-            const stopTracking = await (0, prompt_1.promptOnce)({
-                name: "doNotTrack",
-                type: "confirm",
-                default: false,
+            const stopTracking = await (0, prompt_1.confirm)({
                 message: "Do you wish to continue?",
-            }, options);
+                nonInteractive: options.nonInteractive,
+                force: options.force,
+            });
             if (stopTracking) {
                 delete secret.labels[secretManager_2.FIREBASE_MANAGED];
                 await (0, secretManager_1.patchSecret)(secret.projectId, secret.name, secret.labels);
@@ -84,13 +83,13 @@ async function ensureSecret(projectId, name, options) {
             if (!options.force) {
                 (0, utils_1.logWarning)("Your secret is not managed by Cloud Functions for Firebase. " +
                     "Firebase managed secrets are automatically pruned to reduce your monthly cost for using Secret Manager. ");
-                const confirm = await (0, prompt_1.promptOnce)({
-                    name: "updateLabels",
-                    type: "confirm",
+                const updateLabels = await (0, prompt_1.confirm)({
                     default: true,
                     message: `Would you like to have your secret ${secret.name} managed by Cloud Functions for Firebase?`,
-                }, options);
-                if (confirm) {
+                    nonInteractive: options.nonInteractive,
+                    force: options.force,
+                });
+                if (updateLabels) {
                     return (0, secretManager_1.patchSecret)(projectId, secret.name, Object.assign(Object.assign({}, secret.labels), (0, secretManager_3.labels)()));
                 }
             }

package/lib/gcp/auth.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.setCustomClaim = exports.disableUser = exports.findUser = exports.updateAuthDomains = exports.getAuthDomains = void 0;
+exports.setDenySmsRegionPolicy = exports.setAllowSmsRegionPolicy = exports.setCustomClaim = exports.disableUser = exports.findUser = exports.updateAuthDomains = exports.getAuthDomains = void 0;
 const apiv2_1 = require("../apiv2");
 const api_1 = require("../api");
 const apiClient = new apiv2_1.Client({ urlPrefix: (0, api_1.identityOrigin)(), auth: true });
@@ -72,3 +72,31 @@ async function setCustomClaim(project, uid, claim, options) {
     return user;
 }
 exports.setCustomClaim = setCustomClaim;
+async function setAllowSmsRegionPolicy(project, countryCodes) {
+    const res = await apiClient.patch(`/admin/v2/projects/${project}/config?updateMask=sms_region_config`, {
+        sms_region_config: {
+            allowlist_only: {
+                allowed_regions: countryCodes,
+            },
+        },
+    });
+    if (res.status !== 200) {
+        throw new Error("SMS Region Policy failed to be configured");
+    }
+    return true;
+}
+exports.setAllowSmsRegionPolicy = setAllowSmsRegionPolicy;
+async function setDenySmsRegionPolicy(project, countryCodes) {
+    const res = await apiClient.patch(`/admin/v2/projects/${project}/config?updateMask=sms_region_config`, {
+        sms_region_config: {
+            allow_by_default: {
+                disallowed_regions: countryCodes,
+            },
+        },
+    });
+    if (res.status !== 200) {
+        throw new Error("SMS Region Policy failed to be configured");
+    }
+    return true;
+}
+exports.setDenySmsRegionPolicy = setDenySmsRegionPolicy;

package/lib/gif/fdcExperience.js

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateOperation = exports.chatWithFirebase = exports.generateSchema = void 0;
+const apiv2_1 = require("../apiv2");
+const api_1 = require("../api");
+const apiClient = new apiv2_1.Client({ urlPrefix: (0, api_1.cloudCompanionOrigin)(), auth: true });
+const schemaGeneratorExperience = "/appeco/firebase/fdc-schema-generator";
+const geminiInFirebaseChatExperience = "/appeco/firebase/firebase-chat/free";
+const operationGeneratorExperience = "/appeco/firebase/fdc-query-generator";
+async function generateSchema(prompt, project) {
+    const res = await apiClient.post(`/v1beta/projects/${project}/locations/global/instances/default:completeTask`, {
+        input: { messages: [{ content: prompt, author: "USER" }] },
+        experienceContext: {
+            experience: schemaGeneratorExperience,
+        },
+    });
+    return res.body.output.messages[0].content;
+}
+exports.generateSchema = generateSchema;
+async function chatWithFirebase(prompt, project) {
+    const res = await apiClient.post(`/v1beta/projects/${project}/locations/global/instances/default:completeTask`, {
+        input: { messages: [{ content: prompt, author: "USER" }] },
+        experienceContext: {
+            experience: geminiInFirebaseChatExperience,
+        },
+    });
+    return res.body;
+}
+exports.chatWithFirebase = chatWithFirebase;
+async function generateOperation(prompt, service, project) {
+    const res = await apiClient.post(`/v1beta/projects/${project}/locations/global/instances/default:completeTask`, {
+        input: { messages: [{ content: prompt, author: "USER" }] },
+        experienceContext: {
+            experience: operationGeneratorExperience,
+        },
+        clientContext: {
+            additionalContext: {
+                "@type": "type.googleapis.com/google.cloud.cloudaicompanion.v1main.FirebaseChatRequestContext",
+                fdcInfo: { fdcServiceName: service, requiresQuery: true },
+            },
+        },
+    });
+    return res.body.output.messages[0].content;
+}
+exports.generateOperation = generateOperation;

package/lib/hosting/interactive.js

@@ -4,8 +4,8 @@ exports.interactiveCreateHostingSite = void 0;
 const error_1 = require("../error");
 const utils_1 = require("../utils");
 const projectUtils_1 = require("../projectUtils");
-const prompt_1 = require("../prompt");
 const api_1 = require("./api");
+const prompt_1 = require("../prompt");
 const nameSuggestion = new RegExp("try something like `(.+)`");
 const prompt = "Please provide an unique, URL-friendly id for your site. Your site's URL will be <site-id>.web.app. " +
     'We recommend using letters, numbers, and hyphens (e.g. "{project-id}-{random-hash}"):';
@@ -26,8 +26,7 @@ async function interactiveCreateHostingSite(siteId, appId, options) {
     }
     while (!newSite) {
         if (!id || suggestion) {
-            id = await (0, prompt_1.promptOnce)({
-                type: "input",
+            id = await (0, prompt_1.input)({
                 message: prompt,
                 validate: (s) => s.length > 0,
                 default: suggestion,

package/lib/init/features/account.js

@@ -4,8 +4,8 @@ exports.doSetup = void 0;
 const logger_1 = require("../../logger");
 const utils = require("../../utils");
 const auth_1 = require("../../auth");
-const prompt_1 = require("../../prompt");
 const error_1 = require("../../error");
+const prompt_1 = require("../../prompt");
 async function promptForAccount() {
     logger_1.logger.info();
     logger_1.logger.info(`Which account do you want to use for this project? Choose an account or add a new one now`);
@@ -21,9 +21,7 @@ async function promptForAccount() {
         name: "(add a new account)",
         value: "__add__",
     });
-    const emailChoice = await (0, prompt_1.promptOnce)({
-        type: "list",
-        name: "email",
+    const emailChoice = await (0, prompt_1.select)({
         message: "Please select an option:",
         choices,
     });

package/lib/init/features/database.js

@@ -37,8 +37,7 @@ function writeDBRules(rules, logMessagePrefix, filename, config) {
     logger_1.logger.info(clc.bold("firebase deploy") + ".");
 }
 async function createDefaultDatabaseInstance(project) {
-    const selectedLocation = await (0, prompt_1.promptOnce)({
-        type: "list",
+    const selectedLocation = await (0, prompt_1.select)({
         message: "Please choose the location for your default Realtime Database instance:",
         choices: [
             { name: "us-central1", value: database_1.DatabaseLocation.US_CENTRAL1 },
@@ -79,13 +78,11 @@ async function doSetup(setup, config) {
             instanceDetails = await (0, database_1.getDatabaseInstanceDetails)(setup.projectId, setup.instance);
         }
         else {
-            const confirm = await (0, prompt_1.promptOnce)({
-                type: "confirm",
-                name: "confirm",
-                default: true,
+            const createDefault = await (0, prompt_1.confirm)({
                 message: "It seems like you haven’t initialized Realtime Database in your project yet. Do you want to set it up?",
+                default: true,
             });
-            if (confirm) {
+            if (createDefault) {
                 instanceDetails = await createDefaultDatabaseInstance(setup.projectId);
             }
         }
@@ -95,14 +92,12 @@ async function doSetup(setup, config) {
     logger_1.logger.info("Firebase Realtime Database Security Rules allow you to define how your data should be");
     logger_1.logger.info("structured and when your data can be read from and written to.");
     logger_1.logger.info();
-    await (0, prompt_1.prompt)(setup.config.database, [
-        {
-            type: "input",
-            name: "rules",
-            message: "What file should be used for Realtime Database Security Rules?",
-            default: "database.rules.json",
-        },
-    ]);
+    setup.config.database.rules =
+        setup.config.database.rules ||
+            (await (0, prompt_1.input)({
+                message: "What file should be used for Realtime Database Security Rules?",
+                default: "database.rules.json",
+            }));
     const filename = setup.config.database.rules;
     if (!filename) {
         throw new error_1.FirebaseError("Must specify location for Realtime Database rules file.");
@@ -113,11 +108,7 @@ async function doSetup(setup, config) {
             ? `the Realtime Database Security Rules for ${clc.bold(instanceDetails.name)} from the Firebase console`
             : "default rules";
         const msg = `File ${clc.bold(filename)} already exists. Do you want to overwrite it with ${rulesDescription}?`;
-        writeRules = await (0, prompt_1.promptOnce)({
-            type: "confirm",
-            message: msg,
-            default: false,
-        });
+        writeRules = await (0, prompt_1.confirm)(msg);
     }
     if (writeRules) {
         if (instanceDetails) {

package/lib/init/features/dataconnect/index.js

@@ -211,9 +211,8 @@ async function promptForExistingServices(setup, info, isBillingEnabled) {
                 };
             });
             choices.push({ name: "Create a new service", value: undefined });
-            choice = await (0, prompt_1.promptOnce)({
+            choice = await (0, prompt_1.select)({
                 message: "Your project already has existing services. Which would you like to set up local files for?",
-                type: "list",
                 choices,
             });
         }
@@ -271,9 +270,8 @@ async function promptForCloudSQL(setup, info) {
             else {
                 choices.push({ name: "Create a new CloudSQL instance", value: "", location: "" });
             }
-            info.cloudSqlInstanceId = await (0, prompt_1.promptOnce)({
+            info.cloudSqlInstanceId = await (0, prompt_1.select)({
                 message: `Which CloudSQL instance would you like to use?`,
-                type: "list",
                 choices,
             });
             if (info.cloudSqlInstanceId !== "") {
@@ -283,17 +281,15 @@ async function promptForCloudSQL(setup, info) {
     }
     if (info.cloudSqlInstanceId === "") {
         info.isNewInstance = true;
-        info.cloudSqlInstanceId = await (0, prompt_1.promptOnce)({
+        info.cloudSqlInstanceId = await (0, prompt_1.input)({
             message: `What ID would you like to use for your new CloudSQL instance?`,
-            type: "input",
             default: `${info.serviceId.toLowerCase() || "app"}-fdc`,
         });
     }
     if (info.locationId === "") {
         const choices = await locationChoices(setup);
-        info.locationId = await (0, prompt_1.promptOnce)({
+        info.locationId = await (0, prompt_1.select)({
             message: "What location would like to use?",
-            type: "list",
             choices,
         });
     }
@@ -305,9 +301,8 @@ async function promptForCloudSQL(setup, info) {
             });
             choices.push({ name: "Create a new database", value: "" });
             if (dbs.length) {
-                info.cloudSqlDatabase = await (0, prompt_1.promptOnce)({
+                info.cloudSqlDatabase = await (0, prompt_1.select)({
                     message: `Which database in ${info.cloudSqlInstanceId} would you like to use?`,
-                    type: "list",
                     choices,
                 });
             }
@@ -318,9 +313,8 @@ async function promptForCloudSQL(setup, info) {
     }
     if (info.cloudSqlDatabase === "") {
         info.isNewDatabase = true;
-        info.cloudSqlDatabase = await (0, prompt_1.promptOnce)({
+        info.cloudSqlDatabase = await (0, prompt_1.input)({
             message: `What ID would you like to use for your new database in ${info.cloudSqlInstanceId}?`,
-            type: "input",
             default: `fdcdb`,
         });
     }
@@ -328,9 +322,8 @@ async function promptForCloudSQL(setup, info) {
 }
 async function promptForService(info) {
     if (info.serviceId === "") {
-        info.serviceId = await (0, prompt_1.promptOnce)({
+        info.serviceId = await (0, prompt_1.input)({
             message: "What ID would you like to use for this service?",
-            type: "input",
             default: (0, path_1.basename)(process.cwd()),
         });
     }

package/lib/init/features/dataconnect/sdk.js

@@ -41,7 +41,7 @@ async function askQuestions(setup, config) {
     let appDir = process.env[exports.FDC_APP_FOLDER] || process.cwd();
     let targetPlatform = await (0, fileUtils_1.getPlatformFromFolder)(appDir);
     if (targetPlatform === types_1.Platform.NONE && !((_a = process.env[exports.FDC_APP_FOLDER]) === null || _a === void 0 ? void 0 : _a.length)) {
-        appDir = await (0, prompt_1.promptForDirectory)({
+        appDir = await (0, utils_1.promptForDirectory)({
             config,
             message: "Where is your app directory? Leave blank to set up a generated SDK in your current directory.",
         });
@@ -60,18 +60,16 @@ async function askQuestions(setup, config) {
             { name: "Android (Kotlin)", value: types_1.Platform.ANDROID },
             { name: "Flutter (Dart)", value: types_1.Platform.FLUTTER },
         ];
-        targetPlatform = await (0, prompt_1.promptOnce)({
+        targetPlatform = await (0, prompt_1.select)({
             message: "Which platform do you want to set up a generated SDK for?",
-            type: "list",
             choices: platforms,
         });
     }
     else {
         (0, utils_1.logSuccess)(`Detected ${targetPlatform} app in directory ${appDir}`);
     }
-    const connectorInfo = await (0, prompt_1.promptOnce)({
+    const connectorInfo = await (0, prompt_1.select)({
         message: "Which connector do you want set up a generated SDK for?",
-        type: "list",
         choices: connectorChoices,
     });
     const connectorYaml = JSON.parse(JSON.stringify(connectorInfo.connectorYaml));
@@ -79,23 +77,18 @@ async function askQuestions(setup, config) {
     if (targetPlatform === types_1.Platform.WEB) {
         const unusedFrameworks = fileUtils_1.SUPPORTED_FRAMEWORKS.filter((framework) => { var _a; return !((_a = newConnectorYaml.generate) === null || _a === void 0 ? void 0 : _a.javascriptSdk[framework]); });
         if (unusedFrameworks.length > 0) {
-            const additionalFrameworks = await (0, prompt_1.prompt)(setup, [
-                {
-                    type: "checkbox",
-                    name: "fdcFrameworks",
-                    message: `Which frameworks would you like to generate SDKs for? ` +
-                        "Press Space to select features, then Enter to confirm your choices.",
-                    choices: fileUtils_1.SUPPORTED_FRAMEWORKS.map((frameworkStr) => {
-                        var _a, _b;
-                        return ({
-                            value: frameworkStr,
-                            name: frameworkStr,
-                            checked: (_b = (_a = newConnectorYaml === null || newConnectorYaml === void 0 ? void 0 : newConnectorYaml.generate) === null || _a === void 0 ? void 0 : _a.javascriptSdk) === null || _b === void 0 ? void 0 : _b[frameworkStr],
-                        });
-                    }),
-                },
-            ]);
-            for (const framework of additionalFrameworks.fdcFrameworks) {
+            const additionalFrameworks = await (0, prompt_1.checkbox)({
+                message: "Which frameworks would you like to generate SDKs for? " +
+                    "Press Space to select features, then Enter to confirm your choices.",
+                choices: fileUtils_1.SUPPORTED_FRAMEWORKS.map((frameworkStr) => {
+                    var _a, _b;
+                    return ({
+                        value: frameworkStr,
+                        checked: (_b = (_a = newConnectorYaml === null || newConnectorYaml === void 0 ? void 0 : newConnectorYaml.generate) === null || _a === void 0 ? void 0 : _a.javascriptSdk) === null || _b === void 0 ? void 0 : _b[frameworkStr],
+                    });
+                }),
+            });
+            for (const framework of additionalFrameworks) {
                 newConnectorYaml.generate.javascriptSdk[framework] = true;
             }
         }

package/lib/init/features/emulators.js

@@ -14,19 +14,15 @@ async function doSetup(setup, config) {
         return {
             value: e,
             name: constants_1.Constants.description(e),
-            checked: config && (config.has(e) || config.has(`emulators.${e}`)),
+            checked: (config === null || config === void 0 ? void 0 : config.has(e)) || (config === null || config === void 0 ? void 0 : config.has(`emulators.${e}`)),
         };
     });
     const selections = {};
-    await (0, prompt_1.prompt)(selections, [
-        {
-            type: "checkbox",
-            name: "emulators",
-            message: "Which Firebase emulators do you want to set up? " +
-                "Press Space to select emulators, then Enter to confirm your choices.",
-            choices: choices,
-        },
-    ]);
+    selections.emulators = await (0, prompt_1.checkbox)({
+        message: "Which Firebase emulators do you want to set up? " +
+            "Press Space to select emulators, then Enter to confirm your choices.",
+        choices: choices,
+    });
     if (!selections.emulators) {
         return;
     }
@@ -38,14 +34,10 @@ async function doSetup(setup, config) {
             utils.logBullet(`Port for ${selected} already configured: ${clc.cyan(currentPort)}`);
         }
         else {
-            await (0, prompt_1.prompt)(setup.config.emulators[selected], [
-                {
-                    type: "number",
-                    name: "port",
-                    message: `Which port do you want to use for the ${clc.underline(selected)} emulator?`,
-                    default: constants_1.Constants.getDefaultPort(selected),
-                },
-            ]);
+            setup.config.emulators[selected].port = await (0, prompt_1.number)({
+                message: `Which port do you want to use for the ${clc.underline(selected)} emulator?`,
+                default: constants_1.Constants.getDefaultPort(selected),
+            });
         }
         const additionalInitFn = initEmulators_1.AdditionalInitFns[selected];
         if (additionalInitFn) {
@@ -64,34 +56,20 @@ async function doSetup(setup, config) {
         else {
             const ui = setup.config.emulators.ui || {};
             setup.config.emulators.ui = ui;
-            await (0, prompt_1.prompt)(ui, [
-                {
-                    name: "enabled",
-                    type: "confirm",
-                    message: `Would you like to enable the ${uiDesc}?`,
-                    default: true,
-                },
-            ]);
+            ui.enalbed = await (0, prompt_1.confirm)({
+                message: `Would you like to enable the ${uiDesc}?`,
+                default: true,
+            });
             if (ui.enabled) {
-                await (0, prompt_1.prompt)(ui, [
-                    {
-                        type: "input",
-                        name: "port",
-                        message: `Which port do you want to use for the ${clc.underline(uiDesc)} (leave empty to use any available port)?`,
-                    },
-                ]);
+                ui.port = await (0, prompt_1.number)(`Which port do you want to use for the ${clc.underline(uiDesc)} (leave empty to use any available port)?`);
                 const portNum = Number.parseInt(ui.port);
                 ui.port = isNaN(portNum) ? undefined : portNum;
             }
         }
-        await (0, prompt_1.prompt)(selections, [
-            {
-                name: "download",
-                type: "confirm",
-                message: "Would you like to download the emulators now?",
-                default: true,
-            },
-        ]);
+        selections.download = await (0, prompt_1.confirm)({
+            message: "Would you like to download the emulators now?",
+            default: true,
+        });
     }
     if (setup.config.emulators.singleProjectMode === undefined) {
         setup.config.emulators.singleProjectMode = true;

package/lib/init/features/firestore/index.js

@@ -37,12 +37,8 @@ async function getDatabaseType(setup) {
         return dbType;
     }
 }
-async function selectDatabaseByPrompting() {
-    const database = await (0, prompt_1.promptOnce)({
-        type: "input",
-        message: "Please input the name of the Native Firestore database you would like to use:",
-    });
-    return database;
+function selectDatabaseByPrompting() {
+    return (0, prompt_1.input)("Please input the name of the Native Firestore database you would like to use:");
 }
 async function doSetup(setup, config, options) {
     if (setup.projectId) {

package/lib/init/features/firestore/indexes.js

@@ -10,46 +10,32 @@ const logger_1 = require("../../../logger");
 const templates_1 = require("../../../templates");
 const indexes = new api.FirestoreApi();
 const INDEXES_TEMPLATE = (0, templates_1.readTemplateSync)("init/firestore/firestore.indexes.json");
-function initIndexes(setup, config) {
+async function initIndexes(setup, config) {
     logger_1.logger.info();
     logger_1.logger.info("Firestore indexes allow you to perform complex queries while");
     logger_1.logger.info("maintaining performance that scales with the size of the result");
     logger_1.logger.info("set. You can keep index definitions in your project directory");
     logger_1.logger.info("and publish them with " + clc.bold("firebase deploy") + ".");
     logger_1.logger.info();
-    return (0, prompt_1.prompt)(setup.config.firestore, [
-        {
-            type: "input",
-            name: "indexes",
+    const filename = setup.config.firestore.indexes ||
+        (await (0, prompt_1.input)({
             message: "What file should be used for Firestore indexes?",
             default: "firestore.indexes.json",
-        },
-    ])
-        .then(() => {
-        const filename = setup.config.firestore.indexes;
-        if (fsutils.fileExistsSync(filename)) {
-            const msg = "File " +
-                clc.bold(filename) +
-                " already exists." +
-                " Do you want to overwrite it with the Firestore Indexes from the Firebase Console?";
-            return (0, prompt_1.promptOnce)({
-                type: "confirm",
-                message: msg,
-                default: false,
-            });
+        }));
+    if (fsutils.fileExistsSync(filename)) {
+        const msg = "File " +
+            clc.bold(filename) +
+            " already exists." +
+            " Do you want to overwrite it with the Firestore Indexes from the Firebase Console?";
+        if (!(await (0, prompt_1.confirm)(msg))) {
+            return;
         }
-        return Promise.resolve(true);
-    })
-        .then((overwrite) => {
-        if (!overwrite) {
-            return Promise.resolve();
-        }
-        if (!setup.projectId) {
-            return config.writeProjectFile(setup.config.firestore.indexes, INDEXES_TEMPLATE);
-        }
-        return getIndexesFromConsole(setup.projectId, setup.databaseId).then((contents) => {
-            return config.writeProjectFile(setup.config.firestore.indexes, contents);
-        });
+    }
+    if (!setup.projectId) {
+        return config.writeProjectFile(setup.config.firestore.indexes, INDEXES_TEMPLATE);
+    }
+    return getIndexesFromConsole(setup.projectId, setup.databaseId).then((contents) => {
+        return config.writeProjectFile(setup.config.firestore.indexes, contents);
     });
 }
 exports.initIndexes = initIndexes;

package/lib/init/features/firestore/rules.js

@@ -10,45 +10,31 @@ const utils = require("../../../utils");
 const templates_1 = require("../../../templates");
 const DEFAULT_RULES_FILE = "firestore.rules";
 const RULES_TEMPLATE = (0, templates_1.readTemplateSync)("init/firestore/firestore.rules");
-function initRules(setup, config) {
+async function initRules(setup, config) {
     logger_1.logger.info();
     logger_1.logger.info("Firestore Security Rules allow you to define how and when to allow");
     logger_1.logger.info("requests. You can keep these rules in your project directory");
     logger_1.logger.info("and publish them with " + clc.bold("firebase deploy") + ".");
     logger_1.logger.info();
-    return (0, prompt_1.prompt)(setup.config.firestore, [
-        {
-            type: "input",
-            name: "rules",
+    const filename = setup.config.firestore.rules ||
+        (await (0, prompt_1.input)({
             message: "What file should be used for Firestore Rules?",
             default: DEFAULT_RULES_FILE,
-        },
-    ])
-        .then(() => {
-        const filename = setup.config.firestore.rules;
-        if (fsutils.fileExistsSync(filename)) {
-            const msg = "File " +
-                clc.bold(filename) +
-                " already exists." +
-                " Do you want to overwrite it with the Firestore Rules from the Firebase Console?";
-            return (0, prompt_1.promptOnce)({
-                type: "confirm",
-                message: msg,
-                default: false,
-            });
+        }));
+    if (fsutils.fileExistsSync(filename)) {
+        const msg = "File " +
+            clc.bold(filename) +
+            " already exists." +
+            " Do you want to overwrite it with the Firestore Rules from the Firebase Console?";
+        if (!(await (0, prompt_1.confirm)(msg))) {
+            return;
         }
-        return Promise.resolve(true);
-    })
-        .then((overwrite) => {
-        if (!overwrite) {
-            return Promise.resolve();
-        }
-        if (!setup.projectId) {
-            return config.writeProjectFile(setup.config.firestore.rules, getDefaultRules());
-        }
-        return getRulesFromConsole(setup.projectId).then((contents) => {
-            return config.writeProjectFile(setup.config.firestore.rules, contents);
-        });
+    }
+    if (!setup.projectId) {
+        return config.writeProjectFile(setup.config.firestore.rules, getDefaultRules());
+    }
+    return getRulesFromConsole(setup.projectId).then((contents) => {
+        return config.writeProjectFile(setup.config.firestore.rules, contents);
     });
 }
 exports.initRules = initRules;