Merge branch 'main' of github.com:devforth/adminforth-upload

Author: yaroslav8765

index.ts

@@ -16,13 +16,18 @@ export default class UploadPlugin extends AdminForthPlugin {
 
   resourceConfig: AdminForthResource;
 
+  rateLimiter: RateLimiter;
+
   constructor(options: PluginOptions) {
     super(options, import.meta.url);
     this.options = options;
 
     // for calcualting average time
     this.totalCalls = 0;
     this.totalDuration = 0;
+    if (this.options.generation?.rateLimit?.limit) {
+    this.rateLimiter = new RateLimiter(this.options.generation.rateLimit?.limit)
+    }
   }
 
   instanceUniqueRepresentation(pluginOptions: any) : string {
@@ -344,15 +349,14 @@ export default class UploadPlugin extends AdminForthPlugin {
       path: `/plugin/${this.pluginInstanceId}/generate_images`,
       handler: async ({ body, adminUser, headers }) => {
         const { prompt, recordId } = body;
-        if (this.options.generation.rateLimit?.limit) {
+        if (this.rateLimiter) {
           // rate limit
           // const { error } = RateLimiter.checkRateLimit(
           //   this.pluginInstanceId, 
           //   this.options.generation.rateLimit?.limit,
           //   this.adminforth.auth.getClientIp(headers),
           // );
-          const rateLimiter = new RateLimiter(this.options.generation.rateLimit?.limit);
-          if (!rateLimiter.consume(`${this.pluginInstanceId}-${this.adminforth.auth.getClientIp(headers)}`)) {
+          if (!await this.rateLimiter.consume(`${this.pluginInstanceId}-${this.adminforth.auth.getClientIp(headers)}`)) {
             return { error: this.options.generation.rateLimit.errorMessage };
           }
         }