spikard 0.3.5 → 0.3.6

data/vendor/crates/spikard-rb/src/sse.rs

@@ -1,231 +1,231 @@
-//! Ruby SSE producer bindings
-//!
-//! This module provides the bridge between Ruby blocks/procs and Rust's SSE system.
-//! Uses magnus to safely call Ruby code from Rust async tasks.
-
-use magnus::{RHash, Value, prelude::*, value::Opaque};
-use serde_json::Value as JsonValue;
-use spikard_http::{SseEvent, SseEventProducer};
-use tracing::{debug, error};
-
-/// Ruby implementation of SseEventProducer
-pub struct RubySseEventProducer {
-    /// Producer name for debugging
-    name: String,
-    /// Ruby proc/callable for next_event (Opaque for Send safety)
-    next_event_proc: Opaque<Value>,
-    /// Ruby proc/callable for on_connect (Opaque for Send safety)
-    on_connect_proc: Option<Opaque<Value>>,
-    /// Ruby proc/callable for on_disconnect (Opaque for Send safety)
-    on_disconnect_proc: Option<Opaque<Value>>,
-}
-
-impl RubySseEventProducer {
-    /// Create a new Ruby SSE event producer
-    #[allow(dead_code)]
-    pub fn new(
-        name: String,
-        next_event_proc: Value,
-        on_connect_proc: Option<Value>,
-        on_disconnect_proc: Option<Value>,
-    ) -> Self {
-        Self {
-            name,
-            next_event_proc: next_event_proc.into(),
-            on_connect_proc: on_connect_proc.map(|v| v.into()),
-            on_disconnect_proc: on_disconnect_proc.map(|v| v.into()),
-        }
-    }
-
-    /// Convert Ruby value to JSON
-    fn ruby_to_json(ruby: &magnus::Ruby, value: Value) -> Result<JsonValue, String> {
-        if value.is_nil() {
-            return Ok(JsonValue::Null);
-        }
-
-        let json_module: Value = ruby
-            .class_object()
-            .const_get("JSON")
-            .map_err(|e| format!("JSON module not available: {}", e))?;
-
-        let json_str: String = json_module
-            .funcall("generate", (value,))
-            .map_err(|e| format!("Failed to generate JSON: {}", e))?;
-
-        serde_json::from_str(&json_str).map_err(|e| format!("Failed to parse JSON: {}", e))
-    }
-}
-
-impl SseEventProducer for RubySseEventProducer {
-    async fn next_event(&self) -> Option<SseEvent> {
-        debug!("Ruby SSE producer '{}': next_event", self.name);
-
-        match magnus::Ruby::get()
-            .map_err(|e| format!("Failed to get Ruby: {}", e))
-            .and_then(|ruby| {
-                debug!("Ruby SSE producer: acquired Ruby VM");
-
-                let proc_value = ruby.get_inner(self.next_event_proc);
-                let result: Value = proc_value
-                    .funcall("call", ())
-                    .map_err(|e| format!("Producer '{}' call failed: {}", self.name, e))?;
-
-                debug!("Ruby SSE producer: called next_event proc");
-
-                if result.is_nil() {
-                    debug!("Ruby SSE producer: received nil, ending stream");
-                    return Ok(None);
-                }
-
-                let result_hash = if let Some(hash) = RHash::from_value(result) {
-                    hash
-                } else {
-                    let hash_value: Value = result.funcall("to_h", ()).map_err(|e| {
-                        format!(
-                            "next_event must return a Hash/SseEvent convertible via to_h ({}): {}",
-                            unsafe { result.classname() },
-                            e
-                        )
-                    })?;
-                    RHash::from_value(hash_value).ok_or_else(|| {
-                        format!("next_event to_h must return a Hash, got {}", unsafe {
-                            hash_value.classname()
-                        })
-                    })?
-                };
-
-                let data_value = result_hash
-                    .get(ruby.to_symbol("data"))
-                    .ok_or_else(|| "next_event Hash must have :data key".to_string())?;
-
-                let data_json = Self::ruby_to_json(&ruby, data_value)?;
-
-                let event_type: Option<String> = result_hash
-                    .get(ruby.to_symbol("event_type"))
-                    .and_then(|v| if v.is_nil() { None } else { String::try_convert(v).ok() });
-
-                let id: Option<String> = result_hash
-                    .get(ruby.to_symbol("id"))
-                    .and_then(|v| if v.is_nil() { None } else { String::try_convert(v).ok() });
-
-                let retry: Option<u64> = result_hash
-                    .get(ruby.to_symbol("retry"))
-                    .and_then(|v| if v.is_nil() { None } else { u64::try_convert(v).ok() });
-
-                let mut event = if let Some(et) = event_type {
-                    SseEvent::with_type(et, data_json)
-                } else {
-                    SseEvent::new(data_json)
-                };
-
-                if let Some(id_str) = id {
-                    event = event.with_id(id_str);
-                }
-
-                if let Some(retry_ms) = retry {
-                    event = event.with_retry(retry_ms);
-                }
-
-                Ok(Some(event))
-            }) {
-            Ok(event) => event,
-            Err(e) => {
-                error!("Ruby error in next_event: {}", e);
-                None
-            }
-        }
-    }
-
-    async fn on_connect(&self) {
-        debug!("Ruby SSE producer '{}': on_connect", self.name);
-
-        if let Some(on_connect_proc) = self.on_connect_proc
-            && let Err(e) = magnus::Ruby::get()
-                .map_err(|e| format!("Failed to get Ruby: {}", e))
-                .and_then(|ruby| {
-                    debug!("Ruby SSE producer: on_connect acquired Ruby VM");
-                    let proc_value = ruby.get_inner(on_connect_proc);
-                    proc_value
-                        .funcall::<_, _, Value>("call", ())
-                        .map_err(|e| format!("on_connect '{}' call failed: {}", self.name, e))?;
-                    debug!("Ruby SSE producer: on_connect completed");
-                    Ok(())
-                })
-        {
-            error!("on_connect error: {}", e);
-        }
-    }
-
-    async fn on_disconnect(&self) {
-        debug!("Ruby SSE producer '{}': on_disconnect", self.name);
-
-        if let Some(on_disconnect_proc) = self.on_disconnect_proc
-            && let Err(e) = magnus::Ruby::get()
-                .map_err(|e| format!("Failed to get Ruby: {}", e))
-                .and_then(|ruby| {
-                    let proc_value = ruby.get_inner(on_disconnect_proc);
-                    proc_value
-                        .funcall::<_, _, Value>("call", ())
-                        .map_err(|e| format!("on_disconnect '{}' call failed: {}", self.name, e))?;
-                    debug!("Ruby SSE producer: on_disconnect completed");
-                    Ok(())
-                })
-        {
-            error!("on_disconnect error: {}", e);
-        }
-    }
-}
-
-unsafe impl Send for RubySseEventProducer {}
-unsafe impl Sync for RubySseEventProducer {}
-
-/// Create SseState from Ruby producer object
-///
-/// This function is designed to be called from Ruby to register SSE producers.
-#[allow(dead_code)]
-pub fn create_sse_state(
-    ruby: &magnus::Ruby,
-    producer_obj: Value,
-) -> Result<spikard_http::SseState<RubySseEventProducer>, magnus::Error> {
-    let next_event_proc: Value = producer_obj
-        .funcall("method", (ruby.to_symbol("next_event"),))
-        .map_err(|e| {
-            magnus::Error::new(
-                ruby.exception_arg_error(),
-                format!("next_event method not found: {}", e),
-            )
-        })?;
-
-    let on_connect_proc = producer_obj
-        .funcall::<_, _, Value>("method", (ruby.to_symbol("on_connect"),))
-        .ok();
-
-    let on_disconnect_proc = producer_obj
-        .funcall::<_, _, Value>("method", (ruby.to_symbol("on_disconnect"),))
-        .ok();
-
-    let event_schema = producer_obj
-        .funcall::<_, _, Value>("instance_variable_get", (ruby.to_symbol("@_event_schema"),))
-        .ok()
-        .and_then(|v| {
-            if v.is_nil() {
-                None
-            } else {
-                RubySseEventProducer::ruby_to_json(ruby, v).ok()
-            }
-        });
-
-    let ruby_producer = RubySseEventProducer::new(
-        "SseEventProducer".to_string(),
-        next_event_proc,
-        on_connect_proc,
-        on_disconnect_proc,
-    );
-
-    if event_schema.is_some() {
-        spikard_http::SseState::with_schema(ruby_producer, event_schema)
-            .map_err(|e| magnus::Error::new(ruby.exception_runtime_error(), e))
-    } else {
-        Ok(spikard_http::SseState::new(ruby_producer))
-    }
-}
+//! Ruby SSE producer bindings
+//!
+//! This module provides the bridge between Ruby blocks/procs and Rust's SSE system.
+//! Uses magnus to safely call Ruby code from Rust async tasks.
+
+use magnus::{RHash, Value, prelude::*, value::Opaque};
+use serde_json::Value as JsonValue;
+use spikard_http::{SseEvent, SseEventProducer};
+use tracing::{debug, error};
+
+/// Ruby implementation of SseEventProducer
+pub struct RubySseEventProducer {
+    /// Producer name for debugging
+    name: String,
+    /// Ruby proc/callable for next_event (Opaque for Send safety)
+    next_event_proc: Opaque<Value>,
+    /// Ruby proc/callable for on_connect (Opaque for Send safety)
+    on_connect_proc: Option<Opaque<Value>>,
+    /// Ruby proc/callable for on_disconnect (Opaque for Send safety)
+    on_disconnect_proc: Option<Opaque<Value>>,
+}
+
+impl RubySseEventProducer {
+    /// Create a new Ruby SSE event producer
+    #[allow(dead_code)]
+    pub fn new(
+        name: String,
+        next_event_proc: Value,
+        on_connect_proc: Option<Value>,
+        on_disconnect_proc: Option<Value>,
+    ) -> Self {
+        Self {
+            name,
+            next_event_proc: next_event_proc.into(),
+            on_connect_proc: on_connect_proc.map(|v| v.into()),
+            on_disconnect_proc: on_disconnect_proc.map(|v| v.into()),
+        }
+    }
+
+    /// Convert Ruby value to JSON
+    fn ruby_to_json(ruby: &magnus::Ruby, value: Value) -> Result<JsonValue, String> {
+        if value.is_nil() {
+            return Ok(JsonValue::Null);
+        }
+
+        let json_module: Value = ruby
+            .class_object()
+            .const_get("JSON")
+            .map_err(|e| format!("JSON module not available: {}", e))?;
+
+        let json_str: String = json_module
+            .funcall("generate", (value,))
+            .map_err(|e| format!("Failed to generate JSON: {}", e))?;
+
+        serde_json::from_str(&json_str).map_err(|e| format!("Failed to parse JSON: {}", e))
+    }
+}
+
+impl SseEventProducer for RubySseEventProducer {
+    async fn next_event(&self) -> Option<SseEvent> {
+        debug!("Ruby SSE producer '{}': next_event", self.name);
+
+        match magnus::Ruby::get()
+            .map_err(|e| format!("Failed to get Ruby: {}", e))
+            .and_then(|ruby| {
+                debug!("Ruby SSE producer: acquired Ruby VM");
+
+                let proc_value = ruby.get_inner(self.next_event_proc);
+                let result: Value = proc_value
+                    .funcall("call", ())
+                    .map_err(|e| format!("Producer '{}' call failed: {}", self.name, e))?;
+
+                debug!("Ruby SSE producer: called next_event proc");
+
+                if result.is_nil() {
+                    debug!("Ruby SSE producer: received nil, ending stream");
+                    return Ok(None);
+                }
+
+                let result_hash = if let Some(hash) = RHash::from_value(result) {
+                    hash
+                } else {
+                    let hash_value: Value = result.funcall("to_h", ()).map_err(|e| {
+                        format!(
+                            "next_event must return a Hash/SseEvent convertible via to_h ({}): {}",
+                            unsafe { result.classname() },
+                            e
+                        )
+                    })?;
+                    RHash::from_value(hash_value).ok_or_else(|| {
+                        format!("next_event to_h must return a Hash, got {}", unsafe {
+                            hash_value.classname()
+                        })
+                    })?
+                };
+
+                let data_value = result_hash
+                    .get(ruby.to_symbol("data"))
+                    .ok_or_else(|| "next_event Hash must have :data key".to_string())?;
+
+                let data_json = Self::ruby_to_json(&ruby, data_value)?;
+
+                let event_type: Option<String> = result_hash
+                    .get(ruby.to_symbol("event_type"))
+                    .and_then(|v| if v.is_nil() { None } else { String::try_convert(v).ok() });
+
+                let id: Option<String> = result_hash
+                    .get(ruby.to_symbol("id"))
+                    .and_then(|v| if v.is_nil() { None } else { String::try_convert(v).ok() });
+
+                let retry: Option<u64> = result_hash
+                    .get(ruby.to_symbol("retry"))
+                    .and_then(|v| if v.is_nil() { None } else { u64::try_convert(v).ok() });
+
+                let mut event = if let Some(et) = event_type {
+                    SseEvent::with_type(et, data_json)
+                } else {
+                    SseEvent::new(data_json)
+                };
+
+                if let Some(id_str) = id {
+                    event = event.with_id(id_str);
+                }
+
+                if let Some(retry_ms) = retry {
+                    event = event.with_retry(retry_ms);
+                }
+
+                Ok(Some(event))
+            }) {
+            Ok(event) => event,
+            Err(e) => {
+                error!("Ruby error in next_event: {}", e);
+                None
+            }
+        }
+    }
+
+    async fn on_connect(&self) {
+        debug!("Ruby SSE producer '{}': on_connect", self.name);
+
+        if let Some(on_connect_proc) = self.on_connect_proc
+            && let Err(e) = magnus::Ruby::get()
+                .map_err(|e| format!("Failed to get Ruby: {}", e))
+                .and_then(|ruby| {
+                    debug!("Ruby SSE producer: on_connect acquired Ruby VM");
+                    let proc_value = ruby.get_inner(on_connect_proc);
+                    proc_value
+                        .funcall::<_, _, Value>("call", ())
+                        .map_err(|e| format!("on_connect '{}' call failed: {}", self.name, e))?;
+                    debug!("Ruby SSE producer: on_connect completed");
+                    Ok(())
+                })
+        {
+            error!("on_connect error: {}", e);
+        }
+    }
+
+    async fn on_disconnect(&self) {
+        debug!("Ruby SSE producer '{}': on_disconnect", self.name);
+
+        if let Some(on_disconnect_proc) = self.on_disconnect_proc
+            && let Err(e) = magnus::Ruby::get()
+                .map_err(|e| format!("Failed to get Ruby: {}", e))
+                .and_then(|ruby| {
+                    let proc_value = ruby.get_inner(on_disconnect_proc);
+                    proc_value
+                        .funcall::<_, _, Value>("call", ())
+                        .map_err(|e| format!("on_disconnect '{}' call failed: {}", self.name, e))?;
+                    debug!("Ruby SSE producer: on_disconnect completed");
+                    Ok(())
+                })
+        {
+            error!("on_disconnect error: {}", e);
+        }
+    }
+}
+
+unsafe impl Send for RubySseEventProducer {}
+unsafe impl Sync for RubySseEventProducer {}
+
+/// Create SseState from Ruby producer object
+///
+/// This function is designed to be called from Ruby to register SSE producers.
+#[allow(dead_code)]
+pub fn create_sse_state(
+    ruby: &magnus::Ruby,
+    producer_obj: Value,
+) -> Result<spikard_http::SseState<RubySseEventProducer>, magnus::Error> {
+    let next_event_proc: Value = producer_obj
+        .funcall("method", (ruby.to_symbol("next_event"),))
+        .map_err(|e| {
+            magnus::Error::new(
+                ruby.exception_arg_error(),
+                format!("next_event method not found: {}", e),
+            )
+        })?;
+
+    let on_connect_proc = producer_obj
+        .funcall::<_, _, Value>("method", (ruby.to_symbol("on_connect"),))
+        .ok();
+
+    let on_disconnect_proc = producer_obj
+        .funcall::<_, _, Value>("method", (ruby.to_symbol("on_disconnect"),))
+        .ok();
+
+    let event_schema = producer_obj
+        .funcall::<_, _, Value>("instance_variable_get", (ruby.to_symbol("@_event_schema"),))
+        .ok()
+        .and_then(|v| {
+            if v.is_nil() {
+                None
+            } else {
+                RubySseEventProducer::ruby_to_json(ruby, v).ok()
+            }
+        });
+
+    let ruby_producer = RubySseEventProducer::new(
+        "SseEventProducer".to_string(),
+        next_event_proc,
+        on_connect_proc,
+        on_disconnect_proc,
+    );
+
+    if event_schema.is_some() {
+        spikard_http::SseState::with_schema(ruby_producer, event_schema)
+            .map_err(|e| magnus::Error::new(ruby.exception_runtime_error(), e))
+    } else {
+        Ok(spikard_http::SseState::new(ruby_producer))
+    }
+}