botium-core 1.12.1 → 1.12.2

package/dist/botium-cjs.js

@@ -79,7 +79,7 @@ var express__default = /*#__PURE__*/_interopDefaultLegacy(express);
 var bodyParser__default = /*#__PURE__*/_interopDefaultLegacy(bodyParser);
 
 var name = "botium-core";
-var version$1 = "1.12.1";
+var version$1 = "1.12.2";
 var description = "The Selenium for Chatbots";
 var main = "index.js";
 var module$1 = "dist/botium-es.js";
@@ -357,6 +357,7 @@ var Capabilities = {
   // API Calls Rate Limiting
   RATELIMIT_USERSAYS_MAXCONCURRENT: 'RATELIMIT_USERSAYS_MAXCONCURRENT',
   RATELIMIT_USERSAYS_MINTIME: 'RATELIMIT_USERSAYS_MINTIME',
+  RATELIMIT_BOTTLENECK_FN: 'RATELIMIT_BOTTLENECK_FN',
   SECURITY_ALLOW_UNSAFE: 'SECURITY_ALLOW_UNSAFE',
   PRECOMPILERS: 'PRECOMPILERS'
 };
@@ -491,6 +492,7 @@ Capabilities.LOGIC_HOOKS;
 Capabilities.USER_INPUTS;
 Capabilities.RATELIMIT_USERSAYS_MAXCONCURRENT;
 Capabilities.RATELIMIT_USERSAYS_MINTIME;
+Capabilities.RATELIMIT_BOTTLENECK_FN;
 Capabilities.SECURITY_ALLOW_UNSAFE;
 Capabilities.PRECOMPILERS;
 
@@ -7580,7 +7582,7 @@ var BaseContainer_1 = class BaseContainer {
     this.tempDirectory = tempDirectory;
     this.cleanupTasks = [];
     this.queues = {};
-    this.userSaysLimiter = null;
+    this.bottleneck = null;
   }
 
   Validate() {
@@ -7590,18 +7592,33 @@ var BaseContainer_1 = class BaseContainer {
     this.onBotResponseHook = getHook$1(this.caps, this.caps[Capabilities.CUSTOMHOOK_ONBOTRESPONSE]);
     this.onStopHook = getHook$1(this.caps, this.caps[Capabilities.CUSTOMHOOK_ONSTOP]);
     this.onCleanHook = getHook$1(this.caps, this.caps[Capabilities.CUSTOMHOOK_ONCLEAN]);
-    return Promise.resolve();
-  }
 
-  Build() {
-    if (this.caps[Capabilities.RATELIMIT_USERSAYS_MAXCONCURRENT] || this.caps[Capabilities.RATELIMIT_USERSAYS_MINTIME]) {
+    if (this.caps[Capabilities.RATELIMIT_BOTTLENECK_FN]) {
+      if (lodash__default["default"].isFunction(this.caps[Capabilities.RATELIMIT_BOTTLENECK_FN])) {
+        this.bottleneck = this.caps[Capabilities.RATELIMIT_BOTTLENECK_FN];
+        debug$7('Validate: Applying userSays rate limits from capability');
+      } else {
+        const limiter = new bottleneck__default["default"](this.caps[Capabilities.RATELIMIT_BOTTLENECK_FN]);
+
+        this.bottleneck = fn => limiter.schedule(fn);
+
+        debug$7(`Validate: Applying userSays rate limits ${util__default["default"].inspect(this.caps[Capabilities.RATELIMIT_BOTTLENECK_FN])}`);
+      }
+    } else if (this.caps[Capabilities.RATELIMIT_USERSAYS_MAXCONCURRENT] || this.caps[Capabilities.RATELIMIT_USERSAYS_MINTIME]) {
       const opts = {};
       if (this.caps[Capabilities.RATELIMIT_USERSAYS_MAXCONCURRENT]) opts.maxConcurrent = this.caps[Capabilities.RATELIMIT_USERSAYS_MAXCONCURRENT];
       if (this.caps[Capabilities.RATELIMIT_USERSAYS_MINTIME]) opts.minTime = this.caps[Capabilities.RATELIMIT_USERSAYS_MINTIME];
-      this.userSaysLimiter = new bottleneck__default["default"](opts);
-      debug$7(`Build: Applying userSays rate limits ${util__default["default"].inspect(opts)}`);
+      const limiter = new bottleneck__default["default"](opts);
+
+      this.bottleneck = fn => limiter.schedule(fn);
+
+      debug$7(`Validate: Applying userSays rate limits ${util__default["default"].inspect(opts)}`);
     }
 
+    return Promise.resolve();
+  }
+
+  Build() {
     return new Promise((resolve, reject) => {
       this._RunCustomHook('onBuild', this.onBuildHook).then(() => resolve(this)).catch(err => reject(err));
     });
@@ -7627,8 +7644,8 @@ var BaseContainer_1 = class BaseContainer {
       meMsg
     }).then(() => this.UserSaysImpl(meMsg));
 
-    if (this.userSaysLimiter) {
-      return this.userSaysLimiter.schedule(run);
+    if (this.bottleneck) {
+      return this.bottleneck(run);
     } else {
       return run();
     }
@@ -8116,10 +8133,14 @@ mustache__default["default"].escape = s => s;
 var SimpleRestContainer_1 = class SimpleRestContainer {
   constructor({
     queueBotSays,
-    caps
+    caps,
+    bottleneck
   }) {
     this.queueBotSays = queueBotSays;
     this.caps = Object.assign({}, Defaults, caps);
+
+    this.bottleneck = bottleneck || (fn => fn());
+
     this.processInbound = false;
     this.redisTopic = this.caps[Capabilities.SIMPLEREST_REDIS_TOPIC] || 'SIMPLEREST_INBOUND_SUBSCRIPTION';
 
@@ -8658,7 +8679,7 @@ var SimpleRestContainer_1 = class SimpleRestContainer {
         err,
         response,
         body
-      } = await new Promise(resolve => {
+      } = await this.bottleneck(() => new Promise(resolve => {
         request__default["default"](pingConfig, (err, response, body) => {
           resolve({
             err,
@@ -8666,7 +8687,7 @@ var SimpleRestContainer_1 = class SimpleRestContainer {
             body
           });
         });
-      });
+      }));
 
       if (err) {
         debug$4(`_waitForUrlResponse error on url check ${pingConfig.uri}: ${err}`);
@@ -9190,6 +9211,7 @@ var PluginConnectorContainer_1 = class PluginConnectorContainer extends BaseCont
     this.pluginInstance = tryLoadPlugin(this.caps[Capabilities.CONTAINERMODE], this.caps[Capabilities.PLUGINMODULEPATH], {
       container: this,
       queueBotSays: msg => this._QueueBotSays(msg),
+      bottleneck: this.bottleneck,
       eventEmitter: this.eventEmitter,
       caps: this.caps,
       sources: this.sources,