opensubtitles-api/lib/identify.js

opensubtitles.org api wrapper

const libhash = require('./hash.js')

const path = require('path')
const fs = require('fs')

module.exports = new class LibID {
    readNFO(file) {
        let p = path.parse(file)
        let nfo = path.join(p.dir, p.name + '.nfo')
        let detectedId

        if (!fs.existsSync(nfo)) {
            for (let f of fs.readdirSync(p.dir)) {
                if (f.match(/\.nfo/i) !== null) {
                    nfo = path.join(p.dir, f)
                    break
                }
            }
        }

        if (fs.existsSync(nfo)) detectedId = fs.readFileSync(nfo).toString().match(/tt\d+/i)
        return detectedId ? detectedId[0] : undefined
    }


    parseResponse(response, info) {
        if (response.data === String() || !response.status.match(/200/)) throw new Error(response.status || 'OpenSubtitles unknown error')

        let data = Object()

        if (response.data[info.moviehash] && Object.keys(response.data[info.moviehash]).length) { // we got valid info
            data = {
                subcount: response.data[info.moviehash].SubCount,
                added: response.data.accepted_moviehashes ? true : false,
                metadata: {
                    imdbid: response.data[info.moviehash].MovieImdbID !== '0' ? 'tt' + response.data[info.moviehash].MovieImdbID : undefined,
                    title: response.data[info.moviehash].MovieName,
                    year: response.data[info.moviehash].MovieYear,
                }
            }
            if (response.data[info.moviehash].SeriesEpisode + response.data[info.moviehash].SeriesSeason !== '00') {
                data.metadata.episode = response.data[info.moviehash].SeriesEpisode
                data.metadata.season = response.data[info.moviehash].SeriesSeason
                data.metadata.episode_title = data.metadata.title.split('"')[2].trim()
                data.metadata.title = data.metadata.title.split('"')[1].trim()
            }

        } else if (response.data.accepted_moviehashes && response.data.accepted_moviehashes.length) { // we pushed new hash
            data.added = true
        } else { // this was a waste of time
            data.added = false
        }

        data.moviehash = info.moviehash // inject moviehash
        data.moviebytesize = info.moviebytesize // inject moviebytesize

        if (info.imdb && !data.metadata) { // inject imdb if possible
            data.metadata = {
                imdbid: 'tt' + info.imdb.replace('tt', '')
            }
        }

        return data
    }

    extend(data, response) {
        if (response.data !== String() && response.status.match(/200/)) {
            if (response.data.kind && response.data.kind.match(/episode/i)) {
                data.type = 'episode'
            } else if (response.data.kind && response.data.kind.match(/movie/i)) {
                data.type = 'movie'
            } else {
                delete data.metadata
                return data // it is impossible that a single file is an entire show
            }

            data.metadata.cast = response.data.cast ? response.data.cast : Object()
            data.metadata.country = response.data.country ? response.data.country : Array()
            data.metadata.cover = response.data.cover
            data.metadata.directors = response.data.directors ? response.data.directors : Object()
            data.metadata.duration = response.data.duration
            data.metadata.genres = response.data.genres ? response.data.genres : Array()
            data.metadata.rating = response.data.rating
            data.metadata.synopsis = response.data.plot && response.data.plot.match(/add a plot/i) !== null ? undefined : response.data.plot
            data.metadata.year = response.data.year
            data.metadata.trivia = response.data.trivia ? response.data.trivia.replace('See more >>', '').trim() : undefined
            data.metadata.goofs = response.data.goofs
            data.metadata.votes = response.data.votes
            data.metadata.language = response.data.language ? response.data.language : Array()
            data.metadata.aka = response.data.aka ? response.data.aka : Array()
            data.metadata.awards = response.data.awards ? response.data.awards : Array()
            data.metadata.tagline = response.data.tagline

            if (data.type === 'episode' && response.data.episodeof) {
                data.metadata.show_imdbid = Object.keys(response.data.episodeof).length > 0 ? Object.keys(response.data.episodeof)[0].replace('_','tt') : undefined
            }
        }

        return data
    }
}