375 lines
10 KiB
C++
375 lines
10 KiB
C++
#include <iostream>
|
|
#include <set>
|
|
|
|
#ifdef _WIN32
|
|
|
|
#include <fcntl.h>
|
|
#include <io.h>
|
|
#include <windows.h>
|
|
|
|
using char_t = wchar_t;
|
|
using string = std::wstring;
|
|
|
|
#define cerr std::wcerr
|
|
#define cout std::wcout
|
|
#define cin std::wcin
|
|
|
|
#else
|
|
|
|
#include <getopt.h>
|
|
|
|
using char_t = char;
|
|
using string = std::string;
|
|
|
|
#define cerr std::cerr
|
|
#define cout std::cout
|
|
#define cin std::cin
|
|
|
|
#define TEXT( a ) a
|
|
|
|
#endif
|
|
|
|
#include "filesystem.hpp"
|
|
#include "functions.hpp"
|
|
#include "tv_rename.hpp"
|
|
|
|
// DB flags
|
|
#define DB_ADD 0x0001
|
|
#define DB_REFRESH 0x0002
|
|
#define DB_UPDATE 0x0004
|
|
#define DB_CLEAN 0x0008
|
|
#define DB_REMOVE 0x0010
|
|
#define DB_PATTERN 0x0020
|
|
|
|
// return 0 - all went as expected, 1 - request of help or print languages,
|
|
// -1 - error
|
|
int handleArgument( char_t c, string &show, std::set< int > &seasons_num,
|
|
string &path, string &language, string &pattern,
|
|
size_t &tv_flags, size_t &db_flags, string &db_pattern,
|
|
char_t *optional, int &i ) {
|
|
switch ( c ) {
|
|
case 's':
|
|
show = optional;
|
|
i++;
|
|
break;
|
|
case 'n':
|
|
parseSeasonNumbers( seasons_num, optional );
|
|
i++;
|
|
break;
|
|
case 'c':
|
|
tv_flags |= TV_CHDIR;
|
|
break;
|
|
case 'p':
|
|
path = optional;
|
|
i++;
|
|
// if path provided, assume it's correct
|
|
tv_flags &= ~TV_CHDIR;
|
|
break;
|
|
case 't':
|
|
tv_flags |= TV_TRUST;
|
|
break;
|
|
case 'x':
|
|
tv_flags |= TV_LINUX;
|
|
break;
|
|
case 'd':
|
|
tv_flags |= TV_DVD;
|
|
break;
|
|
case 'l':
|
|
if ( findLanguage( optional ) ) {
|
|
language = optional;
|
|
} else {
|
|
cerr << "Invalid language choice" << std::endl;
|
|
printLangs();
|
|
return -1;
|
|
}
|
|
i++;
|
|
break;
|
|
case 'h':
|
|
printHelp();
|
|
return 1;
|
|
case '0':
|
|
printLangs();
|
|
return 1;
|
|
case '1':
|
|
pattern = optional;
|
|
i++;
|
|
break;
|
|
case 'a':
|
|
db_flags |= DB_ADD;
|
|
break;
|
|
case 'r':
|
|
db_flags |= DB_REFRESH;
|
|
break;
|
|
case 'u':
|
|
db_flags |= DB_UPDATE;
|
|
break;
|
|
case '2':
|
|
db_flags |= DB_CLEAN;
|
|
break;
|
|
case '3':
|
|
db_flags |= DB_REMOVE;
|
|
break;
|
|
case '4':
|
|
db_pattern = optional;
|
|
db_flags |= DB_PATTERN;
|
|
i++;
|
|
break;
|
|
default:
|
|
return -1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
|
|
string getOptions( const char_t *option ) {
|
|
if ( option[0] != '-' )
|
|
return L"";
|
|
else if ( option[1] != '-' )
|
|
return option + 1;
|
|
else if ( !wcscmp( option, L"--show" ) )
|
|
return L"s";
|
|
else if ( !wcscmp( option, L"--season" ) )
|
|
return L"n";
|
|
else if ( !wcscmp( option, L"--correct-path" ) )
|
|
return L"c";
|
|
else if ( !wcscmp( option, L"--trust" ) )
|
|
return L"t";
|
|
else if ( !wcscmp( option, L"--linux" ) )
|
|
return L"x";
|
|
else if ( !wcscmp( option, L"--dvd" ) )
|
|
return L"d";
|
|
else if ( !wcscmp( option, L"--lang" ) )
|
|
return L"l";
|
|
else if ( !wcscmp( option, L"--help" ) )
|
|
return L"h";
|
|
else if ( !wcscmp( option, L"--print-langs" ) )
|
|
return L"0";
|
|
else if ( !wcscmp( option, L"--name-pattern" ) )
|
|
return L"1";
|
|
else if ( !wcscmp( option, L"--db-add" ) )
|
|
return L"a";
|
|
else if ( !wcscmp( option, L"--db-refresh" ) )
|
|
return L"r";
|
|
else if ( !wcscmp( option, L"--db-update" ) )
|
|
return L"u";
|
|
else if ( !wcscmp( option, L"--db-clean" ) )
|
|
return L"2";
|
|
else if ( !wcscmp( option, L"--db-remove" ) )
|
|
return L"3";
|
|
else if ( !wcscmp( option, L"--db-name-pattern" ) )
|
|
return L"4";
|
|
return L"";
|
|
}
|
|
|
|
// there's no getopt for windows, so just use wcscmp
|
|
int parseCommandLine( string &show, std::set< int > &seasons_num, string &path,
|
|
string &language, string &pattern, size_t &tv_flags,
|
|
size_t &db_flags, string &db_pattern, const int argc,
|
|
char_t **argv ) {
|
|
string options{};
|
|
char_t *optional;
|
|
for ( auto i = 1; i < argc; i++ ) {
|
|
options = getOptions( argv[i] );
|
|
if ( options == L"" ) {
|
|
options = L"p";
|
|
optional = argv[i];
|
|
} else {
|
|
optional = ( i < argc - 1 ) ? argv[i + 1] : nullptr;
|
|
}
|
|
for ( const auto &x : options ) {
|
|
auto res =
|
|
handleArgument( x, show, seasons_num, path, language, pattern,
|
|
tv_flags, db_flags, db_pattern, optional, i );
|
|
if ( res != 0 )
|
|
return res;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
#else
|
|
|
|
// parse command line arguments using getopt
|
|
int parseCommandLine( string &show, std::set< int > &seasons_num, string &path,
|
|
string &language, string &pattern, size_t &tv_flags,
|
|
size_t &db_flags, string &db_pattern, int argc,
|
|
char **argv ) {
|
|
static struct option long_options[] = {
|
|
{ "show", required_argument, 0, 's' },
|
|
{ "season", required_argument, 0, 'n' },
|
|
{ "correct-path", no_argument, 0, 'c' },
|
|
{ "trust", no_argument, 0, 't' },
|
|
{ "linux", no_argument, 0, 'x' },
|
|
{ "dvd", no_argument, 0, 'd' },
|
|
{ "lang", required_argument, 0, 'l' },
|
|
{ "help", no_argument, 0, 'h' },
|
|
{ "print-langs", no_argument, 0, '0' },
|
|
{ "name-pattern", required_argument, 0, '1' },
|
|
{ "db-add", no_argument, 0, 'a' },
|
|
{ "db-refresh", no_argument, 0, 'r' },
|
|
{ "db-update", no_argument, 0, 'u' },
|
|
{ "db-clean", no_argument, 0, '2' },
|
|
{ "db-remove", no_argument, 0, '3' },
|
|
{ "db-name-pattern", required_argument, 0, '4' },
|
|
{ 0, 0, 0, 0 }
|
|
};
|
|
|
|
int i{}; // this is useless, but needed for handleArgument
|
|
|
|
while ( 1 ) {
|
|
int option_index{ 0 };
|
|
auto c = getopt_long( argc, argv, "s:n:cp:txl:arudh", long_options,
|
|
&option_index );
|
|
if ( c == -1 )
|
|
break;
|
|
auto res =
|
|
handleArgument( c, show, seasons_num, path, language, pattern,
|
|
tv_flags, db_flags, db_pattern, optarg, i );
|
|
if ( res != 0 )
|
|
return res;
|
|
}
|
|
|
|
if ( optind < argc ) {
|
|
path = string( argv[optind] );
|
|
// if path provided, assume it's correct
|
|
tv_flags &= ~TV_CHDIR;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
#endif
|
|
|
|
// windows needs wmain for unicode support
|
|
#ifdef _WIN32
|
|
int wmain
|
|
#else
|
|
int main
|
|
#endif
|
|
( int argc, char_t **argv ) {
|
|
#ifdef _WIN32
|
|
// set console to unicode
|
|
_setmode( _fileno( stdout ), _O_U16TEXT );
|
|
#endif
|
|
string show{};
|
|
std::set< int > seasons_num{};
|
|
size_t tv_flags{};
|
|
size_t db_flags{};
|
|
string path{ TEXT( "." ) };
|
|
string language{ TEXT( "en" ) };
|
|
string pattern{ TEXT( "%filename - %epname" ) };
|
|
string db_pattern{};
|
|
|
|
authenticate( "42B66F5E-C6BF-423F-ADF9-CC97163472F6" );
|
|
|
|
{
|
|
auto tmp =
|
|
parseCommandLine( show, seasons_num, path, language, pattern,
|
|
tv_flags, db_flags, db_pattern, argc, argv );
|
|
if ( tmp == -1 )
|
|
return 1;
|
|
else if ( tmp == 1 )
|
|
return 0;
|
|
}
|
|
|
|
if ( !FSLib::isDirectory( path ) && FSLib::exists( path ) ) {
|
|
// specified file, not directory
|
|
std::map< int, string > file_set{};
|
|
size_t season_pos{}, ep_pos{};
|
|
if ( !searchSeason( path.c_str(), season_pos, ep_pos ) ) {
|
|
cerr << "Specified file does not conform to filename pattern, "
|
|
<< "cannot rename." << std::endl;
|
|
return 1;
|
|
} else {
|
|
file_set[std::stoi( path.c_str() + ep_pos )] = path;
|
|
auto season = std::stoi( path.c_str() + season_pos );
|
|
singleSeason( path, show, season, TEXT( "" ), language, pattern,
|
|
tv_flags, &file_set, true );
|
|
return 0;
|
|
}
|
|
} else if ( !FSLib::isDirectory( path ) ) {
|
|
tv_flags |= TV_CHDIR;
|
|
}
|
|
|
|
if ( !db_pattern.empty() ) {
|
|
changeDBPattern( db_pattern );
|
|
}
|
|
|
|
if ( db_flags & DB_REMOVE && db_flags & DB_ADD ) {
|
|
cerr << "You can't remove and add at the same time" << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
if ( db_flags & DB_REMOVE ) {
|
|
removeFromDB( FSLib::canonical( path ) );
|
|
}
|
|
if ( db_flags & DB_ADD ) {
|
|
addToDB( show, path, language, tv_flags & TV_LINUX );
|
|
cout << "Added to database" << std::endl;
|
|
}
|
|
if ( db_flags & DB_REFRESH ) {
|
|
refreshDB( tv_flags & TV_LINUX );
|
|
cout << "Refreshed database" << std::endl;
|
|
}
|
|
if ( db_flags & DB_UPDATE ) {
|
|
updateDB( tv_flags & TV_LINUX );
|
|
cout << "Updated database" << std::endl;
|
|
}
|
|
if ( db_flags & DB_CLEAN ) {
|
|
cleanDB();
|
|
cout << "Database cleaned" << std::endl;
|
|
}
|
|
|
|
// if db operations happened don't continue
|
|
if ( db_flags )
|
|
return 0;
|
|
|
|
while ( tv_flags & TV_CHDIR ) {
|
|
if ( !FSLib::isDirectory( path ) ) {
|
|
cout << "This directory doesn't exist, please insert a correct "
|
|
"path: "
|
|
<< std::endl;
|
|
std::getline( cin, path );
|
|
continue;
|
|
}
|
|
cout << "Is this the right directory? " << FSLib::canonical( path )
|
|
<< std::endl;
|
|
string response;
|
|
cin >> response;
|
|
cin.ignore( 1, '\n' );
|
|
cin.clear();
|
|
if ( response[0] == 'y' || response[0] == 'Y' ) {
|
|
tv_flags &= ~TV_CHDIR;
|
|
} else {
|
|
cout << "Insert correct path:" << std::endl;
|
|
std::getline( cin, path );
|
|
}
|
|
}
|
|
|
|
if ( show.empty() ) {
|
|
show = FSLib::canonical( path );
|
|
auto pos = show.find_last_of( '/' );
|
|
if ( pos != string::npos )
|
|
show = show.substr( ++pos );
|
|
cout << "Is this the right show name? " << show << std::endl;
|
|
string response;
|
|
cin >> response;
|
|
cin.ignore( 1, '\n' );
|
|
cin.clear();
|
|
if ( response[0] != 'y' && response[0] != 'Y' ) {
|
|
cout << "Insert the correct show name: " << std::endl;
|
|
std::getline( cin, show );
|
|
}
|
|
}
|
|
|
|
if ( seasons_num.size() == 1 ) {
|
|
singleSeason( path, show, *seasons_num.begin(), string(), language,
|
|
pattern, tv_flags, nullptr, true );
|
|
} else if ( seasons_num.size() != 0 ) {
|
|
multipleSeasons( path, show, seasons_num, language, pattern, tv_flags );
|
|
} else {
|
|
allSeasons( path, show, language, pattern, tv_flags );
|
|
}
|
|
}
|