90 lines
3.2 KiB
JavaScript
90 lines
3.2 KiB
JavaScript
const namedQueue = require('named-queue');
|
|
const options = require('./options');
|
|
const realdebrid = require('./realdebrid');
|
|
const premiumize = require('./premiumize');
|
|
const alldebrid = require('./alldebrid');
|
|
const StaticResponse = require('./static');
|
|
const { cacheWrapResolvedUrl } = require('../lib/cache');
|
|
|
|
const RESOLVER_HOST = process.env.RESOLVER_HOST || 'http://localhost:7050';
|
|
const MOCHS = {
|
|
'realdebrid': {
|
|
key: 'realdebrid',
|
|
instance: realdebrid,
|
|
shortName: 'RD'
|
|
},
|
|
'premiumize': {
|
|
key: 'premiumize',
|
|
instance: premiumize,
|
|
shortName: 'PM'
|
|
},
|
|
'alldebrid': {
|
|
key: 'alldebrid',
|
|
instance: alldebrid,
|
|
shortName: 'AD'
|
|
}
|
|
};
|
|
|
|
const unrestrictQueue = new namedQueue((task, callback) => task.method()
|
|
.then(result => callback(false, result))
|
|
.catch((error => callback(error))));
|
|
|
|
async function applyMochs(streams, config) {
|
|
if (!streams || !streams.length) {
|
|
return streams;
|
|
}
|
|
const includeDownloadLinks = options.includeDownloadLinks(config);
|
|
|
|
return Promise.all(Object.keys(config)
|
|
.filter(configKey => MOCHS[configKey])
|
|
.map(configKey => MOCHS[configKey])
|
|
.map(moch => moch.instance.getCachedStreams(streams, config[moch.key])
|
|
.then(mochStreams => ({ moch, mochStreams }))
|
|
.catch(error => console.warn(error))))
|
|
.then(mochResults => mochResults
|
|
.filter(result => result && result.mochStreams)
|
|
.reduce((resultStreams, { moch, mochStreams }) => {
|
|
resultStreams
|
|
.filter(stream => stream.infoHash)
|
|
.filter(stream => mochStreams[stream.infoHash])
|
|
.forEach(stream => {
|
|
const cachedEntry = mochStreams[stream.infoHash];
|
|
if (cachedEntry.cached) {
|
|
stream.name = `[${moch.shortName}+] ${stream.name}`;
|
|
stream.url = `${RESOLVER_HOST}/${moch.key}/${cachedEntry.url}`;
|
|
delete stream.infoHash;
|
|
delete stream.fileIndex;
|
|
} else if (includeDownloadLinks) {
|
|
resultStreams.push({
|
|
name: `[${moch.shortName} download] ${stream.name}`,
|
|
title: stream.title,
|
|
url: `${RESOLVER_HOST}/${moch.key}/${cachedEntry.url}`
|
|
})
|
|
}
|
|
});
|
|
return resultStreams;
|
|
}, streams));
|
|
}
|
|
|
|
async function resolve(parameters) {
|
|
const moch = MOCHS[parameters.mochKey];
|
|
if (!moch) {
|
|
return Promise.reject('Not a valid moch provider');
|
|
}
|
|
|
|
if (!parameters.apiKey || !parameters.infoHash || !parameters.cachedEntryInfo) {
|
|
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))
|
|
.catch(error => {
|
|
console.warn(error);
|
|
return StaticResponse.FAILED_UNEXPECTED;
|
|
});
|
|
|
|
return new Promise(((resolve, reject) => {
|
|
unrestrictQueue.push({ id, method }, (error, result) => result ? resolve(result) : reject(error));
|
|
}));
|
|
}
|
|
|
|
module.exports = { applyMochs, resolve } |