use crate::tlv;
use anyhow;
use serde_json;
use crate::clusters::helpers::{serialize_opt_bytes_as_hex};
#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
#[repr(u8)]
pub enum Characteristic {
Forcedsubtitles = 0,
Describesvideo = 1,
Easytoread = 2,
Framebased = 3,
Mainprogram = 4,
Originalcontent = 5,
Voiceovertranslation = 6,
Caption = 7,
Subtitle = 8,
Alternate = 9,
Supplementary = 10,
Commentary = 11,
Dubbedtranslation = 12,
Description = 13,
Metadata = 14,
Enhancedaudiointelligibility = 15,
Emergency = 16,
Karaoke = 17,
}
impl Characteristic {
pub fn from_u8(value: u8) -> Option<Self> {
match value {
0 => Some(Characteristic::Forcedsubtitles),
1 => Some(Characteristic::Describesvideo),
2 => Some(Characteristic::Easytoread),
3 => Some(Characteristic::Framebased),
4 => Some(Characteristic::Mainprogram),
5 => Some(Characteristic::Originalcontent),
6 => Some(Characteristic::Voiceovertranslation),
7 => Some(Characteristic::Caption),
8 => Some(Characteristic::Subtitle),
9 => Some(Characteristic::Alternate),
10 => Some(Characteristic::Supplementary),
11 => Some(Characteristic::Commentary),
12 => Some(Characteristic::Dubbedtranslation),
13 => Some(Characteristic::Description),
14 => Some(Characteristic::Metadata),
15 => Some(Characteristic::Enhancedaudiointelligibility),
16 => Some(Characteristic::Emergency),
17 => Some(Characteristic::Karaoke),
_ => None,
}
}
pub fn to_u8(self) -> u8 {
self as u8
}
}
impl From<Characteristic> for u8 {
fn from(val: Characteristic) -> Self {
val as u8
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
#[repr(u8)]
pub enum PlaybackState {
Playing = 0,
Paused = 1,
Notplaying = 2,
Buffering = 3,
}
impl PlaybackState {
pub fn from_u8(value: u8) -> Option<Self> {
match value {
0 => Some(PlaybackState::Playing),
1 => Some(PlaybackState::Paused),
2 => Some(PlaybackState::Notplaying),
3 => Some(PlaybackState::Buffering),
_ => None,
}
}
pub fn to_u8(self) -> u8 {
self as u8
}
}
impl From<PlaybackState> for u8 {
fn from(val: PlaybackState) -> Self {
val as u8
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
#[repr(u8)]
pub enum Status {
Success = 0,
Invalidstateforcommand = 1,
Notallowed = 2,
Notactive = 3,
Speedoutofrange = 4,
Seekoutofrange = 5,
}
impl Status {
pub fn from_u8(value: u8) -> Option<Self> {
match value {
0 => Some(Status::Success),
1 => Some(Status::Invalidstateforcommand),
2 => Some(Status::Notallowed),
3 => Some(Status::Notactive),
4 => Some(Status::Speedoutofrange),
5 => Some(Status::Seekoutofrange),
_ => None,
}
}
pub fn to_u8(self) -> u8 {
self as u8
}
}
impl From<Status> for u8 {
fn from(val: Status) -> Self {
val as u8
}
}
#[derive(Debug, serde::Serialize)]
pub struct PlaybackPosition {
pub updated_at: Option<u64>,
pub position: Option<u64>,
}
#[derive(Debug, serde::Serialize)]
pub struct TrackAttributes {
pub language_code: Option<String>,
pub characteristics: Option<Vec<Characteristic>>,
pub display_name: Option<String>,
}
#[derive(Debug, serde::Serialize)]
pub struct Track {
pub id: Option<String>,
pub track_attributes: Option<TrackAttributes>,
}
pub fn encode_rewind(audio_advance_unmuted: bool) -> anyhow::Result<Vec<u8>> {
let tlv = tlv::TlvItemEnc {
tag: 0,
value: tlv::TlvItemValueEnc::StructInvisible(vec![
(0, tlv::TlvItemValueEnc::Bool(audio_advance_unmuted)).into(),
]),
};
Ok(tlv.encode()?)
}
pub fn encode_fast_forward(audio_advance_unmuted: bool) -> anyhow::Result<Vec<u8>> {
let tlv = tlv::TlvItemEnc {
tag: 0,
value: tlv::TlvItemValueEnc::StructInvisible(vec![
(0, tlv::TlvItemValueEnc::Bool(audio_advance_unmuted)).into(),
]),
};
Ok(tlv.encode()?)
}
pub fn encode_skip_forward(delta_position_milliseconds: u64) -> anyhow::Result<Vec<u8>> {
let tlv = tlv::TlvItemEnc {
tag: 0,
value: tlv::TlvItemValueEnc::StructInvisible(vec![
(0, tlv::TlvItemValueEnc::UInt64(delta_position_milliseconds)).into(),
]),
};
Ok(tlv.encode()?)
}
pub fn encode_skip_backward(delta_position_milliseconds: u64) -> anyhow::Result<Vec<u8>> {
let tlv = tlv::TlvItemEnc {
tag: 0,
value: tlv::TlvItemValueEnc::StructInvisible(vec![
(0, tlv::TlvItemValueEnc::UInt64(delta_position_milliseconds)).into(),
]),
};
Ok(tlv.encode()?)
}
pub fn encode_seek(position: u64) -> anyhow::Result<Vec<u8>> {
let tlv = tlv::TlvItemEnc {
tag: 0,
value: tlv::TlvItemValueEnc::StructInvisible(vec![
(0, tlv::TlvItemValueEnc::UInt64(position)).into(),
]),
};
Ok(tlv.encode()?)
}
pub fn encode_activate_audio_track(track_id: String, audio_output_index: Option<u8>) -> anyhow::Result<Vec<u8>> {
let tlv = tlv::TlvItemEnc {
tag: 0,
value: tlv::TlvItemValueEnc::StructInvisible(vec![
(0, tlv::TlvItemValueEnc::String(track_id)).into(),
(1, tlv::TlvItemValueEnc::UInt8(audio_output_index.unwrap_or(0))).into(),
]),
};
Ok(tlv.encode()?)
}
pub fn encode_activate_text_track(track_id: String) -> anyhow::Result<Vec<u8>> {
let tlv = tlv::TlvItemEnc {
tag: 0,
value: tlv::TlvItemValueEnc::StructInvisible(vec![
(0, tlv::TlvItemValueEnc::String(track_id)).into(),
]),
};
Ok(tlv.encode()?)
}
pub fn decode_current_state(inp: &tlv::TlvItemValue) -> anyhow::Result<PlaybackState> {
if let tlv::TlvItemValue::Int(v) = inp {
PlaybackState::from_u8(*v as u8).ok_or_else(|| anyhow::anyhow!("Invalid enum value"))
} else {
Err(anyhow::anyhow!("Expected Integer"))
}
}
pub fn decode_start_time(inp: &tlv::TlvItemValue) -> anyhow::Result<Option<u64>> {
if let tlv::TlvItemValue::Int(v) = inp {
Ok(Some(*v))
} else {
Ok(None)
}
}
pub fn decode_duration(inp: &tlv::TlvItemValue) -> anyhow::Result<Option<u64>> {
if let tlv::TlvItemValue::Int(v) = inp {
Ok(Some(*v))
} else {
Ok(None)
}
}
pub fn decode_sampled_position(inp: &tlv::TlvItemValue) -> anyhow::Result<Option<PlaybackPosition>> {
if let tlv::TlvItemValue::List(_fields) = inp {
let item = tlv::TlvItem { tag: 0, value: inp.clone() };
Ok(Some(PlaybackPosition {
updated_at: item.get_int(&[0]),
position: item.get_int(&[1]),
}))
} else {
Ok(None)
}
}
pub fn decode_playback_speed(inp: &tlv::TlvItemValue) -> anyhow::Result<u8> {
if let tlv::TlvItemValue::Int(v) = inp {
Ok(*v as u8)
} else {
Err(anyhow::anyhow!("Expected UInt8"))
}
}
pub fn decode_seek_range_end(inp: &tlv::TlvItemValue) -> anyhow::Result<Option<u64>> {
if let tlv::TlvItemValue::Int(v) = inp {
Ok(Some(*v))
} else {
Ok(None)
}
}
pub fn decode_seek_range_start(inp: &tlv::TlvItemValue) -> anyhow::Result<Option<u64>> {
if let tlv::TlvItemValue::Int(v) = inp {
Ok(Some(*v))
} else {
Ok(None)
}
}
pub fn decode_active_audio_track(inp: &tlv::TlvItemValue) -> anyhow::Result<Option<Track>> {
if let tlv::TlvItemValue::List(_fields) = inp {
let item = tlv::TlvItem { tag: 0, value: inp.clone() };
Ok(Some(Track {
id: item.get_string_owned(&[0]),
track_attributes: {
if let Some(nested_tlv) = item.get(&[1]) {
if let tlv::TlvItemValue::List(_) = nested_tlv {
let nested_item = tlv::TlvItem { tag: 1, value: nested_tlv.clone() };
Some(TrackAttributes {
language_code: nested_item.get_string_owned(&[0]),
characteristics: {
if let Some(tlv::TlvItemValue::List(l)) = nested_item.get(&[1]) {
let items: Vec<Characteristic> = l.iter().filter_map(|e| { if let tlv::TlvItemValue::Int(v) = &e.value { Characteristic::from_u8(*v as u8) } else { None } }).collect();
Some(items)
} else {
None
}
},
display_name: nested_item.get_string_owned(&[2]),
})
} else {
None
}
} else {
None
}
},
}))
} else {
Ok(None)
}
}
pub fn decode_available_audio_tracks(inp: &tlv::TlvItemValue) -> anyhow::Result<Vec<Track>> {
let mut res = Vec::new();
if let tlv::TlvItemValue::List(v) = inp {
for item in v {
res.push(Track {
id: item.get_string_owned(&[0]),
track_attributes: {
if let Some(nested_tlv) = item.get(&[1]) {
if let tlv::TlvItemValue::List(_) = nested_tlv {
let nested_item = tlv::TlvItem { tag: 1, value: nested_tlv.clone() };
Some(TrackAttributes {
language_code: nested_item.get_string_owned(&[0]),
characteristics: {
if let Some(tlv::TlvItemValue::List(l)) = nested_item.get(&[1]) {
let items: Vec<Characteristic> = l.iter().filter_map(|e| { if let tlv::TlvItemValue::Int(v) = &e.value { Characteristic::from_u8(*v as u8) } else { None } }).collect();
Some(items)
} else {
None
}
},
display_name: nested_item.get_string_owned(&[2]),
})
} else {
None
}
} else {
None
}
},
});
}
}
Ok(res)
}
pub fn decode_active_text_track(inp: &tlv::TlvItemValue) -> anyhow::Result<Option<Track>> {
if let tlv::TlvItemValue::List(_fields) = inp {
let item = tlv::TlvItem { tag: 0, value: inp.clone() };
Ok(Some(Track {
id: item.get_string_owned(&[0]),
track_attributes: {
if let Some(nested_tlv) = item.get(&[1]) {
if let tlv::TlvItemValue::List(_) = nested_tlv {
let nested_item = tlv::TlvItem { tag: 1, value: nested_tlv.clone() };
Some(TrackAttributes {
language_code: nested_item.get_string_owned(&[0]),
characteristics: {
if let Some(tlv::TlvItemValue::List(l)) = nested_item.get(&[1]) {
let items: Vec<Characteristic> = l.iter().filter_map(|e| { if let tlv::TlvItemValue::Int(v) = &e.value { Characteristic::from_u8(*v as u8) } else { None } }).collect();
Some(items)
} else {
None
}
},
display_name: nested_item.get_string_owned(&[2]),
})
} else {
None
}
} else {
None
}
},
}))
} else {
Ok(None)
}
}
pub fn decode_available_text_tracks(inp: &tlv::TlvItemValue) -> anyhow::Result<Vec<Track>> {
let mut res = Vec::new();
if let tlv::TlvItemValue::List(v) = inp {
for item in v {
res.push(Track {
id: item.get_string_owned(&[0]),
track_attributes: {
if let Some(nested_tlv) = item.get(&[1]) {
if let tlv::TlvItemValue::List(_) = nested_tlv {
let nested_item = tlv::TlvItem { tag: 1, value: nested_tlv.clone() };
Some(TrackAttributes {
language_code: nested_item.get_string_owned(&[0]),
characteristics: {
if let Some(tlv::TlvItemValue::List(l)) = nested_item.get(&[1]) {
let items: Vec<Characteristic> = l.iter().filter_map(|e| { if let tlv::TlvItemValue::Int(v) = &e.value { Characteristic::from_u8(*v as u8) } else { None } }).collect();
Some(items)
} else {
None
}
},
display_name: nested_item.get_string_owned(&[2]),
})
} else {
None
}
} else {
None
}
},
});
}
}
Ok(res)
}
pub fn decode_attribute_json(cluster_id: u32, attribute_id: u32, tlv_value: &crate::tlv::TlvItemValue) -> String {
if cluster_id != 0x0506 {
return format!("{{\"error\": \"Invalid cluster ID. Expected 0x0506, got {}\"}}", cluster_id);
}
match attribute_id {
0x0000 => {
match decode_current_state(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0001 => {
match decode_start_time(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0002 => {
match decode_duration(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0003 => {
match decode_sampled_position(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0004 => {
match decode_playback_speed(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0005 => {
match decode_seek_range_end(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0006 => {
match decode_seek_range_start(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0007 => {
match decode_active_audio_track(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0008 => {
match decode_available_audio_tracks(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0009 => {
match decode_active_text_track(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x000A => {
match decode_available_text_tracks(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
_ => format!("{{\"error\": \"Unknown attribute ID: {}\"}}", attribute_id),
}
}
pub fn get_attribute_list() -> Vec<(u32, &'static str)> {
vec![
(0x0000, "CurrentState"),
(0x0001, "StartTime"),
(0x0002, "Duration"),
(0x0003, "SampledPosition"),
(0x0004, "PlaybackSpeed"),
(0x0005, "SeekRangeEnd"),
(0x0006, "SeekRangeStart"),
(0x0007, "ActiveAudioTrack"),
(0x0008, "AvailableAudioTracks"),
(0x0009, "ActiveTextTrack"),
(0x000A, "AvailableTextTracks"),
]
}
#[derive(Debug, serde::Serialize)]
pub struct PlaybackResponse {
pub status: Option<Status>,
pub data: Option<String>,
}
pub fn decode_playback_response(inp: &tlv::TlvItemValue) -> anyhow::Result<PlaybackResponse> {
if let tlv::TlvItemValue::List(_fields) = inp {
let item = tlv::TlvItem { tag: 0, value: inp.clone() };
Ok(PlaybackResponse {
status: item.get_int(&[0]).and_then(|v| Status::from_u8(v as u8)),
data: item.get_string_owned(&[1]),
})
} else {
Err(anyhow::anyhow!("Expected struct fields"))
}
}
#[derive(Debug, serde::Serialize)]
pub struct StateChangedEvent {
pub current_state: Option<PlaybackState>,
pub start_time: Option<u64>,
pub duration: Option<u64>,
pub sampled_position: Option<PlaybackPosition>,
pub playback_speed: Option<u8>,
pub seek_range_end: Option<u64>,
pub seek_range_start: Option<u64>,
#[serde(serialize_with = "serialize_opt_bytes_as_hex")]
pub data: Option<Vec<u8>>,
pub audio_advance_unmuted: Option<bool>,
}
pub fn decode_state_changed_event(inp: &tlv::TlvItemValue) -> anyhow::Result<StateChangedEvent> {
if let tlv::TlvItemValue::List(_fields) = inp {
let item = tlv::TlvItem { tag: 0, value: inp.clone() };
Ok(StateChangedEvent {
current_state: item.get_int(&[0]).and_then(|v| PlaybackState::from_u8(v as u8)),
start_time: item.get_int(&[1]),
duration: item.get_int(&[2]),
sampled_position: {
if let Some(nested_tlv) = item.get(&[3]) {
if let tlv::TlvItemValue::List(_) = nested_tlv {
let nested_item = tlv::TlvItem { tag: 3, value: nested_tlv.clone() };
Some(PlaybackPosition {
updated_at: nested_item.get_int(&[0]),
position: nested_item.get_int(&[1]),
})
} else {
None
}
} else {
None
}
},
playback_speed: item.get_int(&[4]).map(|v| v as u8),
seek_range_end: item.get_int(&[5]),
seek_range_start: item.get_int(&[6]),
data: item.get_octet_string_owned(&[7]),
audio_advance_unmuted: item.get_bool(&[8]),
})
} else {
Err(anyhow::anyhow!("Expected struct fields"))
}
}