mirrors/lemmy
1
0
Fork 0
mirror of https://github.com/LemmyNet/lemmy.git synced 2024-09-28 22:31:57 +00:00
Code Issues Projects Releases Wiki Activity

add after hook

Browse source
This commit is contained in:
Felix Ableitner 2024-05-06 14:13:06 +02:00
parent ea5a5522fb
commit caf9d20dea
1 changed files with 26 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
src/plugin_middleware.rs
View file

@ -1,4 +1,4 @@
use actix_http::h1::Payload; use actix_http::{body::BoxBody, h1::Payload};
use actix_web::{ use actix_web::{
body::MessageBody, body::MessageBody,
dev::{forward_ready, Service, ServiceRequest, ServiceResponse, Transform}, dev::{forward_ready, Service, ServiceRequest, ServiceResponse, Transform},
@ -22,13 +22,12 @@ impl PluginMiddleware {
PluginMiddleware {} PluginMiddleware {}
} }
} }
impl<S, B> Transform<S, ServiceRequest> for PluginMiddleware impl<S> Transform<S, ServiceRequest> for PluginMiddleware
where where
S: Service<ServiceRequest, Response = ServiceResponse<B>, Error = Error> + 'static, S: Service<ServiceRequest, Response = ServiceResponse<BoxBody>, Error = Error> + 'static,
S::Future: 'static, S::Future: 'static,
B: MessageBody + 'static,
{ {
type Response = ServiceResponse<B>; type Response = ServiceResponse<BoxBody>;
type Error = Error; type Error = Error;
type Transform = SessionService<S>; type Transform = SessionService<S>;
type InitError = (); type InitError = ();
@ -45,13 +44,12 @@ pub struct SessionService<S> {
service: Rc<S>, service: Rc<S>,
} }
impl<S, B> Service<ServiceRequest> for SessionService<S> impl<S> Service<ServiceRequest> for SessionService<S>
where where
S: Service<ServiceRequest, Response = ServiceResponse<B>, Error = Error> + 'static, S: Service<ServiceRequest, Response = ServiceResponse<BoxBody>, Error = Error> + 'static,
S::Future: 'static, S::Future: 'static,
B: 'static,
{ {
type Response = ServiceResponse<B>; type Response = ServiceResponse<BoxBody>;
type Error = Error; type Error = Error;
type Future = LocalBoxFuture<'static, Result<Self::Response, Self::Error>>; type Future = LocalBoxFuture<'static, Result<Self::Response, Self::Error>>;
@ -61,27 +59,39 @@ where
let svc = self.service.clone(); let svc = self.service.clone();
Box::pin(async move { Box::pin(async move {
let method = service_req.method(); let method = service_req.method().clone();
let path = service_req.path().replace("/api/v3/", "").replace("/", "_"); let path = service_req.path().replace("/api/v3/", "").replace("/", "_");
// TODO: naming can be a bit silly, `POST /api/v3/post` becomes `api_before_post_post` // TODO: naming can be a bit silly, `POST /api/v3/post` becomes `api_before_post_post`
let plugin_hook = format!("api_before_{method}_{path}").to_lowercase(); let before_plugin_hook = format!("api_before_{method}_{path}").to_lowercase();
info!("Calling plugin hook {}", &plugin_hook); info!("Calling plugin hook {}", &before_plugin_hook);
if let Some(mut plugins) = load_plugins()? { if let Some(mut plugins) = load_plugins()? {
if plugins.function_exists(&plugin_hook) { if plugins.function_exists(&before_plugin_hook) {
let payload = service_req.extract::<Bytes>().await?; let payload = service_req.extract::<Bytes>().await?;
let mut json: Value = serde_json::from_slice(&payload.to_vec())?; let mut json: Value = serde_json::from_slice(&payload.to_vec())?;
call_plugin(plugins, &plugin_hook, &mut json)?; call_plugin(plugins, &before_plugin_hook, &mut json)?;
let (_, mut new_payload) = Payload::create(true); let (_, mut new_payload) = Payload::create(true);
new_payload.unread_data(Bytes::from(serde_json::to_vec_pretty(&json)?)); new_payload.unread_data(Bytes::from(serde_json::to_vec(&json)?));
service_req.set_payload(new_payload.into()); service_req.set_payload(new_payload.into());
} }
} }
let res = svc.call(service_req).await?; let mut res = svc.call(service_req).await?;
// TODO: add after hook // TODO: add after hook
let after_plugin_hook = format!("api_after_{method}_{path}").to_lowercase();
info!("Calling plugin hook {}", &after_plugin_hook);
if let Some(mut plugins) = load_plugins()? {
if plugins.function_exists(&before_plugin_hook) {
res = res.map_body(|_, body| {
let mut json: Value =
serde_json::from_slice(&body.try_into_bytes().unwrap().to_vec()).unwrap();
call_plugin(plugins, &after_plugin_hook, &mut json).unwrap();
BoxBody::new(Bytes::from(serde_json::to_vec(&json).unwrap()))
});
}
}
Ok(res) Ok(res)
}) })