94 lines
2.5 KiB
Rust
94 lines
2.5 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use async_trait::async_trait;
|
|
use rumqttc::Publish;
|
|
use serde::Deserialize;
|
|
use tracing::{debug, warn};
|
|
|
|
use crate::config::MqttDeviceConfig;
|
|
use crate::devices::Device;
|
|
use crate::event::{self, Event, EventChannel, OnMqtt};
|
|
use crate::messages::PresenceMessage;
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
pub struct PresenceConfig {
|
|
#[serde(flatten)]
|
|
pub mqtt: MqttDeviceConfig,
|
|
}
|
|
|
|
pub const DEFAULT_PRESENCE: bool = false;
|
|
|
|
#[derive(Debug)]
|
|
pub struct Presence {
|
|
tx: event::Sender,
|
|
mqtt: MqttDeviceConfig,
|
|
devices: HashMap<String, bool>,
|
|
current_overall_presence: bool,
|
|
}
|
|
|
|
impl Presence {
|
|
pub fn new(config: PresenceConfig, event_channel: &EventChannel) -> Self {
|
|
Self {
|
|
tx: event_channel.get_tx(),
|
|
mqtt: config.mqtt,
|
|
devices: HashMap::new(),
|
|
current_overall_presence: DEFAULT_PRESENCE,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Device for Presence {
|
|
fn get_id(&self) -> &str {
|
|
"presence"
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl OnMqtt for Presence {
|
|
fn topics(&self) -> Vec<&str> {
|
|
vec![&self.mqtt.topic]
|
|
}
|
|
|
|
async fn on_mqtt(&mut self, message: Publish) {
|
|
let offset = self
|
|
.mqtt
|
|
.topic
|
|
.find('+')
|
|
.or(self.mqtt.topic.find('#'))
|
|
.expect("Presence::create fails if it does not contain wildcards");
|
|
let device_name = message.topic[offset..].into();
|
|
|
|
if message.payload.is_empty() {
|
|
// Remove the device from the map
|
|
debug!("State of device [{device_name}] has been removed");
|
|
self.devices.remove(&device_name);
|
|
} else {
|
|
let present = match PresenceMessage::try_from(message) {
|
|
Ok(state) => state.presence(),
|
|
Err(err) => {
|
|
warn!("Failed to parse message: {err}");
|
|
return;
|
|
}
|
|
};
|
|
|
|
debug!("State of device [{device_name}] has changed: {}", present);
|
|
self.devices.insert(device_name, present);
|
|
}
|
|
|
|
let overall_presence = self.devices.iter().any(|(_, v)| *v);
|
|
if overall_presence != self.current_overall_presence {
|
|
debug!("Overall presence updated: {overall_presence}");
|
|
self.current_overall_presence = overall_presence;
|
|
|
|
if self
|
|
.tx
|
|
.send(Event::Presence(overall_presence))
|
|
.await
|
|
.is_err()
|
|
{
|
|
warn!("There are no receivers on the event channel");
|
|
}
|
|
}
|
|
}
|
|
}
|