moq-karp 0.4.0

Media over QUIC
Documentation 
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

use moq_transfork::{coding::*, Path, Session};

use super::{Error, Timestamp};
use crate::catalog;

pub struct BroadcastProducer {
	catalog: catalog::Broadcast,
	catalog_track: Option<moq_transfork::TrackProducer>,
	session: Session,
	path: Path,
}

impl BroadcastProducer {
	pub fn new(session: Session, path: Path) -> Self {
		Self {
			session,
			path,
			catalog: catalog::Broadcast::default(),
			catalog_track: None,
		}
	}

	pub fn create_video(&mut self, info: catalog::Video) -> Result<TrackProducer, Error> {
		let (producer, consumer) = moq_transfork::Track {
			path: self.path.clone().push(&info.track.name),
			priority: info.track.priority,
			..Default::default()
		}
		.produce();

		self.session.publish(consumer)?;
		let track = TrackProducer::new(producer);

		self.catalog.video.push(info);
		Ok(track)
	}

	pub fn create_audio(&mut self, info: catalog::Audio) -> Result<TrackProducer, Error> {
		let (producer, consumer) = moq_transfork::Track {
			path: self.path.clone().push(&info.track.name),
			priority: info.track.priority,
			..Default::default()
		}
		.produce();

		self.session.publish(consumer)?;
		let track = TrackProducer::new(producer);

		self.catalog.audio.push(info);
		Ok(track)
	}

	pub fn catalog(&self) -> &catalog::Broadcast {
		&self.catalog
	}

	pub fn publish(&mut self) -> Result<(), Error> {
		if let Some(track) = self.catalog_track.as_mut() {
			return Ok(self.catalog.update(track)?);
		}

		let path = self.path.clone().push("catalog.json");
		self.catalog_track = self.catalog.publish(&mut self.session, path)?.into();

		Ok(())
	}

	pub async fn closed(&self) {
		self.session.closed().await;
	}
}

pub struct TrackProducer {
	inner: moq_transfork::TrackProducer,
	group: Option<moq_transfork::GroupProducer>,
}

impl TrackProducer {
	fn new(inner: moq_transfork::TrackProducer) -> Self {
		Self { inner, group: None }
	}

	pub fn keyframe(&mut self) {
		// The take() is important, it means we'll create a new group on the next write.
		if let Some(group) = self.group.take() {
			tracing::debug!(sequence = group.sequence, frames = group.frame_count(), "keyframe");
		}
	}

	pub fn write(&mut self, timestamp: Timestamp, payload: Bytes) {
		let timestamp = timestamp.as_micros();
		let mut header = BytesMut::with_capacity(timestamp.encode_size());
		timestamp.encode(&mut header);

		let mut group = match self.group.take() {
			Some(group) => group,
			None => self.inner.append_group(),
		};

		let mut frame = group.create_frame(header.len() + payload.len());
		frame.write(header.freeze());
		frame.write(payload);

		self.group.replace(group);
	}
}