Add range header handling for ump streams.

See TeamPiped/Piped#3196
TeamPiped · Nov 30, 2023 · 2295ad3 · 2295ad3
1 parent 75f7c4e
commit 2295ad3
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/main.rs b/src/main.rs
@@ -142,7 +142,7 @@ async fn index(req: HttpRequest) -> Result<HttpResponse, Box<dyn Error>> {
     }
 
     // parse query string
-    let query = QString::from(req.query_string());
+    let mut query = QString::from(req.query_string());
 
     #[cfg(feature = "qhash")]
     {
@@ -236,6 +236,14 @@ async fn index(req: HttpRequest) -> Result<HttpResponse, Box<dyn Error>> {
 
     let mime_type = query.get("mime").map(|s| s.to_string());
 
+    if is_ump && !query.has("range") {
+        if let Some(range) = req.headers().get("range") {
+            let range = range.to_str().unwrap();
+            let range = range.replace("bytes=", "");
+            query.add_pair(("range", range));
+        }
+    }
+
     let qs = {
         let collected = query
             .into_pairs()