feat: support props interval and update default timeout and interval

Author: lvqq

package.json

@@ -1,6 +1,6 @@
 {
   "name": "midjourney-fetch",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "description": "",
   "type": "module",
   "main": "./dist/index.js",

src/config.ts

@@ -1,4 +1,4 @@
 export const configs = {
-  timeout: 2 * 60 * 1000, // 2min
-  interval: 2 * 1000, // every 2 second
+  timeout: 5 * 60 * 1000, // 5 min
+  interval: 15 * 1000, // every 15 second
 };

src/interface.ts

@@ -6,7 +6,8 @@ export interface MidjourneyProps {
   channelId: string;
   serverId: string;
   token: string;
-  timeout?: number; // default timeout: 2 min
+  timeout?: number; // default timeout: 5 min
+  interval?: number; // default interval: 15s
 }
 
 export interface MessageAttachment {

src/midjourney.ts

@@ -15,14 +15,25 @@ export class Midjourney {
 
   timeout: number;
 
+  interval: number;
+
   debugger: boolean;
 
   constructor(props: MidjourneyProps) {
-    const { channelId, serverId, token, timeout = configs.timeout } = props;
+    const {
+      channelId,
+      serverId,
+      token,
+      timeout = configs.timeout,
+      interval = configs.interval,
+    } = props;
     this.channelId = channelId;
     this.serverId = serverId;
     this.token = token;
+
     this.timeout = timeout;
+    this.interval = interval;
+
     this.debugger = false;
   }
 
@@ -120,13 +131,13 @@ export class Midjourney {
    */
   async imagine(prompt: string) {
     await this.interactions(prompt);
-    const times = this.timeout / configs.interval;
+    const times = this.timeout / this.interval;
     let count = 0;
     let image: MessageAttachment | null = null;
     while (count < times) {
       try {
         count += 1;
-        await new Promise((res) => setTimeout(res, configs.interval));
+        await new Promise((res) => setTimeout(res, this.interval));
         this.log(count);
         const message = await this.getMessage(prompt);
         if (message && !isInProgress(message)) {