☰

IMDb Tomatoes

Add Rotten Tomatoes ratings to IMDb movie pages

当前为 2021-02-18 提交的版本，查看 最新版本。

// ==UserScript==
// @name          IMDb Tomatoes
// @description   Add Rotten Tomatoes ratings to IMDb movie pages
// @author        chocolateboy
// @copyright     chocolateboy
// @version       3.0.1
// @namespace     https://github.com/chocolateboy/userscripts
// @license       GPL: https://www.gnu.org/copyleft/gpl.html
// @include       http://*.imdb.tld/title/tt*
// @include       http://*.imdb.tld/*/title/tt*
// @include       https://*.imdb.tld/title/tt*
// @include       https://*.imdb.tld/*/title/tt*
// @require       https://code.jquery.com/jquery-3.5.1.min.js
// @require       https://cdn.jsdelivr.net/gh/urin/jquery.balloon.js@8b79aab63b9ae34770bfa81c9bfe30019d9a13b0/jquery.balloon.js
// @require       https://unpkg.com/@chocolateboy/[email protected]/dist/polyfill.iife.min.js
// @require       https://unpkg.com/[email protected]/dist/index.umd.min.js
// @require       https://unpkg.com/[email protected]/dist/index.iife.min.js
// @resource      query https://pastebin.com/raw/EdgTfhij
// @resource      fallback https://cdn.jsdelivr.net/gh/chocolateboy/[email protected]/data/omdb-tomatoes.json
// @grant         GM_addStyle
// @grant         GM_deleteValue
// @grant         GM_getResourceText
// @grant         GM_getValue
// @grant         GM_listValues
// @grant         GM_registerMenuCommand
// @grant         GM_setValue
// @grant         GM_xmlhttpRequest
// @run-at        document-start
// @noframes
// ==/UserScript==

/*
 * OK:
 *
 *   - https://www.imdb.com/title/tt0309698/ - 4 widgets
 *   - https://www.imdb.com/title/tt0086312/ - 3 widgets
 *   - https://www.imdb.com/title/tt0037638/ - 2 widgets
 *
 * Fixed:
 *
 *   layout:
 *
 *     - https://www.imdb.com/title/tt0162346/  - 4 widgets
 *     - https://www.imdb.com/title/tt0159097/  - 4 widgets
 *     - https://www.imdb.com/title/tt0129387/ - 2 .plot_summary_wrapper DIVs
 *
 *   RT/OMDb alias [1]:
 *
 *     - https://www.imdb.com/title/tt0120755/ - Mission: Impossible II
 */

// [1] unaliased and incorrectly aliased titles are common:
// http://web.archive.org/web/20151105080717/http://developer.rottentomatoes.com/forum/read/110751/2

// XXX metadata mismatch: Zéro de conduite [2] is a "Movie" in the old UI, but a
// "Short" in the new one
//
// [2] https://www.imdb.com/title/tt0024803/

'use strict';

const ANALYTICS_KEY   = 'ue_ibe'
const NO_CONSENSUS    = 'No consensus yet.'
const NO_RESULTS      = 'no results found'
const ONE_DAY         = 1000 * 60 * 60 * 24
const ONE_WEEK        = ONE_DAY * 7
const SCRIPT_NAME     = GM_info.script.name
const SCRIPT_VERSION  = GM_info.script.version
const THIS_YEAR       = new Date().getFullYear()

const COLOR = {
    tbd: '#d9d9d9',
    favorable: '#66cc33',
    unfavorable: '#ff0000',
}

const COMPACT_LAYOUT = [
    '.plot_summary_wrapper .minPlotHeightWithPoster', // XXX probably obsolete
    '.plot_summary_wrapper .minPlotHeightWithPosterAndWatchlistButton', // XXX probably obsolete
    '.minPosterWithPlotSummaryHeight .plot_summary_wrapper',
].join(', ')

// the version of each cached record is a combination of the schema version and
// the <major>.<minor> parts of the script's (SemVer) version, e.g. 3 (schema
// version) + 1.7.0 (script version) gives a version of "3/1.7"
//
// this means cached records are invalidated either a) when the schema changes
// or b) when the major or minor version (i.e. not the patch version) of the
// script changes
const SCHEMA_VERSION = 4
const DATA_VERSION = SCHEMA_VERSION + '/' + SCRIPT_VERSION.replace(/\.\d+$/, '') // e.g. 3/1.7

