reintroduce the variadic macro for ruma handler.
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
parent
f871d8fd4e
commit
aebae11c82
1 changed files with 30 additions and 18 deletions
|
@ -1,6 +1,7 @@
|
|||
use std::future::Future;
|
||||
|
||||
use axum::{
|
||||
extract::FromRequestParts,
|
||||
response::IntoResponse,
|
||||
routing::{on, MethodFilter},
|
||||
Router,
|
||||
|
@ -32,13 +33,17 @@ pub(in super::super) trait RumaHandler<T> {
|
|||
fn add_route(&self, router: Router, path: &str) -> Router;
|
||||
}
|
||||
|
||||
impl<Req, E, F, Fut> RumaHandler<Ruma<Req>> for F
|
||||
where
|
||||
macro_rules! ruma_handler {
|
||||
( $($tx:ident),* $(,)? ) => {
|
||||
#[allow(non_snake_case)]
|
||||
impl<Req, Ret, Fut, Fun, $($tx,)*> RumaHandler<($($tx,)* Ruma<Req>,)> for Fun
|
||||
where
|
||||
Req: IncomingRequest + Send + 'static,
|
||||
F: FnOnce(Ruma<Req>) -> Fut + Clone + Send + Sync + 'static,
|
||||
Fut: Future<Output = Result<Req::OutgoingResponse, E>> + Send,
|
||||
E: IntoResponse,
|
||||
{
|
||||
Ret: IntoResponse,
|
||||
Fut: Future<Output = Result<Req::OutgoingResponse, Ret>> + Send,
|
||||
Fun: FnOnce($($tx,)* Ruma<Req>) -> Fut + Clone + Send + Sync + 'static,
|
||||
$( $tx: FromRequestParts<()> + Send + 'static, )*
|
||||
{
|
||||
fn add_routes(&self, router: Router) -> Router {
|
||||
Req::METADATA
|
||||
.history
|
||||
|
@ -49,10 +54,17 @@ where
|
|||
fn add_route(&self, router: Router, path: &str) -> Router {
|
||||
let handle = self.clone();
|
||||
let method = method_to_filter(&Req::METADATA.method);
|
||||
let action = |req| async { handle(req).await.map(RumaResponse) };
|
||||
let action = |$($tx,)* req| async { handle($($tx,)* req).await.map(RumaResponse) };
|
||||
router.route(path, on(method, action))
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
ruma_handler!();
|
||||
ruma_handler!(T1);
|
||||
ruma_handler!(T1, T2);
|
||||
ruma_handler!(T1, T2, T3);
|
||||
ruma_handler!(T1, T2, T3, T4);
|
||||
|
||||
const fn method_to_filter(method: &Method) -> MethodFilter {
|
||||
match *method {
|
||||
|
|
Loading…
Add table
Reference in a new issue