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

[scraper] match subtitles by equals titles instead of includes

Browse Source
This commit is contained in:
TheBeastLT
2020-05-02 20:21:42 +02:00
parent 4507b1bd1d
commit 452f29896c
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
scraper/lib/torrentSubtitles.js
View File

@@ -23,22 +23,24 @@ function assignSubtitles({ contents, videos, subtitles }) {
} }
function _parseVideo(video) { function _parseVideo(video) {
const fileName = video.title.replace(/\.(\w{2,4})$/, '');
const folderName = video.title.replace(/\/?[^/]+$/, '');
return { return {
videoFile: video, videoFile: video,
fileName: video.title.replace(/\.(\w{2,4})$/, ''), fileName: fileName,
folderName: video.title.replace(/\/?[^/]+$/, ''), folderName: folderName,
...parse(video.title) ...parse(fileName)
}; };
} }
function _mostProbableSubtitleVideo(subtitle, parsedVideos) { function _mostProbableSubtitleVideo(subtitle, parsedVideos) {
const subTitle = subtitle.title || subtitle.path; const subTitle = subtitle.title || subtitle.path;
const parsedSub = parse(subTitle); const parsedSub = parse(subTitle.replace(/\.(\w{2,4})$/, ''));
const byFileName = parsedVideos.filter(video => subTitle.includes(video.fileName)); const byFileName = parsedVideos.filter(video => subTitle.includes(video.fileName));
if (byFileName.length === 1) { if (byFileName.length === 1) {
return byFileName[0].videoFile; return byFileName[0].videoFile;
} }
const byTitleSeasonEpisode = parsedVideos.filter(video => parsedSub.title.includes(video.title) const byTitleSeasonEpisode = parsedVideos.filter(video => video.title === parsedSub.title
&& video.seasons === parsedSub.seasons && video.seasons === parsedSub.seasons
&& JSON.stringify(video.episodes) === JSON.stringify(parsedSub.episodes)); && JSON.stringify(video.episodes) === JSON.stringify(parsedSub.episodes));
if (byTitleSeasonEpisode.length === 1) { if (byTitleSeasonEpisode.length === 1) {
@@ -49,7 +51,7 @@ function _mostProbableSubtitleVideo(subtitle, parsedVideos) {
if (bySeasonEpisode.length === 1) { if (bySeasonEpisode.length === 1) {
return bySeasonEpisode[0].videoFile; return bySeasonEpisode[0].videoFile;
} }
const byTitle = parsedVideos.filter(video => parsedSub.title.includes(video.title)); const byTitle = parsedVideos.filter(video => video.title && video.title === parsedSub.title);
if (byTitle.length === 1) { if (byTitle.length === 1) {
return byTitle[0].videoFile; return byTitle[0].videoFile;
} }