spikard 0.12.0 → 0.15.2
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/Steepfile +6 -0
- data/ext/spikard_rb/extconf.rb +1 -2
- data/ext/spikard_rb/{Cargo.lock → native/Cargo.lock} +897 -451
- data/ext/spikard_rb/native/Cargo.toml +24 -0
- data/ext/spikard_rb/src/lib.rs +5366 -3
- data/lib/spikard/native.rb +86 -0
- data/lib/spikard/version.rb +6 -1
- data/lib/spikard.rb +8 -45
- data/lib/spikard_rb.so +0 -0
- data/sig/types.rbs +427 -0
- metadata +14 -242
- data/LICENSE +0 -1
- data/README.md +0 -267
- data/ext/spikard_rb/Cargo.toml +0 -17
- data/lib/spikard/app.rb +0 -428
- data/lib/spikard/background.rb +0 -58
- data/lib/spikard/config.rb +0 -506
- data/lib/spikard/converters.rb +0 -13
- data/lib/spikard/grpc.rb +0 -182
- data/lib/spikard/handler_wrapper.rb +0 -113
- data/lib/spikard/provide.rb +0 -214
- data/lib/spikard/response.rb +0 -173
- data/lib/spikard/schema.rb +0 -243
- data/lib/spikard/sse.rb +0 -111
- data/lib/spikard/streaming_response.rb +0 -44
- data/lib/spikard/testing.rb +0 -432
- data/lib/spikard/upload_file.rb +0 -131
- data/lib/spikard/websocket.rb +0 -59
- data/sig/spikard.rbs +0 -719
- data/vendor/crates/spikard-bindings-shared/Cargo.toml +0 -80
- data/vendor/crates/spikard-bindings-shared/examples/config_extraction.rs +0 -132
- data/vendor/crates/spikard-bindings-shared/src/config_extractor.rs +0 -905
- data/vendor/crates/spikard-bindings-shared/src/conversion_traits.rs +0 -210
- data/vendor/crates/spikard-bindings-shared/src/di_traits.rs +0 -252
- data/vendor/crates/spikard-bindings-shared/src/error_response.rs +0 -404
- data/vendor/crates/spikard-bindings-shared/src/grpc_metadata.rs +0 -199
- data/vendor/crates/spikard-bindings-shared/src/handler_base.rs +0 -252
- data/vendor/crates/spikard-bindings-shared/src/json_conversion.rs +0 -829
- data/vendor/crates/spikard-bindings-shared/src/lazy_cache.rs +0 -587
- data/vendor/crates/spikard-bindings-shared/src/lib.rs +0 -33
- data/vendor/crates/spikard-bindings-shared/src/lifecycle_base.rs +0 -298
- data/vendor/crates/spikard-bindings-shared/src/lifecycle_executor.rs +0 -594
- data/vendor/crates/spikard-bindings-shared/src/response_builder.rs +0 -743
- data/vendor/crates/spikard-bindings-shared/src/response_interpreter.rs +0 -944
- data/vendor/crates/spikard-bindings-shared/src/test_client_base.rs +0 -260
- data/vendor/crates/spikard-bindings-shared/src/validation_helpers.rs +0 -369
- data/vendor/crates/spikard-bindings-shared/tests/config_extractor_behavior.rs +0 -192
- data/vendor/crates/spikard-bindings-shared/tests/error_response_edge_cases.rs +0 -383
- data/vendor/crates/spikard-bindings-shared/tests/full_coverage.rs +0 -459
- data/vendor/crates/spikard-bindings-shared/tests/handler_base_integration.rs +0 -280
- data/vendor/crates/spikard-bindings-shared/tests/integration_tests.rs +0 -669
- data/vendor/crates/spikard-core/Cargo.toml +0 -60
- data/vendor/crates/spikard-core/src/bindings/mod.rs +0 -3
- data/vendor/crates/spikard-core/src/bindings/response.rs +0 -130
- data/vendor/crates/spikard-core/src/debug.rs +0 -127
- data/vendor/crates/spikard-core/src/di/container.rs +0 -702
- data/vendor/crates/spikard-core/src/di/dependency.rs +0 -273
- data/vendor/crates/spikard-core/src/di/error.rs +0 -118
- data/vendor/crates/spikard-core/src/di/factory.rs +0 -538
- data/vendor/crates/spikard-core/src/di/graph.rs +0 -507
- data/vendor/crates/spikard-core/src/di/mod.rs +0 -192
- data/vendor/crates/spikard-core/src/di/resolved.rs +0 -428
- data/vendor/crates/spikard-core/src/di/value.rs +0 -282
- data/vendor/crates/spikard-core/src/errors.rs +0 -72
- data/vendor/crates/spikard-core/src/http.rs +0 -492
- data/vendor/crates/spikard-core/src/lib.rs +0 -29
- data/vendor/crates/spikard-core/src/lifecycle.rs +0 -1273
- data/vendor/crates/spikard-core/src/metadata.rs +0 -378
- data/vendor/crates/spikard-core/src/parameters.rs +0 -2546
- data/vendor/crates/spikard-core/src/problem.rs +0 -358
- data/vendor/crates/spikard-core/src/request_data.rs +0 -1146
- data/vendor/crates/spikard-core/src/router.rs +0 -530
- data/vendor/crates/spikard-core/src/schema_registry.rs +0 -197
- data/vendor/crates/spikard-core/src/type_hints.rs +0 -311
- data/vendor/crates/spikard-core/src/validation/error_mapper.rs +0 -710
- data/vendor/crates/spikard-core/src/validation/mod.rs +0 -470
- data/vendor/crates/spikard-core/tests/bindings_response_tests.rs +0 -136
- data/vendor/crates/spikard-core/tests/di_dependency_defaults.rs +0 -37
- data/vendor/crates/spikard-core/tests/error_mapper.rs +0 -761
- data/vendor/crates/spikard-core/tests/parameters_edge_cases.rs +0 -106
- data/vendor/crates/spikard-core/tests/parameters_full.rs +0 -701
- data/vendor/crates/spikard-core/tests/parameters_schema_and_formats.rs +0 -301
- data/vendor/crates/spikard-core/tests/request_data_roundtrip.rs +0 -67
- data/vendor/crates/spikard-core/tests/validation_coverage.rs +0 -250
- data/vendor/crates/spikard-core/tests/validation_error_paths.rs +0 -45
- data/vendor/crates/spikard-http/Cargo.toml +0 -87
- data/vendor/crates/spikard-http/examples/sse-notifications.rs +0 -148
- data/vendor/crates/spikard-http/examples/websocket-chat.rs +0 -92
- data/vendor/crates/spikard-http/src/auth.rs +0 -301
- data/vendor/crates/spikard-http/src/background.rs +0 -1860
- data/vendor/crates/spikard-http/src/bindings/mod.rs +0 -3
- data/vendor/crates/spikard-http/src/bindings/response.rs +0 -1
- data/vendor/crates/spikard-http/src/body_metadata.rs +0 -8
- data/vendor/crates/spikard-http/src/cors.rs +0 -1026
- data/vendor/crates/spikard-http/src/debug.rs +0 -128
- data/vendor/crates/spikard-http/src/di_handler.rs +0 -1672
- data/vendor/crates/spikard-http/src/grpc/framing.rs +0 -469
- data/vendor/crates/spikard-http/src/grpc/handler.rs +0 -1122
- data/vendor/crates/spikard-http/src/grpc/mod.rs +0 -434
- data/vendor/crates/spikard-http/src/grpc/service.rs +0 -622
- data/vendor/crates/spikard-http/src/grpc/streaming.rs +0 -319
- data/vendor/crates/spikard-http/src/handler_response.rs +0 -901
- data/vendor/crates/spikard-http/src/handler_trait.rs +0 -1015
- data/vendor/crates/spikard-http/src/handler_trait_tests.rs +0 -290
- data/vendor/crates/spikard-http/src/jsonrpc/http_handler.rs +0 -502
- data/vendor/crates/spikard-http/src/jsonrpc/method_registry.rs +0 -648
- data/vendor/crates/spikard-http/src/jsonrpc/mod.rs +0 -58
- data/vendor/crates/spikard-http/src/jsonrpc/protocol.rs +0 -1207
- data/vendor/crates/spikard-http/src/jsonrpc/router.rs +0 -2262
- data/vendor/crates/spikard-http/src/lib.rs +0 -548
- data/vendor/crates/spikard-http/src/lifecycle/adapter.rs +0 -230
- data/vendor/crates/spikard-http/src/lifecycle.rs +0 -1193
- data/vendor/crates/spikard-http/src/middleware/mod.rs +0 -560
- data/vendor/crates/spikard-http/src/middleware/multipart.rs +0 -912
- data/vendor/crates/spikard-http/src/middleware/urlencoded.rs +0 -513
- data/vendor/crates/spikard-http/src/middleware/validation.rs +0 -768
- data/vendor/crates/spikard-http/src/openapi/mod.rs +0 -309
- data/vendor/crates/spikard-http/src/openapi/parameter_extraction.rs +0 -535
- data/vendor/crates/spikard-http/src/openapi/schema_conversion.rs +0 -1363
- data/vendor/crates/spikard-http/src/openapi/spec_generation.rs +0 -667
- data/vendor/crates/spikard-http/src/query_parser.rs +0 -793
- data/vendor/crates/spikard-http/src/response.rs +0 -720
- data/vendor/crates/spikard-http/src/server/fast_router.rs +0 -186
- data/vendor/crates/spikard-http/src/server/grpc_routing.rs +0 -858
- data/vendor/crates/spikard-http/src/server/handler.rs +0 -1661
- data/vendor/crates/spikard-http/src/server/lifecycle_execution.rs +0 -253
- data/vendor/crates/spikard-http/src/server/mod.rs +0 -1649
- data/vendor/crates/spikard-http/src/server/request_extraction.rs +0 -871
- data/vendor/crates/spikard-http/src/server/routing_factory.rs +0 -618
- data/vendor/crates/spikard-http/src/sse.rs +0 -1409
- data/vendor/crates/spikard-http/src/testing/form.rs +0 -52
- data/vendor/crates/spikard-http/src/testing/multipart.rs +0 -64
- data/vendor/crates/spikard-http/src/testing/test_client.rs +0 -787
- data/vendor/crates/spikard-http/src/testing.rs +0 -617
- data/vendor/crates/spikard-http/src/websocket.rs +0 -1477
- data/vendor/crates/spikard-http/tests/auth_integration.rs +0 -645
- data/vendor/crates/spikard-http/tests/background_behavior.rs +0 -832
- data/vendor/crates/spikard-http/tests/common/grpc_helpers.rs +0 -1012
- data/vendor/crates/spikard-http/tests/common/handlers.rs +0 -309
- data/vendor/crates/spikard-http/tests/common/mod.rs +0 -33
- data/vendor/crates/spikard-http/tests/common/test_builders.rs +0 -628
- data/vendor/crates/spikard-http/tests/di_handler_error_responses.rs +0 -162
- data/vendor/crates/spikard-http/tests/di_integration.rs +0 -192
- data/vendor/crates/spikard-http/tests/doc_snippets.rs +0 -5
- data/vendor/crates/spikard-http/tests/grpc_bidirectional_streaming.rs +0 -430
- data/vendor/crates/spikard-http/tests/grpc_client_streaming.rs +0 -738
- data/vendor/crates/spikard-http/tests/grpc_error_handling_test.rs +0 -652
- data/vendor/crates/spikard-http/tests/grpc_integration_test.rs +0 -334
- data/vendor/crates/spikard-http/tests/grpc_metadata_test.rs +0 -532
- data/vendor/crates/spikard-http/tests/grpc_server_integration.rs +0 -495
- data/vendor/crates/spikard-http/tests/grpc_server_streaming.rs +0 -974
- data/vendor/crates/spikard-http/tests/lifecycle_execution.rs +0 -1093
- data/vendor/crates/spikard-http/tests/middleware_stack_integration.rs +0 -389
- data/vendor/crates/spikard-http/tests/multipart_behavior.rs +0 -656
- data/vendor/crates/spikard-http/tests/request_extraction_full.rs +0 -513
- data/vendor/crates/spikard-http/tests/server_auth_middleware_behavior.rs +0 -328
- data/vendor/crates/spikard-http/tests/server_config_builder.rs +0 -314
- data/vendor/crates/spikard-http/tests/server_configured_router_behavior.rs +0 -200
- data/vendor/crates/spikard-http/tests/server_cors_preflight.rs +0 -83
- data/vendor/crates/spikard-http/tests/server_handler_wrappers.rs +0 -464
- data/vendor/crates/spikard-http/tests/server_method_router_additional_behavior.rs +0 -286
- data/vendor/crates/spikard-http/tests/server_method_router_coverage.rs +0 -118
- data/vendor/crates/spikard-http/tests/server_middleware_behavior.rs +0 -99
- data/vendor/crates/spikard-http/tests/server_middleware_branches.rs +0 -204
- data/vendor/crates/spikard-http/tests/server_openapi_jsonrpc_static.rs +0 -421
- data/vendor/crates/spikard-http/tests/server_router_behavior.rs +0 -121
- data/vendor/crates/spikard-http/tests/sse_behavior.rs +0 -620
- data/vendor/crates/spikard-http/tests/sse_full_behavior.rs +0 -584
- data/vendor/crates/spikard-http/tests/sse_handler_behavior.rs +0 -130
- data/vendor/crates/spikard-http/tests/test_client_requests.rs +0 -167
- data/vendor/crates/spikard-http/tests/testing_helpers.rs +0 -87
- data/vendor/crates/spikard-http/tests/testing_module_coverage.rs +0 -155
- data/vendor/crates/spikard-http/tests/urlencoded_content_type.rs +0 -82
- data/vendor/crates/spikard-http/tests/websocket_behavior.rs +0 -663
- data/vendor/crates/spikard-http/tests/websocket_full_behavior.rs +0 -440
- data/vendor/crates/spikard-http/tests/websocket_integration.rs +0 -150
- data/vendor/crates/spikard-rb/Cargo.toml +0 -68
- data/vendor/crates/spikard-rb/build.rs +0 -200
- data/vendor/crates/spikard-rb/src/background.rs +0 -63
- data/vendor/crates/spikard-rb/src/config/mod.rs +0 -5
- data/vendor/crates/spikard-rb/src/config/server_config.rs +0 -401
- data/vendor/crates/spikard-rb/src/conversion.rs +0 -688
- data/vendor/crates/spikard-rb/src/di/builder.rs +0 -100
- data/vendor/crates/spikard-rb/src/di/mod.rs +0 -375
- data/vendor/crates/spikard-rb/src/grpc/handler.rs +0 -834
- data/vendor/crates/spikard-rb/src/grpc/mod.rs +0 -13
- data/vendor/crates/spikard-rb/src/gvl.rs +0 -80
- data/vendor/crates/spikard-rb/src/handler.rs +0 -699
- data/vendor/crates/spikard-rb/src/integration/mod.rs +0 -3
- data/vendor/crates/spikard-rb/src/lib.rs +0 -2264
- data/vendor/crates/spikard-rb/src/lifecycle.rs +0 -303
- data/vendor/crates/spikard-rb/src/metadata/mod.rs +0 -5
- data/vendor/crates/spikard-rb/src/metadata/route_extraction.rs +0 -507
- data/vendor/crates/spikard-rb/src/request.rs +0 -439
- data/vendor/crates/spikard-rb/src/runtime/mod.rs +0 -5
- data/vendor/crates/spikard-rb/src/runtime/server_runner.rs +0 -344
- data/vendor/crates/spikard-rb/src/server.rs +0 -307
- data/vendor/crates/spikard-rb/src/sse.rs +0 -231
- data/vendor/crates/spikard-rb/src/testing/client.rs +0 -698
- data/vendor/crates/spikard-rb/src/testing/mod.rs +0 -7
- data/vendor/crates/spikard-rb/src/testing/sse.rs +0 -108
- data/vendor/crates/spikard-rb/src/testing/websocket.rs +0 -573
- data/vendor/crates/spikard-rb/src/websocket.rs +0 -475
- data/vendor/crates/spikard-rb-macros/Cargo.toml +0 -25
- data/vendor/crates/spikard-rb-macros/src/lib.rs +0 -51
|
@@ -1,699 +0,0 @@
|
|
|
1
|
-
//! Ruby handler wrapper implementing the Handler trait.
|
|
2
|
-
//!
|
|
3
|
-
//! This module provides the `RubyHandler` struct that wraps Ruby Proc objects
|
|
4
|
-
//! and implements Spikard's `Handler` trait for async request processing.
|
|
5
|
-
|
|
6
|
-
#![allow(dead_code)]
|
|
7
|
-
#![deny(clippy::unwrap_used)]
|
|
8
|
-
|
|
9
|
-
use axum::body::Body;
|
|
10
|
-
use axum::http::{HeaderName, HeaderValue, Request, StatusCode};
|
|
11
|
-
use magnus::prelude::*;
|
|
12
|
-
use magnus::value::LazyId;
|
|
13
|
-
use magnus::value::{InnerValue, Opaque};
|
|
14
|
-
use magnus::{Error, RHash, RString, Ruby, TryConvert, Value, gc::Marker};
|
|
15
|
-
use serde_json::Value as JsonValue;
|
|
16
|
-
use spikard_bindings_shared::ErrorResponseBuilder;
|
|
17
|
-
use spikard_core::problem::ProblemDetails;
|
|
18
|
-
use spikard_http::SchemaValidator;
|
|
19
|
-
use spikard_http::{Handler, HandlerResponse, HandlerResult, RequestData};
|
|
20
|
-
use std::collections::HashMap;
|
|
21
|
-
use std::panic::AssertUnwindSafe;
|
|
22
|
-
use std::pin::Pin;
|
|
23
|
-
use std::sync::Arc;
|
|
24
|
-
|
|
25
|
-
use crate::conversion::ruby_value_to_json;
|
|
26
|
-
use crate::gvl::with_gvl;
|
|
27
|
-
use crate::request::NativeRequest;
|
|
28
|
-
|
|
29
|
-
static KEY_PATH_PARAMS: LazyId = LazyId::new("path_params");
|
|
30
|
-
static KEY_QUERY: LazyId = LazyId::new("query");
|
|
31
|
-
static KEY_BODY: LazyId = LazyId::new("body");
|
|
32
|
-
|
|
33
|
-
/// Response payload with status, headers, and body data.
|
|
34
|
-
pub struct HandlerResponsePayload {
|
|
35
|
-
pub status: u16,
|
|
36
|
-
pub headers: HashMap<String, String>,
|
|
37
|
-
pub body: Option<JsonValue>,
|
|
38
|
-
pub raw_body: Option<Vec<u8>>,
|
|
39
|
-
}
|
|
40
|
-
|
|
41
|
-
/// Streaming response variant containing an enumerator and metadata.
|
|
42
|
-
pub struct StreamingResponsePayload {
|
|
43
|
-
pub enumerator: Arc<Opaque<Value>>,
|
|
44
|
-
pub status: u16,
|
|
45
|
-
pub headers: HashMap<String, String>,
|
|
46
|
-
}
|
|
47
|
-
|
|
48
|
-
/// Handler result: either a payload or a streaming response.
|
|
49
|
-
pub enum RubyHandlerResult {
|
|
50
|
-
Payload(HandlerResponsePayload),
|
|
51
|
-
Streaming(StreamingResponsePayload),
|
|
52
|
-
}
|
|
53
|
-
|
|
54
|
-
impl StreamingResponsePayload {
|
|
55
|
-
/// Convert streaming response into a `HandlerResponse`.
|
|
56
|
-
pub fn into_response(self) -> Result<HandlerResponse, Error> {
|
|
57
|
-
let ruby = Ruby::get().map_err(|_| {
|
|
58
|
-
Error::new(
|
|
59
|
-
magnus::exception::runtime_error(),
|
|
60
|
-
"Ruby VM became unavailable during streaming response construction",
|
|
61
|
-
)
|
|
62
|
-
})?;
|
|
63
|
-
|
|
64
|
-
let status = StatusCode::from_u16(self.status).map_err(|err| {
|
|
65
|
-
Error::new(
|
|
66
|
-
ruby.exception_arg_error(),
|
|
67
|
-
format!("Invalid streaming status code {}: {}", self.status, err),
|
|
68
|
-
)
|
|
69
|
-
})?;
|
|
70
|
-
|
|
71
|
-
let header_pairs = self
|
|
72
|
-
.headers
|
|
73
|
-
.into_iter()
|
|
74
|
-
.map(|(name, value)| {
|
|
75
|
-
let header_name = HeaderName::from_bytes(name.as_bytes()).map_err(|err| {
|
|
76
|
-
Error::new(
|
|
77
|
-
ruby.exception_arg_error(),
|
|
78
|
-
format!("Invalid header name '{name}': {err}"),
|
|
79
|
-
)
|
|
80
|
-
})?;
|
|
81
|
-
let header_value = HeaderValue::from_str(&value).map_err(|err| {
|
|
82
|
-
Error::new(
|
|
83
|
-
ruby.exception_arg_error(),
|
|
84
|
-
format!("Invalid header value for '{name}': {err}"),
|
|
85
|
-
)
|
|
86
|
-
})?;
|
|
87
|
-
Ok((header_name, header_value))
|
|
88
|
-
})
|
|
89
|
-
.collect::<Result<Vec<_>, Error>>()?;
|
|
90
|
-
|
|
91
|
-
let enumerator = self.enumerator.clone();
|
|
92
|
-
let body_stream = async_stream::stream! {
|
|
93
|
-
loop {
|
|
94
|
-
match poll_stream_chunk(&enumerator) {
|
|
95
|
-
Ok(Some(bytes)) => yield Ok(bytes),
|
|
96
|
-
Ok(None) => break,
|
|
97
|
-
Err(err) => {
|
|
98
|
-
yield Err(Box::new(err));
|
|
99
|
-
break;
|
|
100
|
-
}
|
|
101
|
-
}
|
|
102
|
-
}
|
|
103
|
-
};
|
|
104
|
-
|
|
105
|
-
let mut response = HandlerResponse::stream(body_stream).with_status(status);
|
|
106
|
-
for (name, value) in header_pairs {
|
|
107
|
-
response = response.with_header(name, value);
|
|
108
|
-
}
|
|
109
|
-
Ok(response)
|
|
110
|
-
}
|
|
111
|
-
}
|
|
112
|
-
|
|
113
|
-
/// Poll a single chunk from a Ruby enumerator.
|
|
114
|
-
fn poll_stream_chunk(enumerator: &Arc<Opaque<Value>>) -> Result<Option<bytes::Bytes>, std::io::Error> {
|
|
115
|
-
let ruby = Ruby::get().map_err(|err| std::io::Error::other(err.to_string()))?;
|
|
116
|
-
let enum_value = enumerator.get_inner_with(&ruby);
|
|
117
|
-
match enum_value.funcall::<_, _, Value>("next", ()) {
|
|
118
|
-
Ok(chunk) => crate::conversion::ruby_value_to_bytes(chunk).map(Some),
|
|
119
|
-
Err(err) => {
|
|
120
|
-
if err.is_kind_of(ruby.exception_stop_iteration()) {
|
|
121
|
-
Ok(None)
|
|
122
|
-
} else {
|
|
123
|
-
Err(std::io::Error::other(err.to_string()))
|
|
124
|
-
}
|
|
125
|
-
}
|
|
126
|
-
}
|
|
127
|
-
}
|
|
128
|
-
|
|
129
|
-
/// Inner state of a Ruby handler.
|
|
130
|
-
pub struct RubyHandlerInner {
|
|
131
|
-
pub handler_proc: Opaque<Value>,
|
|
132
|
-
pub handler_name: String,
|
|
133
|
-
pub method: String,
|
|
134
|
-
pub path: String,
|
|
135
|
-
method_value: Opaque<Value>,
|
|
136
|
-
path_value: Opaque<Value>,
|
|
137
|
-
pub json_module: Opaque<Value>,
|
|
138
|
-
pub response_validator: Option<Arc<SchemaValidator>>,
|
|
139
|
-
pub upload_file_class: Option<Opaque<Value>>,
|
|
140
|
-
}
|
|
141
|
-
|
|
142
|
-
/// Wrapper around a Ruby Proc that implements the Handler trait.
|
|
143
|
-
#[derive(Clone)]
|
|
144
|
-
pub struct RubyHandler {
|
|
145
|
-
pub inner: Arc<RubyHandlerInner>,
|
|
146
|
-
}
|
|
147
|
-
|
|
148
|
-
impl RubyHandler {
|
|
149
|
-
/// Create a new RubyHandler from a route and handler Proc.
|
|
150
|
-
pub fn new(route: &spikard_http::Route, handler_value: Value, json_module: Value) -> Result<Self, Error> {
|
|
151
|
-
let upload_file_class = if route.file_params.is_some() {
|
|
152
|
-
lookup_upload_file_class()?
|
|
153
|
-
} else {
|
|
154
|
-
None
|
|
155
|
-
};
|
|
156
|
-
let method = route.method.as_str().to_string();
|
|
157
|
-
let path = route.path.clone();
|
|
158
|
-
|
|
159
|
-
let Ok(ruby) = Ruby::get() else {
|
|
160
|
-
return Err(Error::new(
|
|
161
|
-
magnus::exception::runtime_error(),
|
|
162
|
-
"Ruby VM unavailable while creating handler",
|
|
163
|
-
));
|
|
164
|
-
};
|
|
165
|
-
let handler_value = crate::conversion::ensure_callable(&ruby, handler_value, &route.handler_name)?;
|
|
166
|
-
let method_value = Opaque::from(ruby.str_new(&method).as_value());
|
|
167
|
-
let path_value = Opaque::from(ruby.str_new(&path).as_value());
|
|
168
|
-
|
|
169
|
-
Ok(Self {
|
|
170
|
-
inner: Arc::new(RubyHandlerInner {
|
|
171
|
-
handler_proc: Opaque::from(handler_value),
|
|
172
|
-
handler_name: route.handler_name.clone(),
|
|
173
|
-
method,
|
|
174
|
-
path,
|
|
175
|
-
method_value,
|
|
176
|
-
path_value,
|
|
177
|
-
json_module: Opaque::from(json_module),
|
|
178
|
-
response_validator: route.response_validator.clone(),
|
|
179
|
-
upload_file_class,
|
|
180
|
-
}),
|
|
181
|
-
})
|
|
182
|
-
}
|
|
183
|
-
|
|
184
|
-
/// Create a new RubyHandler for server mode
|
|
185
|
-
///
|
|
186
|
-
/// This is used by run_server to create handlers from Ruby Procs
|
|
187
|
-
pub fn new_for_server(
|
|
188
|
-
ruby: &Ruby,
|
|
189
|
-
handler_value: Value,
|
|
190
|
-
handler_name: String,
|
|
191
|
-
method: String,
|
|
192
|
-
path: String,
|
|
193
|
-
json_module: Value,
|
|
194
|
-
route: &spikard_http::Route,
|
|
195
|
-
) -> Result<Self, Error> {
|
|
196
|
-
let upload_file_class = if route.file_params.is_some() {
|
|
197
|
-
lookup_upload_file_class()?
|
|
198
|
-
} else {
|
|
199
|
-
None
|
|
200
|
-
};
|
|
201
|
-
let handler_value = crate::conversion::ensure_callable(ruby, handler_value, &handler_name)?;
|
|
202
|
-
let method_value = Opaque::from(ruby.str_new(&method).as_value());
|
|
203
|
-
let path_value = Opaque::from(ruby.str_new(&path).as_value());
|
|
204
|
-
|
|
205
|
-
Ok(Self {
|
|
206
|
-
inner: Arc::new(RubyHandlerInner {
|
|
207
|
-
handler_proc: Opaque::from(handler_value),
|
|
208
|
-
handler_name,
|
|
209
|
-
method,
|
|
210
|
-
path,
|
|
211
|
-
method_value,
|
|
212
|
-
path_value,
|
|
213
|
-
json_module: Opaque::from(json_module),
|
|
214
|
-
response_validator: route.response_validator.clone(),
|
|
215
|
-
upload_file_class,
|
|
216
|
-
}),
|
|
217
|
-
})
|
|
218
|
-
}
|
|
219
|
-
|
|
220
|
-
/// Required by Ruby GC; invoked through the magnus mark hook.
|
|
221
|
-
#[allow(dead_code)]
|
|
222
|
-
pub fn mark(&self, marker: &Marker) {
|
|
223
|
-
if let Ok(ruby) = Ruby::get() {
|
|
224
|
-
let proc_val = self.inner.handler_proc.get_inner_with(&ruby);
|
|
225
|
-
marker.mark(proc_val);
|
|
226
|
-
marker.mark(self.inner.method_value.get_inner_with(&ruby));
|
|
227
|
-
marker.mark(self.inner.path_value.get_inner_with(&ruby));
|
|
228
|
-
}
|
|
229
|
-
}
|
|
230
|
-
|
|
231
|
-
/// Handle a request synchronously.
|
|
232
|
-
pub fn handle(&self, request_data: RequestData) -> HandlerResult {
|
|
233
|
-
with_gvl(|| {
|
|
234
|
-
let result = std::panic::catch_unwind(AssertUnwindSafe(|| self.handle_inner(request_data)));
|
|
235
|
-
match result {
|
|
236
|
-
Ok(res) => res,
|
|
237
|
-
Err(_) => Err(ErrorResponseBuilder::structured_error(
|
|
238
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
239
|
-
"panic",
|
|
240
|
-
"Unexpected panic while executing Ruby handler",
|
|
241
|
-
)),
|
|
242
|
-
}
|
|
243
|
-
})
|
|
244
|
-
}
|
|
245
|
-
|
|
246
|
-
fn handle_inner(&self, mut request_data: RequestData) -> HandlerResult {
|
|
247
|
-
let ruby = Ruby::get().map_err(|_| {
|
|
248
|
-
ErrorResponseBuilder::structured_error(
|
|
249
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
250
|
-
"ruby_vm_unavailable",
|
|
251
|
-
"Ruby VM unavailable while invoking handler",
|
|
252
|
-
)
|
|
253
|
-
})?;
|
|
254
|
-
|
|
255
|
-
// Extract validated_params with Arc::try_unwrap to eliminate clone if possible.
|
|
256
|
-
let validated_params = request_data
|
|
257
|
-
.validated_params
|
|
258
|
-
.take()
|
|
259
|
-
.map(|arc| Arc::try_unwrap(arc).unwrap_or_else(|a| (*a).clone()));
|
|
260
|
-
|
|
261
|
-
// Use NativeRequest for lazy field conversion — fields are only converted
|
|
262
|
-
// to Ruby objects when accessed, avoiding work for unused fields.
|
|
263
|
-
let native_request =
|
|
264
|
-
NativeRequest::from_request_data(request_data, validated_params, self.inner.upload_file_class);
|
|
265
|
-
let request_value = ruby.obj_wrap(native_request).as_value();
|
|
266
|
-
|
|
267
|
-
let handler_proc = self.inner.handler_proc.get_inner_with(&ruby);
|
|
268
|
-
let response_value = match call_handler_proc(&ruby, handler_proc, request_value) {
|
|
269
|
-
Ok(value) => value,
|
|
270
|
-
Err(err) => return Err(problem_from_ruby_error(&ruby, &self.inner, err)),
|
|
271
|
-
};
|
|
272
|
-
|
|
273
|
-
let handler_result = interpret_handler_response(&ruby, &self.inner, response_value).map_err(|err| {
|
|
274
|
-
ErrorResponseBuilder::structured_error(
|
|
275
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
276
|
-
"response_interpret_error",
|
|
277
|
-
format!(
|
|
278
|
-
"Failed to interpret response from '{}': {}",
|
|
279
|
-
self.inner.handler_name, err
|
|
280
|
-
),
|
|
281
|
-
)
|
|
282
|
-
})?;
|
|
283
|
-
|
|
284
|
-
let payload = match handler_result {
|
|
285
|
-
RubyHandlerResult::Streaming(streaming) => {
|
|
286
|
-
let response = streaming.into_response().map_err(|err| {
|
|
287
|
-
ErrorResponseBuilder::structured_error(
|
|
288
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
289
|
-
"streaming_response_error",
|
|
290
|
-
format!("Failed to build streaming response: {}", err),
|
|
291
|
-
)
|
|
292
|
-
})?;
|
|
293
|
-
return Ok(response.into_response());
|
|
294
|
-
}
|
|
295
|
-
RubyHandlerResult::Payload(payload) => payload,
|
|
296
|
-
};
|
|
297
|
-
|
|
298
|
-
if let Some(validator) = &self.inner.response_validator {
|
|
299
|
-
let candidate_body = match payload.body.clone() {
|
|
300
|
-
Some(body) => Some(body),
|
|
301
|
-
None => match try_parse_raw_body(payload.raw_body.as_ref()) {
|
|
302
|
-
Ok(parsed) => parsed,
|
|
303
|
-
Err(err) => {
|
|
304
|
-
return Err(ErrorResponseBuilder::structured_error(
|
|
305
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
306
|
-
"response_body_decode_error",
|
|
307
|
-
err,
|
|
308
|
-
));
|
|
309
|
-
}
|
|
310
|
-
},
|
|
311
|
-
};
|
|
312
|
-
|
|
313
|
-
match candidate_body {
|
|
314
|
-
Some(json_body) => {
|
|
315
|
-
if let Err(errors) = validator.validate(&json_body) {
|
|
316
|
-
let problem = ProblemDetails::from_validation_error(&errors);
|
|
317
|
-
return Err(ErrorResponseBuilder::problem_details_response(&problem));
|
|
318
|
-
}
|
|
319
|
-
}
|
|
320
|
-
None => {
|
|
321
|
-
return Err(ErrorResponseBuilder::structured_error(
|
|
322
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
323
|
-
"response_validation_failed",
|
|
324
|
-
"Response validator requires JSON body but handler returned raw bytes",
|
|
325
|
-
));
|
|
326
|
-
}
|
|
327
|
-
}
|
|
328
|
-
}
|
|
329
|
-
|
|
330
|
-
let HandlerResponsePayload {
|
|
331
|
-
status,
|
|
332
|
-
headers,
|
|
333
|
-
body,
|
|
334
|
-
raw_body,
|
|
335
|
-
} = payload;
|
|
336
|
-
|
|
337
|
-
let mut response_builder = axum::http::Response::builder().status(status);
|
|
338
|
-
let mut has_content_type = false;
|
|
339
|
-
|
|
340
|
-
for (name, value) in headers.iter() {
|
|
341
|
-
if name.eq_ignore_ascii_case("content-type") {
|
|
342
|
-
has_content_type = true;
|
|
343
|
-
}
|
|
344
|
-
let header_name = HeaderName::from_bytes(name.as_bytes()).map_err(|err| {
|
|
345
|
-
(
|
|
346
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
347
|
-
format!("Invalid header name '{name}': {err}"),
|
|
348
|
-
)
|
|
349
|
-
})?;
|
|
350
|
-
let header_value = HeaderValue::from_str(value).map_err(|err| {
|
|
351
|
-
(
|
|
352
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
353
|
-
format!("Invalid header value for '{name}': {err}"),
|
|
354
|
-
)
|
|
355
|
-
})?;
|
|
356
|
-
|
|
357
|
-
response_builder = response_builder.header(header_name, header_value);
|
|
358
|
-
}
|
|
359
|
-
|
|
360
|
-
if !has_content_type && body.is_some() {
|
|
361
|
-
response_builder = response_builder.header(
|
|
362
|
-
HeaderName::from_static("content-type"),
|
|
363
|
-
HeaderValue::from_static("application/json"),
|
|
364
|
-
);
|
|
365
|
-
}
|
|
366
|
-
|
|
367
|
-
let body_bytes = if let Some(raw) = raw_body {
|
|
368
|
-
raw
|
|
369
|
-
} else if let Some(json_value) = body {
|
|
370
|
-
serde_json::to_vec(&json_value).map_err(|err| {
|
|
371
|
-
(
|
|
372
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
373
|
-
format!("Failed to serialise response body: {err}"),
|
|
374
|
-
)
|
|
375
|
-
})?
|
|
376
|
-
} else {
|
|
377
|
-
Vec::new()
|
|
378
|
-
};
|
|
379
|
-
|
|
380
|
-
response_builder.body(Body::from(body_bytes)).map_err(|err| {
|
|
381
|
-
(
|
|
382
|
-
StatusCode::INTERNAL_SERVER_ERROR,
|
|
383
|
-
format!("Failed to build response: {err}"),
|
|
384
|
-
)
|
|
385
|
-
})
|
|
386
|
-
}
|
|
387
|
-
}
|
|
388
|
-
|
|
389
|
-
impl Handler for RubyHandler {
|
|
390
|
-
fn call(
|
|
391
|
-
&self,
|
|
392
|
-
_req: Request<Body>,
|
|
393
|
-
request_data: RequestData,
|
|
394
|
-
) -> Pin<Box<dyn std::future::Future<Output = HandlerResult> + Send + '_>> {
|
|
395
|
-
let handler = self.clone();
|
|
396
|
-
Box::pin(async move { handler.handle(request_data) })
|
|
397
|
-
}
|
|
398
|
-
}
|
|
399
|
-
|
|
400
|
-
fn try_parse_raw_body(raw_body: Option<&Vec<u8>>) -> Result<Option<JsonValue>, String> {
|
|
401
|
-
let Some(bytes) = raw_body else {
|
|
402
|
-
return Ok(None);
|
|
403
|
-
};
|
|
404
|
-
if bytes.is_empty() {
|
|
405
|
-
return Ok(None);
|
|
406
|
-
}
|
|
407
|
-
// PERFORMANCE: Use from_slice directly to avoid String allocation.
|
|
408
|
-
// serde_json handles UTF-8 validation internally.
|
|
409
|
-
serde_json::from_slice(bytes)
|
|
410
|
-
.map(Some)
|
|
411
|
-
.map_err(|e| format!("Failed to parse response body as JSON: {e}"))
|
|
412
|
-
}
|
|
413
|
-
|
|
414
|
-
fn lookup_upload_file_class() -> Result<Option<Opaque<Value>>, Error> {
|
|
415
|
-
let ruby = match Ruby::get() {
|
|
416
|
-
Ok(ruby) => ruby,
|
|
417
|
-
Err(_) => return Ok(None),
|
|
418
|
-
};
|
|
419
|
-
|
|
420
|
-
let upload_file = ruby.eval::<Value>("Spikard::UploadFile").ok();
|
|
421
|
-
Ok(upload_file.map(Opaque::from))
|
|
422
|
-
}
|
|
423
|
-
|
|
424
|
-
fn call_handler_proc(ruby: &Ruby, handler_proc: Value, request_value: Value) -> Result<Value, Error> {
|
|
425
|
-
let arity: i64 = handler_proc.funcall("arity", ())?;
|
|
426
|
-
if arity == 0 {
|
|
427
|
-
return handler_proc.funcall("call", ());
|
|
428
|
-
}
|
|
429
|
-
|
|
430
|
-
if arity == 1 {
|
|
431
|
-
return handler_proc.funcall("call", (request_value,));
|
|
432
|
-
}
|
|
433
|
-
|
|
434
|
-
let (params_value, query_value, body_value) = if let Ok(request) = <&NativeRequest>::try_convert(request_value) {
|
|
435
|
-
(
|
|
436
|
-
NativeRequest::path_params(ruby, request).unwrap_or_else(|_| ruby.qnil().as_value()),
|
|
437
|
-
NativeRequest::query(ruby, request).unwrap_or_else(|_| ruby.qnil().as_value()),
|
|
438
|
-
NativeRequest::body(ruby, request).unwrap_or_else(|_| ruby.qnil().as_value()),
|
|
439
|
-
)
|
|
440
|
-
} else if let Some(hash) = RHash::from_value(request_value) {
|
|
441
|
-
(
|
|
442
|
-
hash.get(*KEY_PATH_PARAMS).unwrap_or_else(|| ruby.qnil().as_value()),
|
|
443
|
-
hash.get(*KEY_QUERY).unwrap_or_else(|| ruby.qnil().as_value()),
|
|
444
|
-
hash.get(*KEY_BODY).unwrap_or_else(|| ruby.qnil().as_value()),
|
|
445
|
-
)
|
|
446
|
-
} else {
|
|
447
|
-
(ruby.qnil().as_value(), ruby.qnil().as_value(), ruby.qnil().as_value())
|
|
448
|
-
};
|
|
449
|
-
|
|
450
|
-
if arity == 2 {
|
|
451
|
-
return handler_proc.funcall("call", (params_value, query_value));
|
|
452
|
-
}
|
|
453
|
-
|
|
454
|
-
handler_proc.funcall("call", (params_value, query_value, body_value))
|
|
455
|
-
}
|
|
456
|
-
|
|
457
|
-
fn problem_from_ruby_error(ruby: &Ruby, handler: &RubyHandlerInner, err: Error) -> (StatusCode, String) {
|
|
458
|
-
let mut status = StatusCode::INTERNAL_SERVER_ERROR;
|
|
459
|
-
let mut extensions: HashMap<String, JsonValue> = HashMap::new();
|
|
460
|
-
let mut detail = ruby_error_message(ruby, &err);
|
|
461
|
-
|
|
462
|
-
if err.is_kind_of(ruby.exception_arg_error()) {
|
|
463
|
-
status = StatusCode::BAD_REQUEST;
|
|
464
|
-
}
|
|
465
|
-
|
|
466
|
-
if let Some(exception) = err.value() {
|
|
467
|
-
if matches!(exception.respond_to("status", false), Ok(true)) {
|
|
468
|
-
if let Ok(code) = exception.funcall::<_, _, i64>("status", ()) {
|
|
469
|
-
status = StatusCode::from_u16(code as u16).unwrap_or(StatusCode::INTERNAL_SERVER_ERROR);
|
|
470
|
-
}
|
|
471
|
-
} else if matches!(exception.respond_to("status_code", false), Ok(true))
|
|
472
|
-
&& let Ok(code) = exception.funcall::<_, _, i64>("status_code", ())
|
|
473
|
-
{
|
|
474
|
-
status = StatusCode::from_u16(code as u16).unwrap_or(StatusCode::INTERNAL_SERVER_ERROR);
|
|
475
|
-
}
|
|
476
|
-
|
|
477
|
-
let json_module = handler.json_module.get_inner_with(ruby);
|
|
478
|
-
if matches!(exception.respond_to("code", false), Ok(true))
|
|
479
|
-
&& let Ok(value) = exception.funcall::<_, _, Value>("code", ())
|
|
480
|
-
&& let Ok(json_value) = ruby_value_to_json(ruby, json_module, value)
|
|
481
|
-
{
|
|
482
|
-
extensions.insert("code".to_string(), json_value);
|
|
483
|
-
}
|
|
484
|
-
|
|
485
|
-
if matches!(exception.respond_to("details", false), Ok(true))
|
|
486
|
-
&& let Ok(value) = exception.funcall::<_, _, Value>("details", ())
|
|
487
|
-
&& let Ok(json_value) = ruby_value_to_json(ruby, json_module, value)
|
|
488
|
-
{
|
|
489
|
-
extensions.insert("details".to_string(), json_value);
|
|
490
|
-
}
|
|
491
|
-
}
|
|
492
|
-
|
|
493
|
-
detail = sanitize_error_detail(&detail);
|
|
494
|
-
|
|
495
|
-
let mut problem = problem_for_status(status, detail);
|
|
496
|
-
for (key, value) in extensions {
|
|
497
|
-
problem = problem.with_extension(key, value);
|
|
498
|
-
}
|
|
499
|
-
|
|
500
|
-
ErrorResponseBuilder::problem_details_response(&problem)
|
|
501
|
-
}
|
|
502
|
-
|
|
503
|
-
fn ruby_error_message(_ruby: &Ruby, err: &Error) -> String {
|
|
504
|
-
if let Some(exception) = err.value()
|
|
505
|
-
&& matches!(exception.respond_to("message", false), Ok(true))
|
|
506
|
-
&& let Ok(message) = exception.funcall::<_, _, String>("message", ())
|
|
507
|
-
{
|
|
508
|
-
return message;
|
|
509
|
-
}
|
|
510
|
-
err.to_string()
|
|
511
|
-
}
|
|
512
|
-
|
|
513
|
-
fn problem_for_status(status: StatusCode, detail: String) -> ProblemDetails {
|
|
514
|
-
match status {
|
|
515
|
-
StatusCode::BAD_REQUEST => ProblemDetails::bad_request(detail),
|
|
516
|
-
StatusCode::UNAUTHORIZED => {
|
|
517
|
-
ProblemDetails::new("https://spikard.dev/errors/unauthorized", "Unauthorized", status).with_detail(detail)
|
|
518
|
-
}
|
|
519
|
-
StatusCode::FORBIDDEN => {
|
|
520
|
-
ProblemDetails::new("https://spikard.dev/errors/forbidden", "Forbidden", status).with_detail(detail)
|
|
521
|
-
}
|
|
522
|
-
StatusCode::NOT_FOUND => ProblemDetails::not_found(detail),
|
|
523
|
-
StatusCode::UNPROCESSABLE_ENTITY => ProblemDetails::new(
|
|
524
|
-
ProblemDetails::TYPE_VALIDATION_ERROR,
|
|
525
|
-
"Request Validation Failed",
|
|
526
|
-
status,
|
|
527
|
-
)
|
|
528
|
-
.with_detail(detail),
|
|
529
|
-
_ => ProblemDetails::internal_server_error(detail),
|
|
530
|
-
}
|
|
531
|
-
}
|
|
532
|
-
|
|
533
|
-
fn sanitize_error_detail(detail: &str) -> String {
|
|
534
|
-
let mut tokens = Vec::new();
|
|
535
|
-
let mut redact_next = false;
|
|
536
|
-
|
|
537
|
-
for token in detail.split_whitespace() {
|
|
538
|
-
let lower = token.to_lowercase();
|
|
539
|
-
if token.starts_with('/') || token.contains(".rb:") {
|
|
540
|
-
tokens.push("[redacted]".to_string());
|
|
541
|
-
redact_next = false;
|
|
542
|
-
continue;
|
|
543
|
-
}
|
|
544
|
-
|
|
545
|
-
if lower.starts_with("password=") {
|
|
546
|
-
tokens.push("password=[redacted]".to_string());
|
|
547
|
-
redact_next = false;
|
|
548
|
-
continue;
|
|
549
|
-
}
|
|
550
|
-
|
|
551
|
-
if lower.starts_with("host=") {
|
|
552
|
-
tokens.push("host=[redacted]".to_string());
|
|
553
|
-
redact_next = false;
|
|
554
|
-
continue;
|
|
555
|
-
}
|
|
556
|
-
|
|
557
|
-
if lower.starts_with("token=") || lower.starts_with("secret=") {
|
|
558
|
-
tokens.push("[redacted]".to_string());
|
|
559
|
-
redact_next = false;
|
|
560
|
-
continue;
|
|
561
|
-
}
|
|
562
|
-
|
|
563
|
-
if redact_next {
|
|
564
|
-
tokens.push("[redacted]".to_string());
|
|
565
|
-
redact_next = false;
|
|
566
|
-
continue;
|
|
567
|
-
}
|
|
568
|
-
|
|
569
|
-
if token.eq_ignore_ascii_case("in") {
|
|
570
|
-
tokens.push(token.to_string());
|
|
571
|
-
redact_next = true;
|
|
572
|
-
continue;
|
|
573
|
-
}
|
|
574
|
-
|
|
575
|
-
tokens.push(token.to_string());
|
|
576
|
-
}
|
|
577
|
-
|
|
578
|
-
let mut sanitized = tokens.join(" ");
|
|
579
|
-
sanitized = sanitized.replace("SELECT *", "[redacted]");
|
|
580
|
-
sanitized = sanitized.replace("select *", "[redacted]");
|
|
581
|
-
sanitized = sanitized.replace("FROM users", "[redacted]");
|
|
582
|
-
sanitized = sanitized.replace("from users", "[redacted]");
|
|
583
|
-
sanitized
|
|
584
|
-
}
|
|
585
|
-
|
|
586
|
-
/// Interpret a Ruby handler response into our response types.
|
|
587
|
-
fn interpret_handler_response(
|
|
588
|
-
ruby: &Ruby,
|
|
589
|
-
handler: &RubyHandlerInner,
|
|
590
|
-
value: Value,
|
|
591
|
-
) -> Result<RubyHandlerResult, Error> {
|
|
592
|
-
if value.is_nil() {
|
|
593
|
-
return Ok(RubyHandlerResult::Payload(HandlerResponsePayload {
|
|
594
|
-
status: 200,
|
|
595
|
-
headers: HashMap::new(),
|
|
596
|
-
body: None,
|
|
597
|
-
raw_body: None,
|
|
598
|
-
}));
|
|
599
|
-
}
|
|
600
|
-
|
|
601
|
-
if is_streaming_response(ruby, value)? {
|
|
602
|
-
let stream_value: Value = value.funcall("stream", ())?;
|
|
603
|
-
let status: i64 = value.funcall("status_code", ())?;
|
|
604
|
-
let headers_value: Value = value.funcall("headers", ())?;
|
|
605
|
-
|
|
606
|
-
let status_u16 = u16::try_from(status).map_err(|_| {
|
|
607
|
-
Error::new(
|
|
608
|
-
ruby.exception_arg_error(),
|
|
609
|
-
"StreamingResponse status_code must be between 0 and 65535",
|
|
610
|
-
)
|
|
611
|
-
})?;
|
|
612
|
-
|
|
613
|
-
let headers = value_to_string_map(ruby, headers_value)?;
|
|
614
|
-
|
|
615
|
-
return Ok(RubyHandlerResult::Streaming(StreamingResponsePayload {
|
|
616
|
-
enumerator: Arc::new(Opaque::from(stream_value)),
|
|
617
|
-
status: status_u16,
|
|
618
|
-
headers,
|
|
619
|
-
}));
|
|
620
|
-
}
|
|
621
|
-
|
|
622
|
-
let status_symbol = ruby.intern("status_code");
|
|
623
|
-
if value.respond_to(status_symbol, false)? {
|
|
624
|
-
let status: i64 = value.funcall("status_code", ())?;
|
|
625
|
-
let status_u16 = u16::try_from(status)
|
|
626
|
-
.map_err(|_| Error::new(ruby.exception_arg_error(), "status_code must be between 0 and 65535"))?;
|
|
627
|
-
|
|
628
|
-
let headers_value: Value = value.funcall("headers", ())?;
|
|
629
|
-
let headers = if headers_value.is_nil() {
|
|
630
|
-
HashMap::new()
|
|
631
|
-
} else {
|
|
632
|
-
let hash = RHash::try_convert(headers_value)?;
|
|
633
|
-
hash.to_hash_map::<String, String>()?
|
|
634
|
-
};
|
|
635
|
-
|
|
636
|
-
let content_value: Value = value.funcall("content", ())?;
|
|
637
|
-
let mut raw_body = None;
|
|
638
|
-
let body = if content_value.is_nil() {
|
|
639
|
-
None
|
|
640
|
-
} else if let Ok(str_value) = RString::try_convert(content_value) {
|
|
641
|
-
let slice = unsafe { str_value.as_slice() };
|
|
642
|
-
raw_body = Some(slice.to_vec());
|
|
643
|
-
None
|
|
644
|
-
} else {
|
|
645
|
-
Some(ruby_value_to_json(
|
|
646
|
-
ruby,
|
|
647
|
-
handler.json_module.get_inner_with(ruby),
|
|
648
|
-
content_value,
|
|
649
|
-
)?)
|
|
650
|
-
};
|
|
651
|
-
|
|
652
|
-
return Ok(RubyHandlerResult::Payload(HandlerResponsePayload {
|
|
653
|
-
status: status_u16,
|
|
654
|
-
headers,
|
|
655
|
-
body,
|
|
656
|
-
raw_body,
|
|
657
|
-
}));
|
|
658
|
-
}
|
|
659
|
-
|
|
660
|
-
if let Ok(str_value) = RString::try_convert(value) {
|
|
661
|
-
let slice = unsafe { str_value.as_slice() };
|
|
662
|
-
return Ok(RubyHandlerResult::Payload(HandlerResponsePayload {
|
|
663
|
-
status: 200,
|
|
664
|
-
headers: HashMap::new(),
|
|
665
|
-
body: None,
|
|
666
|
-
raw_body: Some(slice.to_vec()),
|
|
667
|
-
}));
|
|
668
|
-
}
|
|
669
|
-
|
|
670
|
-
let body_json = ruby_value_to_json(ruby, handler.json_module.get_inner_with(ruby), value)?;
|
|
671
|
-
|
|
672
|
-
Ok(RubyHandlerResult::Payload(HandlerResponsePayload {
|
|
673
|
-
status: 200,
|
|
674
|
-
headers: HashMap::new(),
|
|
675
|
-
body: Some(body_json),
|
|
676
|
-
raw_body: None,
|
|
677
|
-
}))
|
|
678
|
-
}
|
|
679
|
-
|
|
680
|
-
/// Convert a Ruby value to a string HashMap.
|
|
681
|
-
fn value_to_string_map(ruby: &Ruby, value: Value) -> Result<HashMap<String, String>, Error> {
|
|
682
|
-
if value.is_nil() {
|
|
683
|
-
return Ok(HashMap::new());
|
|
684
|
-
}
|
|
685
|
-
let hash = RHash::try_convert(value)?;
|
|
686
|
-
hash.to_hash_map::<String, String>().map_err(|err| {
|
|
687
|
-
Error::new(
|
|
688
|
-
ruby.exception_arg_error(),
|
|
689
|
-
format!("Expected headers hash of strings: {}", err),
|
|
690
|
-
)
|
|
691
|
-
})
|
|
692
|
-
}
|
|
693
|
-
|
|
694
|
-
/// Check if a Ruby value is a streaming response.
|
|
695
|
-
fn is_streaming_response(ruby: &Ruby, value: Value) -> Result<bool, Error> {
|
|
696
|
-
let stream_sym = ruby.intern("stream");
|
|
697
|
-
let status_sym = ruby.intern("status_code");
|
|
698
|
-
Ok(value.respond_to(stream_sym, false)? && value.respond_to(status_sym, false)?)
|
|
699
|
-
}
|