124 lines
3.1 KiB
Rust
124 lines
3.1 KiB
Rust
use std::ops::Deref;
|
|
|
|
// TODO: Enable and disable functions based on query_only and command_only
|
|
|
|
pub trait PartialUserData<T> {
|
|
fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M);
|
|
|
|
fn interface_name() -> Option<&'static str> {
|
|
None
|
|
}
|
|
}
|
|
|
|
pub struct Device;
|
|
|
|
impl<T> PartialUserData<T> for Device
|
|
where
|
|
T: crate::device::Device + 'static,
|
|
{
|
|
fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M) {
|
|
methods.add_async_method("get_id", async |_lua, this, _: ()| Ok(this.get_id()));
|
|
}
|
|
|
|
fn interface_name() -> Option<&'static str> {
|
|
Some("DeviceInterface")
|
|
}
|
|
}
|
|
|
|
pub struct OnOff;
|
|
|
|
impl<T> PartialUserData<T> for OnOff
|
|
where
|
|
T: google_home::traits::OnOff + 'static,
|
|
{
|
|
fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M) {
|
|
methods.add_async_method("set_on", async |_lua, this, on: bool| {
|
|
this.deref().set_on(on).await.unwrap();
|
|
|
|
Ok(())
|
|
});
|
|
|
|
methods.add_async_method("on", async |_lua, this, ()| {
|
|
Ok(this.deref().on().await.unwrap())
|
|
});
|
|
}
|
|
|
|
fn interface_name() -> Option<&'static str> {
|
|
Some("OnOffInterface")
|
|
}
|
|
}
|
|
|
|
pub struct Brightness;
|
|
|
|
impl<T> PartialUserData<T> for Brightness
|
|
where
|
|
T: google_home::traits::Brightness + 'static,
|
|
{
|
|
fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M) {
|
|
methods.add_async_method("set_brightness", async |_lua, this, brightness: u8| {
|
|
this.set_brightness(brightness).await.unwrap();
|
|
|
|
Ok(())
|
|
});
|
|
|
|
methods.add_async_method("brightness", async |_lua, this, _: ()| {
|
|
Ok(this.brightness().await.unwrap())
|
|
});
|
|
}
|
|
|
|
fn interface_name() -> Option<&'static str> {
|
|
Some("BrightnessInterface")
|
|
}
|
|
}
|
|
|
|
pub struct ColorSetting;
|
|
|
|
impl<T> PartialUserData<T> for ColorSetting
|
|
where
|
|
T: google_home::traits::ColorSetting + 'static,
|
|
{
|
|
fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M) {
|
|
methods.add_async_method(
|
|
"set_color_temperature",
|
|
async |_lua, this, temperature: u32| {
|
|
this.set_color(google_home::traits::Color { temperature })
|
|
.await
|
|
.unwrap();
|
|
|
|
Ok(())
|
|
},
|
|
);
|
|
|
|
methods.add_async_method("color_temperature", async |_lua, this, ()| {
|
|
Ok(this.color().await.temperature)
|
|
});
|
|
}
|
|
|
|
fn interface_name() -> Option<&'static str> {
|
|
Some("ColorSettingInterface")
|
|
}
|
|
}
|
|
|
|
pub struct OpenClose;
|
|
|
|
impl<T> PartialUserData<T> for OpenClose
|
|
where
|
|
T: google_home::traits::OpenClose + 'static,
|
|
{
|
|
fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M) {
|
|
methods.add_async_method("set_open_percent", async |_lua, this, open_percent: u8| {
|
|
this.set_open_percent(open_percent).await.unwrap();
|
|
|
|
Ok(())
|
|
});
|
|
|
|
methods.add_async_method("open_percent", async |_lua, this, _: ()| {
|
|
Ok(this.open_percent().await.unwrap())
|
|
});
|
|
}
|
|
|
|
fn interface_name() -> Option<&'static str> {
|
|
Some("OpenCloseInterface")
|
|
}
|
|
}
|