115 lines
3.9 KiB
JavaScript
115 lines
3.9 KiB
JavaScript
// Dependencies
|
|
const cheerio = require('cheerio');
|
|
const request = require('request');
|
|
// Config
|
|
const config = require('../config');
|
|
|
|
class FlixInfo {
|
|
|
|
constructor(opt = '') {
|
|
if (opt == '') throw new Error('missing tmdb api key');
|
|
|
|
this.tmdbApiKey = opt;
|
|
}
|
|
|
|
getInfo(netflixId, cb) {
|
|
|
|
// Imdb id page
|
|
const flixableURL = config.filexibleURL + netflixId;
|
|
// The Movie DB api page
|
|
const theMovieDbURL = config.theMovieDbURL;
|
|
// API Key
|
|
const tmdbApiKey = this.tmdbApiKey;
|
|
|
|
var returnResult = {};
|
|
|
|
request(flixableURL, (err, response, body) => {
|
|
|
|
if (err) {
|
|
returnResult = { error: 1, errorMsg: 'flixable.com error' };
|
|
cb(returnResult)
|
|
}
|
|
else {
|
|
const $ = cheerio.load(body, {
|
|
normalizeWhitespace: true,
|
|
xmlMode: true
|
|
});
|
|
|
|
const imdbResult = $('[class="imdbRatingPlugin"]').length;
|
|
|
|
if (imdbResult) {
|
|
$('[class="imdbRatingPlugin"]').each(function (i, element) {
|
|
|
|
const imdbId = $(this).attr('data-title');
|
|
const tmdbURL = theMovieDbURL + imdbId + '?api_key=' + tmdbApiKey + '&language=en-EN&external_source=imdb_id'
|
|
|
|
request(tmdbURL, (err, resultBody, body) => {
|
|
|
|
const getJson = JSON.parse(resultBody.body);
|
|
|
|
if(getJson.status_code > 1){
|
|
// if wrong api key error
|
|
returnResult = { error: 1, errorMsg: 'tmdb wrong api key error' };
|
|
cb(returnResult)
|
|
}
|
|
else{
|
|
|
|
if (err) {
|
|
// if thmdb api connection error
|
|
returnResult = { error: 1, errorMsg: 'tmdb api error' };
|
|
cb(returnResult)
|
|
}
|
|
else {
|
|
if (getJson.tv_results.length) {
|
|
// If result is tv
|
|
var result = getJson.tv_results[0];
|
|
var originalName = result.original_name;
|
|
var name = result.name;
|
|
var year = result.first_air_date;
|
|
if (result.poster_path != null) {
|
|
var poster = 'https://image.tmdb.org/t/p/original/' + result.poster_path;
|
|
}
|
|
if (result.poster_path != null) {
|
|
var backdrop = 'https://image.tmdb.org/t/p/original/' + result.backdrop_path;
|
|
}
|
|
var country = result.origin_country[0];
|
|
var rate = result.vote_average;
|
|
var overviewEN = result.overview;
|
|
|
|
}
|
|
else if (getJson.movie_results.length) {
|
|
// If result is movie
|
|
var result = getJson.movie_results[0];
|
|
var originalName = result.original_title;
|
|
var name = result.title;
|
|
var year = result.release_date;
|
|
if (result.poster_path != null) {
|
|
var poster = 'https://image.tmdb.org/t/p/original/' + result.poster_path;
|
|
}
|
|
if (result.poster_path != null) {
|
|
var backdrop = 'https://image.tmdb.org/t/p/original/' + result.backdrop_path;
|
|
}
|
|
var country = result.original_language;
|
|
var rate = result.vote_average;
|
|
var overviewEN = result.overview;
|
|
}
|
|
|
|
returnResult = { result: 1, error: 0, originalName, name, year, poster, backdrop, country, rate, overviewEN };
|
|
// If there is no error return the result
|
|
cb(returnResult);
|
|
}
|
|
}
|
|
});
|
|
})
|
|
}
|
|
else {
|
|
// Return the error if imdb id cannot be found
|
|
const returnResult = { result: 0, error: 0 };
|
|
cb(returnResult)
|
|
}
|
|
}
|
|
});
|
|
}
|
|
}
|
|
module.exports = FlixInfo;
|