diff --git a/addon/moch/alldebrid.js b/addon/moch/alldebrid.js new file mode 100644 index 0000000..1ca3716 --- /dev/null +++ b/addon/moch/alldebrid.js @@ -0,0 +1,93 @@ +const AllDebridClient = require('all-debrid-api'); +const namedQueue = require('named-queue'); +const isVideo = require('../lib/video'); +const { getRandomProxy, getRandomUserAgent } = require('../lib/request_helper'); +const { cacheWrapResolvedUrl, cacheWrapProxy, cacheUserAgent } = require('../lib/cache'); + +const unrestrictQueue = new namedQueue((task, callback) => task.method() + .then(result => callback(false, result)) + .catch((error => callback(error)))); + +async function getCachedStreams(streams, apiKey) { + const options = await getDefaultOptions(apiKey); + const AD = new AllDebridClient(apiKey, options); + const hashes = streams.map(stream => stream.infoHash); + const available = await AD.magnet.instant(hashes) + .catch(error => { + console.warn('Failed AllDebrid cached torrent availability request: ', error); + return undefined; + }); + return available && available.data && available.data.magnets + .filter(magnet => magnet.instant) + .reduce((cachedStreams, magnet) => { + const stream = streams.find(stream => stream.infoHash === magnet.hash.toLowerCase()); + if (stream) { + const streamTitleParts = stream.title.replace(/\n👤.*/s, '').split('\n'); + const fileName = streamTitleParts[streamTitleParts.length - 1]; + const fileIndex = streamTitleParts.length === 2 ? stream.fileIdx : null; + const encodedFileName = encodeURIComponent(fileName); + cachedStreams[stream.infoHash] = `${apiKey}/${stream.infoHash}/${encodedFileName}/${fileIndex}`; + } + return cachedStreams; + }, {}) +} + +async function resolve({ ip, apiKey, infoHash, cachedEntryInfo, fileIndex }) { + if (!apiKey || !infoHash || !cachedEntryInfo) { + return Promise.reject("No valid parameters passed"); + } + const id = `${apiKey}_${infoHash}_${fileIndex}`; + const method = () => cacheWrapResolvedUrl(id, () => _unrestrict(ip, apiKey, infoHash, cachedEntryInfo, fileIndex)); + + return new Promise(((resolve, reject) => { + unrestrictQueue.push({ id, method }, (error, result) => result ? resolve(result) : reject(error)); + })); +} + +async function _unrestrict(ip, apiKey, infoHash, encodedFileName, fileIndex) { + console.log(`Unrestricting ${infoHash} [${fileIndex}]`); + const options = await getDefaultOptions(apiKey, ip); + const AD = new AllDebridClient(apiKey, options); + const cachedTorrent = await _createOrFindTorrent(AD, infoHash); + if (cachedTorrent && cachedTorrent.status === 'Ready') { + const targetFileName = decodeURIComponent(encodedFileName); + const videos = cachedTorrent.links.filter(link => isVideo(link.filename)); + const targetVideo = Number.isInteger(fileIndex) + ? videos.find(video => targetFileName.includes(video.filename)) + : videos.sort((a, b) => b.size - a.size)[0]; + const unrestrictedLink = await _unrestrictLink(AD, targetVideo.link); + console.log(`Unrestricted ${infoHash} [${fileIndex}] to ${unrestrictedLink}`); + return unrestrictedLink; + } + return Promise.reject("Failed AllDebrid adding torrent"); +} + +async function _createOrFindTorrent(AD, infoHash) { + return AD.magnet.status() + .then(response => response.data.magnets) + .then(torrents => torrents.find(torrent => torrent.hash === infoHash)) + .then(torrent => torrent || Promise.reject('No recent torrent found')) + .catch(() => AD.magnet.upload(infoHash) + .then(response => AD.magnet.status(response.data.magnets[0].id) + .then(statusResponse => statusResponse.data.magnets))) + .catch(error => { + console.warn('Failed AllDebrid torrent retrieval', error); + return undefined; + }); +} + +async function _unrestrictLink(AD, link) { + if (!link || !link.length) { + return Promise.reject("No available links found"); + } + return AD.link.unlock(link).then(response => response.data.link); +} + +async function getDefaultOptions(id, ip) { + const userAgent = await cacheUserAgent(id, () => getRandomUserAgent()).catch(() => getRandomUserAgent()); + const proxy = await cacheWrapProxy('moch', () => getRandomProxy()).catch(() => getRandomProxy()); + + return { proxy: proxy, headers: { 'User-Agent': userAgent } }; +} + +module.exports = { getCachedStreams, resolve }; \ No newline at end of file diff --git a/addon/moch/moch.js b/addon/moch/moch.js index 033bdff..b9687fb 100644 --- a/addon/moch/moch.js +++ b/addon/moch/moch.js @@ -1,5 +1,6 @@ const realdebrid = require('./realdebrid'); const premiumize = require('./premiumize'); +const alldebrid = require('./alldebrid'); const RESOLVER_HOST = process.env.RESOLVER_HOST || 'http://localhost:7050'; const MOCHS = { @@ -12,6 +13,11 @@ const MOCHS = { key: 'premiumize', instance: premiumize, shortName: 'PM' + }, + 'alldebrid': { + key: 'alldebrid', + instance: alldebrid, + shortName: 'AD' } }; diff --git a/addon/package-lock.json b/addon/package-lock.json index fbc7f03..d56dd5b 100644 --- a/addon/package-lock.json +++ b/addon/package-lock.json @@ -72,6 +72,14 @@ "uri-js": "^4.2.2" } }, + "all-debrid-api": { + "version": "1.0.0", + "resolved": "https://registry.npmjs.org/all-debrid-api/-/all-debrid-api-1.0.0.tgz", + "integrity": "sha512-PxaNrM36Cp8+J3M11MCCc8aGWDSaYj+7ciMfzuiQCg8LgjBdsd4dvB4txepitFbBZVVNo5Rl4meLjOD6iLa6IQ==", + "requires": { + "request": "^2.83.0" + } + }, "ansi-escapes": { "version": "3.2.0", "resolved": "https://registry.npmjs.org/ansi-escapes/-/ansi-escapes-3.2.0.tgz", diff --git a/addon/package.json b/addon/package.json index 7707d41..968dcec 100644 --- a/addon/package.json +++ b/addon/package.json @@ -8,6 +8,7 @@ "author": "TheBeastLT ", "license": "MIT", "dependencies": { + "all-debrid-api": "^1.0.0", "bottleneck": "^2.19.5", "cache-manager": "^2.11.1", "cache-manager-mongodb": "^0.2.2",