Run main binary by default

.cargo/config.toml

@@ -1,2 +1,3 @@
-[env]
+[build]
-RUST_LOG = "automation=debug"
+target = "x86_64-unknown-linux-gnu"
+rustflags = ["--cfg", "tokio_unstable"]

.dockerignore

@@ -1,4 +1,2 @@
 /target
 .env
-# Use the rust environment provided by the container
-rust-toolchain.toml

.gitattributes

@@ -1 +0,0 @@
-*.xcf filter=lfs diff=lfs merge=lfs -text

.gitea/workflows/build.yml

@@ -1,24 +1,116 @@
-name: Build and deploy
+# Based on: https://pastebin.com/99Fq2b2w
+name: Build and deploy automation_rs
 on:
     push:
         branches:
             - master
             - feature/**
-        tags:
-            - v*.*.*
 
 jobs:
     build:
-        uses: dreaded_x/workflows/.gitea/workflows/rust-kubernetes.yaml@22ee0c1788a8d2157db87d6a6f8dbe520fe48592
+        name: Build automation_rs
-        secrets: inherit
-        with:
-            upload_manifests: false
-
-    deploy:
-        name: Deploy container
         runs-on: ubuntu-latest
         container: catthehacker/ubuntu:act-latest
-        needs: build
+        steps:
+            - name: Checkout
+              uses: actions/checkout@v4
+
+            - name: Setup Rust
+              uses: actions-rust-lang/setup-rust-toolchain@v1
+              with:
+                  rustflags: ""
+
+            - name: Build
+              run: cargo build --release --all-targets
+
+            - name: Upload artifact
+              uses: actions/upload-artifact@v3
+              with:
+                  name: build
+                  path: target/x86_64-unknown-linux-gnu/release/automation
+
+    check:
+        name: Run pre-commit checks
+        runs-on: ubuntu-latest
+        # The lua definition check require that generate_definitions is build first
+        needs: [build]
+        container: catthehacker/ubuntu:act-latest
+        steps:
+            - name: Checkout
+              uses: actions/checkout@v4
+
+            - name: Setup go
+              uses: actions/setup-go@v3
+              with:
+                  go-version: ">=1.20.1"
+            - name: Calculate cache hash
+              uses: https://gitea.com/actions/go-hashfiles@v0.0.1
+              id: hash-go
+              with:
+                  patterns: |
+                      .pre-commit-config.yaml
+            - name: Setup cache
+              uses: actions/cache@v4
+              with:
+                  path: ~/.cache/pre-commit
+                  key: pre-commit-3|${{ steps.hash-go.outputs.hash }}
+
+            - name: Install pre-commit
+              run: python -m pip install pre-commit
+              shell: bash
+
+            - name: Setup Rust
+              uses: actions-rust-lang/setup-rust-toolchain@v1
+              with:
+                  rustflags: ""
+
+            - name: Set rustup default
+              run: rustup default stable
+              shell: bash
+
+            - name: Run pre-commit
+              run: pre-commit run --show-diff-on-failure --color=always --all-files
+              shell: bash
+
+    container:
+        name: Build Docker image
+        runs-on: ubuntu-latest
+        needs: [build, checks]
+        container: catthehacker/ubuntu:act-latest
+        steps:
+            - name: Checkout
+              uses: actions/checkout@v4
+
+            - name: Download artifact
+              uses: actions/download-artifact@v3
+              with:
+                  name: build
+
+            - name: Set permissions
+              run: |
+                  chown 65532:65532 ./build/*
+                  chmod 0755 ./build/*
+
+            - name: Login to registry
+              uses: https://github.com/docker/login-action@v3
+              with:
+                  registry: git.huizinga.dev
+                  username: ${{ gitea.actor }}
+                  password: ${{ secrets.REGISTRY_TOKEN }}
+
+            - name: Build and push Docker image
+              uses: https://github.com/docker/build-push-action@v5
+              with:
+                  context: .
+                  push: ${{ gitea.ref == 'refs/heads/master' }}
+                  # TODO: Automatically add the correct tags here
+                  tags: git.huizinga.dev/dreaded_x/automation_rs:latest
+
+    deploy:
+        name: Deploy Docker container
+        runs-on: ubuntu-latest
+        container: catthehacker/ubuntu:act-latest
+        needs: [container]
         if: gitea.ref == 'refs/heads/master'
         steps:
             - name: Stop and remove current container
@@ -34,12 +126,14 @@ jobs:
                     --name automation_rs \
                     --network mqtt \
                     -e RUST_LOG=automation=debug \
-                    -e AUTOMATION__SECRETS__MQTT_PASSWORD=${{ secrets.MQTT_PASSWORD }} \
+                    -e MQTT_PASSWORD=${{ secrets.MQTT_PASSWORD }} \
-                    -e AUTOMATION__SECRETS__HUE_TOKEN=${{ secrets.HUE_TOKEN }} \
+                    -e HUE_TOKEN=${{ secrets.HUE_TOKEN }} \
-                    -e AUTOMATION__SECRETS__NTFY_TOPIC=${{ secrets.NTFY_TOPIC }} \
+                    -e NTFY_TOPIC=${{ secrets.NTFY_TOPIC }} \
-                    git.huizinga.dev/dreaded_x/automation_rs@${{ needs.build.outputs.digest }}
+                    git.huizinga.dev/dreaded_x/automation_rs:latest
 
                   docker network connect web automation_rs
 
             - name: Start container
               run: docker start automation_rs
+
+            # TODO: Perform a healthcheck

.gitignore

@@ -1,3 +1,2 @@
 /target
 .env
-automation.toml

.pre-commit-config.yaml

@@ -1,101 +1,42 @@
-default_install_hook_types:
-  - pre-commit
-  - commit-msg
-default_stages:
-  - pre-commit
-
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v6.0.0
+    rev: v4.6.0
     hooks:
       - id: trailing-whitespace
       - id: end-of-file-fixer
       - id: check-yaml
-        args:
-          - --allow-multiple-documents
       - id: check-toml
       - id: check-added-large-files
       - id: check-merge-conflict
 
-  - repo: https://github.com/compilerla/conventional-pre-commit
+  - repo: https://github.com/doublify/pre-commit-rust
-    rev: v4.2.0
+    rev: v1.0
     hooks:
-      - id: conventional-pre-commit
+      - id: clippy
-        stages: [commit-msg]
+      - id: fmt
-        args: [--verbose]
 
   - repo: https://github.com/JohnnyMorganz/StyLua
-    rev: v2.1.0
+    rev: v0.20.0
     hooks:
       - id: stylua
 
   - repo: https://github.com/crate-ci/typos
-    rev: v1.36.1
+    rev: v1.21.0
     hooks:
       - id: typos
         args: ["--force-exclude"]
 
+  - repo: https://github.com/pryorda/dockerfilelint-precommit-hooks
+    rev: v0.1.0
+    hooks:
+      - id: dockerfilelint
+
   - repo: local
     hooks:
-      - id: fmt
+      - id: lua-definitions
-        name: fmt
+        name: Generate lua definitions
-        description: Format files with cargo fmt.
+        description: This hook generates up to date lua definitions.
-        entry: cargo +nightly fmt
+        entry: cargo run --release --bin generate_definitions
-        language: system
-        types: [rust]
-        args: ["--", "--check"]
-        # For some reason some formatting is different depending on how you invoke?
-        pass_filenames: false
-
-      - id: clippy
-        name: clippy
-        description: Lint rust sources
-        entry: cargo clippy
-        language: system
-        args: ["--", "-D", "warnings"]
-        types: [file]
-        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
-
-      - id: test
-        name: test
-        description: Rust test
-        entry: cargo test
-        language: system
-        args: ["--workspace"]
-        types: [file]
-        files: (\.rs|Cargo.lock)$
-        pass_filenames: false
-
-      - id: udeps
-        name: unused
-        description: Check for unused crates
-        entry: cargo udeps
-        args: ["--workspace"]
-        language: system
-        types: [file]
-        files: (\.rs|Cargo.lock)$
-        pass_filenames: false
-
-      - id: audit
-        name: audit
-        description: Audit packages
-        entry: cargo audit
-        args: ["--deny", "warnings"]
-        language: system
-        pass_filenames: false
-        verbose: true
-        always_run: true
-
-  - repo: https://github.com/hadolint/hadolint
-    rev: v2.13.1
-    hooks:
-      - id: hadolint

.typos.toml

@@ -0,0 +1,2 @@
+[default.extend-words]
+mosquitto = "mosquitto"

Cargo.toml

@@ -1,99 +1,56 @@
 [package]
 name = "automation"
 version = "0.1.0"
-edition = "2024"
+edition = "2021"
 default-run = "automation"
 
 [workspace]
-members = [
+members = ["google-home", "automation_macro", "automation_cast"]
-  "automation_cast",
-  "automation_devices",
-  "automation_lib",
-  "automation_macro",
-  "google_home/google_home",
-  "google_home/google_home_macro",
-]
 
-[workspace.dependencies]
+[dependencies]
-air_filter_types = { git = "https://git.huizinga.dev/Dreaded_X/airfilter", tag = "v0.4.4" }
-anyhow = "1.0.99"
-async-trait = "0.1.89"
-automation_cast = { path = "./automation_cast" }
-automation_devices = { path = "./automation_devices" }
-automation_lib = { path = "./automation_lib" }
 automation_macro = { path = "./automation_macro" }
-axum = "0.8.4"
+automation_cast = { path = "./automation_cast/" }
-bytes = "1.10.1"
+rumqttc = "0.18"
-dyn-clone = "1.0.20"
+serde = { version = "1.0.149", features = ["derive"] }
-eui48 = { version = "1.1.0", features = [
+serde_json = "1.0.89"
-  "disp_hexstring",
+google-home = { path = "./google-home" }
-  "serde",
+paste = "1.0.10"
-], default-features = false }
+tokio = { version = "1", features = ["rt-multi-thread"] }
-futures = "0.3.31"
+dotenvy = "0.15.0"
-google_home = { path = "./google_home/google_home" }
+reqwest = { version = "0.11.13", features = [
-google_home_macro = { path = "./google_home/google_home_macro" }
-hostname = "0.4.1"
-inventory = "0.3.21"
-itertools = "0.14.0"
-json_value_merge = "2.0.1"
-lua_typed = { git = "https://git.huizinga.dev/Dreaded_X/lua_typed" }
-mlua = { version = "0.11.3", features = [
-  "lua54",
-  "vendored",
-  "macros",
-  "serialize",
-  "async",
-  "send",
-] }
-proc-macro2 = "1.0.101"
-quote = "1.0.40"
-reqwest = { version = "0.12.23", features = [
   "json",
   "rustls-tls",
 ], default-features = false } # Use rustls, since the other packages also use rustls
-rumqttc = "0.24.0"
+axum = "0.6.1"
-serde = { version = "1.0.219", features = ["derive"] }
+serde_repr = "0.1.10"
-serde_json = "1.0.143"
+tracing = "0.1.37"
-serde_repr = "0.1.20"
+bytes = "1.3.0"
-syn = { version = "2.0.106" }
+pollster = "0.2.5"
-thiserror = "2.0.16"
+regex = "1.7.0"
-tokio = { version = "1", features = ["rt-multi-thread"] }
+async-trait = "0.1.61"
-tokio-cron-scheduler = "0.15.0"
+futures = "0.3.25"
-tracing = "0.1.41"
+eui48 = { version = "1.1.0", default-features = false, features = [
-tracing-subscriber = "0.3.20"
+  "disp_hexstring",
-wakey = "0.3.0"
+  "serde",
-
-[dependencies]
-anyhow = { workspace = true }
-async-trait = { workspace = true }
-automation_devices = { workspace = true }
-automation_lib = { workspace = true }
-automation_macro = { path = "./automation_macro" }
-axum = { workspace = true }
-config = { version = "0.15.15", default-features = false, features = [
-  "async",
-  "toml",
 ] }
-git-version = "0.3.9"
+thiserror = "1.0.38"
-google_home = { workspace = true }
+anyhow = "1.0.68"
-lua_typed = { workspace = true }
+wakey = "0.3.0"
-inventory = { workspace = true }
+console-subscriber = "0.1.8"
-mlua = { workspace = true }
+tracing-subscriber = "0.3.16"
-reqwest = { workspace = true }
+serde_with = "3.2.0"
-rumqttc = { workspace = true }
+enum_dispatch = "0.3.12"
-serde = { workspace = true }
+indexmap = { version = "2.0.0", features = ["serde"] }
-serde_json = { workspace = true }
+serde_yaml = "0.9.27"
-thiserror = { workspace = true }
+tokio-cron-scheduler = "0.9.4"
-tokio = { workspace = true }
+mlua = { version = "0.9.7", features = ["lua54", "vendored", "macros", "serialize", "async", "send"] }
-tokio-cron-scheduler = { workspace = true }
+once_cell = "1.19.0"
-tracing = { workspace = true }
+hostname = "0.4.0"
-tracing-subscriber = { workspace = true }
+tokio-util = { version = "0.7.11", features = ["full"] }
+uuid = "1.8.0"
 
 [patch.crates-io]
 wakey = { git = "https://git.huizinga.dev/Dreaded_X/wakey" }
 
 [profile.release]
 lto = true
-
-[package.metadata.typos.default.extend-words]
-mosquitto = "mosquitto"

Dockerfile

@@ -1,25 +1,8 @@
-FROM rust:1.89 AS base
+FROM gcr.io/distroless/cc-debian12:nonroot
-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
 
-FROM base AS planner
+ENV AUTOMATION_CONFIG=/app/config.lua
-COPY . .
+COPY ./config.lua /app/config.lua
-RUN cargo chef prepare --recipe-path recipe.json
 
-FROM base AS builder
+COPY ./build/automation /app/automation
-# HACK: Now we can use unstable feature while on stable rust!
-ENV RUSTC_BOOTSTRAP=1
-COPY --from=planner /app/recipe.json recipe.json
-RUN cargo chef cook --release --recipe-path recipe.json
 
-COPY . .
+CMD ["/app/automation"]
-ARG RELEASE_VERSION
-ENV RELEASE_VERSION=${RELEASE_VERSION}
-RUN cargo auditable build --release
-
-FROM gcr.io/distroless/cc-debian12:nonroot AS runtime
-COPY --from=builder /app/target/release/automation /app/automation
-COPY ./config /app/config
-CMD [ "/app/automation" ]

automation_cast/Cargo.toml

@@ -1,6 +1,8 @@
 [package]
 name = "automation_cast"
 version = "0.1.0"
-edition = "2024"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]

automation_cast/src/lib.rs

@@ -6,6 +6,7 @@ use std::marker::Unsize;
 
 pub trait Cast<P: ?Sized> {
     fn cast(&self) -> Option<&P>;
+    fn cast_mut(&mut self) -> Option<&mut P>;
 }
 
 impl<D, P> Cast<P> for D
@@ -15,6 +16,10 @@ where
     default fn cast(&self) -> Option<&P> {
         None
     }
+
+    default fn cast_mut(&mut self) -> Option<&mut P> {
+        None
+    }
 }
 
 impl<D, P> Cast<P> for D
@@ -25,4 +30,8 @@ where
     fn cast(&self) -> Option<&P> {
         Some(self)
     }
+
+    fn cast_mut(&mut self) -> Option<&mut P> {
+        Some(self)
+    }
 }

automation_devices/Cargo.toml

@@ -1,27 +0,0 @@
-[package]
-name = "automation_devices"
-version = "0.1.0"
-edition = "2024"
-
-[dependencies]
-air_filter_types = { workspace = true }
-anyhow = { workspace = true }
-async-trait = { workspace = true }
-automation_lib = { workspace = true }
-automation_macro = { workspace = true }
-bytes = { workspace = true }
-dyn-clone = { workspace = true }
-eui48 = { workspace = true }
-google_home = { workspace = true }
-inventory = { workspace = true }
-lua_typed = { workspace = true }
-mlua = { workspace = true }
-reqwest = { workspace = true }
-rumqttc = { workspace = true }
-serde = { workspace = true }
-serde_json = { workspace = true }
-serde_repr = { workspace = true }
-thiserror = { workspace = true }
-tokio = { workspace = true }
-tracing = { workspace = true }
-wakey = { workspace = true }

automation_devices/src/air_filter.rs

@@ -1,232 +0,0 @@
-use async_trait::async_trait;
-use automation_lib::config::InfoConfig;
-use automation_lib::device::{Device, LuaDeviceCreate};
-use automation_macro::{Device, LuaDeviceConfig};
-use google_home::device::Name;
-use google_home::errors::ErrorCode;
-use google_home::traits::{
-    AvailableSpeeds, FanSpeed, HumiditySetting, OnOff, Speed, SpeedValue, TemperatureControl,
-    TemperatureUnit,
-};
-use google_home::types::Type;
-use lua_typed::Typed;
-use thiserror::Error;
-use tracing::{debug, trace};
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "AirFilterConfig")]
-pub struct Config {
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub info: InfoConfig,
-    pub url: String,
-}
-crate::register_type!(Config);
-
-#[derive(Debug, Clone, Device)]
-#[device(traits(OnOff))]
-pub struct AirFilter {
-    config: Config,
-}
-crate::register_device!(AirFilter);
-
-#[derive(Debug, Error)]
-pub enum Error {
-    #[error("Connection error")]
-    ReqwestError(#[from] reqwest::Error),
-}
-
-impl From<Error> for google_home::errors::ErrorCode {
-    fn from(value: Error) -> Self {
-        match value {
-            // Assume that if we encounter a ReqwestError the device is offline
-            Error::ReqwestError(_) => {
-                Self::DeviceError(google_home::errors::DeviceError::DeviceOffline)
-            }
-        }
-    }
-}
-
-// TODO: Handle error properly
-impl AirFilter {
-    async fn set_fan_speed(&self, speed: air_filter_types::FanSpeed) -> Result<(), Error> {
-        let message = air_filter_types::SetFanSpeed::new(speed);
-        let url = format!("{}/state/fan", self.config.url);
-        let client = reqwest::Client::new();
-        client.put(url).json(&message).send().await?;
-
-        Ok(())
-    }
-
-    async fn get_fan_state(&self) -> Result<air_filter_types::FanState, Error> {
-        let url = format!("{}/state/fan", self.config.url);
-        Ok(reqwest::get(url).await?.json().await?)
-    }
-
-    async fn get_sensor_data(&self) -> Result<air_filter_types::SensorData, Error> {
-        let url = format!("{}/state/sensor", self.config.url);
-        Ok(reqwest::get(url).await?.json().await?)
-    }
-}
-
-#[async_trait]
-impl LuaDeviceCreate for AirFilter {
-    type Config = Config;
-    type Error = rumqttc::ClientError;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = config.info.identifier(), "Setting up AirFilter");
-
-        Ok(Self { config })
-    }
-}
-
-impl Device for AirFilter {
-    fn get_id(&self) -> String {
-        self.config.info.identifier()
-    }
-}
-
-#[async_trait]
-impl google_home::Device for AirFilter {
-    fn get_device_type(&self) -> Type {
-        Type::AirPurifier
-    }
-
-    fn get_device_name(&self) -> Name {
-        Name::new(&self.config.info.name)
-    }
-
-    fn get_id(&self) -> String {
-        Device::get_id(self)
-    }
-
-    async fn is_online(&self) -> bool {
-        self.get_sensor_data().await.is_ok()
-    }
-
-    fn get_room_hint(&self) -> Option<&str> {
-        self.config.info.room.as_deref()
-    }
-
-    fn will_report_state(&self) -> bool {
-        false
-    }
-}
-
-#[async_trait]
-impl OnOff for AirFilter {
-    async fn on(&self) -> Result<bool, ErrorCode> {
-        Ok(self.get_fan_state().await?.speed != air_filter_types::FanSpeed::Off)
-    }
-
-    async fn set_on(&self, on: bool) -> Result<(), ErrorCode> {
-        debug!("Turning on air filter: {on}");
-
-        if on {
-            self.set_fan_speed(air_filter_types::FanSpeed::High).await?;
-        } else {
-            self.set_fan_speed(air_filter_types::FanSpeed::Off).await?;
-        }
-
-        Ok(())
-    }
-}
-
-#[async_trait]
-impl FanSpeed for AirFilter {
-    fn available_fan_speeds(&self) -> AvailableSpeeds {
-        AvailableSpeeds {
-            speeds: vec![
-                Speed {
-                    speed_name: "off".into(),
-                    speed_values: vec![SpeedValue {
-                        speed_synonym: vec!["Off".into()],
-                        lang: "en".into(),
-                    }],
-                },
-                Speed {
-                    speed_name: "low".into(),
-                    speed_values: vec![SpeedValue {
-                        speed_synonym: vec!["Low".into()],
-                        lang: "en".into(),
-                    }],
-                },
-                Speed {
-                    speed_name: "medium".into(),
-                    speed_values: vec![SpeedValue {
-                        speed_synonym: vec!["Medium".into()],
-                        lang: "en".into(),
-                    }],
-                },
-                Speed {
-                    speed_name: "high".into(),
-                    speed_values: vec![SpeedValue {
-                        speed_synonym: vec!["High".into()],
-                        lang: "en".into(),
-                    }],
-                },
-            ],
-            ordered: true,
-        }
-    }
-
-    async fn current_fan_speed_setting(&self) -> Result<String, ErrorCode> {
-        let speed = self.get_fan_state().await?.speed;
-        let speed = match speed {
-            air_filter_types::FanSpeed::Off => "off",
-            air_filter_types::FanSpeed::Low => "low",
-            air_filter_types::FanSpeed::Medium => "medium",
-            air_filter_types::FanSpeed::High => "high",
-        };
-
-        Ok(speed.into())
-    }
-
-    async fn set_fan_speed(&self, fan_speed: String) -> Result<(), ErrorCode> {
-        let fan_speed = fan_speed.as_str();
-        let speed = if fan_speed == "off" {
-            air_filter_types::FanSpeed::Off
-        } else if fan_speed == "low" {
-            air_filter_types::FanSpeed::Low
-        } else if fan_speed == "medium" {
-            air_filter_types::FanSpeed::Medium
-        } else if fan_speed == "high" {
-            air_filter_types::FanSpeed::High
-        } else {
-            return Err(google_home::errors::DeviceError::TransientError.into());
-        };
-
-        self.set_fan_speed(speed).await?;
-
-        Ok(())
-    }
-}
-
-#[async_trait]
-impl HumiditySetting for AirFilter {
-    fn query_only_humidity_setting(&self) -> Option<bool> {
-        Some(true)
-    }
-
-    async fn humidity_ambient_percent(&self) -> Result<isize, ErrorCode> {
-        Ok(self.get_sensor_data().await?.humidity().round() as isize)
-    }
-}
-
-#[async_trait]
-impl TemperatureControl for AirFilter {
-    fn query_only_temperature_control(&self) -> Option<bool> {
-        Some(true)
-    }
-
-    #[allow(non_snake_case)]
-    fn temperatureUnitForUX(&self) -> TemperatureUnit {
-        TemperatureUnit::Celsius
-    }
-
-    async fn temperature_ambient_celsius(&self) -> Result<f32, ErrorCode> {
-        // HACK: Round to one decimal place
-        Ok((10.0 * self.get_sensor_data().await?.temperature()).round() / 10.0)
-    }
-}

automation_devices/src/contact_sensor.rs

@@ -1,192 +0,0 @@
-use std::sync::Arc;
-
-use async_trait::async_trait;
-use automation_lib::action_callback::ActionCallback;
-use automation_lib::config::{InfoConfig, MqttDeviceConfig};
-use automation_lib::device::{Device, LuaDeviceCreate};
-use automation_lib::error::DeviceConfigError;
-use automation_lib::event::OnMqtt;
-use automation_lib::messages::ContactMessage;
-use automation_lib::mqtt::WrappedAsyncClient;
-use automation_macro::{Device, LuaDeviceConfig};
-use google_home::device;
-use google_home::errors::{DeviceError, ErrorCode};
-use google_home::traits::OpenClose;
-use google_home::types::Type;
-use lua_typed::Typed;
-use serde::Deserialize;
-use tokio::sync::{RwLock, RwLockReadGuard, RwLockWriteGuard};
-use tracing::{debug, error, trace};
-
-#[derive(Debug, Clone, Deserialize, PartialEq, Eq, Copy, Typed)]
-pub enum SensorType {
-    Door,
-    Drawer,
-    Window,
-}
-crate::register_type!(SensorType);
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "ContactSensorConfig")]
-pub struct Config {
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub info: InfoConfig,
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub mqtt: MqttDeviceConfig,
-
-    #[device_config(default(SensorType::Window))]
-    #[typed(default)]
-    pub sensor_type: SensorType,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub callback: ActionCallback<(ContactSensor, bool)>,
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub battery_callback: ActionCallback<(ContactSensor, f32)>,
-
-    #[device_config(from_lua)]
-    #[typed(default)]
-    pub client: WrappedAsyncClient,
-}
-crate::register_type!(Config);
-
-#[derive(Debug)]
-struct State {
-    is_closed: bool,
-}
-
-#[derive(Debug, Clone, Device)]
-#[device(traits(OpenClose))]
-pub struct ContactSensor {
-    config: Config,
-    state: Arc<RwLock<State>>,
-}
-crate::register_device!(ContactSensor);
-
-impl ContactSensor {
-    async fn state(&self) -> RwLockReadGuard<'_, State> {
-        self.state.read().await
-    }
-
-    async fn state_mut(&self) -> RwLockWriteGuard<'_, State> {
-        self.state.write().await
-    }
-}
-
-#[async_trait]
-impl LuaDeviceCreate for ContactSensor {
-    type Config = Config;
-    type Error = DeviceConfigError;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = config.info.identifier(), "Setting up ContactSensor");
-
-        config
-            .client
-            .subscribe(&config.mqtt.topic, rumqttc::QoS::AtLeastOnce)
-            .await?;
-
-        let state = State { is_closed: true };
-        let state = Arc::new(RwLock::new(state));
-
-        Ok(Self { config, state })
-    }
-}
-
-impl Device for ContactSensor {
-    fn get_id(&self) -> String {
-        self.config.info.identifier()
-    }
-}
-
-#[async_trait]
-impl google_home::Device for ContactSensor {
-    fn get_device_type(&self) -> google_home::types::Type {
-        match self.config.sensor_type {
-            SensorType::Door => Type::Door,
-            SensorType::Drawer => Type::Drawer,
-            SensorType::Window => Type::Window,
-        }
-    }
-
-    fn get_id(&self) -> String {
-        Device::get_id(self)
-    }
-
-    fn get_device_name(&self) -> google_home::device::Name {
-        device::Name::new(&self.config.info.name)
-    }
-
-    fn get_room_hint(&self) -> Option<&str> {
-        self.config.info.room.as_deref()
-    }
-
-    fn will_report_state(&self) -> bool {
-        false
-    }
-
-    async fn is_online(&self) -> bool {
-        true
-    }
-}
-
-#[async_trait]
-impl OpenClose for ContactSensor {
-    fn discrete_only_open_close(&self) -> Option<bool> {
-        Some(true)
-    }
-
-    fn query_only_open_close(&self) -> Option<bool> {
-        Some(true)
-    }
-
-    async fn open_percent(&self) -> Result<u8, ErrorCode> {
-        if self.state().await.is_closed {
-            Ok(0)
-        } else {
-            Ok(100)
-        }
-    }
-
-    async fn set_open_percent(&self, _open_percent: u8) -> Result<(), ErrorCode> {
-        Err(DeviceError::ActionNotAvailable.into())
-    }
-}
-
-#[async_trait]
-impl OnMqtt for ContactSensor {
-    async fn on_mqtt(&self, message: rumqttc::Publish) {
-        if !rumqttc::matches(&message.topic, &self.config.mqtt.topic) {
-            return;
-        }
-
-        let message = match ContactMessage::try_from(message) {
-            Ok(message) => message,
-            Err(err) => {
-                error!(id = self.get_id(), "Failed to parse message: {err}");
-                return;
-            }
-        };
-
-        if let Some(is_closed) = message.contact {
-            if is_closed == self.state().await.is_closed {
-                return;
-            }
-
-            self.config.callback.call((self.clone(), !is_closed)).await;
-
-            debug!(id = self.get_id(), "Updating state to {is_closed}");
-            self.state_mut().await.is_closed = is_closed;
-        }
-
-        if let Some(battery) = message.battery {
-            self.config
-                .battery_callback
-                .call((self.clone(), battery))
-                .await;
-        }
-    }
-}

automation_devices/src/hue_group.rs

@@ -1,168 +0,0 @@
-use std::net::SocketAddr;
-
-use anyhow::Result;
-use async_trait::async_trait;
-use automation_macro::{Device, LuaDeviceConfig};
-use google_home::errors::ErrorCode;
-use google_home::traits::OnOff;
-use lua_typed::Typed;
-use tracing::{error, trace, warn};
-
-use super::{Device, LuaDeviceCreate};
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "HueGroupConfig")]
-pub struct Config {
-    pub identifier: String,
-    #[device_config(rename("ip"), with(|ip| SocketAddr::new(ip, 80)))]
-    #[typed(as = "ip")]
-    pub addr: SocketAddr,
-    pub login: String,
-    pub group_id: isize,
-    pub scene_id: String,
-}
-crate::register_type!(Config);
-
-#[derive(Debug, Clone, Device)]
-#[device(traits(OnOff))]
-pub struct HueGroup {
-    config: Config,
-}
-crate::register_device!(HueGroup);
-
-// Couple of helper function to get the correct urls
-#[async_trait]
-impl LuaDeviceCreate for HueGroup {
-    type Config = Config;
-    type Error = rumqttc::ClientError;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = config.identifier, "Setting up AudioSetup");
-
-        Ok(Self { config })
-    }
-}
-
-impl HueGroup {
-    fn url_base(&self) -> String {
-        format!("http://{}/api/{}", self.config.addr, self.config.login)
-    }
-
-    fn url_set_action(&self) -> String {
-        format!("{}/groups/{}/action", self.url_base(), self.config.group_id)
-    }
-
-    fn url_get_state(&self) -> String {
-        format!("{}/groups/{}", self.url_base(), self.config.group_id)
-    }
-}
-
-impl Device for HueGroup {
-    fn get_id(&self) -> String {
-        self.config.identifier.clone()
-    }
-}
-
-#[async_trait]
-impl OnOff for HueGroup {
-    async fn set_on(&self, on: bool) -> Result<(), ErrorCode> {
-        let message = if on {
-            message::Action::scene(self.config.scene_id.clone())
-        } else {
-            message::Action::on(false)
-        };
-
-        let res = reqwest::Client::new()
-            .put(self.url_set_action())
-            .json(&message)
-            .send()
-            .await;
-
-        match res {
-            Ok(res) => {
-                let status = res.status();
-                if !status.is_success() {
-                    warn!(id = self.get_id(), "Status code is not success: {status}");
-                }
-            }
-            Err(err) => error!(id = self.get_id(), "Error: {err}"),
-        }
-
-        Ok(())
-    }
-
-    async fn on(&self) -> Result<bool, ErrorCode> {
-        let res = reqwest::Client::new()
-            .get(self.url_get_state())
-            .send()
-            .await;
-
-        match res {
-            Ok(res) => {
-                let status = res.status();
-                if !status.is_success() {
-                    warn!(id = self.get_id(), "Status code is not success: {status}");
-                }
-
-                let on = match res.json::<message::Info>().await {
-                    Ok(info) => info.any_on(),
-                    Err(err) => {
-                        error!(id = self.get_id(), "Failed to parse message: {err}");
-                        // TODO: Error code
-                        return Ok(false);
-                    }
-                };
-
-                return Ok(on);
-            }
-            Err(err) => error!(id = self.get_id(), "Error: {err}"),
-        }
-
-        Ok(false)
-    }
-}
-
-mod message {
-    use serde::{Deserialize, Serialize};
-
-    #[derive(Debug, Serialize, Deserialize)]
-    pub struct Action {
-        #[serde(skip_serializing_if = "Option::is_none")]
-        on: Option<bool>,
-        #[serde(skip_serializing_if = "Option::is_none")]
-        scene: Option<String>,
-    }
-
-    impl Action {
-        pub fn on(on: bool) -> Self {
-            Self {
-                on: Some(on),
-                scene: None,
-            }
-        }
-
-        pub fn scene(scene: String) -> Self {
-            Self {
-                on: None,
-                scene: Some(scene),
-            }
-        }
-    }
-
-    #[derive(Debug, Serialize, Deserialize)]
-    struct State {
-        all_on: bool,
-        any_on: bool,
-    }
-
-    #[derive(Debug, Serialize, Deserialize)]
-    pub struct Info {
-        state: State,
-    }
-
-    impl Info {
-        pub fn any_on(&self) -> bool {
-            self.state.any_on
-        }
-    }
-}

automation_devices/src/hue_switch.rs

@@ -1,147 +0,0 @@
-use async_trait::async_trait;
-use automation_lib::action_callback::ActionCallback;
-use automation_lib::config::{InfoConfig, MqttDeviceConfig};
-use automation_lib::device::{Device, LuaDeviceCreate};
-use automation_lib::event::OnMqtt;
-use automation_lib::mqtt::WrappedAsyncClient;
-use automation_macro::{Device, LuaDeviceConfig};
-use lua_typed::Typed;
-use rumqttc::{Publish, matches};
-use serde::Deserialize;
-use tracing::{debug, trace, warn};
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "HueSwitchConfig")]
-pub struct Config {
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub info: InfoConfig,
-
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub mqtt: MqttDeviceConfig,
-
-    #[device_config(from_lua)]
-    pub client: WrappedAsyncClient,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub left_callback: ActionCallback<HueSwitch>,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub right_callback: ActionCallback<HueSwitch>,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub left_hold_callback: ActionCallback<HueSwitch>,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub right_hold_callback: ActionCallback<HueSwitch>,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub battery_callback: ActionCallback<(HueSwitch, f32)>,
-}
-crate::register_type!(Config);
-
-#[derive(Debug, Copy, Clone, Deserialize)]
-#[serde(rename_all = "snake_case")]
-enum Action {
-    LeftPress,
-    LeftPressRelease,
-    LeftHold,
-    LeftHoldRelease,
-    RightPress,
-    RightPressRelease,
-    RightHold,
-    RightHoldRelease,
-}
-
-#[derive(Debug, Clone, Deserialize)]
-struct State {
-    action: Option<Action>,
-    battery: Option<f32>,
-}
-
-#[derive(Debug, Clone, Device)]
-pub struct HueSwitch {
-    config: Config,
-}
-crate::register_device!(HueSwitch);
-
-impl Device for HueSwitch {
-    fn get_id(&self) -> String {
-        self.config.info.identifier()
-    }
-}
-
-#[async_trait]
-impl LuaDeviceCreate for HueSwitch {
-    type Config = Config;
-    type Error = rumqttc::ClientError;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = config.info.identifier(), "Setting up HueSwitch");
-
-        config
-            .client
-            .subscribe(&config.mqtt.topic, rumqttc::QoS::AtLeastOnce)
-            .await?;
-
-        Ok(Self { config })
-    }
-}
-
-#[async_trait]
-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(&message.topic, &self.config.mqtt.topic) {
-            let message = match serde_json::from_slice::<State>(&message.payload) {
-                Ok(message) => message,
-                Err(err) => {
-                    warn!(id = Device::get_id(self), "Failed to parse message: {err}");
-                    return;
-                }
-            };
-
-            if let Some(action) = message.action {
-                debug!(
-                    id = Device::get_id(self),
-                    ?message.action,
-                    "Action received",
-                );
-
-                match action {
-                    Action::LeftPressRelease => self.config.left_callback.call(self.clone()).await,
-                    Action::RightPressRelease => {
-                        self.config.right_callback.call(self.clone()).await
-                    }
-                    Action::LeftHold => self.config.left_hold_callback.call(self.clone()).await,
-                    Action::RightHold => self.config.right_hold_callback.call(self.clone()).await,
-                    // If there is no hold action, the switch will act like a normal release
-                    Action::RightHoldRelease => {
-                        if self.config.right_hold_callback.is_empty() {
-                            self.config.right_callback.call(self.clone()).await
-                        }
-                    }
-                    Action::LeftHoldRelease => {
-                        if self.config.left_hold_callback.is_empty() {
-                            self.config.left_callback.call(self.clone()).await
-                        }
-                    }
-                    _ => {}
-                }
-            }
-
-            if let Some(battery) = message.battery {
-                self.config
-                    .battery_callback
-                    .call((self.clone(), battery))
-                    .await;
-            }
-        }
-    }
-}

automation_devices/src/ikea_remote.rs

@@ -1,112 +0,0 @@
-use async_trait::async_trait;
-use automation_lib::action_callback::ActionCallback;
-use automation_lib::config::{InfoConfig, MqttDeviceConfig};
-use automation_lib::device::{Device, LuaDeviceCreate};
-use automation_lib::event::OnMqtt;
-use automation_lib::messages::{RemoteAction, RemoteMessage};
-use automation_lib::mqtt::WrappedAsyncClient;
-use automation_macro::{Device, LuaDeviceConfig};
-use lua_typed::Typed;
-use rumqttc::{Publish, matches};
-use tracing::{debug, error, trace};
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "IkeaRemoteConfig")]
-pub struct Config {
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub info: InfoConfig,
-
-    #[device_config(default)]
-    #[typed(default)]
-    pub single_button: bool,
-
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub mqtt: MqttDeviceConfig,
-
-    #[device_config(from_lua)]
-    pub client: WrappedAsyncClient,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub callback: ActionCallback<(IkeaRemote, bool)>,
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub battery_callback: ActionCallback<(IkeaRemote, f32)>,
-}
-crate::register_type!(Config);
-
-#[derive(Debug, Clone, Device)]
-pub struct IkeaRemote {
-    config: Config,
-}
-crate::register_device!(IkeaRemote);
-
-impl Device for IkeaRemote {
-    fn get_id(&self) -> String {
-        self.config.info.identifier()
-    }
-}
-
-#[async_trait]
-impl LuaDeviceCreate for IkeaRemote {
-    type Config = Config;
-    type Error = rumqttc::ClientError;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = config.info.identifier(), "Setting up IkeaRemote");
-
-        config
-            .client
-            .subscribe(&config.mqtt.topic, rumqttc::QoS::AtLeastOnce)
-            .await?;
-
-        Ok(Self { config })
-    }
-}
-
-#[async_trait]
-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(&message.topic, &self.config.mqtt.topic) {
-            let message = match RemoteMessage::try_from(message) {
-                Ok(message) => message,
-                Err(err) => {
-                    error!(id = Device::get_id(self), "Failed to parse message: {err}");
-                    return;
-                }
-            };
-
-            if let Some(action) = message.action {
-                debug!(id = Device::get_id(self), "Remote action = {:?}", action);
-
-                let on = if self.config.single_button {
-                    match action {
-                        RemoteAction::On => Some(true),
-                        RemoteAction::BrightnessMoveUp => Some(false),
-                        _ => None,
-                    }
-                } else {
-                    match action {
-                        RemoteAction::On => Some(true),
-                        RemoteAction::Off => Some(false),
-                        _ => None,
-                    }
-                };
-
-                if let Some(on) = on {
-                    self.config.callback.call((self.clone(), on)).await;
-                }
-            }
-
-            if let Some(battery) = message.battery {
-                self.config
-                    .battery_callback
-                    .call((self.clone(), battery))
-                    .await;
-            }
-        }
-    }
-}

automation_devices/src/lib.rs

@@ -1,130 +0,0 @@
-#![feature(iter_intersperse)]
-mod air_filter;
-mod contact_sensor;
-mod hue_bridge;
-mod hue_group;
-mod hue_switch;
-mod ikea_remote;
-mod kasa_outlet;
-mod light_sensor;
-mod ntfy;
-mod presence;
-mod wake_on_lan;
-mod washer;
-mod zigbee;
-
-use automation_lib::Module;
-use automation_lib::device::{Device, LuaDeviceCreate};
-use tracing::{debug, warn};
-
-type DeviceNameFn = fn() -> String;
-type RegisterDeviceFn = fn(lua: &mlua::Lua) -> mlua::Result<mlua::AnyUserData>;
-
-pub struct RegisteredDevice {
-    name_fn: DeviceNameFn,
-    register_fn: RegisterDeviceFn,
-}
-
-impl RegisteredDevice {
-    pub const fn new(name_fn: DeviceNameFn, register_fn: RegisterDeviceFn) -> Self {
-        Self {
-            name_fn,
-            register_fn,
-        }
-    }
-
-    pub fn get_name(&self) -> String {
-        (self.name_fn)()
-    }
-
-    pub fn register(&self, lua: &mlua::Lua) -> mlua::Result<mlua::AnyUserData> {
-        (self.register_fn)(lua)
-    }
-}
-
-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> {
-    let devices = lua.create_table()?;
-
-    debug!("Loading devices...");
-    for device in inventory::iter::<RegisteredDevice> {
-        let name = device.get_name();
-        debug!(name, "Registering device");
-        let proxy = device.register(lua)?;
-        devices.set(name, proxy)?;
-    }
-
-    Ok(devices)
-}
-
-type TypeNameFn = fn() -> String;
-type TypeDefinitionFn = fn() -> Option<String>;
-
-pub struct RegisteredType {
-    name_fn: TypeNameFn,
-    definition_fn: TypeDefinitionFn,
-}
-
-impl RegisteredType {
-    pub const fn new(name_fn: TypeNameFn, definition_fn: TypeDefinitionFn) -> Self {
-        Self {
-            name_fn,
-            definition_fn,
-        }
-    }
-
-    pub fn get_name(&self) -> String {
-        (self.name_fn)()
-    }
-
-    pub fn register(&self) -> Option<String> {
-        (self.definition_fn)()
-    }
-}
-
-macro_rules! register_type {
-    ($ty:ty) => {
-        ::inventory::submit!(crate::RegisteredType::new(
-            <$ty as ::lua_typed::Typed>::type_name,
-            <$ty as ::lua_typed::Typed>::generate_full
-        ));
-    };
-}
-pub(crate) use register_type;
-
-inventory::collect!(RegisteredType);
-
-fn generate_definitions() -> String {
-    let mut output = String::new();
-
-    let mut types: Vec<_> = inventory::iter::<RegisteredType>.into_iter().collect();
-    types.sort_by_key(|ty| ty.get_name());
-
-    output += "---@meta\n\nlocal devices\n\n";
-    for ty in types {
-        if let Some(def) = (ty.definition_fn)() {
-            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");
-        }
-    }
-    output += "return devices";
-
-    output
-}
-
-inventory::submit! {Module::new("automation:devices", create_module, Some(generate_definitions))}

automation_devices/src/light_sensor.rs

@@ -1,127 +0,0 @@
-use std::sync::Arc;
-
-use async_trait::async_trait;
-use automation_lib::action_callback::ActionCallback;
-use automation_lib::config::MqttDeviceConfig;
-use automation_lib::device::{Device, LuaDeviceCreate};
-use automation_lib::event::OnMqtt;
-use automation_lib::messages::BrightnessMessage;
-use automation_lib::mqtt::WrappedAsyncClient;
-use automation_macro::{Device, LuaDeviceConfig};
-use lua_typed::Typed;
-use rumqttc::Publish;
-use tokio::sync::{RwLock, RwLockReadGuard, RwLockWriteGuard};
-use tracing::{debug, trace, warn};
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "LightSensorConfig")]
-pub struct Config {
-    pub identifier: String,
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub mqtt: MqttDeviceConfig,
-    pub min: isize,
-    pub max: isize,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub callback: ActionCallback<(LightSensor, bool)>,
-
-    #[device_config(from_lua)]
-    pub client: WrappedAsyncClient,
-}
-crate::register_type!(Config);
-
-const DEFAULT: bool = false;
-
-#[derive(Debug)]
-pub struct State {
-    is_dark: bool,
-}
-
-#[derive(Debug, Clone, Device)]
-pub struct LightSensor {
-    config: Config,
-    state: Arc<RwLock<State>>,
-}
-crate::register_device!(LightSensor);
-
-impl LightSensor {
-    async fn state(&self) -> RwLockReadGuard<'_, State> {
-        self.state.read().await
-    }
-
-    async fn state_mut(&self) -> RwLockWriteGuard<'_, State> {
-        self.state.write().await
-    }
-}
-
-#[async_trait]
-impl LuaDeviceCreate for LightSensor {
-    type Config = Config;
-    type Error = rumqttc::ClientError;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = config.identifier, "Setting up LightSensor");
-
-        config
-            .client
-            .subscribe(&config.mqtt.topic, rumqttc::QoS::AtLeastOnce)
-            .await?;
-
-        let state = State { is_dark: DEFAULT };
-        let state = Arc::new(RwLock::new(state));
-
-        Ok(Self { config, state })
-    }
-}
-
-impl Device for LightSensor {
-    fn get_id(&self) -> String {
-        self.config.identifier.clone()
-    }
-}
-
-#[async_trait]
-impl OnMqtt for LightSensor {
-    async fn on_mqtt(&self, message: Publish) {
-        if !rumqttc::matches(&message.topic, &self.config.mqtt.topic) {
-            return;
-        }
-
-        let illuminance = match BrightnessMessage::try_from(message) {
-            Ok(state) => state.illuminance(),
-            Err(err) => {
-                warn!("Failed to parse message: {err}");
-                return;
-            }
-        };
-
-        // TODO: Move this logic to lua at some point
-        debug!("Illuminance: {illuminance}");
-        let is_dark = if illuminance <= self.config.min {
-            trace!("It is dark");
-            true
-        } else if illuminance >= self.config.max {
-            trace!("It is light");
-            false
-        } else {
-            let is_dark = self.state().await.is_dark;
-            trace!(
-                "In between min ({}) and max ({}) value, keeping current state: {}",
-                self.config.min, self.config.max, is_dark
-            );
-            is_dark
-        };
-
-        if is_dark != self.state().await.is_dark {
-            debug!("Dark state has changed: {is_dark}");
-            self.state_mut().await.is_dark = is_dark;
-
-            self.config
-                .callback
-                .call((self.clone(), !self.state().await.is_dark))
-                .await;
-        }
-    }
-}

automation_devices/src/ntfy.rs

@@ -1,161 +0,0 @@
-use std::collections::HashMap;
-use std::convert::Infallible;
-
-use async_trait::async_trait;
-use automation_lib::device::{Device, LuaDeviceCreate};
-use automation_lib::lua::traits::PartialUserData;
-use automation_macro::{Device, LuaDeviceConfig};
-use lua_typed::Typed;
-use mlua::LuaSerdeExt;
-use serde::{Deserialize, Serialize};
-use serde_repr::*;
-use tracing::{error, trace, warn};
-
-#[derive(Debug, Serialize_repr, Deserialize, Clone, Copy, Typed)]
-#[repr(u8)]
-#[serde(rename_all = "snake_case")]
-#[typed(rename_all = "snake_case")]
-pub enum Priority {
-    Min = 1,
-    Low,
-    Default,
-    High,
-    Max,
-}
-crate::register_type!(Priority);
-
-#[derive(Debug, Serialize, Deserialize, Clone, Typed)]
-#[serde(rename_all = "snake_case", tag = "action")]
-#[typed(rename_all = "snake_case", tag = "action")]
-pub enum ActionType {
-    Broadcast {
-        #[serde(skip_serializing_if = "HashMap::is_empty")]
-        #[serde(default)]
-        #[typed(default)]
-        extras: HashMap<String, String>,
-    },
-    // View,
-    // Http
-}
-
-#[derive(Debug, Serialize, Deserialize, Clone, Typed)]
-pub struct Action {
-    #[serde(flatten)]
-    #[typed(flatten)]
-    pub action: ActionType,
-    pub label: String,
-    pub clear: Option<bool>,
-}
-crate::register_type!(Action);
-
-#[derive(Serialize, Deserialize, Typed)]
-struct NotificationFinal {
-    topic: String,
-    #[serde(flatten)]
-    #[typed(flatten)]
-    inner: Notification,
-}
-
-#[derive(Debug, Serialize, Clone, Deserialize, Typed)]
-pub struct Notification {
-    title: String,
-    message: Option<String>,
-    #[serde(skip_serializing_if = "Vec::is_empty", default = "Default::default")]
-    #[typed(default)]
-    tags: Vec<String>,
-    #[serde(skip_serializing_if = "Option::is_none")]
-    priority: Option<Priority>,
-    #[serde(skip_serializing_if = "Vec::is_empty", default = "Default::default")]
-    #[typed(default)]
-    actions: Vec<Action>,
-}
-crate::register_type!(Notification);
-
-impl Notification {
-    fn finalize(self, topic: &str) -> NotificationFinal {
-        NotificationFinal {
-            topic: topic.into(),
-            inner: self,
-        }
-    }
-}
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "NtfyConfig")]
-pub struct Config {
-    #[device_config(default("https://ntfy.sh".into()))]
-    #[typed(default)]
-    pub url: String,
-    pub topic: String,
-}
-crate::register_type!(Config);
-
-#[derive(Debug, Clone, Device)]
-#[device(extra_user_data = SendNotification)]
-pub struct Ntfy {
-    config: Config,
-}
-crate::register_device!(Ntfy);
-
-struct SendNotification;
-impl PartialUserData<Ntfy> for SendNotification {
-    fn add_methods<M: mlua::UserDataMethods<Ntfy>>(methods: &mut M) {
-        methods.add_async_method(
-            "send_notification",
-            async |lua, this, notification: mlua::Value| {
-                let notification: Notification = lua.from_value(notification)?;
-
-                this.send(notification).await;
-
-                Ok(())
-            },
-        );
-    }
-
-    fn definitions() -> Option<String> {
-        Some(format!(
-            "---@async\n---@param notification {}\nfunction {}:send_notification(notification) end\n",
-            <Notification as Typed>::type_name(),
-            <Ntfy as Typed>::type_name(),
-        ))
-    }
-}
-
-#[async_trait]
-impl LuaDeviceCreate for Ntfy {
-    type Config = Config;
-    type Error = Infallible;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = "ntfy", "Setting up Ntfy");
-        Ok(Self { config })
-    }
-}
-
-impl Device for Ntfy {
-    fn get_id(&self) -> String {
-        "ntfy".to_string()
-    }
-}
-
-impl Ntfy {
-    async fn send(&self, notification: Notification) {
-        let notification = notification.finalize(&self.config.topic);
-
-        // Create the request
-        let res = reqwest::Client::new()
-            .post(self.config.url.clone())
-            .json(&notification)
-            .send()
-            .await;
-
-        if let Err(err) = res {
-            error!("Something went wrong while sending the notification: {err}");
-        } else if let Ok(res) = res {
-            let status = res.status();
-            if !status.is_success() {
-                warn!("Received status {status} when sending notification");
-            }
-        }
-    }
-}

automation_devices/src/presence.rs

@@ -1,149 +0,0 @@
-use std::collections::HashMap;
-use std::sync::Arc;
-
-use async_trait::async_trait;
-use automation_lib::action_callback::ActionCallback;
-use automation_lib::config::MqttDeviceConfig;
-use automation_lib::device::{Device, LuaDeviceCreate};
-use automation_lib::event::OnMqtt;
-use automation_lib::lua::traits::PartialUserData;
-use automation_lib::messages::PresenceMessage;
-use automation_lib::mqtt::WrappedAsyncClient;
-use automation_macro::{Device, LuaDeviceConfig};
-use lua_typed::Typed;
-use rumqttc::Publish;
-use tokio::sync::{RwLock, RwLockReadGuard, RwLockWriteGuard};
-use tracing::{debug, trace, warn};
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "PresenceConfig")]
-pub struct Config {
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub mqtt: MqttDeviceConfig,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub callback: ActionCallback<(Presence, bool)>,
-
-    #[device_config(from_lua)]
-    pub client: WrappedAsyncClient,
-}
-crate::register_type!(Config);
-
-pub const DEFAULT_PRESENCE: bool = false;
-
-#[derive(Debug)]
-pub struct State {
-    devices: HashMap<String, bool>,
-    current_overall_presence: bool,
-}
-
-#[derive(Debug, Clone, Device)]
-#[device(extra_user_data = OverallPresence)]
-pub struct Presence {
-    config: Config,
-    state: Arc<RwLock<State>>,
-}
-crate::register_device!(Presence);
-
-struct OverallPresence;
-impl PartialUserData<Presence> for OverallPresence {
-    fn add_methods<M: mlua::UserDataMethods<Presence>>(methods: &mut M) {
-        methods.add_async_method("overall_presence", async |_lua, this, ()| {
-            Ok(this.state().await.current_overall_presence)
-        });
-    }
-
-    fn definitions() -> Option<String> {
-        Some(format!(
-            "---@async\n---@return boolean\nfunction {}:overall_presence() end\n",
-            <Presence as Typed>::type_name(),
-        ))
-    }
-}
-
-impl Presence {
-    async fn state(&self) -> RwLockReadGuard<'_, State> {
-        self.state.read().await
-    }
-
-    async fn state_mut(&self) -> RwLockWriteGuard<'_, State> {
-        self.state.write().await
-    }
-}
-
-#[async_trait]
-impl LuaDeviceCreate for Presence {
-    type Config = Config;
-    type Error = rumqttc::ClientError;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = "presence", "Setting up Presence");
-
-        config
-            .client
-            .subscribe(&config.mqtt.topic, rumqttc::QoS::AtLeastOnce)
-            .await?;
-
-        let state = State {
-            devices: HashMap::new(),
-            current_overall_presence: DEFAULT_PRESENCE,
-        };
-        let state = Arc::new(RwLock::new(state));
-
-        Ok(Self { config, state })
-    }
-}
-
-impl Device for Presence {
-    fn get_id(&self) -> String {
-        "presence".to_string()
-    }
-}
-
-#[async_trait]
-impl OnMqtt for Presence {
-    async fn on_mqtt(&self, message: Publish) {
-        if !rumqttc::matches(&message.topic, &self.config.mqtt.topic) {
-            return;
-        }
-
-        let offset = self
-            .config
-            .mqtt
-            .topic
-            .find('+')
-            .or(self.config.mqtt.topic.find('#'))
-            .expect("Presence::create fails if it does not contain wildcards");
-        let device_name = message.topic[offset..].into();
-
-        if message.payload.is_empty() {
-            // Remove the device from the map
-            debug!("State of device [{device_name}] has been removed");
-            self.state_mut().await.devices.remove(&device_name);
-        } else {
-            let present = match PresenceMessage::try_from(message) {
-                Ok(state) => state.presence(),
-                Err(err) => {
-                    warn!("Failed to parse message: {err}");
-                    return;
-                }
-            };
-
-            debug!("State of device [{device_name}] has changed: {}", present);
-            self.state_mut().await.devices.insert(device_name, present);
-        }
-
-        let overall_presence = self.state().await.devices.iter().any(|(_, v)| *v);
-        if overall_presence != self.state().await.current_overall_presence {
-            debug!("Overall presence updated: {overall_presence}");
-            self.state_mut().await.current_overall_presence = overall_presence;
-
-            self.config
-                .callback
-                .call((self.clone(), overall_presence))
-                .await;
-        }
-    }
-}

automation_devices/src/zigbee/light.rs

@@ -1,447 +0,0 @@
-use std::fmt::Debug;
-use std::ops::Deref;
-use std::sync::Arc;
-
-use anyhow::Result;
-use async_trait::async_trait;
-use automation_lib::action_callback::ActionCallback;
-use automation_lib::config::{InfoConfig, MqttDeviceConfig};
-use automation_lib::device::{Device, LuaDeviceCreate};
-use automation_lib::event::OnMqtt;
-use automation_lib::helpers::serialization::state_deserializer;
-use automation_lib::mqtt::WrappedAsyncClient;
-use automation_macro::{Device, LuaDeviceConfig, LuaSerialize};
-use google_home::device;
-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, matches};
-use serde::{Deserialize, Serialize};
-use serde_json::json;
-use tokio::sync::{RwLock, RwLockReadGuard, RwLockWriteGuard};
-use tracing::{debug, trace, warn};
-
-pub trait LightState:
-    Debug + Clone + Default + Sync + Send + Serialize + Into<StateOnOff> + Typed + 'static
-{
-}
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "ConfigLight")]
-pub struct Config<T: LightState>
-where
-    Light<T>: Typed,
-{
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub info: InfoConfig,
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub mqtt: MqttDeviceConfig,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub callback: ActionCallback<(Light<T>, T)>,
-
-    #[device_config(from_lua)]
-    #[typed(default)]
-    pub client: WrappedAsyncClient,
-}
-crate::register_type!(Config<StateOnOff>);
-crate::register_type!(Config<StateBrightness>);
-crate::register_type!(Config<StateColorTemperature>);
-
-#[derive(Debug, Clone, Default, Serialize, Deserialize, LuaSerialize, Typed)]
-#[typed(as = "LightStateOnOff")]
-pub struct StateOnOff {
-    #[serde(deserialize_with = "state_deserializer")]
-    state: bool,
-}
-
-impl LightState for StateOnOff {}
-crate::register_type!(StateOnOff);
-
-#[derive(Debug, Clone, Default, Serialize, Deserialize, LuaSerialize, Typed)]
-#[typed(as = "LightStateBrightness")]
-pub struct StateBrightness {
-    #[serde(deserialize_with = "state_deserializer")]
-    state: bool,
-    brightness: f32,
-}
-
-impl LightState for StateBrightness {}
-crate::register_type!(StateBrightness);
-
-impl From<StateBrightness> for StateOnOff {
-    fn from(state: StateBrightness) -> Self {
-        StateOnOff { state: state.state }
-    }
-}
-
-#[derive(Debug, Clone, Default, Serialize, Deserialize, LuaSerialize, Typed)]
-#[typed(as = "LightStateColorTemperature")]
-pub struct StateColorTemperature {
-    #[serde(deserialize_with = "state_deserializer")]
-    state: bool,
-    brightness: f32,
-    color_temp: u32,
-}
-crate::register_type!(StateColorTemperature);
-
-impl LightState for StateColorTemperature {}
-
-impl From<StateColorTemperature> for StateOnOff {
-    fn from(state: StateColorTemperature) -> Self {
-        StateOnOff { state: state.state }
-    }
-}
-
-impl From<StateColorTemperature> for StateBrightness {
-    fn from(state: StateColorTemperature) -> Self {
-        StateBrightness {
-            state: state.state,
-            brightness: state.brightness,
-        }
-    }
-}
-
-#[derive(Debug, Clone, Device)]
-#[device(traits(OnOff for LightOnOff, LightBrightness, LightColorTemperature))]
-#[device(traits(Brightness for LightBrightness, LightColorTemperature))]
-#[device(traits(ColorSetting for LightColorTemperature))]
-pub struct Light<T: LightState>
-where
-    Light<T>: Typed,
-{
-    config: Config<T>,
-
-    state: Arc<RwLock<T>>,
-}
-
-pub type LightOnOff = Light<StateOnOff>;
-crate::register_device!(LightOnOff);
-
-pub type LightBrightness = Light<StateBrightness>;
-crate::register_device!(LightBrightness);
-
-pub type LightColorTemperature = Light<StateColorTemperature>;
-crate::register_device!(LightColorTemperature);
-
-impl<T: LightState> Light<T>
-where
-    Light<T>: Typed,
-{
-    async fn state(&self) -> RwLockReadGuard<'_, T> {
-        self.state.read().await
-    }
-
-    async fn state_mut(&self) -> RwLockWriteGuard<'_, T> {
-        self.state.write().await
-    }
-}
-
-#[async_trait]
-impl<T: LightState> LuaDeviceCreate for Light<T>
-where
-    Light<T>: Typed,
-{
-    type Config = Config<T>;
-    type Error = rumqttc::ClientError;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = config.info.identifier(), "Setting up IkeaOutlet");
-
-        config
-            .client
-            .subscribe(&config.mqtt.topic, rumqttc::QoS::AtLeastOnce)
-            .await?;
-
-        Ok(Self {
-            config,
-            state: Default::default(),
-        })
-    }
-}
-
-impl<T: LightState> Device for Light<T>
-where
-    Light<T>: Typed,
-{
-    fn get_id(&self) -> String {
-        self.config.info.identifier()
-    }
-}
-
-#[async_trait]
-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(&message.topic, &self.config.mqtt.topic) {
-            let state = match serde_json::from_slice::<StateOnOff>(&message.payload) {
-                Ok(state) => state,
-                Err(err) => {
-                    warn!(id = Device::get_id(self), "Failed to parse message: {err}");
-                    return;
-                }
-            };
-
-            // No need to do anything if the state has not changed
-            if state.state == self.state().await.state {
-                return;
-            }
-
-            self.state_mut().await.state = state.state;
-            debug!(
-                id = Device::get_id(self),
-                "Updating state to {:?}",
-                self.state().await
-            );
-
-            self.config
-                .callback
-                .call((self.clone(), self.state().await.clone()))
-                .await;
-        }
-    }
-}
-
-#[async_trait]
-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(&message.topic, &self.config.mqtt.topic) {
-            let state = match serde_json::from_slice::<StateBrightness>(&message.payload) {
-                Ok(state) => state,
-                Err(err) => {
-                    warn!(id = Device::get_id(self), "Failed to parse message: {err}");
-                    return;
-                }
-            };
-
-            {
-                let current_state = self.state().await;
-                // No need to do anything if the state has not changed
-                if state.state == current_state.state
-                    && state.brightness == current_state.brightness
-                {
-                    return;
-                }
-            }
-
-            self.state_mut().await.state = state.state;
-            self.state_mut().await.brightness = state.brightness;
-            debug!(
-                id = Device::get_id(self),
-                "Updating state to {:?}",
-                self.state().await
-            );
-
-            self.config
-                .callback
-                .call((self.clone(), self.state().await.clone()))
-                .await;
-        }
-    }
-}
-
-#[async_trait]
-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(&message.topic, &self.config.mqtt.topic) {
-            let state = match serde_json::from_slice::<StateColorTemperature>(&message.payload) {
-                Ok(state) => state,
-                Err(err) => {
-                    warn!(id = Device::get_id(self), "Failed to parse message: {err}");
-                    return;
-                }
-            };
-
-            {
-                let current_state = self.state().await;
-                // No need to do anything if the state has not changed
-                if state.state == current_state.state
-                    && state.brightness == current_state.brightness
-                    && state.color_temp == current_state.color_temp
-                {
-                    return;
-                }
-            }
-
-            self.state_mut().await.state = state.state;
-            self.state_mut().await.brightness = state.brightness;
-            self.state_mut().await.color_temp = state.color_temp;
-            debug!(
-                id = Device::get_id(self),
-                "Updating state to {:?}",
-                self.state().await
-            );
-
-            self.config
-                .callback
-                .call((self.clone(), self.state().await.clone()))
-                .await;
-        }
-    }
-}
-
-#[async_trait]
-impl<T: LightState> google_home::Device for Light<T>
-where
-    Light<T>: Typed,
-{
-    fn get_device_type(&self) -> Type {
-        Type::Light
-    }
-
-    fn get_device_name(&self) -> device::Name {
-        device::Name::new(&self.config.info.name)
-    }
-
-    fn get_id(&self) -> String {
-        Device::get_id(self)
-    }
-
-    async fn is_online(&self) -> bool {
-        true
-    }
-
-    fn get_room_hint(&self) -> Option<&str> {
-        self.config.info.room.as_deref()
-    }
-
-    fn will_report_state(&self) -> bool {
-        // TODO: Implement state reporting
-        false
-    }
-}
-
-#[async_trait]
-impl<T> OnOff for Light<T>
-where
-    T: LightState,
-    Light<T>: Typed,
-{
-    async fn on(&self) -> Result<bool, ErrorCode> {
-        let state = self.state().await;
-        let state: StateOnOff = state.deref().clone().into();
-        Ok(state.state)
-    }
-
-    async fn set_on(&self, on: bool) -> Result<(), ErrorCode> {
-        let message = json!({
-            "state": if on { "ON" } else { "OFF"}
-        });
-
-        debug!(id = Device::get_id(self), "{message}");
-
-        let topic = format!("{}/set", self.config.mqtt.topic);
-        // TODO: Handle potential errors here
-        self.config
-            .client
-            .publish(
-                &topic,
-                rumqttc::QoS::AtLeastOnce,
-                false,
-                serde_json::to_string(&message).unwrap(),
-            )
-            .await
-            .map_err(|err| warn!("Failed to update state on {topic}: {err}"))
-            .ok();
-
-        Ok(())
-    }
-}
-
-const FACTOR: f32 = 30.0;
-
-#[async_trait]
-impl<T> Brightness for Light<T>
-where
-    T: LightState,
-    T: Into<StateBrightness>,
-    Light<T>: Typed,
-{
-    async fn brightness(&self) -> Result<u8, ErrorCode> {
-        let state = self.state().await;
-        let state: StateBrightness = state.deref().clone().into();
-        let brightness =
-            100.0 * f32::log10(state.brightness / FACTOR + 1.0) / f32::log10(254.0 / FACTOR + 1.0);
-
-        Ok(brightness.clamp(0.0, 100.0).round() as u8)
-    }
-
-    async fn set_brightness(&self, brightness: u8) -> Result<(), ErrorCode> {
-        let brightness =
-            FACTOR * ((FACTOR / (FACTOR + 254.0)).powf(-(brightness as f32) / 100.0) - 1.0);
-
-        let message = json!({
-            "brightness": brightness.clamp(0.0, 254.0).round() as u8
-        });
-
-        let topic = format!("{}/set", self.config.mqtt.topic);
-        // TODO: Handle potential errors here
-        self.config
-            .client
-            .publish(
-                &topic,
-                rumqttc::QoS::AtLeastOnce,
-                false,
-                serde_json::to_string(&message).unwrap(),
-            )
-            .await
-            .map_err(|err| warn!("Failed to update state on {topic}: {err}"))
-            .ok();
-
-        Ok(())
-    }
-}
-
-#[async_trait]
-impl<T> ColorSetting for Light<T>
-where
-    T: LightState,
-    T: Into<StateColorTemperature>,
-    Light<T>: Typed,
-{
-    fn color_temperature_range(&self) -> ColorTemperatureRange {
-        ColorTemperatureRange {
-            temperature_min_k: 2200,
-            temperature_max_k: 4000,
-        }
-    }
-
-    async fn color(&self) -> Color {
-        let state = self.state().await;
-        let state: StateColorTemperature = state.deref().clone().into();
-
-        let temperature = 1_000_000 / state.color_temp;
-
-        Color { temperature }
-    }
-
-    async fn set_color(&self, color: Color) -> Result<(), ErrorCode> {
-        let temperature = 1_000_000 / color.temperature;
-
-        let message = json!({
-            "color_temp": temperature,
-        });
-
-        let topic = format!("{}/set", self.config.mqtt.topic);
-        // TODO: Handle potential errors here
-        self.config
-            .client
-            .publish(
-                &topic,
-                rumqttc::QoS::AtLeastOnce,
-                false,
-                serde_json::to_string(&message).unwrap(),
-            )
-            .await
-            .map_err(|err| warn!("Failed to update state on {topic}: {err}"))
-            .ok();
-
-        Ok(())
-    }
-}

automation_devices/src/zigbee/mod.rs

@@ -1,2 +0,0 @@
-pub mod light;
-pub mod outlet;

automation_devices/src/zigbee/outlet.rs

@@ -1,297 +0,0 @@
-use std::fmt::Debug;
-use std::ops::Deref;
-use std::sync::Arc;
-
-use anyhow::Result;
-use async_trait::async_trait;
-use automation_lib::action_callback::ActionCallback;
-use automation_lib::config::{InfoConfig, MqttDeviceConfig};
-use automation_lib::device::{Device, LuaDeviceCreate};
-use automation_lib::event::OnMqtt;
-use automation_lib::helpers::serialization::state_deserializer;
-use automation_lib::mqtt::WrappedAsyncClient;
-use automation_macro::{Device, LuaDeviceConfig, LuaSerialize};
-use google_home::device;
-use google_home::errors::ErrorCode;
-use google_home::traits::OnOff;
-use google_home::types::Type;
-use lua_typed::Typed;
-use rumqttc::{Publish, matches};
-use serde::{Deserialize, Serialize};
-use serde_json::json;
-use tokio::sync::{RwLock, RwLockReadGuard, RwLockWriteGuard};
-use tracing::{debug, trace, warn};
-
-pub trait OutletState:
-    Debug + Clone + Default + Sync + Send + Serialize + Into<StateOnOff> + Typed + 'static
-{
-}
-
-#[derive(Debug, Clone, Deserialize, PartialEq, Eq, Copy, Typed)]
-pub enum OutletType {
-    Outlet,
-    Kettle,
-}
-crate::register_type!(OutletType);
-
-impl From<OutletType> for Type {
-    fn from(outlet: OutletType) -> Self {
-        match outlet {
-            OutletType::Outlet => Type::Outlet,
-            OutletType::Kettle => Type::Kettle,
-        }
-    }
-}
-
-#[derive(Debug, Clone, LuaDeviceConfig, Typed)]
-#[typed(as = "ConfigOutlet")]
-pub struct Config<T: OutletState>
-where
-    Outlet<T>: Typed,
-{
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub info: InfoConfig,
-    #[device_config(flatten)]
-    #[typed(flatten)]
-    pub mqtt: MqttDeviceConfig,
-    #[device_config(default(OutletType::Outlet))]
-    #[typed(default)]
-    pub outlet_type: OutletType,
-
-    #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub callback: ActionCallback<(Outlet<T>, T)>,
-
-    #[device_config(from_lua)]
-    pub client: WrappedAsyncClient,
-}
-crate::register_type!(Config<StateOnOff>);
-crate::register_type!(Config<StatePower>);
-
-#[derive(Debug, Clone, Default, Serialize, Deserialize, LuaSerialize, Typed)]
-#[typed(as = "OutletStateOnOff")]
-pub struct StateOnOff {
-    #[serde(deserialize_with = "state_deserializer")]
-    state: bool,
-}
-crate::register_type!(StateOnOff);
-
-impl OutletState for StateOnOff {}
-
-#[derive(Debug, Clone, Default, Serialize, Deserialize, LuaSerialize, Typed)]
-#[typed(as = "OutletStatePower")]
-pub struct StatePower {
-    #[serde(deserialize_with = "state_deserializer")]
-    state: bool,
-    power: f64,
-}
-crate::register_type!(StatePower);
-
-impl OutletState for StatePower {}
-
-impl From<StatePower> for StateOnOff {
-    fn from(state: StatePower) -> Self {
-        StateOnOff { state: state.state }
-    }
-}
-
-#[derive(Debug, Clone, Device)]
-#[device(traits(OnOff for OutletOnOff, OutletPower))]
-pub struct Outlet<T: OutletState>
-where
-    Outlet<T>: Typed,
-{
-    config: Config<T>,
-
-    state: Arc<RwLock<T>>,
-}
-
-pub type OutletOnOff = Outlet<StateOnOff>;
-crate::register_device!(OutletOnOff);
-
-pub type OutletPower = Outlet<StatePower>;
-crate::register_device!(OutletPower);
-
-impl<T: OutletState> Outlet<T>
-where
-    Outlet<T>: Typed,
-{
-    async fn state(&self) -> RwLockReadGuard<'_, T> {
-        self.state.read().await
-    }
-
-    async fn state_mut(&self) -> RwLockWriteGuard<'_, T> {
-        self.state.write().await
-    }
-}
-
-#[async_trait]
-impl<T: OutletState> LuaDeviceCreate for Outlet<T>
-where
-    Outlet<T>: Typed,
-{
-    type Config = Config<T>;
-    type Error = rumqttc::ClientError;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error> {
-        trace!(id = config.info.identifier(), "Setting up IkeaOutlet");
-
-        config
-            .client
-            .subscribe(&config.mqtt.topic, rumqttc::QoS::AtLeastOnce)
-            .await?;
-
-        Ok(Self {
-            config,
-            state: Default::default(),
-        })
-    }
-}
-
-impl<T: OutletState> Device for Outlet<T>
-where
-    Outlet<T>: Typed,
-{
-    fn get_id(&self) -> String {
-        self.config.info.identifier()
-    }
-}
-
-#[async_trait]
-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(&message.topic, &self.config.mqtt.topic) {
-            let state = match serde_json::from_slice::<StateOnOff>(&message.payload) {
-                Ok(state) => state,
-                Err(err) => {
-                    warn!(id = Device::get_id(self), "Failed to parse message: {err}");
-                    return;
-                }
-            };
-
-            // No need to do anything if the state has not changed
-            if state.state == self.state().await.state {
-                return;
-            }
-
-            self.state_mut().await.state = state.state;
-            debug!(
-                id = Device::get_id(self),
-                "Updating state to {:?}",
-                self.state().await
-            );
-
-            self.config
-                .callback
-                .call((self.clone(), self.state().await.clone()))
-                .await;
-        }
-    }
-}
-
-#[async_trait]
-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(&message.topic, &self.config.mqtt.topic) {
-            let state = match serde_json::from_slice::<StatePower>(&message.payload) {
-                Ok(state) => state,
-                Err(err) => {
-                    warn!(id = Device::get_id(self), "Failed to parse message: {err}");
-                    return;
-                }
-            };
-
-            {
-                let current_state = self.state().await;
-                // No need to do anything if the state has not changed
-                if state.state == current_state.state && state.power == current_state.power {
-                    return;
-                }
-            }
-
-            self.state_mut().await.state = state.state;
-            self.state_mut().await.power = state.power;
-            debug!(
-                id = Device::get_id(self),
-                "Updating state to {:?}",
-                self.state().await
-            );
-
-            self.config
-                .callback
-                .call((self.clone(), self.state().await.clone()))
-                .await;
-        }
-    }
-}
-
-#[async_trait]
-impl<T: OutletState> google_home::Device for Outlet<T>
-where
-    Outlet<T>: Typed,
-{
-    fn get_device_type(&self) -> Type {
-        self.config.outlet_type.into()
-    }
-
-    fn get_device_name(&self) -> device::Name {
-        device::Name::new(&self.config.info.name)
-    }
-
-    fn get_id(&self) -> String {
-        Device::get_id(self)
-    }
-
-    async fn is_online(&self) -> bool {
-        true
-    }
-
-    fn get_room_hint(&self) -> Option<&str> {
-        self.config.info.room.as_deref()
-    }
-
-    fn will_report_state(&self) -> bool {
-        // TODO: Implement state reporting
-        false
-    }
-}
-
-#[async_trait]
-impl<T> OnOff for Outlet<T>
-where
-    T: OutletState,
-    Outlet<T>: Typed,
-{
-    async fn on(&self) -> Result<bool, ErrorCode> {
-        let state = self.state().await;
-        let state: StateOnOff = state.deref().clone().into();
-        Ok(state.state)
-    }
-
-    async fn set_on(&self, on: bool) -> Result<(), ErrorCode> {
-        let message = json!({
-            "state": if on { "ON" } else { "OFF"}
-        });
-
-        debug!(id = Device::get_id(self), "{message}");
-
-        let topic = format!("{}/set", self.config.mqtt.topic);
-        // TODO: Handle potential errors here
-        self.config
-            .client
-            .publish(
-                &topic,
-                rumqttc::QoS::AtLeastOnce,
-                false,
-                serde_json::to_string(&message).unwrap(),
-            )
-            .await
-            .map_err(|err| warn!("Failed to update state on {topic}: {err}"))
-            .ok();
-
-        Ok(())
-    }
-}

automation_lib/Cargo.toml

@@ -1,23 +0,0 @@
-[package]
-name = "automation_lib"
-version = "0.1.0"
-edition = "2024"
-
-[dependencies]
-automation_macro = { workspace = true }
-async-trait = { workspace = true }
-automation_cast = { workspace = true }
-bytes = { workspace = true }
-dyn-clone = { workspace = true }
-futures = { workspace = true }
-google_home = { workspace = true }
-hostname = { workspace = true }
-inventory = { workspace = true }
-lua_typed = { workspace = true }
-mlua = { workspace = true }
-rumqttc = { workspace = true }
-serde = { workspace = true }
-serde_json = { workspace = true }
-thiserror = { workspace = true }
-tokio = { workspace = true }
-tracing = { workspace = true }

automation_lib/src/action_callback.rs

@@ -1,94 +0,0 @@
-use std::marker::PhantomData;
-
-use futures::future::try_join_all;
-use lua_typed::Typed;
-use mlua::{FromLua, IntoLuaMulti};
-
-#[derive(Debug, Clone)]
-pub struct ActionCallback<P> {
-    callbacks: Vec<mlua::Function>,
-    _parameters: PhantomData<P>,
-}
-
-impl Typed for ActionCallback<()> {
-    fn type_name() -> String {
-        "fun() | fun()[]".into()
-    }
-}
-
-impl<A: Typed> Typed for ActionCallback<A> {
-    fn type_name() -> String {
-        let type_name = A::type_name();
-        format!("fun(_: {type_name}) | fun(_: {type_name})[]")
-    }
-}
-
-impl<A: Typed, B: Typed> Typed for ActionCallback<(A, B)> {
-    fn type_name() -> String {
-        let type_name_a = A::type_name();
-        let type_name_b = B::type_name();
-        format!(
-            "fun(_: {type_name_a}, _: {type_name_b}) | fun(_: {type_name_a}, _: {type_name_b})[]"
-        )
-    }
-}
-
-// NOTE: For some reason the derive macro combined with PhantomData leads to issues where it
-// requires all types part of P to implement default, even if they never actually get constructed.
-// By manually implemented Default it works fine.
-impl<P> Default for ActionCallback<P> {
-    fn default() -> Self {
-        Self {
-            callbacks: Default::default(),
-            _parameters: Default::default(),
-        }
-    }
-}
-
-impl<P> FromLua for ActionCallback<P> {
-    fn from_lua(value: mlua::Value, _lua: &mlua::Lua) -> mlua::Result<Self> {
-        let callbacks = match value {
-            mlua::Value::Function(f) => vec![f],
-            mlua::Value::Table(table) => table
-                .pairs::<mlua::Value, mlua::Function>()
-                .map(|pair| {
-                    let (_, f) = pair?;
-
-                    Ok::<_, mlua::Error>(f)
-                })
-                .try_collect()?,
-            _ => {
-                return Err(mlua::Error::FromLuaConversionError {
-                    from: value.type_name(),
-                    to: "ActionCallback".into(),
-                    message: Some("expected function or table of functions".into()),
-                });
-            }
-        };
-
-        Ok(ActionCallback {
-            callbacks,
-            _parameters: PhantomData::<P>,
-        })
-    }
-}
-
-// TODO: Return proper error here
-impl<P> ActionCallback<P>
-where
-    P: IntoLuaMulti + Sync + Clone,
-{
-    pub async fn call(&self, parameters: P) {
-        try_join_all(
-            self.callbacks
-                .iter()
-                .map(async |f| f.call_async::<()>(parameters.clone()).await),
-        )
-        .await
-        .unwrap();
-    }
-
-    pub fn is_empty(&self) -> bool {
-        self.callbacks.is_empty()
-    }
-}

automation_lib/src/config.rs

@@ -1,23 +0,0 @@
-use lua_typed::Typed;
-use serde::Deserialize;
-
-#[derive(Debug, Clone, Deserialize, Typed)]
-pub struct InfoConfig {
-    pub name: String,
-    pub room: Option<String>,
-}
-
-impl InfoConfig {
-    pub fn identifier(&self) -> String {
-        (if let Some(room) = &self.room {
-            room.to_ascii_lowercase().replace(' ', "_") + "_"
-        } else {
-            String::new()
-        }) + &self.name.to_ascii_lowercase().replace(' ', "_")
-    }
-}
-
-#[derive(Debug, Clone, Deserialize, Typed)]
-pub struct MqttDeviceConfig {
-    pub topic: String,
-}

automation_lib/src/device.rs

@@ -1,54 +0,0 @@
-use std::fmt::Debug;
-
-use automation_cast::Cast;
-use dyn_clone::DynClone;
-use lua_typed::Typed;
-use mlua::ObjectLike;
-
-use crate::event::OnMqtt;
-
-#[async_trait::async_trait]
-pub trait LuaDeviceCreate {
-    type Config;
-    type Error;
-
-    async fn create(config: Self::Config) -> Result<Self, Self::Error>
-    where
-        Self: Sized;
-}
-
-pub trait Device:
-    Debug + DynClone + Sync + Send + Cast<dyn google_home::Device> + Cast<dyn OnMqtt>
-{
-    fn get_id(&self) -> String;
-}
-
-impl mlua::FromLua for Box<dyn Device> {
-    fn from_lua(value: mlua::Value, _lua: &mlua::Lua) -> mlua::Result<Self> {
-        match value {
-            mlua::Value::UserData(ud) => {
-                let ud = if ud.is::<Self>() {
-                    ud
-                } else {
-                    ud.call_method::<_>("__box", ())?
-                };
-
-                let b = ud.borrow::<Self>()?.clone();
-                Ok(b)
-            }
-            _ => Err(mlua::Error::runtime(format!(
-                "Expected user data, instead found: {}",
-                value.type_name()
-            ))),
-        }
-    }
-}
-impl mlua::UserData for Box<dyn Device> {}
-
-impl Typed for Box<dyn Device> {
-    fn type_name() -> String {
-        "DeviceInterface".into()
-    }
-}
-
-dyn_clone::clone_trait_object!(Device);

automation_lib/src/device_manager.rs

@@ -1,89 +0,0 @@
-use std::collections::HashMap;
-use std::sync::Arc;
-
-use futures::future::join_all;
-use tokio::sync::{RwLock, RwLockReadGuard};
-use tracing::{debug, instrument, trace};
-
-use crate::device::Device;
-use crate::event::{Event, EventChannel, OnMqtt};
-
-pub type DeviceMap = HashMap<String, Box<dyn Device>>;
-
-#[derive(Clone)]
-pub struct DeviceManager {
-    devices: Arc<RwLock<DeviceMap>>,
-    event_channel: EventChannel,
-}
-
-impl DeviceManager {
-    pub async fn new() -> Self {
-        let (event_channel, mut event_rx) = EventChannel::new();
-
-        let device_manager = Self {
-            devices: Arc::new(RwLock::new(HashMap::new())),
-            event_channel,
-        };
-
-        tokio::spawn({
-            let device_manager = device_manager.clone();
-            async move {
-                loop {
-                    if let Some(event) = event_rx.recv().await {
-                        device_manager.handle_event(event).await;
-                    } else {
-                        todo!("Handle errors with the event channel properly")
-                    }
-                }
-            }
-        });
-
-        device_manager
-    }
-
-    pub async fn add(&self, device: Box<dyn Device>) {
-        let id = device.get_id();
-
-        debug!(id, "Adding device");
-
-        self.devices.write().await.insert(id, device);
-    }
-
-    pub fn event_channel(&self) -> EventChannel {
-        self.event_channel.clone()
-    }
-
-    pub async fn get(&self, name: &str) -> Option<Box<dyn Device>> {
-        self.devices.read().await.get(name).cloned()
-    }
-
-    pub async fn devices(&self) -> RwLockReadGuard<'_, DeviceMap> {
-        self.devices.read().await
-    }
-
-    #[instrument(skip(self))]
-    async fn handle_event(&self, event: Event) {
-        match event {
-            Event::MqttMessage(message) => {
-                let devices = self.devices.read().await;
-                let iter = devices.iter().map(async |(id, device)| {
-                    let device: Option<&dyn OnMqtt> = device.cast();
-                    if let Some(device) = device {
-                        // let subscribed = device
-                        //     .topics()
-                        //     .iter()
-                        //     .any(|topic| matches(&message.topic, topic));
-                        //
-                        // if subscribed {
-                        trace!(id, "Handling");
-                        device.on_mqtt(message.clone()).await;
-                        trace!(id, "Done");
-                        // }
-                    }
-                });
-
-                join_all(iter).await;
-            }
-        }
-    }
-}

automation_lib/src/helpers/mod.rs

@@ -1 +0,0 @@
-pub mod serialization;

automation_lib/src/helpers/serialization.rs

@@ -1,16 +0,0 @@
-use serde::de::{self, Unexpected};
-use serde::{Deserialize, Deserializer};
-
-pub fn state_deserializer<'de, D>(deserializer: D) -> Result<bool, D::Error>
-where
-    D: Deserializer<'de>,
-{
-    match String::deserialize(deserializer)?.as_ref() {
-        "ON" => Ok(true),
-        "OFF" => Ok(false),
-        other => Err(de::Error::invalid_value(
-            Unexpected::Str(other),
-            &"Value expected was either ON or OFF",
-        )),
-    }
-}

automation_lib/src/lib.rs

@@ -1,62 +0,0 @@
-#![feature(iterator_try_collect)]
-#![feature(with_negative_coherence)]
-
-use tracing::debug;
-
-pub mod action_callback;
-pub mod config;
-pub mod device;
-pub mod device_manager;
-pub mod error;
-pub mod event;
-pub mod helpers;
-pub mod lua;
-pub mod messages;
-pub mod mqtt;
-
-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,
-        definitions_fn: Option<DefinitionsFn>,
-    ) -> Self {
-        Self {
-            name,
-            register_fn,
-            definitions_fn,
-        }
-    }
-
-    pub const fn get_name(&self) -> &'static str {
-        self.name
-    }
-
-    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<()> {
-    for module in inventory::iter::<Module> {
-        debug!(name = module.get_name(), "Loading module");
-        let table = module.register(lua)?;
-        lua.register_module(module.get_name(), table)?;
-    }
-
-    Ok(())
-}
-
-inventory::collect!(Module);

automation_lib/src/lua/mod.rs

@@ -1,3 +0,0 @@
-pub mod traits;
-
-mod utils;

automation_lib/src/lua/traits.rs

@@ -1,127 +0,0 @@
-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
-    }
-
-    fn definitions() -> Option<String> {
-        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")
-    }
-}

automation_lib/src/lua/utils/mod.rs

@@ -1,48 +0,0 @@
-mod timeout;
-
-use std::time::{SystemTime, UNIX_EPOCH};
-
-use lua_typed::Typed;
-pub use timeout::Timeout;
-
-use crate::Module;
-
-fn create_module(lua: &mlua::Lua) -> mlua::Result<mlua::Table> {
-    let utils = lua.create_table()?;
-
-    utils.set(Timeout::type_name(), lua.create_proxy::<Timeout>()?)?;
-
-    let get_hostname = lua.create_function(|_lua, ()| {
-        hostname::get()
-            .map(|name| name.to_str().unwrap_or("unknown").to_owned())
-            .map_err(mlua::ExternalError::into_lua_err)
-    })?;
-    utils.set("get_hostname", get_hostname)?;
-    let get_epoch = lua.create_function(|_lua, ()| {
-        Ok(SystemTime::now()
-            .duration_since(UNIX_EPOCH)
-            .expect("Time is after UNIX EPOCH")
-            .as_millis())
-    })?;
-    utils.set("get_epoch", get_epoch)?;
-
-    Ok(utils)
-}
-
-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))}

automation_lib/src/lua/utils/timeout.rs

@@ -1,118 +0,0 @@
-use std::sync::Arc;
-use std::time::Duration;
-
-use lua_typed::Typed;
-use tokio::sync::RwLock;
-use tokio::task::JoinHandle;
-use tracing::debug;
-
-use crate::action_callback::ActionCallback;
-
-#[derive(Debug, Default)]
-pub struct State {
-    handle: Option<JoinHandle<()>>,
-}
-
-#[derive(Debug, Clone)]
-pub struct Timeout {
-    state: Arc<RwLock<State>>,
-}
-
-impl mlua::UserData for Timeout {
-    fn add_methods<M: mlua::UserDataMethods<Self>>(methods: &mut M) {
-        methods.add_function("new", |_lua, ()| {
-            let device = Self {
-                state: Default::default(),
-            };
-
-            Ok(device)
-        });
-
-        methods.add_async_method(
-            "start",
-            async |_lua, this, (timeout, callback): (f32, ActionCallback<()>)| {
-                if let Some(handle) = this.state.write().await.handle.take() {
-                    handle.abort();
-                }
-
-                debug!("Running timeout callback after {timeout}s");
-
-                let timeout = Duration::from_secs_f32(timeout);
-
-                this.state.write().await.handle = Some(tokio::spawn({
-                    async move {
-                        tokio::time::sleep(timeout).await;
-
-                        callback.call(()).await;
-                    }
-                }));
-
-                Ok(())
-            },
-        );
-
-        methods.add_async_method("cancel", async |_lua, this, ()| {
-            debug!("Canceling timeout callback");
-
-            if let Some(handle) = this.state.write().await.handle.take() {
-                handle.abort();
-            }
-
-            Ok(())
-        });
-
-        methods.add_async_method("is_waiting", async |_lua, this, ()| {
-            debug!("Canceling timeout callback");
-
-            if let Some(handle) = this.state.read().await.handle.as_ref() {
-                debug!("Join handle: {}", handle.is_finished());
-                return Ok(!handle.is_finished());
-            }
-
-            debug!("Join handle: None");
-
-            Ok(false)
-        });
-    }
-}
-
-impl Typed for Timeout {
-    fn type_name() -> String {
-        "Timeout".into()
-    }
-
-    fn generate_header() -> Option<String> {
-        let type_name = Self::type_name();
-        Some(format!("---@class {type_name}\nlocal {type_name}\n"))
-    }
-
-    fn generate_members() -> Option<String> {
-        let mut output = String::new();
-
-        let type_name = Self::type_name();
-
-        output += &format!(
-            "---@async\n---@param timeout number\n---@param callback {}\nfunction {type_name}:start(timeout, callback) end\n",
-            ActionCallback::<()>::type_name()
-        );
-
-        output += &format!("---@async\nfunction {type_name}:cancel() end\n",);
-
-        output +=
-            &format!("---@async\n---@return boolean\nfunction {type_name}:is_waiting() end\n",);
-
-        Some(output)
-    }
-
-    fn generate_footer() -> Option<String> {
-        let mut output = String::new();
-
-        let type_name = Self::type_name();
-
-        output += &format!("utils.{type_name} = {{}}\n");
-        output += &format!("---@return {type_name}\n");
-        output += &format!("function utils.{type_name}.new() end\n");
-
-        Some(output)
-    }
-}

automation_lib/src/mqtt.rs

@@ -1,128 +0,0 @@
-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, Transport};
-use serde::Deserialize;
-use tracing::{debug, warn};
-
-use crate::event::{self, EventChannel};
-
-#[derive(Debug, Clone, LuaDeviceConfig, Deserialize, Typed)]
-pub struct MqttConfig {
-    pub host: String,
-    pub port: u16,
-    pub client_name: String,
-    pub username: String,
-    pub password: String,
-    #[serde(default)]
-    #[typed(default)]
-    pub tls: bool,
-}
-
-impl From<MqttConfig> for MqttOptions {
-    fn from(value: MqttConfig) -> Self {
-        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(Duration::from_secs(5));
-
-        if value.tls {
-            mqtt_options.set_transport(Transport::tls_with_default_config());
-        }
-
-        mqtt_options
-    }
-}
-
-#[derive(Debug, Clone, FromLua)]
-pub struct WrappedAsyncClient(pub AsyncClient);
-
-impl Typed for WrappedAsyncClient {
-    fn type_name() -> String {
-        "AsyncClient".into()
-    }
-
-    fn generate_header() -> Option<String> {
-        let type_name = Self::type_name();
-        Some(format!("---@class {type_name}\nlocal {type_name}\n"))
-    }
-
-    fn generate_members() -> Option<String> {
-        let mut output = String::new();
-
-        let type_name = Self::type_name();
-
-        output += &format!(
-            "---@async\n---@param topic string\n---@param message table?\nfunction {type_name}:send_message(topic, message) end\n"
-        );
-
-        Some(output)
-    }
-}
-
-impl Deref for WrappedAsyncClient {
-    type Target = AsyncClient;
-
-    fn deref(&self) -> &Self::Target {
-        &self.0
-    }
-}
-
-impl DerefMut for WrappedAsyncClient {
-    fn deref_mut(&mut self) -> &mut Self::Target {
-        &mut self.0
-    }
-}
-
-impl mlua::UserData for WrappedAsyncClient {
-    fn add_methods<M: mlua::UserDataMethods<Self>>(methods: &mut M) {
-        methods.add_async_method(
-            "send_message",
-            async |_lua, this, (topic, message): (String, mlua::Value)| {
-                // serde_json converts nil => "null", but we actually want nil to send an empty
-                // message
-                let message = if message.is_nil() {
-                    "".into()
-                } else {
-                    serde_json::to_string(&message).unwrap()
-                };
-
-                debug!("message = {message}");
-
-                this.0
-                    .publish(topic, rumqttc::QoS::AtLeastOnce, true, message)
-                    .await
-                    .unwrap();
-
-                Ok(())
-            },
-        );
-    }
-}
-
-pub fn start(config: MqttConfig, event_channel: &EventChannel) -> WrappedAsyncClient {
-    let tx = event_channel.get_tx();
-    let (client, mut eventloop) = AsyncClient::new(config.into(), 100);
-
-    tokio::spawn(async move {
-        debug!("Listening for MQTT events");
-        loop {
-            let notification = eventloop.poll().await;
-            match notification {
-                Ok(Event::Incoming(Incoming::Publish(p))) => {
-                    tx.send(event::Event::MqttMessage(p)).await.ok();
-                }
-                Ok(..) => continue,
-                Err(err) => {
-                    // Something has gone wrong
-                    // We stay in the loop as that will attempt to reconnect
-                    warn!("{}", err);
-                }
-            }
-        }
-    });
-
-    WrappedAsyncClient(client)
-}

automation_macro/Cargo.toml

@@ -1,16 +1,13 @@
 [package]
 name = "automation_macro"
 version = "0.1.0"
-edition = "2024"
+edition = "2021"
 
 [lib]
 proc-macro = true
 
 [dependencies]
-itertools = { workspace = true }
+itertools = "0.12.1"
-proc-macro2 = { workspace = true }
+proc-macro2 = "1.0.81"
-quote = { workspace = true }
+quote = "1.0.36"
-syn = { workspace = true }
+syn = { version = "2.0.60", features = ["extra-traits", "full"] }
-
-[dev-dependencies]
-mlua = { workspace = true }

automation_macro/src/device.rs

@@ -1,265 +0,0 @@
-use std::collections::HashMap;
-
-use proc_macro2::TokenStream as TokenStream2;
-use quote::quote;
-use syn::parse::{Parse, ParseStream};
-use syn::punctuated::Punctuated;
-use syn::spanned::Spanned;
-use syn::{Attribute, DeriveInput, Token, parenthesized};
-
-enum Attr {
-    Trait(TraitAttr),
-    ExtraUserData(ExtraUserDataAttr),
-}
-
-impl Attr {
-    fn parse(attr: &Attribute) -> syn::Result<Self> {
-        let mut parsed = None;
-        attr.parse_nested_meta(|meta| {
-            if meta.path.is_ident("traits") {
-                let input;
-                _ = parenthesized!(input in meta.input);
-                parsed = Some(Attr::Trait(input.parse()?));
-            } else if meta.path.is_ident("extra_user_data") {
-                let value = meta.value()?;
-                parsed = Some(Attr::ExtraUserData(value.parse()?));
-            } else {
-                return Err(syn::Error::new(meta.path.span(), "Unknown attribute"));
-            }
-
-            Ok(())
-        })?;
-
-        Ok(parsed.expect("Parsed should be set"))
-    }
-}
-
-struct TraitAttr {
-    traits: Traits,
-    aliases: Aliases,
-}
-
-impl Parse for TraitAttr {
-    fn parse(input: ParseStream) -> syn::Result<Self> {
-        Ok(Self {
-            traits: input.parse()?,
-            aliases: input.parse()?,
-        })
-    }
-}
-
-#[derive(Default)]
-struct Traits(Vec<syn::Ident>);
-
-impl Traits {
-    fn extend(&mut self, other: &Traits) {
-        self.0.extend_from_slice(&other.0);
-    }
-}
-
-impl Parse for Traits {
-    fn parse(input: ParseStream) -> syn::Result<Self> {
-        input
-            .call(Punctuated::<_, Token![,]>::parse_separated_nonempty)
-            .map(|traits| traits.into_iter().collect())
-            .map(Self)
-    }
-}
-
-#[derive(Default)]
-struct Aliases(Vec<syn::Ident>);
-
-impl Aliases {
-    fn has_aliases(&self) -> bool {
-        !self.0.is_empty()
-    }
-}
-
-impl Parse for Aliases {
-    fn parse(input: ParseStream) -> syn::Result<Self> {
-        if !input.peek(Token![for]) {
-            if input.is_empty() {
-                return Ok(Default::default());
-            } else {
-                return Err(input.error("Expected ')' or 'for'"));
-            }
-        }
-
-        _ = input.parse::<syn::Token![for]>()?;
-
-        input
-            .call(Punctuated::<_, Token![,]>::parse_separated_nonempty)
-            .map(|aliases| aliases.into_iter().collect())
-            .map(Self)
-    }
-}
-
-#[derive(Clone)]
-struct ExtraUserDataAttr(syn::Ident);
-
-impl Parse for ExtraUserDataAttr {
-    fn parse(input: ParseStream) -> syn::Result<Self> {
-        Ok(Self(input.parse()?))
-    }
-}
-
-struct Implementation {
-    name: syn::Ident,
-    traits: Traits,
-    extra_user_data: Vec<ExtraUserDataAttr>,
-}
-
-impl quote::ToTokens for Implementation {
-    fn to_tokens(&self, tokens: &mut TokenStream2) {
-        let Self {
-            name,
-            traits,
-            extra_user_data,
-        } = &self;
-        let Traits(traits) = traits;
-        let extra_user_data: Vec<_> = extra_user_data.iter().map(|tr| tr.0.clone()).collect();
-
-        tokens.extend(quote! {
-            impl mlua::UserData for #name {
-                fn add_methods<M: mlua::UserDataMethods<Self>>(methods: &mut M) {
-                    methods.add_async_function("new", async |_lua, config| {
-                        let device: Self = LuaDeviceCreate::create(config)
-                            .await
-                            .map_err(mlua::ExternalError::into_lua_err)?;
-
-                        Ok(device)
-                    });
-
-                    methods.add_method("__box", |_lua, this, _: ()| {
-                        let b: Box<dyn Device> = Box::new(this.clone());
-                        Ok(b)
-                    });
-
-                    <::automation_lib::lua::traits::Device as ::automation_lib::lua::traits::PartialUserData<#name>>::add_methods(methods);
-
-                    #(
-                        <::automation_lib::lua::traits::#traits as ::automation_lib::lua::traits::PartialUserData<#name>>::add_methods(methods);
-                    )*
-
-                    #(
-                        <#extra_user_data as ::automation_lib::lua::traits::PartialUserData<#name>>::add_methods(methods);
-                    )*
-                }
-            }
-
-            impl ::lua_typed::Typed for #name {
-                fn type_name() -> String {
-                    stringify!(#name).into()
-                }
-
-                fn generate_header() -> std::option::Option<::std::string::String> {
-                    let type_name = <Self as ::lua_typed::Typed>::type_name();
-                    let mut output = String::new();
-
-                    let interfaces: String = [
-                        <::automation_lib::lua::traits::Device as ::automation_lib::lua::traits::PartialUserData<#name>>::interface_name(),
-                        #(
-                            <::automation_lib::lua::traits::#traits as ::automation_lib::lua::traits::PartialUserData<#name>>::interface_name(),
-                        )*
-                    ].into_iter().flatten().intersperse(", ").collect();
-
-                    let interfaces = if interfaces.is_empty() {
-                        "".into()
-                    } else {
-                        format!(": {interfaces}")
-                    };
-
-                    Some(format!("---@class {type_name}{interfaces}\nlocal {type_name}\n"))
-                }
-
-                fn generate_members() -> Option<String> {
-                    let mut output = String::new();
-
-                    let type_name = <Self as ::lua_typed::Typed>::type_name();
-                    output += &format!("devices.{type_name} = {{}}\n");
-                    let config_name = <<Self as ::automation_lib::device::LuaDeviceCreate>::Config as ::lua_typed::Typed>::type_name();
-                    output += &format!("---@param config {config_name}\n");
-                    output += &format!("---@return {type_name}\n");
-                    output += &format!("function devices.{type_name}.new(config) end\n");
-
-                    output += &<::automation_lib::lua::traits::Device as ::automation_lib::lua::traits::PartialUserData<#name>>::definitions().unwrap_or("".into());
-
-                    #(
-                        output += &<::automation_lib::lua::traits::#traits as ::automation_lib::lua::traits::PartialUserData<#name>>::definitions().unwrap_or("".into());
-                    )*
-                    #(
-                        output += &<#extra_user_data as ::automation_lib::lua::traits::PartialUserData<#name>>::definitions().unwrap_or("".into());
-                    )*
-
-
-                    Some(output)
-                }
-            }
-        });
-    }
-}
-
-struct Implementations(Vec<Implementation>);
-
-impl Implementations {
-    fn from_attr(attributes: Vec<Attr>, name: syn::Ident) -> Self {
-        let mut add_methods = Vec::new();
-        let mut all = Traits::default();
-        let mut implementations: HashMap<_, Traits> = HashMap::new();
-        for attribute in attributes {
-            match attribute {
-                Attr::Trait(attribute) => {
-                    if attribute.aliases.has_aliases() {
-                        for alias in &attribute.aliases.0 {
-                            implementations
-                                .entry(Some(alias.clone()))
-                                .or_default()
-                                .extend(&attribute.traits);
-                        }
-                    } else {
-                        all.extend(&attribute.traits);
-                    }
-                }
-                Attr::ExtraUserData(attribute) => add_methods.push(attribute),
-            }
-        }
-
-        if implementations.is_empty() {
-            implementations.entry(None).or_default().extend(&all);
-        } else {
-            for traits in implementations.values_mut() {
-                traits.extend(&all);
-            }
-        }
-
-        Self(
-            implementations
-                .into_iter()
-                .map(|(alias, traits)| Implementation {
-                    name: alias.unwrap_or(name.clone()),
-                    traits,
-                    extra_user_data: add_methods.clone(),
-                })
-                .collect(),
-        )
-    }
-}
-
-pub fn device(input: DeriveInput) -> TokenStream2 {
-    let Implementations(imp) = match input
-        .attrs
-        .iter()
-        .filter(|attr| attr.path().is_ident("device"))
-        .map(Attr::parse)
-        .try_collect::<Vec<_>>()
-    {
-        Ok(attr) => Implementations::from_attr(attr, input.ident),
-        Err(err) => return err.into_compile_error(),
-    };
-
-    quote! {
-        #(
-            #imp
-        )*
-    }
-}

automation_macro/src/lib.rs

@@ -1,11 +1,18 @@
-#![feature(iter_intersperse)]
+mod lua_device;
-#![feature(iterator_try_collect)]
-mod device;
 mod lua_device_config;
+mod lua_type_definition;
 
+use lua_device::impl_lua_device_macro;
 use lua_device_config::impl_lua_device_config_macro;
-use quote::quote;
+use lua_type_definition::impl_lua_type_definition;
-use syn::{DeriveInput, parse_macro_input};
+use syn::{parse_macro_input, DeriveInput};
+
+#[proc_macro_derive(LuaDevice)]
+pub fn lua_device_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
+    let ast = parse_macro_input!(input as DeriveInput);
+
+    impl_lua_device_macro(&ast).into()
+}
 
 #[proc_macro_derive(LuaDeviceConfig, attributes(device_config))]
 pub fn lua_device_config_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
@@ -14,60 +21,9 @@ pub fn lua_device_config_derive(input: proc_macro::TokenStream) -> proc_macro::T
     impl_lua_device_config_macro(&ast).into()
 }
 
-#[proc_macro_derive(LuaSerialize, attributes(traits))]
+#[proc_macro_derive(LuaTypeDefinition, attributes(device_config))]
-pub fn lua_serialize(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
+pub fn lua_type_definition_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
     let ast = parse_macro_input!(input as DeriveInput);
 
-    let name = &ast.ident;
+    impl_lua_type_definition(&ast).into()
-
-    quote! {
-        impl ::mlua::IntoLua for #name {
-            fn into_lua(self, lua: &::mlua::Lua) -> ::mlua::Result<::mlua::Value> {
-                ::mlua::LuaSerdeExt::to_value(lua, &self)
-            }
-        }
-    }
-    .into()
-}
-
-/// Derive macro generating an impl for the trait `::mlua::UserData`
-///
-/// # Device traits
-/// The `device(traits)` attribute can be used to tell the macro what traits are implemented so that
-/// the appropriate methods can automatically be registered.
-/// If the struct does not have any type parameters the syntax is very simple:
-/// ```rust
-/// #[device(traits(TraitA, TraitB))]
-/// ```
-///
-/// If the type does have type parameters you will have to manually specify all variations that
-/// have the trait available:
-/// ```rust
-/// #[device(traits(TraitA, TraitB for <StateA>, <StateB>))]
-/// ```
-/// If multiple of these attributes are specified they will all combined appropriately.
-///
-///
-/// ## NOTE
-/// If your type _has_ type parameters any instance of the traits attribute that does not specify
-/// any type parameters will have the traits applied to _all_ other type parameter variations
-/// listed in the other trait attributes. This behavior only applies if there is at least one
-/// instance with type parameters specified.
-///
-/// # Additional methods
-/// Additional methods can be added by using the `device(add_methods)` attribute. This attribute
-/// takes the path to a function with the following signature that can register the additional methods:
-///
-/// ```rust
-/// # struct D;
-/// fn top_secret<M: mlua::UserDataMethods<D>>(methods: &mut M) {}
-/// ```
-/// It can then be registered with:
-/// ```rust
-/// #[device(add_methods = top_secret)]
-/// ```
-#[proc_macro_derive(Device, attributes(device))]
-pub fn device(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
-    let ast = parse_macro_input!(input as DeriveInput);
-    device::device(ast).into()
 }

automation_macro/src/lua_device.rs

@@ -0,0 +1,42 @@
+use proc_macro2::TokenStream;
+use quote::quote;
+use syn::DeriveInput;
+
+pub fn impl_lua_device_macro(ast: &DeriveInput) -> TokenStream {
+    let name = &ast.ident;
+    let gen = quote! {
+        impl #name {
+            pub fn register_with_lua(lua: &mlua::Lua) -> mlua::Result<()> {
+                lua.globals().set(stringify!(#name), lua.create_proxy::<#name>()?)
+            }
+
+            pub fn generate_lua_definition() -> String {
+                // TODO: Do not hardcode the name of the config type
+                let def = format!(
+                    r#"--- @class {0}
+{0} = {{}}
+--- @param config {0}Config
+--- @return WrappedDevice
+function {0}.new(config) end
+"#, stringify!(#name)
+                );
+
+                def
+            }
+        }
+        impl mlua::UserData for #name {
+            fn add_methods<'lua, M: mlua::UserDataMethods<'lua, Self>>(methods: &mut M) {
+                methods.add_async_function("new", |lua, config: mlua::Value| async {
+                    let config = mlua::FromLua::from_lua(config, lua)?;
+
+                    // TODO: Using crate:: could cause issues
+                    let device: #name = crate::devices::LuaDeviceCreate::create(config).await.map_err(mlua::ExternalError::into_lua_err)?;
+
+                    Ok(crate::device_manager::WrappedDevice::new(Box::new(device)))
+                });
+            }
+        }
+    };
+
+    gen
+}

automation_macro/src/lua_device_config.rs

@@ -6,8 +6,8 @@ use syn::punctuated::Punctuated;
 use syn::spanned::Spanned;
 use syn::token::Paren;
 use syn::{
-    Data, DataStruct, DeriveInput, Expr, Field, Fields, FieldsNamed, LitStr, Result, Token, Type,
+    parenthesized, Data, DataStruct, DeriveInput, Expr, Field, Fields, FieldsNamed, LitStr, Result,
-    parenthesized,
+    Token, Type,
 };
 
 mod kw {
@@ -23,7 +23,7 @@ mod kw {
 }
 
 #[derive(Debug)]
-enum Argument {
+pub enum Argument {
     Flatten {
         _keyword: kw::flatten,
     },
@@ -107,8 +107,8 @@ impl Parse for Argument {
 }
 
 #[derive(Debug)]
-struct Args {
+pub(crate) struct Args {
-    args: Punctuated<Argument, Token![,]>,
+    pub(crate) args: Punctuated<Argument, Token![,]>,
 }
 
 impl Parse for Args {
@@ -155,7 +155,7 @@ fn field_from_lua(field: &Field) -> TokenStream {
         [] => field.ident.clone().unwrap().to_string(),
         [rename] => rename.to_owned(),
         _ => {
-            return quote_spanned! {field.span() => compile_error!("Field contains duplicate 'rename'")};
+            return quote_spanned! {field.span() => compile_error!("Field contains duplicate 'rename'")}
         }
     };
 
@@ -174,7 +174,7 @@ fn field_from_lua(field: &Field) -> TokenStream {
         [] => quote! {panic!(#missing)},
         [default] => default.to_owned(),
         _ => {
-            return quote_spanned! {field.span() => compile_error!("Field contains duplicate 'default'")};
+            return quote_spanned! {field.span() => compile_error!("Field contains duplicate 'default'")}
         }
     };
 
@@ -218,6 +218,21 @@ fn field_from_lua(field: &Field) -> TokenStream {
                     temp.into()
                 }
             }),
+            _ => None,
+        })
+        .collect::<Vec<_>>()
+        .as_slice()
+    {
+        [] => value,
+        [value] => value.to_owned(),
+        _ => {
+            return quote_spanned! {field.span() => compile_error!("Field contains duplicate 'from'")}
+        }
+    };
+
+    let value = match args
+        .iter()
+        .filter_map(|arg| match arg {
             Argument::With { expr, .. } => Some(quote! {
                 {
                     let temp = #value;
@@ -232,7 +247,7 @@ fn field_from_lua(field: &Field) -> TokenStream {
         [] => value,
         [value] => value.to_owned(),
         _ => {
-            return quote_spanned! {field.span() => compile_error!("Only one of either 'from' or 'with' is allowed")};
+            return quote_spanned! {field.span() => compile_error!("Field contains duplicate 'with'")}
         }
     };
 
@@ -260,10 +275,9 @@ pub fn impl_lua_device_config_macro(ast: &DeriveInput) -> TokenStream {
         })
         .collect();
 
-    let (impl_generics, type_generics, where_clause) = ast.generics.split_for_impl();
     let impl_from_lua = quote! {
-        impl #impl_generics mlua::FromLua for #name #type_generics #where_clause {
+        impl<'lua> mlua::FromLua<'lua> for #name {
-            fn from_lua(value: mlua::Value, lua: &mlua::Lua) -> mlua::Result<Self> {
+            fn from_lua(value: mlua::Value<'lua>, lua: &'lua mlua::Lua) -> mlua::Result<Self> {
                 if !value.is_table() {
                     panic!("Expected table");
                 }

automation_macro/src/lua_type_definition.rs

@@ -0,0 +1,137 @@
+use itertools::Itertools;
+use proc_macro2::TokenStream;
+use quote::{quote, quote_spanned};
+use syn::spanned::Spanned;
+use syn::{
+    AngleBracketedGenericArguments, Data, DataStruct, DeriveInput, Field, Fields, FieldsNamed,
+    PathArguments, Type, TypePath,
+};
+
+use crate::lua_device_config::{Args, Argument};
+
+fn field_definition(field: &Field) -> TokenStream {
+    let (args, _): (Vec<_>, Vec<_>) = field
+        .attrs
+        .iter()
+        .filter_map(|attr| {
+            if attr.path().is_ident("device_config") {
+                Some(attr.parse_args::<Args>().map(|args| args.args))
+            } else {
+                None
+            }
+        })
+        .partition_result();
+    let args: Vec<_> = args.into_iter().flatten().collect();
+
+    let field_name = if let Some(field_name) = args.iter().find_map(|arg| match arg {
+        Argument::Rename { ident, .. } => Some(ident),
+        _ => None,
+    }) {
+        field_name.value()
+    } else {
+        format!("{}", field.ident.clone().unwrap())
+    };
+
+    let mut optional = args
+        .iter()
+        .filter(|arg| matches!(arg, Argument::Default { .. } | Argument::DefaultExpr { .. }))
+        .count()
+        >= 1;
+
+    if args
+        .iter()
+        .filter(|arg| matches!(arg, Argument::Flatten { .. }))
+        .count()
+        >= 1
+    {
+        let field_type = &field.ty;
+        quote! {
+            #field_type::generate_lua_fields().as_str()
+        }
+    } else {
+        let path = if let Some(ty) = args.iter().find_map(|arg| match arg {
+            Argument::From { ty, .. } => Some(ty),
+            _ => None,
+        }) {
+            if let Type::Path(TypePath { path, .. }) = ty {
+                path.clone()
+            } else {
+                todo!();
+            }
+        } else if let Type::Path(TypePath { path, .. }) = field.ty.clone() {
+            path
+        } else {
+            todo!()
+        };
+
+        let seg = path.segments.first().unwrap();
+        let field_type = if seg.ident == "Option" {
+            if let PathArguments::AngleBracketed(AngleBracketedGenericArguments { args, .. }) =
+                seg.arguments.clone()
+            {
+                optional = true;
+                quote! { stringify!(#args) }
+            } else {
+                unreachable!("Option should always have angle brackets");
+            }
+        } else if seg.ident == "Vec" {
+            if let PathArguments::AngleBracketed(AngleBracketedGenericArguments { args, .. }) =
+                seg.arguments.clone()
+            {
+                optional = true;
+                quote! { stringify!(#args[]) }
+            } else {
+                unreachable!("Option should always have angle brackets");
+            }
+        } else {
+            quote! { stringify!(#path).replace(" :: ", "_") }
+        };
+
+        let mut format = "--- @field {} {}".to_string();
+        if optional {
+            format += "|nil";
+        }
+        format += "\n";
+
+        quote! {
+            format!(#format, #field_name, #field_type).as_str()
+        }
+    }
+}
+
+pub fn impl_lua_type_definition(ast: &DeriveInput) -> TokenStream {
+    let name = &ast.ident;
+    let fields = if let Data::Struct(DataStruct {
+        fields: Fields::Named(FieldsNamed { ref named, .. }),
+        ..
+    }) = ast.data
+    {
+        named
+    } else {
+        return quote_spanned! {ast.span() => compile_error!("This macro only works on named structs")};
+    };
+
+    let fields: Vec<_> = fields.iter().map(field_definition).collect();
+
+    let gen = quote! {
+        impl #name {
+            pub fn generate_lua_definition() -> String {
+                let mut def = format!("--- @class {}\n", stringify!(#name));
+
+                def += #name::generate_lua_fields().as_str();
+
+                def
+            }
+
+            pub fn generate_lua_fields() -> String {
+                let mut def = String::new();
+
+                #(def += #fields;)*
+
+                def
+            }
+        }
+    };
+
+    gen
+}

config.lua

@@ -0,0 +1,181 @@
+print("Hello from lua")
+
+local host = automation.util.get_hostname()
+print("Running @" .. host)
+
+local debug, value = pcall(automation.util.get_env, "DEBUG")
+if debug and value ~= "true" then
+	debug = false
+end
+
+local function mqtt_z2m(topic)
+	return "zigbee2mqtt/" .. topic
+end
+
+local function mqtt_automation(topic)
+	return "automation/" .. topic
+end
+
+automation.fulfillment = {
+	openid_url = "https://login.huizinga.dev/api/oidc",
+}
+
+local mqtt_client = automation.new_mqtt_client({
+	host = (host == "zeus" and "olympus.lan.huizinga.dev")
+		or (host == "hephaestus" and "olympus.vpn.huizinga.dev")
+		or "mosquitto",
+	port = 8883,
+	client_name = "automation-" .. host,
+	username = "mqtt",
+	password = automation.util.get_env("MQTT_PASSWORD"),
+	tls = host == "zeus" or host == "hephaestus",
+})
+
+automation.device_manager:add(Ntfy.new({
+	topic = automation.util.get_env("NTFY_TOPIC"),
+	event_channel = automation.device_manager:event_channel(),
+}))
+
+automation.device_manager:add(Presence.new({
+	topic = "automation_dev/presence/+/#",
+	client = mqtt_client,
+	event_channel = automation.device_manager:event_channel(),
+}))
+
+automation.device_manager:add(DebugBridge.new({
+	identifier = "debug_bridge",
+	topic = mqtt_automation("debug"),
+	client = mqtt_client,
+}))
+
+local hue_ip = "10.0.0.146"
+local hue_token = automation.util.get_env("HUE_TOKEN")
+
+automation.device_manager:add(HueBridge.new({
+	identifier = "hue_bridge",
+	ip = hue_ip,
+	login = hue_token,
+	flags = {
+		presence = 41,
+		darkness = 43,
+	},
+}))
+
+automation.device_manager:add(LightSensor.new({
+	identifier = "living_light_sensor",
+	topic = mqtt_z2m("living/light"),
+	client = mqtt_client,
+	min = 22000,
+	max = 23500,
+	event_channel = automation.device_manager:event_channel(),
+}))
+
+automation.device_manager:add(WakeOnLAN.new({
+	name = "Zeus",
+	room = "Living Room",
+	topic = mqtt_automation("appliance/living_room/zeus"),
+	client = mqtt_client,
+	mac_address = "30:9c:23:60:9c:13",
+	broadcast_ip = "10.0.0.255",
+}))
+
+local living_mixer = KasaOutlet.new({ identifier = "living_mixer", ip = "10.0.0.49" })
+automation.device_manager:add(living_mixer)
+local living_speakers = KasaOutlet.new({ identifier = "living_speakers", ip = "10.0.0.182" })
+automation.device_manager:add(living_speakers)
+
+automation.device_manager:add(AudioSetup.new({
+	identifier = "living_audio",
+	topic = mqtt_z2m("living/remote"),
+	client = mqtt_client,
+	mixer = living_mixer,
+	speakers = living_speakers,
+}))
+
+automation.device_manager:add(IkeaOutlet.new({
+	outlet_type = "Kettle",
+	name = "Kettle",
+	room = "Kitchen",
+	topic = mqtt_z2m("kitchen/kettle"),
+	client = mqtt_client,
+	timeout = debug and 5 or 300,
+	remotes = {
+		{ topic = mqtt_z2m("bedroom/remote") },
+		{ topic = mqtt_z2m("kitchen/remote") },
+	},
+}))
+
+automation.device_manager:add(IkeaOutlet.new({
+	outlet_type = "Light",
+	name = "Light",
+	room = "Bathroom",
+	topic = mqtt_z2m("batchroom/light"),
+	client = mqtt_client,
+	timeout = debug and 60 or 45 * 60,
+}))
+
+automation.device_manager:add(Washer.new({
+	identifier = "bathroom_washer",
+	topic = mqtt_z2m("batchroom/washer"),
+	client = mqtt_client,
+	threshold = 1,
+	event_channel = automation.device_manager:event_channel(),
+}))
+
+automation.device_manager:add(IkeaOutlet.new({
+	outlet_type = "Charger",
+	name = "Charger",
+	room = "Workbench",
+	topic = mqtt_z2m("workbench/charger"),
+	client = mqtt_client,
+	timeout = debug and 5 or 20 * 3600,
+}))
+
+automation.device_manager:add(IkeaOutlet.new({
+	name = "Outlet",
+	room = "Workbench",
+	topic = mqtt_z2m("workbench/outlet"),
+	client = mqtt_client,
+}))
+local hallway_lights = HueGroup.new({
+	identifier = "hallway_lights",
+	ip = hue_ip,
+	login = hue_token,
+	group_id = 81,
+	scene_id = "3qWKxGVadXFFG4o",
+	timer_id = 1,
+	remotes = {
+		{ topic = mqtt_z2m("hallway/remote") },
+	},
+	client = mqtt_client,
+})
+automation.device_manager:add(hallway_lights)
+
+automation.device_manager:add(ContactSensor.new({
+	identifier = "hallway_frontdoor",
+	topic = mqtt_z2m("hallway/frontdoor"),
+	client = mqtt_client,
+	presence = {
+		topic = mqtt_automation("presence/contact/frontdoor"),
+		timeout = debug and 10 or 15 * 60,
+	},
+	trigger = {
+		devices = { hallway_lights },
+		timeout = debug and 10 or 2 * 60,
+	},
+}))
+
+local bedroom_air_filter = AirFilter.new({
+	name = "Air Filter",
+	room = "Bedroom",
+	topic = "pico/filter/bedroom",
+	client = mqtt_client,
+})
+automation.device_manager:add(bedroom_air_filter)
+
+automation.device_manager:schedule("0 0 19 * * *", function()
+	bedroom_air_filter:set_on(true)
+end)
+automation.device_manager:schedule("0 0 20 * * *", function()
+	bedroom_air_filter:set_on(false)
+end)

config/battery.lua

@@ -1,47 +0,0 @@
-local ntfy = require("config.ntfy")
-
---- @class BatteryModule: Module
-local module = {}
-
---- @type {[string]: number}
-local low_battery = {}
-
---- @param device DeviceInterface
---- @param battery number
-function module.callback(device, battery)
-	local id = device:get_id()
-	if battery < 15 then
-		print("Device '" .. id .. "' has low battery: " .. tostring(battery))
-		low_battery[id] = battery
-	else
-		low_battery[id] = nil
-	end
-end
-
-local function notify_low_battery()
-	-- Don't send notifications if there are now devices with low battery
-	if next(low_battery) == nil then
-		print("No devices with low battery")
-		return
-	end
-
-	local lines = {}
-	for name, battery in pairs(low_battery) do
-		table.insert(lines, name .. ": " .. tostring(battery) .. "%")
-	end
-	local message = table.concat(lines, "\n")
-
-	ntfy.send_notification({
-		title = "Low battery",
-		message = message,
-		tags = { "battery" },
-		priority = "default",
-	})
-end
-
---- @type Schedule
-module.schedule = {
-	["0 0 21 */1 * *"] = notify_low_battery,
-}
-
-return module

