#![allow(clippy::too_many_arguments)]
use crate::tlv;
use anyhow;
use serde_json;
#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
#[repr(u8)]
pub enum EffectIdentifier {
Blink = 0,
Breathe = 1,
Okay = 2,
Channelchange = 11,
Finisheffect = 254,
Stopeffect = 255,
}
impl EffectIdentifier {
pub fn from_u8(value: u8) -> Option<Self> {
match value {
0 => Some(EffectIdentifier::Blink),
1 => Some(EffectIdentifier::Breathe),
2 => Some(EffectIdentifier::Okay),
11 => Some(EffectIdentifier::Channelchange),
254 => Some(EffectIdentifier::Finisheffect),
255 => Some(EffectIdentifier::Stopeffect),
_ => None,
}
}
pub fn to_u8(self) -> u8 {
self as u8
}
}
impl From<EffectIdentifier> for u8 {
fn from(val: EffectIdentifier) -> Self {
val as u8
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
#[repr(u8)]
pub enum EffectVariant {
Default = 0,
}
impl EffectVariant {
pub fn from_u8(value: u8) -> Option<Self> {
match value {
0 => Some(EffectVariant::Default),
_ => None,
}
}
pub fn to_u8(self) -> u8 {
self as u8
}
}
impl From<EffectVariant> for u8 {
fn from(val: EffectVariant) -> Self {
val as u8
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
#[repr(u8)]
pub enum IdentifyType {
None = 0,
Lightoutput = 1,
Visibleindicator = 2,
Audiblebeep = 3,
Display = 4,
Actuator = 5,
}
impl IdentifyType {
pub fn from_u8(value: u8) -> Option<Self> {
match value {
0 => Some(IdentifyType::None),
1 => Some(IdentifyType::Lightoutput),
2 => Some(IdentifyType::Visibleindicator),
3 => Some(IdentifyType::Audiblebeep),
4 => Some(IdentifyType::Display),
5 => Some(IdentifyType::Actuator),
_ => None,
}
}
pub fn to_u8(self) -> u8 {
self as u8
}
}
impl From<IdentifyType> for u8 {
fn from(val: IdentifyType) -> Self {
val as u8
}
}
pub fn encode_identify(identify_time: u16) -> anyhow::Result<Vec<u8>> {
let tlv = tlv::TlvItemEnc {
tag: 0,
value: tlv::TlvItemValueEnc::StructInvisible(vec![
(0, tlv::TlvItemValueEnc::UInt16(identify_time)).into(),
]),
};
Ok(tlv.encode()?)
}
pub fn encode_trigger_effect(effect_identifier: EffectIdentifier, effect_variant: EffectVariant) -> anyhow::Result<Vec<u8>> {
let tlv = tlv::TlvItemEnc {
tag: 0,
value: tlv::TlvItemValueEnc::StructInvisible(vec![
(0, tlv::TlvItemValueEnc::UInt8(effect_identifier.to_u8())).into(),
(1, tlv::TlvItemValueEnc::UInt8(effect_variant.to_u8())).into(),
]),
};
Ok(tlv.encode()?)
}
pub fn decode_identify_time(inp: &tlv::TlvItemValue) -> anyhow::Result<u16> {
if let tlv::TlvItemValue::Int(v) = inp {
Ok(*v as u16)
} else {
Err(anyhow::anyhow!("Expected UInt16"))
}
}
pub fn decode_identify_type(inp: &tlv::TlvItemValue) -> anyhow::Result<IdentifyType> {
if let tlv::TlvItemValue::Int(v) = inp {
IdentifyType::from_u8(*v as u8).ok_or_else(|| anyhow::anyhow!("Invalid enum value"))
} else {
Err(anyhow::anyhow!("Expected Integer"))
}
}
pub fn decode_attribute_json(cluster_id: u32, attribute_id: u32, tlv_value: &crate::tlv::TlvItemValue) -> String {
if cluster_id != 0x0003 {
return format!("{{\"error\": \"Invalid cluster ID. Expected 0x0003, got {}\"}}", cluster_id);
}
match attribute_id {
0x0000 => {
match decode_identify_time(tlv_value) {
Ok(value) => serde_json::to_string(&value).unwrap_or_else(|_| "null".to_string()),
Err(e) => format!("{{\"error\": \"{}\"}}", e),
}
}
0x0001 => {
match decode_identify_type(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, "IdentifyTime"),
(0x0001, "IdentifyType"),
]
}
pub fn get_command_list() -> Vec<(u32, &'static str)> {
vec![
(0x00, "Identify"),
(0x40, "TriggerEffect"),
]
}
pub fn get_command_name(cmd_id: u32) -> Option<&'static str> {
match cmd_id {
0x00 => Some("Identify"),
0x40 => Some("TriggerEffect"),
_ => None,
}
}
pub fn get_command_schema(cmd_id: u32) -> Option<Vec<crate::clusters::codec::CommandField>> {
match cmd_id {
0x00 => Some(vec![
crate::clusters::codec::CommandField { tag: 0, name: "identify_time", kind: crate::clusters::codec::FieldKind::U16, optional: false, nullable: false },
]),
0x40 => Some(vec![
crate::clusters::codec::CommandField { tag: 0, name: "effect_identifier", kind: crate::clusters::codec::FieldKind::Enum { name: "EffectIdentifier", variants: &[(0, "Blink"), (1, "Breathe"), (2, "Okay"), (11, "Channelchange"), (254, "Finisheffect"), (255, "Stopeffect")] }, optional: false, nullable: false },
crate::clusters::codec::CommandField { tag: 1, name: "effect_variant", kind: crate::clusters::codec::FieldKind::Enum { name: "EffectVariant", variants: &[(0, "Default")] }, optional: false, nullable: false },
]),
_ => None,
}
}
pub fn encode_command_json(cmd_id: u32, args: &serde_json::Value) -> anyhow::Result<Vec<u8>> {
match cmd_id {
0x00 => {
let identify_time = crate::clusters::codec::json_util::get_u16(args, "identify_time")?;
encode_identify(identify_time)
}
0x40 => {
let effect_identifier = {
let n = crate::clusters::codec::json_util::get_u64(args, "effect_identifier")?;
EffectIdentifier::from_u8(n as u8).ok_or_else(|| anyhow::anyhow!("invalid EffectIdentifier: {}", n))?
};
let effect_variant = {
let n = crate::clusters::codec::json_util::get_u64(args, "effect_variant")?;
EffectVariant::from_u8(n as u8).ok_or_else(|| anyhow::anyhow!("invalid EffectVariant: {}", n))?
};
encode_trigger_effect(effect_identifier, effect_variant)
}
_ => Err(anyhow::anyhow!("unknown command ID: 0x{:02X}", cmd_id)),
}
}
pub async fn identify(conn: &crate::controller::Connection, endpoint: u16, identify_time: u16) -> anyhow::Result<()> {
conn.invoke_request(endpoint, crate::clusters::defs::CLUSTER_ID_IDENTIFY, crate::clusters::defs::CLUSTER_IDENTIFY_CMD_ID_IDENTIFY, &encode_identify(identify_time)?).await?;
Ok(())
}
pub async fn trigger_effect(conn: &crate::controller::Connection, endpoint: u16, effect_identifier: EffectIdentifier, effect_variant: EffectVariant) -> anyhow::Result<()> {
conn.invoke_request(endpoint, crate::clusters::defs::CLUSTER_ID_IDENTIFY, crate::clusters::defs::CLUSTER_IDENTIFY_CMD_ID_TRIGGEREFFECT, &encode_trigger_effect(effect_identifier, effect_variant)?).await?;
Ok(())
}
pub async fn read_identify_time(conn: &crate::controller::Connection, endpoint: u16) -> anyhow::Result<u16> {
let tlv = conn.read_request2(endpoint, crate::clusters::defs::CLUSTER_ID_IDENTIFY, crate::clusters::defs::CLUSTER_IDENTIFY_ATTR_ID_IDENTIFYTIME).await?;
decode_identify_time(&tlv)
}
pub async fn read_identify_type(conn: &crate::controller::Connection, endpoint: u16) -> anyhow::Result<IdentifyType> {
let tlv = conn.read_request2(endpoint, crate::clusters::defs::CLUSTER_ID_IDENTIFY, crate::clusters::defs::CLUSTER_IDENTIFY_ATTR_ID_IDENTIFYTYPE).await?;
decode_identify_type(&tlv)
}