katoss/utils/searchSickBeard.js at master · same31/katoss · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
process.env['NODE_TLS_REJECT_UNAUTHORIZED'] = '0';

var hasToReplaceLowQuality = ~process.argv.indexOf('--replace-low-quality'),
    config                 = require('../katoss/config.json'),
    search                 = require('../katoss/search'),
    formatShowNumber       = require('../katoss/src/utils').formatShowNumber,
    allSettled             = require('../katoss/src/utils').allSettled,
    sendSickBeardAPICmd    = require('./include/sendSickBeardAPICmd'),
	showAlternativeTitles  = require('../katoss/showAlternativeTitles.json'),
    hasToSearchEpisode,
    addEpisodeToSearch,
    maxQuality,
    minDate;

if (hasToReplaceLowQuality) {
    maxQuality = config.qualityOrder[0].toLowerCase();
    // Do not search every episode in 2160p or it would search for practically the entire base
    // ---------------------------------------------------------------------------------------
    maxQuality === '2160p' && (maxQuality = config.qualityOrder[1].toLowerCase());
    minDate = new Date();
    minDate.setMonth(minDate.getMonth() - 6);
}

hasToSearchEpisode = hasToReplaceLowQuality
    ? episodeInfo => episodeInfo.status === 'Downloaded' && episodeInfo.quality.indexOf(maxQuality) === -1 && (new Date(episodeInfo.airdate)) > minDate
    : episodeInfo => episodeInfo.status === 'Wanted';

addEpisodeToSearch = (searchJSONShow, seasonNumber, episodeNumber) => {
    searchJSONShow.seasons[seasonNumber] || (searchJSONShow.seasons[seasonNumber] = []);
    searchJSONShow.seasons[seasonNumber].push(episodeNumber);
};

if (hasToReplaceLowQuality) {
    var originalAddEpisodeToSearch = addEpisodeToSearch;
    addEpisodeToSearch             = (searchJSONShow, seasonNumber, episodeNumber, currentQuality) => {
        originalAddEpisodeToSearch(searchJSONShow, seasonNumber, episodeNumber);

        searchJSONShow.currentQualities || (searchJSONShow.currentQualities = {});
        searchJSONShow.currentQualities[seasonNumber] || (searchJSONShow.currentQualities[seasonNumber] = []);
        searchJSONShow.currentQualities[seasonNumber].push(currentQuality);
    };
}

function notifySickBeard (tvdbid, season, episode) {
    return sendSickBeardAPICmd(
        'episode.setstatus',
        {
            'tvdbid':  tvdbid,
            'season':  parseInt(season),
            'episode': parseInt(episode),
            'status':  'skipped'
        }
    );
}

// Get show id list
// ----------------
sendSickBeardAPICmd('shows', { 'sort': 'name', 'paused': 0 }).then(showList => {
    var searchJSON  = {},
        promiseList = [];
    for (var showName in showList) {
        if (!showList.hasOwnProperty(showName)) {
            continue;
        }
        var show = showList[showName];
        ((tvdbid, showName) => {
            promiseList.push(sendSickBeardAPICmd('show.seasons', { tvdbid: tvdbid }).then(seasonList => {
                for (var seasonNumber in seasonList) {
                    if (!seasonList.hasOwnProperty(seasonNumber)) {
                        continue;
                    }
                    var season      = formatShowNumber(seasonNumber),
                        episodeList = seasonList[seasonNumber];
                    for (var episodeNumber in episodeList) {
                        if (!episodeList.hasOwnProperty(episodeNumber)) {
                            continue;
                        }
                        var episodeInfo     = episodeList[episodeNumber];
                        episodeInfo.quality = episodeInfo.quality.replace('4K UHD', maxQuality); // Pretend it is the desired max quality
                        if (hasToSearchEpisode(episodeInfo)) {
                            searchJSON[showName] || (searchJSON[showName] = { seasons: {}, tvdbid: tvdbid });
                            addEpisodeToSearch(searchJSON[showName], season, formatShowNumber(episodeNumber), episodeInfo.quality);
                        }
                    }
                }
            }));
        })(show.tvdbid, showAlternativeTitles[showName] || showName);
    }

    allSettled(promiseList).then(() => {
        console.log(searchJSON);
        console.log('\n');
        search(searchJSON, notifySickBeard);
    });
});