Faster mangakatana search on exact match

1 files changed, 9 insertions, 3 deletions

diff --git a/src/plugins/Youtube.cpp b/src/plugins/Youtube.cpp
index 4934c43..b490e97 100644
--- a/src/plugins/Youtube.cpp
+++ b/src/plugins/Youtube.cpp
@@ -237,20 +237,26 @@ namespace QuickMedia {
         std::vector<CommandArg> additional_args = get_cookies();
         additional_args.push_back({"--no-buffer", ""});
 
+        std::vector<std::string> response_headers;
         const int max_redirects = 5;
         for(int i = 0; i < max_redirects; ++i) {
             std::string response_body;
-            std::string response_headers;
+            response_headers.clear();
             download_to_string(playback_url, response_body, additional_args, true, true, false, &response_headers, 4096);
 
-            std::string content_type = header_extract_value(response_headers, "content-type");
+            if(response_headers.empty()) {
+                fprintf(stderr, "Youtube video header not found\n");
+                return "";
+            }
+
+            std::string content_type = header_extract_value(response_headers.back(), "content-type");
             if(content_type.empty()) {
                 fprintf(stderr, "Failed to find content-type in youtube video header\n");
                 return "";
             }
 
             if(string_starts_with(content_type, "video") || string_starts_with(content_type, "audio")) {
-                std::string content_length_str = header_extract_value(response_headers, "content-length");
+                std::string content_length_str = header_extract_value(response_headers.back(), "content-length");
                 if(content_length_str.empty())
                     return "";