config/config.lua

@@ -1,32 +0,0 @@
-local utils = require("automation:utils")
-local secrets = require("automation:secrets")
-
-local host = utils.get_hostname()
-print("Lua " .. _VERSION .. " running on " .. utils.get_hostname())
-
----@type Config
-return {
-	fulfillment = {
-		openid_url = "https://login.huizinga.dev/api/oidc",
-	},
-	mqtt = {
-		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 = host == "zeus" or host == "hephaestus",
-	},
-	modules = {
-		require("config.battery"),
-		require("config.debug"),
-		require("config.hallway_automation"),
-		require("config.helper"),
-		require("config.hue_bridge"),
-		require("config.light"),
-		require("config.ntfy"),
-		require("config.presence"),
-		require("config.rooms"),
-		require("config.windows"),
-	},
-}

config/debug.lua

@@ -1,35 +0,0 @@
-local helper = require("config.helper")
-local light = require("config.light")
-local presence = require("config.presence")
-local utils = require("automation:utils")
-local variables = require("automation:variables")
-
---- @class DebugModule: Module
-local module = {}
-
-if variables.debug == "true" then
-	module.debug_mode = true
-elseif not variables.debug or variables.debug == "false" then
-	module.debug_mode = false
-else
-	error("Variable debug has invalid value '" .. variables.debug .. "', expected 'true' or 'false'")
-end
-
---- @type SetupFunction
-function module.setup(mqtt_client)
-	presence.add_callback(function(p)
-		mqtt_client:send_message(helper.mqtt_automation("debug") .. "/presence", {
-			state = p,
-			updated = utils.get_epoch(),
-		})
-	end)
-
-	light.add_callback(function(l)
-		mqtt_client:send_message(helper.mqtt_automation("debug") .. "/darkness", {
-			state = not l,
-			updated = utils.get_epoch(),
-		})
-	end)
-end
-
-return module