const BALLOON_OPTIONS = {
    classname: 'rt-consensus-balloon',
    css: {
        maxWidth: '31rem',
        fontFamily: 'Roboto, Helvetica, Arial, sans-serif',
        fontSize: '0.9rem',
        padding: '0.75rem',
    },
    html: true,
    position: 'bottom',
}

// log a message to the console
const log = console.log

// run a function when the +pageshow+ event fires, or immediately if it has fired
// already
const onPageShow = when('pageshow')

// a custom version of get-wild's `get` function which uses a simpler/faster
// path parser since we don't use the extended syntax
const pluck = exports.getter({ split: '.' })

// register a jQuery plugin which extracts and returns JSON-LD data for the
// current page.
//
// used to extract metadata on legacy IMDb and Rotten Tomatoes
//
// @ts-ignore
$.fn.jsonLd = function jsonLd (id) {
    const $script = this.find('script[type="application/ld+json"]')

    let data

    if ($script.length) {
        try {
            data = JSON.parse($script.first().text().trim())
        } catch (e) {
            throw new Error(`Can't parse JSON-LD data for ${id}: ${e}`)
        }
    } else {
        throw new Error(`Can't find JSON-LD data for ${id}`)
    }

    return data
}

// take a record (object) from the OMDb fallback data (object) and convert it
// into the parsed format we expect to get back from the API, e.g.:
//
// before:
//
//     {
//         Title: "Example",
//         Ratings: [
//             {
//                 Source: "Rotten Tomatoes",
//                 Value: "42%"
//             }
//         ],
//         tomatoURL: "https://www.rottentomatoes.com/m/example"
//     }
//
// after:
//
//     {
//         CriticRating: 42,
//         RTConsensus: undefined,
//         RTUrl: "https://www.rottentomatoes.com/m/example",
//     }

function adaptOmdbData (data) {
    const ratings = data.Ratings || []
    const rating = ratings.find(it => it.Source === 'Rotten Tomatoes') || {}
    const score = rating.Value && parseInt(rating.Value)

    return {
        CriticRating: (Number.isInteger(score) ? score : null),
        RTConsensus: rating.tomatoConsensus,
        RTUrl: data.tomatoURL,
    }
}

// add a Rotten Tomatoes widget to the review bar in the legacy UI
function addLegacyWidget ($target, { balloonOptions, rating, style, url }) {
    // reduce the amount of space taken up by the Metacritic widget
    // and make it consistent with our style (i.e. site name rather
    // than domain name)
    $target.find('a[href="http://www.metacritic.com"]').text('Metacritic')

    // 4 review widgets is too many for the "compact" layout (i.e.
    // a poster but no trailer). it's designed for a maximum of 3.
    // to work around this, we hoist the review bar out of the
    // movie-info block (.plot_summary_wrapper) and float it left
    // beneath the poster, e.g.:
    //
    // before:
    //
    // [  [        ] [                    ] ]
    // [  [        ] [                    ] ]
    // [  [ Poster ] [        Info        ] ]
    // [  [        ] [                    ] ]
    // [  [        ] [ [MC] [IMDb] [etc.] ] ]
    //
    // after:
    //
    // [  [        ] [                    ] ]
    // [  [        ] [                    ] ]
    // [  [ Poster ] [        Info        ] ]
    // [  [        ] [                    ] ]
    // [  [        ] [                    ] ]
    // [                                    ]
    // [  [RT] [MC] [IMDb] [etc.]           ]

    if ($(COMPACT_LAYOUT).length && $target.find('.titleReviewBarItem').length > 2) {
        const $clear = $('<div class="clear">&nbsp;</div>')

        // sometimes there are two Info (.plot_summary_wrapper) DIVs (e.g.
        // [1]). the first is (currently) empty and the second contains the
        // actual markup. this may be a transient error in the markup, or
        // may be used somehow (e.g. for mobile). if targeted, the first one
        // is displayed above the visible Plot/Info row, whereas the second
        // one is to the right of the poster, as expected, so we target that
        //
        // [1] https://www.imdb.com/title/tt0129387/
        $('.plot_summary_wrapper').last().after($target.remove())

        $target.before($clear).after($clear).css({
            'float':          'left',
            'padding-top':    '11px',
            'padding-bottom': '0px',
        })
    }

    const score = rating === -1 ? 'N/A' : rating

    const html = `
        <div id="rt-rating" class="titleReviewBarItem">
            <a href="${url}"><div
                class="rt-consensus metacriticScore score_${style} titleReviewBarSubItem"><span>${score}</span></div></a>
           <div class="titleReviewBarSubItem">
               <div>
                   <a href="${url}">Tomatometer</a>
               </div>
               <div>
                   <span class="subText">
                       From <a href="https://www.rottentomatoes.com" target="_blank">Rotten Tomatoes</a>
                   </span>
               </div>
            </div>
        </div>
        <div class="divider"></div>
    `
    $target.prepend(html)
    $target.find('.rt-consensus').balloon(balloonOptions)
    log('added widget')
}

