Fix suggest lookup

Author: bbqsrc

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "divvun-runtime"
-version = "0.2.1"
+version = "0.2.2"
 edition = "2024"
 repository = "https://github.com/divvun/divvun-runtime"
 license = "MIT OR Apache-2.0"

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "divvun-runtime-macros"
-version = "0.2.1"
+version = "0.2.2"
 edition = "2021"
 
 [lib]

macros/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "divvun-rt-playground"
-version = "0.2.1"
+version = "0.2.2"
 description = "A Tauri App"
 authors = ["you"]
 edition = "2021"

playground/src-tauri/Cargo.toml

@@ -506,11 +506,19 @@ impl Pipe {
         })
     }
 
-    pub fn command<T: CommandRunner>(&self, key: &str) -> Option<&T> {
+    pub fn command<T: CommandRunner>(&self, key: Option<&str>) -> Option<&T> {
+        if let Some(key) = key {
+            return self
+                .modules
+                .get(key)
+                .map(|x| &**x as &(dyn Any + Send + Sync))
+                .and_then(|x| x.downcast_ref::<T>());
+        }
+
         self.modules
-            .get(key)
+            .values()
             .map(|x| &**x as &(dyn Any + Send + Sync))
-            .and_then(|x| x.downcast_ref::<T>())
+            .find_map(|x| x.downcast_ref::<T>())
     }
 
     pub async fn create_stream(

src/ast/mod.rs

@@ -214,7 +214,7 @@ impl Bundle {
         &self.pipe.defn
     }
 
-    pub fn command<T: modules::CommandRunner>(&self, key: &str) -> Option<&T> {
+    pub fn command<T: modules::CommandRunner>(&self, key: Option<&str>) -> Option<&T> {
         self.pipe.command(key)
     }