70 lines
2.3 KiB
JavaScript
70 lines
2.3 KiB
JavaScript
const options = require('./options');
|
|
const realdebrid = require('./realdebrid');
|
|
const premiumize = require('./premiumize');
|
|
const alldebrid = require('./alldebrid');
|
|
|
|
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'
|
|
}
|
|
};
|
|
|
|
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');
|
|
}
|
|
return moch.instance.resolve(parameters);
|
|
}
|
|
|
|
module.exports = { applyMochs, resolve } |