tools/knightcrawler
1
0
Fork 0
mirror of https://github.com/knightcrawler-stremio/knightcrawler.git synced 2024-12-20 03:29:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[addon] passed cached file ids in the urls

Browse Source
This commit is contained in:
TheBeastLT
2020-03-17 22:34:33 +01:00
parent a1d6e5fa80
commit 6490b2fbe8
2 changed files with 31 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
addon/moch/realdebrid.js
View File

@@ -15,12 +15,16 @@ async function applyMoch(streams, apiKey) {
const available = await _instantAvailability(RD, hashes); const available = await _instantAvailability(RD, hashes);
if (available) { if (available) {
Object.entries(available) Object.entries(available)
.filter(([key, value]) => getCachedFileIds(streamMapping[key.toLowerCase()].fileIdx, value).length) .map(([key, value]) => ({
.map(([key]) => key.toLowerCase()) cachedInfoHash: key.toLowerCase(),
.map(cachedInfoHash => streamMapping[cachedInfoHash]) cachedFileIds: getCachedFileIds(streamMapping[key.toLowerCase()].fileIdx, value)
.forEach(stream => { }))
.filter(cachedEntry => cachedEntry.cachedFileIds && cachedEntry.cachedFileIds.length)
.forEach(cachedEntry => {
const stream = streamMapping[cachedEntry.cachedInfoHash];
const cachedIds = cachedEntry.cachedFileIds.join(',');
stream.name = `[RD Cached]\n${stream.name}`; stream.name = `[RD Cached]\n${stream.name}`;
stream.url = `${ADDON_HOST}/realdebrid/${apiKey}/${stream.infoHash}/${stream.fileIdx}`; stream.url = `${ADDON_HOST}/realdebrid/${apiKey}/${stream.infoHash}/${cachedIds}/${stream.fileIdx}`;
delete stream.infoHash; delete stream.infoHash;
delete stream.fileIndex; delete stream.fileIndex;
}) })
@@ -29,18 +33,18 @@ async function applyMoch(streams, apiKey) {
return streams; return streams;
} }
async function unrestrict(apiKey, infoHash, fileIndex) { async function unrestrict(apiKey, infoHash, cachedFileIds, fileIndex) {
if (!apiKey || !infoHash) { if (!apiKey || !infoHash || !cachedFileIds || !cachedFileIds.length) {
return Promise.reject("No valid parameters passed"); return Promise.reject("No valid parameters passed");
} }
const key = `${apiKey}_${infoHash}_${fileIndex}`; const key = `${apiKey}_${infoHash}_${fileIndex}`;
return cacheWrapUnrestricted(key, () => _unrestrict(apiKey, infoHash, fileIndex)) return cacheWrapUnrestricted(key, () => _unrestrict(apiKey, infoHash, cachedFileIds, fileIndex))
} }
async function _unrestrict(apiKey, infoHash, fileIndex) { async function _unrestrict(apiKey, infoHash, cachedFileIds, fileIndex) {
console.log(`Unrestricting ${infoHash} [${fileIndex}]`); console.log(`Unrestricting ${infoHash} [${fileIndex}]`);
const RD = new RealDebridClient(apiKey); const RD = new RealDebridClient(apiKey);
const torrentId = await _createOrFindTorrentId(RD, infoHash); const torrentId = await _createOrFindTorrentId(RD, infoHash, cachedFileIds);
console.log(`Retrieved torrentId: ${torrentId}`); console.log(`Retrieved torrentId: ${torrentId}`);
if (torrentId) { if (torrentId) {
const info = await RD.torrents.info(torrentId); const info = await RD.torrents.info(torrentId);
@@ -55,12 +59,12 @@ async function _unrestrict(apiKey, infoHash, fileIndex) {
return Promise.reject("Failed adding torrent"); return Promise.reject("Failed adding torrent");
} }
async function _createOrFindTorrentId(RD, infoHash) { async function _createOrFindTorrentId(RD, infoHash, cachedFileIds) {
return RD.torrents.get(0, 1) return RD.torrents.get(0, 1)
.then(torrents => torrents.find(torrent => torrent.hash.toLowerCase() === infoHash)) .then(torrents => torrents.find(torrent => torrent.hash.toLowerCase() === infoHash))
.then(torrent => torrent && torrent.id || Promise.reject('No recent torrent found')) .then(torrent => torrent && torrent.id || Promise.reject('No recent torrent found'))
.catch((error) => RD.torrents.addMagnet(encode({ infoHash })) .catch((error) => RD.torrents.addMagnet(encode({ infoHash }))
.then(response => RD.torrents.selectFiles(response.id) .then(response => RD.torrents.selectFiles(response.id, cachedFileIds)
.then((() => response.id)))) .then((() => response.id))))
.catch(error => { .catch(error => {
console.warn('Failed RealDebrid torrent retrieval', error); console.warn('Failed RealDebrid torrent retrieval', error);
@@ -81,9 +85,12 @@ async function _unrestrictLink(RD, link) {
return Promise.reject("No available links found"); return Promise.reject("No available links found");
} }
return RD._post('unrestrict/link', { form: { link }, proxy: getProxy() }) return RD._post('unrestrict/link', { form: { link }, proxy: getProxy() })
.then(unrestrictedLink => { .then(unrestrictedLink => RD.streaming.transcode(unrestrictedLink.id))
console.log(`Unrestricted ${link} to ${unrestrictedLink.download}`); .then(transcodedLink => {
return Promise.resolve(unrestrictedLink.download); const url = transcodedLink.apple && transcodedLink.apple.full
|| transcodedLink[Object.keys(transcodedLink)[0]].full;
console.log(`Unrestricted ${link} to ${url}`);
return url;
}); });
} }
@@ -92,11 +99,13 @@ function getCachedFileIds(fileIndex, hosterResults) {
return []; return [];
} }
// if not all cached files are videos, then the torrent will be zipped to a rar // if not all cached files are videos, then the torrent will be zipped to a rar
const cachedTorrent = Object.values(hosterResults) const cachedTorrents = Object.values(hosterResults)
.reduce((a, b) => a.concat(b), []) .reduce((a, b) => a.concat(b), [])
.filter(cached => isNaN(fileIndex) && Object.keys(cached).length || cached[fileIndex + 1]) .filter(cached => !Number.isInteger(fileIndex) && Object.keys(cached).length || cached[fileIndex + 1])
.find(cached => Object.values(cached).every(file => isVideo(file.filename))); .filter(cached => Object.values(cached).every(file => isVideo(file.filename)))
return cachedTorrent && Object.keys(cachedTorrent) || []; .map(cached => Object.keys(cached))
.sort((a, b) => b.length - a.length);
return cachedTorrents.length && cachedTorrents[0] || [];
} }
function getProxy() { function getProxy() {

6
addon/serverless.js
View File

@@ -69,10 +69,10 @@ router.get('/:configuration/:resource/:type/:id.json', (req, res, next) => {
}); });
}); });
router.get('/realdebrid/:apiKey/:infoHash/:fileIndex?', (req, res) => { router.get('/realdebrid/:apiKey/:infoHash/:cachedFileIds/:fileIndex?', (req, res) => {
const { apiKey, infoHash, fileIndex } = req.params; const { apiKey, infoHash, cachedFileIds, fileIndex } = req.params;
console.time(infoHash); console.time(infoHash);
realDebrid.unrestrict(apiKey, infoHash, isNaN(fileIndex) ? undefined : parseInt(fileIndex)) realDebrid.unrestrict(apiKey, infoHash, cachedFileIds, isNaN(fileIndex) ? undefined : parseInt(fileIndex))
.then(url => { .then(url => {
console.timeEnd(infoHash); console.timeEnd(infoHash);
res.writeHead(301, { Location: url }); res.writeHead(301, { Location: url });