89 lines
2.1 KiB
Rust
89 lines
2.1 KiB
Rust
use async_trait::async_trait;
|
|
use rumqttc::AsyncClient;
|
|
use serde::Deserialize;
|
|
use tracing::warn;
|
|
|
|
use crate::{
|
|
config::MqttDeviceConfig,
|
|
devices::Device,
|
|
mqtt::{DarknessMessage, PresenceMessage},
|
|
traits::OnDarkness,
|
|
traits::OnPresence,
|
|
};
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
pub struct DebugBridgeConfig {
|
|
#[serde(flatten)]
|
|
pub mqtt: MqttDeviceConfig,
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct DebugBridge {
|
|
mqtt: MqttDeviceConfig,
|
|
client: AsyncClient,
|
|
}
|
|
|
|
impl DebugBridge {
|
|
pub fn new(
|
|
config: DebugBridgeConfig,
|
|
client: &AsyncClient,
|
|
) -> Result<Self, crate::error::CreateDeviceError> {
|
|
Ok(Self {
|
|
mqtt: config.mqtt,
|
|
client: client.clone(),
|
|
})
|
|
}
|
|
}
|
|
|
|
impl Device for DebugBridge {
|
|
fn get_id(&self) -> &str {
|
|
"debug_bridge"
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl OnPresence for DebugBridge {
|
|
async fn on_presence(&mut self, presence: bool) {
|
|
let message = PresenceMessage::new(presence);
|
|
let topic = format!("{}/presence", self.mqtt.topic);
|
|
self.client
|
|
.publish(
|
|
topic,
|
|
rumqttc::QoS::AtLeastOnce,
|
|
true,
|
|
serde_json::to_string(&message).unwrap(),
|
|
)
|
|
.await
|
|
.map_err(|err| {
|
|
warn!(
|
|
"Failed to update presence on {}/presence: {err}",
|
|
self.mqtt.topic
|
|
)
|
|
})
|
|
.ok();
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl OnDarkness for DebugBridge {
|
|
async fn on_darkness(&mut self, dark: bool) {
|
|
let message = DarknessMessage::new(dark);
|
|
let topic = format!("{}/darkness", self.mqtt.topic);
|
|
self.client
|
|
.publish(
|
|
topic,
|
|
rumqttc::QoS::AtLeastOnce,
|
|
true,
|
|
serde_json::to_string(&message).unwrap(),
|
|
)
|
|
.await
|
|
.map_err(|err| {
|
|
warn!(
|
|
"Failed to update presence on {}/presence: {err}",
|
|
self.mqtt.topic
|
|
)
|
|
})
|
|
.ok();
|
|
}
|
|
}
|