neutrino/src/gui/tmdb.cpp

/*
	Copyright (C) 2015-2020 TangoCash

	License: GPLv2

	This program is free software; you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation;

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program; if not, write to the Free Software
	Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/


#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <errno.h>
#include <fstream>

#include <set>
#include <string>

#include <neutrino.h>

#include "system/settings.h"
#include "system/set_threadname.h"
#include "gui/widget/hintbox.h"

#include <driver/screen_max.h>

#include <global.h>

#include "tmdb.h"

cTmdb* cTmdb::getInstance()
{
	static cTmdb* tmdb = NULL;
	if(!tmdb)
		tmdb = new cTmdb();
	return tmdb;
}

cTmdb::cTmdb()
{
#ifdef TMDB_API_KEY
	key = TMDB_API_KEY;
#else
	key = g_settings.tmdb_api_key;
#endif
}

cTmdb::~cTmdb()
{
    cleanup();
}

void cTmdb::setTitle(std::string epgtitle)
{
	minfo.epgtitle = epgtitle;

	CHintBox hintbox(LOCALE_MESSAGEBOX_INFO, g_Locale->getText(LOCALE_TMDB_READ_DATA));
	hintbox.paint();

	std::string lang = Lang2ISO639_1(g_settings.language);
	GetMovieDetails(lang);
	if ((minfo.result < 1 || minfo.overview.empty()) && lang != "en")
		GetMovieDetails("en", true);

	hintbox.hide();
}

bool cTmdb::GetData(std::string url, Json::Value *root)
{
	std::string answer;
	if (!getUrl(url, answer))
		return false;

	std::string errMsg = "";
	bool ok = parseJsonFromString(answer, root, &errMsg);
	if (!ok) {
		printf("Failed to parse JSON\n");
		printf("%s\n", errMsg.c_str());
		return false;
	}
	return true;
}

bool cTmdb::GetMovieDetails(std::string lang, bool second)
{
	printf("[TMDB]: %s\n",__func__);
	Json::Value root;
	const std::string urlapi = "http://api.themoviedb.org/3/";
	std::string url	= urlapi + "search/multi?api_key="+key+"&language="+lang+"&query=" + encodeUrl(minfo.epgtitle);
	if(!(GetData(url, &root)))
		return false;

	minfo.result = root.get("total_results",0).asInt();
	if(minfo.result == 0){
		std::string title = minfo.epgtitle;
		size_t pos1 = title.find_last_of("(");
		size_t pos2 = title.find_last_of(")");
		if(pos1 != std::string::npos && pos2 != std::string::npos && pos2 > pos1){
			printf("[TMDB]: second try\n");
			title.replace(pos1, pos2-pos1+1, "");
			url	= urlapi + "search/multi?api_key="+key+"&language="+lang+"&query=" + encodeUrl(title);
			if(!(GetData(url, &root)))
				return false;
		}
		minfo.result = root.get("total_results",0).asInt();
	}
	printf("[TMDB]: results: %d\n",minfo.result);

	if (minfo.result > 0) {
		Json::Value elements = root["results"];
		int use_result = 0;

		if ((minfo.result > 1) && (!second))
			selectResult(elements, minfo.result, use_result);

		if (!second) {
			minfo.id = elements[use_result].get("id",-1).asInt();
			minfo.media_type = elements[use_result].get("media_type","").asString();
		}
		if (minfo.id > -1) {
			url = urlapi+minfo.media_type+"/"+to_string(minfo.id)+"?api_key="+key+"&language="+lang+"&append_to_response=credits";
			if(!(GetData(url, &root)))
				return false;

			minfo.overview = root.get("overview","").asString();
			minfo.poster_path = root.get("poster_path","").asString();
			minfo.original_title = root.get("original_title","").asString();;
			minfo.release_date = root.get("release_date","").asString();;
			minfo.vote_average = root.get("vote_average","").asString();;
			minfo.vote_count = root.get("vote_count",0).asInt();;
			minfo.runtime = root.get("runtime",0).asInt();;
			if (minfo.media_type == "tv") {
				minfo.original_title = root.get("original_name","").asString();;
				minfo.episodes = root.get("number_of_episodes",0).asInt();;
				minfo.seasons = root.get("number_of_seasons",0).asInt();;
				minfo.release_date = root.get("first_air_date","").asString();;
				elements = root["episode_run_time"];
				minfo.runtimes = elements[0].asString();
				for (unsigned int i= 1; i<elements.size();i++) {
					minfo.runtimes +=  + ", "+elements[i].asString();
				}
			}

			elements = root["genres"];
			minfo.genres = elements[0].get("name","").asString();
			for (unsigned int i= 1; i<elements.size();i++) {
				minfo.genres += ", " + elements[i].get("name","").asString();
			}

			elements = root["credits"]["cast"];
			minfo.cast.clear();
			for (unsigned int i= 0; i<elements.size() && i<10;i++) {
				minfo.cast +=  "  "+elements[i].get("character","").asString()+" ("+elements[i].get("name","").asString() + ")\n";
				//printf("test: %s (%s)\n",elements[i].get("character","").asString().c_str(),elements[i].get("name","").asString().c_str());
			}

			unlink(TMDB_COVER);
			if (hasCover())
				getBigCover(TMDB_COVER);
			//printf("[TMDB]: %s (%s) %s\n %s\n %d\n",minfo.epgtitle.c_str(),minfo.original_title.c_str(),minfo.release_date.c_str(),minfo.overview.c_str(),minfo.found);

			return true;
		}
	} else
		return false;

	return false;
}

std::string cTmdb::CreateEPGText()
{
	std::string epgtext;
	epgtext += "Vote: "+minfo.vote_average.substr(0,3)+"/10 Votecount: "+to_string(minfo.vote_count)+"\n";
	epgtext += "\n";
	epgtext += minfo.overview+"\n";
	epgtext += "\n";
	if (minfo.media_type == "tv")
		epgtext += (std::string)g_Locale->getText(LOCALE_EPGVIEWER_LENGTH)+": "+minfo.runtimes+"\n";
	else
		epgtext += (std::string)g_Locale->getText(LOCALE_EPGVIEWER_LENGTH)+": "+to_string(minfo.runtime)+"\n";
	epgtext += (std::string)g_Locale->getText(LOCALE_EPGVIEWER_GENRE)+": "+minfo.genres+"\n";
	epgtext += (std::string)g_Locale->getText(LOCALE_EPGEXTENDED_ORIGINAL_TITLE) +" : "+ minfo.original_title+"\n";
	epgtext += (std::string)g_Locale->getText(LOCALE_EPGEXTENDED_YEAR_OF_PRODUCTION)+" : "+ minfo.release_date.substr(0,4) +"\n";
	if (minfo.media_type == "tv")
		epgtext += "Seasons/Episodes: "+to_string(minfo.seasons)+"/"+to_string(minfo.episodes)+"\n";
	if (!minfo.cast.empty())
		epgtext += (std::string)g_Locale->getText(LOCALE_EPGEXTENDED_ACTORS)+":\n"+ minfo.cast+"\n";
	return epgtext;
}

void cTmdb::cleanup()
{
	if (access(TMDB_COVER, F_OK) == 0)
		unlink(TMDB_COVER);
}

void cTmdb::selectResult(Json::Value elements, int results, int &use_result)
{
	int select = 0;

	CMenuWidget *m = new CMenuWidget(LOCALE_TMDB_READ_DATA, NEUTRINO_ICON_SETTINGS);
	CMenuSelectorTarget * selector = new CMenuSelectorTarget(&select);

	// we don't show introitems, so we add a separator for a smoother view
	m->addItem(GenericMenuSeparator);
	CMenuForwarder* mf;
	int counter = std::min(results, 10);
	for (int i = 0; i != counter; i++)
	{
		if (elements[i].get("media_type","").asString() == "movie")
			mf = new CMenuForwarder(elements[i].get("title","").asString(), true, NULL, selector, to_string(i).c_str());
		else
			mf = new CMenuForwarder(elements[i].get("name","").asString(), true, NULL, selector, to_string(i).c_str());
		m->addItem(mf);
	}

	m->enableSaveScreen();
	m->exec(NULL, "");
	if (!m->gotAction())
		return;
	delete selector;
	m->hide();
	use_result = select;
}