Cache thumbnails

1 files changed, 24 insertions, 0 deletions

diff --git a/src/DownloadUtils.cpp b/src/DownloadUtils.cpp
index a61d0a1..6f31c2c 100644
--- a/src/DownloadUtils.cpp
+++ b/src/DownloadUtils.cpp
@@ -1,6 +1,8 @@
 #include "../include/DownloadUtils.hpp"
 #include "../include/Program.h"
+#include "../include/Storage.hpp"
 #include <SFML/System/Clock.hpp>
+#include <cppcodec/base64_rfc4648.hpp>
 
 static int accumulate_string(char *data, int size, void *userdata) {
     std::string *str = (std::string*)userdata;
@@ -29,6 +31,28 @@ namespace QuickMedia {
         return DownloadResult::OK;
     }
 
+    DownloadResult download_to_string_cache(const std::string &url, std::string &result, const std::vector<CommandArg> &additional_args, bool use_tor) {
+        Path media_dir = get_cache_dir().join("media");
+        Path media_file_path = Path(media_dir).join(cppcodec::base64_rfc4648::encode(url));
+        Path media_finished_path(media_file_path.data + ".finished");
+        if(get_file_type(media_finished_path) != FileType::FILE_NOT_FOUND && get_file_type(media_file_path) == FileType::REGULAR) {
+            if(file_get_content(media_file_path, result) == 0) {
+                return DownloadResult::OK;
+            } else {
+                fprintf(stderr, "Failed to get content of cached media file: %s\n", media_file_path.data.c_str());
+                return DownloadResult::ERR;
+            }
+        } else {
+            DownloadResult download_result = download_to_string(url, result, additional_args, use_tor);
+            if(download_result == DownloadResult::OK) {
+                if(create_directory_recursive(media_dir) == 0 && file_overwrite(media_file_path, result) == 0) {
+                    create_lock_file(media_finished_path);
+                }
+            }
+            return download_result;
+        }
+    }
+
     std::vector<CommandArg> create_command_args_from_form_data(const std::vector<FormData> &form_data) {
         // TODO: This boundary value might need to change, depending on the content. What if the form data contains the boundary value?
         const std::string boundary = "-----------------------------119561554312148213571335532670";