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);
if (available) {
Object.entries(available)
.filter(([key, value]) => getCachedFileIds(streamMapping[key.toLowerCase()].fileIdx, value).length)
.map(([key]) => key.toLowerCase())
.map(cachedInfoHash => streamMapping[cachedInfoHash])
.forEach(stream => {
.map(([key, value]) => ({
cachedInfoHash: key.toLowerCase(),
cachedFileIds: getCachedFileIds(streamMapping[key.toLowerCase()].fileIdx, value)
}))
.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.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.fileIndex;
})
@@ -29,18 +33,18 @@ async function applyMoch(streams, apiKey) {
return streams;
}
async function unrestrict(apiKey, infoHash, fileIndex) {
if (!apiKey || !infoHash) {
async function unrestrict(apiKey, infoHash, cachedFileIds, fileIndex) {
if (!apiKey || !infoHash || !cachedFileIds || !cachedFileIds.length) {
return Promise.reject("No valid parameters passed");
}
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}]`);
const RD = new RealDebridClient(apiKey);
const torrentId = await _createOrFindTorrentId(RD, infoHash);
const torrentId = await _createOrFindTorrentId(RD, infoHash, cachedFileIds);
console.log(`Retrieved torrentId: ${torrentId}`);
if (torrentId) {
const info = await RD.torrents.info(torrentId);
@@ -55,12 +59,12 @@ async function _unrestrict(apiKey, infoHash, fileIndex) {
return Promise.reject("Failed adding torrent");
}
async function _createOrFindTorrentId(RD, infoHash) {
async function _createOrFindTorrentId(RD, infoHash, cachedFileIds) {
return RD.torrents.get(0, 1)
.then(torrents => torrents.find(torrent => torrent.hash.toLowerCase() === infoHash))
.then(torrent => torrent && torrent.id || Promise.reject('No recent torrent found'))
.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))))
.catch(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 RD._post('unrestrict/link', { form: { link }, proxy: getProxy() })
.then(unrestrictedLink => {
console.log(`Unrestricted ${link} to ${unrestrictedLink.download}`);
return Promise.resolve(unrestrictedLink.download);
.then(unrestrictedLink => RD.streaming.transcode(unrestrictedLink.id))
.then(transcodedLink => {
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 [];
}
// 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), [])
.filter(cached => isNaN(fileIndex) && Object.keys(cached).length || cached[fileIndex + 1])
.find(cached => Object.values(cached).every(file => isVideo(file.filename)));
return cachedTorrent && Object.keys(cachedTorrent) || [];
.filter(cached => !Number.isInteger(fileIndex) && Object.keys(cached).length || cached[fileIndex + 1])
.filter(cached => Object.values(cached).every(file => isVideo(file.filename)))
.map(cached => Object.keys(cached))
.sort((a, b) => b.length - a.length);
return cachedTorrents.length && cachedTorrents[0] || [];
}
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) => {
const { apiKey, infoHash, fileIndex } = req.params;
router.get('/realdebrid/:apiKey/:infoHash/:cachedFileIds/:fileIndex?', (req, res) => {
const { apiKey, infoHash, cachedFileIds, fileIndex } = req.params;
console.time(infoHash);
realDebrid.unrestrict(apiKey, infoHash, isNaN(fileIndex) ? undefined : parseInt(fileIndex))
realDebrid.unrestrict(apiKey, infoHash, cachedFileIds, isNaN(fileIndex) ? undefined : parseInt(fileIndex))
.then(url => {
console.timeEnd(infoHash);
res.writeHead(301, { Location: url });