spikard 0.3.4 → 0.3.6
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/LICENSE +1 -1
- data/README.md +659 -659
- data/ext/spikard_rb/Cargo.toml +17 -17
- data/ext/spikard_rb/extconf.rb +10 -10
- data/ext/spikard_rb/src/lib.rs +6 -6
- data/lib/spikard/app.rb +386 -386
- data/lib/spikard/background.rb +27 -27
- data/lib/spikard/config.rb +396 -396
- data/lib/spikard/converters.rb +13 -13
- data/lib/spikard/handler_wrapper.rb +113 -113
- data/lib/spikard/provide.rb +214 -214
- data/lib/spikard/response.rb +173 -173
- data/lib/spikard/schema.rb +243 -243
- data/lib/spikard/sse.rb +111 -111
- data/lib/spikard/streaming_response.rb +44 -44
- data/lib/spikard/testing.rb +221 -221
- data/lib/spikard/upload_file.rb +131 -131
- data/lib/spikard/version.rb +5 -5
- data/lib/spikard/websocket.rb +59 -59
- data/lib/spikard.rb +43 -43
- data/sig/spikard.rbs +366 -360
- data/vendor/crates/spikard-core/Cargo.toml +40 -40
- data/vendor/crates/spikard-core/src/bindings/mod.rs +3 -3
- data/vendor/crates/spikard-core/src/bindings/response.rs +133 -133
- data/vendor/crates/spikard-core/src/debug.rs +63 -63
- data/vendor/crates/spikard-core/src/di/container.rs +726 -726
- data/vendor/crates/spikard-core/src/di/dependency.rs +273 -273
- data/vendor/crates/spikard-core/src/di/error.rs +118 -118
- data/vendor/crates/spikard-core/src/di/factory.rs +538 -538
- data/vendor/crates/spikard-core/src/di/graph.rs +545 -545
- data/vendor/crates/spikard-core/src/di/mod.rs +192 -192
- data/vendor/crates/spikard-core/src/di/resolved.rs +411 -411
- data/vendor/crates/spikard-core/src/di/value.rs +283 -283
- data/vendor/crates/spikard-core/src/errors.rs +39 -39
- data/vendor/crates/spikard-core/src/http.rs +153 -153
- data/vendor/crates/spikard-core/src/lib.rs +29 -29
- data/vendor/crates/spikard-core/src/lifecycle.rs +422 -422
- data/vendor/crates/spikard-core/src/parameters.rs +722 -722
- data/vendor/crates/spikard-core/src/problem.rs +310 -310
- data/vendor/crates/spikard-core/src/request_data.rs +189 -189
- data/vendor/crates/spikard-core/src/router.rs +249 -249
- data/vendor/crates/spikard-core/src/schema_registry.rs +183 -183
- data/vendor/crates/spikard-core/src/type_hints.rs +304 -304
- data/vendor/crates/spikard-core/src/validation.rs +699 -699
- data/vendor/crates/spikard-http/Cargo.toml +68 -58
- data/vendor/crates/spikard-http/src/auth.rs +247 -247
- data/vendor/crates/spikard-http/src/background.rs +249 -249
- data/vendor/crates/spikard-http/src/bindings/mod.rs +3 -3
- data/vendor/crates/spikard-http/src/bindings/response.rs +1 -1
- data/vendor/crates/spikard-http/src/body_metadata.rs +8 -8
- data/vendor/crates/spikard-http/src/cors.rs +490 -490
- data/vendor/crates/spikard-http/src/debug.rs +63 -63
- data/vendor/crates/spikard-http/src/di_handler.rs +423 -423
- data/vendor/crates/spikard-http/src/handler_response.rs +190 -190
- data/vendor/crates/spikard-http/src/handler_trait.rs +228 -228
- data/vendor/crates/spikard-http/src/handler_trait_tests.rs +284 -284
- data/vendor/crates/spikard-http/src/lib.rs +529 -529
- data/vendor/crates/spikard-http/src/lifecycle/adapter.rs +149 -149
- data/vendor/crates/spikard-http/src/lifecycle.rs +428 -428
- data/vendor/crates/spikard-http/src/middleware/mod.rs +285 -285
- data/vendor/crates/spikard-http/src/middleware/multipart.rs +86 -86
- data/vendor/crates/spikard-http/src/middleware/urlencoded.rs +147 -147
- data/vendor/crates/spikard-http/src/middleware/validation.rs +287 -287
- data/vendor/crates/spikard-http/src/openapi/mod.rs +309 -309
- data/vendor/crates/spikard-http/src/openapi/parameter_extraction.rs +190 -190
- data/vendor/crates/spikard-http/src/openapi/schema_conversion.rs +308 -308
- data/vendor/crates/spikard-http/src/openapi/spec_generation.rs +195 -195
- data/vendor/crates/spikard-http/src/parameters.rs +1 -1
- data/vendor/crates/spikard-http/src/problem.rs +1 -1
- data/vendor/crates/spikard-http/src/query_parser.rs +369 -369
- data/vendor/crates/spikard-http/src/response.rs +399 -399
- data/vendor/crates/spikard-http/src/router.rs +1 -1
- data/vendor/crates/spikard-http/src/schema_registry.rs +1 -1
- data/vendor/crates/spikard-http/src/server/handler.rs +87 -87
- data/vendor/crates/spikard-http/src/server/lifecycle_execution.rs +98 -98
- data/vendor/crates/spikard-http/src/server/mod.rs +805 -805
- data/vendor/crates/spikard-http/src/server/request_extraction.rs +119 -119
- data/vendor/crates/spikard-http/src/sse.rs +447 -447
- data/vendor/crates/spikard-http/src/testing/form.rs +14 -14
- data/vendor/crates/spikard-http/src/testing/multipart.rs +60 -60
- data/vendor/crates/spikard-http/src/testing/test_client.rs +285 -285
- data/vendor/crates/spikard-http/src/testing.rs +377 -377
- data/vendor/crates/spikard-http/src/type_hints.rs +1 -1
- data/vendor/crates/spikard-http/src/validation.rs +1 -1
- data/vendor/crates/spikard-http/src/websocket.rs +324 -324
- data/vendor/crates/spikard-rb/Cargo.toml +42 -42
- data/vendor/crates/spikard-rb/build.rs +8 -8
- data/vendor/crates/spikard-rb/src/background.rs +63 -63
- data/vendor/crates/spikard-rb/src/config.rs +294 -294
- data/vendor/crates/spikard-rb/src/conversion.rs +453 -453
- data/vendor/crates/spikard-rb/src/di.rs +409 -409
- data/vendor/crates/spikard-rb/src/handler.rs +625 -625
- data/vendor/crates/spikard-rb/src/lib.rs +2771 -2771
- data/vendor/crates/spikard-rb/src/lifecycle.rs +274 -274
- data/vendor/crates/spikard-rb/src/server.rs +283 -283
- data/vendor/crates/spikard-rb/src/sse.rs +231 -231
- data/vendor/crates/spikard-rb/src/test_client.rs +404 -404
- data/vendor/crates/spikard-rb/src/test_sse.rs +143 -143
- data/vendor/crates/spikard-rb/src/test_websocket.rs +221 -221
- data/vendor/crates/spikard-rb/src/websocket.rs +233 -233
- metadata +1 -79
- data/vendor/spikard-core/Cargo.toml +0 -40
- data/vendor/spikard-core/src/bindings/mod.rs +0 -3
- data/vendor/spikard-core/src/bindings/response.rs +0 -133
- data/vendor/spikard-core/src/debug.rs +0 -63
- data/vendor/spikard-core/src/di/container.rs +0 -726
- data/vendor/spikard-core/src/di/dependency.rs +0 -273
- data/vendor/spikard-core/src/di/error.rs +0 -118
- data/vendor/spikard-core/src/di/factory.rs +0 -538
- data/vendor/spikard-core/src/di/graph.rs +0 -545
- data/vendor/spikard-core/src/di/mod.rs +0 -192
- data/vendor/spikard-core/src/di/resolved.rs +0 -411
- data/vendor/spikard-core/src/di/value.rs +0 -283
- data/vendor/spikard-core/src/http.rs +0 -153
- data/vendor/spikard-core/src/lib.rs +0 -28
- data/vendor/spikard-core/src/lifecycle.rs +0 -422
- data/vendor/spikard-core/src/parameters.rs +0 -719
- data/vendor/spikard-core/src/problem.rs +0 -310
- data/vendor/spikard-core/src/request_data.rs +0 -189
- data/vendor/spikard-core/src/router.rs +0 -249
- data/vendor/spikard-core/src/schema_registry.rs +0 -183
- data/vendor/spikard-core/src/type_hints.rs +0 -304
- data/vendor/spikard-core/src/validation.rs +0 -699
- data/vendor/spikard-http/Cargo.toml +0 -58
- data/vendor/spikard-http/src/auth.rs +0 -247
- data/vendor/spikard-http/src/background.rs +0 -249
- data/vendor/spikard-http/src/bindings/mod.rs +0 -3
- data/vendor/spikard-http/src/bindings/response.rs +0 -1
- data/vendor/spikard-http/src/body_metadata.rs +0 -8
- data/vendor/spikard-http/src/cors.rs +0 -490
- data/vendor/spikard-http/src/debug.rs +0 -63
- data/vendor/spikard-http/src/di_handler.rs +0 -423
- data/vendor/spikard-http/src/handler_response.rs +0 -190
- data/vendor/spikard-http/src/handler_trait.rs +0 -228
- data/vendor/spikard-http/src/handler_trait_tests.rs +0 -284
- data/vendor/spikard-http/src/lib.rs +0 -529
- data/vendor/spikard-http/src/lifecycle/adapter.rs +0 -149
- data/vendor/spikard-http/src/lifecycle.rs +0 -428
- data/vendor/spikard-http/src/middleware/mod.rs +0 -285
- data/vendor/spikard-http/src/middleware/multipart.rs +0 -86
- data/vendor/spikard-http/src/middleware/urlencoded.rs +0 -147
- data/vendor/spikard-http/src/middleware/validation.rs +0 -287
- data/vendor/spikard-http/src/openapi/mod.rs +0 -309
- data/vendor/spikard-http/src/openapi/parameter_extraction.rs +0 -190
- data/vendor/spikard-http/src/openapi/schema_conversion.rs +0 -308
- data/vendor/spikard-http/src/openapi/spec_generation.rs +0 -195
- data/vendor/spikard-http/src/parameters.rs +0 -1
- data/vendor/spikard-http/src/problem.rs +0 -1
- data/vendor/spikard-http/src/query_parser.rs +0 -369
- data/vendor/spikard-http/src/response.rs +0 -399
- data/vendor/spikard-http/src/router.rs +0 -1
- data/vendor/spikard-http/src/schema_registry.rs +0 -1
- data/vendor/spikard-http/src/server/handler.rs +0 -80
- data/vendor/spikard-http/src/server/lifecycle_execution.rs +0 -98
- data/vendor/spikard-http/src/server/mod.rs +0 -805
- data/vendor/spikard-http/src/server/request_extraction.rs +0 -119
- data/vendor/spikard-http/src/sse.rs +0 -447
- data/vendor/spikard-http/src/testing/form.rs +0 -14
- data/vendor/spikard-http/src/testing/multipart.rs +0 -60
- data/vendor/spikard-http/src/testing/test_client.rs +0 -285
- data/vendor/spikard-http/src/testing.rs +0 -377
- data/vendor/spikard-http/src/type_hints.rs +0 -1
- data/vendor/spikard-http/src/validation.rs +0 -1
- data/vendor/spikard-http/src/websocket.rs +0 -324
- data/vendor/spikard-rb/Cargo.toml +0 -42
- data/vendor/spikard-rb/build.rs +0 -8
- data/vendor/spikard-rb/src/background.rs +0 -63
- data/vendor/spikard-rb/src/config.rs +0 -294
- data/vendor/spikard-rb/src/conversion.rs +0 -392
- data/vendor/spikard-rb/src/di.rs +0 -409
- data/vendor/spikard-rb/src/handler.rs +0 -534
- data/vendor/spikard-rb/src/lib.rs +0 -2020
- data/vendor/spikard-rb/src/lifecycle.rs +0 -267
- data/vendor/spikard-rb/src/server.rs +0 -283
- data/vendor/spikard-rb/src/sse.rs +0 -231
- data/vendor/spikard-rb/src/test_client.rs +0 -404
- data/vendor/spikard-rb/src/test_sse.rs +0 -143
- data/vendor/spikard-rb/src/test_websocket.rs +0 -221
- data/vendor/spikard-rb/src/websocket.rs +0 -233
|
@@ -1,274 +1,274 @@
|
|
|
1
|
-
//! Ruby lifecycle hooks implementation
|
|
2
|
-
//!
|
|
3
|
-
//! This module provides the bridge between Ruby blocks/procs and Rust's lifecycle hook system.
|
|
4
|
-
//! Uses magnus to safely call Ruby code from Rust async tasks.
|
|
5
|
-
|
|
6
|
-
use axum::{
|
|
7
|
-
body::Body,
|
|
8
|
-
http::{Request, Response, StatusCode},
|
|
9
|
-
};
|
|
10
|
-
use magnus::{RHash, Value, gc::Marker, prelude::*, value::InnerValue, value::Opaque};
|
|
11
|
-
use serde_json::Value as JsonValue;
|
|
12
|
-
use spikard_http::lifecycle::{HookResult, LifecycleHook};
|
|
13
|
-
use std::future::Future;
|
|
14
|
-
use std::pin::Pin;
|
|
15
|
-
|
|
16
|
-
/// Ruby lifecycle hook wrapper
|
|
17
|
-
///
|
|
18
|
-
/// Wraps a Ruby proc/block and makes it callable from Rust's lifecycle system.
|
|
19
|
-
/// Handles conversion between Rust HTTP types and Ruby request/response objects.
|
|
20
|
-
pub struct RubyLifecycleHook {
|
|
21
|
-
name: String,
|
|
22
|
-
/// Ruby proc/callable object (Opaque for Send safety)
|
|
23
|
-
func: Opaque<Value>,
|
|
24
|
-
}
|
|
25
|
-
|
|
26
|
-
impl RubyLifecycleHook {
|
|
27
|
-
/// Create a new Ruby lifecycle hook
|
|
28
|
-
pub fn new(name: String, func: Value) -> Self {
|
|
29
|
-
Self {
|
|
30
|
-
name,
|
|
31
|
-
func: func.into(),
|
|
32
|
-
}
|
|
33
|
-
}
|
|
34
|
-
|
|
35
|
-
/// Mark Ruby values for GC
|
|
36
|
-
pub fn mark(&self, marker: &Marker) {
|
|
37
|
-
if let Ok(ruby) = magnus::Ruby::get() {
|
|
38
|
-
marker.mark(self.func.get_inner_with(&ruby));
|
|
39
|
-
}
|
|
40
|
-
}
|
|
41
|
-
}
|
|
42
|
-
|
|
43
|
-
impl LifecycleHook<Request<Body>, Response<Body>> for RubyLifecycleHook {
|
|
44
|
-
fn name(&self) -> &str {
|
|
45
|
-
&self.name
|
|
46
|
-
}
|
|
47
|
-
|
|
48
|
-
fn execute_request<'a>(
|
|
49
|
-
&'a self,
|
|
50
|
-
req: Request<Body>,
|
|
51
|
-
) -> Pin<Box<dyn Future<Output = Result<HookResult<Request<Body>, Response<Body>>, String>> + Send + 'a>> {
|
|
52
|
-
let func = self.func;
|
|
53
|
-
let name = self.name.clone();
|
|
54
|
-
let name_for_error = name.clone();
|
|
55
|
-
|
|
56
|
-
Box::pin(async move {
|
|
57
|
-
let (parts, body) = req.into_parts();
|
|
58
|
-
let body_bytes = axum::body::to_bytes(body, usize::MAX)
|
|
59
|
-
.await
|
|
60
|
-
.map_err(|e| format!("Failed to read request body: {}", e))?;
|
|
61
|
-
|
|
62
|
-
let body_value: JsonValue = if body_bytes.is_empty() {
|
|
63
|
-
JsonValue::Null
|
|
64
|
-
} else {
|
|
65
|
-
serde_json::from_slice(&body_bytes)
|
|
66
|
-
.unwrap_or_else(|_| JsonValue::String(String::from_utf8_lossy(&body_bytes).to_string()))
|
|
67
|
-
};
|
|
68
|
-
|
|
69
|
-
let result = magnus::Ruby::get()
|
|
70
|
-
.map_err(|e| format!("Failed to get Ruby: {}", e))
|
|
71
|
-
.and_then(|ruby| {
|
|
72
|
-
let request_hash = RHash::new();
|
|
73
|
-
|
|
74
|
-
request_hash
|
|
75
|
-
.aset(ruby.to_symbol("method"), ruby.str_new(parts.method.as_str()))
|
|
76
|
-
.map_err(|e| format!("Failed to set method: {}", e))?;
|
|
77
|
-
request_hash
|
|
78
|
-
.aset(ruby.to_symbol("path"), ruby.str_new(parts.uri.path()))
|
|
79
|
-
.map_err(|e| format!("Failed to set path: {}", e))?;
|
|
80
|
-
|
|
81
|
-
let headers_hash = RHash::new();
|
|
82
|
-
for (key, value) in parts.headers.iter() {
|
|
83
|
-
headers_hash
|
|
84
|
-
.aset(ruby.str_new(key.as_str()), ruby.str_new(value.to_str().unwrap_or("")))
|
|
85
|
-
.map_err(|e| format!("Failed to set header: {}", e))?;
|
|
86
|
-
}
|
|
87
|
-
request_hash
|
|
88
|
-
.aset(ruby.to_symbol("headers"), headers_hash)
|
|
89
|
-
.map_err(|e| format!("Failed to set headers: {}", e))?;
|
|
90
|
-
|
|
91
|
-
let body_str =
|
|
92
|
-
serde_json::to_string(&body_value).map_err(|e| format!("Failed to serialize body: {}", e))?;
|
|
93
|
-
request_hash
|
|
94
|
-
.aset(ruby.to_symbol("body"), ruby.str_new(&body_str))
|
|
95
|
-
.map_err(|e| format!("Failed to set body: {}", e))?;
|
|
96
|
-
|
|
97
|
-
let func_value = ruby.get_inner(func);
|
|
98
|
-
let result: Value = func_value
|
|
99
|
-
.funcall("call", (request_hash,))
|
|
100
|
-
.map_err(|e| format!("Hook '{}' call failed: {}", name, e))?;
|
|
101
|
-
|
|
102
|
-
if let Some(result_hash) = RHash::from_value(result) {
|
|
103
|
-
if let Some(status_value) = result_hash.get(ruby.to_symbol("status_code")) {
|
|
104
|
-
let status = i64::try_convert(status_value)
|
|
105
|
-
.map_err(|e| format!("Failed to convert status code: {}", e))?;
|
|
106
|
-
|
|
107
|
-
let content = result_hash
|
|
108
|
-
.get(ruby.to_symbol("content"))
|
|
109
|
-
.or_else(|| result_hash.get(ruby.to_symbol("body")))
|
|
110
|
-
.unwrap_or_else(|| ruby.qnil().as_value());
|
|
111
|
-
|
|
112
|
-
let body_str = if content.is_nil() {
|
|
113
|
-
"{}".to_string()
|
|
114
|
-
} else {
|
|
115
|
-
String::try_convert(content).unwrap_or_else(|_| {
|
|
116
|
-
content
|
|
117
|
-
.to_r_string()
|
|
118
|
-
.map(|s| s.to_string().unwrap_or_else(|_| "{}".to_string()))
|
|
119
|
-
.unwrap_or_else(|_| "{}".to_string())
|
|
120
|
-
})
|
|
121
|
-
};
|
|
122
|
-
|
|
123
|
-
let response = Response::builder()
|
|
124
|
-
.status(StatusCode::from_u16(status as u16).unwrap_or(StatusCode::OK))
|
|
125
|
-
.header("content-type", "application/json")
|
|
126
|
-
.body(Body::from(body_str))
|
|
127
|
-
.map_err(|e| format!("Failed to build response: {}", e))?;
|
|
128
|
-
|
|
129
|
-
return Ok(HookResult::ShortCircuit(response));
|
|
130
|
-
}
|
|
131
|
-
|
|
132
|
-
let method = result_hash
|
|
133
|
-
.get(ruby.to_symbol("method"))
|
|
134
|
-
.and_then(|v| String::try_convert(v).ok())
|
|
135
|
-
.unwrap_or_else(|| "GET".to_string());
|
|
136
|
-
let path = result_hash
|
|
137
|
-
.get(ruby.to_symbol("path"))
|
|
138
|
-
.and_then(|v| String::try_convert(v).ok())
|
|
139
|
-
.unwrap_or_else(|| "/".to_string());
|
|
140
|
-
|
|
141
|
-
let req_builder = Request::builder().method(method.as_str()).uri(path);
|
|
142
|
-
|
|
143
|
-
let body = if let Some(body_val) = result_hash.get(ruby.to_symbol("body")) {
|
|
144
|
-
if body_val.is_nil() {
|
|
145
|
-
Body::empty()
|
|
146
|
-
} else {
|
|
147
|
-
let body_str = String::try_convert(body_val).unwrap_or_else(|_| {
|
|
148
|
-
body_val
|
|
149
|
-
.to_r_string()
|
|
150
|
-
.map(|s| s.to_string().unwrap_or_default())
|
|
151
|
-
.unwrap_or_default()
|
|
152
|
-
});
|
|
153
|
-
Body::from(body_str)
|
|
154
|
-
}
|
|
155
|
-
} else {
|
|
156
|
-
Body::empty()
|
|
157
|
-
};
|
|
158
|
-
|
|
159
|
-
let request = req_builder
|
|
160
|
-
.body(body)
|
|
161
|
-
.map_err(|e| format!("Failed to build request: {}", e))?;
|
|
162
|
-
|
|
163
|
-
Ok(HookResult::Continue(request))
|
|
164
|
-
} else {
|
|
165
|
-
Err(format!("Hook must return a Hash, got {}", unsafe {
|
|
166
|
-
result.classname()
|
|
167
|
-
}))
|
|
168
|
-
}
|
|
169
|
-
});
|
|
170
|
-
|
|
171
|
-
result.map_err(|e| format!("Hook '{}' task error: {}", name_for_error, e))
|
|
172
|
-
})
|
|
173
|
-
}
|
|
174
|
-
|
|
175
|
-
fn execute_response<'a>(
|
|
176
|
-
&'a self,
|
|
177
|
-
resp: Response<Body>,
|
|
178
|
-
) -> Pin<Box<dyn Future<Output = Result<HookResult<Response<Body>, Response<Body>>, String>> + Send + 'a>> {
|
|
179
|
-
let func = self.func;
|
|
180
|
-
let name = self.name.clone();
|
|
181
|
-
let name_for_error = name.clone();
|
|
182
|
-
|
|
183
|
-
Box::pin(async move {
|
|
184
|
-
let (parts, body) = resp.into_parts();
|
|
185
|
-
let body_bytes = axum::body::to_bytes(body, usize::MAX)
|
|
186
|
-
.await
|
|
187
|
-
.map_err(|e| format!("Failed to read response body: {}", e))?;
|
|
188
|
-
|
|
189
|
-
let body_value: JsonValue = if body_bytes.is_empty() {
|
|
190
|
-
JsonValue::Null
|
|
191
|
-
} else {
|
|
192
|
-
serde_json::from_slice(&body_bytes)
|
|
193
|
-
.unwrap_or_else(|_| JsonValue::String(String::from_utf8_lossy(&body_bytes).to_string()))
|
|
194
|
-
};
|
|
195
|
-
|
|
196
|
-
let result = magnus::Ruby::get()
|
|
197
|
-
.map_err(|e| format!("Failed to get Ruby: {}", e))
|
|
198
|
-
.and_then(|ruby| {
|
|
199
|
-
let response_hash = RHash::new();
|
|
200
|
-
|
|
201
|
-
response_hash
|
|
202
|
-
.aset(
|
|
203
|
-
ruby.to_symbol("status_code"),
|
|
204
|
-
ruby.integer_from_i64(parts.status.as_u16() as i64),
|
|
205
|
-
)
|
|
206
|
-
.map_err(|e| format!("Failed to set status_code: {}", e))?;
|
|
207
|
-
|
|
208
|
-
let headers_hash = RHash::new();
|
|
209
|
-
for (key, value) in parts.headers.iter() {
|
|
210
|
-
headers_hash
|
|
211
|
-
.aset(ruby.str_new(key.as_str()), ruby.str_new(value.to_str().unwrap_or("")))
|
|
212
|
-
.map_err(|e| format!("Failed to set header: {}", e))?;
|
|
213
|
-
}
|
|
214
|
-
response_hash
|
|
215
|
-
.aset(ruby.to_symbol("headers"), headers_hash)
|
|
216
|
-
.map_err(|e| format!("Failed to set headers: {}", e))?;
|
|
217
|
-
|
|
218
|
-
let body_str =
|
|
219
|
-
serde_json::to_string(&body_value).map_err(|e| format!("Failed to serialize body: {}", e))?;
|
|
220
|
-
response_hash
|
|
221
|
-
.aset(ruby.to_symbol("content"), ruby.str_new(&body_str))
|
|
222
|
-
.map_err(|e| format!("Failed to set content: {}", e))?;
|
|
223
|
-
|
|
224
|
-
let func_value = ruby.get_inner(func);
|
|
225
|
-
let result: Value = func_value
|
|
226
|
-
.funcall("call", (response_hash,))
|
|
227
|
-
.map_err(|e| format!("Hook '{}' call failed: {}", name, e))?;
|
|
228
|
-
|
|
229
|
-
if let Some(result_hash) = RHash::from_value(result) {
|
|
230
|
-
let status = result_hash
|
|
231
|
-
.get(ruby.to_symbol("status_code"))
|
|
232
|
-
.and_then(|v| i64::try_convert(v).ok())
|
|
233
|
-
.unwrap_or(200);
|
|
234
|
-
|
|
235
|
-
let content = result_hash
|
|
236
|
-
.get(ruby.to_symbol("content"))
|
|
237
|
-
.or_else(|| result_hash.get(ruby.to_symbol("body")))
|
|
238
|
-
.unwrap_or_else(|| ruby.qnil().as_value());
|
|
239
|
-
|
|
240
|
-
let body_str = if content.is_nil() {
|
|
241
|
-
"{}".to_string()
|
|
242
|
-
} else {
|
|
243
|
-
String::try_convert(content).unwrap_or_else(|_| {
|
|
244
|
-
content
|
|
245
|
-
.to_r_string()
|
|
246
|
-
.map(|s| s.to_string().unwrap_or_else(|_| "{}".to_string()))
|
|
247
|
-
.unwrap_or_else(|_| "{}".to_string())
|
|
248
|
-
})
|
|
249
|
-
};
|
|
250
|
-
|
|
251
|
-
let mut response_builder =
|
|
252
|
-
Response::builder().status(StatusCode::from_u16(status as u16).unwrap_or(StatusCode::OK));
|
|
253
|
-
|
|
254
|
-
response_builder = response_builder.header("content-type", "application/json");
|
|
255
|
-
|
|
256
|
-
let response = response_builder
|
|
257
|
-
.body(Body::from(body_str))
|
|
258
|
-
.map_err(|e| format!("Failed to build response: {}", e))?;
|
|
259
|
-
|
|
260
|
-
Ok(HookResult::Continue(response))
|
|
261
|
-
} else {
|
|
262
|
-
Err(format!("Hook must return a Hash, got {}", unsafe {
|
|
263
|
-
result.classname()
|
|
264
|
-
}))
|
|
265
|
-
}
|
|
266
|
-
});
|
|
267
|
-
|
|
268
|
-
result.map_err(|e| format!("Hook '{}' task error: {}", name_for_error, e))
|
|
269
|
-
})
|
|
270
|
-
}
|
|
271
|
-
}
|
|
272
|
-
|
|
273
|
-
unsafe impl Send for RubyLifecycleHook {}
|
|
274
|
-
unsafe impl Sync for RubyLifecycleHook {}
|
|
1
|
+
//! Ruby lifecycle hooks implementation
|
|
2
|
+
//!
|
|
3
|
+
//! This module provides the bridge between Ruby blocks/procs and Rust's lifecycle hook system.
|
|
4
|
+
//! Uses magnus to safely call Ruby code from Rust async tasks.
|
|
5
|
+
|
|
6
|
+
use axum::{
|
|
7
|
+
body::Body,
|
|
8
|
+
http::{Request, Response, StatusCode},
|
|
9
|
+
};
|
|
10
|
+
use magnus::{RHash, Value, gc::Marker, prelude::*, value::InnerValue, value::Opaque};
|
|
11
|
+
use serde_json::Value as JsonValue;
|
|
12
|
+
use spikard_http::lifecycle::{HookResult, LifecycleHook};
|
|
13
|
+
use std::future::Future;
|
|
14
|
+
use std::pin::Pin;
|
|
15
|
+
|
|
16
|
+
/// Ruby lifecycle hook wrapper
|
|
17
|
+
///
|
|
18
|
+
/// Wraps a Ruby proc/block and makes it callable from Rust's lifecycle system.
|
|
19
|
+
/// Handles conversion between Rust HTTP types and Ruby request/response objects.
|
|
20
|
+
pub struct RubyLifecycleHook {
|
|
21
|
+
name: String,
|
|
22
|
+
/// Ruby proc/callable object (Opaque for Send safety)
|
|
23
|
+
func: Opaque<Value>,
|
|
24
|
+
}
|
|
25
|
+
|
|
26
|
+
impl RubyLifecycleHook {
|
|
27
|
+
/// Create a new Ruby lifecycle hook
|
|
28
|
+
pub fn new(name: String, func: Value) -> Self {
|
|
29
|
+
Self {
|
|
30
|
+
name,
|
|
31
|
+
func: func.into(),
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
|
|
35
|
+
/// Mark Ruby values for GC
|
|
36
|
+
pub fn mark(&self, marker: &Marker) {
|
|
37
|
+
if let Ok(ruby) = magnus::Ruby::get() {
|
|
38
|
+
marker.mark(self.func.get_inner_with(&ruby));
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
impl LifecycleHook<Request<Body>, Response<Body>> for RubyLifecycleHook {
|
|
44
|
+
fn name(&self) -> &str {
|
|
45
|
+
&self.name
|
|
46
|
+
}
|
|
47
|
+
|
|
48
|
+
fn execute_request<'a>(
|
|
49
|
+
&'a self,
|
|
50
|
+
req: Request<Body>,
|
|
51
|
+
) -> Pin<Box<dyn Future<Output = Result<HookResult<Request<Body>, Response<Body>>, String>> + Send + 'a>> {
|
|
52
|
+
let func = self.func;
|
|
53
|
+
let name = self.name.clone();
|
|
54
|
+
let name_for_error = name.clone();
|
|
55
|
+
|
|
56
|
+
Box::pin(async move {
|
|
57
|
+
let (parts, body) = req.into_parts();
|
|
58
|
+
let body_bytes = axum::body::to_bytes(body, usize::MAX)
|
|
59
|
+
.await
|
|
60
|
+
.map_err(|e| format!("Failed to read request body: {}", e))?;
|
|
61
|
+
|
|
62
|
+
let body_value: JsonValue = if body_bytes.is_empty() {
|
|
63
|
+
JsonValue::Null
|
|
64
|
+
} else {
|
|
65
|
+
serde_json::from_slice(&body_bytes)
|
|
66
|
+
.unwrap_or_else(|_| JsonValue::String(String::from_utf8_lossy(&body_bytes).to_string()))
|
|
67
|
+
};
|
|
68
|
+
|
|
69
|
+
let result = magnus::Ruby::get()
|
|
70
|
+
.map_err(|e| format!("Failed to get Ruby: {}", e))
|
|
71
|
+
.and_then(|ruby| {
|
|
72
|
+
let request_hash = RHash::new();
|
|
73
|
+
|
|
74
|
+
request_hash
|
|
75
|
+
.aset(ruby.to_symbol("method"), ruby.str_new(parts.method.as_str()))
|
|
76
|
+
.map_err(|e| format!("Failed to set method: {}", e))?;
|
|
77
|
+
request_hash
|
|
78
|
+
.aset(ruby.to_symbol("path"), ruby.str_new(parts.uri.path()))
|
|
79
|
+
.map_err(|e| format!("Failed to set path: {}", e))?;
|
|
80
|
+
|
|
81
|
+
let headers_hash = RHash::new();
|
|
82
|
+
for (key, value) in parts.headers.iter() {
|
|
83
|
+
headers_hash
|
|
84
|
+
.aset(ruby.str_new(key.as_str()), ruby.str_new(value.to_str().unwrap_or("")))
|
|
85
|
+
.map_err(|e| format!("Failed to set header: {}", e))?;
|
|
86
|
+
}
|
|
87
|
+
request_hash
|
|
88
|
+
.aset(ruby.to_symbol("headers"), headers_hash)
|
|
89
|
+
.map_err(|e| format!("Failed to set headers: {}", e))?;
|
|
90
|
+
|
|
91
|
+
let body_str =
|
|
92
|
+
serde_json::to_string(&body_value).map_err(|e| format!("Failed to serialize body: {}", e))?;
|
|
93
|
+
request_hash
|
|
94
|
+
.aset(ruby.to_symbol("body"), ruby.str_new(&body_str))
|
|
95
|
+
.map_err(|e| format!("Failed to set body: {}", e))?;
|
|
96
|
+
|
|
97
|
+
let func_value = ruby.get_inner(func);
|
|
98
|
+
let result: Value = func_value
|
|
99
|
+
.funcall("call", (request_hash,))
|
|
100
|
+
.map_err(|e| format!("Hook '{}' call failed: {}", name, e))?;
|
|
101
|
+
|
|
102
|
+
if let Some(result_hash) = RHash::from_value(result) {
|
|
103
|
+
if let Some(status_value) = result_hash.get(ruby.to_symbol("status_code")) {
|
|
104
|
+
let status = i64::try_convert(status_value)
|
|
105
|
+
.map_err(|e| format!("Failed to convert status code: {}", e))?;
|
|
106
|
+
|
|
107
|
+
let content = result_hash
|
|
108
|
+
.get(ruby.to_symbol("content"))
|
|
109
|
+
.or_else(|| result_hash.get(ruby.to_symbol("body")))
|
|
110
|
+
.unwrap_or_else(|| ruby.qnil().as_value());
|
|
111
|
+
|
|
112
|
+
let body_str = if content.is_nil() {
|
|
113
|
+
"{}".to_string()
|
|
114
|
+
} else {
|
|
115
|
+
String::try_convert(content).unwrap_or_else(|_| {
|
|
116
|
+
content
|
|
117
|
+
.to_r_string()
|
|
118
|
+
.map(|s| s.to_string().unwrap_or_else(|_| "{}".to_string()))
|
|
119
|
+
.unwrap_or_else(|_| "{}".to_string())
|
|
120
|
+
})
|
|
121
|
+
};
|
|
122
|
+
|
|
123
|
+
let response = Response::builder()
|
|
124
|
+
.status(StatusCode::from_u16(status as u16).unwrap_or(StatusCode::OK))
|
|
125
|
+
.header("content-type", "application/json")
|
|
126
|
+
.body(Body::from(body_str))
|
|
127
|
+
.map_err(|e| format!("Failed to build response: {}", e))?;
|
|
128
|
+
|
|
129
|
+
return Ok(HookResult::ShortCircuit(response));
|
|
130
|
+
}
|
|
131
|
+
|
|
132
|
+
let method = result_hash
|
|
133
|
+
.get(ruby.to_symbol("method"))
|
|
134
|
+
.and_then(|v| String::try_convert(v).ok())
|
|
135
|
+
.unwrap_or_else(|| "GET".to_string());
|
|
136
|
+
let path = result_hash
|
|
137
|
+
.get(ruby.to_symbol("path"))
|
|
138
|
+
.and_then(|v| String::try_convert(v).ok())
|
|
139
|
+
.unwrap_or_else(|| "/".to_string());
|
|
140
|
+
|
|
141
|
+
let req_builder = Request::builder().method(method.as_str()).uri(path);
|
|
142
|
+
|
|
143
|
+
let body = if let Some(body_val) = result_hash.get(ruby.to_symbol("body")) {
|
|
144
|
+
if body_val.is_nil() {
|
|
145
|
+
Body::empty()
|
|
146
|
+
} else {
|
|
147
|
+
let body_str = String::try_convert(body_val).unwrap_or_else(|_| {
|
|
148
|
+
body_val
|
|
149
|
+
.to_r_string()
|
|
150
|
+
.map(|s| s.to_string().unwrap_or_default())
|
|
151
|
+
.unwrap_or_default()
|
|
152
|
+
});
|
|
153
|
+
Body::from(body_str)
|
|
154
|
+
}
|
|
155
|
+
} else {
|
|
156
|
+
Body::empty()
|
|
157
|
+
};
|
|
158
|
+
|
|
159
|
+
let request = req_builder
|
|
160
|
+
.body(body)
|
|
161
|
+
.map_err(|e| format!("Failed to build request: {}", e))?;
|
|
162
|
+
|
|
163
|
+
Ok(HookResult::Continue(request))
|
|
164
|
+
} else {
|
|
165
|
+
Err(format!("Hook must return a Hash, got {}", unsafe {
|
|
166
|
+
result.classname()
|
|
167
|
+
}))
|
|
168
|
+
}
|
|
169
|
+
});
|
|
170
|
+
|
|
171
|
+
result.map_err(|e| format!("Hook '{}' task error: {}", name_for_error, e))
|
|
172
|
+
})
|
|
173
|
+
}
|
|
174
|
+
|
|
175
|
+
fn execute_response<'a>(
|
|
176
|
+
&'a self,
|
|
177
|
+
resp: Response<Body>,
|
|
178
|
+
) -> Pin<Box<dyn Future<Output = Result<HookResult<Response<Body>, Response<Body>>, String>> + Send + 'a>> {
|
|
179
|
+
let func = self.func;
|
|
180
|
+
let name = self.name.clone();
|
|
181
|
+
let name_for_error = name.clone();
|
|
182
|
+
|
|
183
|
+
Box::pin(async move {
|
|
184
|
+
let (parts, body) = resp.into_parts();
|
|
185
|
+
let body_bytes = axum::body::to_bytes(body, usize::MAX)
|
|
186
|
+
.await
|
|
187
|
+
.map_err(|e| format!("Failed to read response body: {}", e))?;
|
|
188
|
+
|
|
189
|
+
let body_value: JsonValue = if body_bytes.is_empty() {
|
|
190
|
+
JsonValue::Null
|
|
191
|
+
} else {
|
|
192
|
+
serde_json::from_slice(&body_bytes)
|
|
193
|
+
.unwrap_or_else(|_| JsonValue::String(String::from_utf8_lossy(&body_bytes).to_string()))
|
|
194
|
+
};
|
|
195
|
+
|
|
196
|
+
let result = magnus::Ruby::get()
|
|
197
|
+
.map_err(|e| format!("Failed to get Ruby: {}", e))
|
|
198
|
+
.and_then(|ruby| {
|
|
199
|
+
let response_hash = RHash::new();
|
|
200
|
+
|
|
201
|
+
response_hash
|
|
202
|
+
.aset(
|
|
203
|
+
ruby.to_symbol("status_code"),
|
|
204
|
+
ruby.integer_from_i64(parts.status.as_u16() as i64),
|
|
205
|
+
)
|
|
206
|
+
.map_err(|e| format!("Failed to set status_code: {}", e))?;
|
|
207
|
+
|
|
208
|
+
let headers_hash = RHash::new();
|
|
209
|
+
for (key, value) in parts.headers.iter() {
|
|
210
|
+
headers_hash
|
|
211
|
+
.aset(ruby.str_new(key.as_str()), ruby.str_new(value.to_str().unwrap_or("")))
|
|
212
|
+
.map_err(|e| format!("Failed to set header: {}", e))?;
|
|
213
|
+
}
|
|
214
|
+
response_hash
|
|
215
|
+
.aset(ruby.to_symbol("headers"), headers_hash)
|
|
216
|
+
.map_err(|e| format!("Failed to set headers: {}", e))?;
|
|
217
|
+
|
|
218
|
+
let body_str =
|
|
219
|
+
serde_json::to_string(&body_value).map_err(|e| format!("Failed to serialize body: {}", e))?;
|
|
220
|
+
response_hash
|
|
221
|
+
.aset(ruby.to_symbol("content"), ruby.str_new(&body_str))
|
|
222
|
+
.map_err(|e| format!("Failed to set content: {}", e))?;
|
|
223
|
+
|
|
224
|
+
let func_value = ruby.get_inner(func);
|
|
225
|
+
let result: Value = func_value
|
|
226
|
+
.funcall("call", (response_hash,))
|
|
227
|
+
.map_err(|e| format!("Hook '{}' call failed: {}", name, e))?;
|
|
228
|
+
|
|
229
|
+
if let Some(result_hash) = RHash::from_value(result) {
|
|
230
|
+
let status = result_hash
|
|
231
|
+
.get(ruby.to_symbol("status_code"))
|
|
232
|
+
.and_then(|v| i64::try_convert(v).ok())
|
|
233
|
+
.unwrap_or(200);
|
|
234
|
+
|
|
235
|
+
let content = result_hash
|
|
236
|
+
.get(ruby.to_symbol("content"))
|
|
237
|
+
.or_else(|| result_hash.get(ruby.to_symbol("body")))
|
|
238
|
+
.unwrap_or_else(|| ruby.qnil().as_value());
|
|
239
|
+
|
|
240
|
+
let body_str = if content.is_nil() {
|
|
241
|
+
"{}".to_string()
|
|
242
|
+
} else {
|
|
243
|
+
String::try_convert(content).unwrap_or_else(|_| {
|
|
244
|
+
content
|
|
245
|
+
.to_r_string()
|
|
246
|
+
.map(|s| s.to_string().unwrap_or_else(|_| "{}".to_string()))
|
|
247
|
+
.unwrap_or_else(|_| "{}".to_string())
|
|
248
|
+
})
|
|
249
|
+
};
|
|
250
|
+
|
|
251
|
+
let mut response_builder =
|
|
252
|
+
Response::builder().status(StatusCode::from_u16(status as u16).unwrap_or(StatusCode::OK));
|
|
253
|
+
|
|
254
|
+
response_builder = response_builder.header("content-type", "application/json");
|
|
255
|
+
|
|
256
|
+
let response = response_builder
|
|
257
|
+
.body(Body::from(body_str))
|
|
258
|
+
.map_err(|e| format!("Failed to build response: {}", e))?;
|
|
259
|
+
|
|
260
|
+
Ok(HookResult::Continue(response))
|
|
261
|
+
} else {
|
|
262
|
+
Err(format!("Hook must return a Hash, got {}", unsafe {
|
|
263
|
+
result.classname()
|
|
264
|
+
}))
|
|
265
|
+
}
|
|
266
|
+
});
|
|
267
|
+
|
|
268
|
+
result.map_err(|e| format!("Hook '{}' task error: {}", name_for_error, e))
|
|
269
|
+
})
|
|
270
|
+
}
|
|
271
|
+
}
|
|
272
|
+
|
|
273
|
+
unsafe impl Send for RubyLifecycleHook {}
|
|
274
|
+
unsafe impl Sync for RubyLifecycleHook {}
|