Add --disable-triggers flag to database write commands

CHANGELOG.md

@@ -1,3 +1,4 @@
 - Releases RTDB Emulator v4.11.0: Wire protocol update for `startAfter`, `endBefore`.
 - Changes `superstatic` dependency to `v8`, addressing Hosting emulator issues on Windows.
 - Fixes internal library that was not being correctly published.
+- Adds `--disable-triggers` flag to RTDB write commands.

src/commands/database-push.ts

@@ -21,6 +21,7 @@ export const command = new Command("database:push <path> [infile]")
     "--instance <instance>",
     "use the database <instance>.firebaseio.com (if omitted, use default database instance)"
   )
+  .option("--disable-triggers", "suppress any Cloud functions triggered by this operation")
   .before(requirePermissions, ["firebasedatabase.instances.update"])
   .before(requireDatabaseInstance)
   .before(populateInstanceDetails)
@@ -33,6 +34,9 @@ export const command = new Command("database:push <path> [infile]")
       utils.stringToStream(options.data) || (infile ? fs.createReadStream(infile) : process.stdin);
     const origin = realtimeOriginOrEmulatorOrCustomUrl(options.instanceDetails.databaseUrl);
     const u = new URL(utils.getDatabaseUrl(origin, options.instance, path + ".json"));
+    if (options.disableTriggers) {
+      u.searchParams.set("disableTriggers", "true");
+    }
 
     if (!infile && !options.data) {
       utils.explainStdin();
@@ -46,6 +50,7 @@ export const command = new Command("database:push <path> [infile]")
         method: "POST",
         path: u.pathname,
         body: inStream,
+        queryParams: u.searchParams,
       });
     } catch (err: any) {
       logger.debug(err);

src/commands/database-remove.ts

@@ -17,6 +17,7 @@ export const command = new Command("database:remove <path>")
     "--instance <instance>",
     "use the database <instance>.firebaseio.com (if omitted, use default database instance)"
   )
+  .option("--disable-triggers", "suppress any Cloud functions triggered by this operation")
   .before(requirePermissions, ["firebasedatabase.instances.update"])
   .before(requireDatabaseInstance)
   .before(populateInstanceDetails)
@@ -40,7 +41,7 @@ export const command = new Command("database:remove <path>")
       return utils.reject("Command aborted.", { exit: 1 });
     }
 
-    const removeOps = new DatabaseRemove(options.instance, path, origin);
+    const removeOps = new DatabaseRemove(options.instance, path, origin, !!options.disableTriggers);
     await removeOps.execute();
     utils.logSuccess("Data removed successfully");
   });

src/commands/database-set.ts

@@ -23,6 +23,7 @@ export const command = new Command("database:set <path> [infile]")
     "--instance <instance>",
     "use the database <instance>.firebaseio.com (if omitted, use default database instance)"
   )
+  .option("--disable-triggers", "suppress any Cloud functions triggered by this operation")
   .before(requirePermissions, ["firebasedatabase.instances.update"])
   .before(requireDatabaseInstance)
   .before(populateInstanceDetails)
@@ -34,6 +35,9 @@ export const command = new Command("database:set <path> [infile]")
     const origin = realtimeOriginOrEmulatorOrCustomUrl(options.instanceDetails.databaseUrl);
     const dbPath = utils.getDatabaseUrl(origin, options.instance, path);
     const dbJsonURL = new URL(utils.getDatabaseUrl(origin, options.instance, path + ".json"));
+    if (options.disableTriggers) {
+      dbJsonURL.searchParams.set("disableTriggers", "true");
+    }
 
     const confirm = await promptOnce(
       {
@@ -61,6 +65,7 @@ export const command = new Command("database:set <path> [infile]")
         method: "PUT",
         path: dbJsonURL.pathname,
         body: inStream,
+        queryParams: dbJsonURL.searchParams,
       });
     } catch (err: any) {
       logger.debug(err);

src/commands/database-update.ts

@@ -23,6 +23,7 @@ export const command = new Command("database:update <path> [infile]")
     "--instance <instance>",
     "use the database <instance>.firebaseio.com (if omitted, use default database instance)"
   )
+  .option("--disable-triggers", "suppress any Cloud functions triggered by this operation")
   .before(requirePermissions, ["firebasedatabase.instances.update"])
   .before(requireDatabaseInstance)
   .before(populateInstanceDetails)