config/hallway_automation.lua

@@ -1,85 +0,0 @@
-local debug = require("config.debug")
-local utils = require("automation:utils")
-
---- @class HallwayAutomationModule: Module
-local module = {}
-
-local timeout = utils.Timeout.new()
-local forced = false
---- @type OpenCloseInterface?
-local trash = nil
---- @type OpenCloseInterface?
-local door = nil
-
---- @type fun(on: boolean)[]
-local callbacks = {}
-
---- @param on boolean
-local function callback(on)
-	for _, f in ipairs(callbacks) do
-		f(on)
-	end
-end
-
----@type fun(device: DeviceInterface, on: boolean)
-function module.switch_callback(_, on)
-	timeout:cancel()
-	callback(on)
-	forced = on
-end
-
----@type fun(device: DeviceInterface, open: boolean)
-function module.door_callback(_, open)
-	if open then
-		timeout:cancel()
-
-		callback(true)
-	elseif not forced then
-		timeout:start(debug.debug_mode and 10 or 2 * 60, function()
-			if trash == nil or trash:open_percent() == 0 then
-				callback(false)
-			end
-		end)
-	end
-end
-
----@type fun(device: DeviceInterface, open: boolean)
-function module.trash_callback(_, open)
-	if open then
-		callback(true)
-	else
-		if not forced and not timeout:is_waiting() and (door == nil or door:open_percent() == 0) then
-			callback(false)
-		end
-	end
-end
-
----@type fun(device: DeviceInterface, state: { state: boolean })
-function module.light_callback(_, state)
-	print("LIGHT = " .. tostring(state.state))
-	if state.state and (trash == nil or trash:open_percent()) == 0 and (door == nil or door:open_percent() == 0) then
-		-- If the door and trash are not open, that means the light got turned on manually
-		timeout:cancel()
-		forced = true
-	elseif not state.state then
-		-- The light is never forced when it is off
-		forced = false
-	end
-end
-
---- @param t OpenCloseInterface
-function module.set_trash(t)
-	trash = t
-end
-
---- @param d OpenCloseInterface
-function module.set_door(d)
-	door = d
-end
-
---- @param c fun(on: boolean)
-function module.add_callback(c)
-	table.insert(callbacks, c)
-end
-
-return module