// add a Rotten Tomatoes widget to the review bar in the new UI
function addReactWidget ($target, { balloonOptions, rating, style, url }) {
    log('adding widget')

    // clone the IMDb rating widget: https://git.io/JtXpQ
    const $imdbRating = $target.children().first()
    const $rtRating = $imdbRating.clone().attr('id', 'rt-rating')

    // 1) set the star (SVG) to the right color
    GM_addStyle(`#rt-rating svg { color: ${COLOR[style]} }`)

    // 2) remove the review count and its preceding spacer element
    const $reviewCount = $rtRating
        .find('[class^="AggregateRatingButton__TotalRatingAmount-"]')

    $reviewCount.add($reviewCount.prev()).remove()

    // 3) replace "IMDb Rating" with "RT Rating"
    $rtRating.find('[class^="TitleBlockButtonBase__Header-"]')
        .text('RT RATING')

    // 4) remove the "/ 10" suffix
    const $score = $rtRating
        .find('[data-testid="hero-title-block__aggregate-rating__score"]')
        .children()

    $score.last().remove()

    // 5) replace the IMDb rating with the RT score
    const score = rating === -1 ? 'N/A' : `${rating}%`

    $score.first().text(score)

    // 6) add the tooltip class to the link and update its label and URL
    $rtRating.find('a[role="button"]')
        .addClass('rt-consensus')
        .balloon(balloonOptions)
        .attr('aria-label', 'View RT Rating')
        .attr('href', url)

    // 7) prepend the element to the review bar

    // defer to work around React reconciliation
    onPageShow(() => {
        $target.prepend($rtRating)
        log('added widget')
    })
}

// add a Rotten Tomatoes widget to the review bar
function addWidget ($target, data) {
    const { consensus, rating } = data
    const balloonOptions = Object.assign({}, BALLOON_OPTIONS, { contents: consensus })

    let style

    if (rating === -1) {
        style = 'tbd'
    } else if (rating < 60) {
        style = 'unfavorable'
    } else {
        style = 'favorable'
    }

    const options = { ...data, balloonOptions, style }

    if ($target.hasClass('titleReviewBar')) { // legacy UI
        addLegacyWidget($target, options)
    } else { // new UI
        addReactWidget($target, options)
    }
}

// disable the debug/analytics code on the new site
function disableAnalytics () {
    GMCompat.unsafeWindow[ANALYTICS_KEY] = 1
}

// URL-encode the supplied query parameter and replace encoded spaces ("%20")
// with plus signs ("+")
function encodeParam (param) {
    return encodeURIComponent(param).replace(/%20/g, '+')
}

// encode a dictionary of params as a query parameter string. this is similar to
// jQuery.params, but we additionally replace spaces ("%20") with plus signs
// ("+")
function encodeParams (params) {
    const pairs = []

    for (const [key, value] of Object.entries(params)) {
        pairs.push(`${encodeParam(key)}=${encodeParam(value)}`)
    }

    return pairs.join('&')
}

