@shushed/helpers 0.0.53 → 0.0.55

package/dist/index.d.ts

@@ -28789,6 +28789,7 @@ declare const validateGoogleAuth: (opts: Opts & {
 }, headers: Record<string, string>, firestore: Firestore) => Promise<void>;
 declare const parseDateOrDefault: (date: string | null | Date, defaultValue?: Date | null) => Date | null;
 declare function getEventTime(x: any, defaultValue: Date | null): Date | null;
+declare function shortHash(input: string): string;
 
 type Level = 'env' | 'workflow' | 'trigger';
 declare class EnvEngine extends Runtime {
@@ -28844,17 +28845,34 @@ declare class PubSubHelper extends Runtime {
     }, payload: AtLeastOne<MessageBody, LastModifiedKeys>, orgMessage?: Message): Promise<string[]>;
     createOrUpdate(opts: {
         topicName: string;
-        dlq: boolean;
-        bigQueryTableId?: string | null;
-        retryLimit: number;
-        retryDelay: number;
-        ackDeadline: number;
+        createDeadLetterQueues: boolean;
+        pull?: null | {
+            retryLimit: number;
+            ackDeadline: number;
+            retryMinDelay: number;
+            retryMaxDelay: number;
+        };
+        push?: null | {
+            retryLimit: number;
+            ackDeadline: number;
+            retryMinDelay: number;
+            retryMaxDelay: number;
+        };
+        table?: null | {
+            tableId: string;
+            retryLimit: number;
+            retryMinDelay: number;
+            retryMaxDelay: number;
+            ackDeadline: number;
+        };
     }): Promise<{
-        topic: Topic;
-        dlqTopic: Topic | null;
-        bqSubscription: Subscription | null;
-        dlqSubscription: Subscription | null;
-        subscription: Subscription;
+        mainTopic: Topic | undefined;
+        mainDlqTopic: Topic | undefined;
+        tableDlqTopic: Topic | undefined;
+        mainSubscription: Subscription | null;
+        mainDlqSubscription: Subscription | null;
+        tableDlqSubscription: Subscription | null;
+        tableSubscription: Subscription | null;
     }>;
     private findTopic;
     private ensureTopicExists;
