relay_core_script/
engine_trait.rs

1use async_trait::async_trait;
2use relay_core_api::flow::{Flow, WebSocketMessage};
3use relay_core_lib::interceptor::{HttpBody, RequestAction, ResponseAction, WebSocketMessageAction, BoxError};
4
5#[async_trait]
6pub trait ScriptEngineTrait: Send + Sync {
7    async fn load_script(&mut self, script: &str) -> Result<(), BoxError>;
8    
9    async fn on_request_headers(&self, _flow: &mut Flow) -> Result<Option<Flow>, BoxError> {
10        Ok(None)
11    }
12
13    async fn on_request(&self, flow: &mut Flow, body: HttpBody) -> Result<RequestAction, BoxError>;
14    
15    async fn on_response_headers(&self, _flow: &mut Flow) -> Result<Option<Flow>, BoxError> {
16        Ok(None)
17    }
18
19    async fn on_response(&self, flow: &mut Flow, body: HttpBody) -> Result<ResponseAction, BoxError>;
20
21    async fn on_websocket_message(&self, _flow: &mut Flow, _message: &mut WebSocketMessage) -> Result<WebSocketMessageAction, BoxError> {
22        Ok(WebSocketMessageAction::Continue(_message.clone()))
23    }
24}
relay_core_script/engine_trait.rs

relay_core_script/
engine_trait.rs
