147 lines
4.0 KiB
Rust
147 lines
4.0 KiB
Rust
use std::marker::PhantomData;
|
|
use std::ops::Deref;
|
|
|
|
use lua_typed::Typed;
|
|
|
|
// TODO: Enable and disable functions based on query_only and command_only
|
|
|
|
pub struct OnOff<T> {
|
|
_phantom: PhantomData<T>,
|
|
}
|
|
|
|
impl<T: google_home::traits::OnOff + Typed> OnOff<T> {
|
|
pub fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M)
|
|
where
|
|
T: Sized + 'static,
|
|
{
|
|
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())
|
|
});
|
|
}
|
|
|
|
pub fn generate_definitions() -> String {
|
|
let type_name = T::type_name();
|
|
let mut output = String::new();
|
|
|
|
output +=
|
|
&format!("---@async\n---@param on boolean\nfunction {type_name}:set_on(on) end\n");
|
|
output += &format!("---@async\n---@return boolean\nfunction {type_name}:on() end\n");
|
|
|
|
output
|
|
}
|
|
}
|
|
|
|
pub struct Brightness<T> {
|
|
_phantom: PhantomData<T>,
|
|
}
|
|
|
|
impl<T: google_home::traits::Brightness + Typed> Brightness<T> {
|
|
pub fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M)
|
|
where
|
|
T: Sized + 'static,
|
|
{
|
|
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())
|
|
});
|
|
}
|
|
|
|
pub fn generate_definitions() -> String {
|
|
let type_name = T::type_name();
|
|
let mut output = String::new();
|
|
|
|
output += &format!(
|
|
"---@async\n---@param brightness integer\nfunction {type_name}:set_brightness(brightness) end\n"
|
|
);
|
|
output +=
|
|
&format!("---@async\n---@return integer\nfunction {type_name}:brightness() end\n");
|
|
|
|
output
|
|
}
|
|
}
|
|
|
|
pub struct ColorSetting<T> {
|
|
_phantom: PhantomData<T>,
|
|
}
|
|
|
|
impl<T: google_home::traits::ColorSetting + Typed> ColorSetting<T> {
|
|
pub fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M)
|
|
where
|
|
T: Sized + 'static,
|
|
{
|
|
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)
|
|
});
|
|
}
|
|
|
|
pub fn generate_definitions() -> String {
|
|
let type_name = T::type_name();
|
|
let mut output = String::new();
|
|
|
|
output += &format!(
|
|
"---@async\n---@param temperature integer\nfunction {type_name}:set_color_temperature(temperature) end\n"
|
|
);
|
|
output += &format!(
|
|
"---@async\n---@return integer\nfunction {type_name}:color_temperature() end\n"
|
|
);
|
|
|
|
output
|
|
}
|
|
}
|
|
|
|
pub struct OpenClose<T> {
|
|
_phantom: PhantomData<T>,
|
|
}
|
|
|
|
impl<T: google_home::traits::OpenClose + Typed> OpenClose<T> {
|
|
pub fn add_methods<M: mlua::UserDataMethods<T>>(methods: &mut M)
|
|
where
|
|
T: Sized + 'static,
|
|
{
|
|
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())
|
|
});
|
|
}
|
|
|
|
pub fn generate_definitions() -> String {
|
|
let type_name = T::type_name();
|
|
let mut output = String::new();
|
|
|
|
output += &format!(
|
|
"---@async\n---@param open_percent integer\nfunction {type_name}:set_open_percent(open_percent) end\n"
|
|
);
|
|
output +=
|
|
&format!("---@async\n---@return integer\nfunction {type_name}:open_percent() end\n");
|
|
|
|
output
|
|
}
|
|
}
|