1use crate::draft20::message::ControlMessage;
2use crate::fields::{FieldMap as Map, FieldValue as Value};
3use crate::kvp::{KeyValuePair, KvpValue};
4use crate::range_filter::RangeFilter;
5use crate::types::*;
6use crate::varint::{Moqt18 as Wire, VarInt};
7
8fn vi(v: u64) -> Value {
9 Value::Uint(v)
10}
11
12fn ns_to_json(ns: &TrackNamespace) -> Value {
13 Value::Array(
14 ns.0.iter().map(|e| Value::Text(String::from_utf8_lossy(e).into_owned())).collect(),
15 )
16}
17
18fn d20_param_name(key: u64) -> Option<&'static str> {
22 match key {
23 0x02 => Some("object_delivery_timeout"),
24 0x03 => Some("authorization_token"),
25 0x04 => Some("rendezvous_timeout"),
26 0x06 => Some("subgroup_delivery_timeout"),
27 0x08 => Some("expires"),
28 0x09 => Some("largest_object"),
29 0x0A => Some("fill_timeout"),
30 0x10 => Some("forward"),
31 0x20 => Some("subscriber_priority"),
32 0x21 => Some("location_filter"),
33 0x22 => Some("group_order"),
34 0x23 => Some("fill_parameters"),
35 0x25 => Some("subgroup_filter"),
36 0x26 => Some("objectid_filter"),
37 0x27 => Some("priority_filter"),
38 0x28 => Some("object_property_filter"),
39 0x29 => Some("track_property_filter"),
40 0x32 => Some("new_group_request"),
41 0x34 => Some("track_namespace_prefix"),
42 0x35 => Some("include_properties"),
43 _ => None,
44 }
45}
46
47fn d20_option_name(key: u64) -> Option<&'static str> {
49 match key {
50 0x01 => Some("path"),
51 0x03 => Some("authorization_token"),
52 0x04 => Some("max_auth_token_cache_size"),
53 0x05 => Some("authority"),
54 0x06 => Some("max_filter_ranges"),
55 0x07 => Some("moqt_implementation"),
56 0x08 => Some("max_request_updates"),
57 _ => None,
58 }
59}
60
61fn decode_range_filter(bytes: &[u8], parameter_type: u64) -> Value {
109 let mut o = Map::new();
110 if bytes.is_empty() {
111 o.insert("removed".into(), Value::Bool(true));
112 return Value::Map(o);
113 }
114 let Ok(filter) = RangeFilter::decode_moqt_structure::<Wire>(parameter_type, bytes) else {
115 return Value::Bytes(bytes.to_vec());
116 };
117 if let Err(broken) = filter.check_its_own_types() {
118 o.insert("violates".into(), Value::Text(broken.to_string()));
119 }
120 o.insert("set_id".into(), vi(filter.set_id as u64));
121 if let Some(property_type) = filter.property_type {
122 o.insert("property_type".into(), vi(property_type));
123 }
124 let ranges = filter
125 .ranges
126 .iter()
127 .map(|range| {
128 let mut r = Map::new();
129 r.insert("start".into(), vi(range.start));
130 if let Some(end) = range.end {
131 r.insert("end".into(), vi(end));
132 }
133 Value::Map(r)
134 })
135 .collect();
136 o.insert("ranges".into(), Value::Array(ranges));
137 Value::Map(o)
138}
139
140fn decode_location_filter(bytes: &[u8]) -> Value {
152 const NAMES: [&str; 4] = ["start_group", "start_object", "end_group_delta", "end_object"];
153 let fields = crate::draft20::message::decode_location_filter(bytes)
154 .expect("the decoder accepted this parameter, so its value parses");
155 let mut obj = Map::new();
156 if fields.is_empty() {
157 obj.insert("removed".into(), Value::Bool(true));
158 return Value::Map(obj);
159 }
160 for (name, value) in NAMES.iter().zip(fields) {
161 obj.insert((*name).to_string(), vi(value));
162 }
163 Value::Map(obj)
164}
165
166fn decode_fill_parameters(bytes: &[u8]) -> Value {
173 let nested = crate::draft20::message::decode_fill_parameters(bytes)
174 .expect("the decoder accepted this parameter, so its value parses");
175 params_to_json(&nested)
176}
177
178fn auth_token_to_json_d20(bytes: &[u8]) -> Value {
179 let mut buf = bytes;
180 let alias_type = match VarInt::decode_moqt::<Wire>(&mut buf) {
181 Ok(v) => v,
182 Err(_) => return Value::Bytes(bytes.to_vec()),
183 };
184 let at = alias_type.into_inner();
185 let mut o = Map::new();
186 o.insert("alias_type".into(), vi(at));
187 match at {
188 0 | 2 => {
189 if let Ok(ta) = VarInt::decode_moqt::<Wire>(&mut buf) {
190 o.insert("token_alias".into(), vi(ta.into_inner()));
191 }
192 }
193 1 => {
194 if let Ok(ta) = VarInt::decode_moqt::<Wire>(&mut buf) {
195 o.insert("token_alias".into(), vi(ta.into_inner()));
196 }
197 if let Ok(tt) = VarInt::decode_moqt::<Wire>(&mut buf) {
198 o.insert("token_type".into(), vi(tt.into_inner()));
199 }
200 o.insert("token_value".into(), Value::Bytes(buf.to_vec()));
203 }
204 _ => {
205 if let Ok(tt) = VarInt::decode_moqt::<Wire>(&mut buf) {
206 o.insert("token_type".into(), vi(tt.into_inner()));
207 }
208 o.insert("token_value".into(), Value::Bytes(buf.to_vec()));
209 }
210 }
211 Value::Map(o)
212}
213
214fn decode_largest_object(bytes: &[u8]) -> Value {
227 let mut buf = bytes;
228 let Ok(group) = VarInt::decode_moqt::<Wire>(&mut buf) else {
229 return Value::Bytes(bytes.to_vec());
230 };
231 let Ok(object) = VarInt::decode_moqt::<Wire>(&mut buf) else {
232 return Value::Bytes(bytes.to_vec());
233 };
234 let (group, object) = (group.into_inner(), object.into_inner());
235 let mut obj = Map::new();
236 obj.insert("group".into(), vi(group));
237 obj.insert("object".into(), vi(object));
238 Value::Map(obj)
239}
240
241fn decode_track_namespace_prefix(bytes: &[u8]) -> Value {
242 let mut buf = bytes;
243 match TrackNamespace::decode_allow_empty_moqt::<Wire>(&mut buf) {
244 Ok(ns) => ns_to_json(&ns),
245 Err(_) => Value::Bytes(bytes.to_vec()),
246 }
247}
248
249fn params_to_json(params: &[KeyValuePair]) -> Value {
250 crate::fields::kvp_entries(params, |key, value| {
251 let Some(name) = d20_param_name(key) else {
252 return (None, None);
253 };
254 let rendered = match (value, key) {
255 (KvpValue::Bytes(b), 0x21) => decode_location_filter(b),
256 (KvpValue::Bytes(b), 0x23) => decode_fill_parameters(b),
257 (KvpValue::Bytes(b), 0x25..=0x29) => decode_range_filter(b, key),
262 (KvpValue::Bytes(b), 0x09) => decode_largest_object(b),
263 (KvpValue::Bytes(b), 0x34) => decode_track_namespace_prefix(b),
264 (KvpValue::Bytes(b), _) if name == "authorization_token" => auth_token_to_json_d20(b),
265 (KvpValue::Varint(v), _) => vi(v.into_inner()),
266 (KvpValue::Bytes(b), _) => Value::Text(String::from_utf8_lossy(b).into_owned()),
267 };
268 (Some(name), Some(rendered))
269 })
270}
271
272pub(crate) fn options_to_json(options: &[KeyValuePair]) -> Value {
273 crate::fields::kvp_entries(options, |key, value| {
274 let Some(name) = d20_option_name(key) else {
275 return (None, None);
276 };
277 let rendered = match value {
278 KvpValue::Varint(v) => vi(v.into_inner()),
279 KvpValue::Bytes(b) if name == "authorization_token" => auth_token_to_json_d20(b),
280 KvpValue::Bytes(b) => Value::Text(String::from_utf8_lossy(b).into_owned()),
281 };
282 (Some(name), Some(rendered))
283 })
284}
285
286fn d20_track_prop_name(key: u64) -> Option<&'static str> {
287 match key {
288 0x02 => Some("object_delivery_timeout"),
289 0x04 => Some("max_cache_duration"),
290 0x06 => Some("subgroup_delivery_timeout"),
291 0x0b => Some("immutable_properties"),
292 0x0e => Some("default_publisher_priority"),
293 0x22 => Some("default_publisher_group_order"),
294 0x30 => Some("dynamic_groups"),
295 _ => None,
296 }
297}
298
299fn track_props_to_json(props: &[KeyValuePair]) -> Value {
300 crate::fields::kvp_entries(props, |key, value| {
301 let name = d20_track_prop_name(key);
302 let rendered = match value {
303 KvpValue::Varint(v) => Some(vi(v.into_inner())),
304 KvpValue::Bytes(_) => None,
308 };
309 (name, rendered)
310 })
311}
312
313pub fn message_fields(msg: &ControlMessage) -> Map {
319 let obj = match msg {
320 ControlMessage::Setup(m) => {
321 let mut o = Map::new();
322 o.insert("options".into(), options_to_json(&m.options));
323 o
324 }
325 ControlMessage::GoAway(m) => {
326 let mut o = Map::new();
327 o.insert(
328 "new_session_uri".into(),
329 Value::Text(String::from_utf8_lossy(&m.new_session_uri).into_owned()),
330 );
331 o.insert("timeout".into(), vi(m.timeout.into_inner()));
332 o
333 }
334 ControlMessage::RequestOk(m) => {
335 let mut o = Map::new();
336 o.insert("parameters".into(), params_to_json(&m.parameters));
337 o.insert("track_properties".into(), track_props_to_json(&m.track_properties));
338 o
339 }
340 ControlMessage::RequestError(m) => {
341 let mut o = Map::new();
342 o.insert("error_code".into(), vi(m.error_code.into_inner()));
343 o.insert("retry_interval".into(), vi(m.retry_interval.into_inner()));
344 o.insert(
345 "reason_phrase".into(),
346 Value::Text(String::from_utf8_lossy(&m.reason_phrase).into_owned()),
347 );
348 if let Some(r) = &m.redirect {
349 let mut r_obj = Map::new();
350 r_obj.insert(
351 "connect_uri".into(),
352 Value::Text(String::from_utf8_lossy(&r.connect_uri).into_owned()),
353 );
354 r_obj.insert("track_namespace".into(), ns_to_json(&r.track_namespace));
355 r_obj.insert(
356 "track_name".into(),
357 Value::Text(String::from_utf8_lossy(&r.track_name).into_owned()),
358 );
359 o.insert("redirect".into(), Value::Map(r_obj));
360 }
361 o
362 }
363 ControlMessage::Subscribe(m) => {
364 let mut o = Map::new();
365 o.insert("request_id".into(), vi(m.request_id.into_inner()));
366 o.insert("track_namespace".into(), ns_to_json(&m.track_namespace));
367 o.insert(
368 "track_name".into(),
369 Value::Text(String::from_utf8_lossy(&m.track_name).into_owned()),
370 );
371 o.insert("parameters".into(), params_to_json(&m.parameters));
372 o
373 }
374 ControlMessage::SubscribeOk(m) => {
375 let mut o = Map::new();
376 o.insert("track_alias".into(), vi(m.track_alias.into_inner()));
377 o.insert("parameters".into(), params_to_json(&m.parameters));
378 o.insert("track_properties".into(), track_props_to_json(&m.track_properties));
379 o
380 }
381 ControlMessage::RequestUpdate(m) => {
382 let mut o = Map::new();
383 o.insert("request_id".into(), vi(m.request_id.into_inner()));
384 o.insert("parameters".into(), params_to_json(&m.parameters));
385 o
386 }
387 ControlMessage::Publish(m) => {
388 let mut o = Map::new();
389 o.insert("request_id".into(), vi(m.request_id.into_inner()));
390 o.insert("track_namespace".into(), ns_to_json(&m.track_namespace));
391 o.insert(
392 "track_name".into(),
393 Value::Text(String::from_utf8_lossy(&m.track_name).into_owned()),
394 );
395 o.insert("track_alias".into(), vi(m.track_alias.into_inner()));
396 o.insert("parameters".into(), params_to_json(&m.parameters));
397 o.insert("track_properties".into(), track_props_to_json(&m.track_properties));
398 o
399 }
400 ControlMessage::PublishStateNotify(m) => {
403 let mut o = Map::new();
404 o.insert("parameters".into(), params_to_json(&m.parameters));
405 o
406 }
407 ControlMessage::PublishDone(m) => {
408 let mut o = Map::new();
409 o.insert("status_code".into(), vi(m.status_code.into_inner()));
410 o.insert("stream_count".into(), vi(m.stream_count.into_inner()));
411 o.insert(
412 "reason_phrase".into(),
413 Value::Text(String::from_utf8_lossy(&m.reason_phrase).into_owned()),
414 );
415 o
416 }
417 ControlMessage::PublishNamespace(m) => {
418 let mut o = Map::new();
419 o.insert("request_id".into(), vi(m.request_id.into_inner()));
420 o.insert("track_namespace".into(), ns_to_json(&m.track_namespace));
421 o.insert("parameters".into(), params_to_json(&m.parameters));
422 o
423 }
424 ControlMessage::Namespace(m) => {
425 let mut o = Map::new();
426 o.insert("namespace_suffix".into(), ns_to_json(&m.namespace_suffix));
427 o
428 }
429 ControlMessage::NamespaceDone(m) => {
430 let mut o = Map::new();
431 o.insert("namespace_suffix".into(), ns_to_json(&m.namespace_suffix));
432 o
433 }
434 ControlMessage::SubscribeNamespace(m) => {
435 let mut o = Map::new();
436 o.insert("request_id".into(), vi(m.request_id.into_inner()));
437 o.insert("namespace_prefix".into(), ns_to_json(&m.namespace_prefix));
438 o.insert("parameters".into(), params_to_json(&m.parameters));
439 o
440 }
441 ControlMessage::SubscribeTracks(m) => {
442 let mut o = Map::new();
443 o.insert("request_id".into(), vi(m.request_id.into_inner()));
444 o.insert("namespace_prefix".into(), ns_to_json(&m.namespace_prefix));
445 o.insert("parameters".into(), params_to_json(&m.parameters));
446 o
447 }
448 ControlMessage::TrackStatus(m) => {
449 let mut o = Map::new();
450 o.insert("request_id".into(), vi(m.request_id.into_inner()));
451 o.insert("track_namespace".into(), ns_to_json(&m.track_namespace));
452 o.insert(
453 "track_name".into(),
454 Value::Text(String::from_utf8_lossy(&m.track_name).into_owned()),
455 );
456 o.insert("parameters".into(), params_to_json(&m.parameters));
457 o
458 }
459 ControlMessage::Fetch(m) => {
462 let mut o = Map::new();
463 o.insert("request_id".into(), vi(m.request_id.into_inner()));
464 o.insert("track_namespace".into(), ns_to_json(&m.track_namespace));
465 o.insert(
466 "track_name".into(),
467 Value::Text(String::from_utf8_lossy(&m.track_name).into_owned()),
468 );
469 o.insert("parameters".into(), params_to_json(&m.parameters));
470 o
471 }
472 ControlMessage::FetchOk(m) => {
473 let mut o = Map::new();
474 o.insert("end_of_track".into(), vi(m.end_of_track as u64));
475 o.insert("end_group".into(), vi(m.end_group.into_inner()));
476 o.insert("end_object".into(), vi(m.end_object.into_inner()));
477 o.insert("parameters".into(), params_to_json(&m.parameters));
478 o.insert("track_properties".into(), track_props_to_json(&m.track_properties));
479 o
480 }
481 ControlMessage::PublishSkipped(m) => {
482 let mut o = Map::new();
483 o.insert("namespace_suffix".into(), ns_to_json(&m.namespace_suffix));
484 o.insert(
485 "track_name".into(),
486 Value::Text(String::from_utf8_lossy(&m.track_name).into_owned()),
487 );
488 o
489 }
490 };
491 obj
492}