config/helper.lua

@@ -1,49 +0,0 @@
-local utils = require("automation:utils")
-
---- @class HelperModule: Module
-local module = {}
-
---- @param topic string
---- @return string
-function module.mqtt_z2m(topic)
-	return "zigbee2mqtt/" .. topic
-end
-
---- @param topic string
---- @return string
-function module.mqtt_automation(topic)
-	return "automation/" .. topic
-end
-
---- @return fun(self: OnOffInterface, state: {state: boolean, power: number})
-function module.auto_off()
-	local timeout = utils.Timeout.new()
-
-	return function(self, state)
-		if state.state and state.power < 100 then
-			timeout:start(3, function()
-				self:set_on(false)
-			end)
-		else
-			timeout:cancel()
-		end
-	end
-end
-
---- @param duration number
---- @return fun(self: OnOffInterface, state: {state: boolean})
-function module.off_timeout(duration)
-	local timeout = utils.Timeout.new()
-
-	return function(self, state)
-		if state.state then
-			timeout:start(duration, function()
-				self:set_on(false)
-			end)
-		else
-			timeout:cancel()
-		end
-	end
-end
-
-return module

config/hue_bridge.lua

@@ -1,41 +0,0 @@
-local devices = require("automation:devices")
-local light = require("config.light")
-local presence = require("config.presence")
-local secrets = require("automation:secrets")
-
---- @class HueBridgeModule: Module
-local module = {}
-
-module.ip = "10.0.0.102"
-module.token = secrets.hue_token
-
-if module.token == nil then
-	error("Hue token is not specified")
-end
-
---- @type SetupFunction
-function module.setup()
-	local bridge = devices.HueBridge.new({
-		identifier = "hue_bridge",
-		ip = module.ip,
-		login = module.token,
-		flags = {
-			presence = 41,
-			darkness = 43,
-		},
-	})
-
-	light.add_callback(function(l)
-		bridge:set_flag("darkness", not l)
-	end)
-
-	presence.add_callback(function(p)
-		bridge:set_flag("presence", p)
-	end)
-
-	return {
-		bridge,
-	}
-end
-
-return module

