Utility library for Rotten Tomatoes. Provides an API for grabbing info from rottentomatoes.com
目前为
此脚本不应直接安装，它是一个供其他脚本使用的外部库。如果您需要使用该库，请在脚本元属性加入：// @require https://update.greasyfork.icu/scripts/389810/854004/RottenTomatoes%20Utility%20Library%20%28custom%20API%29.js
// ==UserScript==
// @name RottenTomatoes Utility Library (custom API)
// @namespace driver8.net
// @version 0.1.5
// @description Utility library for Rotten Tomatoes. Provides an API for grabbing info from rottentomatoes.com
// @author driver8
// @grant GM_xmlhttpRequest
// @connect rottentomatoes.com
// ==/UserScript==
console.log('hi rt api lib');
const MAX_YEAR_DIFF = 2;
const MAX_RESULTS = 50;
function _parse(query, regex, doc) {
doc = doc || document;
try {
let text = doc.querySelector(query).textContent.trim();
if (regex) {
text = text.match(regex)[1];
}
return text.trim();
} catch (e) {
console.log('error', e);
return '';
}
};
function jsonParse(j) {
try {
result = JSON.parse(j);
return result;
} catch(e) {
console.log('RT error', e);
return null;
}
}
function getRtIdFromTitle(title, tv, year) {
tv = tv || false;
year = parseInt(year) || 1800;
return new Promise(function(resolve, reject) {
GM_xmlhttpRequest({
method: 'GET',
responseType: 'json',
url: `https://www.rottentomatoes.com/api/private/v2.0/search/?limit=${MAX_RESULTS}&q=${title}`,
onload: (resp) => {
let movies = tv ? resp.response.tvSeries : resp.response.movies;
if (!Array.isArray(movies) || movies.length < 1) {
console.log('no search results');
reject('no results');
return;
}
let sorted = movies.concat();
if (year && sorted) {
sorted.sort((a, b) => {
if (Math.abs(a.year - year) !== Math.abs(b.year - year)) {
// Prefer closest year to the given one
return Math.abs(a.year - year) - Math.abs(b.year - year);
} else {
return b.year - a.year; // In a tie, later year should come first
}
});
}
// Search for matches with exact title in order of proximity by year
let bestMatch, closeMatch;
for (let m of sorted) {
m.title = m.title || m.name;
if (m.title.toLowerCase() === title.toLowerCase()) {
bestMatch = bestMatch || m;
// RT often includes original titles in parentheses for foreign films, so only check if they start the same
} else if (m.title.toLowerCase().startsWith(title.toLowerCase())) {
closeMatch = closeMatch || m;
}
if (bestMatch && closeMatch) {
break;
}
}
//console.log('sorted', sorted, 'bestMatch', bestMatch, 'closeMatch', closeMatch, movies);
// Fall back on closest year match if within 2 years, or whatever the first result was.
// RT years are often one year later than imdb, or even two
function yearComp(imdb, rt) {
return rt - imdb <= MAX_YEAR_DIFF && imdb - rt < MAX_YEAR_DIFF;
}
if (year && (!bestMatch || !yearComp(year, bestMatch.year))) {
if (closeMatch && yearComp(year, closeMatch.year)) {
bestMatch = closeMatch;
} else if (yearComp(year, sorted[0].year)) {
bestMatch = sorted[0];
}
}
bestMatch = bestMatch || closeMatch || movies[0];
if (bestMatch) {
let id = bestMatch && bestMatch.url.replace(/\/s\d{2}\/?$/, ''); // remove season suffix from tv matches
console.log('found id', id);
resolve(id);
} else {
console.log('no match found on rt');
reject('no suitable match');
}
}
});
});
}
function getRtInfoFromId(id) {
return new Promise(function(resolve, reject) {
if (!id || typeof id !== 'string' || id.length < 3) {
console.log('invalid id');
reject('invalid id');
}
let url = 'https://www.rottentomatoes.com' + id + (id.startsWith('/tv/') ? '/s01' : ''); // Look up season 1 for TV shows
GM_xmlhttpRequest({
method: 'GET',
responseType: 'document',
url: url,
onload: (resp) => {
let text = resp.responseText;
//console.log('text', text);
// Create DOM from responseText
const doc = document.implementation.createHTMLDocument().documentElement;
doc.innerHTML = text;
let year = parseInt(_parse('.h3.year, .movie_title .h3.subtle, .meta-row .meta-value time', /(\d{4})/, doc));
// Find the javascript snippet storing the tomatometer/score info.
// Everything is named different for TV shows for some stupid reason.
let m = text.match(/root\.RottenTomatoes\.context\.scoreInfo = ({.+});/);
m = m || text.match(/root\.RottenTomatoes\.context\.scoreboardCriticInfo = ({.+});/);
let dataString = m[1];
let scoreInfo = jsonParse(dataString);
//console.log('scoreInfo', scoreInfo);
let all = scoreInfo.tomatometerAllCritics ?? scoreInfo.all ?? false,
top = scoreInfo.tomatometerTopCritics ?? scoreInfo.top ?? {};
//console.log('scoreInfo', scoreInfo);
// TV consensus is stored in a totally different object :/
m = text.match(/root\.RottenTomatoes\.context\.result =\s+({.+});\n/);
//console.log('m[1]', m?.[1]);
let fixedJson = m?.[1].replace(/:undefined/g, ':null');
//console.log('fixedJson', fixedJson);
let contextResult = m && jsonParse(fixedJson);
if (all) {
// Try field names used for movie data, then TV show data.
const data = {
id: id,
score: all.score ?? all.tomatometer ?? -1,
rating: all.avgScore ?? all.averageRating ?? -1,
votes: all.numberOfReviews ?? all.totalCount ?? all.ratingCount ?? all.reviewCount ?? 0,
consensus: all.consensus ?? (contextResult?.seasonData?.tomatometer?.consensus) ?? doc.querySelector('.mop-ratings-wrap__text--concensus')?.innerHTML, // TV consensus is stored in a totally different object :/
state: all.tomatometerState ?? all.state,
topScore: parseInt(top.score ?? top.tomatometer ?? -1),
topRating: parseFloat(top.avgScore ?? top.averageRating ?? -1),
topVotes: top.numberOfReviews ?? top.totalCount ?? top.ratingCount ?? top.reviewCount ?? 0,
year: year,
fetched: new Date()
};
console.log('found data', data);
resolve(data);
} else {
reject('error getting rt info for id ' + id);
}
}
});
});
}
function getRtInfoFromTitle(title, tv, year) {
return getRtIdFromTitle(title, tv, year).then((id) => {
return getRtInfoFromId(id);
})
}