Fix request header not passed on cached extractor response & support …

…HEAD method to video endpoint
mhdzumair · Nov 17, 2024 · 641fd17 · 641fd17
1 parent f69b67d
commit 641fd17
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mediaflow_proxy/routes/extractor.py b/mediaflow_proxy/routes/extractor.py
@@ -19,6 +19,7 @@
 logger = logging.getLogger(__name__)
 
 
+@extractor_router.head("/video")
 @extractor_router.get("/video")
 async def extract_url(
     extractor_params: Annotated[ExtractorURLParams, Query()],
@@ -33,6 +34,8 @@ async def extract_url(
             extractor = ExtractorFactory.get_extractor(extractor_params.host, proxy_headers.request)
             response = await extractor.extract(extractor_params.destination, **extractor_params.extra_params)
             await set_cache_extractor_result(cache_key, response)
+        else:
+            response["request_headers"].update(proxy_headers.request)
 
         response["mediaflow_proxy_url"] = str(
             request.url_for(response.pop("mediaflow_endpoint")).replace(scheme=get_original_scheme(request))