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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
/*
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) 2024 metamuffin <metamuffin.org>
*/
use anyhow::{anyhow, bail, Context, Result};
use jellybase::CONF;
use jellycommon::{jhls::SubtitleCue, stream::StreamSpec, LocalTrack, Node};
use jellyremuxer::extract::extract_track;
use jellytranscoder::subtitles::{
parse_ass_blocks, parse_pgs_blocks, parse_webvtt_blocks, write_webvtt,
};
use tokio::io::{AsyncWriteExt, DuplexStream};
pub async fn vtt_stream(
json: bool,
node: Node,
local_tracks: Vec<LocalTrack>,
spec: StreamSpec,
mut b: DuplexStream,
) -> Result<()> {
// TODO cache
// TODO should use snippets too? big films take too long...
let tracki = *spec.tracks.get(0).ok_or(anyhow!("no track selected"))?;
let local_track = local_tracks.get(0).ok_or(anyhow!("no tracks"))?.clone();
let track = &node.public.media.unwrap().tracks[tracki];
let write = |blocks: Vec<SubtitleCue>| -> anyhow::Result<()> {
let output = if json {
serde_json::to_string(&blocks)?
} else {
write_webvtt(node.public.title.clone().unwrap_or_default(), blocks)
.context("writing webvtt")?
};
tokio::task::spawn(async move {
let _ = b.write_all(output.as_bytes()).await;
});
Ok(())
};
match track.codec.as_str() {
"D_WEBVTT/SUBTITLES" => {
let webvtt_blocks = tokio::task::spawn_blocking(move || {
extract_track(CONF.media_path.clone(), local_track)
})
.await??;
let subtitles = parse_webvtt_blocks(webvtt_blocks).context("parsing subtitles")?;
write(subtitles)?;
}
"S_HDMV/PGS" => {
let webvtt_blocks = tokio::task::spawn_blocking(move || {
extract_track(CONF.media_path.clone(), local_track)
})
.await??;
let subtitles = parse_pgs_blocks(webvtt_blocks).context("parsing subtitles")?;
write(subtitles)?;
}
"S_HDMV/TEXTST" => bail!("no HDMV/PGSs yet"),
"S_ARISUB" => bail!("no arisub yet"),
"S_TEXT/UTF8" => bail!("no subrip yet"),
"S_VOBSUB" => bail!("no vobsub yet"),
"S_TEXT/ASS" => {
let codec_private = local_track
.codec_private
.clone()
.ok_or(anyhow!("ASS is missing required codec private data"))?;
let ass_blocks = tokio::task::spawn_blocking(move || {
extract_track(CONF.media_path.clone(), local_track)
})
.await??;
let subtitles =
parse_ass_blocks(codec_private, ass_blocks).context("parsing subtitles")?;
write(subtitles)?;
}
x => bail!("unknown sub codec {x:?}"),
};
Ok(())
}
|
