diff --git a/src/color.h b/src/color.h new file mode 100644 index 0000000..ec63d9e --- /dev/null +++ b/src/color.h @@ -0,0 +1,29 @@ +// +// Created by moss on 12/24/22. +// + +#pragma once + +#define RESET "\033[0m" +#define BLACK "\033[30m" /* Black */ +#define RED "\033[31m" /* Red */ +#define GREEN "\033[32m" /* Green */ +#define YELLOW "\033[33m" /* Yellow */ +#define BLUE "\033[34m" /* Blue */ +#define MAGENTA "\033[35m" /* Magenta */ +#define CYAN "\033[36m" /* Cyan */ +#define WHITE "\033[37m" /* White */ +#define BOLDBLACK "\033[1m\033[30m" /* Bold Black */ +#define BOLDRED "\033[1m\033[31m" /* Bold Red */ +#define FAIL "\033[31mFAIL: \033[0m" // Test Failure +#define BOLDFAIL "\033[1m\033[31mFAIL: \033[0m" // Test Failure +#define BOLDGREEN "\033[1m\033[32m" /* Bold Green */ +#define SUCCESS "\033[32mSUCCESS: \033[0m" /* Test Success */ +#define BOLDSUCCESS "\033[1m\033[32mSUCCESS: \033[0m" /* Test Success */ +#define BOLDYELLOW "\033[1m\033[33m" /* Bold Yellow */ +#define WARN "\033[1m\033[33mWARNING: \033[0m" /* Test Warning */ +#define BOLDBLUE "\033[1m\033[34m" /* Bold Blue */ +#define TESTNAME "\033[1m\033[34mTEST: \033[0m" /* Bold Blue */ +#define BOLDMAGENTA "\033[1m\033[35m" /* Bold Magenta */ +#define BOLDCYAN "\033[1m\033[36m" /* Bold Cyan */ +#define BOLDWHITE "\033[1m\033[37m" /* Bold White */ diff --git a/src/episode.cpp b/src/episode.cpp index edd162d..9b72d28 100644 --- a/src/episode.cpp +++ b/src/episode.cpp @@ -265,7 +265,6 @@ namespace dropout_dl { // Skip "VIDEO_TITLE", the following colon, and the opening quotation mark. i += video_title_title.size() + 2; - int j; for (j = 0; meta_data[i + j] != '"' && i + j < meta_data.size(); j++) { // skip checking for quotes if prefaced by a forward slash @@ -518,6 +517,31 @@ namespace dropout_dl { } + void episode::download_quality(const std::string& quality, const std::string& base_directory, const std::string& filename) { + if (!std::filesystem::is_directory(base_directory)) { + std::filesystem::create_directories(base_directory); + if (this->verbose) { + std::cout << "Creating quality directory" << '\n'; + } + } + + std::string filepath = base_directory + "/" + filename; + + if(std::filesystem::is_regular_file(filepath)) { + std::cout << YELLOW << "File already exists: " << filepath << RESET << '\n'; + return; + } + std::fstream out(filepath, + std::ios_base::in | std::ios_base::out | std::ios_base::trunc); + + out << this->get_video_data(quality, filepath) << std::endl; + + std::cout << GREEN << filepath << RESET; + + std::cout << '\n'; + } + + void episode::download(const std::string& quality, const std::string& series_directory, std::string filename) { if (filename.empty()) { if (this->episode_number != 0) { @@ -537,32 +561,12 @@ namespace dropout_dl { if (quality == "all") { for (const auto &possible_quality: this->qualities) { - if (!std::filesystem::is_directory(series_directory + "/" + possible_quality)) { - std::filesystem::create_directories(series_directory + "/" + possible_quality); - if (this->verbose) { - std::cout << "Creating quality directory" << '\n'; - } - } - std::fstream out(series_directory + "/" + possible_quality + "/" + filename, - std::ios_base::in | std::ios_base::out | std::ios_base::trunc); - - out << this->get_video_data(possible_quality, series_directory + "/" + possible_quality + "/" + filename) << std::endl; + this->download_quality(possible_quality, series_directory + possible_quality, filename); } } else { - if (!std::filesystem::is_directory(series_directory)) { - std::filesystem::create_directories(series_directory); - if (this->verbose) { - std::cout << "Creating quality directory" << '\n'; - } - } - - std::fstream out(series_directory + "/" + filename, - std::ios_base::in | std::ios_base::out | std::ios_base::trunc); - - out << this->get_video_data(quality, series_directory + "/" + filename) << std::endl; + this->download_quality(quality, series_directory, filename); } - std::cout << '\n'; } diff --git a/src/episode.h b/src/episode.h index 08faf9b..9cccf56 100644 --- a/src/episode.h +++ b/src/episode.h @@ -9,6 +9,7 @@ #include #include #include +#include "color.h" #include #ifdef DROPOUT_DL_GCRYPT #include @@ -259,13 +260,13 @@ namespace dropout_dl { /// The name of the season that the episode belongs to std::string season; /// The number of the season (only set when downloading a season or series) - int season_number; + int season_number = 0; /// The json metadata of the episode std::string metadata; /// The name of the episode std::string name; /// The number of the episode (only set when downloading a season or series) - int episode_number; + int episode_number = 0; /// The url for the main episode page std::string episode_url; /// The data of the main episode page @@ -382,6 +383,18 @@ namespace dropout_dl { */ std::string get_video_data(const std::string& quality, const std::string& filename = ""); + + /** + * + * @param quality - The quality of the video + * @param base_directory - The directory which the episode is downloaded into + * @param filename - The name of the file (Will default if empty) + * + * Downloads the episode using the get_video_data function and places it into the filename file in the base_directory directory. + * If the file already exists it will output the name in yellow and will not redownload. + */ + void download_quality(const std::string& quality, const std::string& base_directory, const std::string& filename); + /** * * @param quality - The quality of the video