refactor: improve CLI perf with lazy loading

Signed-off-by: Jiaxiao Zhou <duibao55328@gmail.com>

Author: Mossaka

.claude/agents/rust-pro.md

@@ -0,0 +1,36 @@
+---
+name: rust-pro
+description: Use this agent to write rust code
+model: sonnet
+color: orange
+---
+
+You are a Rust expert specializing in safe, performant systems programming.
+
+## Focus Areas
+
+- Ownership, borrowing, and lifetime annotations
+- Trait design and generic programming
+- Async/await with Tokio/async-std
+- Safe concurrency with Arc, Mutex, channels
+- Error handling with Result and custom errors
+- FFI and unsafe code when necessary
+
+## Approach
+
+1. Leverage the type system for correctness
+2. Zero-cost abstractions over runtime checks
+3. Explicit error handling - no panics in libraries
+4. Use iterators over manual loops
+5. Minimize unsafe blocks with clear invariants
+
+## Output
+
+- Idiomatic Rust with proper error handling
+- Trait implementations with derive macros
+- Async code with proper cancellation
+- Unit tests and documentation tests
+- Benchmarks with criterion.rs
+- Cargo.toml with feature flags
+
+Follow clippy lints. Include examples in doc comments.

.claude/settings.local.json

@@ -0,0 +1,20 @@
+{
+  "permissions": {
+    "allow": [
+      "Bash(git fetch:*)",
+      "Bash(git rebase:*)",
+      "Bash(git add:*)",
+      "Bash(cargo check:*)",
+      "Bash(cargo build:*)",
+      "mcp__wassette__list-components",
+      "Bash(cargo test:*)",
+      "Bash(just build)",
+      "Bash(just test:*)",
+      "Bash(find:*)",
+      "Bash(cargo doc:*)"
+    ],
+    "deny": [],
+    "ask": [],
+    "defaultMode": "acceptEdits"
+  }
+}

config.yaml

@@ -0,0 +1,21 @@
+binds:
+- port: 3000
+  listeners:
+  - routes:
+    - policies:
+        cors:
+          allowOrigins:
+            - "*"
+          allowHeaders:
+            - mcp-protocol-version
+            - content-type
+            - cache-control
+      backends:
+      - mcp:
+          targets:
+          - name: wassette
+            stdio:
+              cmd: wassette
+              args: ["serve", "--stdio",]
+              env:
+                RUST_LOG: "off"

crates/mcp-server/src/components.rs

@@ -15,7 +15,8 @@ use wassette::LifecycleManager;
 #[instrument(skip(lifecycle_manager))]
 pub(crate) async fn get_component_tools(lifecycle_manager: &LifecycleManager) -> Result<Vec<Tool>> {
     debug!("Listing components");
-    let component_ids = lifecycle_manager.list_components().await;
+    // Use known components (loaded or present on disk) for fast listing
+    let component_ids = lifecycle_manager.list_components_known().await;
 
     info!(count = component_ids.len(), "Found components");
     let mut tools = Vec::new();
@@ -139,7 +140,8 @@ pub async fn handle_list_components(
 ) -> Result<CallToolResult> {
     info!("Listing loaded components");
 
-    let component_ids = lifecycle_manager.list_components().await;
+    // Use known components (loaded or present on disk) for fast listing
+    let component_ids = lifecycle_manager.list_components_known().await;
 
     let components_info = stream::iter(component_ids)
         .map(|id| async move {

crates/mcp-server/src/tools.rs

@@ -463,14 +463,11 @@ pub async fn handle_get_policy(
 
     info!("Getting policy for component {}", component_id);
 
-    // First check if the component exists
-    let component_exists = lifecycle_manager
-        .get_component(component_id)
+    // Ensure the component is available (compile lazily if needed)
+    lifecycle_manager
+        .ensure_component_loaded(component_id)
         .await
-        .is_some();
-    if !component_exists {
-        return Err(anyhow::anyhow!("Component not found: {}", component_id));
-    }
+        .map_err(|e| anyhow::anyhow!("Component not found: {} ({})", component_id, e))?;
 
     let policy_info = lifecycle_manager.get_policy_info(component_id).await;
 
@@ -520,6 +517,12 @@ pub async fn handle_grant_storage_permission(
 
     info!("Granting storage permission to component {}", component_id);
 
+    // Ensure component is loaded (lazy compile)
+    lifecycle_manager
+        .ensure_component_loaded(component_id)
+        .await
+        .map_err(|e| anyhow::anyhow!("Component not found: {} ({})", component_id, e))?;
+
     let result = lifecycle_manager
         .grant_permission(component_id, "storage", details)
         .await;
@@ -570,6 +573,11 @@ pub async fn handle_grant_network_permission(
 
     info!("Granting network permission to component {}", component_id);
 
+    lifecycle_manager
+        .ensure_component_loaded(component_id)
+        .await
+        .map_err(|e| anyhow::anyhow!("Component not found: {} ({})", component_id, e))?;
+
     let result = lifecycle_manager
         .grant_permission(component_id, "network", details)
         .await;
@@ -623,6 +631,11 @@ pub async fn handle_grant_environment_variable_permission(
         component_id
     );
 
+    lifecycle_manager
+        .ensure_component_loaded(component_id)
+        .await
+        .map_err(|e| anyhow::anyhow!("Component not found: {} ({})", component_id, e))?;
+
     let result = lifecycle_manager
         .grant_permission(component_id, "environment", details)
         .await;
@@ -673,6 +686,11 @@ pub async fn handle_grant_memory_permission(
 
     info!("Granting memory permission to component {}", component_id);
 
+    lifecycle_manager
+        .ensure_component_loaded(component_id)
+        .await
+        .map_err(|e| anyhow::anyhow!("Component not found: {} ({})", component_id, e))?;
+
     let result = lifecycle_manager
         .grant_permission(component_id, "resource", details)
         .await;
@@ -731,6 +749,11 @@ pub async fn handle_revoke_storage_permission(
         uri, component_id
     );
 
+    lifecycle_manager
+        .ensure_component_loaded(component_id)
+        .await
+        .map_err(|e| anyhow::anyhow!("Component not found: {} ({})", component_id, e))?;
+
     let result = lifecycle_manager
         .revoke_storage_permission_by_uri(component_id, uri)
         .await;
@@ -784,6 +807,11 @@ pub async fn handle_revoke_network_permission(
         component_id
     );
 
+    lifecycle_manager
+        .ensure_component_loaded(component_id)
+        .await
+        .map_err(|e| anyhow::anyhow!("Component not found: {} ({})", component_id, e))?;
+
     let result = lifecycle_manager
         .revoke_permission(component_id, "network", details)
         .await;
@@ -837,6 +865,11 @@ pub async fn handle_revoke_environment_variable_permission(
         component_id
     );
 
+    lifecycle_manager
+        .ensure_component_loaded(component_id)
+        .await
+        .map_err(|e| anyhow::anyhow!("Component not found: {} ({})", component_id, e))?;
+
     let result = lifecycle_manager
         .revoke_permission(component_id, "environment", details)
         .await;
@@ -883,6 +916,11 @@ pub async fn handle_reset_permission(
 
     info!("Resetting all permissions for component {}", component_id);
 
+    lifecycle_manager
+        .ensure_component_loaded(component_id)
+        .await
+        .map_err(|e| anyhow::anyhow!("Component not found: {} ({})", component_id, e))?;
+
     let result = lifecycle_manager.reset_permission(component_id).await;
 
     match result {