aboutsummaryrefslogtreecommitdiff
path: root/stream/src/segment.rs
blob: d09d357089a2a2831bfee57dd5d17b1ca41696a6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
/*
    This file is part of jellything (https://codeberg.org/metamuffin/jellything)
    which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
    Copyright (C) 2023 metamuffin <metamuffin.org>
*/
use anyhow::{anyhow, bail, Result};
use jellybase::CONF;
use jellycommon::{stream::StreamSpec, LocalTrack, Node};
use log::warn;
use tokio::io::DuplexStream;
use tokio_util::io::SyncIoBridge;

pub async fn stream_segment(
    node: Node,
    track_sources: Vec<LocalTrack>,
    spec: StreamSpec,
    b: DuplexStream,
) -> Result<()> {
    let b = SyncIoBridge::new(b);

    if spec.tracks.len() != 1 {
        bail!("unsupported number of tracks for segment, must be exactly one");
    }
    let track = spec.tracks[0];
    let n = spec.index.ok_or(anyhow!("segment index missing"))?;

    tokio::task::spawn_blocking(move || {
        if let Err(err) = jellyremuxer::write_snippet_into(
            b,
            CONF.library_path.clone(),
            node.public.clone(),
            track_sources,
            track,
            spec.webm.unwrap_or(false),
            n,
        ) {
            warn!("segment stream error: {err}");
        }
    });

    Ok(())
}