config/light.lua

@@ -1,44 +0,0 @@
-local devices = require("automation:devices")
-local helper = require("config.helper")
-
---- @class LightModule: Module
-local module = {}
-
---- @class OnPresence
---- @field [integer] fun(light: boolean)
-local callbacks = {}
-
---- @param callback fun(light: boolean)
-function module.add_callback(callback)
-	table.insert(callbacks, callback)
-end
-
---- @param _ DeviceInterface
---- @param light boolean
-local function callback(_, light)
-	for _, f in ipairs(callbacks) do
-		f(light)
-	end
-end
-
---- @type LightSensor?
-module.device = nil
-
---- @type SetupFunction
-function module.setup(mqtt_client)
-	module.device = devices.LightSensor.new({
-		identifier = "living_light_sensor",
-		topic = helper.mqtt_z2m("living/light"),
-		client = mqtt_client,
-		min = 22000,
-		max = 23500,
-		callback = callback,
-	})
-
-	--- @type Module
-	return {
-		module.device,
-	}
-end
-
-return module

config/ntfy.lua

@@ -1,34 +0,0 @@
-local devices = require("automation:devices")
-local secrets = require("automation:secrets")
-
---- @class NtfyModule: Module
-local module = {}
-
-local ntfy_topic = secrets.ntfy_topic
-if ntfy_topic == nil then
-	error("Ntfy topic is not specified")
-end
-
---- @type Ntfy?
-local ntfy = nil
-
---- @param notification Notification
-function module.send_notification(notification)
-	if ntfy then
-		ntfy:send_notification(notification)
-	end
-end
-
---- @type SetupFunction
-function module.setup()
-	ntfy = devices.Ntfy.new({
-		topic = ntfy_topic,
-	})
-
-	--- @type Module
-	return {
-		ntfy,
-	}
-end
-
-return module

config/presence.lua

@@ -1,80 +0,0 @@
-local devices = require("automation:devices")
-local helper = require("config.helper")
-local ntfy = require("config.ntfy")
-
---- @class PresenceModule: Module
-local module = {}
-
---- @class OnPresence
---- @field [integer] fun(presence: boolean)
-local callbacks = {}
-
---- @param callback fun(presence: boolean)
-function module.add_callback(callback)
-	table.insert(callbacks, callback)
-end
-
---- @param device OnOffInterface
-function module.turn_off_when_away(device)
-	module.add_callback(function(presence)
-		if not presence then
-			device:set_on(false)
-		end
-	end)
-end
-
---- @param _ DeviceInterface
---- @param presence boolean
-local function callback(_, presence)
-	for _, f in ipairs(callbacks) do
-		f(presence)
-	end
-end
-
---- @type Presence?
-local presence = nil
-
---- @type SetupFunction
-function module.setup(mqtt_client)
-	presence = devices.Presence.new({
-		topic = helper.mqtt_automation("presence/+/#"),
-		client = mqtt_client,
-		callback = callback,
-	})
-
-	module.add_callback(function(p)
-		ntfy.send_notification({
-			title = "Presence",
-			message = p and "Home" or "Away",
-			tags = { "house" },
-			priority = "low",
-			actions = {
-				{
-					action = "broadcast",
-					extras = {
-						cmd = "presence",
-						state = p and "0" or "1",
-					},
-					label = p and "Set away" or "Set home",
-					clear = true,
-				},
-			},
-		})
-	end)
-
-	--- @type Module
-	return {
-		presence,
-	}
-end
-
-function module.overall_presence()
-	-- Default to no presence when the device has not been created yet
-	if not presence then
-		return false
-	end
-
-	return presence:overall_presence()
-end
-
-return module

config/rooms.lua

@@ -1,12 +0,0 @@
---- @type Module
-return {
-	require("config.rooms.bathroom"),
-	require("config.rooms.bedroom"),
-	require("config.rooms.guest_bedroom"),
-	require("config.rooms.hallway_bottom"),
-	require("config.rooms.hallway_top"),
-	require("config.rooms.kitchen"),
-	require("config.rooms.living_room"),
-	require("config.rooms.storage"),
-	require("config.rooms.workbench"),
-}

config/rooms/bathroom.lua

@@ -1,40 +0,0 @@
-local debug = require("config.debug")
-local devices = require("automation:devices")
-local helper = require("config.helper")
-local ntfy = require("config.ntfy")
-
---- @type Module
-local module = {}
-
-function module.setup(mqtt_client)
-	local light = devices.LightOnOff.new({
-		name = "Light",
-		room = "Bathroom",
-		topic = helper.mqtt_z2m("bathroom/light"),
-		client = mqtt_client,
-		callback = helper.off_timeout(debug.debug_mode and 60 or 45 * 60),
-	})
-
-	local washer = devices.Washer.new({
-		identifier = "bathroom_washer",
-		topic = helper.mqtt_z2m("bathroom/washer"),
-		client = mqtt_client,
-		threshold = 1,
-		done_callback = function()
-			ntfy.send_notification({
-				title = "Laundy is done",
-				message = "Don't forget to hang it!",
-				tags = { "womans_clothes" },
-				priority = "high",
-			})
-		end,
-	})
-
-	--- @type Module
-	return {
-		light,
-		washer,
-	}
-end
-
-return module

config/rooms/bedroom.lua

@@ -1,78 +0,0 @@
-local battery = require("config.battery")
-local devices = require("automation:devices")
-local helper = require("config.helper")
-local hue_bridge = require("config.hue_bridge")
-local windows = require("config.windows")
-
---- @type Module
-local module = {}
-
---- @type AirFilter?
-local air_filter = nil
-
-function module.setup(mqtt_client)
-	local lights = devices.HueGroup.new({
-		identifier = "bedroom_lights",
-		ip = hue_bridge.ip,
-		login = hue_bridge.token,
-		group_id = 3,
-		scene_id = "PvRs-lGD4VRytL9",
-	})
-	local lights_relax = devices.HueGroup.new({
-		identifier = "bedroom_lights_relax",
-		ip = hue_bridge.ip,
-		login = hue_bridge.token,
-		group_id = 3,
-		scene_id = "60tfTyR168v2csz",
-	})
-
-	air_filter = devices.AirFilter.new({
-		name = "Air Filter",
-		room = "Bedroom",
-		url = "http://10.0.0.103",
-	})
-
-	local switch = devices.HueSwitch.new({
-		name = "Switch",
-		room = "Bedroom",
-		client = mqtt_client,
-		topic = helper.mqtt_z2m("bedroom/switch"),
-		left_callback = function()
-			lights:set_on(not lights:on())
-		end,
-		left_hold_callback = function()
-			lights_relax:set_on(true)
-		end,
-		battery_callback = battery.callback,
-	})
-
-	local window = devices.ContactSensor.new({
-		name = "Window",
-		room = "Bedroom",
-		topic = helper.mqtt_z2m("bedroom/window"),
-		client = mqtt_client,
-		battery_callback = battery.callback,
-	})
-	windows.add(window)
-
-	--- @type Module
-	return {
-		devices = {
-			lights,
-			lights_relax,
-			air_filter,
-			switch,
-			window,
-		},
-		schedule = {
-			["0 0 19 * * *"] = function()
-				air_filter:set_on(true)
-			end,
-			["0 0 20 * * *"] = function()
-				air_filter:set_on(false)
-			end,
-		},
-	}
-end
-
-return module

config/rooms/guest_bedroom.lua

@@ -1,35 +0,0 @@
-local battery = require("config.battery")
-local devices = require("automation:devices")
-local helper = require("config.helper")
-local presence = require("config.presence")
-local windows = require("config.windows")
-
---- @type Module
-local module = {}
-
-function module.setup(mqtt_client)
-	local light = devices.LightOnOff.new({
-		name = "Light",
-		room = "Guest Room",
-		topic = helper.mqtt_z2m("guest/light"),
-		client = mqtt_client,
-	})
-	presence.turn_off_when_away(light)
-
-	local window = devices.ContactSensor.new({
-		name = "Window",
-		room = "Guest Room",
-		topic = helper.mqtt_z2m("guest/window"),
-		client = mqtt_client,
-		battery_callback = battery.callback,
-	})
-	windows.add(window)
-
-	--- @type Module
-	return {
-		light,
-		window,
-	}
-end
-
-return module

config/rooms/hallway_bottom.lua

@@ -1,110 +0,0 @@
-local battery = require("config.battery")
-local debug = require("config.debug")
-local devices = require("automation:devices")
-local hallway_automation = require("config.hallway_automation")
-local helper = require("config.helper")
-local hue_bridge = require("config.hue_bridge")
-local presence = require("config.presence")
-local utils = require("automation:utils")
-local windows = require("config.windows")
-
---- @type Module
-local module = {}
-
-function module.setup(mqtt_client)
-	local main_light = devices.HueGroup.new({
-		identifier = "hallway_main_light",
-		ip = hue_bridge.ip,
-		login = hue_bridge.token,
-		group_id = 81,
-		scene_id = "3qWKxGVadXFFG4o",
-	})
-	hallway_automation.add_callback(function(on)
-		main_light:set_on(on)
-	end)
-
-	local storage_light = devices.LightBrightness.new({
-		name = "Storage",
-		room = "Hallway",
-		topic = helper.mqtt_z2m("hallway/storage"),
-		client = mqtt_client,
-		callback = hallway_automation.light_callback,
-	})
-	presence.turn_off_when_away(storage_light)
-	hallway_automation.add_callback(function(on)
-		if on then
-			storage_light:set_brightness(80)
-		else
-			storage_light:set_on(false)
-		end
-	end)
-
-	local remote = devices.IkeaRemote.new({
-		name = "Remote",
-		room = "Hallway",
-		client = mqtt_client,
-		topic = helper.mqtt_z2m("hallway/remote"),
-		callback = hallway_automation.switch_callback,
-		battery_callback = battery.callback,
-	})
-
-	local trash = devices.ContactSensor.new({
-		name = "Trash",
-		room = "Hallway",
-		sensor_type = "Drawer",
-		topic = helper.mqtt_z2m("hallway/trash"),
-		client = mqtt_client,
-		callback = hallway_automation.trash_callback,
-		battery_callback = battery.callback,
-	})
-	hallway_automation.set_trash(trash)
-
-	---@param duration number
-	---@return fun(_, open: boolean)
-	local function frontdoor_presence(duration)
-		local timeout = utils.Timeout.new()
-
-		return function(_, open)
-			if open then
-				timeout:cancel()
-
-				if presence.overall_presence() then
-					mqtt_client:send_message(helper.mqtt_automation("presence/contact/frontdoor"), {
-						state = true,
-						updated = utils.get_epoch(),
-					})
-				end
-			else
-				timeout:start(duration, function()
-					mqtt_client:send_message(helper.mqtt_automation("presence/contact/frontdoor"), nil)
-				end)
-			end
-		end
-	end
-
-	local frontdoor = devices.ContactSensor.new({
-		name = "Frontdoor",
-		room = "Hallway",
-		sensor_type = "Door",
-		topic = helper.mqtt_z2m("hallway/frontdoor"),
-		client = mqtt_client,
-		callback = {
-			frontdoor_presence(debug.debug_mode and 10 or 15 * 60),
-			hallway_automation.door_callback,
-		},
-		battery_callback = battery.callback,
-	})
-	windows.add(frontdoor)
-	hallway_automation.set_door(frontdoor)
-
-	--- @type Module
-	return {
-		main_light,
-		storage_light,
-		remote,
-		trash,
-		frontdoor,
-	}
-end
-
-return module

config/rooms/hallway_top.lua

@@ -1,48 +0,0 @@
-local battery = require("config.battery")
-local devices = require("automation:devices")
-local helper = require("config.helper")
-local hue_bridge = require("config.hue_bridge")
-
---- @type Module
-local module = {}
-
-function module.setup(mqtt_client)
-	local light = devices.HueGroup.new({
-		identifier = "hallway_top_light",
-		ip = hue_bridge.ip,
-		login = hue_bridge.token,
-		group_id = 83,
-		scene_id = "QeufkFDICEHWeKJ7",
-	})
-
-	local top_switch = devices.HueSwitch.new({
-		name = "SwitchTop",
-		room = "Hallway",
-		client = mqtt_client,
-		topic = helper.mqtt_z2m("hallway/switchtop"),
-		left_callback = function()
-			light:set_on(not light:on())
-		end,
-		battery_callback = battery.callback,
-	})
-
-	local bottom_switch = devices.HueSwitch.new({
-		name = "SwitchBottom",
-		room = "Hallway",
-		client = mqtt_client,
-		topic = helper.mqtt_z2m("hallway/switchbottom"),
-		left_callback = function()
-			light:set_on(not light:on())
-		end,
-		battery_callback = battery.callback,
-	})
-
-	--- @type Module
-	return {
-		light,
-		top_switch,
-		bottom_switch,
-	}
-end
-
-return module