@@ -51,6 +52,9 @@ export const command = new Command("database:update <path> [infile]")
       (infile && fs.createReadStream(infile)) ||
       process.stdin;
     const jsonUrl = new URL(utils.getDatabaseUrl(origin, options.instance, path + ".json"));
+    if (options.disableTriggers) {
+      jsonUrl.searchParams.set("disableTriggers", "true");
+    }
 
     if (!infile && !options.data) {
       utils.explainStdin();
@@ -62,6 +66,7 @@ export const command = new Command("database:update <path> [infile]")
         method: "PATCH",
         path: jsonUrl.pathname,
         body: inStream,
+        queryParams: jsonUrl.searchParams,
       });
     } catch (err: any) {
       throw new FirebaseError("Unexpected error while setting data");

src/database/remove.ts

@@ -29,10 +29,11 @@ export default class DatabaseRemove {
    * @param instance RTBD instance ID.
    * @param path path to delete.
    * @param host db host.
+   * @param disableTriggers if true, suppresses any Cloud functions that would be triggered by this operation.
    */
-  constructor(instance: string, path: string, host: string) {
+  constructor(instance: string, path: string, host: string, disableTriggers: boolean = false) {
     this.path = path;
-    this.remote = new RTDBRemoveRemote(instance, host);
+    this.remote = new RTDBRemoveRemote(instance, host, disableTriggers);
     this.deleteJobStack = new Stack({
       name: "delete stack",
       concurrency: 1,

src/database/removeRemote.ts

@@ -22,10 +22,12 @@ export class RTDBRemoveRemote implements RemoveRemote {
   private instance: string;
   private host: string;
   private apiClient: Client;
+  private disableTriggers: boolean;
 
-  constructor(instance: string, host: string) {
+  constructor(instance: string, host: string, disableTriggers: boolean = false) {
     this.instance = instance;
     this.host = host;
+    this.disableTriggers = disableTriggers;
 
     const url = new URL(utils.getDatabaseUrl(this.host, this.instance, "/"));
     this.apiClient = new Client({ urlPrefix: url.origin, auth: true });
@@ -46,7 +48,11 @@ export class RTDBRemoveRemote implements RemoveRemote {
   private async patch(path: string, body: any, note: string): Promise<boolean> {
     const t0 = Date.now();
     const url = new URL(utils.getDatabaseUrl(this.host, this.instance, path + ".json"));
-    const queryParams = { print: "silent", writeSizeLimit: "tiny" };
+    const queryParams = {
+      print: "silent",
+      writeSizeLimit: "tiny",
+      disableTriggers: this.disableTriggers.toString(),
+    };
     const res = await this.apiClient.request({
       method: "PATCH",
       path: url.pathname,

src/test/database/removeRemote.spec.ts

@@ -17,15 +17,15 @@ describe("RemoveRemote", () => {
   it("should return true when patch is small", () => {
     nock(serverUrl)
       .patch("/a/b.json")
-      .query({ print: "silent", writeSizeLimit: "tiny" })
+      .query({ print: "silent", writeSizeLimit: "tiny", disableTriggers: "false" })
       .reply(200, {});
     return expect(remote.deletePath("/a/b")).to.eventually.eql(true);
   });
 
   it("should return false whem patch is large", () => {
     nock(serverUrl)
       .patch("/a/b.json")
-      .query({ print: "silent", writeSizeLimit: "tiny" })
+      .query({ print: "silent", writeSizeLimit: "tiny", disableTriggers: "false" })
       .reply(400, {
         error:
           "Data requested exceeds the maximum size that can be accessed with a single request.",
@@ -36,15 +36,15 @@ describe("RemoveRemote", () => {
   it("should return true when multi-path patch is small", () => {
     nock(serverUrl)
       .patch("/a/b.json")
-      .query({ print: "silent", writeSizeLimit: "tiny" })
+      .query({ print: "silent", writeSizeLimit: "tiny", disableTriggers: "false" })
       .reply(200, {});
     return expect(remote.deleteSubPath("/a/b", ["1", "2", "3"])).to.eventually.eql(true);
   });
 
   it("should return false when multi-path patch is large", () => {
     nock(serverUrl)
       .patch("/a/b.json")
-      .query({ print: "silent", writeSizeLimit: "tiny" })
+      .query({ print: "silent", writeSizeLimit: "tiny", disableTriggers: "false" })
       .reply(400, {
         error:
           "Data requested exceeds the maximum size that can be accessed with a single request.",