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

adds manual reapply step to decompose episodes again

Browse Source
This commit is contained in:
TheBeastLT
2020-02-25 08:13:08 +01:00
parent 9aad6a5148
commit 75053fccd0
3 changed files with 39 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/torrent.js
View File

@@ -35,7 +35,8 @@ module.exports.updateCurrentSeeders = function (torrent) {
};
module.exports.torrentFiles = function (torrent) {
return filesFromTorrentFile(torrent)
return getFilesFromObject(torrent)
.catch(() => filesFromTorrentFile(torrent))
.catch(() => filesFromTorrentStream(torrent))
.catch(() => filesFromCache(torrent.infoHash))
.then((files) => filterVideos(files))
@@ -43,6 +44,13 @@ module.exports.torrentFiles = function (torrent) {
.then((files) => filterExtras(files));
};
function getFilesFromObject(torrent) {
if (torrent.files && torrent.files.length) {
return Promise.resolve(torrent.files);
}
return Promise.reject("No files in the object");
}
function filesFromCache(infoHash) {
return retrieveTorrentFiles(infoHash)
.then((files) => files.map((file) => ({

5
lib/torrentFiles.js
View File

@@ -138,8 +138,9 @@ async function decomposeEpisodes(torrent, files, metadata = { episodeCount: [] }
&& sortedEpisodes.every(ep => metadata.episodeCount[div100(ep) - 1] >= mod100(ep))
&& files.every(file => !file.season || file.episodes.every(ep => div100(ep) === file.season))) {
decomposeConcatSeasonAndEpisodeFiles(torrent, files, metadata);
} else if ((files.every(file => !file.season && file.episodes) || files.some(file => file.season && file.episodes
&& file.episodes.every(ep => metadata.episodeCount[file.season - 1] < ep)))
} else if ((files.filter(file => !file.isMovie && file.episodes).every(file => !file.season && file.episodes) ||
files.some(file => file.season && file.episodes && file.episodes
.every(ep => metadata.episodeCount[file.season - 1] < ep)))
&& (sortedEpisodes.length <= 1 || sortedEpisodes.slice(1).every((ep, i) => ep - sortedEpisodes[i] <= 2))) {
decomposeAbsoluteEpisodeFiles(torrent, files, metadata);
} else if (files.every(file => (!file.season || !metadata.episodeCount[file.season - 1]) && file.date)) {

29
manual/manual.js
View File

@@ -48,6 +48,30 @@ async function updateMovieCollections() {
}));
}
async function reapplyEpisodeDecomposing() {
const infoHash = '84fadd061f0d0bc356235b7fa6495a0f51fff311';
const imdbId = 'tt0988824';
const torrent = await repository.getTorrent({ infoHash });
const storedFiles = await repository.getFiles({ infoHash });
const fileIndexMap = storedFiles
.reduce((map, next) => (map[next.fileIndex] = (map[next.fileIndex] || []).concat(next), map), {});
const files = Object.values(fileIndexMap)
.map(sameIndexFiles => sameIndexFiles[0])
.map(file => ({ fileIndex: file.fileIndex, name: file.title, path: file.title, size: file.size }));
return parseTorrentFiles({ ...torrent, imdbId, files })
.then(newFiles => newFiles.map(file => {
const originalFile = fileIndexMap[file.fileIndex].shift();
originalFile.imdbSeason = file.imdbSeason;
originalFile.imdbEpisode = file.imdbEpisode;
originalFile.kitsuId = file.kitsuId;
originalFile.kitsuEpisode = file.kitsuEpisode;
return originalFile;
}))
.then(updatedFiles => Promise.all(updatedFiles.map(file => file.save())))
.then(() => console.log(`Updated files for ${torrent.title}`));
}
async function findAllFiles() {
/* Test cases */
/* Anime Season and absolute episodes */
@@ -122,5 +146,6 @@ async function findAllFiles() {
}
//addMissingEpisodes().then(() => console.log('Finished'));
findAllFiles().then(() => console.log('Finished'));
//updateMovieCollections().then(() => console.log('Finished'));
//findAllFiles().then(() => console.log('Finished'));
//updateMovieCollections().then(() => console.log('Finished'));
reapplyEpisodeDecomposing().then(() => console.log('Finished'));