diff --git a/src/episode.h b/src/episode.h index 4b49451..1fe4453 100644 --- a/src/episode.h +++ b/src/episode.h @@ -113,7 +113,7 @@ namespace dropout_dl { std::replace(this->series.begin(), this->series.end(), ',', '_'); - this->filename = this->series + "/S" + (this->season_number.size() < 2 ? "0" + this->season_number : this->season_number) + "E" + (this->episode_number.size() < 2 ? "0" + this->episode_number : this->episode_number) + this->name + ".mp4"; + this->filename = "S" + (this->season_number.size() < 2 ? "0" + this->season_number : this->season_number) + "E" + (this->episode_number.size() < 2 ? "0" + this->episode_number : this->episode_number) + this->name + ".mp4"; std::replace(filename.begin(), filename.end(), ' ', '_'); diff --git a/src/main.cpp b/src/main.cpp index 76c4806..59dbd2e 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -11,7 +11,6 @@ #endif #endif - static int sqlite_write_callback(void* data, int argc, char** argv, char** azColName) { if (argc < 1) { @@ -298,8 +297,8 @@ std::vector get_cookies(bool verbose = false) { int main(int argc, char** argv) { - bool verbose = true; - std::string quality = "1080p"; + bool verbose = false; + std::string quality = "all"; std::string firefox_profile; std::string chrome_profile; @@ -325,12 +324,6 @@ int main(int argc, char** argv) { dropout_dl::episode ep(episode_url, cookies, verbose); - std::string video_url = ep.get_video_url(quality); - if (verbose) { - std::cout << "Got video url: " << video_url << '\n'; - } - - if (!std::filesystem::is_directory(ep.series)) { std::filesystem::create_directories(ep.series); if (verbose) { @@ -338,9 +331,25 @@ int main(int argc, char** argv) { } } - std::fstream out(ep.filename, std::ios_base::in|std::ios_base::out|std::ios_base::trunc); + if (quality == "all") { + for (const auto& possible_quality : ep.qualities) { + if (!std::filesystem::is_directory(ep.series + "/" + possible_quality)) { + std::filesystem::create_directories(ep.series + "/" + possible_quality); + if (verbose) { + std::cout << "Creating series directory" << '\n'; + } + } + std::fstream out(ep.series + "/" + possible_quality + "/" + ep.filename, + std::ios_base::in | std::ios_base::out | std::ios_base::trunc); - out << ep.get_video_data(quality) << std::endl; + out << ep.get_video_data(possible_quality) << std::endl; + } + } + else { + std::fstream out(ep.series + "/" + ep.filename, std::ios_base::in | std::ios_base::out | std::ios_base::trunc); + + out << ep.get_video_data(quality) << std::endl; + } return 0; }