287 lines
9.0 KiB
C++
287 lines
9.0 KiB
C++
// API - 2ebc8e784a4072da457fae5c0d291e48
|
||
// API READ ONLY -
|
||
// eyJhbGciOiJIUzI1NiJ9.eyJhdWQiOiIyZWJjOGU3ODRhNDA3MmRhNDU3ZmFlNWMwZDI5MWU0OCIsInN1YiI6IjYwZTJlNGI5MjJlNDgwMDA2MDJmZDMzMyIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9uIjoxfQ.c0y7bTCI5KSsfQRw7igPx1FR40mbMF6hGTJTHn0HXH8
|
||
#include <algorithm>
|
||
#include <iostream>
|
||
#include <map>
|
||
|
||
#include "../filesystem/filesystem.hpp"
|
||
#include "functions.hpp"
|
||
#include "../library.hpp"
|
||
|
||
#ifdef _WIN32
|
||
|
||
#include "rapidjson/document.h"
|
||
|
||
#else
|
||
|
||
#include <rapidjson/document.h>
|
||
|
||
#endif
|
||
|
||
#include <sstream>
|
||
#include <vector>
|
||
|
||
#ifdef _WIN32
|
||
|
||
#include <windows.h>
|
||
|
||
#include <codecvt>
|
||
#include <fcntl.h>
|
||
#include <io.h>
|
||
#include <locale>
|
||
|
||
constexpr const char_t *_tv_rename_dir_divider = L"\\";
|
||
|
||
#define toString( a ) utf8_to_wstring( a )
|
||
|
||
#else
|
||
|
||
constexpr const char_t *_tv_rename_dir_divider = "/";
|
||
|
||
#define toString( a ) a
|
||
|
||
#endif
|
||
|
||
string _moviedb_api_token{};
|
||
Request _moviedb_request;
|
||
|
||
struct MovieNames {
|
||
string name;
|
||
string original_name;
|
||
};
|
||
|
||
struct MovieInfo {
|
||
string name;
|
||
string original_name;
|
||
string year;
|
||
string id;
|
||
};
|
||
|
||
bool init( const string &config_path ) {
|
||
Request &request = _moviedb_request;
|
||
#ifdef _WIN32
|
||
request.setServer( TEXT( "api.themoviedb.org/3" ) );
|
||
#else
|
||
request.setServer( "https://api.themoviedb.org/3" );
|
||
#endif
|
||
_moviedb_api_token =
|
||
"eyJhbGciOiJIUzI1NiJ9."
|
||
"eyJhdWQiOiIyZWJjOGU3ODRhNDA3MmRhNDU3ZmFlNWMwZDI5MWU0OCIsInN1YiI6IjYwZT"
|
||
"JlNGI5MjJlNDgwMDA2MDJmZDMzMyIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9u"
|
||
"IjoxfQ.c0y7bTCI5KSsfQRw7igPx1FR40mbMF6hGTJTHn0HXH8";
|
||
return true;
|
||
}
|
||
|
||
bool hasKey( const rapidjson::GenericValue< rapidjson::UTF8<> > &object,
|
||
const std::string &key ) {
|
||
return object.FindMember( key.c_str() ) != object.MemberEnd();
|
||
}
|
||
|
||
std::vector< MovieInfo >
|
||
searchMovie( const string &movie, const string &language, const string &year ) {
|
||
Request &request = _moviedb_request;
|
||
request.addHeader( TEXT( "Accept: application/json" ) );
|
||
request.addHeader( TEXT( "Authorization: Bearer " ) + _moviedb_api_token );
|
||
|
||
auto encoded_show = encodeUrl( movie );
|
||
int pages = 0;
|
||
int cur_page = 0;
|
||
|
||
std::vector< MovieInfo > ret;
|
||
|
||
do {
|
||
cur_page++;
|
||
rapidjson::Document json;
|
||
auto request_uri = TEXT( "/search/movie?query=" ) + encoded_show +
|
||
TEXT( "&language=" ) + language + TEXT( "&page=" ) +
|
||
toString( std::to_string( cur_page ) );
|
||
if ( !year.empty() ) {
|
||
request_uri += TEXT( "&year=" ) + year;
|
||
}
|
||
json.Parse( request.get( request_uri ).c_str() );
|
||
if ( json.HasParseError() )
|
||
return {};
|
||
|
||
pages = json["total_pages"].GetInt();
|
||
|
||
const rapidjson::Value &results = json["results"];
|
||
if ( !results.IsArray() ) {
|
||
return {};
|
||
}
|
||
|
||
// find all possible movies
|
||
for ( size_t i = 0; i < results.Size(); i++ ) {
|
||
if ( !hasKey( results[i], "title" ) ||
|
||
!hasKey( results[i], "id" ) ||
|
||
!hasKey( results[i], "release_date" ) ||
|
||
!hasKey( results[i], "original_title" ) ) {
|
||
continue;
|
||
}
|
||
auto movie = toString( results[i]["title"].GetString() );
|
||
auto id = toString( std::to_string( results[i]["id"].GetInt() ) );
|
||
string year = toString( results[i]["release_date"].GetString() );
|
||
string original =
|
||
toString( results[i]["original_title"].GetString() );
|
||
if ( year.empty() ) {
|
||
year = "0000";
|
||
} else {
|
||
year = year.substr( 0, year.find( '-' ) );
|
||
}
|
||
MovieInfo tmp;
|
||
tmp.name = std::move( movie );
|
||
tmp.id = std::move( id );
|
||
tmp.original_name = std::move( original );
|
||
tmp.year = std::move( year );
|
||
ret.push_back( std::move( tmp ) );
|
||
}
|
||
} while ( cur_page < pages && cur_page < 5 );
|
||
request.clearHeader();
|
||
return ret;
|
||
}
|
||
|
||
RenameObject movieToRenameObject( const MovieInfo &movie,
|
||
const std::string &language ) {
|
||
RenameObject result;
|
||
result.setPresentedName( movie.name );
|
||
result.addCustomField( "id", movie.id );
|
||
result.addCustomField( "language", language );
|
||
result.addCustomField( "year", movie.year );
|
||
result.addCustomField( "original_title", movie.original_name );
|
||
result.addCustomField( "use_original", "false" );
|
||
return result;
|
||
}
|
||
|
||
std::vector< RenameObject > getOptions( const RenameObject &search ) {
|
||
string lang = "en-US";
|
||
string year = "";
|
||
if ( search.getCustomFields().find( "language" ) !=
|
||
search.getCustomFields().end() ) {
|
||
lang = search.getCustomFields().at( "language" );
|
||
}
|
||
if ( search.getCustomFields().find( "year" ) !=
|
||
search.getCustomFields().end() ) {
|
||
year = search.getCustomFields().at( "year" );
|
||
}
|
||
string name = search.getPresentedName();
|
||
auto possibilities = searchMovie( name, lang, year );
|
||
std::vector< RenameObject > result{};
|
||
for ( auto &movie : possibilities ) {
|
||
result.push_back( movieToRenameObject( movie, lang ) );
|
||
}
|
||
return result;
|
||
}
|
||
|
||
std::string removeIllegalCharacters( const std::string &input ) {
|
||
// replace '/' with '|'
|
||
std::string ret = input;
|
||
for ( size_t i = 0; i < ret.size(); i++ ) {
|
||
if ( ret[i] == '/' ) {
|
||
ret[i] = '|';
|
||
}
|
||
}
|
||
// replace characters illegal in windows
|
||
ret.erase( std::remove_if( ret.begin(), ret.end(),
|
||
[]( char_t x ) {
|
||
return x == '?' || x == '"' || x == '\\' ||
|
||
x == '*';
|
||
} ),
|
||
ret.end() );
|
||
for ( size_t i = 0; i < ret.size(); i++ ) {
|
||
if ( ret[i] == '|' ) {
|
||
ret[i] = '-';
|
||
} else if ( ret[i] == '<' ) {
|
||
ret.erase( i, 1 );
|
||
ret.insert( i, TEXT( "＜" ) );
|
||
} else if ( ret[i] == '>' ) {
|
||
ret.erase( i, 1 );
|
||
ret.insert( i, TEXT( "＞" ) );
|
||
} else if ( ret[i] == ':' ) {
|
||
ret[i] = ' ';
|
||
ret.insert( i + 1, TEXT( "- " ) );
|
||
}
|
||
}
|
||
for ( size_t i = 0; i < ret.size(); i++ ) {
|
||
if ( ret[i] == ' ' && ret[i + 1] == ' ' ) {
|
||
ret.erase( i, 1 );
|
||
i--;
|
||
}
|
||
}
|
||
return ret;
|
||
}
|
||
|
||
MovieNames movieFromId( const string &id, const string &language ) {
|
||
string uri = "/movie/" + id + "?language=" + language;
|
||
|
||
Request &request = _moviedb_request;
|
||
request.addHeader( TEXT( "Accept: application/json" ) );
|
||
request.addHeader( TEXT( "Authorization: Bearer " ) + _moviedb_api_token );
|
||
rapidjson::Document json;
|
||
json.Parse( request.get( uri ).c_str() );
|
||
if ( json.HasParseError() ) {
|
||
return { "", "" };
|
||
}
|
||
return { removeIllegalCharacters( json["title"].GetString() ),
|
||
removeIllegalCharacters( json["original_title"].GetString() ) };
|
||
}
|
||
|
||
bool renameMovie( const string &path, const string &name, const string &year ) {
|
||
return FSLib::rename(
|
||
path, FSLib::canonical( FSLib::getContainingDirectory( path ) ) + "/" +
|
||
name + " (" + year + ")." + FSLib::getFileExtension( path ) );
|
||
}
|
||
|
||
bool renamePath( const string &path, const RenameObject &renamer ) {
|
||
string id = "";
|
||
string lang = "en-US";
|
||
MovieNames movie = { "", "" };
|
||
string year = "";
|
||
bool use_original = false;
|
||
|
||
if ( renamer.getCustomFields().find( "language" ) !=
|
||
renamer.getCustomFields().end() ) {
|
||
lang = renamer.getCustomFields().at( "language" );
|
||
}
|
||
|
||
if ( renamer.getCustomFields().find( "year" ) !=
|
||
renamer.getCustomFields().end() ) {
|
||
year = renamer.getCustomFields().at( "year" );
|
||
}
|
||
|
||
if ( renamer.getCustomFields().find( "use_original" ) !=
|
||
renamer.getCustomFields().end() ) {
|
||
auto use = renamer.getCustomFields().at( "use_original" );
|
||
use_original = ( use == "true" || use == "True" || use == "TRUE" );
|
||
}
|
||
|
||
if ( renamer.getCustomFields().find( "id" ) ==
|
||
renamer.getCustomFields().end() ||
|
||
renamer.getCustomFields().at( "id" ).empty() ) {
|
||
auto results = searchMovie( renamer.getPresentedName(), lang, year );
|
||
if ( results.empty() )
|
||
return false;
|
||
id = results[0].id;
|
||
movie = { removeIllegalCharacters( results[0].name ),
|
||
removeIllegalCharacters( results[0].original_name ) };
|
||
year = results[0].year;
|
||
} else {
|
||
id = renamer.getCustomFields().at( "id" );
|
||
movie = movieFromId( id, lang );
|
||
}
|
||
|
||
if ( ( !use_original && movie.name == "" ) ||
|
||
( use_original && movie.original_name == "" ) )
|
||
return false;
|
||
|
||
return renameMovie( path, use_original ? movie.original_name : movie.name,
|
||
year );
|
||
}
|
||
|
||
std::vector< string > getCustomKeys() {
|
||
return { "id", "language", "year", "original_title", "use_original" };
|
||
}
|
||
|
||
const string getName() {
|
||
return "themoviedb";
|
||
}
|