1 files changed, 24 insertions, 8 deletions

diff --git a/stream/src/fragment.rs b/stream/src/fragment.rs
index 26746fc..2ce3c78 100644
--- a/stream/src/fragment.rs
+++ b/stream/src/fragment.rs
@@ -4,8 +4,9 @@
     Copyright (C) 2025 metamuffin <metamuffin.org>
 */
 use crate::{stream_info, SMediaInfo};
-use anyhow::{anyhow, Result};
+use anyhow::{anyhow, bail, Result};
 use jellybase::common::stream::StreamContainer;
+use jellyremuxer::matroska_to_mpeg4;
 use jellytranscoder::fragment::transcode;
 use log::warn;
 use std::sync::Arc;
@@ -55,14 +56,13 @@ pub async fn fragment_stream(
             &format!("{path:?} {track_num} {index} {format_num} {container}"), // TODO maybe not use the entire source
             track.kind,
             format,
-            container,
             move |b| {
                 tokio::task::spawn_blocking(move || {
                     if let Err(err) = jellyremuxer::write_fragment_into(
                         SyncIoBridge::new(b),
                         &path,
                         track_num,
-                        container == StreamContainer::WebM,
+                        false,
                         &info.name.unwrap_or_default(),
                         index,
                     ) {
@@ -72,12 +72,28 @@ pub async fn fragment_stream(
             },
         )
         .await?;
-        let mut output = File::open(location.abs()).await?;
-        tokio::task::spawn(async move {
-            if let Err(err) = tokio::io::copy(&mut output, &mut b).await {
-                warn!("cannot write stream: {err}")
+        eprintln!("{:?}", location.abs());
+        let mut frag = File::open(location.abs()).await?;
+        match container {
+            StreamContainer::WebM => {}
+            StreamContainer::Matroska => {
+                tokio::task::spawn(async move {
+                    if let Err(err) = tokio::io::copy(&mut frag, &mut b).await {
+                        warn!("cannot write stream: {err}")
+                    }
+                });
             }
-        });
+            StreamContainer::MPEG4 => {
+                tokio::task::spawn_blocking(move || {
+                    if let Err(err) =
+                        matroska_to_mpeg4(SyncIoBridge::new(frag), SyncIoBridge::new(b))
+                    {
+                        warn!("mpeg4 transmux failed: {err}");
+                    }
+                });
+            }
+            _ => bail!("unsupported"),
+        }
     }
 
     Ok(())