// parse the API's response and extract the RT rating and consensus.
//
// if there's no consensus, default to "No consensus yet."
// if there's no rating, default to -1
async function getRTData ({ response, imdbId, title, fallback }) {
    function fail (message) {
        throw new Error(message)
    }

    let results

    try {
        results = JSON.parse(JSON.parse(response)) // ಠ_ಠ
    } catch (e) {
        fail(`can't parse response: ${e}`)
    }

    if (!results) {
        fail('no response')
    }

    if (!Array.isArray(results)) {
        const type = {}.toString.call(results)
        fail(`invalid response: ${type}`)
    }

    let movie = results.find(it => it.imdbID === imdbId)

    if (!movie) {
        if (fallback) {
            log(`no results for ${imdbId} - using fallback data`)
            movie = adaptOmdbData(fallback)
        } else {
            fail(NO_RESULTS)
        }
    }

    let { RTConsensus: consensus, CriticRating: rating, RTUrl: url } = movie
    let updated = false

    if (url) {
        // the new way: the RT URL is provided: scrape the consensus from
        // that page

        log(`loading RT URL: ${url}`)
        const res = await httpGet(url)
        log(`response: ${res.status} ${res.statusText}`)

        const parser = new DOMParser()
        const dom = parser.parseFromString(res.responseText, 'text/html')
        const $rt = $(dom)
        const $consensus = $rt.find('.what-to-know__section-body > span')

        if ($consensus.length) {
            consensus = $consensus.html().trim()
        }

        // update the rating
        // @ts-ignore
        const meta = $rt.jsonLd(url)
        const newRating = meta.aggregateRating.ratingValue

        if (newRating !== rating) {
            log(`updating rating: ${rating} -> ${newRating}`)
            rating = newRating
            updated = true
        }
    } else {
        // the old way: a rating but no RT URL (or consensus).
        // may still be used for some old and new releases
        log(`no Rotten Tomatoes URL for ${imdbId}`)
        url = `https://www.rottentomatoes.com/search/?search=${encodeURIComponent(title)}`
    }

    if (rating == null) {
        rating = -1
    }

    consensus = consensus ? consensus.replace(/--/g, '&#8212;') : NO_CONSENSUS

    return { data: { consensus, rating, url }, updated }
}

// extract metadata from the GraphQL data embedded in the page
function graphQlMetadata (imdbId) {
    const meta = JSON.parse($('#__NEXT_DATA__').text())
    const pageType = pluck(meta, 'props.requestContext.pageType')

    // there are multiple matching subtrees (with different but partially
    // overlapping keys). select the first one with the required properties
    let title, type

    const found = pluck(meta, 'props.urqlState.*.data', [])
        .find(it => {
            return pluck(it, 'title.id') === imdbId
                && (title = pluck(it, 'title.titleText.text'))
                && (type = pluck(it, 'title.titleType.text'))
        })

    if (!found) {
        if (!title) {
            throw new TypeError("Can't find title in metadata")
        }

        if (!type) {
            throw new TypeError("Can't find type in metadata")
        }
    }

    return { pageType, title, type }
}

// promisified cross-origin HTTP requests
function httpGet (url, options = {}) {
    if (options.params) {
        url = url + '?' + encodeParams(options.params)
    }

    const request = Object.assign({ method: 'GET', url }, options.request || {})

    return new Promise((resolve, reject) => {
        request.onload = resolve

        // XXX the +onerror+ response object doesn't contain any useful info
        request.onerror = _res => {
            reject(new Error(`error fetching ${options.title || url}`))
        }

        GM_xmlhttpRequest(request)
    })
}

// extract metadata from the JSON+LD data embedded in the page and from metadata
// elements
function jsonLdMetadata (imdbId) {
    // @ts-ignore
    const meta = $(document).jsonLd(imdbId)

    // override the original title (e.g. "Le fabuleux destin d'Amélie Poulain")
    // with the English language (US) title (e.g. "Amélie") if available
    // (the API only supports English-language titles)
    const title = $('#star-rating-widget').data('title') || meta.name

    return {
        pageType: prop('pageType'),
        title,
        type: meta['@type']
    }
}

// extract a property from a META element, or return null if the property is
// not defined
function prop (name) {
    const $meta = $(`meta[property="${name}"]`)
    return $meta.length ? $meta.attr('content') : null
}

// purge expired entries
function purgeCached (date) {
    for (const key of GM_listValues()) {
        const json = GM_getValue(key)
        const value = JSON.parse(json)

        if (value.expires === -1) { // persistent storage (currently unused)
            if (value.version !== SCHEMA_VERSION) {
                log(`purging invalid value (obsolete schema version): ${key}`)
                GM_deleteValue(key)
            }
        } else if (value.version !== DATA_VERSION) {
            log(`purging invalid value (obsolete data version): ${key}`)
            GM_deleteValue(key)
        } else if (date === -1 || (typeof value.expires !== 'number') || (date > value.expires)) {
            log(`purging expired value: ${key}`)
            GM_deleteValue(key)
        }
    }
}

