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,594 +0,0 @@
|
|
|
1
|
-
//! Shared lifecycle hook executor infrastructure
|
|
2
|
-
//!
|
|
3
|
-
//! This module provides a language-agnostic abstraction for executing lifecycle hooks.
|
|
4
|
-
//! It extracts ~960 lines of duplicated logic from Python, Node.js, Ruby, and PHP bindings
|
|
5
|
-
//! into reusable trait-based components.
|
|
6
|
-
//!
|
|
7
|
-
//! # Design
|
|
8
|
-
//!
|
|
9
|
-
//! The executor uses a trait-based pattern where language bindings implement
|
|
10
|
-
//! `LanguageLifecycleHook` to provide language-specific hook invocation, while
|
|
11
|
-
//! `LifecycleExecutor` handles the common logic:
|
|
12
|
-
//!
|
|
13
|
-
//! - Hook result type handling (Continue vs `ShortCircuit`)
|
|
14
|
-
//! - Response/Request building from hook results
|
|
15
|
-
//! - Error handling and conversion
|
|
16
|
-
//!
|
|
17
|
-
//! # Example
|
|
18
|
-
//!
|
|
19
|
-
//! ```ignore
|
|
20
|
-
//! struct MyLanguageHook { ... }
|
|
21
|
-
//!
|
|
22
|
-
//! impl LanguageLifecycleHook for MyLanguageHook {
|
|
23
|
-
//! fn prepare_hook_data(&self, req: &Request<Body>) -> Result<Self::HookData, String> {
|
|
24
|
-
//! // Convert Request<Body> to language-specific representation
|
|
25
|
-
//! }
|
|
26
|
-
//!
|
|
27
|
-
//! async fn invoke_hook(&self, data: Self::HookData) -> Result<HookResultData, String> {
|
|
28
|
-
//! // Call language function and return structured result
|
|
29
|
-
//! }
|
|
30
|
-
//! }
|
|
31
|
-
//! ```
|
|
32
|
-
|
|
33
|
-
use axum::{
|
|
34
|
-
body::Body,
|
|
35
|
-
http::{Request, Response, StatusCode},
|
|
36
|
-
};
|
|
37
|
-
use std::collections::HashMap;
|
|
38
|
-
use std::future::Future;
|
|
39
|
-
use std::pin::Pin;
|
|
40
|
-
use std::sync::Arc;
|
|
41
|
-
|
|
42
|
-
/// Data returned from language-specific hook invocation
|
|
43
|
-
///
|
|
44
|
-
/// This is a normalized representation of hook results that abstracts away
|
|
45
|
-
/// language-specific details. Bindings convert their native hook results
|
|
46
|
-
/// into this common format.
|
|
47
|
-
#[derive(Debug, Clone)]
|
|
48
|
-
pub struct HookResultData {
|
|
49
|
-
/// Whether to continue execution (true) or short-circuit (false)
|
|
50
|
-
pub continue_execution: bool,
|
|
51
|
-
/// Optional status code for short-circuit responses
|
|
52
|
-
pub status_code: Option<u16>,
|
|
53
|
-
/// Optional headers to include in response
|
|
54
|
-
pub headers: Option<HashMap<String, String>>,
|
|
55
|
-
/// Optional body bytes for response
|
|
56
|
-
pub body: Option<Vec<u8>>,
|
|
57
|
-
/// Optional request modifications (method, path, headers, body)
|
|
58
|
-
pub request_modifications: Option<RequestModifications>,
|
|
59
|
-
}
|
|
60
|
-
|
|
61
|
-
/// Modifications to apply to a request
|
|
62
|
-
#[derive(Debug, Clone)]
|
|
63
|
-
pub struct RequestModifications {
|
|
64
|
-
/// New HTTP method (e.g., "GET", "POST")
|
|
65
|
-
pub method: Option<String>,
|
|
66
|
-
/// New request path
|
|
67
|
-
pub path: Option<String>,
|
|
68
|
-
/// New or updated headers
|
|
69
|
-
pub headers: Option<HashMap<String, String>>,
|
|
70
|
-
/// New request body
|
|
71
|
-
pub body: Option<Vec<u8>>,
|
|
72
|
-
}
|
|
73
|
-
|
|
74
|
-
impl HookResultData {
|
|
75
|
-
/// Create a Continue result (pass through)
|
|
76
|
-
#[must_use]
|
|
77
|
-
pub const fn continue_execution() -> Self {
|
|
78
|
-
Self {
|
|
79
|
-
continue_execution: true,
|
|
80
|
-
status_code: None,
|
|
81
|
-
headers: None,
|
|
82
|
-
body: None,
|
|
83
|
-
request_modifications: None,
|
|
84
|
-
}
|
|
85
|
-
}
|
|
86
|
-
|
|
87
|
-
/// Create a short-circuit response result
|
|
88
|
-
#[must_use]
|
|
89
|
-
pub const fn short_circuit(status_code: u16, body: Vec<u8>, headers: Option<HashMap<String, String>>) -> Self {
|
|
90
|
-
Self {
|
|
91
|
-
continue_execution: false,
|
|
92
|
-
status_code: Some(status_code),
|
|
93
|
-
headers,
|
|
94
|
-
body: Some(body),
|
|
95
|
-
request_modifications: None,
|
|
96
|
-
}
|
|
97
|
-
}
|
|
98
|
-
|
|
99
|
-
/// Create a request modification result
|
|
100
|
-
#[must_use]
|
|
101
|
-
pub const fn modify_request(modifications: RequestModifications) -> Self {
|
|
102
|
-
Self {
|
|
103
|
-
continue_execution: true,
|
|
104
|
-
status_code: None,
|
|
105
|
-
headers: None,
|
|
106
|
-
body: None,
|
|
107
|
-
request_modifications: Some(modifications),
|
|
108
|
-
}
|
|
109
|
-
}
|
|
110
|
-
}
|
|
111
|
-
|
|
112
|
-
/// Trait for language-specific lifecycle hook implementations
|
|
113
|
-
///
|
|
114
|
-
/// Each language binding implements this trait to provide language-specific
|
|
115
|
-
/// hook invocation while delegating common logic to `LifecycleExecutor`.
|
|
116
|
-
pub trait LanguageLifecycleHook: Send + Sync {
|
|
117
|
-
/// Language-specific hook data type
|
|
118
|
-
type HookData: Send;
|
|
119
|
-
|
|
120
|
-
/// Prepare hook data from the incoming request/response
|
|
121
|
-
///
|
|
122
|
-
/// This should convert axum HTTP types to language-specific representations.
|
|
123
|
-
///
|
|
124
|
-
/// # Errors
|
|
125
|
-
///
|
|
126
|
-
/// Returns an error if hook data preparation fails.
|
|
127
|
-
fn prepare_hook_data(&self, req: &Request<Body>) -> Result<Self::HookData, String>;
|
|
128
|
-
|
|
129
|
-
/// Invoke the language hook and return normalized result data
|
|
130
|
-
///
|
|
131
|
-
/// This should call the language function and convert its result to `HookResultData`.
|
|
132
|
-
fn invoke_hook(&self, data: Self::HookData)
|
|
133
|
-
-> Pin<Box<dyn Future<Output = Result<HookResultData, String>> + Send>>;
|
|
134
|
-
}
|
|
135
|
-
|
|
136
|
-
/// Executor that handles common lifecycle hook logic
|
|
137
|
-
///
|
|
138
|
-
/// This executor is generic over any language binding that implements
|
|
139
|
-
/// `LanguageLifecycleHook`. It provides common logic for:
|
|
140
|
-
/// - Executing request hooks and handling short-circuits
|
|
141
|
-
/// - Executing response hooks and building modified responses
|
|
142
|
-
/// - Converting hook results to axum Request/Response types
|
|
143
|
-
pub struct LifecycleExecutor<L: LanguageLifecycleHook> {
|
|
144
|
-
hook: Arc<L>,
|
|
145
|
-
}
|
|
146
|
-
|
|
147
|
-
impl<L: LanguageLifecycleHook> LifecycleExecutor<L> {
|
|
148
|
-
/// Create a new executor for the given hook
|
|
149
|
-
pub const fn new(hook: Arc<L>) -> Self {
|
|
150
|
-
Self { hook }
|
|
151
|
-
}
|
|
152
|
-
|
|
153
|
-
/// Execute a request hook, handling Continue/ShortCircuit semantics
|
|
154
|
-
///
|
|
155
|
-
/// Returns either the modified request or a short-circuit response.
|
|
156
|
-
///
|
|
157
|
-
/// # Errors
|
|
158
|
-
///
|
|
159
|
-
/// Returns an error if hook execution or modification fails.
|
|
160
|
-
pub async fn execute_request_hook(
|
|
161
|
-
&self,
|
|
162
|
-
req: Request<Body>,
|
|
163
|
-
) -> Result<Result<Request<Body>, Response<Body>>, String> {
|
|
164
|
-
let hook_data = self.hook.prepare_hook_data(&req)?;
|
|
165
|
-
let result = self.hook.invoke_hook(hook_data).await?;
|
|
166
|
-
|
|
167
|
-
if !result.continue_execution {
|
|
168
|
-
let response = Self::build_response_from_hook_result(&result)?;
|
|
169
|
-
return Ok(Err(response));
|
|
170
|
-
}
|
|
171
|
-
|
|
172
|
-
if let Some(modifications) = result.request_modifications {
|
|
173
|
-
let modified_req = Self::apply_request_modifications(req, modifications)?;
|
|
174
|
-
Ok(Ok(modified_req))
|
|
175
|
-
} else {
|
|
176
|
-
Ok(Ok(req))
|
|
177
|
-
}
|
|
178
|
-
}
|
|
179
|
-
|
|
180
|
-
/// Execute a response hook, handling response modification
|
|
181
|
-
///
|
|
182
|
-
/// Response hooks can only continue or modify the response,
|
|
183
|
-
/// never short-circuit.
|
|
184
|
-
/// Execute the lifecycle hook on an outgoing response
|
|
185
|
-
///
|
|
186
|
-
/// # Errors
|
|
187
|
-
///
|
|
188
|
-
/// Returns an error if hook execution or response building fails.
|
|
189
|
-
pub async fn execute_response_hook(&self, resp: Response<Body>) -> Result<Response<Body>, String> {
|
|
190
|
-
let (parts, body) = resp.into_parts();
|
|
191
|
-
let body_bytes = extract_body(body).await?;
|
|
192
|
-
|
|
193
|
-
let dummy_req = Request::builder()
|
|
194
|
-
.method("GET")
|
|
195
|
-
.uri("/")
|
|
196
|
-
.body(Body::empty())
|
|
197
|
-
.map_err(|e| format!("Failed to build dummy request: {e}"))?;
|
|
198
|
-
|
|
199
|
-
let hook_data = self.hook.prepare_hook_data(&dummy_req)?;
|
|
200
|
-
let result = self.hook.invoke_hook(hook_data).await?;
|
|
201
|
-
|
|
202
|
-
if let Some(modifications) = result.request_modifications {
|
|
203
|
-
let mut builder = Response::builder().status(parts.status);
|
|
204
|
-
|
|
205
|
-
let header_mod_keys: Vec<String> = modifications
|
|
206
|
-
.headers
|
|
207
|
-
.as_ref()
|
|
208
|
-
.map(|mods| mods.keys().map(|k| k.to_lowercase()).collect())
|
|
209
|
-
.unwrap_or_default();
|
|
210
|
-
|
|
211
|
-
if let Some(header_mods) = modifications.headers {
|
|
212
|
-
for (key, value) in header_mods {
|
|
213
|
-
builder = builder.header(&key, &value);
|
|
214
|
-
}
|
|
215
|
-
}
|
|
216
|
-
|
|
217
|
-
for (name, value) in &parts.headers {
|
|
218
|
-
let key_str = name.as_str().to_lowercase();
|
|
219
|
-
if !header_mod_keys.contains(&key_str) {
|
|
220
|
-
builder = builder.header(name, value);
|
|
221
|
-
}
|
|
222
|
-
}
|
|
223
|
-
|
|
224
|
-
let body = modifications.body.unwrap_or(body_bytes);
|
|
225
|
-
return builder
|
|
226
|
-
.body(Body::from(body))
|
|
227
|
-
.map_err(|e| format!("Failed to build modified response: {e}"));
|
|
228
|
-
}
|
|
229
|
-
|
|
230
|
-
let mut builder = Response::builder().status(parts.status);
|
|
231
|
-
for (name, value) in parts.headers {
|
|
232
|
-
if let Some(name) = name {
|
|
233
|
-
builder = builder.header(name, value);
|
|
234
|
-
}
|
|
235
|
-
}
|
|
236
|
-
builder
|
|
237
|
-
.body(Body::from(body_bytes))
|
|
238
|
-
.map_err(|e| format!("Failed to rebuild response: {e}"))
|
|
239
|
-
}
|
|
240
|
-
|
|
241
|
-
/// Build an axum Response from hook result data
|
|
242
|
-
fn build_response_from_hook_result(result: &HookResultData) -> Result<Response<Body>, String> {
|
|
243
|
-
let status_code = result.status_code.unwrap_or(200);
|
|
244
|
-
let status =
|
|
245
|
-
StatusCode::from_u16(status_code).map_err(|e| format!("Invalid status code {status_code}: {e}"))?;
|
|
246
|
-
|
|
247
|
-
let mut builder = Response::builder().status(status);
|
|
248
|
-
|
|
249
|
-
if let Some(ref headers) = result.headers {
|
|
250
|
-
for (key, value) in headers {
|
|
251
|
-
builder = builder.header(key, value);
|
|
252
|
-
}
|
|
253
|
-
}
|
|
254
|
-
|
|
255
|
-
if !builder.headers_ref().is_some_and(|h| h.contains_key("content-type")) {
|
|
256
|
-
builder = builder.header("content-type", "application/json");
|
|
257
|
-
}
|
|
258
|
-
|
|
259
|
-
let body = result.body.clone().unwrap_or_else(|| b"{}".to_vec());
|
|
260
|
-
|
|
261
|
-
builder
|
|
262
|
-
.body(Body::from(body))
|
|
263
|
-
.map_err(|e| format!("Failed to build response: {e}"))
|
|
264
|
-
}
|
|
265
|
-
|
|
266
|
-
/// Apply request modifications to a request
|
|
267
|
-
fn apply_request_modifications(req: Request<Body>, mods: RequestModifications) -> Result<Request<Body>, String> {
|
|
268
|
-
let (mut parts, body) = req.into_parts();
|
|
269
|
-
|
|
270
|
-
if let Some(method) = &mods.method {
|
|
271
|
-
parts.method = method.parse().map_err(|e| format!("Invalid method '{method}': {e}"))?;
|
|
272
|
-
}
|
|
273
|
-
|
|
274
|
-
if let Some(path) = &mods.path {
|
|
275
|
-
parts.uri = path.parse().map_err(|e| format!("Invalid path '{path}': {e}"))?;
|
|
276
|
-
}
|
|
277
|
-
|
|
278
|
-
if let Some(new_headers) = &mods.headers {
|
|
279
|
-
for (key, value) in new_headers {
|
|
280
|
-
let header_name: http::header::HeaderName =
|
|
281
|
-
key.parse().map_err(|_| format!("Invalid header name: {key}"))?;
|
|
282
|
-
let header_value: http::header::HeaderValue = value
|
|
283
|
-
.parse()
|
|
284
|
-
.map_err(|_| format!("Invalid header value for {key}: {value}"))?;
|
|
285
|
-
parts.headers.insert(header_name, header_value);
|
|
286
|
-
}
|
|
287
|
-
}
|
|
288
|
-
|
|
289
|
-
let body = mods.body.map_or(body, Body::from);
|
|
290
|
-
|
|
291
|
-
Ok(Request::from_parts(parts, body))
|
|
292
|
-
}
|
|
293
|
-
}
|
|
294
|
-
|
|
295
|
-
/// Extract body bytes from an axum Body
|
|
296
|
-
///
|
|
297
|
-
/// This is a helper used by lifecycle executors to read response bodies.
|
|
298
|
-
///
|
|
299
|
-
/// # Errors
|
|
300
|
-
///
|
|
301
|
-
/// Returns an error if body collection fails.
|
|
302
|
-
pub async fn extract_body(body: Body) -> Result<Vec<u8>, String> {
|
|
303
|
-
use http_body_util::BodyExt;
|
|
304
|
-
|
|
305
|
-
let bytes = body
|
|
306
|
-
.collect()
|
|
307
|
-
.await
|
|
308
|
-
.map_err(|e| format!("Failed to read body: {e}"))?
|
|
309
|
-
.to_bytes();
|
|
310
|
-
Ok(bytes.to_vec())
|
|
311
|
-
}
|
|
312
|
-
|
|
313
|
-
#[cfg(test)]
|
|
314
|
-
mod tests {
|
|
315
|
-
use super::*;
|
|
316
|
-
|
|
317
|
-
#[test]
|
|
318
|
-
fn test_hook_result_data_continue() {
|
|
319
|
-
let result = HookResultData::continue_execution();
|
|
320
|
-
assert!(result.continue_execution);
|
|
321
|
-
assert_eq!(result.status_code, None);
|
|
322
|
-
assert_eq!(result.body, None);
|
|
323
|
-
}
|
|
324
|
-
|
|
325
|
-
#[test]
|
|
326
|
-
fn test_hook_result_data_short_circuit() {
|
|
327
|
-
let body = b"error".to_vec();
|
|
328
|
-
let mut headers = HashMap::new();
|
|
329
|
-
headers.insert("x-error".to_string(), "true".to_string());
|
|
330
|
-
|
|
331
|
-
let result = HookResultData::short_circuit(400, body.clone(), Some(headers.clone()));
|
|
332
|
-
assert!(!result.continue_execution);
|
|
333
|
-
assert_eq!(result.status_code, Some(400));
|
|
334
|
-
assert_eq!(result.body, Some(body));
|
|
335
|
-
assert_eq!(result.headers, Some(headers));
|
|
336
|
-
}
|
|
337
|
-
|
|
338
|
-
#[test]
|
|
339
|
-
fn test_hook_result_data_modify_request() {
|
|
340
|
-
let mods = RequestModifications {
|
|
341
|
-
method: Some("POST".to_string()),
|
|
342
|
-
path: Some("/new-path".to_string()),
|
|
343
|
-
headers: None,
|
|
344
|
-
body: None,
|
|
345
|
-
};
|
|
346
|
-
|
|
347
|
-
let result = HookResultData::modify_request(mods);
|
|
348
|
-
assert!(result.continue_execution);
|
|
349
|
-
assert_eq!(result.status_code, None);
|
|
350
|
-
assert_eq!(
|
|
351
|
-
result.request_modifications.as_ref().unwrap().method,
|
|
352
|
-
Some("POST".to_string())
|
|
353
|
-
);
|
|
354
|
-
assert_eq!(
|
|
355
|
-
result.request_modifications.as_ref().unwrap().path,
|
|
356
|
-
Some("/new-path".to_string())
|
|
357
|
-
);
|
|
358
|
-
}
|
|
359
|
-
|
|
360
|
-
#[test]
|
|
361
|
-
fn test_request_modifications_creation() {
|
|
362
|
-
let mods = RequestModifications {
|
|
363
|
-
method: Some("PUT".to_string()),
|
|
364
|
-
path: Some("/api/resource".to_string()),
|
|
365
|
-
headers: None,
|
|
366
|
-
body: Some(b"data".to_vec()),
|
|
367
|
-
};
|
|
368
|
-
|
|
369
|
-
assert_eq!(mods.method, Some("PUT".to_string()));
|
|
370
|
-
assert_eq!(mods.path, Some("/api/resource".to_string()));
|
|
371
|
-
assert_eq!(mods.body, Some(b"data".to_vec()));
|
|
372
|
-
}
|
|
373
|
-
|
|
374
|
-
struct MockHook {
|
|
375
|
-
result: HookResultData,
|
|
376
|
-
}
|
|
377
|
-
|
|
378
|
-
impl LanguageLifecycleHook for MockHook {
|
|
379
|
-
type HookData = ();
|
|
380
|
-
|
|
381
|
-
fn prepare_hook_data(&self, _req: &Request<Body>) -> Result<Self::HookData, String> {
|
|
382
|
-
Ok(())
|
|
383
|
-
}
|
|
384
|
-
|
|
385
|
-
fn invoke_hook(
|
|
386
|
-
&self,
|
|
387
|
-
_data: Self::HookData,
|
|
388
|
-
) -> Pin<Box<dyn Future<Output = Result<HookResultData, String>> + Send>> {
|
|
389
|
-
let result = self.result.clone();
|
|
390
|
-
Box::pin(async move { Ok(result) })
|
|
391
|
-
}
|
|
392
|
-
}
|
|
393
|
-
|
|
394
|
-
#[tokio::test]
|
|
395
|
-
async fn test_execute_request_hook_continue() {
|
|
396
|
-
let hook = Arc::new(MockHook {
|
|
397
|
-
result: HookResultData::continue_execution(),
|
|
398
|
-
});
|
|
399
|
-
let executor = LifecycleExecutor::new(hook);
|
|
400
|
-
|
|
401
|
-
let req = Request::builder().body(Body::empty()).unwrap();
|
|
402
|
-
let result = executor.execute_request_hook(req).await.unwrap();
|
|
403
|
-
|
|
404
|
-
assert!(result.is_ok());
|
|
405
|
-
}
|
|
406
|
-
|
|
407
|
-
#[tokio::test]
|
|
408
|
-
async fn test_execute_request_hook_short_circuit() {
|
|
409
|
-
let hook = Arc::new(MockHook {
|
|
410
|
-
result: HookResultData::short_circuit(403, b"Forbidden".to_vec(), None),
|
|
411
|
-
});
|
|
412
|
-
let executor = LifecycleExecutor::new(hook);
|
|
413
|
-
|
|
414
|
-
let req = Request::builder().body(Body::empty()).unwrap();
|
|
415
|
-
let result = executor.execute_request_hook(req).await.unwrap();
|
|
416
|
-
|
|
417
|
-
assert!(result.is_err());
|
|
418
|
-
let response = result.unwrap_err();
|
|
419
|
-
assert_eq!(response.status(), StatusCode::FORBIDDEN);
|
|
420
|
-
}
|
|
421
|
-
|
|
422
|
-
#[tokio::test]
|
|
423
|
-
async fn test_execute_request_hook_modify_request() {
|
|
424
|
-
let mods = RequestModifications {
|
|
425
|
-
method: Some("POST".to_string()),
|
|
426
|
-
path: Some("/new-path".to_string()),
|
|
427
|
-
headers: None,
|
|
428
|
-
body: Some(b"new body".to_vec()),
|
|
429
|
-
};
|
|
430
|
-
let hook = Arc::new(MockHook {
|
|
431
|
-
result: HookResultData::modify_request(mods),
|
|
432
|
-
});
|
|
433
|
-
let executor = LifecycleExecutor::new(hook);
|
|
434
|
-
|
|
435
|
-
let req = Request::builder()
|
|
436
|
-
.method("GET")
|
|
437
|
-
.uri("/old-path")
|
|
438
|
-
.body(Body::empty())
|
|
439
|
-
.unwrap();
|
|
440
|
-
let result = executor.execute_request_hook(req).await.unwrap();
|
|
441
|
-
|
|
442
|
-
assert!(result.is_ok());
|
|
443
|
-
let modified_req = result.unwrap();
|
|
444
|
-
assert_eq!(modified_req.method(), "POST");
|
|
445
|
-
assert_eq!(modified_req.uri().path(), "/new-path");
|
|
446
|
-
}
|
|
447
|
-
|
|
448
|
-
#[tokio::test]
|
|
449
|
-
async fn test_execute_response_hook_with_modifications() {
|
|
450
|
-
let mods = RequestModifications {
|
|
451
|
-
method: None,
|
|
452
|
-
path: None,
|
|
453
|
-
headers: Some({
|
|
454
|
-
let mut h = HashMap::new();
|
|
455
|
-
h.insert("X-Modified".to_string(), "true".to_string());
|
|
456
|
-
h
|
|
457
|
-
}),
|
|
458
|
-
body: Some(b"modified response".to_vec()),
|
|
459
|
-
};
|
|
460
|
-
let hook = Arc::new(MockHook {
|
|
461
|
-
result: HookResultData::modify_request(mods),
|
|
462
|
-
});
|
|
463
|
-
let executor = LifecycleExecutor::new(hook);
|
|
464
|
-
|
|
465
|
-
let resp = Response::builder().status(200).body(Body::from("original")).unwrap();
|
|
466
|
-
let result = executor.execute_response_hook(resp).await.unwrap();
|
|
467
|
-
|
|
468
|
-
assert_eq!(result.status(), StatusCode::OK);
|
|
469
|
-
assert_eq!(result.headers().get("X-Modified").unwrap().to_str().unwrap(), "true");
|
|
470
|
-
}
|
|
471
|
-
|
|
472
|
-
#[tokio::test]
|
|
473
|
-
async fn test_build_response_from_hook_result_with_headers() {
|
|
474
|
-
let mut headers = HashMap::new();
|
|
475
|
-
headers.insert("X-Custom".to_string(), "value".to_string());
|
|
476
|
-
|
|
477
|
-
let result = HookResultData::short_circuit(201, b"Created".to_vec(), Some(headers));
|
|
478
|
-
|
|
479
|
-
let response = LifecycleExecutor::<MockHook>::build_response_from_hook_result(&result).unwrap();
|
|
480
|
-
assert_eq!(response.status(), StatusCode::CREATED);
|
|
481
|
-
assert_eq!(response.headers().get("X-Custom").unwrap().to_str().unwrap(), "value");
|
|
482
|
-
}
|
|
483
|
-
|
|
484
|
-
#[tokio::test]
|
|
485
|
-
async fn test_build_response_from_hook_result_default_content_type() {
|
|
486
|
-
let result = HookResultData::short_circuit(200, b"{}".to_vec(), None);
|
|
487
|
-
|
|
488
|
-
let response = LifecycleExecutor::<MockHook>::build_response_from_hook_result(&result).unwrap();
|
|
489
|
-
assert_eq!(
|
|
490
|
-
response.headers().get("content-type").unwrap().to_str().unwrap(),
|
|
491
|
-
"application/json"
|
|
492
|
-
);
|
|
493
|
-
}
|
|
494
|
-
|
|
495
|
-
#[tokio::test]
|
|
496
|
-
async fn test_apply_request_modifications_method() {
|
|
497
|
-
let mods = RequestModifications {
|
|
498
|
-
method: Some("PATCH".to_string()),
|
|
499
|
-
path: None,
|
|
500
|
-
headers: None,
|
|
501
|
-
body: None,
|
|
502
|
-
};
|
|
503
|
-
|
|
504
|
-
let req = Request::builder().method("GET").body(Body::empty()).unwrap();
|
|
505
|
-
let modified = LifecycleExecutor::<MockHook>::apply_request_modifications(req, mods).unwrap();
|
|
506
|
-
|
|
507
|
-
assert_eq!(modified.method(), "PATCH");
|
|
508
|
-
}
|
|
509
|
-
|
|
510
|
-
#[tokio::test]
|
|
511
|
-
async fn test_apply_request_modifications_path() {
|
|
512
|
-
let mods = RequestModifications {
|
|
513
|
-
method: None,
|
|
514
|
-
path: Some("/api/v2/users".to_string()),
|
|
515
|
-
headers: None,
|
|
516
|
-
body: None,
|
|
517
|
-
};
|
|
518
|
-
|
|
519
|
-
let req = Request::builder().uri("/api/v1/users").body(Body::empty()).unwrap();
|
|
520
|
-
let modified = LifecycleExecutor::<MockHook>::apply_request_modifications(req, mods).unwrap();
|
|
521
|
-
|
|
522
|
-
assert_eq!(modified.uri().path(), "/api/v2/users");
|
|
523
|
-
}
|
|
524
|
-
|
|
525
|
-
#[tokio::test]
|
|
526
|
-
async fn test_apply_request_modifications_headers() {
|
|
527
|
-
let mut new_headers = HashMap::new();
|
|
528
|
-
new_headers.insert("Authorization".to_string(), "Bearer token".to_string());
|
|
529
|
-
|
|
530
|
-
let mods = RequestModifications {
|
|
531
|
-
method: None,
|
|
532
|
-
path: None,
|
|
533
|
-
headers: Some(new_headers),
|
|
534
|
-
body: None,
|
|
535
|
-
};
|
|
536
|
-
|
|
537
|
-
let req = Request::builder().body(Body::empty()).unwrap();
|
|
538
|
-
let modified = LifecycleExecutor::<MockHook>::apply_request_modifications(req, mods).unwrap();
|
|
539
|
-
|
|
540
|
-
assert_eq!(
|
|
541
|
-
modified.headers().get("Authorization").unwrap().to_str().unwrap(),
|
|
542
|
-
"Bearer token"
|
|
543
|
-
);
|
|
544
|
-
}
|
|
545
|
-
|
|
546
|
-
#[tokio::test]
|
|
547
|
-
async fn test_apply_request_modifications_body() {
|
|
548
|
-
let new_body = b"modified body".to_vec();
|
|
549
|
-
let mods = RequestModifications {
|
|
550
|
-
method: None,
|
|
551
|
-
path: None,
|
|
552
|
-
headers: None,
|
|
553
|
-
body: Some(new_body.clone()),
|
|
554
|
-
};
|
|
555
|
-
|
|
556
|
-
let req = Request::builder().body(Body::from("original body")).unwrap();
|
|
557
|
-
let modified = LifecycleExecutor::<MockHook>::apply_request_modifications(req, mods).unwrap();
|
|
558
|
-
|
|
559
|
-
let body_bytes = extract_body(modified.into_body()).await.unwrap();
|
|
560
|
-
assert_eq!(body_bytes, new_body);
|
|
561
|
-
}
|
|
562
|
-
|
|
563
|
-
#[tokio::test]
|
|
564
|
-
async fn test_apply_request_modifications_invalid_method() {
|
|
565
|
-
let mods = RequestModifications {
|
|
566
|
-
method: Some(String::new()),
|
|
567
|
-
path: None,
|
|
568
|
-
headers: None,
|
|
569
|
-
body: None,
|
|
570
|
-
};
|
|
571
|
-
|
|
572
|
-
let req = Request::builder().body(Body::empty()).unwrap();
|
|
573
|
-
let result = LifecycleExecutor::<MockHook>::apply_request_modifications(req, mods);
|
|
574
|
-
|
|
575
|
-
assert!(result.is_err());
|
|
576
|
-
assert!(result.unwrap_err().contains("Invalid method"));
|
|
577
|
-
}
|
|
578
|
-
|
|
579
|
-
#[tokio::test]
|
|
580
|
-
async fn test_extract_body_helper() {
|
|
581
|
-
let body = Body::from("test data");
|
|
582
|
-
let bytes = extract_body(body).await.unwrap();
|
|
583
|
-
|
|
584
|
-
assert_eq!(bytes, b"test data");
|
|
585
|
-
}
|
|
586
|
-
|
|
587
|
-
#[tokio::test]
|
|
588
|
-
async fn test_extract_body_empty() {
|
|
589
|
-
let body = Body::empty();
|
|
590
|
-
let bytes = extract_body(body).await.unwrap();
|
|
591
|
-
|
|
592
|
-
assert_eq!(bytes.len(), 0);
|
|
593
|
-
}
|
|
594
|
-
}
|