101 lines
3.1 KiB
Rust
101 lines
3.1 KiB
Rust
use std::net::Ipv4Addr;
|
|
|
|
use async_trait::async_trait;
|
|
use google_home::{GoogleHomeDevice, types::Type, device, traits::{self, Scene}, errors::ErrorCode};
|
|
use tracing::{debug, error};
|
|
use rumqttc::{AsyncClient, Publish, matches};
|
|
use eui48::MacAddress;
|
|
|
|
use crate::{config::{InfoConfig, MqttDeviceConfig}, mqtt::{OnMqtt, ActivateMessage}, error::DeviceError};
|
|
|
|
use super::Device;
|
|
|
|
#[derive(Debug)]
|
|
pub struct WakeOnLAN {
|
|
identifier: String,
|
|
info: InfoConfig,
|
|
mqtt: MqttDeviceConfig,
|
|
mac_address: MacAddress,
|
|
broadcast_ip: Ipv4Addr,
|
|
}
|
|
|
|
impl WakeOnLAN {
|
|
pub async fn build(identifier: &str, info: InfoConfig, mqtt: MqttDeviceConfig, mac_address: MacAddress, broadcast_ip: Ipv4Addr, client: AsyncClient) -> Result<Self, DeviceError> {
|
|
// @TODO Handle potential errors here
|
|
client.subscribe(mqtt.topic.clone(), rumqttc::QoS::AtLeastOnce).await?;
|
|
|
|
Ok(Self { identifier: identifier.to_owned(), info, mqtt, mac_address, broadcast_ip })
|
|
}
|
|
}
|
|
|
|
impl Device for WakeOnLAN {
|
|
fn get_id(&self) -> &str {
|
|
&self.identifier
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl OnMqtt for WakeOnLAN {
|
|
async fn on_mqtt(&mut self, message: &Publish) {
|
|
if !matches(&message.topic, &self.mqtt.topic) {
|
|
return;
|
|
}
|
|
|
|
let activate = match ActivateMessage::try_from(message) {
|
|
Ok(message) => message.activate(),
|
|
Err(err) => {
|
|
error!(id = self.identifier, "Failed to parse message: {err}");
|
|
return;
|
|
}
|
|
};
|
|
|
|
self.set_active(activate).ok();
|
|
}
|
|
}
|
|
|
|
impl GoogleHomeDevice for WakeOnLAN {
|
|
fn get_device_type(&self) -> Type {
|
|
Type::Scene
|
|
}
|
|
|
|
fn get_device_name(&self) -> device::Name {
|
|
let mut name = device::Name::new(&self.info.name);
|
|
name.add_default_name("Computer");
|
|
|
|
return name;
|
|
}
|
|
|
|
fn get_id(&self) -> &str {
|
|
Device::get_id(self)
|
|
}
|
|
|
|
fn is_online(&self) -> bool {
|
|
true
|
|
}
|
|
|
|
fn get_room_hint(&self) -> Option<&str> {
|
|
self.info.room.as_deref()
|
|
}
|
|
}
|
|
|
|
impl traits::Scene for WakeOnLAN {
|
|
fn set_active(&self, activate: bool) -> Result<(), ErrorCode> {
|
|
if activate {
|
|
debug!(id = self.identifier, "Activating Computer: {} (Sending to {})", self.mac_address, self.broadcast_ip);
|
|
let wol = wakey::WolPacket::from_bytes(&self.mac_address.to_array()).map_err(|err| {
|
|
error!(id = self.identifier, "invalid mac address: {err}");
|
|
google_home::errors::DeviceError::TransientError
|
|
})?;
|
|
|
|
wol.send_magic_to((Ipv4Addr::new(0, 0, 0, 0), 0), (self.broadcast_ip, 9)).map_err(|err| {
|
|
error!(id = self.identifier, "Failed to activate computer: {err}");
|
|
google_home::errors::DeviceError::TransientError.into()
|
|
}).map(|_| debug!(id = self.identifier, "Success!"))
|
|
} else {
|
|
debug!(id = self.identifier, "Trying to deactive computer, this is not currently supported");
|
|
// We do not support deactivating this scene
|
|
Err(ErrorCode::DeviceError(google_home::errors::DeviceError::ActionNotAvailable))
|
|
}
|
|
}
|
|
}
|