feat(config)!: In config devices can now also be a (table of) function(s)

This function receives the mqtt client as an argument. In the future
this will be the only way to create devices that require the mqtt client.

src/config.rs

@@ -1,10 +1,12 @@
-use std::collections::HashMap;
+use std::collections::{HashMap, VecDeque};
 use std::net::{Ipv4Addr, SocketAddr};
 
 use automation_lib::action_callback::ActionCallback;
 use automation_lib::device::Device;
+use automation_lib::mqtt::WrappedAsyncClient;
 use automation_macro::LuaDeviceConfig;
 use lua_typed::Typed;
+use mlua::FromLua;
 use serde::Deserialize;
 
 #[derive(Debug, Deserialize)]
@@ -32,12 +34,78 @@ pub struct FulfillmentConfig {
     pub port: u16,
 }
 
+#[derive(Debug, Default)]
+pub struct Devices(mlua::Value);
+
+impl Devices {
+    pub async fn get(
+        self,
+        lua: &mlua::Lua,
+        client: &WrappedAsyncClient,
+    ) -> mlua::Result<Vec<Box<dyn Device>>> {
+        let mut devices = Vec::new();
+        let initial_table = match self.0 {
+            mlua::Value::Table(table) => table,
+            mlua::Value::Function(f) => f.call_async(client.clone()).await?,
+            _ => Err(mlua::Error::runtime(format!(
+                "Expected table or function, instead found: {}",
+                self.0.type_name()
+            )))?,
+        };
+
+        let mut queue: VecDeque<mlua::Table> = [initial_table].into();
+        loop {
+            let Some(table) = queue.pop_front() else {
+                break;
+            };
+
+            for pair in table.pairs() {
+                let (_, value): (mlua::Value, _) = pair?;
+
+                match value {
+                    mlua::Value::UserData(_) => devices.push(Box::from_lua(value, lua)?),
+                    mlua::Value::Function(f) => {
+                        queue.push_back(f.call_async(client.clone()).await?);
+                    }
+                    _ => Err(mlua::Error::runtime(format!(
+                        "Expected a device, table, or function, instead found: {}",
+                        value.type_name()
+                    )))?,
+                }
+            }
+        }
+
+        Ok(devices)
+    }
+}
+
+impl FromLua for Devices {
+    fn from_lua(value: mlua::Value, _lua: &mlua::Lua) -> mlua::Result<Self> {
+        Ok(Devices(value))
+    }
+}
+
+impl Typed for Devices {
+    fn type_name() -> String {
+        "Devices".into()
+    }
+
+    fn generate_header() -> Option<String> {
+        Some(format!(
+            "---@alias {} (DeviceInterface | fun(client: {}): Devices)[]\n",
+            <Self as Typed>::type_name(),
+            <WrappedAsyncClient as Typed>::type_name()
+        ))
+    }
+}
+
 #[derive(Debug, LuaDeviceConfig, Typed)]
 pub struct Config {
     pub fulfillment: FulfillmentConfig,
     #[device_config(from_lua, default)]
-    #[typed(default)]
-    pub devices: Vec<Box<dyn Device>>,
+    pub devices: Option<Devices>,
+    #[device_config(from_lua)]
+    pub mqtt: WrappedAsyncClient,
     #[device_config(from_lua, default)]
     #[typed(default)]
     pub schedule: HashMap<String, ActionCallback<()>>,