Compare commits
7 Commits
b327d32177
...
1d6f95e0a0
| Author | SHA1 | Date | |
|---|---|---|---|
|
1d6f95e0a0
|
|||
|
df64804b00
|
|||
|
11b9787890
|
|||
|
8961101fdf
|
|||
|
17a68e8991
|
|||
|
be1602d0e2
|
|||
|
9bddeae54e
|
@@ -87,6 +87,14 @@ repos:
|
||||
files: (\.rs|Cargo.lock)$
|
||||
pass_filenames: false
|
||||
|
||||
- id: generate_definitions
|
||||
name: generate definitions
|
||||
description: Generate lua definitions
|
||||
entry: cargo run --bin generate_definitions
|
||||
language: system
|
||||
types: [rust]
|
||||
pass_filenames: false
|
||||
|
||||
- repo: https://github.com/hadolint/hadolint
|
||||
rev: v2.13.1
|
||||
hooks:
|
||||
|
||||
1
Cargo.lock
generated
1
Cargo.lock
generated
@@ -98,6 +98,7 @@ dependencies = [
|
||||
"config",
|
||||
"git-version",
|
||||
"google_home",
|
||||
"inventory",
|
||||
"mlua",
|
||||
"reqwest",
|
||||
"rumqttc",
|
||||
|
||||
@@ -2,6 +2,7 @@
|
||||
name = "automation"
|
||||
version = "0.1.0"
|
||||
edition = "2024"
|
||||
default-run = "automation"
|
||||
|
||||
[workspace]
|
||||
members = [
|
||||
@@ -76,6 +77,7 @@ config = { version = "0.15.15", default-features = false, features = [
|
||||
] }
|
||||
git-version = "0.3.9"
|
||||
google_home = { workspace = true }
|
||||
inventory = { workspace = true }
|
||||
mlua = { workspace = true }
|
||||
reqwest = { workspace = true }
|
||||
rumqttc = { workspace = true }
|
||||
|
||||
@@ -15,20 +15,7 @@ mod zigbee;
|
||||
|
||||
use automation_lib::Module;
|
||||
use automation_lib::device::{Device, LuaDeviceCreate};
|
||||
use tracing::debug;
|
||||
|
||||
macro_rules! register_device {
|
||||
($device:ty) => {
|
||||
::inventory::submit!(crate::RegisteredDevice::new(
|
||||
<$device as ::lua_typed::Typed>::type_name,
|
||||
::mlua::Lua::create_proxy::<$device>
|
||||
));
|
||||
|
||||
crate::register_type!($device);
|
||||
};
|
||||
}
|
||||
|
||||
pub(crate) use register_device;
|
||||
use tracing::{debug, warn};
|
||||
|
||||
type DeviceNameFn = fn() -> String;
|
||||
type RegisterDeviceFn = fn(lua: &mlua::Lua) -> mlua::Result<mlua::AnyUserData>;
|
||||
@@ -55,6 +42,18 @@ impl RegisteredDevice {
|
||||
}
|
||||
}
|
||||
|
||||
macro_rules! register_device {
|
||||
($device:ty) => {
|
||||
::inventory::submit!(crate::RegisteredDevice::new(
|
||||
<$device as ::lua_typed::Typed>::type_name,
|
||||
::mlua::Lua::create_proxy::<$device>
|
||||
));
|
||||
|
||||
crate::register_type!($device);
|
||||
};
|
||||
}
|
||||
pub(crate) use register_device;
|
||||
|
||||
inventory::collect!(RegisteredDevice);
|
||||
|
||||
pub fn create_module(lua: &mlua::Lua) -> mlua::Result<mlua::Table> {
|
||||
@@ -71,7 +70,9 @@ pub fn create_module(lua: &mlua::Lua) -> mlua::Result<mlua::Table> {
|
||||
Ok(devices)
|
||||
}
|
||||
|
||||
inventory::submit! {Module::new("automation:devices", create_module)}
|
||||
type RegisterTypeFn = fn() -> Option<String>;
|
||||
|
||||
pub struct RegisteredType(RegisterTypeFn);
|
||||
|
||||
macro_rules! register_type {
|
||||
($ty:ty) => {
|
||||
@@ -80,20 +81,25 @@ macro_rules! register_type {
|
||||
));
|
||||
};
|
||||
}
|
||||
|
||||
pub(crate) use register_type;
|
||||
|
||||
type RegisterTypeFn = fn() -> Option<String>;
|
||||
|
||||
pub struct RegisteredType(RegisterTypeFn);
|
||||
|
||||
inventory::collect!(RegisteredType);
|
||||
|
||||
pub fn generate_definitions() {
|
||||
println!("---@meta\n\nlocal devices\n");
|
||||
fn generate_definitions() -> String {
|
||||
let mut output = String::new();
|
||||
|
||||
output += "---@meta\n\nlocal devices\n\n";
|
||||
for ty in inventory::iter::<RegisteredType> {
|
||||
let def = ty.0().unwrap();
|
||||
println!("{def}");
|
||||
if let Some(def) = ty.0() {
|
||||
output += &(def + "\n");
|
||||
} else {
|
||||
// NOTE: Due to how this works the typed is erased, so we don't know the cause
|
||||
warn!("Registered type is missing generate_full function");
|
||||
}
|
||||
println!("return devices")
|
||||
}
|
||||
output += "return devices";
|
||||
|
||||
output
|
||||
}
|
||||
|
||||
inventory::submit! {Module::new("automation:devices", create_module, Some(generate_definitions))}
|
||||
|
||||
@@ -5,6 +5,7 @@ use std::sync::Arc;
|
||||
use futures::Future;
|
||||
use futures::future::join_all;
|
||||
use lua_typed::Typed;
|
||||
use mlua::FromLua;
|
||||
use tokio::sync::{RwLock, RwLockReadGuard};
|
||||
use tokio_cron_scheduler::{Job, JobScheduler};
|
||||
use tracing::{debug, instrument, trace};
|
||||
@@ -14,7 +15,7 @@ use crate::event::{Event, EventChannel, OnMqtt};
|
||||
|
||||
pub type DeviceMap = HashMap<String, Box<dyn Device>>;
|
||||
|
||||
#[derive(Clone)]
|
||||
#[derive(Clone, FromLua)]
|
||||
pub struct DeviceManager {
|
||||
devices: Arc<RwLock<DeviceMap>>,
|
||||
event_channel: EventChannel,
|
||||
|
||||
@@ -1,4 +1,3 @@
|
||||
#![allow(incomplete_features)]
|
||||
#![feature(iterator_try_collect)]
|
||||
#![feature(with_negative_coherence)]
|
||||
|
||||
@@ -17,15 +16,25 @@ pub mod mqtt;
|
||||
pub mod schedule;
|
||||
|
||||
type RegisterFn = fn(lua: &mlua::Lua) -> mlua::Result<mlua::Table>;
|
||||
type DefinitionsFn = fn() -> String;
|
||||
|
||||
pub struct Module {
|
||||
name: &'static str,
|
||||
register_fn: RegisterFn,
|
||||
definitions_fn: Option<DefinitionsFn>,
|
||||
}
|
||||
|
||||
impl Module {
|
||||
pub const fn new(name: &'static str, register_fn: RegisterFn) -> Self {
|
||||
Self { name, register_fn }
|
||||
pub const fn new(
|
||||
name: &'static str,
|
||||
register_fn: RegisterFn,
|
||||
definitions_fn: Option<DefinitionsFn>,
|
||||
) -> Self {
|
||||
Self {
|
||||
name,
|
||||
register_fn,
|
||||
definitions_fn,
|
||||
}
|
||||
}
|
||||
|
||||
pub const fn get_name(&self) -> &'static str {
|
||||
@@ -35,6 +44,10 @@ impl Module {
|
||||
pub fn register(&self, lua: &mlua::Lua) -> mlua::Result<mlua::Table> {
|
||||
(self.register_fn)(lua)
|
||||
}
|
||||
|
||||
pub fn definitions(&self) -> Option<String> {
|
||||
self.definitions_fn.map(|f| f())
|
||||
}
|
||||
}
|
||||
|
||||
pub fn load_modules(lua: &mlua::Lua) -> mlua::Result<()> {
|
||||
|
||||
@@ -2,6 +2,7 @@ mod timeout;
|
||||
|
||||
use std::time::{SystemTime, UNIX_EPOCH};
|
||||
|
||||
use lua_typed::Typed;
|
||||
pub use timeout::Timeout;
|
||||
|
||||
use crate::Module;
|
||||
@@ -9,7 +10,7 @@ use crate::Module;
|
||||
fn create_module(lua: &mlua::Lua) -> mlua::Result<mlua::Table> {
|
||||
let utils = lua.create_table()?;
|
||||
|
||||
utils.set("Timeout", lua.create_proxy::<Timeout>()?)?;
|
||||
utils.set(Timeout::type_name(), lua.create_proxy::<Timeout>()?)?;
|
||||
|
||||
let get_hostname = lua.create_function(|_lua, ()| {
|
||||
hostname::get()
|
||||
@@ -28,4 +29,20 @@ fn create_module(lua: &mlua::Lua) -> mlua::Result<mlua::Table> {
|
||||
Ok(utils)
|
||||
}
|
||||
|
||||
inventory::submit! {Module::new("automation:utils", create_module)}
|
||||
fn generate_definitions() -> String {
|
||||
let mut output = String::new();
|
||||
|
||||
output += "---@meta\n\nlocal utils\n\n";
|
||||
|
||||
output += &Timeout::generate_full().expect("Timeout should have generate_full");
|
||||
output += "\n";
|
||||
|
||||
output += "---@return string\nfunction utils.get_hostname() end\n\n";
|
||||
output += "---@return integer\nfunction utils.get_epoch() end\n\n";
|
||||
|
||||
output += "return utils";
|
||||
|
||||
output
|
||||
}
|
||||
|
||||
inventory::submit! {Module::new("automation:utils", create_module, Some(generate_definitions))}
|
||||
|
||||
@@ -1,11 +1,13 @@
|
||||
use std::ops::{Deref, DerefMut};
|
||||
|
||||
use lua_typed::Typed;
|
||||
use mlua::FromLua;
|
||||
use mlua::{FromLua, LuaSerdeExt};
|
||||
use rumqttc::{AsyncClient, Event, EventLoop, Incoming};
|
||||
use tracing::{debug, warn};
|
||||
|
||||
use crate::Module;
|
||||
use crate::config::MqttConfig;
|
||||
use crate::device_manager::DeviceManager;
|
||||
use crate::event::{self, EventChannel};
|
||||
|
||||
#[derive(Debug, Clone, FromLua)]
|
||||
@@ -39,9 +41,10 @@ impl Typed for WrappedAsyncClient {
|
||||
let type_name = Self::type_name();
|
||||
|
||||
output += &format!("mqtt.{type_name} = {{}}\n");
|
||||
output += &format!("---@param device_manager {}\n", DeviceManager::type_name());
|
||||
output += &format!("---@param config {}\n", MqttConfig::type_name());
|
||||
output += &format!("---@return {type_name}\n");
|
||||
output += "function mqtt.new(config) end\n";
|
||||
output += "function mqtt.new(device_manager, config) end\n";
|
||||
|
||||
Some(output)
|
||||
}
|
||||
@@ -108,3 +111,41 @@ pub fn start(mut eventloop: EventLoop, event_channel: &EventChannel) {
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
fn create_module(lua: &mlua::Lua) -> mlua::Result<mlua::Table> {
|
||||
let mqtt = lua.create_table()?;
|
||||
let mqtt_new = lua.create_function(
|
||||
move |lua, (device_manager, config): (DeviceManager, mlua::Value)| {
|
||||
let event_channel = device_manager.event_channel();
|
||||
let config: MqttConfig = lua.from_value(config)?;
|
||||
|
||||
// Create a mqtt client
|
||||
// TODO: When starting up, the devices are not yet created, this could lead to a device being out of sync
|
||||
let (client, eventloop) = AsyncClient::new(config.into(), 100);
|
||||
start(eventloop, &event_channel);
|
||||
|
||||
Ok(WrappedAsyncClient(client))
|
||||
},
|
||||
)?;
|
||||
mqtt.set("new", mqtt_new)?;
|
||||
|
||||
Ok(mqtt)
|
||||
}
|
||||
|
||||
fn generate_definitions() -> String {
|
||||
let mut output = String::new();
|
||||
|
||||
output += "---@meta\n\nlocal mqtt\n\n";
|
||||
|
||||
output += &MqttConfig::generate_full().expect("WrappedAsyncClient should have generate_full");
|
||||
output += "\n";
|
||||
output +=
|
||||
&WrappedAsyncClient::generate_full().expect("WrappedAsyncClient should have generate_full");
|
||||
output += "\n";
|
||||
|
||||
output += "return mqtt";
|
||||
|
||||
output
|
||||
}
|
||||
|
||||
inventory::submit! {Module::new("automation:mqtt", create_module, Some(generate_definitions))}
|
||||
|
||||
@@ -21,7 +21,7 @@ local fulfillment = {
|
||||
openid_url = "https://login.huizinga.dev/api/oidc",
|
||||
}
|
||||
|
||||
local mqtt_client = require("automation:mqtt").new({
|
||||
local mqtt_client = require("automation:mqtt").new(device_manager, {
|
||||
host = ((host == "zeus" or host == "hephaestus") and "olympus.lan.huizinga.dev") or "mosquitto",
|
||||
port = 8883,
|
||||
client_name = "automation-" .. host,
|
||||
|
||||
337
definitions/automation:devices.lua
Normal file
337
definitions/automation:devices.lua
Normal file
@@ -0,0 +1,337 @@
|
||||
-- DO NOT MODIFY, FILE IS AUTOMATICALLY GENERATED
|
||||
---@meta
|
||||
|
||||
local devices
|
||||
|
||||
---@class KasaOutletConfig
|
||||
---@field identifier string
|
||||
---@field ip string
|
||||
local KasaOutletConfig
|
||||
|
||||
---@class KasaOutlet: DeviceInterface, OnOffInterface
|
||||
local KasaOutlet
|
||||
devices.KasaOutlet = {}
|
||||
---@param config KasaOutletConfig
|
||||
---@return KasaOutlet
|
||||
function devices.KasaOutlet.new(config) end
|
||||
|
||||
---@class AirFilter: DeviceInterface, OnOffInterface
|
||||
local AirFilter
|
||||
devices.AirFilter = {}
|
||||
---@param config AirFilterConfig
|
||||
---@return AirFilter
|
||||
function devices.AirFilter.new(config) end
|
||||
|
||||
---@class AirFilterConfig
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field url string
|
||||
local AirFilterConfig
|
||||
|
||||
---@class Presence: DeviceInterface
|
||||
local Presence
|
||||
devices.Presence = {}
|
||||
---@param config PresenceConfig
|
||||
---@return Presence
|
||||
function devices.Presence.new(config) end
|
||||
---@async
|
||||
---@return boolean
|
||||
function Presence:overall_presence() end
|
||||
|
||||
---@class PresenceConfig
|
||||
---@field topic string
|
||||
---@field callback fun(_: Presence, _: boolean) | fun(_: Presence, _: boolean)[]?
|
||||
---@field client AsyncClient
|
||||
local PresenceConfig
|
||||
|
||||
---@class HueBridge: DeviceInterface
|
||||
local HueBridge
|
||||
devices.HueBridge = {}
|
||||
---@param config HueBridgeConfig
|
||||
---@return HueBridge
|
||||
function devices.HueBridge.new(config) end
|
||||
---@async
|
||||
---@param flag Flag
|
||||
---@param value boolean
|
||||
function HueBridge:set_flag(flag, value) end
|
||||
|
||||
---@class FlagIDs
|
||||
---@field presence integer
|
||||
---@field darkness integer
|
||||
local FlagIDs
|
||||
|
||||
---@class HueBridgeConfig
|
||||
---@field identifier string
|
||||
---@field ip string
|
||||
---@field login string
|
||||
---@field flags FlagIDs
|
||||
local HueBridgeConfig
|
||||
|
||||
---@alias Flag
|
||||
---| "presence"
|
||||
---| "darkness"
|
||||
|
||||
---@class WasherConfig
|
||||
---@field identifier string
|
||||
---@field topic string
|
||||
---@field threshold number
|
||||
---@field done_callback fun(_: Washer) | fun(_: Washer)[]?
|
||||
---@field client AsyncClient
|
||||
local WasherConfig
|
||||
|
||||
---@class Washer: DeviceInterface
|
||||
local Washer
|
||||
devices.Washer = {}
|
||||
---@param config WasherConfig
|
||||
---@return Washer
|
||||
function devices.Washer.new(config) end
|
||||
|
||||
---@class LightSensor: DeviceInterface
|
||||
local LightSensor
|
||||
devices.LightSensor = {}
|
||||
---@param config LightSensorConfig
|
||||
---@return LightSensor
|
||||
function devices.LightSensor.new(config) end
|
||||
|
||||
---@class LightSensorConfig
|
||||
---@field identifier string
|
||||
---@field topic string
|
||||
---@field min integer
|
||||
---@field max integer
|
||||
---@field callback fun(_: LightSensor, _: boolean) | fun(_: LightSensor, _: boolean)[]?
|
||||
---@field client AsyncClient
|
||||
local LightSensorConfig
|
||||
|
||||
---@class HueGroupConfig
|
||||
---@field identifier string
|
||||
---@field ip string
|
||||
---@field login string
|
||||
---@field group_id integer
|
||||
---@field scene_id string
|
||||
local HueGroupConfig
|
||||
|
||||
---@class HueGroup: DeviceInterface, OnOffInterface
|
||||
local HueGroup
|
||||
devices.HueGroup = {}
|
||||
---@param config HueGroupConfig
|
||||
---@return HueGroup
|
||||
function devices.HueGroup.new(config) end
|
||||
|
||||
---@class IkeaRemote: DeviceInterface
|
||||
local IkeaRemote
|
||||
devices.IkeaRemote = {}
|
||||
---@param config IkeaRemoteConfig
|
||||
---@return IkeaRemote
|
||||
function devices.IkeaRemote.new(config) end
|
||||
|
||||
---@class IkeaRemoteConfig
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field single_button boolean?
|
||||
---@field topic string
|
||||
---@field client AsyncClient
|
||||
---@field callback fun(_: IkeaRemote, _: boolean) | fun(_: IkeaRemote, _: boolean)[]?
|
||||
---@field battery_callback fun(_: IkeaRemote, _: number) | fun(_: IkeaRemote, _: number)[]?
|
||||
local IkeaRemoteConfig
|
||||
|
||||
---@class WolConfig
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field topic string
|
||||
---@field mac_address string
|
||||
---@field broadcast_ip string?
|
||||
---@field client AsyncClient
|
||||
local WolConfig
|
||||
|
||||
---@class WakeOnLAN: DeviceInterface
|
||||
local WakeOnLAN
|
||||
devices.WakeOnLAN = {}
|
||||
---@param config WolConfig
|
||||
---@return WakeOnLAN
|
||||
function devices.WakeOnLAN.new(config) end
|
||||
|
||||
---@class ConfigOutletOutletStateOnOff
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field topic string
|
||||
---@field outlet_type OutletType?
|
||||
---@field callback fun(_: OutletOnOff, _: OutletStateOnOff) | fun(_: OutletOnOff, _: OutletStateOnOff)[]?
|
||||
---@field client AsyncClient
|
||||
local ConfigOutletOutletStateOnOff
|
||||
|
||||
---@class OutletStatePower
|
||||
---@field state boolean
|
||||
---@field power number
|
||||
local OutletStatePower
|
||||
|
||||
---@class ConfigOutletOutletStatePower
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field topic string
|
||||
---@field outlet_type OutletType?
|
||||
---@field callback fun(_: OutletPower, _: OutletStatePower) | fun(_: OutletPower, _: OutletStatePower)[]?
|
||||
---@field client AsyncClient
|
||||
local ConfigOutletOutletStatePower
|
||||
|
||||
---@alias OutletType
|
||||
---| "Outlet"
|
||||
---| "Kettle"
|
||||
|
||||
---@class OutletStateOnOff
|
||||
---@field state boolean
|
||||
local OutletStateOnOff
|
||||
|
||||
---@class OutletOnOff: DeviceInterface, OnOffInterface
|
||||
local OutletOnOff
|
||||
devices.OutletOnOff = {}
|
||||
---@param config ConfigOutletOutletStateOnOff
|
||||
---@return OutletOnOff
|
||||
function devices.OutletOnOff.new(config) end
|
||||
|
||||
---@class OutletPower: DeviceInterface, OnOffInterface
|
||||
local OutletPower
|
||||
devices.OutletPower = {}
|
||||
---@param config ConfigOutletOutletStatePower
|
||||
---@return OutletPower
|
||||
function devices.OutletPower.new(config) end
|
||||
|
||||
---@class ContactSensorConfig
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field topic string
|
||||
---@field sensor_type SensorType?
|
||||
---@field callback fun(_: ContactSensor, _: boolean) | fun(_: ContactSensor, _: boolean)[]?
|
||||
---@field battery_callback fun(_: ContactSensor, _: number) | fun(_: ContactSensor, _: number)[]?
|
||||
---@field client AsyncClient?
|
||||
local ContactSensorConfig
|
||||
|
||||
---@alias SensorType
|
||||
---| "Door"
|
||||
---| "Drawer"
|
||||
---| "Window"
|
||||
|
||||
---@class ContactSensor: DeviceInterface, OpenCloseInterface
|
||||
local ContactSensor
|
||||
devices.ContactSensor = {}
|
||||
---@param config ContactSensorConfig
|
||||
---@return ContactSensor
|
||||
function devices.ContactSensor.new(config) end
|
||||
|
||||
---@class LightStateOnOff
|
||||
---@field state boolean
|
||||
local LightStateOnOff
|
||||
|
||||
---@class ConfigLightLightStateColorTemperature
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field topic string
|
||||
---@field callback fun(_: LightColorTemperature, _: LightStateColorTemperature) | fun(_: LightColorTemperature, _: LightStateColorTemperature)[]?
|
||||
---@field client AsyncClient?
|
||||
local ConfigLightLightStateColorTemperature
|
||||
|
||||
---@class LightStateBrightness
|
||||
---@field state boolean
|
||||
---@field brightness number
|
||||
local LightStateBrightness
|
||||
|
||||
---@class LightStateColorTemperature
|
||||
---@field state boolean
|
||||
---@field brightness number
|
||||
---@field color_temp integer
|
||||
local LightStateColorTemperature
|
||||
|
||||
---@class LightBrightness: DeviceInterface, OnOffInterface, BrightnessInterface
|
||||
local LightBrightness
|
||||
devices.LightBrightness = {}
|
||||
---@param config ConfigLightLightStateBrightness
|
||||
---@return LightBrightness
|
||||
function devices.LightBrightness.new(config) end
|
||||
|
||||
---@class LightColorTemperature: DeviceInterface, OnOffInterface, BrightnessInterface, ColorSettingInterface
|
||||
local LightColorTemperature
|
||||
devices.LightColorTemperature = {}
|
||||
---@param config ConfigLightLightStateColorTemperature
|
||||
---@return LightColorTemperature
|
||||
function devices.LightColorTemperature.new(config) end
|
||||
|
||||
---@class ConfigLightLightStateOnOff
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field topic string
|
||||
---@field callback fun(_: LightOnOff, _: LightStateOnOff) | fun(_: LightOnOff, _: LightStateOnOff)[]?
|
||||
---@field client AsyncClient?
|
||||
local ConfigLightLightStateOnOff
|
||||
|
||||
---@class ConfigLightLightStateBrightness
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field topic string
|
||||
---@field callback fun(_: LightBrightness, _: LightStateBrightness) | fun(_: LightBrightness, _: LightStateBrightness)[]?
|
||||
---@field client AsyncClient?
|
||||
local ConfigLightLightStateBrightness
|
||||
|
||||
---@class LightOnOff: DeviceInterface, OnOffInterface
|
||||
local LightOnOff
|
||||
devices.LightOnOff = {}
|
||||
---@param config ConfigLightLightStateOnOff
|
||||
---@return LightOnOff
|
||||
function devices.LightOnOff.new(config) end
|
||||
|
||||
---@class Action
|
||||
---@field action
|
||||
---| "broadcast"
|
||||
---@field extras table<string, string>?
|
||||
---@field label string
|
||||
---@field clear boolean?
|
||||
local Action
|
||||
|
||||
---@class NtfyConfig
|
||||
---@field url string?
|
||||
---@field topic string
|
||||
local NtfyConfig
|
||||
|
||||
---@class Notification
|
||||
---@field title string
|
||||
---@field message string?
|
||||
---@field tags string[]?
|
||||
---@field priority Priority?
|
||||
---@field actions Action[]?
|
||||
local Notification
|
||||
|
||||
---@alias Priority
|
||||
---| "min"
|
||||
---| "low"
|
||||
---| "default"
|
||||
---| "high"
|
||||
---| "max"
|
||||
|
||||
---@class Ntfy: DeviceInterface
|
||||
local Ntfy
|
||||
devices.Ntfy = {}
|
||||
---@param config NtfyConfig
|
||||
---@return Ntfy
|
||||
function devices.Ntfy.new(config) end
|
||||
---@async
|
||||
---@param notification Notification
|
||||
function Ntfy:send_notification(notification) end
|
||||
|
||||
---@class HueSwitch: DeviceInterface
|
||||
local HueSwitch
|
||||
devices.HueSwitch = {}
|
||||
---@param config HueSwitchConfig
|
||||
---@return HueSwitch
|
||||
function devices.HueSwitch.new(config) end
|
||||
|
||||
---@class HueSwitchConfig
|
||||
---@field name string
|
||||
---@field room string?
|
||||
---@field topic string
|
||||
---@field client AsyncClient
|
||||
---@field left_callback fun(_: HueSwitch) | fun(_: HueSwitch)[]?
|
||||
---@field right_callback fun(_: HueSwitch) | fun(_: HueSwitch)[]?
|
||||
---@field left_hold_callback fun(_: HueSwitch) | fun(_: HueSwitch)[]?
|
||||
---@field right_hold_callback fun(_: HueSwitch) | fun(_: HueSwitch)[]?
|
||||
---@field battery_callback fun(_: HueSwitch, _: number) | fun(_: HueSwitch, _: number)[]?
|
||||
local HueSwitchConfig
|
||||
|
||||
return devices
|
||||
27
definitions/automation:mqtt.lua
Normal file
27
definitions/automation:mqtt.lua
Normal file
@@ -0,0 +1,27 @@
|
||||
-- DO NOT MODIFY, FILE IS AUTOMATICALLY GENERATED
|
||||
---@meta
|
||||
|
||||
local mqtt
|
||||
|
||||
---@class MqttConfig
|
||||
---@field host string
|
||||
---@field port integer
|
||||
---@field client_name string
|
||||
---@field username string
|
||||
---@field password string
|
||||
---@field tls boolean
|
||||
local MqttConfig
|
||||
|
||||
---@class AsyncClient
|
||||
local AsyncClient
|
||||
---@async
|
||||
---@param topic string
|
||||
---@param message table?
|
||||
function AsyncClient:send_message(topic, message) end
|
||||
mqtt.AsyncClient = {}
|
||||
---@param device_manager DeviceManager
|
||||
---@param config MqttConfig
|
||||
---@return AsyncClient
|
||||
function mqtt.new(device_manager, config) end
|
||||
|
||||
return mqtt
|
||||
6
definitions/automation:secrets.lua
Normal file
6
definitions/automation:secrets.lua
Normal file
@@ -0,0 +1,6 @@
|
||||
---@meta
|
||||
|
||||
---@type table<string, string?>
|
||||
local secrets
|
||||
|
||||
return secrets
|
||||
27
definitions/automation:utils.lua
Normal file
27
definitions/automation:utils.lua
Normal file
@@ -0,0 +1,27 @@
|
||||
-- DO NOT MODIFY, FILE IS AUTOMATICALLY GENERATED
|
||||
---@meta
|
||||
|
||||
local utils
|
||||
|
||||
---@class Timeout
|
||||
local Timeout
|
||||
---@async
|
||||
---@param timeout number
|
||||
---@param callback fun() | fun()[]
|
||||
function Timeout:start(timeout, callback) end
|
||||
---@async
|
||||
function Timeout:cancel() end
|
||||
---@async
|
||||
---@return boolean
|
||||
function Timeout:is_waiting() end
|
||||
utils.Timeout = {}
|
||||
---@return Timeout
|
||||
function utils.Timeout.new() end
|
||||
|
||||
---@return string
|
||||
function utils.get_hostname() end
|
||||
|
||||
---@return integer
|
||||
function utils.get_epoch() end
|
||||
|
||||
return utils
|
||||
6
definitions/automation:variables.lua
Normal file
6
definitions/automation:variables.lua
Normal file
@@ -0,0 +1,6 @@
|
||||
---@meta
|
||||
|
||||
---@type table<string, string?>
|
||||
local variables
|
||||
|
||||
return variables
|
||||
33
src/bin/generate_definitions.rs
Normal file
33
src/bin/generate_definitions.rs
Normal file
@@ -0,0 +1,33 @@
|
||||
use std::fs::{self, File};
|
||||
use std::io::Write;
|
||||
|
||||
use automation_lib::Module;
|
||||
use tracing::{info, warn};
|
||||
|
||||
extern crate automation_devices;
|
||||
|
||||
fn main() -> std::io::Result<()> {
|
||||
tracing_subscriber::fmt::init();
|
||||
|
||||
let definitions_directory =
|
||||
std::path::Path::new(std::env!("CARGO_MANIFEST_DIR")).join("definitions");
|
||||
fs::create_dir_all(&definitions_directory)?;
|
||||
|
||||
for module in inventory::iter::<Module> {
|
||||
if let Some(definitions) = module.definitions() {
|
||||
info!(name = module.get_name(), "Generating definitions");
|
||||
|
||||
let filename = format!("{}.lua", module.get_name());
|
||||
let mut file = File::create(definitions_directory.join(filename))?;
|
||||
|
||||
file.write_all(b"-- DO NOT MODIFY, FILE IS AUTOMATICALLY GENERATED\n")?;
|
||||
file.write_all(definitions.as_bytes())?;
|
||||
file.write_all(b"\n")?;
|
||||
} else {
|
||||
warn!(name = module.get_name(), "No definitions");
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
// automation_devices::generate_definitions()
|
||||
}
|
||||
19
src/main.rs
19
src/main.rs
@@ -9,9 +9,8 @@ use std::path::Path;
|
||||
use std::process;
|
||||
|
||||
use ::config::{Environment, File};
|
||||
use automation_lib::config::{FulfillmentConfig, MqttConfig};
|
||||
use automation_lib::config::FulfillmentConfig;
|
||||
use automation_lib::device_manager::DeviceManager;
|
||||
use automation_lib::mqtt::{self, WrappedAsyncClient};
|
||||
use axum::extract::{FromRef, State};
|
||||
use axum::http::StatusCode;
|
||||
use axum::routing::post;
|
||||
@@ -19,7 +18,6 @@ use axum::{Json, Router};
|
||||
use config::Config;
|
||||
use google_home::{GoogleHome, Request, Response};
|
||||
use mlua::LuaSerdeExt;
|
||||
use rumqttc::AsyncClient;
|
||||
use tokio::net::TcpListener;
|
||||
use tracing::{debug, error, info, warn};
|
||||
use web::{ApiError, User};
|
||||
@@ -138,21 +136,6 @@ async fn app() -> anyhow::Result<()> {
|
||||
|
||||
automation_lib::load_modules(&lua)?;
|
||||
|
||||
let mqtt = lua.create_table()?;
|
||||
let event_channel = device_manager.event_channel();
|
||||
let mqtt_new = lua.create_function(move |lua, config: mlua::Value| {
|
||||
let config: MqttConfig = lua.from_value(config)?;
|
||||
|
||||
// Create a mqtt client
|
||||
// TODO: When starting up, the devices are not yet created, this could lead to a device being out of sync
|
||||
let (client, eventloop) = AsyncClient::new(config.into(), 100);
|
||||
mqtt::start(eventloop, &event_channel);
|
||||
|
||||
Ok(WrappedAsyncClient(client))
|
||||
})?;
|
||||
mqtt.set("new", mqtt_new)?;
|
||||
lua.register_module("automation:mqtt", mqtt)?;
|
||||
|
||||
lua.register_module("automation:device_manager", device_manager.clone())?;
|
||||
|
||||
lua.register_module("automation:variables", lua.to_value(&config.variables)?)?;
|
||||
|
||||
Reference in New Issue
Block a user