config/rooms/kitchen.lua

@@ -1,71 +0,0 @@
-local battery = require("config.battery")
-local devices = require("automation:devices")
-local helper = require("config.helper")
-local hue_bridge = require("config.hue_bridge")
-local presence = require("config.presence")
-
---- @class KitchenModule: Module
-local module = {}
-
---- @type HueGroup?
-local lights = nil
-
---- @type SetupFunction
-function module.setup(mqtt_client)
-	lights = devices.HueGroup.new({
-		identifier = "kitchen_lights",
-		ip = hue_bridge.ip,
-		login = hue_bridge.token,
-		group_id = 7,
-		scene_id = "7MJLG27RzeRAEVJ",
-	})
-
-	local kettle = devices.OutletPower.new({
-		outlet_type = "Kettle",
-		name = "Kettle",
-		room = "Kitchen",
-		topic = helper.mqtt_z2m("kitchen/kettle"),
-		client = mqtt_client,
-		callback = helper.auto_off(),
-	})
-	presence.turn_off_when_away(kettle)
-
-	local kettle_remote = devices.IkeaRemote.new({
-		name = "Remote",
-		room = "Kitchen",
-		client = mqtt_client,
-		topic = helper.mqtt_z2m("kitchen/remote"),
-		single_button = true,
-		callback = function(_, on)
-			kettle:set_on(on)
-		end,
-		battery_callback = battery.callback,
-	})
-
-	local kettle_remote_bedroom = devices.IkeaRemote.new({
-		name = "Remote",
-		room = "Bedroom",
-		client = mqtt_client,
-		topic = helper.mqtt_z2m("bedroom/remote"),
-		single_button = true,
-		callback = function(_, on)
-			kettle:set_on(on)
-		end,
-		battery_callback = battery.callback,
-	})
-
-	return {
-		lights,
-		kettle,
-		kettle_remote,
-		kettle_remote_bedroom,
-	}
-end
-
-function module.toggle_lights()
-	if lights then
-		lights:set_on(not lights:on())
-	end
-end
-
-return module

config/rooms/living_room.lua

@@ -1,126 +0,0 @@
-local battery = require("config.battery")
-local devices = require("automation:devices")
-local helper = require("config.helper")
-local hue_bridge = require("config.hue_bridge")
-local presence = require("config.presence")
-local windows = require("config.windows")
-
---- @type Module
-local module = {}
-
-function module.setup(mqtt_client)
-	local lights = devices.HueGroup.new({
-		identifier = "living_lights",
-		ip = hue_bridge.ip,
-		login = hue_bridge.token,
-		group_id = 1,
-		scene_id = "SNZw7jUhQ3cXSjkj",
-	})
-
-	local lights_relax = devices.HueGroup.new({
-		identifier = "living_lights_relax",
-		ip = hue_bridge.ip,
-		login = hue_bridge.token,
-		group_id = 1,
-		scene_id = "eRJ3fvGHCcb6yNw",
-	})
-
-	local switch = devices.HueSwitch.new({
-		name = "Switch",
-		room = "Living",
-		client = mqtt_client,
-		topic = helper.mqtt_z2m("living/switch"),
-		left_callback = require("config.rooms.kitchen").toggle_lights,
-		right_callback = function()
-			lights:set_on(not lights:on())
-		end,
-		right_hold_callback = function()
-			lights_relax:set_on(true)
-		end,
-		battery_callback = battery.callback,
-	})
-
-	local pc = devices.WakeOnLAN.new({
-		name = "Zeus",
-		room = "Living Room",
-		topic = helper.mqtt_automation("appliance/living_room/zeus"),
-		client = mqtt_client,
-		mac_address = "30:9c:23:60:9c:13",
-		broadcast_ip = "10.0.3.255",
-	})
-
-	local mixer = devices.OutletOnOff.new({
-		name = "Mixer",
-		room = "Living Room",
-		topic = helper.mqtt_z2m("living/mixer"),
-		client = mqtt_client,
-	})
-	presence.turn_off_when_away(mixer)
-
-	local speakers = devices.OutletOnOff.new({
-		name = "Speakers",
-		room = "Living Room",
-		topic = helper.mqtt_z2m("living/speakers"),
-		client = mqtt_client,
-	})
-	presence.turn_off_when_away(speakers)
-
-	local audio_remote = devices.IkeaRemote.new({
-		name = "Remote",
-		room = "Living Room",
-		client = mqtt_client,
-		topic = helper.mqtt_z2m("living/remote"),
-		single_button = true,
-		callback = function(_, on)
-			if on then
-				if mixer:on() then
-					mixer:set_on(false)
-					speakers:set_on(false)
-				else
-					mixer:set_on(true)
-					speakers:set_on(true)
-				end
-			else
-				if not mixer:on() then
-					mixer:set_on(true)
-				else
-					speakers:set_on(not speakers:on())
-				end
-			end
-		end,
-		battery_callback = battery.callback,
-	})
-
-	local balcony = devices.ContactSensor.new({
-		name = "Balcony",
-		room = "Living Room",
-		sensor_type = "Door",
-		topic = helper.mqtt_z2m("living/balcony"),
-		client = mqtt_client,
-		battery_callback = battery.callback,
-	})
-	windows.add(balcony)
-	local window = devices.ContactSensor.new({
-		name = "Window",
-		room = "Living Room",
-		topic = helper.mqtt_z2m("living/window"),
-		client = mqtt_client,
-		battery_callback = battery.callback,
-	})
-	windows.add(window)
-
-	--- @type Module
-	return {
-		lights,
-		lights_relax,
-		switch,
-		pc,
-		mixer,
-		speakers,
-		audio_remote,
-		balcony,
-		window,
-	}
-end
-
-return module

config/rooms/storage.lua

@@ -1,41 +0,0 @@
-local battery = require("config.battery")
-local devices = require("automation:devices")
-local helper = require("config.helper")
-local presence = require("config.presence")
-
---- @type Module
-local module = {}
-
-function module.setup(mqtt_client)
-	local light = devices.LightBrightness.new({
-		name = "Light",
-		room = "Storage",
-		topic = helper.mqtt_z2m("storage/light"),
-		client = mqtt_client,
-	})
-	presence.turn_off_when_away(light)
-
-	local door = devices.ContactSensor.new({
-		name = "Door",
-		room = "Storage",
-		sensor_type = "Door",
-		topic = helper.mqtt_z2m("storage/door"),
-		client = mqtt_client,
-		callback = function(_, open)
-			if open then
-				light:set_brightness(100)
-			else
-				light:set_on(false)
-			end
-		end,
-		battery_callback = battery.callback,
-	})
-
-	--- @type Module
-	return {
-		light,
-		door,
-	}
-end
-
-return module

config/rooms/workbench.lua

@@ -1,69 +0,0 @@
-local battery = require("config.battery")
-local debug = require("config.debug")
-local devices = require("automation:devices")
-local helper = require("config.helper")
-local presence = require("config.presence")
-local utils = require("automation:utils")
-
---- @type Module
-local module = {}
-
-function module.setup(mqtt_client)
-	local charger = devices.OutletOnOff.new({
-		name = "Charger",
-		room = "Workbench",
-		topic = helper.mqtt_z2m("workbench/charger"),
-		client = mqtt_client,
-		callback = helper.off_timeout(debug.debug_mode and 5 or 20 * 3600),
-	})
-
-	local outlets = devices.OutletOnOff.new({
-		name = "Outlets",
-		room = "Workbench",
-		topic = helper.mqtt_z2m("workbench/outlet"),
-		client = mqtt_client,
-	})
-	presence.turn_off_when_away(outlets)
-
-	local light = devices.LightColorTemperature.new({
-		name = "Light",
-		room = "Workbench",
-		topic = helper.mqtt_z2m("workbench/light"),
-		client = mqtt_client,
-	})
-	presence.turn_off_when_away(light)
-
-	local delay_color_temp = utils.Timeout.new()
-	local remote = devices.IkeaRemote.new({
-		name = "Remote",
-		room = "Workbench",
-		client = mqtt_client,
-		topic = helper.mqtt_z2m("workbench/remote"),
-		callback = function(_, on)
-			delay_color_temp:cancel()
-			if on then
-				light:set_brightness(82)
-				-- NOTE: This light does NOT support changing both the brightness and color
-				-- temperature at the same time, so we first change the brightness and once
-				-- that is complete we change the color temperature, as that is less likely
-				-- to have to actually change.
-				delay_color_temp:start(0.5, function()
-					light:set_color_temperature(3333)
-				end)
-			else
-				light:set_on(false)
-			end
-		end,
-		battery_callback = battery.callback,
-	})
-
-	--- @type Module
-	return {
-		charger,
-		outlets,
-		light,
-		remote,
-	}
-end
-
-return module

config/windows.lua

@@ -1,43 +0,0 @@
-local ntfy = require("config.ntfy")
-local presence = require("config.presence")
-
---- @class WindowsModule: Module
-local module = {}
-
---- @class OnPresence
---- @field [integer] OpenCloseInterface
-local sensors = {}
-
---- @param sensor OpenCloseInterface
-function module.add(sensor)
-	table.insert(sensors, sensor)
-end
-
---- @type SetupFunction
-function module.setup()
-	presence.add_callback(function(p)
-		if not p then
-			local open = {}
-			for _, sensor in ipairs(sensors) do
-				if sensor:open_percent() > 0 then
-					local id = sensor:get_id()
-					print("Open window detected: " .. id)
-					table.insert(open, id)
-				end
-			end
-
-			if #open > 0 then
-				local message = table.concat(open, "\n")
-
-				ntfy.send_notification({
-					title = "Windows are open",
-					message = message,
-					tags = { "window" },
-					priority = "high",
-				})
-			end
-		end
-	end)
-end
-
-return module

definitions/automation:devices.lua

@@ -1,337 +0,0 @@
--- DO NOT MODIFY, FILE IS AUTOMATICALLY GENERATED
----@meta
-
-local devices
-
----@class Action
----@field action
----| "broadcast"
----@field extras (table<string, string>)?
----@field label string
----@field clear (boolean)?
-local Action
-
----@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 ConfigLightLightStateBrightness
----@field name string
----@field room (string)?
----@field topic string
----@field callback (fun(_: LightBrightness, _: LightStateBrightness) | fun(_: LightBrightness, _: LightStateBrightness)[])?
----@field client (AsyncClient)?
-local ConfigLightLightStateBrightness
-
----@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 ConfigLightLightStateOnOff
----@field name string
----@field room (string)?
----@field topic string
----@field callback (fun(_: LightOnOff, _: LightStateOnOff) | fun(_: LightOnOff, _: LightStateOnOff)[])?
----@field client (AsyncClient)?
-local ConfigLightLightStateOnOff
-
----@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 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
-
----@class ContactSensor: DeviceInterface, OpenCloseInterface
-local ContactSensor
-devices.ContactSensor = {}
----@param config ContactSensorConfig
----@return ContactSensor
-function devices.ContactSensor.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 Flag
----| "presence"
----| "darkness"
-
----@class FlagIDs
----@field presence integer
----@field darkness integer
-local FlagIDs
-
----@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 HueBridgeConfig
----@field identifier string
----@field ip string
----@field login string
----@field flags FlagIDs
-local HueBridgeConfig
-
----@class HueGroup: DeviceInterface, OnOffInterface
-local HueGroup
-devices.HueGroup = {}
----@param config HueGroupConfig
----@return HueGroup
-function devices.HueGroup.new(config) end
-
----@class HueGroupConfig
----@field identifier string
----@field ip string
----@field login string
----@field group_id integer
----@field scene_id string
-local HueGroupConfig
-
----@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
-
----@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 KasaOutlet: DeviceInterface, OnOffInterface
-local KasaOutlet
-devices.KasaOutlet = {}
----@param config KasaOutletConfig
----@return KasaOutlet
-function devices.KasaOutlet.new(config) end
-
----@class KasaOutletConfig
----@field identifier string
----@field ip string
-local KasaOutletConfig
-
----@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 LightOnOff: DeviceInterface, OnOffInterface
-local LightOnOff
-devices.LightOnOff = {}
----@param config ConfigLightLightStateOnOff
----@return LightOnOff
-function devices.LightOnOff.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 LightStateBrightness
----@field state boolean
----@field brightness number
-local LightStateBrightness
-
----@class LightStateColorTemperature
----@field state boolean
----@field brightness number
----@field color_temp integer
-local LightStateColorTemperature
-
----@class LightStateOnOff
----@field state boolean
-local LightStateOnOff
-
----@class Notification
----@field title string
----@field message (string)?
----@field tags ((string)[])?
----@field priority (Priority)?
----@field actions ((Action)[])?
-local Notification
-
----@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 NtfyConfig
----@field url (string)?
----@field topic string
-local NtfyConfig
-
----@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 OutletStateOnOff
----@field state boolean
-local OutletStateOnOff
-
----@class OutletStatePower
----@field state boolean
----@field power number
-local OutletStatePower
-
----@alias OutletType
----| "Outlet"
----| "Kettle"
-
----@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
-
----@alias Priority
----| "min"
----| "low"
----| "default"
----| "high"
----| "max"
-
----@alias SensorType
----| "Door"
----| "Drawer"
----| "Window"
-
----@class WakeOnLAN: DeviceInterface
-local WakeOnLAN
-devices.WakeOnLAN = {}
----@param config WolConfig
----@return WakeOnLAN
-function devices.WakeOnLAN.new(config) end
-
----@class Washer: DeviceInterface
-local Washer
-devices.Washer = {}
----@param config WasherConfig
----@return Washer
-function devices.Washer.new(config) end
-
----@class WasherConfig
----@field identifier string
----@field topic string
----@field threshold number
----@field done_callback (fun(_: Washer) | fun(_: Washer)[])?
----@field client AsyncClient
-local WasherConfig
-
----@class WolConfig
----@field name string
----@field room (string)?
----@field topic string
----@field mac_address string
----@field broadcast_ip (string)?
----@field client AsyncClient
-local WolConfig
-
-return devices

definitions/automation:secrets.lua

@@ -1,6 +0,0 @@
----@meta
-
----@type table<string, string?>
-local secrets
-
-return secrets

definitions/automation:utils.lua

@@ -1,27 +0,0 @@
--- 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

definitions/automation:variables.lua

@@ -1,6 +0,0 @@
----@meta
-
----@type table<string, string?>
-local variables
-
-return variables

definitions/config.lua

@@ -1,41 +0,0 @@
--- DO NOT MODIFY, FILE IS AUTOMATICALLY GENERATED
----@meta
-
----@class FulfillmentConfig
----@field openid_url string
----@field ip (string)?
----@field port (integer)?
-local FulfillmentConfig
-
----@class Config
----@field fulfillment FulfillmentConfig
----@field modules (Module)[]
----@field mqtt MqttConfig
-local Config
-
----@alias SetupFunction fun(mqtt_client: AsyncClient): Module | DeviceInterface[] | nil
-
----@alias Schedule table<string, fun() | fun()[]>
-
----@class Module
----@field setup (SetupFunction)?
----@field devices (DeviceInterface)[]?
----@field schedule Schedule?
----@field [number] (Module)[]?
-local Module
-
----@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

definitions/definitions.lua

@@ -0,0 +1,40 @@
+--- @meta
+
+--- @class WrappedDevice
+WrappedDevice = {}
+--- @return string
+function WrappedDevice:get_id() end
+
+--- @class WrappedAsyncClient
+
+--- @class EventChannel
+--- @return EventChannel
+function automation.device_manager:event_channel() end
+
+automation = {}
+
+automation.device_manager = {}
+--- @param device WrappedDevice
+function automation.device_manager:add(device) end
+
+--- @param when string
+--- @param func function
+function automation.device_manager:schedule(when, func) end
+
+automation.util = {}
+--- @param env string
+--- @return string
+function automation.util.get_env(env) end
+
+--- @class Fulfillment
+--- @field openid_url string|nil
+automation.fulfillment = {}
+
+--- @class MqttConfig
+--- @param config MqttConfig
+--- @return WrappedAsyncClient
+function automation.new_mqtt_client(config) end
+
+--- TODO: Generate this automatically
+--- @alias OutletType "Outlet"|"Kettle"|"Charger"|"Light"
+--- @alias TriggerDevicesHelper WrappedDevice[]

definitions/generated.lua

@@ -0,0 +1,184 @@
+-- WARN: This file is automatically generated, do not manually edit
+
+---@meta
+--- @class MqttDeviceConfig
+--- @field topic String
+
+--- @class AirFilter
+AirFilter = {}
+--- @param config AirFilterConfig
+--- @return WrappedDevice
+function AirFilter.new(config) end
+
+--- @class AirFilterConfig
+--- @field name String
+--- @field room String|nil
+--- @field topic String
+--- @field client WrappedAsyncClient
+
+--- @class AudioSetup
+AudioSetup = {}
+--- @param config AudioSetupConfig
+--- @return WrappedDevice
+function AudioSetup.new(config) end
+
+--- @class AudioSetupConfig
+--- @field identifier String
+--- @field topic String
+--- @field mixer WrappedDevice
+--- @field speakers WrappedDevice
+--- @field client WrappedAsyncClient
+
+--- @class ContactSensor
+ContactSensor = {}
+--- @param config ContactSensorConfig
+--- @return WrappedDevice
+function ContactSensor.new(config) end
+
+--- @class ContactSensorConfig
+--- @field identifier String
+--- @field topic String
+--- @field presence PresenceDeviceConfig|nil
+--- @field trigger TriggerConfig|nil
+--- @field client WrappedAsyncClient
+
+--- @class PresenceDeviceConfig
+--- @field topic String
+--- @field timeout u64
+
+--- @class TriggerConfig
+--- @field devices TriggerDevicesHelper
+--- @field timeout u64|nil
+
+--- @class DebugBridge
+DebugBridge = {}
+--- @param config DebugBridgeConfig
+--- @return WrappedDevice
+function DebugBridge.new(config) end
+
+--- @class DebugBridgeConfig
+--- @field identifier String
+--- @field topic String
+--- @field client WrappedAsyncClient
+
+--- @class HueBridge
+HueBridge = {}
+--- @param config HueBridgeConfig
+--- @return WrappedDevice
+function HueBridge.new(config) end
+
+--- @class HueBridgeConfig
+--- @field identifier String
+--- @field ip Ipv4Addr
+--- @field login String
+--- @field flags FlagIDs
+
+--- @class FlagIDs
+--- @field presence isize
+--- @field darkness isize
+
+--- @class HueGroup
+HueGroup = {}
+--- @param config HueGroupConfig
+--- @return WrappedDevice
+function HueGroup.new(config) end
+
+--- @class HueGroupConfig
+--- @field identifier String
+--- @field ip Ipv4Addr
+--- @field login String
+--- @field group_id isize
+--- @field timer_id isize
+--- @field scene_id String
+--- @field remotes MqttDeviceConfig []|nil
+--- @field client WrappedAsyncClient
+
+--- @class IkeaOutlet
+IkeaOutlet = {}
+--- @param config IkeaOutletConfig
+--- @return WrappedDevice
+function IkeaOutlet.new(config) end
+
+--- @class IkeaOutletConfig
+--- @field name String
+--- @field room String|nil
+--- @field topic String
+--- @field outlet_type OutletType|nil
+--- @field timeout u64|nil
+--- @field remotes MqttDeviceConfig []|nil
+--- @field client WrappedAsyncClient
+
+--- @class KasaOutlet
+KasaOutlet = {}
+--- @param config KasaOutletConfig
+--- @return WrappedDevice
+function KasaOutlet.new(config) end
+
+--- @class KasaOutletConfig
+--- @field identifier String
+--- @field ip Ipv4Addr
+
+--- @class LightSensor
+LightSensor = {}
+--- @param config LightSensorConfig
+--- @return WrappedDevice
+function LightSensor.new(config) end
+
+--- @class LightSensorConfig
+--- @field identifier String
+--- @field topic String
+--- @field min isize
+--- @field max isize
+--- @field event_channel EventChannel
+--- @field client WrappedAsyncClient
+
+--- @class Ntfy
+Ntfy = {}
+--- @param config NtfyConfig
+--- @return WrappedDevice
+function Ntfy.new(config) end
+
+--- @class NtfyConfig
+--- @field url String|nil
+--- @field topic String
+--- @field event_channel EventChannel
+
+--- @class Presence
+Presence = {}
+--- @param config PresenceConfig
+--- @return WrappedDevice
+function Presence.new(config) end
+
+--- @class PresenceConfig
+--- @field topic String
+--- @field event_channel EventChannel
+--- @field client WrappedAsyncClient
+
+--- @class WakeOnLAN
+WakeOnLAN = {}
+--- @param config WakeOnLANConfig
+--- @return WrappedDevice
+function WakeOnLAN.new(config) end
+
+--- @class WakeOnLANConfig
+--- @field name String
+--- @field room String|nil
+--- @field topic String
+--- @field mac_address MacAddress
+--- @field broadcast_ip Ipv4Addr|nil
+--- @field client WrappedAsyncClient
+
+--- @class Washer
+Washer = {}
+--- @param config WasherConfig
+--- @return WrappedDevice
+function Washer.new(config) end
+
+--- @class WasherConfig
+--- @field identifier String
+--- @field topic String
+--- @field threshold f32
+--- @field event_channel EventChannel
+--- @field client WrappedAsyncClient
+
+-- End of generated file

definitions/interfaces.lua

@@ -1,42 +0,0 @@
---- @meta
-
----@class DeviceInterface
-local DeviceInterface
----@return string
-function DeviceInterface:get_id() end
-
----@class OnOffInterface: DeviceInterface
-local OnOffInterface
----@async
----@param on boolean
-function OnOffInterface:set_on(on) end
----@async
----@return boolean
-function OnOffInterface:on() end
-
----@class BrightnessInterface: DeviceInterface
-local BrightnessInterface
----@async
----@param brightness integer
-function BrightnessInterface:set_brightness(brightness) end
----@async
----@return integer
-function BrightnessInterface:brightness() end
-
----@class ColorSettingInterface: DeviceInterface
-local ColorSettingInterface
----@async
----@param temperature integer
-function ColorSettingInterface:set_color_temperature(temperature) end
----@async
----@return integer
-function ColorSettingInterface:color_temperature() end
-
----@class OpenCloseInterface: DeviceInterface
-local OpenCloseInterface
----@async
----@param open_percent integer
-function OpenCloseInterface:set_open_percent(open_percent) end
----@async
----@return integer
-function OpenCloseInterface:open_percent() end

definitions/rust.lua

@@ -0,0 +1,10 @@
+--- @meta
+
+--- @alias String string
+
+--- @alias u64 number
+--- @alias isize number
+--- @alias f32 number
+
+--- @alias Ipv4Addr string
+--- @alias MacAddress string

google-home/Cargo.toml

@@ -0,0 +1,16 @@
+[package]
+name = "google-home"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+automation_cast = { path = "../automation_cast/" }
+serde = { version = "1.0.149", features = ["derive"] }
+serde_json = "1.0.89"
+thiserror = "1.0.37"
+tokio = { version = "1", features = ["sync"] }
+async-trait = "0.1.61"
+futures = "0.3.25"
+anyhow = "1.0.75"

google-home/src/attributes.rs

@@ -0,0 +1,22 @@
+use serde::Serialize;
+
+use crate::traits::AvailableSpeeds;
+
+#[derive(Debug, Default, Serialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Attributes {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub command_only_on_off: Option<bool>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub query_only_on_off: Option<bool>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub scene_reversible: Option<bool>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub reversible: Option<bool>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub command_only_fan_speed: Option<bool>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub available_fan_speeds: Option<AvailableSpeeds>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub query_only_humidity_setting: Option<bool>,
+}

google-home/src/device.rs

@@ -0,0 +1,174 @@
+use async_trait::async_trait;
+use automation_cast::Cast;
+use serde::Serialize;
+
+use crate::errors::{DeviceError, ErrorCode};
+use crate::request::execute::CommandType;
+use crate::response;
+use crate::traits::{FanSpeed, HumiditySetting, OnOff, Scene, Trait};
+use crate::types::Type;
+
+#[async_trait]
+pub trait GoogleHomeDevice:
+    Sync + Send + Cast<dyn OnOff> + Cast<dyn Scene> + Cast<dyn FanSpeed> + Cast<dyn HumiditySetting>
+{
+    fn get_device_type(&self) -> Type;
+    fn get_device_name(&self) -> Name;
+    fn get_id(&self) -> String;
+    fn is_online(&self) -> bool;
+
+    // Default values that can optionally be overridden
+    fn will_report_state(&self) -> bool {
+        false
+    }
+    fn get_room_hint(&self) -> Option<&str> {
+        None
+    }
+    fn get_device_info(&self) -> Option<Info> {
+        None
+    }
+
+    async fn sync(&self) -> response::sync::Device {
+        let name = self.get_device_name();
+        let mut device =
+            response::sync::Device::new(&self.get_id(), &name.name, self.get_device_type());
+
+        device.name = name;
+        device.will_report_state = self.will_report_state();
+        // notification_supported_by_agent
+        if let Some(room) = self.get_room_hint() {
+            device.room_hint = Some(room.into());
+        }
+        device.device_info = self.get_device_info();
+
+        let mut traits = Vec::new();
+
+        // OnOff
+        if let Some(on_off) = self.cast() as Option<&dyn OnOff> {
+            traits.push(Trait::OnOff);
+            device.attributes.command_only_on_off = on_off.is_command_only();
+            device.attributes.query_only_on_off = on_off.is_query_only();
+        }
+
+        // Scene
+        if let Some(scene) = self.cast() as Option<&dyn Scene> {
+            traits.push(Trait::Scene);
+            device.attributes.scene_reversible = scene.is_scene_reversible();
+        }
+
+        // FanSpeed
+        if let Some(fan_speed) = self.cast() as Option<&dyn FanSpeed> {
+            traits.push(Trait::FanSpeed);
+            device.attributes.command_only_fan_speed = fan_speed.command_only_fan_speed();
+            device.attributes.available_fan_speeds = Some(fan_speed.available_speeds());
+        }
+
+        if let Some(humidity_setting) = self.cast() as Option<&dyn HumiditySetting> {
+            traits.push(Trait::HumiditySetting);
+            device.attributes.query_only_humidity_setting =
+                humidity_setting.query_only_humidity_setting();
+        }
+
+        device.traits = traits;
+
+        device
+    }
+
+    async fn query(&self) -> response::query::Device {
+        let mut device = response::query::Device::new();
+        if !self.is_online() {
+            device.set_offline();
+        }
+
+        // OnOff
+        if let Some(on_off) = self.cast() as Option<&dyn OnOff> {
+            device.state.on = on_off
+                .is_on()
+                .await
+                .map_err(|err| device.set_error(err))
+                .ok();
+        }
+
+        // FanSpeed
+        if let Some(fan_speed) = self.cast() as Option<&dyn FanSpeed> {
+            device.state.current_fan_speed_setting = Some(fan_speed.current_speed().await);
+        }
+
+        if let Some(humidity_setting) = self.cast() as Option<&dyn HumiditySetting> {
+            device.state.humidity_ambient_percent =
+                Some(humidity_setting.humidity_ambient_percent().await);
+        }
+
+        device
+    }
+
+    async fn execute(&mut self, command: &CommandType) -> Result<(), ErrorCode> {
+        match command {
+            CommandType::OnOff { on } => {
+                if let Some(t) = self.cast_mut() as Option<&mut dyn OnOff> {
+                    t.set_on(*on).await?;
+                } else {
+                    return Err(DeviceError::ActionNotAvailable.into());
+                }
+            }
+            CommandType::ActivateScene { deactivate } => {
+                if let Some(t) = self.cast_mut() as Option<&mut dyn Scene> {
+                    t.set_active(!deactivate).await?;
+                } else {
+                    return Err(DeviceError::ActionNotAvailable.into());
+                }
+            }
+            CommandType::SetFanSpeed { fan_speed } => {
+                if let Some(t) = self.cast_mut() as Option<&mut dyn FanSpeed> {
+                    t.set_speed(fan_speed).await?;
+                }
+            }
+        }
+
+        Ok(())
+    }
+}
+
+#[derive(Debug, Serialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Name {
+    #[serde(skip_serializing_if = "Vec::is_empty")]
+    default_names: Vec<String>,
+    name: String,
+    #[serde(skip_serializing_if = "Vec::is_empty")]
+    nicknames: Vec<String>,
+}
+
+impl Name {
+    pub fn new(name: &str) -> Self {
+        Self {
+            default_names: Vec::new(),
+            name: name.into(),
+            nicknames: Vec::new(),
+        }
+    }
+
+    pub fn add_default_name(&mut self, name: &str) {
+        self.default_names.push(name.into());
+    }
+
+    pub fn add_nickname(&mut self, name: &str) {
+        self.nicknames.push(name.into());
+    }
+}
+
+#[derive(Debug, Default, Serialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Info {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub manufacturer: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub model: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub hw_version: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub sw_version: Option<String>,
+    // attributes
+    // customData
+    // otherDeviceIds
+}