async function run () {
    // @ts-ignore
    const imdbId = location.pathname.match(/\/title\/(tt\d+)/)[1]

    log('id:', imdbId)

    const $classicReviewBar = $('.titleReviewBar')
    const $reactReviewBar = $('[class^="TitleBlock__ButtonContainer-"]')
    const $target = [$classicReviewBar, $reactReviewBar].find(it => it.length)

    if (!$target) {
        console.info(`Can't find target for ${imdbId}`)
        return
    }

    let meta

    if ($target === $classicReviewBar) {
        meta = jsonLdMetadata(imdbId)
        /* make the background color more legible (darker) if the rating is N/A */
        GM_addStyle(`.score_tbd { background-color: ${COLOR.tbd} }`)
    } else if ($target === $reactReviewBar) {
        meta = graphQlMetadata(imdbId)
    } else {
        console.warn(`can't find metadata for ${imdbId}`)
        return
    }

    log('metadata:', meta)

    const { pageType, title, type } = meta

    if (type !== 'Movie') {
        console.info(`invalid type for ${imdbId}: ${type}`)
        return
    }

    if (pageType !== 'title') {
        console.info(`invalid page type for ${imdbId}: ${pageType}`)
        return
    }

    const now = Date.now()

    purgeCached(now)

    // get the cached result for this page
    const cached = JSON.parse(GM_getValue(imdbId, 'null'))

    if (cached) {
        const expires = new Date(cached.expires).toLocaleString()

        if (cached.error) {
            log(`cached error (expires: ${expires}):`, cached.error)
        } else {
            log(`cached result (expires: ${expires}):`, cached.data)
            addWidget($target, cached.data)
        }

        return
    } else {
        log('not cached')
    }

    // add a { version, expires, data|error } entry to the cache
    const store = (dataOrError, ttl) => {
        const expires = now + ttl
        const cached = { version: DATA_VERSION, expires, ...dataOrError }
        const json = JSON.stringify(cached)

        GM_setValue(imdbId, json)
    }

    const query = JSON.parse(GM_getResourceText('query'))

    Object.assign(query.params, { searchTerm: title, yearMax: THIS_YEAR })

    try {
        log(`querying API for ${JSON.stringify(title)}`)

        const requestOptions = Object.assign({}, query, { title: `data for ${imdbId}` })
        const response = await httpGet(query.api, requestOptions)
        const fallback = JSON.parse(GM_getResourceText('fallback'))

        log(`response: ${response.status} ${response.statusText}`)

        const { data, updated } = await getRTData({
            response: response.responseText,
            imdbId,
            title,
            fallback: fallback[imdbId],
        })

        if (updated) {
            log(`caching result for: one day`)
            store({ data }, ONE_DAY)
        } else {
            log(`caching result for: one week`)
            store({ data }, ONE_WEEK)
        }

        addWidget($target, data)
    } catch (error) {
        const message = error.message || String(error) // stringify

        log(`caching error for one day: ${message}`)
        store({ error: message }, ONE_DAY)

        if (message !== NO_RESULTS) {
            console.error(error)
        }
    }
}

// returns a function which runs a function when the specified event has fired,
// or immediately if it has already fired
function when (event, { target = window } = {}) {
    let ready

    const callbacks = new Set()

    const listener = /** @this {any} */ function (...args) {
        ready = { this: this, args }

        if (callbacks.size) {
            for (const callback of callbacks.values()) {
                callback.apply(ready.this, ready.args)
            }

            callbacks.clear()
        }
    }

    const onEvent = callback => {
        if (ready) {
            queueMicrotask(() => callback.apply(ready.this, ready.args))
        } else {
            callbacks.add(callback)
        }

        return !!ready
    }

    target.addEventListener(event, listener, { once: true })

    return onEvent
}

// register this first so data can be cleared even if there's an error
GM_registerMenuCommand(SCRIPT_NAME + ': clear cache', () => { purgeCached(-1) })

// disable the debugging/analytics code on the new site
disableAnalytics()

$(window).on('DOMContentLoaded', run)

QingJ © 2025

镜像随时可能失效，请加Q群300939539或关注我们的公众号极客氢云获取最新地址