From 7b2ed93798dd59ec1a887054bbc592f4931cb350 Mon Sep 17 00:00:00 2001
From: hosted-fornet <hosted-fornet@protonmail.com>
Date: Mon, 18 Aug 2025 17:18:41 -0700
Subject: [PATCH] build: add #[ws_client]`

---
 src/build/wit_generator.rs | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/build/wit_generator.rs b/src/build/wit_generator.rs
index 15ead1b4..9338e748 100644
--- a/src/build/wit_generator.rs
+++ b/src/build/wit_generator.rs
@@ -903,9 +903,10 @@ fn process_rust_project(project_path: &Path, api_dir: &Path) -> Result<Option<(S
             let has_http = method.attrs.iter().any(|a| a.path().is_ident("http"));
             let has_init = method.attrs.iter().any(|a| a.path().is_ident("init"));
             let has_ws = method.attrs.iter().any(|a| a.path().is_ident("ws"));
+            let has_ws_client = method.attrs.iter().any(|a| a.path().is_ident("ws_client"));
 
-            if has_remote || has_local || has_http || has_init || has_ws {
-                debug!(remote=%has_remote, local=%has_local, http=%has_http, init=%has_init, ws=%has_ws, "Method attributes found");
+            if has_remote || has_local || has_http || has_init || has_ws || has_ws_client {
+                debug!(remote=%has_remote, local=%has_local, http=%has_http, init=%has_init, ws=%has_ws, ws_client=%has_ws_client, "Method attributes found");
                 // Validate original Rust function name
                 validate_name(&method_name, "Function")?; // Error early if name invalid
                 let func_kebab_name = to_kebab_case(&method_name);
@@ -920,6 +921,11 @@ fn process_rust_project(project_path: &Path, api_dir: &Path) -> Result<Option<(S
                     continue;
                 }
 
+                if has_ws_client {
+                    debug!(method_name = %method_name, "Found [ws_client] function, skipping signature generation (websocket handlers are ignored by WIT generator)");
+                    continue;
+                }
+
                 // Generate signature structs. `generate_signature_struct` calls `rust_type_to_wit`,
                 // which populates `global_used_types` with all custom types found in parameters/return types.
                 if has_remote {
@@ -952,7 +958,7 @@ fn process_rust_project(project_path: &Path, api_dir: &Path) -> Result<Option<(S
             } else {
                 // Method in hyperprocess impl lacks required attribute - Error
                 return Err(eyre!(
-                         "Method '{}' in the #[hyperprocess] impl block is missing a required attribute ([remote], [local], [http], [init], or [ws]). Only methods with these attributes should be included.",
+                         "Method '{}' in the #[hyperprocess] impl block is missing a required attribute ([remote], [local], [http], [init], [ws] or [ws_client]). Only methods with these attributes should be included.",
                          method_name
                      ));
             }