google-home/src/fulfillment.rs

@@ -2,14 +2,14 @@ use std::collections::HashMap;
 use std::sync::Arc;
 
 use automation_cast::Cast;
-use futures::future::{OptionFuture, join_all};
+use futures::future::{join_all, OptionFuture};
 use thiserror::Error;
-use tokio::sync::Mutex;
+use tokio::sync::{Mutex, RwLock};
 
-use crate::Device;
 use crate::errors::{DeviceError, ErrorCode};
 use crate::request::{self, Intent, Request};
-use crate::response::{self, Response, ResponsePayload, execute, query, sync};
+use crate::response::{self, execute, query, sync, Response, ResponsePayload, State};
+use crate::GoogleHomeDevice;
 
 #[derive(Debug)]
 pub struct GoogleHome {
@@ -30,23 +30,25 @@ impl GoogleHome {
         }
     }
 
-    pub async fn handle_request<T: Cast<dyn Device> + ?Sized + 'static>(
+    pub async fn handle_request<T: Cast<dyn GoogleHomeDevice> + ?Sized + 'static>(
         &self,
         request: Request,
-        devices: &HashMap<String, Box<T>>,
+        devices: &HashMap<String, Arc<RwLock<Box<T>>>>,
     ) -> Result<Response, FulfillmentError> {
         // TODO: What do we do if we actually get more then one thing in the input array, right now
         // we only respond to the first thing
         let intent = request.inputs.into_iter().next();
 
         let payload: OptionFuture<_> = intent
-            .map(async |intent| match intent {
+            .map(|intent| async move {
-                Intent::Sync => ResponsePayload::Sync(self.sync(devices).await),
+                match intent {
-                Intent::Query(payload) => {
+                    Intent::Sync => ResponsePayload::Sync(self.sync(devices).await),
-                    ResponsePayload::Query(self.query(payload, devices).await)
+                    Intent::Query(payload) => {
-                }
+                        ResponsePayload::Query(self.query(payload, devices).await)
-                Intent::Execute(payload) => {
+                    }
-                    ResponsePayload::Execute(self.execute(payload, devices).await)
+                    Intent::Execute(payload) => {
+                        ResponsePayload::Execute(self.execute(payload, devices).await)
+                    }
                 }
             })
             .into();
@@ -57,14 +59,14 @@ impl GoogleHome {
             .map(|payload| Response::new(&request.request_id, payload))
     }
 
-    async fn sync<T: Cast<dyn Device> + ?Sized + 'static>(
+    async fn sync<T: Cast<dyn GoogleHomeDevice> + ?Sized + 'static>(
         &self,
-        devices: &HashMap<String, Box<T>>,
+        devices: &HashMap<String, Arc<RwLock<Box<T>>>>,
     ) -> sync::Payload {
         let mut resp_payload = sync::Payload::new(&self.user_id);
-        let f = devices.values().map(async |device| {
+        let f = devices.iter().map(|(_, device)| async move {
-            if let Some(device) = device.as_ref().cast() {
+            if let Some(device) = device.read().await.as_ref().cast() {
-                Some(Device::sync(device).await)
+                Some(device.sync().await)
             } else {
                 None
             }
@@ -74,22 +76,22 @@ impl GoogleHome {
         resp_payload
     }
 
-    async fn query<T: Cast<dyn Device> + ?Sized + 'static>(
+    async fn query<T: Cast<dyn GoogleHomeDevice> + ?Sized + 'static>(
         &self,
         payload: request::query::Payload,
-        devices: &HashMap<String, Box<T>>,
+        devices: &HashMap<String, Arc<RwLock<Box<T>>>>,
     ) -> query::Payload {
         let mut resp_payload = query::Payload::new();
         let f = payload
             .devices
             .into_iter()
             .map(|device| device.id)
-            .map(async |id| {
+            .map(|id| async move {
                 // NOTE: Requires let_chains feature
                 let device = if let Some(device) = devices.get(id.as_str())
-                    && let Some(device) = device.as_ref().cast()
+                    && let Some(device) = device.read().await.as_ref().cast()
                 {
-                    Device::query(device).await
+                    device.query().await
                 } else {
                     let mut device = query::Device::new();
                     device.set_offline();
@@ -106,84 +108,91 @@ impl GoogleHome {
         resp_payload
     }
 
-    async fn execute<T: Cast<dyn Device> + ?Sized + 'static>(
+    async fn execute<T: Cast<dyn GoogleHomeDevice> + ?Sized + 'static>(
         &self,
         payload: request::execute::Payload,
-        devices: &HashMap<String, Box<T>>,
+        devices: &HashMap<String, Arc<RwLock<Box<T>>>>,
     ) -> execute::Payload {
         let resp_payload = Arc::new(Mutex::new(response::execute::Payload::new()));
 
-        let f = payload.commands.into_iter().map(async |command| {
+        let f = payload.commands.into_iter().map(|command| {
-            let mut success = response::execute::Command::new(execute::Status::Success);
+            let resp_payload = resp_payload.clone();
-            success.states = Some(execute::States {
+            async move {
-                online: true,
+                let mut success = response::execute::Command::new(execute::Status::Success);
-                state: Default::default(),
+                success.states = Some(execute::States {
-            });
+                    online: true,
-            let mut offline = response::execute::Command::new(execute::Status::Offline);
+                    state: State::default(),
-            offline.states = Some(execute::States {
+                });
-                online: false,
+                let mut offline = response::execute::Command::new(execute::Status::Offline);
-                state: Default::default(),
+                offline.states = Some(execute::States {
-            });
+                    online: false,
-            let mut errors: HashMap<ErrorCode, response::execute::Command> = HashMap::new();
+                    state: State::default(),
+                });
+                let mut errors: HashMap<ErrorCode, response::execute::Command> = HashMap::new();
 
-            let f = command
+                let f = command
-                .devices
+                    .devices
-                .into_iter()
+                    .into_iter()
-                .map(|device| device.id)
+                    .map(|device| device.id)
-                .map(async |id| {
+                    .map(|id| {
-                    if let Some(device) = devices.get(id.as_str())
+                        let execution = command.execution.clone();
-                        && let Some(device) = device.as_ref().cast()
+                        async move {
-                    {
+                            if let Some(device) = devices.get(id.as_str())
-                        if !device.is_online().await {
+                                && let Some(device) = device.write().await.as_mut().cast_mut()
-                            return (id, Ok(false));
+                            {
+                                if !device.is_online() {
+                                    return (id, Ok(false));
+                                }
+
+                                // NOTE: We can not use .map here because async =(
+                                let mut results = Vec::new();
+                                for cmd in &execution {
+                                    results.push(device.execute(cmd).await);
+                                }
+
+                                // Convert vec of results to a result with a vec and the first
+                                // encountered error
+                                let results =
+                                    results.into_iter().collect::<Result<Vec<_>, ErrorCode>>();
+
+                                // TODO: We only get one error not all errors
+                                if let Err(err) = results {
+                                    (id, Err(err))
+                                } else {
+                                    (id, Ok(true))
+                                }
+                            } else {
+                                (id.clone(), Err(DeviceError::DeviceNotFound.into()))
+                            }
                         }
+                    });
 
-                        // NOTE: We can not use .map here because async =(
+                let a = join_all(f).await;
-                        let mut results = Vec::new();
+                a.into_iter().for_each(|(id, state)| {
-                        for cmd in &command.execution {
+                    match state {
-                            results.push(Device::execute(device, cmd.clone()).await);
+                        Ok(true) => success.add_id(&id),
-                        }
+                        Ok(false) => offline.add_id(&id),
-
+                        Err(err) => errors
-                        // Convert vec of results to a result with a vec and the first
+                            .entry(err)
-                        // encountered error
+                            .or_insert_with(|| match &err {
-                        let results = results.into_iter().collect::<Result<Vec<_>, ErrorCode>>();
+                                ErrorCode::DeviceError(_) => {
-
+                                    response::execute::Command::new(execute::Status::Error)
-                        // TODO: We only get one error not all errors
+                                }
-                        if let Err(err) = results {
+                                ErrorCode::DeviceException(_) => {
-                            (id, Err(err))
+                                    response::execute::Command::new(execute::Status::Exceptions)
-                        } else {
+                                }
-                            (id, Ok(true))
+                            })
-                        }
+                            .add_id(&id),
-                    } else {
+                    };
-                        (id.clone(), Err(DeviceError::DeviceNotFound.into()))
-                    }
                 });
 
-            let a = join_all(f).await;
+                let mut resp_payload = resp_payload.lock().await;
-            a.into_iter().for_each(|(id, state)| {
+                resp_payload.add_command(success);
-                match state {
+                resp_payload.add_command(offline);
-                    Ok(true) => success.add_id(&id),
+                for (error, mut cmd) in errors {
-                    Ok(false) => offline.add_id(&id),
+                    cmd.error_code = Some(error);
-                    Err(err) => errors
+                    resp_payload.add_command(cmd);
-                        .entry(err)
+                }
-                        .or_insert_with(|| match &err {
-                            ErrorCode::DeviceError(_) => {
-                                response::execute::Command::new(execute::Status::Error)
-                            }
-                            ErrorCode::DeviceException(_) => {
-                                response::execute::Command::new(execute::Status::Exceptions)
-                            }
-                        })
-                        .add_id(&id),
-                };
-            });
-
-            let mut resp_payload = resp_payload.lock().await;
-            resp_payload.add_command(success);
-            resp_payload.add_command(offline);
-            for (error, mut cmd) in errors {
-                cmd.error_code = Some(error);
-                resp_payload.add_command(cmd);
             }
         });
 

google-home/src/lib.rs

@@ -1,16 +1,18 @@
 #![allow(incomplete_features)]
 #![feature(specialization)]
+#![feature(let_chains)]
 pub mod device;
 mod fulfillment;
 
 mod request;
 mod response;
 
+mod attributes;
 pub mod errors;
 pub mod traits;
 pub mod types;
 
-pub use device::Device;
+pub use device::GoogleHomeDevice;
 pub use fulfillment::{FulfillmentError, GoogleHome};
 pub use request::Request;
 pub use response::Response;

google-home/src/request/execute.rs

@@ -0,0 +1,106 @@
+use serde::Deserialize;
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Payload {
+    pub commands: Vec<Command>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Command {
+    pub devices: Vec<Device>,
+    pub execution: Vec<CommandType>,
+}
+
+#[derive(Debug, Deserialize)]
+#[serde(rename_all = "camelCase")]
+pub struct Device {
+    pub id: String,
+    // customData
+}
+
+#[derive(Debug, Deserialize, Clone)]
+#[serde(tag = "command", content = "params")]
+pub enum CommandType {
+    #[serde(rename = "action.devices.commands.OnOff")]
+    OnOff { on: bool },
+    #[serde(rename = "action.devices.commands.ActivateScene")]
+    ActivateScene { deactivate: bool },
+    #[serde(rename = "action.devices.commands.SetFanSpeed")]
+    SetFanSpeed { fan_speed: String },
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::request::{Intent, Request};
+
+    #[test]
+    fn deserialize() {
+        let json = r#"{
+  "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf",
+  "inputs": [
+    {
+      "intent": "action.devices.EXECUTE",
+      "payload": {
+        "commands": [
+          {
+            "devices": [
+              {
+                "id": "123",
+                "customData": {
+                  "fooValue": 74,
+                  "barValue": true,
+                  "bazValue": "sheepdip"
+                }
+              },
+              {
+                "id": "456",
+                "customData": {
+                  "fooValue": 36,
+                  "barValue": false,
+                  "bazValue": "moarsheep"
+                }
+              }
+            ],
+            "execution": [
+              {
+                "command": "action.devices.commands.OnOff",
+                "params": {
+                  "on": true
+                }
+              }
+            ]
+          }
+        ]
+      }
+    }
+  ]
+}"#;
+
+        let req: Request = serde_json::from_str(json).unwrap();
+
+        println!("{:?}", req);
+
+        assert_eq!(
+            req.request_id,
+            "ff36a3cc-ec34-11e6-b1a0-64510650abcf".to_string()
+        );
+        assert_eq!(req.inputs.len(), 1);
+        match &req.inputs[0] {
+            Intent::Execute(payload) => {
+                assert_eq!(payload.commands.len(), 1);
+                assert_eq!(payload.commands[0].devices.len(), 2);
+                assert_eq!(payload.commands[0].devices[0].id, "123");
+                assert_eq!(payload.commands[0].devices[1].id, "456");
+                assert_eq!(payload.commands[0].execution.len(), 1);
+                match payload.commands[0].execution[0] {
+                    CommandType::OnOff { on } => assert!(on),
+                    _ => panic!("Expected OnOff"),
+                }
+            }
+            _ => panic!("Expected Execute intent"),
+        };
+    }
+}

google-home/src/request/query.rs

@@ -15,42 +15,40 @@ pub struct Device {
 
 #[cfg(test)]
 mod tests {
-    use serde_json::json;
-
     use crate::request::{Intent, Request};
 
     #[test]
     fn deserialize() {
-        let req = json!({
+        let json = r#"{
-          "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf",
+  "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf",
-          "inputs": [
+  "inputs": [
-            {
+    {
-              "intent": "action.devices.QUERY",
+      "intent": "action.devices.QUERY",
-              "payload": {
+      "payload": {
-                "devices": [
+        "devices": [
-                  {
+          {
-                    "id": "123",
+            "id": "123",
-                    "customData": {
+            "customData": {
-                      "fooValue": 74,
+              "fooValue": 74,
-                      "barValue": true,
+              "barValue": true,
-                      "bazValue": "foo"
+              "bazValue": "foo"
-                    }
-                  },
-                  {
-                    "id": "456",
-                    "customData": {
-                      "fooValue": 12,
-                      "barValue": false,
-                      "bazValue": "bar"
-                    }
-                  }
-                ]
-              }
             }
-          ]
+          },
-        });
+          {
+            "id": "456",
+            "customData": {
+              "fooValue": 12,
+              "barValue": false,
+              "bazValue": "bar"
+            }
+          }
+        ]
+      }
+    }
+  ]
+}"#;
 
-        let req: Request = serde_json::from_value(req).unwrap();
+        let req: Request = serde_json::from_str(json).unwrap();
 
         println!("{:?}", req);
 

google-home/src/request/sync.rs

@@ -1,21 +1,19 @@
 #[cfg(test)]
 mod tests {
-    use serde_json::json;
-
     use crate::request::{Intent, Request};
 
     #[test]
     fn deserialize() {
-        let req = json!({
+        let json = r#"{
-          "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf",
+      "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf",
-          "inputs": [
+      "inputs": [
-            {
+        {
-              "intent": "action.devices.SYNC"
+          "intent": "action.devices.SYNC"
-            }
+        }
-          ]
+      ]
-        });
+    }"#;
 
-        let req: Request = serde_json::from_value(req).unwrap();
+        let req: Request = serde_json::from_str(json).unwrap();
 
         println!("{:?}", req);
 

google-home/src/response.rs

@@ -27,3 +27,16 @@ pub enum ResponsePayload {
     Query(query::Payload),
     Execute(execute::Payload),
 }
+
+#[derive(Debug, Default, Serialize, Clone)]
+#[serde(rename_all = "camelCase")]
+pub struct State {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub on: Option<bool>,
+
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub current_fan_speed_setting: Option<String>,
+
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub humidity_ambient_percent: Option<isize>,
+}

google-home/src/response/execute.rs

@@ -1,6 +1,7 @@
 use serde::Serialize;
 
 use crate::errors::ErrorCode;
+use crate::response::State;
 
 #[derive(Debug, Serialize, Clone)]
 #[serde(rename_all = "camelCase")]
@@ -71,7 +72,7 @@ pub struct States {
     pub online: bool,
 
     #[serde(flatten)]
-    pub state: serde_json::Value,
+    pub state: State,
 }
 
 #[derive(Debug, Serialize, Clone)]
@@ -86,19 +87,19 @@ pub enum Status {
 
 #[cfg(test)]
 mod tests {
-    use serde_json::json;
-
     use super::*;
     use crate::errors::DeviceError;
-    use crate::response::{Response, ResponsePayload};
+    use crate::response::{Response, ResponsePayload, State};
 
     #[test]
     fn serialize() {
         let mut execute_resp = Payload::new();
 
-        let state = json!({
+        let state = State {
-            "on": true,
+            on: Some(true),
-        });
+            current_fan_speed_setting: None,
+            humidity_ambient_percent: None,
+        };
         let mut command = Command::new(Status::Success);
         command.states = Some(States {
             online: true,
@@ -117,28 +118,10 @@ mod tests {
             ResponsePayload::Execute(execute_resp),
         );
 
-        let resp = serde_json::to_value(resp).unwrap();
+        let json = serde_json::to_string(&resp).unwrap();
 
-        let resp_expected = json!({
+        println!("{}", json);
-            "payload": {
-                "commands": [
-                    {
-                        "states": {
-                            "on": true,
-                            "online": true
-                        },
-                        "ids": ["123"],
-                        "status": "SUCCESS"
-                    }, {
-                        "errorCode": "deviceNotFound",
-                        "ids": ["456"],
-                        "status":"ERROR"
-                    }
-                ]
-            },
-            "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf"
-        });
 
-        assert_eq!(resp, resp_expected);
+        // TODO: Add a known correct output to test against
     }
 }

google-home/src/response/query.rs

@@ -3,6 +3,7 @@ use std::collections::HashMap;
 use serde::Serialize;
 
 use crate::errors::ErrorCode;
+use crate::response::State;
 
 #[derive(Debug, Serialize)]
 #[serde(rename_all = "camelCase")]
@@ -52,7 +53,7 @@ pub struct Device {
     error_code: Option<ErrorCode>,
 
     #[serde(flatten)]
-    pub state: serde_json::Value,
+    pub state: State,
 }
 
 impl Device {
@@ -61,7 +62,7 @@ impl Device {
             online: true,
             status: Status::Success,
             error_code: None,
-            state: Default::default(),
+            state: State::default(),
         }
     }
 
@@ -87,8 +88,6 @@ impl Default for Device {
 
 #[cfg(test)]
 mod tests {
-    use serde_json::json;
-
     use super::*;
     use crate::response::{Response, ResponsePayload};
 
@@ -97,15 +96,11 @@ mod tests {
         let mut query_resp = Payload::new();
 
         let mut device = Device::new();
-        device.state = json!({
+        device.state.on = Some(true);
-            "on": true,
-        });
         query_resp.add_device("123", device);
 
         let mut device = Device::new();
-        device.state = json!({
+        device.state.on = Some(false);
-            "on": true,
-        });
         query_resp.add_device("456", device);
 
         let resp = Response::new(
@@ -113,26 +108,10 @@ mod tests {
             ResponsePayload::Query(query_resp),
         );
 
-        let resp = serde_json::to_value(resp).unwrap();
+        let json = serde_json::to_string(&resp).unwrap();
 
-        let resp_expected = json!({
+        println!("{}", json);
-            "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf",
-            "payload": {
-                "devices": {
-                    "123": {
-                        "online": true,
-                        "status": "SUCCESS",
-                        "on": true
-                    },
-                    "456": {
-                        "online": true,
-                        "status": "SUCCESS",
-                        "on":true
-                    }
-                }
-            }
-        });
 
-        assert_eq!(resp, resp_expected);
+        // TODO: Add a known correct output to test against
     }
 }

google-home/src/response/sync.rs

@@ -1,5 +1,6 @@
 use serde::Serialize;
 
+use crate::attributes::Attributes;
 use crate::device;
 use crate::errors::ErrorCode;
 use crate::traits::Trait;
@@ -46,8 +47,7 @@ pub struct Device {
     pub room_hint: Option<String>,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub device_info: Option<device::Info>,
-    #[serde(skip_serializing_if = "serde_json::Value::is_null")]
+    pub attributes: Attributes,
-    pub attributes: serde_json::Value,
 }
 
 impl Device {
@@ -61,15 +61,13 @@ impl Device {
             notification_supported_by_agent: None,
             room_hint: None,
             device_info: None,
-            attributes: Default::default(),
+            attributes: Attributes::default(),
         }
     }
 }
 
 #[cfg(test)]
 mod tests {
-    use serde_json::json;
-
     use super::*;
     use crate::response::{Response, ResponsePayload};
     use crate::traits::Trait;
@@ -99,35 +97,10 @@ mod tests {
             ResponsePayload::Sync(sync_resp),
         );
 
-        let resp = serde_json::to_value(resp).unwrap();
+        let json = serde_json::to_string(&resp).unwrap();
 
-        let resp_expected = json!({
+        println!("{}", json);
-            "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf",
-            "payload": {
-                "agentUserId": "1836.15267389",
-                "devices": [
-                    {
-                        "id": "123",
-                        "type": "action.devices.types.KETTLE",
-                        "traits": ["action.devices.traits.OnOff"],
-                        "name": {
-                            "defaultNames": ["My Outlet 1234"],
-                            "name": "Night light",
-                            "nicknames": ["wall plug"]
-                        },
-                        "willReportState": false,
-                        "roomHint": "kitchen",
-                        "deviceInfo": {
-                            "manufacturer": "lights-out-inc",
-                            "model": "hs1234",
-                            "hwVersion": "3.2",
-                            "swVersion": "11.4"
-                        }
-                    }
-                ]
-            }
-        });
 
-        assert_eq!(resp, resp_expected);
+        // assert_eq!(json, r#"{"requestId":"ff36a3cc-ec34-11e6-b1a0-64510650abcf","payload":{"agentUserId":"1836.15267389","devices":[{"id":"123","type":"action.devices.types.KETTLE","traits":["action.devices.traits.OnOff"],"name":{"defaultNames":["My Outlet 1234"],"name":"Night light","nicknames":["wall plug"]},"willReportState":false,"roomHint":"kitchen","deviceInfo":{"manufacturer":"lights-out-inc","model":"hs1234","hwVersion":"3.2","swVersion":"11.4"}}]}}"#)
     }
 }

google-home/src/traits.rs

@@ -0,0 +1,83 @@
+use async_trait::async_trait;
+use serde::Serialize;
+
+use crate::errors::ErrorCode;
+
+#[derive(Debug, Serialize)]
+pub enum Trait {
+    #[serde(rename = "action.devices.traits.OnOff")]
+    OnOff,
+    #[serde(rename = "action.devices.traits.Scene")]
+    Scene,
+    #[serde(rename = "action.devices.traits.FanSpeed")]
+    FanSpeed,
+    #[serde(rename = "action.devices.traits.HumiditySetting")]
+    HumiditySetting,
+}
+
+#[async_trait]
+pub trait OnOff: Sync + Send {
+    fn is_command_only(&self) -> Option<bool> {
+        None
+    }
+
+    fn is_query_only(&self) -> Option<bool> {
+        None
+    }
+
+    // TODO: Implement correct error so we can handle them properly
+    async fn is_on(&self) -> Result<bool, ErrorCode>;
+    async fn set_on(&mut self, on: bool) -> Result<(), ErrorCode>;
+}
+
+#[async_trait]
+pub trait Scene: Sync + Send {
+    fn is_scene_reversible(&self) -> Option<bool> {
+        None
+    }
+
+    async fn set_active(&self, activate: bool) -> Result<(), ErrorCode>;
+}
+
+#[derive(Debug, Serialize)]
+pub struct SpeedValues {
+    pub speed_synonym: Vec<String>,
+    pub lang: String,
+}
+
+#[derive(Debug, Serialize)]
+pub struct Speed {
+    pub speed_name: String,
+    pub speed_values: Vec<SpeedValues>,
+}
+
+#[derive(Debug, Serialize)]
+pub struct AvailableSpeeds {
+    pub speeds: Vec<Speed>,
+    pub ordered: bool,
+}
+
+#[async_trait]
+pub trait FanSpeed: Sync + Send {
+    fn reversible(&self) -> Option<bool> {
+        None
+    }
+
+    fn command_only_fan_speed(&self) -> Option<bool> {
+        None
+    }
+
+    fn available_speeds(&self) -> AvailableSpeeds;
+    async fn current_speed(&self) -> String;
+    async fn set_speed(&self, speed: &str) -> Result<(), ErrorCode>;
+}
+
+#[async_trait]
+pub trait HumiditySetting: Sync + Send {
+    // TODO: This implementation is not complete, I have only implemented what I need right now
+    fn query_only_humidity_setting(&self) -> Option<bool> {
+        None
+    }
+
+    async fn humidity_ambient_percent(&self) -> isize;
+}

google-home/src/types.rs

@@ -12,10 +12,4 @@ pub enum Type {
     Scene,
     #[serde(rename = "action.devices.types.AIRPURIFIER")]
     AirPurifier,
-    #[serde(rename = "action.devices.types.DOOR")]
-    Door,
-    #[serde(rename = "action.devices.types.WINDOW")]
-    Window,
-    #[serde(rename = "action.devices.types.DRAWER")]
-    Drawer,
 }

google_home/google_home/Cargo.toml

@@ -1,15 +0,0 @@
-[package]
-name = "google_home"
-version = "0.1.0"
-edition = "2024"
-
-[dependencies]
-async-trait = { workspace = true }
-automation_cast = { workspace = true }
-futures = { workspace = true }
-google_home_macro = { workspace = true }
-json_value_merge = { workspace = true }
-serde = { workspace = true }
-serde_json = { workspace = true }
-thiserror = { workspace = true }
-tokio = { workspace = true }

google_home/google_home/src/device.rs

@@ -1,120 +0,0 @@
-use async_trait::async_trait;
-use serde::Serialize;
-
-use crate::errors::ErrorCode;
-use crate::response;
-use crate::traits::{Command, DeviceFulfillment};
-use crate::types::Type;
-
-#[async_trait]
-pub trait Device: DeviceFulfillment {
-    fn get_device_type(&self) -> Type;
-    fn get_device_name(&self) -> Name;
-    fn get_id(&self) -> String;
-    async fn is_online(&self) -> bool;
-
-    // Default values that can optionally be overridden
-    fn will_report_state(&self) -> bool {
-        false
-    }
-    fn get_room_hint(&self) -> Option<&str> {
-        None
-    }
-    fn get_device_info(&self) -> Option<Info> {
-        None
-    }
-
-    async fn sync(&self) -> response::sync::Device {
-        let name = self.get_device_name();
-        let mut device =
-            response::sync::Device::new(&self.get_id(), &name.name, self.get_device_type());
-
-        device.name = name;
-        device.will_report_state = self.will_report_state();
-        // notification_supported_by_agent
-        if let Some(room) = self.get_room_hint() {
-            device.room_hint = Some(room.into());
-        }
-        device.device_info = self.get_device_info();
-
-        // TODO: Return the appropriate error
-        if let Ok((traits, attributes)) = DeviceFulfillment::sync(self).await {
-            device.traits = traits;
-            device.attributes = attributes;
-        }
-
-        device
-    }
-
-    async fn query(&self) -> response::query::Device {
-        let mut device = response::query::Device::new();
-        if !self.is_online().await {
-            device.set_offline();
-        }
-
-        // TODO: Return the appropriate error
-        if let Ok(state) = DeviceFulfillment::query(self).await {
-            device.state = state;
-        }
-
-        device
-    }
-
-    async fn execute(&self, command: Command) -> Result<(), ErrorCode> {
-        // TODO: Do something with the return value, or just get rut of the return value?
-        if DeviceFulfillment::execute(self, command.clone())
-            .await
-            .is_err()
-        {
-            return Err(ErrorCode::DeviceError(
-                crate::errors::DeviceError::TransientError,
-            ));
-        }
-
-        Ok(())
-    }
-}
-
-#[derive(Debug, Serialize)]
-#[serde(rename_all = "camelCase")]
-pub struct Name {
-    #[serde(skip_serializing_if = "Vec::is_empty")]
-    default_names: Vec<String>,
-    name: String,
-    #[serde(skip_serializing_if = "Vec::is_empty")]
-    nicknames: Vec<String>,
-}
-
-impl Name {
-    pub fn new(name: &str) -> Self {
-        Self {
-            default_names: Vec::new(),
-            name: name.into(),
-            nicknames: Vec::new(),
-        }
-    }
-
-    pub fn add_default_name(&mut self, name: &str) {
-        self.default_names.push(name.into());
-    }
-
-    pub fn add_nickname(&mut self, name: &str) {
-        self.nicknames.push(name.into());
-    }
-}
-
-#[derive(Debug, Default, Serialize)]
-#[serde(rename_all = "camelCase")]
-pub struct Info {
-    #[serde(skip_serializing_if = "Option::is_none")]
-    pub manufacturer: Option<String>,
-    #[serde(skip_serializing_if = "Option::is_none")]
-    pub model: Option<String>,
-    #[serde(skip_serializing_if = "Option::is_none")]
-    pub hw_version: Option<String>,
-    #[serde(skip_serializing_if = "Option::is_none")]
-    pub sw_version: Option<String>,
-    // attributes
-    // customData
-    // otherDeviceIds
-}

google_home/google_home/src/request/execute.rs

@@ -1,142 +0,0 @@
-use serde::Deserialize;
-
-use crate::traits;
-
-#[derive(Debug, Deserialize)]
-#[serde(rename_all = "camelCase")]
-pub struct Payload {
-    pub commands: Vec<Command>,
-}
-
-#[derive(Debug, Deserialize)]
-#[serde(rename_all = "camelCase")]
-pub struct Command {
-    pub devices: Vec<Device>,
-    pub execution: Vec<traits::Command>,
-}
-
-#[derive(Debug, Deserialize)]
-#[serde(rename_all = "camelCase")]
-pub struct Device {
-    pub id: String,
-    // customData
-}
-
-#[cfg(test)]
-mod tests {
-    use serde_json::json;
-
-    use super::*;
-    use crate::request::{Intent, Request};
-
-    #[test]
-    fn deserialize_set_fan_speed() {
-        let req = json!({
-          "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf",
-          "inputs": [
-            {
-              "intent": "action.devices.EXECUTE",
-              "payload": {
-                "commands": [
-                  {
-                    "devices": [],
-                    "execution": [
-                      {
-                        "command": "action.devices.commands.SetFanSpeed",
-                        "params": {
-                          "fanSpeed": "Test"
-                        }
-                      }
-                    ]
-                  }
-                ]
-              }
-            }
-          ]
-        });
-
-        let req: Request = serde_json::from_value(req).unwrap();
-
-        assert_eq!(req.inputs.len(), 1);
-        match &req.inputs[0] {
-            Intent::Execute(payload) => {
-                assert_eq!(payload.commands.len(), 1);
-                assert_eq!(payload.commands[0].devices.len(), 0);
-                assert_eq!(payload.commands[0].execution.len(), 1);
-                match &payload.commands[0].execution[0] {
-                    traits::Command::SetFanSpeed { fan_speed } => assert_eq!(fan_speed, "Test"),
-                    _ => panic!("Expected SetFanSpeed"),
-                }
-            }
-            _ => panic!("Expected Execute intent"),
-        };
-    }
-
-    #[test]
-    fn deserialize() {
-        let req = json!({
-          "requestId": "ff36a3cc-ec34-11e6-b1a0-64510650abcf",
-          "inputs": [
-            {
-              "intent": "action.devices.EXECUTE",
-              "payload": {
-                "commands": [
-                  {
-                    "devices": [
-                      {
-                        "id": "123",
-                        "customData": {
-                          "fooValue": 74,
-                          "barValue": true,
-                          "bazValue": "sheepdip"
-                        }
-                      },
-                      {
-                        "id": "456",
-                        "customData": {
-                          "fooValue": 36,
-                          "barValue": false,
-                          "bazValue": "moarsheep"
-                        }
-                      }
-                    ],
-                    "execution": [
-                      {
-                        "command": "action.devices.commands.OnOff",
-                        "params": {
-                          "on": true
-                        }
-                      }
-                    ]
-                  }
-                ]
-              }
-            }
-          ]
-        });
-
-        let req: Request = serde_json::from_value(req).unwrap();
-
-        println!("{:?}", req);
-
-        assert_eq!(
-            req.request_id,
-            "ff36a3cc-ec34-11e6-b1a0-64510650abcf".to_string()
-        );
-        assert_eq!(req.inputs.len(), 1);
-        match &req.inputs[0] {
-            Intent::Execute(payload) => {
-                assert_eq!(payload.commands.len(), 1);
-                assert_eq!(payload.commands[0].devices.len(), 2);
-                assert_eq!(payload.commands[0].devices[0].id, "123");
-                assert_eq!(payload.commands[0].devices[1].id, "456");
-                assert_eq!(payload.commands[0].execution.len(), 1);
-                match payload.commands[0].execution[0] {
-                    traits::Command::OnOff { on } => assert!(on),
-                    _ => panic!("Expected OnOff"),
-                }
-            }
-            _ => panic!("Expected Execute intent"),
-        };
-    }
-}

google_home/google_home/src/traits.rs

@@ -1,103 +0,0 @@
-#![allow(non_snake_case)]
-use automation_cast::Cast;
-use google_home_macro::traits;
-use serde::{Deserialize, Serialize};
-
-use crate::Device;
-use crate::errors::ErrorCode;
-
-traits! {
-    Device,
-    "action.devices.traits.OnOff" => trait OnOff {
-        command_only_on_off: Option<bool>,
-        query_only_on_off: Option<bool>,
-        async fn on(&self) -> Result<bool, ErrorCode>,
-        "action.devices.commands.OnOff" => async fn set_on(&self, on: bool) -> Result<(), ErrorCode>,
-    },
-    "action.devices.traits.OpenClose" => trait OpenClose {
-        discrete_only_open_close: Option<bool>,
-        command_only_open_close: Option<bool>,
-        query_only_open_close: Option<bool>,
-        async fn open_percent(&self) -> Result<u8, ErrorCode>,
-        "action.devices.commands.OpenClose" => async fn set_open_percent(&self, open_percent: u8) -> Result<(), ErrorCode>,
-    },
-    "action.devices.traits.Brightness" => trait Brightness {
-        command_only_brightness: Option<bool>,
-        async fn brightness(&self) -> Result<u8, ErrorCode>,
-        "action.devices.commands.BrightnessAbsolute" => async fn set_brightness(&self, brightness: u8) -> Result<(), ErrorCode>,
-    },
-    "action.devices.traits.ColorSetting" => trait ColorSetting {
-        color_temperature_range: ColorTemperatureRange,
-
-        async fn color(&self) -> Color,
-        "action.devices.commands.ColorAbsolute" => async fn set_color(&self, color: Color) -> Result<(), ErrorCode>,
-    },
-    "action.devices.traits.Scene" => trait Scene {
-        scene_reversible: Option<bool>,
-
-        "action.devices.commands.ActivateScene" => async fn set_active(&self, deactivate: bool) -> Result<(), ErrorCode>,
-    },
-    "action.devices.traits.FanSpeed" => trait FanSpeed {
-        reversible: Option<bool>,
-        command_only_fan_speed: Option<bool>,
-        available_fan_speeds: AvailableSpeeds,
-
-        async fn current_fan_speed_setting(&self) -> Result<String, ErrorCode>,
-
-        // TODO: Figure out some syntax for optional command?
-        // Probably better to just force the user to always implement commands?
-        "action.devices.commands.SetFanSpeed" => async fn set_fan_speed(&self, fan_speed: String) -> Result<(), ErrorCode>,
-    },
-    "action.devices.traits.HumiditySetting" => trait HumiditySetting {
-        query_only_humidity_setting: Option<bool>,
-
-        async fn humidity_ambient_percent(&self) -> Result<isize, ErrorCode>,
-    },
-    "action.devices.traits.TemperatureControl" => trait TemperatureControl {
-        query_only_temperature_control: Option<bool>,
-        // TODO: Add rename
-        temperatureUnitForUX: TemperatureUnit,
-
-        async fn temperature_ambient_celsius(&self) -> Result<f32, ErrorCode>,
-    }
-}
-
-#[derive(Debug, Serialize)]
-#[serde(rename_all = "camelCase")]
-pub struct ColorTemperatureRange {
-    pub temperature_min_k: u32,
-    pub temperature_max_k: u32,
-}
-
-#[derive(Debug, Copy, Clone, Serialize, Deserialize)]
-#[serde(rename_all = "camelCase")]
-pub struct Color {
-    #[serde(rename(serialize = "temperatureK"))]
-    pub temperature: u32,
-}
-
-#[derive(Debug, Serialize)]
-pub struct SpeedValue {
-    pub speed_synonym: Vec<String>,
-    pub lang: String,
-}
-
-#[derive(Debug, Serialize)]
-pub struct Speed {
-    pub speed_name: String,
-    pub speed_values: Vec<SpeedValue>,
-}
-
-#[derive(Debug, Serialize)]
-pub struct AvailableSpeeds {
-    pub speeds: Vec<Speed>,
-    pub ordered: bool,
-}
-
-#[derive(Debug, Serialize)]
-pub enum TemperatureUnit {
-    #[serde(rename = "C")]
-    Celsius,
-    #[serde(rename = "F")]
-    Fahrenheit,
-}

google_home/google_home_macro/Cargo.toml

@@ -1,12 +0,0 @@
-[package]
-name = "google_home_macro"
-version = "0.1.0"
-edition = "2024"
-
-[lib]
-proc-macro = true
-
-[dependencies]
-proc-macro2 = { workspace = true }
-quote = { workspace = true }
-syn = { workspace = true }

google_home/google_home_macro/src/lib.rs

@@ -1,568 +0,0 @@
-#![feature(iter_intersperse)]
-use proc_macro::TokenStream;
-use quote::quote;
-use syn::parse::Parse;
-use syn::punctuated::Punctuated;
-use syn::token::Brace;
-use syn::{
-    GenericArgument, Ident, LitStr, Path, PathArguments, PathSegment, ReturnType, Signature, Token,
-    Type, TypePath, braced, parse_macro_input,
-};
-
-mod kw {
-    use syn::custom_keyword;
-
-    custom_keyword!(required);
-}
-
-#[derive(Debug)]
-struct FieldAttribute {
-    ident: Ident,
-    _colon_token: Token![:],
-    ty: Type,
-}
-
-impl Parse for FieldAttribute {
-    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
-        Ok(Self {
-            ident: input.parse()?,
-            _colon_token: input.parse()?,
-            ty: input.parse()?,
-        })
-    }
-}
-
-#[derive(Debug)]
-struct FieldState {
-    sign: Signature,
-}
-
-impl Parse for FieldState {
-    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
-        Ok(Self {
-            sign: input.parse()?,
-        })
-    }
-}
-
-#[derive(Debug)]
-struct FieldExecute {
-    name: LitStr,
-    _fat_arrow_token: Token![=>],
-    sign: Signature,
-}
-
-impl Parse for FieldExecute {
-    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
-        Ok(Self {
-            name: input.parse()?,
-            _fat_arrow_token: input.parse()?,
-            sign: input.parse()?,
-        })
-    }
-}
-
-#[derive(Debug)]
-enum Field {
-    Attribute(FieldAttribute),
-    State(FieldState),
-    Execute(FieldExecute),
-}
-
-impl Parse for Field {
-    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
-        if input.peek(Ident) {
-            Ok(Field::Attribute(input.parse()?))
-        } else if input.peek(LitStr) {
-            Ok(Field::Execute(input.parse()?))
-        } else {
-            Ok(Field::State(input.parse()?))
-        }
-    }
-}
-
-#[derive(Debug)]
-struct Trait {
-    name: LitStr,
-    _fat_arrow_token: Token![=>],
-    _trait_token: Token![trait],
-    ident: Ident,
-    _brace_token: Brace,
-    fields: Punctuated<Field, Token![,]>,
-}
-
-impl Parse for Trait {
-    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
-        let content;
-        Ok(Self {
-            name: input.parse()?,
-            _fat_arrow_token: input.parse()?,
-            _trait_token: input.parse()?,
-            ident: input.parse()?,
-            _brace_token: braced!(content in input),
-            fields: content.parse_terminated(Field::parse, Token![,])?,
-        })
-    }
-}
-
-#[derive(Debug)]
-struct Input {
-    ty: TypePath,
-    _comma: Token![,],
-    traits: Punctuated<Trait, Token![,]>,
-}
-
-// TODO: Error on duplicate name?
-impl Parse for Input {
-    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
-        Ok(Self {
-            ty: input.parse()?,
-            _comma: input.parse()?,
-            traits: input.parse_terminated(Trait::parse, Token![,])?,
-        })
-    }
-}
-
-fn extract_type_path(ty: &syn::Type) -> Option<&Path> {
-    match *ty {
-        Type::Path(ref typepath) if typepath.qself.is_none() => Some(&typepath.path),
-        _ => None,
-    }
-}
-
-fn extract_segment<'a>(path: &'a Path, options: &[&str]) -> Option<&'a PathSegment> {
-    let idents_of_path = path
-        .segments
-        .iter()
-        .map(|segment| segment.ident.to_string())
-        .intersperse('|'.into())
-        .collect::<String>();
-
-    options
-        .iter()
-        .find(|s| &idents_of_path == *s)
-        .and_then(|_| path.segments.last())
-}
-
-// Based on: https://stackoverflow.com/a/56264023
-fn extract_type_from_option(ty: &syn::Type) -> Option<&syn::Type> {
-    extract_type_path(ty)
-        .and_then(|path| {
-            extract_segment(path, &["Option", "std|option|Option", "core|option|Option"])
-        })
-        .and_then(|path_seg| {
-            let type_params = &path_seg.arguments;
-            // It should have only on angle-bracketed param ("<String>"):
-            match *type_params {
-                PathArguments::AngleBracketed(ref params) => params.args.first(),
-                _ => None,
-            }
-        })
-        .and_then(|generic_arg| match *generic_arg {
-            GenericArgument::Type(ref ty) => Some(ty),
-            _ => None,
-        })
-}
-
-fn extract_type_from_result(ty: &syn::Type) -> Option<&syn::Type> {
-    extract_type_path(ty)
-        .and_then(|path| {
-            extract_segment(path, &["Result", "std|result|Result", "core|result|Result"])
-        })
-        .and_then(|path_seg| {
-            let type_params = &path_seg.arguments;
-            // It should have only on angle-bracketed param ("<String>"):
-            match *type_params {
-                PathArguments::AngleBracketed(ref params) => params.args.first(),
-                _ => None,
-            }
-        })
-        .and_then(|generic_arg| match *generic_arg {
-            GenericArgument::Type(ref ty) => Some(ty),
-            _ => None,
-        })
-}
-
-fn get_attributes_struct_ident(t: &Trait) -> Ident {
-    syn::Ident::new(&format!("{}Attributes", t.ident), t.ident.span())
-}
-
-fn get_attributes_struct(t: &Trait) -> proc_macro2::TokenStream {
-    let fields = t.fields.iter().filter_map(|f| match f {
-        Field::Attribute(attr) => {
-            let ident = &attr.ident;
-            let ty = &attr.ty;
-
-            // TODO: Extract into function
-            if let Some(ty) = extract_type_from_option(ty) {
-                Some(quote! {
-                    #[serde(skip_serializing_if = "core::option::Option::is_none")]
-                    #ident: ::core::option::Option<#ty>
-                })
-            } else {
-                Some(quote! {
-                    #ident: #ty
-                })
-            }
-        }
-        _ => None,
-    });
-
-    let name = get_attributes_struct_ident(t);
-    quote! {
-        #[derive(Debug, serde::Serialize)]
-        #[serde(rename_all = "camelCase")]
-        struct #name {
-            #(#fields,)*
-        }
-    }
-}
-
-fn get_state_struct_ident(t: &Trait) -> Ident {
-    syn::Ident::new(&format!("{}State", t.ident), t.ident.span())
-}
-
-fn get_state_struct(t: &Trait) -> proc_macro2::TokenStream {
-    let fields = t.fields.iter().filter_map(|f| match f {
-        Field::State(state) => {
-            let ident = &state.sign.ident;
-
-            let ReturnType::Type(_, ty) = &state.sign.output else {
-                return None;
-            };
-
-            let ty = extract_type_from_result(ty).unwrap_or(ty);
-
-            if let Some(ty) = extract_type_from_option(ty) {
-                Some(quote! {
-                    #[serde(skip_serializing_if = "core::option::Option::is_none")]
-                    #ident: ::core::option::Option<#ty>
-                })
-            } else {
-                Some(quote! {#ident: #ty})
-            }
-        }
-        _ => None,
-    });
-
-    let name = get_state_struct_ident(t);
-    quote! {
-        #[derive(Debug, serde::Serialize)]
-        #[serde(rename_all = "camelCase")]
-        struct #name {
-            #(#fields,)*
-        }
-    }
-}
-
-fn get_command_enum(traits: &Punctuated<Trait, Token![,]>) -> proc_macro2::TokenStream {
-    let items = traits.iter().flat_map(|t| {
-        t.fields.iter().filter_map(|f| match f {
-            Field::Execute(execute) => {
-                let name = execute.name.value();
-                let ident = Ident::new(
-                    name.split_at(name.rfind('.').map(|v| v + 1).unwrap_or(0)).1,
-                    execute.name.span(),
-                );
-
-                let parameters = execute.sign.inputs.iter().skip(1);
-
-                Some(quote! {
-                    #[serde(rename = #name, rename_all = "camelCase")]
-                    #ident {
-                        #(#parameters,)*
-                    }
-                })
-            }
-            _ => None,
-        })
-    });
-
-    quote! {
-        #[derive(Debug, Clone, serde::Deserialize)]
-        #[serde(tag = "command", content = "params", rename_all = "camelCase")]
-        pub enum Command {
-            #(#items,)*
-        }
-    }
-}
-
-fn get_trait_enum(traits: &Punctuated<Trait, Token![,]>) -> proc_macro2::TokenStream {
-    let items = traits.iter().map(|t| {
-        let name = &t.name;
-        let ident = &t.ident;
-        quote! {
-            #[serde(rename = #name)]
-            #ident
-        }
-    });
-
-    quote! {
-        #[derive(Debug, serde::Serialize)]
-        pub enum Trait {
-            #(#items,)*
-        }
-    }
-}
-
-fn get_trait(t: &Trait) -> proc_macro2::TokenStream {
-    let fields = t.fields.iter().map(|f| match f {
-        Field::Attribute(attr) => {
-            let name = &attr.ident;
-            let ty = &attr.ty;
-
-            // If the default type is marked as optional, respond None by default
-            if let Some(ty) = extract_type_from_option(ty) {
-                quote! {
-                    fn #name(&self) -> Option<#ty> {
-                        None
-                    }
-                }
-            } else {
-                quote! {
-                    fn #name(&self) -> #ty;
-                }
-            }
-        }
-        Field::State(state) => {
-            let sign = &state.sign;
-
-            let ReturnType::Type(_, ty) = &state.sign.output else {
-                todo!("Handle weird function return types");
-            };
-
-            let inner = extract_type_from_result(ty);
-            // If the default type is marked as optional, respond None by default
-            if extract_type_from_option(inner.unwrap_or(ty)).is_some() {
-                if inner.is_some() {
-                    quote! {
-                        #sign {
-                            Ok(None)
-                        }
-                    }
-                } else {
-                    quote! {
-                        #sign {
-                            None
-                        }
-                    }
-                }
-            } else {
-                quote! {
-                    #sign;
-                }
-            }
-        }
-        Field::Execute(execute) => {
-            let sign = &execute.sign;
-            quote! {
-                #sign;
-            }
-        }
-    });
-
-    let ident = &t.ident;
-
-    let attr_ident = get_attributes_struct_ident(t);
-    let attr = t.fields.iter().filter_map(|f| match f {
-        Field::Attribute(attr) => {
-            let name = &attr.ident;
-
-            Some(quote! {
-                #name: self.#name()
-            })
-        }
-        _ => None,
-    });
-
-    let state_ident = get_state_struct_ident(t);
-    let state = t.fields.iter().filter_map(|f| match f {
-        Field::State(state) => {
-            let ident = &state.sign.ident;
-            let f_ident = &state.sign.ident;
-
-            let asyncness = if state.sign.asyncness.is_some() {
-                quote! {.await}
-            } else {
-                quote! {}
-            };
-
-            let errors = if let ReturnType::Type(_, ty) = &state.sign.output
-                && extract_type_from_result(ty).is_some()
-            {
-                quote! {?}
-            } else {
-                quote! {}
-            };
-
-            Some(quote! {
-                #ident: self.#f_ident() #asyncness #errors,
-            })
-        }
-        _ => None,
-    });
-
-    quote! {
-        #[async_trait::async_trait]
-        pub trait #ident: Sync + Send {
-            #(#fields)*
-
-            fn get_attributes(&self) -> #attr_ident {
-                #attr_ident { #(#attr,)* }
-            }
-
-            async fn get_state(&self) -> Result<#state_ident, Box<dyn ::std::error::Error>> {
-                Ok(#state_ident { #(#state)* })
-            }
-        }
-    }
-}
-
-#[proc_macro]
-pub fn traits(item: TokenStream) -> TokenStream {
-    let input = parse_macro_input!(item as Input);
-    let traits = input.traits;
-
-    let structs = traits.iter().map(|t| {
-        let attr = get_attributes_struct(t);
-        let state = get_state_struct(t);
-        let tra = get_trait(t);
-
-        quote! {
-            #attr
-            #state
-            #tra
-        }
-    });
-
-    let command_enum = get_command_enum(&traits);
-    let trait_enum = get_trait_enum(&traits);
-
-    let sync = traits.iter().map(|t| {
-        let ident = &t.ident;
-
-        quote! {
-            if let Some(t) = self.cast() as Option<&dyn #ident> {
-                traits.push(Trait::#ident);
-                let value = serde_json::to_value(t.get_attributes())?;
-                json_value_merge::Merge::merge(&mut attrs, &value);
-            }
-        }
-    });
-
-    let query = traits.iter().map(|t| {
-        let ident = &t.ident;
-
-        quote! {
-            if let Some(t) = self.cast() as Option<&dyn #ident> {
-                let value = serde_json::to_value(t.get_state().await?)?;
-                json_value_merge::Merge::merge(&mut state, &value);
-            }
-        }
-    });
-
-    let execute = traits.iter().flat_map(|t| {
-        t.fields.iter().filter_map(|f| match f {
-            Field::Execute(execute) => {
-                let ident = &t.ident;
-                let name = execute.name.value();
-                let command_name = Ident::new(
-                    name.split_at(name.rfind('.').map(|v| v + 1).unwrap_or(0)).1,
-                    execute.name.span(),
-                );
-                let f_name = &&execute.sign.ident;
-                let parameters = execute
-                    .sign
-                    .inputs
-                    .iter()
-                    .filter_map(|p| {
-                        if let syn::FnArg::Typed(p) = p {
-                            Some(&p.pat)
-                        } else {
-                            None
-                        }
-                    })
-                    .collect::<Vec<_>>();
-
-                let asyncness = if execute.sign.asyncness.is_some() {
-                    quote! {.await}
-                } else {
-                    quote! {}
-                };
-
-                let errors = if let ReturnType::Type(_, ty) = &execute.sign.output
-                    && extract_type_from_result(ty).is_some()
-                {
-                    quote! {?}
-                } else {
-                    quote! {}
-                };
-
-                Some(quote! {
-                    Command::#command_name {#(#parameters,)*} => {
-                        if let Some(t) = self.cast() as Option<&dyn #ident> {
-                            t.#f_name(#(#parameters,)*) #asyncness #errors;
-                            serde_json::to_value(t.get_state().await?)?
-                        } else {
-                            todo!("Device does not support action, return proper error");
-                        }
-                    }
-                })
-            }
-            _ => None,
-        })
-    });
-
-    let ty = input.ty;
-
-    let fulfillment = Ident::new(
-        &format!("{}Fulfillment", ty.path.segments.last().unwrap().ident),
-        ty.path.segments.last().unwrap().ident.span(),
-    );
-
-    quote! {
-		// TODO: This is always the same, so should not be part of the macro, but instead something
-		// else
-        #[async_trait::async_trait]
-		pub trait #fulfillment: Sync + Send {
-			async fn sync(&self) -> Result<(Vec<Trait>, serde_json::Value), Box<dyn ::std::error::Error>>;
-			async fn query(&self) -> Result<serde_json::Value, Box<dyn ::std::error::Error>>;
-            async fn execute(&self, command: Command) -> Result<serde_json::Value, Box<dyn std::error::Error>>;
-		}
-
-		#(#structs)*
-
-		#command_enum
-		#trait_enum
-
-        #[async_trait::async_trait]
-		impl<D> #fulfillment for D where D: #ty
-		{
-			async fn sync(&self) -> Result<(Vec<Trait>, serde_json::Value), Box<dyn ::std::error::Error>> {
-				let mut traits = Vec::new();
-				let mut attrs = serde_json::Value::Null;
-
-				#(#sync)*
-
-				Ok((traits, attrs))
-			  }
-
-			async fn query(&self) -> Result<serde_json::Value, Box<dyn ::std::error::Error>> {
-				let mut state = serde_json::Value::Null;
-
-				#(#query)*
-
-				Ok(state)
-			}
-
-            async fn execute(&self, command: Command) -> Result<serde_json::Value, Box<dyn std::error::Error>> {
-                let value = match command {
-                    #(#execute)*
-                };
-
-            	Ok(value)
-            }
-		}
-    }
-    .into()
-}

rust-toolchain.toml

@@ -1,4 +1,4 @@
 [toolchain]
-channel = "nightly-2025-08-20"
+channel = "nightly-2023-11-15"
-components = ["rustfmt", "clippy", "rust-analyzer"]
+components = ["rustfmt", "clippy"]
 profile = "minimal"

src/auth.rs

@@ -0,0 +1,62 @@
+use axum::async_trait;
+use axum::extract::{FromRef, FromRequestParts};
+use axum::http::request::Parts;
+use axum::http::StatusCode;
+use serde::Deserialize;
+
+use crate::error::{ApiError, ApiErrorJson};
+
+#[derive(Debug, Deserialize)]
+pub struct User {
+    pub preferred_username: String,
+}
+
+#[async_trait]
+impl<S> FromRequestParts<S> for User
+where
+    String: FromRef<S>,
+    S: Send + Sync,
+{
+    type Rejection = ApiError;
+
+    async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
+        // Get the state
+        let openid_url = String::from_ref(state);
+
+        // Create a request to the auth server
+        // TODO: Do some discovery to find the correct url for this instead of assuming
+        let mut req = reqwest::Client::new().get(format!("{}/userinfo", openid_url));
+
+        // Add auth header to the request if it exists
+        if let Some(auth) = parts.headers.get(axum::http::header::AUTHORIZATION) {
+            req = req.header(reqwest::header::AUTHORIZATION, auth);
+        }
+
+        // Send the request
+        let res = req
+            .send()
+            .await
+            .map_err(|err| ApiError::new(StatusCode::INTERNAL_SERVER_ERROR, err.into()))?;
+
+        // If the request is success full the auth token is valid and we are given userinfo
+        let status = res.status();
+        if status.is_success() {
+            let user = res
+                .json()
+                .await
+                .map_err(|err| ApiError::new(StatusCode::INTERNAL_SERVER_ERROR, err.into()))?;
+
+            return Ok(user);
+        } else {
+            let err: ApiErrorJson = res
+                .json()
+                .await
+                .map_err(|err| ApiError::new(StatusCode::INTERNAL_SERVER_ERROR, err.into()))?;
+
+            let err = ApiError::try_from(err)
+                .map_err(|err| ApiError::new(StatusCode::INTERNAL_SERVER_ERROR, err.into()))?;
+
+            Err(err)
+        }
+    }
+}