From fff207d3cb41c5a76db234196f16d28952dfd91c Mon Sep 17 00:00:00 2001
From: Fuwn <contact@fuwn.me>
Date: Thu, 6 Apr 2023 08:01:52 +0000
Subject: feat(response): async error handler

---
 src/handler/response/error.rs | 19 ++++++++++++++++---
 src/handler/response/route.rs |  4 +---
 2 files changed, 17 insertions(+), 6 deletions(-)

(limited to 'src/handler/response')

diff --git a/src/handler/response/error.rs b/src/handler/response/error.rs
index c433a46..a1a7885 100644
--- a/src/handler/response/error.rs
+++ b/src/handler/response/error.rs
@@ -16,10 +16,23 @@
 // Copyright (C) 2022-2022 Fuwn <contact@fuwn.me>
 // SPDX-License-Identifier: GPL-3.0-only
 
+use async_trait::async_trait;
+
 use crate::{context::ErrorContext, Response};
 
 #[allow(clippy::module_name_repetitions)]
-pub trait ErrorResponse: FnMut(ErrorContext) -> Response + Send + Sync {}
+#[async_trait]
+pub trait ErrorResponse: Send + Sync {
+  async fn call(&mut self, context: ErrorContext) -> Response;
+}
 
-impl<T> ErrorResponse for T where T: FnMut(ErrorContext) -> Response + Send + Sync
-{}
+#[async_trait]
+impl<T, F> ErrorResponse for T
+where
+  T: FnMut(ErrorContext) -> F + Send + Sync,
+  F: std::future::Future<Output = Response> + Send + 'static,
+{
+  async fn call(&mut self, context: ErrorContext) -> Response {
+    (*self)(context).await
+  }
+}
diff --git a/src/handler/response/route.rs b/src/handler/response/route.rs
index 0196363..a65b889 100644
--- a/src/handler/response/route.rs
+++ b/src/handler/response/route.rs
@@ -16,8 +16,6 @@
 // Copyright (C) 2022-2022 Fuwn <contact@fuwn.me>
 // SPDX-License-Identifier: GPL-3.0-only
 
-use std::future::Future;
-
 use async_trait::async_trait;
 
 use crate::{context::RouteContext, Response};
@@ -32,7 +30,7 @@ pub trait RouteResponse: Send + Sync {
 impl<T, F> RouteResponse for T
 where
   T: FnMut(RouteContext<'_>) -> F + Send + Sync,
-  F: Future<Output = Response> + Send + 'static,
+  F: std::future::Future<Output = Response> + Send + 'static,
 {
   async fn call(&mut self, context: RouteContext<'_>) -> Response {
     (*self)(context).await
-- 
cgit v1.2.3