1 files changed, 36 insertions, 7 deletions
diff --git a/server/src/routes/stream.rs b/server/src/routes/stream.rs
index c47f589..018fae5 100644
--- a/server/src/routes/stream.rs
+++ b/server/src/routes/stream.rs
@@ -10,14 +10,29 @@ use jellybase::CONF;
 use jellycommon::{stream::StreamSpec, MediaSource};
 use log::{info, warn};
 use rocket::{
-    get,
-    http::{ContentType, Header, Status},
+    get, head,
+    http::{Header, Status},
     request::{self, FromRequest},
     response::{self, Redirect, Responder},
     Either, Request, Response, State,
 };
 use std::{ops::Range, time::Duration};
-use tokio::io::DuplexStream;
+use tokio::io::{duplex, DuplexStream};
+
+#[head("/n/<_id>/stream?<spec>")]
+pub async fn r_stream_head(
+    _sess: Session,
+    _id: &str,
+    spec: StreamSpec,
+) -> Result<Either<StreamResponse, Redirect>, MyError> {
+    let head = jellystream::stream_head(&spec);
+    Ok(Either::Left(StreamResponse {
+        stream: duplex(0).0,
+        advertise_range: head.range_supported,
+        content_type: head.content_type,
+        range: None,
+    }))
+}
 
 #[get("/n/<id>/stream?<spec>")]
 pub async fn r_stream(
@@ -28,7 +43,10 @@ pub async fn r_stream(
     range: Option<RequestRange>,
     spec: StreamSpec,
 ) -> Result<Either<StreamResponse, Redirect>, MyError> {
-    let node = db.node.get(&id.to_string())?.ok_or(anyhow!("node does not exist"))?;
+    let node = db
+        .node
+        .get(&id.to_string())?
+        .ok_or(anyhow!("node does not exist"))?;
     let source = node
         .private
         .source
@@ -70,8 +88,15 @@ pub async fn r_stream(
         None => 0..(isize::MAX as usize),
     };
 
+    let head = jellystream::stream_head(&spec);
+
     match jellystream::stream(node, spec, urange).await {
-        Ok(stream) => Ok(Either::Left(StreamResponse { stream, range })),
+        Ok(stream) => Ok(Either::Left(StreamResponse {
+            stream,
+            range,
+            advertise_range: head.range_supported,
+            content_type: head.content_type,
+        })),
         Err(e) => {
             warn!("stream error: {e}");
             Err(MyError(e))
@@ -81,6 +106,8 @@ pub async fn r_stream(
 
 pub struct StreamResponse {
     stream: DuplexStream,
+    advertise_range: bool,
+    content_type: &'static str,
     range: Option<RequestRange>,
 }
 
@@ -92,8 +119,10 @@ impl<'r> Responder<'r, 'static> for StreamResponse {
             b.status(Status::PartialContent);
             b.header(Header::new("content-range", range.to_cr_hv()));
         }
-        b.header(Header::new("accept-ranges", "bytes"))
-            .header(ContentType::WEBM)
+        if self.advertise_range {
+            b.header(Header::new("accept-ranges", "bytes"));
+        }
+        b.header(Header::new("content-type", self.content_type))
             .streamed_body(self.stream)
             .ok()
     }