WaitMessage add timestamp options

erictik · Jun 16, 2023 · d198d2f · d198d2f
1 parent a15e8c7
commit d198d2f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/midjourney.message.ts b/src/midjourney.message.ts
@@ -93,8 +93,12 @@ export class MidjourneyMessage {
   UriToHash(uri: string) {
     return uri.split("_").pop()?.split(".")[0] ?? "";
   }
-  async WaitMessage(prompt: string, loading?: LoadingHandler) {
-    var timestamp = Date.now();
+  async WaitMessage(
+    prompt: string,
+    loading?: LoadingHandler,
+    timestamp?: number
+  ) {
+    timestamp = timestamp ?? Date.now();
     for (let i = 0; i < this.config.MaxWait; i++) {
       const msg = await this.FilterMessages(timestamp, prompt, loading);
       if (msg !== null) {