[addon] add a timeout for resolve method

addon/lib/promises.js

@@ -0,0 +1,22 @@
+/**
+ * Delay promise
+ */
+async function delay(duration) {
+  return new Promise((resolve) => setTimeout(resolve, duration));
+}
+
+/**
+ * Timeout promise after a set time in ms
+ */
+async function timeout(timeoutMs, promise, message = 'Timed out') {
+  return Promise.race([
+    promise,
+    new Promise(function (resolve, reject) {
+      setTimeout(function () {
+        reject(message);
+      }, timeoutMs);
+    })
+  ]);
+}
+
+module.exports = { delay, timeout };

addon/moch/debridlink.js

@@ -1,10 +1,10 @@
 const DebridLinkClient = require('debrid-link-api');
 const { Type } = require('../lib/types');
 const { isVideo, isArchive } = require('../lib/extension');
+const { delay } = require('../lib/promises');
 const StaticResponse = require('./static');
 const { getMagnetLink } = require('../lib/magnetHelper');
 const { chunkArray, BadTokenError } = require('./mochHelper');
-const delay = require('./delay');
 
 const KEY = 'debridlink';
 

addon/moch/delay.js

@@ -1 +0,0 @@
-module.exports = duration => new Promise((resolve) => setTimeout(resolve, duration));

addon/moch/moch.js

@@ -7,8 +7,10 @@ const debridlink = require('./debridlink');
 const putio = require('./putio');
 const StaticResponse = require('./static');
 const { cacheWrapResolvedUrl } = require('../lib/cache');
+const { timeout } = require('../lib/promises');
 const { BadTokenError } = require('./mochHelper');
 
+const RESOLVE_TIMEOUT = 10 * 60 * 1000; // 10 minutes
 const MIN_API_KEY_SYMBOLS = 15;
 const TOKEN_BLACKLIST = [];
 const RESOLVER_HOST = process.env.RESOLVER_HOST || 'http://localhost:7050';
@@ -87,12 +89,11 @@ async function resolve(parameters) {
     return Promise.reject("No valid parameters passed");
   }
   const id = `${parameters.mochKey}_${parameters.apiKey}_${parameters.infoHash}_${parameters.fileIndex}`;
-  const method = () => cacheWrapResolvedUrl(id, () => moch.instance.resolve(parameters))
+  const method = () => timeout(RESOLVE_TIMEOUT, cacheWrapResolvedUrl(id, () => moch.instance.resolve(parameters)))
       .catch(error => {
         console.warn(error);
         return StaticResponse.FAILED_UNEXPECTED;
       });
-  console.log(`Starting [${parameters.infoHash}] link resolve with queue size: ${unrestrictQueue.length()}`);
   return new Promise(((resolve, reject) => {
     unrestrictQueue.push({ id, method }, (error, result) => result ? resolve(result) : reject(error));
   }));

addon/moch/putio.js

@@ -1,6 +1,6 @@
 const PutioAPI = require('@putdotio/api-client').default
 const { isVideo } = require('../lib/extension');
-const delay = require('./delay');
+const { delay } = require('../lib/promises');
 const StaticResponse = require('./static');
 const { getMagnetLink } = require('../lib/magnetHelper');
 

addon/moch/realdebrid.js

@@ -1,7 +1,7 @@
 const RealDebridClient = require('real-debrid-api');
 const { Type } = require('../lib/types');
 const { isVideo, isArchive } = require('../lib/extension');
-const delay = require('./delay');
+const { delay } = require('../lib/promises');
 const StaticResponse = require('./static');
 const { getMagnetLink } = require('../lib/magnetHelper');
 const { chunkArray, BadTokenError } = require('./mochHelper');