Compare commits
2
Commits
master
..
1369a628b4
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
1369a628b4
|
||
|
|
b24bb70777
|
@@ -1,5 +1,2 @@
|
||||
[env]
|
||||
RUST_LOG = "automation=debug"
|
||||
|
||||
[target.x86_64-unknown-linux-musl]
|
||||
rustflags = ["-C", "link-arg=-lc"]
|
||||
|
||||
@@ -2,5 +2,3 @@
|
||||
.env
|
||||
# Use the rust environment provided by the container
|
||||
rust-toolchain.toml
|
||||
Dockerfile
|
||||
docker-bake.hcl
|
||||
|
||||
@@ -1,28 +0,0 @@
|
||||
name: Audit
|
||||
|
||||
on:
|
||||
push:
|
||||
paths:
|
||||
- "**/Cargo.toml"
|
||||
- "**/Cargo.lock"
|
||||
schedule:
|
||||
- cron: "0 0 * * *"
|
||||
workflow_dispatch:
|
||||
|
||||
jobs:
|
||||
audit:
|
||||
name: cargo audit
|
||||
runs-on: ubuntu-latest
|
||||
permissions:
|
||||
issues: write
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v6
|
||||
- name: Install rust toolchain
|
||||
uses: dtolnay/rust-toolchain@nightly
|
||||
- name: Setup rust cache
|
||||
uses: Swatinem/rust-cache@v2
|
||||
- name: Audit check
|
||||
uses: https://git.huizinga.dev/infra/rust-audit-check@v3.0.0
|
||||
with:
|
||||
token: ${{ secrets.GITEA_TOKEN }}
|
||||
@@ -9,7 +9,7 @@ on:
|
||||
|
||||
jobs:
|
||||
build:
|
||||
uses: infra/workflows/.gitea/workflows/docker.yaml@2bb37b698effa4fba36040d668cfa95d4abc568e
|
||||
uses: dreaded_x/workflows/.gitea/workflows/docker-kubernetes.yaml@ef78704b98c72e4a6b8340f9bff7b085a7bdd95c
|
||||
secrets: inherit
|
||||
with:
|
||||
push_manifests: false
|
||||
@@ -41,8 +41,6 @@ jobs:
|
||||
-e AUTOMATION__SECRETS__MQTT_PASSWORD=${{ secrets.MQTT_PASSWORD }} \
|
||||
-e AUTOMATION__SECRETS__HUE_TOKEN=${{ secrets.HUE_TOKEN }} \
|
||||
-e AUTOMATION__SECRETS__NTFY_TOPIC=${{ secrets.NTFY_TOPIC }} \
|
||||
-e AUTOMATION__SECRETS__PRINTER_DEVICE_ID=${{ secrets.PRINTER_DEVICE_ID }} \
|
||||
-e AUTOMATION__SECRETS__PRINTER_ACCESS_CODE=${{ secrets.PRINTER_ACCESS_CODE }} \
|
||||
$(echo ${{ toJSON(needs.build.outputs.images) }} | jq .automation -r)
|
||||
|
||||
docker network connect web automation_rs
|
||||
|
||||
Generated
+663
-684
File diff suppressed because it is too large
Load Diff
+5
-5
@@ -34,10 +34,10 @@ google_home = { path = "./google_home/google_home" }
|
||||
google_home_macro = { path = "./google_home/google_home_macro" }
|
||||
hostname = "0.4.2"
|
||||
inventory = "0.3.24"
|
||||
itertools = "0.15.0"
|
||||
itertools = "0.14.0"
|
||||
json_value_merge = "2.0.1"
|
||||
lua_typed = { git = "https://git.huizinga.dev/Dreaded_X/lua_typed" }
|
||||
mlua = { version = "0.12.0", features = [
|
||||
mlua = { version = "0.11.6", features = [
|
||||
"lua54",
|
||||
"vendored",
|
||||
"macros",
|
||||
@@ -51,11 +51,11 @@ reqwest = { version = "0.13.3", features = [
|
||||
"json",
|
||||
"rustls",
|
||||
], default-features = false } # Use rustls, since the other packages also use rustls
|
||||
rumqttc-next = "0.34.0"
|
||||
rumqttc = "0.25.1"
|
||||
serde = { version = "1.0.228", features = ["derive"] }
|
||||
serde_json = "1.0.149"
|
||||
serde_repr = "0.1.20"
|
||||
syn = { version = "3.0.3" }
|
||||
syn = { version = "2.0.117" }
|
||||
thiserror = "2.0.18"
|
||||
tokio = { version = "1", features = ["rt-multi-thread"] }
|
||||
tokio-cron-scheduler = "0.15.1"
|
||||
@@ -80,6 +80,7 @@ lua_typed = { workspace = true }
|
||||
inventory = { workspace = true }
|
||||
mlua = { workspace = true }
|
||||
reqwest = { workspace = true }
|
||||
rumqttc = { workspace = true }
|
||||
serde = { workspace = true }
|
||||
serde_json = { workspace = true }
|
||||
thiserror = { workspace = true }
|
||||
@@ -87,7 +88,6 @@ tokio = { workspace = true }
|
||||
tokio-cron-scheduler = { workspace = true }
|
||||
tracing = { workspace = true }
|
||||
tracing-subscriber = { workspace = true }
|
||||
rumqttc-next = { workspace = true }
|
||||
|
||||
[patch.crates-io]
|
||||
wakey = { git = "https://git.huizinga.dev/Dreaded_X/wakey" }
|
||||
|
||||
+4
-4
@@ -1,7 +1,9 @@
|
||||
FROM rust:1.98-alpine3.24 AS base
|
||||
FROM rust:1.95 AS base
|
||||
ENV CARGO_REGISTRIES_CRATES_IO_PROTOCOL=sparse
|
||||
RUN cargo install cargo-chef --locked --version 0.1.71 && \
|
||||
cargo install cargo-auditable --locked --version 0.6.6
|
||||
WORKDIR /app
|
||||
COPY ./rust-toolchain.toml .
|
||||
RUN rustup toolchain install
|
||||
|
||||
FROM base AS planner
|
||||
@@ -9,7 +11,6 @@ COPY . .
|
||||
RUN cargo chef prepare --recipe-path recipe.json
|
||||
|
||||
FROM base AS builder
|
||||
RUN apk add --no-cache g++=15.2.0-r5 cmake=4.2.3-r0 make=4.4.1-r4 openssl-dev=3.5.8-r0 openssl-libs-static=3.5.8-r0
|
||||
# HACK: Now we can use unstable feature while on stable rust!
|
||||
ENV RUSTC_BOOTSTRAP=1
|
||||
COPY --from=planner /app/recipe.json recipe.json
|
||||
@@ -20,8 +21,7 @@ ARG RELEASE_VERSION
|
||||
ENV RELEASE_VERSION=${RELEASE_VERSION}
|
||||
RUN cargo auditable build --release
|
||||
|
||||
|
||||
FROM gcr.io/distroless/static-debian13:nonroot AS runtime
|
||||
FROM gcr.io/distroless/cc-debian12:nonroot AS runtime
|
||||
COPY --from=builder /app/target/release/automation /app/automation
|
||||
ENV AUTOMATION__ENTRYPOINT=/app/config/config.lua
|
||||
ENV LUA_PATH="/app/?.lua;;"
|
||||
|
||||
@@ -17,7 +17,7 @@ inventory = { workspace = true }
|
||||
lua_typed = { workspace = true }
|
||||
mlua = { workspace = true }
|
||||
reqwest = { workspace = true }
|
||||
rumqttc-next = { workspace = true }
|
||||
rumqttc = { workspace = true }
|
||||
serde = { workspace = true }
|
||||
serde_json = { workspace = true }
|
||||
serde_repr = { workspace = true }
|
||||
@@ -25,4 +25,3 @@ thiserror = { workspace = true }
|
||||
tokio = { workspace = true }
|
||||
tracing = { workspace = true }
|
||||
wakey = { workspace = true }
|
||||
bambulab = { version = "0.4.30", default-features = false }
|
||||
|
||||
@@ -1,142 +0,0 @@
|
||||
use std::convert::Infallible;
|
||||
use std::sync::Arc;
|
||||
use std::sync::atomic::AtomicBool;
|
||||
use std::time::Duration;
|
||||
|
||||
use async_trait::async_trait;
|
||||
use automation_lib::action_callback::ActionCallback;
|
||||
use automation_lib::device::Device;
|
||||
use automation_macro::{Device, LuaDeviceConfig};
|
||||
use bambulab::client::Client;
|
||||
use bambulab::{Command, Message};
|
||||
use google_home::errors::{self};
|
||||
use google_home::traits::OnOff;
|
||||
use lua_typed::Typed;
|
||||
use tracing::{debug, trace};
|
||||
|
||||
use crate::{DebugWrap, LuaDeviceCreate};
|
||||
|
||||
#[derive(Debug, Clone, LuaDeviceConfig, Typed, Default)]
|
||||
#[typed(as = "BambuCallbacks")]
|
||||
pub struct Callbacks {
|
||||
#[device_config(from_lua, default)]
|
||||
#[typed(default)]
|
||||
pub state: ActionCallback<Bambu>,
|
||||
#[device_config(from_lua, default)]
|
||||
#[typed(default)]
|
||||
pub connected: ActionCallback<Bambu>,
|
||||
}
|
||||
crate::register_type!(Callbacks);
|
||||
|
||||
#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
|
||||
#[typed(as = "BambuConfig")]
|
||||
pub struct Config {
|
||||
pub host: String,
|
||||
pub device_id: String,
|
||||
pub access_code: String,
|
||||
#[device_config(from_lua, default)]
|
||||
pub callbacks: Callbacks,
|
||||
}
|
||||
crate::register_type!(Config);
|
||||
|
||||
#[derive(Debug, Clone, Device)]
|
||||
#[device(traits(OnOff))]
|
||||
pub struct Bambu {
|
||||
config: Config,
|
||||
|
||||
client: DebugWrap<Client>,
|
||||
|
||||
state: Arc<AtomicBool>,
|
||||
}
|
||||
crate::register_device!(Bambu);
|
||||
|
||||
#[async_trait]
|
||||
impl LuaDeviceCreate for Bambu {
|
||||
type Config = Config;
|
||||
type Error = Infallible;
|
||||
|
||||
async fn create(config: Self::Config) -> Result<Self, Infallible> {
|
||||
trace!(id = config.device_id, "Setting up bambu");
|
||||
|
||||
let (tx, mut rx) = tokio::sync::broadcast::channel(25);
|
||||
let client = Client::new(&config.host, &config.access_code, &config.device_id, tx);
|
||||
|
||||
let state = Arc::new(AtomicBool::new(false));
|
||||
let bambu = Self {
|
||||
config,
|
||||
client: DebugWrap(client.clone()),
|
||||
state: state.clone(),
|
||||
};
|
||||
|
||||
tokio::spawn({
|
||||
let mut bambu = bambu.clone();
|
||||
async move {
|
||||
// The printer might be offline so periodically try to reconnecct
|
||||
loop {
|
||||
bambu.client.run().await.ok();
|
||||
|
||||
tokio::time::sleep(Duration::from_secs(60)).await;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
tokio::spawn({
|
||||
let bambu = bambu.clone();
|
||||
async move {
|
||||
loop {
|
||||
let message = rx.recv().await.unwrap();
|
||||
|
||||
match message {
|
||||
Message::Print(data) => 'print: {
|
||||
// Extract the state of the chamber light
|
||||
let Some(light_report) = data.print.lights_report else {
|
||||
break 'print;
|
||||
};
|
||||
|
||||
let on = light_report
|
||||
.iter()
|
||||
.find(|report| report.node == "chamber_light")
|
||||
.map(|report| report.mode == "on")
|
||||
.unwrap_or(false);
|
||||
|
||||
state.store(on, std::sync::atomic::Ordering::Relaxed);
|
||||
|
||||
bambu.config.callbacks.state.call(bambu.clone()).await;
|
||||
}
|
||||
Message::Connected => {
|
||||
debug!(id = bambu.config.device_id, "Connected");
|
||||
client.publish(Command::PushAll).await.unwrap();
|
||||
|
||||
bambu.config.callbacks.connected.call(bambu.clone()).await;
|
||||
}
|
||||
// Ignore everything else
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
Ok(bambu)
|
||||
}
|
||||
}
|
||||
|
||||
impl Device for Bambu {
|
||||
fn get_id(&self) -> String {
|
||||
self.config.device_id.clone()
|
||||
}
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
impl OnOff for Bambu {
|
||||
async fn on(&self) -> Result<bool, errors::ErrorCode> {
|
||||
Ok(self.state.load(std::sync::atomic::Ordering::Relaxed))
|
||||
}
|
||||
|
||||
async fn set_on(&self, on: bool) -> Result<(), errors::ErrorCode> {
|
||||
// NOTE: This will error in case the printer is offline, but we don't really care in that
|
||||
// case so we just ignore the error
|
||||
self.client.publish(Command::SetChamberLight(on)).await.ok();
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
@@ -159,10 +159,7 @@ impl OpenClose for ContactSensor {
|
||||
#[async_trait]
|
||||
impl OnMqtt for ContactSensor {
|
||||
async fn on_mqtt(&self, message: rumqttc::Publish) {
|
||||
if !rumqttc::matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if !rumqttc::matches(&message.topic, &self.config.mqtt.topic) {
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
@@ -98,10 +98,7 @@ impl LuaDeviceCreate for HueSwitch {
|
||||
impl OnMqtt for HueSwitch {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
// Check if the message is from the device itself or from a remote
|
||||
if matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if matches(&message.topic, &self.config.mqtt.topic) {
|
||||
let message = match serde_json::from_slice::<State>(&message.payload) {
|
||||
Ok(message) => message,
|
||||
Err(err) => {
|
||||
|
||||
@@ -70,10 +70,7 @@ impl LuaDeviceCreate for IkeaRemote {
|
||||
impl OnMqtt for IkeaRemote {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
// Check if the message is from the deviec itself or from a remote
|
||||
if matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if matches(&message.topic, &self.config.mqtt.topic) {
|
||||
let message = match RemoteMessage::try_from(message) {
|
||||
Ok(message) => message,
|
||||
Err(err) => {
|
||||
|
||||
@@ -1,7 +1,5 @@
|
||||
#![feature(debug_closure_helpers)]
|
||||
#![feature(iter_intersperse)]
|
||||
mod air_filter;
|
||||
mod bambu;
|
||||
mod contact_sensor;
|
||||
mod hue_bridge;
|
||||
mod hue_group;
|
||||
@@ -15,9 +13,6 @@ mod wake_on_lan;
|
||||
mod washer;
|
||||
mod zigbee;
|
||||
|
||||
use std::fmt;
|
||||
use std::ops::{Deref, DerefMut};
|
||||
|
||||
use automation_lib::Module;
|
||||
use automation_lib::device::{Device, LuaDeviceCreate};
|
||||
use tracing::{debug, warn};
|
||||
@@ -25,31 +20,6 @@ use tracing::{debug, warn};
|
||||
type DeviceNameFn = fn() -> String;
|
||||
type RegisterDeviceFn = fn(lua: &mlua::Lua) -> mlua::Result<mlua::AnyUserData>;
|
||||
|
||||
#[derive(Clone)]
|
||||
struct DebugWrap<T: Clone>(T);
|
||||
|
||||
impl<T: Clone> DerefMut for DebugWrap<T> {
|
||||
fn deref_mut(&mut self) -> &mut Self::Target {
|
||||
&mut self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl<T: Clone> Deref for DebugWrap<T> {
|
||||
type Target = T;
|
||||
|
||||
fn deref(&self) -> &Self::Target {
|
||||
&self.0
|
||||
}
|
||||
}
|
||||
|
||||
impl<T: Clone> fmt::Debug for DebugWrap<T> {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
f.debug_tuple("DebugWrap")
|
||||
.field_with(|f| f.write_str(stringify!(T)))
|
||||
.finish()
|
||||
}
|
||||
}
|
||||
|
||||
pub struct RegisteredDevice {
|
||||
name_fn: DeviceNameFn,
|
||||
register_fn: RegisterDeviceFn,
|
||||
|
||||
@@ -85,10 +85,7 @@ impl Device for LightSensor {
|
||||
#[async_trait]
|
||||
impl OnMqtt for LightSensor {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
if !rumqttc::matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if !rumqttc::matches(&message.topic, &self.config.mqtt.topic) {
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
@@ -105,10 +105,7 @@ impl Device for Presence {
|
||||
#[async_trait]
|
||||
impl OnMqtt for Presence {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
if !rumqttc::matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if !rumqttc::matches(&message.topic, &self.config.mqtt.topic) {
|
||||
return;
|
||||
}
|
||||
|
||||
@@ -119,9 +116,7 @@ impl OnMqtt for Presence {
|
||||
.find('+')
|
||||
.or(self.config.mqtt.topic.find('#'))
|
||||
.expect("Presence::create fails if it does not contain wildcards");
|
||||
let device_name: String = str::from_utf8(&message.topic[offset..])
|
||||
.expect("Topic should be valid")
|
||||
.into();
|
||||
let device_name = message.topic[offset..].into();
|
||||
|
||||
if message.payload.is_empty() {
|
||||
// Remove the device from the map
|
||||
|
||||
@@ -66,10 +66,7 @@ impl Device for WakeOnLAN {
|
||||
#[async_trait]
|
||||
impl OnMqtt for WakeOnLAN {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
if !rumqttc::matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if !rumqttc::matches(&message.topic, &self.config.mqtt.topic) {
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
@@ -89,10 +89,7 @@ const HYSTERESIS: isize = 10;
|
||||
#[async_trait]
|
||||
impl OnMqtt for Washer {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
if !rumqttc::matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if !rumqttc::matches(&message.topic, &self.config.mqtt.topic) {
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
@@ -1,4 +1,3 @@
|
||||
use core::str;
|
||||
use std::fmt::Debug;
|
||||
use std::ops::Deref;
|
||||
use std::sync::Arc;
|
||||
@@ -17,7 +16,7 @@ use google_home::errors::ErrorCode;
|
||||
use google_home::traits::{Brightness, Color, ColorSetting, ColorTemperatureRange, OnOff};
|
||||
use google_home::types::Type;
|
||||
use lua_typed::Typed;
|
||||
use rumqttc::{Publish, PublishOptions, matches};
|
||||
use rumqttc::{Publish, matches};
|
||||
use serde::{Deserialize, Serialize};
|
||||
use serde_json::json;
|
||||
use tokio::sync::{RwLock, RwLockReadGuard, RwLockWriteGuard};
|
||||
@@ -178,10 +177,7 @@ where
|
||||
impl OnMqtt for LightOnOff {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
// Check if the message is from the device itself or from a remote
|
||||
if matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if matches(&message.topic, &self.config.mqtt.topic) {
|
||||
let state = match serde_json::from_slice::<StateOnOff>(&message.payload) {
|
||||
Ok(state) => state,
|
||||
Err(err) => {
|
||||
@@ -214,10 +210,7 @@ impl OnMqtt for LightOnOff {
|
||||
impl OnMqtt for LightBrightness {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
// Check if the message is from the deviec itself or from a remote
|
||||
if matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if matches(&message.topic, &self.config.mqtt.topic) {
|
||||
let state = match serde_json::from_slice::<StateBrightness>(&message.payload) {
|
||||
Ok(state) => state,
|
||||
Err(err) => {
|
||||
@@ -256,10 +249,7 @@ impl OnMqtt for LightBrightness {
|
||||
impl OnMqtt for LightColorTemperature {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
// Check if the message is from the deviec itself or from a remote
|
||||
if matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if matches(&message.topic, &self.config.mqtt.topic) {
|
||||
let state = match serde_json::from_slice::<StateColorTemperature>(&message.payload) {
|
||||
Ok(state) => state,
|
||||
Err(err) => {
|
||||
@@ -352,8 +342,9 @@ where
|
||||
.client
|
||||
.publish(
|
||||
&topic,
|
||||
rumqttc::QoS::AtLeastOnce,
|
||||
false,
|
||||
serde_json::to_string(&message).unwrap(),
|
||||
PublishOptions::at_least_once(),
|
||||
)
|
||||
.await
|
||||
.map_err(|err| warn!("Failed to update state on {topic}: {err}"))
|
||||
@@ -395,8 +386,9 @@ where
|
||||
.client
|
||||
.publish(
|
||||
&topic,
|
||||
rumqttc::QoS::AtLeastOnce,
|
||||
false,
|
||||
serde_json::to_string(&message).unwrap(),
|
||||
PublishOptions::at_least_once(),
|
||||
)
|
||||
.await
|
||||
.map_err(|err| warn!("Failed to update state on {topic}: {err}"))
|
||||
@@ -442,8 +434,9 @@ where
|
||||
.client
|
||||
.publish(
|
||||
&topic,
|
||||
rumqttc::QoS::AtLeastOnce,
|
||||
false,
|
||||
serde_json::to_string(&message).unwrap(),
|
||||
PublishOptions::at_least_once(),
|
||||
)
|
||||
.await
|
||||
.map_err(|err| warn!("Failed to update state on {topic}: {err}"))
|
||||
|
||||
@@ -16,7 +16,7 @@ use google_home::errors::ErrorCode;
|
||||
use google_home::traits::OnOff;
|
||||
use google_home::types::Type;
|
||||
use lua_typed::Typed;
|
||||
use rumqttc::{Publish, PublishOptions, matches};
|
||||
use rumqttc::{Publish, matches};
|
||||
use serde::{Deserialize, Serialize};
|
||||
use serde_json::json;
|
||||
use tokio::sync::{RwLock, RwLockReadGuard, RwLockWriteGuard};
|
||||
@@ -162,10 +162,7 @@ where
|
||||
impl OnMqtt for OutletOnOff {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
// Check if the message is from the device itself or from a remote
|
||||
if matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if matches(&message.topic, &self.config.mqtt.topic) {
|
||||
let state = match serde_json::from_slice::<StateOnOff>(&message.payload) {
|
||||
Ok(state) => state,
|
||||
Err(err) => {
|
||||
@@ -198,10 +195,7 @@ impl OnMqtt for OutletOnOff {
|
||||
impl OnMqtt for OutletPower {
|
||||
async fn on_mqtt(&self, message: Publish) {
|
||||
// Check if the message is from the deviec itself or from a remote
|
||||
if matches(
|
||||
str::from_utf8(&message.topic).expect("Topic should be valid"),
|
||||
&self.config.mqtt.topic,
|
||||
) {
|
||||
if matches(&message.topic, &self.config.mqtt.topic) {
|
||||
let state = match serde_json::from_slice::<StatePower>(&message.payload) {
|
||||
Ok(state) => state,
|
||||
Err(err) => {
|
||||
@@ -290,8 +284,9 @@ where
|
||||
.client
|
||||
.publish(
|
||||
&topic,
|
||||
rumqttc::QoS::AtLeastOnce,
|
||||
false,
|
||||
serde_json::to_string(&message).unwrap(),
|
||||
PublishOptions::at_least_once(),
|
||||
)
|
||||
.await
|
||||
.map_err(|err| warn!("Failed to update state on {topic}: {err}"))
|
||||
|
||||
@@ -15,7 +15,7 @@ hostname = { workspace = true }
|
||||
inventory = { workspace = true }
|
||||
lua_typed = { workspace = true }
|
||||
mlua = { workspace = true }
|
||||
rumqttc-next = { workspace = true }
|
||||
rumqttc = { workspace = true }
|
||||
serde = { workspace = true }
|
||||
serde_json = { workspace = true }
|
||||
thiserror = { workspace = true }
|
||||
|
||||
@@ -1,9 +1,10 @@
|
||||
use std::ops::{Deref, DerefMut};
|
||||
use std::time::Duration;
|
||||
|
||||
use automation_macro::LuaDeviceConfig;
|
||||
use lua_typed::Typed;
|
||||
use mlua::FromLua;
|
||||
use rumqttc::{AsyncClient, Event, Incoming, MqttOptions, PublishOptions, Transport};
|
||||
use rumqttc::{AsyncClient, Event, Incoming, MqttOptions, Transport};
|
||||
use serde::Deserialize;
|
||||
use tracing::{debug, warn};
|
||||
|
||||
@@ -23,9 +24,9 @@ pub struct MqttConfig {
|
||||
|
||||
impl From<MqttConfig> for MqttOptions {
|
||||
fn from(value: MqttConfig) -> Self {
|
||||
let mut mqtt_options = MqttOptions::new(value.client_name, (value.host, value.port));
|
||||
let mut mqtt_options = MqttOptions::new(value.client_name, value.host, value.port);
|
||||
mqtt_options.set_credentials(value.username, value.password);
|
||||
mqtt_options.set_keep_alive(5);
|
||||
mqtt_options.set_keep_alive(Duration::from_secs(5));
|
||||
|
||||
if value.tls {
|
||||
mqtt_options.set_transport(Transport::tls_with_default_config());
|
||||
@@ -91,7 +92,7 @@ impl mlua::UserData for WrappedAsyncClient {
|
||||
debug!("message = {message}");
|
||||
|
||||
this.0
|
||||
.publish(topic, message, PublishOptions::at_least_once().retained())
|
||||
.publish(topic, rumqttc::QoS::AtLeastOnce, true, message)
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
@@ -103,7 +104,7 @@ impl mlua::UserData for WrappedAsyncClient {
|
||||
|
||||
pub fn start(config: MqttConfig, event_channel: &EventChannel) -> WrappedAsyncClient {
|
||||
let tx = event_channel.get_tx();
|
||||
let (client, mut eventloop) = AsyncClient::builder(config.into()).capacity(100).build();
|
||||
let (client, mut eventloop) = AsyncClient::new(config.into(), 100);
|
||||
|
||||
tokio::spawn(async move {
|
||||
debug!("Listening for MQTT events");
|
||||
|
||||
@@ -182,7 +182,7 @@ fn field_from_lua(field: &Field) -> TokenStream {
|
||||
.iter()
|
||||
.filter_map(|arg| match arg {
|
||||
Argument::Flatten { .. } => Some(quote! {
|
||||
mlua::LuaSerdeExt::from_value_with(lua, value.clone(), mlua::serde::DeserializeOptions::new().deny_unsupported_types(false))?
|
||||
mlua::LuaSerdeExt::from_value_with(lua, value.clone(), mlua::DeserializeOptions::new().deny_unsupported_types(false))?
|
||||
}),
|
||||
Argument::FromLua { .. } => Some(quote! {
|
||||
if table.contains_key(#table_name)? {
|
||||
|
||||
+2
-2
@@ -10,12 +10,12 @@ return {
|
||||
openid_url = "https://login.huizinga.dev/api/oidc",
|
||||
},
|
||||
mqtt = {
|
||||
host = ((host == "zeus" or host == "hephaestus") and "olympus.huizinga.lan") or "mosquitto",
|
||||
host = ((host == "zeus" or host == "hephaestus") and "olympus.lan.huizinga.dev") or "mosquitto",
|
||||
port = 8883,
|
||||
client_name = "automation-" .. host,
|
||||
username = "mqtt",
|
||||
password = secrets.mqtt_password,
|
||||
tls = false,
|
||||
tls = host == "zeus" or host == "hephaestus",
|
||||
},
|
||||
modules = {
|
||||
require("config.battery"),
|
||||
|
||||
@@ -4,35 +4,15 @@ local helper = require("config.helper")
|
||||
local presence = require("config.presence")
|
||||
local windows = require("config.windows")
|
||||
|
||||
local secrets = require("automation:secrets")
|
||||
|
||||
--- @type Module
|
||||
local module = {}
|
||||
|
||||
function module.setup(mqtt_client)
|
||||
local light = nil
|
||||
|
||||
local bambu = devices.Bambu.new({
|
||||
host = "10.0.0.108",
|
||||
device_id = secrets.printer_device_id,
|
||||
access_code = secrets.printer_access_code,
|
||||
callbacks = {
|
||||
connected = function(self)
|
||||
if light ~= nil then
|
||||
self:set_on(light:on())
|
||||
end
|
||||
end,
|
||||
},
|
||||
})
|
||||
|
||||
light = devices.LightOnOff.new({
|
||||
local light = devices.LightOnOff.new({
|
||||
name = "Light",
|
||||
room = "Guest Room",
|
||||
topic = helper.mqtt_z2m("guest/light"),
|
||||
client = mqtt_client,
|
||||
callback = function(_, state)
|
||||
bambu:set_on(state.state)
|
||||
end,
|
||||
})
|
||||
presence.turn_off_when_away(light)
|
||||
|
||||
@@ -57,7 +37,6 @@ function module.setup(mqtt_client)
|
||||
light,
|
||||
window,
|
||||
printer,
|
||||
bambu,
|
||||
}
|
||||
end
|
||||
|
||||
|
||||
@@ -24,25 +24,6 @@ function devices.AirFilter.new(config) end
|
||||
---@field url string
|
||||
local AirFilterConfig
|
||||
|
||||
---@class Bambu: DeviceInterface, OnOffInterface
|
||||
local Bambu
|
||||
devices.Bambu = {}
|
||||
---@param config BambuConfig
|
||||
---@return Bambu
|
||||
function devices.Bambu.new(config) end
|
||||
|
||||
---@class BambuCallbacks
|
||||
---@field state (fun(_: Bambu) | fun(_: Bambu)[])?
|
||||
---@field connected (fun(_: Bambu) | fun(_: Bambu)[])?
|
||||
local BambuCallbacks
|
||||
|
||||
---@class BambuConfig
|
||||
---@field host string
|
||||
---@field device_id string
|
||||
---@field access_code string
|
||||
---@field callbacks BambuCallbacks
|
||||
local BambuConfig
|
||||
|
||||
---@class ConfigLightLightStateBrightness
|
||||
---@field name string
|
||||
---@field room (string)?
|
||||
|
||||
+1
-3
@@ -5,9 +5,7 @@ group "default" {
|
||||
targets = ["automation"]
|
||||
}
|
||||
|
||||
target "docker-metadata-action" {
|
||||
tags = []
|
||||
}
|
||||
target "docker-metadata-action" {}
|
||||
|
||||
target "automation" {
|
||||
inherits = ["docker-metadata-action"]
|
||||
|
||||
Reference in New Issue
Block a user