diff options
Diffstat (limited to 'stream/src')
-rw-r--r-- | stream/src/fragment.rs | 6 | ||||
-rw-r--r-- | stream/src/fragment_index.rs | 2 | ||||
-rw-r--r-- | stream/src/hls.rs | 4 | ||||
-rw-r--r-- | stream/src/lib.rs | 17 | ||||
-rw-r--r-- | stream/src/stream_info.rs | 8 | ||||
-rw-r--r-- | stream/src/webvtt.rs | 150 |
6 files changed, 83 insertions, 104 deletions
diff --git a/stream/src/fragment.rs b/stream/src/fragment.rs index 49817ca..89ce94f 100644 --- a/stream/src/fragment.rs +++ b/stream/src/fragment.rs @@ -5,8 +5,8 @@ */ use crate::{stream_info, SMediaInfo}; use anyhow::{anyhow, bail, Result}; -use jellycommon::stream::StreamContainer; use jellyremuxer::{matroska_to_mpeg4, matroska_to_webm::matroska_to_webm}; +use jellystream_types::StreamContainer; use jellytranscoder::fragment::transcode; use log::warn; use std::{ @@ -84,9 +84,7 @@ pub async fn fragment_stream( } }); } - StreamContainer::JVTT => { - - } + StreamContainer::JVTT => {} _ => bail!("not yet supported"), } } else { diff --git a/stream/src/fragment_index.rs b/stream/src/fragment_index.rs index 0632f0a..cb54948 100644 --- a/stream/src/fragment_index.rs +++ b/stream/src/fragment_index.rs @@ -5,7 +5,7 @@ */ use crate::{stream_info, SMediaInfo}; use anyhow::{anyhow, Result}; -use jellycommon::stream::{SegmentNum, TrackNum}; +use jellystream_types::{SegmentNum, TrackNum}; use std::sync::Arc; use tokio::io::{AsyncWriteExt, DuplexStream}; diff --git a/stream/src/hls.rs b/stream/src/hls.rs index 41b896b..949ddb4 100644 --- a/stream/src/hls.rs +++ b/stream/src/hls.rs @@ -6,9 +6,7 @@ use crate::{stream_info, SMediaInfo}; use anyhow::{anyhow, Result}; -use jellycommon::stream::{ - FormatNum, SegmentNum, StreamContainer, StreamSpec, TrackKind, TrackNum, -}; +use jellystream_types::{FormatNum, SegmentNum, StreamContainer, StreamSpec, TrackKind, TrackNum}; use std::{fmt::Write, ops::Range, sync::Arc}; use tokio::{ io::{AsyncWriteExt, DuplexStream}, diff --git a/stream/src/lib.rs b/stream/src/lib.rs index 9c62ac8..5b4e8ed 100644 --- a/stream/src/lib.rs +++ b/stream/src/lib.rs @@ -4,20 +4,17 @@ Copyright (C) 2025 metamuffin <metamuffin.org> */ #![feature(iterator_try_collect)] -pub mod fragment; -pub mod fragment_index; -pub mod hls; -pub mod stream_info; -pub mod webvtt; +mod fragment; +mod fragment_index; +mod hls; +mod stream_info; +mod webvtt; use anyhow::{anyhow, bail, Context, Result}; use fragment::fragment_stream; use fragment_index::fragment_index_stream; use hls::{hls_multivariant_stream, hls_supermultivariant_stream, hls_variant_stream}; -use jellycommon::{ - stream::{StreamContainer, StreamSpec}, - Node, -}; +use jellystream_types::{StreamContainer, StreamSpec}; use serde::{Deserialize, Serialize}; use std::{ collections::BTreeSet, @@ -53,7 +50,7 @@ static CONF: LazyLock<Config> = LazyLock::new(|| { #[derive(Debug)] pub struct SMediaInfo { - pub info: Arc<Node>, + pub title: Option<String>, pub files: BTreeSet<PathBuf>, } diff --git a/stream/src/stream_info.rs b/stream/src/stream_info.rs index 920ce69..560ec9b 100644 --- a/stream/src/stream_info.rs +++ b/stream/src/stream_info.rs @@ -5,13 +5,13 @@ */ use crate::{SMediaInfo, CONF}; use anyhow::Result; -use jellycommon::stream::{ - StreamContainer, StreamFormatInfo, StreamInfo, StreamSegmentInfo, StreamTrackInfo, TrackKind, -}; use jellyremuxer::{ metadata::{matroska_metadata, MatroskaMetadata, MatroskaTrackEntry}, seek_index::get_track_sizes, }; +use jellystream_types::{ + StreamContainer, StreamFormatInfo, StreamInfo, StreamSegmentInfo, StreamTrackInfo, TrackKind, +}; use std::{collections::BTreeMap, path::PathBuf, sync::Arc}; use tokio::{ io::{AsyncWriteExt, DuplexStream}, @@ -76,7 +76,7 @@ pub(crate) async fn stream_info(info: Arc<SMediaInfo>) -> Result<(InternalStream track_to_file, }, StreamInfo { - name: info.info.title.clone(), + name: info.title.clone(), segments: vec![segment], }, )) diff --git a/stream/src/webvtt.rs b/stream/src/webvtt.rs index c0bc466..d99111e 100644 --- a/stream/src/webvtt.rs +++ b/stream/src/webvtt.rs @@ -3,91 +3,77 @@ which is licensed under the GNU Affero General Public License (version 3); see /COPYING. Copyright (C) 2025 metamuffin <metamuffin.org> */ -use anyhow::Result; -use jellycommon::{stream::StreamSpec, Node}; -use std::sync::Arc; -use tokio::io::DuplexStream; -pub async fn vtt_stream( - json: bool, - node: Arc<Node>, - spec: StreamSpec, - b: DuplexStream, -) -> Result<()> { - let _ = b; - let _ = spec; - let _ = node; - let _ = json; - // TODO cache +// async fn vtt_stream(json: bool, node: Arc<Node>, spec: StreamSpec, b: DuplexStream) -> Result<()> { +// TODO cache +// TODO should use fragments too? big films take too long... - // TODO should use fragments too? big films take too long... +// let tracki = *spec.track.first().ok_or(anyhow!("no track selected"))?; +// let local_track = local_tracks.first().ok_or(anyhow!("no tracks"))?.clone(); +// let track = &node.media.as_ref().unwrap().tracks[tracki]; +// let cp = local_track.codec_private.clone(); - // let tracki = *spec.track.first().ok_or(anyhow!("no track selected"))?; - // let local_track = local_tracks.first().ok_or(anyhow!("no tracks"))?.clone(); - // let track = &node.media.as_ref().unwrap().tracks[tracki]; - // let cp = local_track.codec_private.clone(); +// let subtitles = async_cache_memory( +// &[ +// "vtt", +// &format!( +// "{} {}", +// local_track.path.to_str().unwrap(), +// local_track.track +// ), +// ], +// move || async move { +// let blocks = tokio::task::spawn_blocking(move || { +// extract_track(CONF.media_path.clone(), local_track) +// }) +// .await??; +// let subtitles = parse_subtitles(&track.codec, cp, blocks)?; +// Ok(subtitles) +// }, +// )spec.track.first().ok_or(anyhow!("no track selected"))?; +// let local_track = local_tracks.first().ok_or(anyhow!("no tracks"))?.clone(); +// let track = &node.media.as_ref().unwrap().tracks[tracki]; +// let cp = local_track.codec_private.clone(); - // let subtitles = async_cache_memory( - // &[ - // "vtt", - // &format!( - // "{} {}", - // local_track.path.to_str().unwrap(), - // local_track.track - // ), - // ], - // move || async move { - // let blocks = tokio::task::spawn_blocking(move || { - // extract_track(CONF.media_path.clone(), local_track) - // }) - // .await??; - // let subtitles = parse_subtitles(&track.codec, cp, blocks)?; - // Ok(subtitles) - // }, - // )spec.track.first().ok_or(anyhow!("no track selected"))?; - // let local_track = local_tracks.first().ok_or(anyhow!("no tracks"))?.clone(); - // let track = &node.media.as_ref().unwrap().tracks[tracki]; - // let cp = local_track.codec_private.clone(); +// let subtitles = async_cache_memory( +// &[ +// "vtt", +// &format!( +// "{} {}", +// local_track.path.to_str().unwrap(), +// local_track.track +// ), +// ], +// move || async move { +// let blocks = tokio::task::spawn_blocking(move || { +// extract_track(CONF.media_path.clone(), local_track) +// }) +// .await??; +// let subtitles = parse_subtitles(&track.codec, cp, blocks)?; +// Ok(subtitles) +// }, +// ) +// .await?; - // let subtitles = async_cache_memory( - // &[ - // "vtt", - // &format!( - // "{} {}", - // local_track.path.to_str().unwrap(), - // local_track.track - // ), - // ], - // move || async move { - // let blocks = tokio::task::spawn_blocking(move || { - // extract_track(CONF.media_path.clone(), local_track) - // }) - // .await??; - // let subtitles = parse_subtitles(&track.codec, cp, blocks)?; - // Ok(subtitles) - // }, - // ) - // .await?; +// let output = if json { +// serde_json::to_string(subtitles.as_ref())? +// } else { +// write_webvtt(node.title.clone().unwrap_or_default(), subtitles.as_ref()) +// .context("writing webvtt")? +// }; +// tokio::task::spawn(async move { +// let _ = b.write_all(output.as_bytes()).await; +// }); +// .await?; - // let output = if json { - // serde_json::to_string(subtitles.as_ref())? - // } else { - // write_webvtt(node.title.clone().unwrap_or_default(), subtitles.as_ref()) - // .context("writing webvtt")? - // }; - // tokio::task::spawn(async move { - // let _ = b.write_all(output.as_bytes()).await; - // }); - // .await?; - - // let output = if json { - // serde_json::to_string(subtitles.as_ref())? - // } else { - // write_webvtt(node.title.clone().unwrap_or_default(), subtitles.as_ref()) - // .context("writing webvtt")? - // }; - // tokio::task::spawn(async move { - // let _ = b.write_all(output.as_bytes()).await; - // }); - Ok(()) -} +// let output = if json { +// serde_json::to_string(subtitles.as_ref())? +// } else { +// write_webvtt(node.title.clone().unwrap_or_default(), subtitles.as_ref()) +// .context("writing webvtt")? +// }; +// tokio::task::spawn(async move { +// let _ = b.write_all(output.as_bytes()).await; +// }); +// Ok(()) +// } |