@@ -28862,24 +28880,41 @@ declare class PubSubHelper extends Runtime {
     private updateSubscription;
     private ensureSubscribtionExists;
     private ensureDLQSubscriptionExists;
+    deleteSubscriptions(topicName: string): Promise<{
+        mainTopic: Topic | undefined;
+        mainDlqTopic: Topic | undefined;
+        tableDlqTopic: Topic | undefined;
+        mainSubscription: Subscription | null;
+        mainDlqSubscription: Subscription | null;
+        tableDlqSubscription: Subscription | null;
+        tableSubscription: Subscription | null;
+    }>;
+    pause(topicName: string): Promise<{
+        subscriptionId: string;
+    } | {
+        subscriptionId: null;
+    }>;
+    resume(topicName: string): Promise<{
+        subscriptionId: string;
+    } | {
+        subscriptionId: null;
+    }>;
     deleteSubscription(subscriptionURN: string): Promise<any>;
 }
 
 declare class SchedulerHelper extends Runtime {
-    private extraQuery;
     private pubSubTarget;
     private pubSubProjectId;
     private schedulerLocationId;
     private schedulerClient;
     constructor(opts: Opts & {
-        extraQuery: Record<string, string>;
         serviceAccount: string;
         defaultServiceAccount: string;
         pubSubTarget?: Topic | null;
         schedulerLocationId?: string | null;
     }, schedulerClient: CloudSchedulerClient);
     get locationIds(): string[];
-    getTarget(): {
+    getTarget(extraQuery?: Record<string, string>): {
         pubsubTarget: {
             topicName: string;
             data: Buffer<ArrayBuffer>;
@@ -28899,9 +28934,10 @@ declare class SchedulerHelper extends Runtime {
         };
         pubsubTarget?: undefined;
     };
-    create({ schedule, timezone }: {
+    create({ schedule, timezone, extraQuery }: {
         schedule: string;
         timezone: string;
+        extraQuery: Record<string, string>;
     }): Promise<{
         status: _google_cloud_scheduler_build_protos_protos.google.rpc.IStatus | null | undefined;
         state: "STATE_UNSPECIFIED" | _google_cloud_scheduler_build_protos_protos.google.cloud.scheduler.v1.Job.State | "ENABLED" | "PAUSED" | "DISABLED" | "UPDATE_FAILED" | null | undefined;
@@ -28909,9 +28945,10 @@ declare class SchedulerHelper extends Runtime {
         httpTarget: _google_cloud_scheduler_build_protos_protos.google.cloud.scheduler.v1.IHttpTarget | null | undefined;
         pubsubTarget: _google_cloud_scheduler_build_protos_protos.google.cloud.scheduler.v1.IPubsubTarget | null | undefined;
     }>;
-    update({ schedule, timezone }: {
+    update({ schedule, timezone, extraQuery }: {
         schedule: string;
         timezone: string;
+        extraQuery: Record<string, string>;
     }): Promise<{
         status: _google_cloud_scheduler_build_protos_protos.google.rpc.IStatus | null | undefined;
         state: "STATE_UNSPECIFIED" | _google_cloud_scheduler_build_protos_protos.google.cloud.scheduler.v1.Job.State | "ENABLED" | "PAUSED" | "DISABLED" | "UPDATE_FAILED" | null | undefined;
@@ -29208,10 +29245,11 @@ declare const index_isPubSubRequest: typeof isPubSubRequest;
 declare const index_parseDateOrDefault: typeof parseDateOrDefault;
 declare const index_sanitize: typeof sanitize;
 declare const index_sanitizeToString: typeof sanitizeToString;
+declare const index_shortHash: typeof shortHash;
 declare const index_validate: typeof validate;
 declare const index_validateGoogleAuth: typeof validateGoogleAuth;
 declare namespace index {
-  export { index_BigQueryHelper as BigQueryHelper, index_EnvEngine as EnvEngine, index_JWKSHelper as JWKSHelper, type index_Message as Message, type index_NodeOptions as NodeOptions, index_PubSubHelper as PubSubHelper, index_Runtime as Runtime, index_SchedulerHelper as SchedulerHelper, index_Secrets as Secrets, type index_TriggerOnCreateOptions as TriggerOnCreateOptions, type index_TriggerOnExecuteOptions as TriggerOnExecuteOptions, index_getEventTime as getEventTime, index_isCronMessage as isCronMessage, index_isPubSubRequest as isPubSubRequest, index_parseDateOrDefault as parseDateOrDefault, index_sanitize as sanitize, index_sanitizeToString as sanitizeToString, index_validate as validate, index_validateGoogleAuth as validateGoogleAuth };
+  export { index_BigQueryHelper as BigQueryHelper, index_EnvEngine as EnvEngine, index_JWKSHelper as JWKSHelper, type index_Message as Message, type index_NodeOptions as NodeOptions, index_PubSubHelper as PubSubHelper, index_Runtime as Runtime, index_SchedulerHelper as SchedulerHelper, index_Secrets as Secrets, type index_TriggerOnCreateOptions as TriggerOnCreateOptions, type index_TriggerOnExecuteOptions as TriggerOnExecuteOptions, index_getEventTime as getEventTime, index_isCronMessage as isCronMessage, index_isPubSubRequest as isPubSubRequest, index_parseDateOrDefault as parseDateOrDefault, index_sanitize as sanitize, index_sanitizeToString as sanitizeToString, index_shortHash as shortHash, index_validate as validate, index_validateGoogleAuth as validateGoogleAuth };
 }
 
 export { index as lib, index$9 as schema, index$1 as types };

package/dist/index.js

@@ -298,6 +298,7 @@ __export(src_public_exports, {
   parseDateOrDefault: () => parseDateOrDefault,
   sanitize: () => sanitize,
   sanitizeToString: () => sanitizeToString,
+  shortHash: () => shortHash,
   validate: () => validate,
   validateGoogleAuth: () => validateGoogleAuth
 });
@@ -354,6 +355,9 @@ var sanitize = new import_deep_redact.DeepRedact({
   replaceStringByLength: true
 });
 
+// src-public/utils.ts
+var import_crypto = __toESM(require("crypto"));
+
 // src-public/jwks.ts
 var import_jose = require("jose");
 
@@ -536,9 +540,16 @@ function getEventTime(x, defaultValue) {
   }
   return defaultValue;
 }
+function shortHash(input) {
+  const hash = import_crypto.default.createHash("sha256").update(input).digest();
+  const base64 = hash.toString("base64").replace(/\+/g, "-").replace(/\//g, "_").replace(/=+$/, "");
+  return base64.substring(0, 8);
+}
 
 // src-public/env.ts
-var crypto = __toESM(require("crypto"));
+var crypto2 = __toESM(require("crypto"));
+global.storeLoading = global.storeLoading || {};
+var storeLoading = global.storeLoading;
 var EnvEngine = class extends Runtime {
   docRef;
   store;
@@ -556,7 +567,13 @@ var EnvEngine = class extends Runtime {
     };
   }
   async set(envs, level = "trigger", encryptWithValue) {
-    await this.initializeIfNeeded(level);
+    const timeout = 300;
+    await Promise.race([this.initializeIfNeeded(level), new Promise((resolve, reject) => setTimeout(() => {
+      reject(new Error(`Waitied ${timeout}ms. Terminating firestore initialize during the set function`));
+    }, timeout))]).catch((err) => {
+      console.warn("Invariant: While initializing the env engine during set - " + err.message + ` - ${this.envName} - ${this.workflowId} - ${this.triggerId}.
+        Failing quietly as the set might proceed, but this is not a good thing.`);
+    });
     let aes = null;
     if (encryptWithValue) {
       aes = new AES256GCM(Buffer.from(encryptWithValue, "utf8"));
@@ -587,13 +604,27 @@ var EnvEngine = class extends Runtime {
   async initializeIfNeeded(level = "trigger") {
     if (!this.store[level]) {
       if (this.docRef[level]) {
-        const docSnapshot = await this.docRef[level].get();
-        const data = docSnapshot.data();
-        this.store[level] = {};
-        for (const k in data) {
-          const v = data[k];
-          this.store[level][k] = typeof v !== "string" ? `${v === null || typeof v === "undefined" ? "" : `${v}`}` : v;
+        if (!storeLoading[this.docRef[level].path]) {
+          const promise = (async () => {
+            const docSnapshot = await this.docRef[level].get();
+            return docSnapshot;
+          })().then((x) => {
+            storeLoading[this.docRef[level].path] = null;
+            return x;
+          }).catch((err) => {
+            storeLoading[this.docRef[level].path] = null;
+            throw err;
+          });
+          storeLoading[this.docRef[level].path] = promise;
         }
+        return storeLoading[this.docRef[level].path].then((docSnapshot) => {
+          const data = docSnapshot.data() || {};
+          this.store[level] = {};
+          for (const k in data) {
+            const v = data[k];
+            this.store[level][k] = typeof v !== "string" ? `${v === null || typeof v === "undefined" ? "" : `${v}`}` : v;
+          }
+        });
       } else {
         this.store[level] = {};
       }
@@ -633,7 +664,7 @@ var EnvEngine = class extends Runtime {
 var AES256GCM = class {
   normalizedKey;
   constructor(key) {
-    this.normalizedKey = crypto.createHash("sha256").update(key).digest().slice(0, 32);
+    this.normalizedKey = crypto2.createHash("sha256").update(key).digest().slice(0, 32);
   }
   // Pad payload to a multiple of 8 bytes
   pad(payload) {
@@ -647,8 +678,8 @@ var AES256GCM = class {
   encrypt(payload) {
     const normalizedWrappingKey = this.normalizedKey;
     const paddedPayload = this.pad(Buffer.from(payload));
-    const iv = crypto.randomBytes(12);
-    const cipher = crypto.createCipheriv(
+    const iv = crypto2.randomBytes(12);
+    const cipher = crypto2.createCipheriv(
       "aes-256-gcm",
       normalizedWrappingKey,
       iv
@@ -665,7 +696,7 @@ var AES256GCM = class {
     const encryptedPayload = Buffer.from(encryptedKey64, "base64");
     const iv = Buffer.from(iv64, "base64");
     const authTag = Buffer.from(authTag64, "base64");
-    const decipher = crypto.createDecipheriv(
+    const decipher = crypto2.createDecipheriv(
       "aes-256-gcm",
       this.normalizedKey,
       iv
@@ -760,53 +791,112 @@ var PubSubHelper = class extends Runtime {
     return Promise.all(promises);
   }
   async createOrUpdate(opts) {
-    const subscriptionName = `${this.envName}---${this.triggerId}`;
+    const [topics] = await this.pubSub.getTopics();
+    const mainDlqTopicName = `${opts.topicName}-${shortHash(this.triggerId)}-dlq`;
+    const tableDlqTopicName = `${opts.topicName}-${shortHash(this.triggerId)}-bq-dlq`;
+    let mainTopic = topics.find((x) => this.getNameFromFullyQualifiedName(x.name) === opts.topicName);
+    let mainDlqTopic = topics.find((x) => this.getNameFromFullyQualifiedName(x.name) === mainDlqTopicName);
+    let tableDlqTopic = topics.find((x) => this.getNameFromFullyQualifiedName(x.name) === tableDlqTopicName);
+    const mainSubscriptionName = `${opts.topicName}--${shortHash(this.triggerId)}`;
+    const mainDlqSubscriptionName = `${opts.topicName}--${shortHash(this.triggerId)}-dlq`;
+    const tableSubscriptionName = `${opts.topicName}--${shortHash(this.triggerId)}-bq`;
+    const tableDlqSubscriptionName = `${opts.topicName}--${shortHash(this.triggerId)}-bq-dlq`;
+    let mainTopicSubscriptions = null;
+    let mainDlqTopicSubscriptions = null;
+    let tableDlqTopicSubscriptions = null;
+    if (mainTopic) {
+      [mainTopicSubscriptions] = await mainTopic.getSubscriptions();
+    }
+    if (mainDlqTopic) {
+      [mainDlqTopicSubscriptions] = await mainDlqTopic.getSubscriptions();
+    }
+    if (tableDlqTopic) {
+      [tableDlqTopicSubscriptions] = await tableDlqTopic.getSubscriptions();
+    }
+    let mainSubscription = mainTopicSubscriptions?.find((x) => this.getNameFromFullyQualifiedName(x.name) === mainSubscriptionName) || null;
+    let mainDlqSubscription = mainDlqTopicSubscriptions?.find((x) => this.getNameFromFullyQualifiedName(x.name) === mainDlqSubscriptionName) || null;
+    let tableSubscription = mainTopicSubscriptions?.find((x) => this.getNameFromFullyQualifiedName(x.name) === tableSubscriptionName) || null;
+    let tableDlqSubscription = tableDlqTopicSubscriptions?.find((x) => this.getNameFromFullyQualifiedName(x.name) === tableDlqSubscriptionName) || null;
     const topic = await this.ensureTopicExists(opts.topicName);
-    let dlqTopic = null;
-    let dlqSubscription = null;
-    if (opts.dlq) {
-      const dlqTopicName = `${opts.topicName}-dlq`;
-      const dlqSubscriptionName = `${subscriptionName}-dlq`;
-      dlqTopic = await this.ensureTopicExists(dlqTopicName);
-      dlqSubscription = await this.ensureDLQSubscriptionExists(dlqSubscriptionName, {
-        topic: dlqTopic
-      });
-    } else {
-      await this.deleteSubscription(`${subscriptionName}-dlq`);
+    if (opts.table || opts.pull || opts.push) {
+      mainTopic = await this.ensureTopicExists(opts.topicName);
     }
-    let bqSubscription = null;
-    if (opts.bigQueryTableId) {
-      const bigQuerySubscriptionName = `${subscriptionName}-bq`;
-      const bigQueryDlqTopicName = `${opts.topicName}-bq-dlq`;
-      const bigQueryDlqSubscriptionName = `${subscriptionName}-bq-dlq`;
-      dlqTopic = await this.ensureTopicExists(bigQueryDlqTopicName);
-      dlqSubscription = await this.ensureDLQSubscriptionExists(bigQueryDlqSubscriptionName, {
-        topic: dlqTopic
-      });
-      bqSubscription = await this.ensureSubscribtionExists(bigQuerySubscriptionName, {
+    if (!opts.table) {
+      if (tableSubscription) {
+        await tableSubscription.delete();
+        tableSubscription = null;
+      }
+      if (tableDlqSubscription) {
+        await tableDlqSubscription.delete();
+        tableDlqSubscription = null;
+      }
+    } else {
+      if (opts.createDeadLetterQueues) {
+        tableDlqTopic = await this.ensureTopicExists(tableDlqTopicName);
+        tableDlqSubscription = await this.ensureDLQSubscriptionExists(tableDlqSubscriptionName, {
+          topic: tableDlqTopic
+        });
+      }
+      tableSubscription = await this.ensureSubscribtionExists(tableSubscriptionName, {
         topic,
-        tableId: opts.bigQueryTableId,
-        dlqTopic
+        tableId: opts.table.tableId,
+        ackDeadline: opts.table.ackDeadline,
+        retryLimit: opts.table.retryLimit,
+        retryMinDelay: opts.table.retryMinDelay,
+        retryMaxDelay: opts.table.retryMaxDelay,
+        dlqTopic: mainTopic
       });
+    }
+    const pullPushSubscription = opts.pull || opts.push;
+    if (!pullPushSubscription) {
+      if (mainSubscription) {
+        await mainSubscription.delete();
+        mainSubscription = null;
+      }
+      if (mainDlqSubscription) {
+        await mainDlqSubscription.delete();
+        mainDlqSubscription = null;
+      }
     } else {
-      await this.deleteSubscription(`${subscriptionName}-bq`);
-      await this.deleteSubscription(`${subscriptionName}-bq-dlq`);
+      if (opts.createDeadLetterQueues) {
+        mainDlqTopic = await this.ensureTopicExists(mainDlqTopicName);
+        mainDlqSubscription = await this.ensureDLQSubscriptionExists(mainDlqSubscriptionName, {
+          topic: mainDlqTopic
+        });
+        mainSubscription = await this.ensureSubscribtionExists(mainSubscriptionName, {
+          topic,
+          dlqTopic: mainDlqTopic,
+          endpointUrl: opts.push ? this.runtimeUrl + `/executeWorkflow/${this.workflowId}/${this.triggerId}?topic_name=${opts.topicName}` : null,
+          retryMinDelay: pullPushSubscription.retryMinDelay,
+          retryMaxDelay: pullPushSubscription.retryMaxDelay,
+          retryLimit: pullPushSubscription.retryLimit,
+          ackDeadline: pullPushSubscription.ackDeadline
+        });
+      }
+    }
+    if (!opts.createDeadLetterQueues) {
+      if (mainDlqSubscription) {
+        await mainDlqSubscription.delete();
+        mainDlqSubscription = null;
+      }
+      if (tableDlqSubscription) {
+        await tableDlqSubscription.delete();
+        tableDlqSubscription = null;
+      }
+      if (tableDlqTopic) {
+        await tableDlqTopic.delete();
+      }
+      if (mainDlqTopic)
+        await mainDlqTopic.delete();
     }
-    const subscription = await this.ensureSubscribtionExists(subscriptionName, {
-      topic,
-      dlqTopic,
-      retryMinDelay: opts.retryDelay,
-      retryMaxDelay: opts.retryDelay,
-      retryLimit: opts.retryLimit,
-      endpointUrl: this.runtimeUrl + `/executeWorkflow/${this.workflowId}/${this.triggerId}`,
-      ackDeadline: opts.ackDeadline
-    });
     return {
-      topic,
-      dlqTopic,
-      bqSubscription,
-      dlqSubscription,
-      subscription
+      mainTopic,
+      mainDlqTopic,
+      tableDlqTopic,
+      mainSubscription,
+      mainDlqSubscription,
+      tableDlqSubscription,
+      tableSubscription
     };
   }
   async findTopic(topicName) {
@@ -959,7 +1049,7 @@ var PubSubHelper = class extends Runtime {
     }
     if (subscription && nextMetadata) {
       try {
-        return null;
+        await subscription.setMetadata(nextMetadata);
       } catch (err) {
         console.error("Invariant: While updating subscription - " + err.message + ` - ${this.pubSub.projectId} - ${subscriptionName} - ${JSON.stringify(nextMetadata)}`);
         throw new Error("Invariant: While updating subscription - " + err.message);
@@ -1000,6 +1090,63 @@ var PubSubHelper = class extends Runtime {
     }
     return subscription;
   }
+  async deleteSubscriptions(topicName) {
+    return this.createOrUpdate({
+      topicName,
+      createDeadLetterQueues: false,
+      pull: null,
+      push: null,
+      table: null
+    });
+  }
+  async pause(topicName) {
+    const [topics] = await this.pubSub.getTopics();
+    const mainTopic = topics.find((x) => this.getNameFromFullyQualifiedName(x.name) === topicName);
+    let mainTopicSubscriptions = null;
+    if (mainTopic) {
+      [mainTopicSubscriptions] = await mainTopic.getSubscriptions();
+    }
+    const mainSubscriptionName = `${topicName}--${shortHash(this.triggerId)}`;
+    const mainSubscription = mainTopicSubscriptions?.find((x) => this.getNameFromFullyQualifiedName(x.name) === mainSubscriptionName) || null;
+    if (mainSubscription) {
+      await mainSubscription.setMetadata({
+        pushConfig: null
+      });
+      return {
+        subscriptionId: mainSubscription.name
+      };
+    }
+    return {
+      subscriptionId: null
+    };
+  }
+  async resume(topicName) {
+    const [topics] = await this.pubSub.getTopics();
+    const mainTopic = topics.find((x) => this.getNameFromFullyQualifiedName(x.name) === topicName);
+    let mainTopicSubscriptions = null;
+    if (mainTopic) {
+      [mainTopicSubscriptions] = await mainTopic.getSubscriptions();
+    }
+    const mainSubscriptionName = `${topicName}--${shortHash(this.triggerId)}`;
+    const mainSubscription = mainTopicSubscriptions?.find((x) => this.getNameFromFullyQualifiedName(x.name) === mainSubscriptionName) || null;
+    if (mainSubscription) {
+      await mainSubscription.setMetadata({
+        pushConfig: {
+          pushEndpoint: this.runtimeUrl + "/executeWorkflow/" + this.workflowId + "/" + this.triggerId + `?topic_name=${topicName}`,
+          oidcToken: {
+            serviceAccountEmail: this.serviceAccountEmail,
+            audience: process.env.GCLOUD_PROJECT + "/" + this.triggerId
+          }
+        }
+      });
+      return {
+        subscriptionId: mainSubscription.name
+      };
+    }
+    return {
+      subscriptionId: null
+    };
+  }
   async deleteSubscription(subscriptionURN) {
     const sub = this.pubSub.subscription(subscriptionURN);
     return sub.delete().catch((err) => {
@@ -1016,14 +1163,12 @@ var pubsub_default = PubSubHelper;
 
 // src-public/scheduler.ts
 var SchedulerHelper = class extends Runtime {
-  extraQuery;
   pubSubTarget;
   pubSubProjectId;
   schedulerLocationId;
   schedulerClient;
   constructor(opts, schedulerClient) {
     super(opts);
-    this.extraQuery = opts.extraQuery;
     this.pubSubTarget = opts.pubSubTarget || null;
     this.pubSubProjectId = this.pubSubTarget?.name?.split("/")[1] || null;
     this.schedulerLocationId = opts.schedulerLocationId || "europe-west2";
@@ -1041,7 +1186,13 @@ var SchedulerHelper = class extends Runtime {
       "us-west1"
     ];
   }
-  getTarget() {
+  getTarget(extraQuery = {}) {
+    extraQuery = Object.assign({}, extraQuery, {
+      "created_by_wid": this.workflowId,
+      "created_by_tid": this.triggerId,
+      "created_by_sevn": this.systemEnvName,
+      "cron": "1"
+    });
     if (this.pubSubTarget) {
       return {
         pubsubTarget: {
@@ -1054,7 +1205,7 @@ var SchedulerHelper = class extends Runtime {
     }
     return {
       httpTarget: {
-        uri: `${this.runtimeUrl}/executeWorkflow/${this.workflowId}/${this.triggerId}?${new URLSearchParams(this.extraQuery).toString()}`,
+        uri: `${this.runtimeUrl}/executeWorkflow/${this.workflowId}/${this.triggerId}?${new URLSearchParams(extraQuery).toString()}`,
         httpMethod: "POST",
         headers: {
           "Content-Type": "application/json"
@@ -1066,7 +1217,7 @@ var SchedulerHelper = class extends Runtime {
       }
     };
   }
-  async create({ schedule, timezone }) {
+  async create({ schedule, timezone, extraQuery }) {
     let parent;
     let name;
     if (this.pubSubProjectId) {
@@ -1085,7 +1236,7 @@ var SchedulerHelper = class extends Runtime {
           name,
           schedule,
           timeZone: timezone,
-          ...this.getTarget()
+          ...this.getTarget(extraQuery)
         }
       });
       return {
@@ -1098,14 +1249,14 @@ var SchedulerHelper = class extends Runtime {
     } catch (error) {
       if (error.code === 6) {
         return this.update(
-          { schedule, timezone }
+          { schedule, timezone, extraQuery }
         );
       } else {
         throw error;
       }
     }
   }
-  async update({ schedule, timezone }) {
+  async update({ schedule, timezone, extraQuery }) {
     let parent;
     let name;
     if (this.pubSubProjectId) {
@@ -1120,7 +1271,7 @@ var SchedulerHelper = class extends Runtime {
     try {
       const response = await this.schedulerClient.updateJob({
         job: {
-          ...this.getTarget(),
+          ...this.getTarget(extraQuery),
           name,
           schedule,
           timeZone: timezone
@@ -1140,7 +1291,7 @@ var SchedulerHelper = class extends Runtime {
           name,
           schedule,
           timeZone: timezone,
-          ...this.getTarget()
+          ...this.getTarget(extraQuery)
         }
       });
       return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shushed/helpers",
-  "version": "0.0.53",
+  "version": "0.0.55",
   "author": "",
   "license": "UNLICENSED",
   "description": "",