From 391038b6a67ddea3d3efb67c44bb3feb64360eae Mon Sep 17 00:00:00 2001
From: Fuwn <contact@fuwn.net>
Date: Wed, 29 Mar 2023 08:28:56 +0000
Subject: feat(router): modify content from post-route callback

---
 src/handler.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'src/handler.rs')

diff --git a/src/handler.rs b/src/handler.rs
index 15dfd6d..c5e23e8 100644
--- a/src/handler.rs
+++ b/src/handler.rs
@@ -16,12 +16,17 @@
 // Copyright (C) 2022-2022 Fuwn <contact@fuwn.me>
 // SPDX-License-Identifier: GPL-3.0-only
 
-use crate::{returnable, returnable::RouteContext, Response};
+use crate::{
+  returnable,
+  returnable::{CallbackContext, RouteContext},
+  Response,
+};
 
 pub type RouteResponse =
   Box<dyn FnMut(RouteContext<'_>) -> Response<'_> + Send + Sync>;
 pub type ErrorResponse =
   Box<dyn FnMut(returnable::ErrorContext<'_>) -> Response<'_> + Send + Sync>;
-pub type Callback =
-  Box<dyn FnMut(returnable::CallbackContext<'_>) + Send + Sync>;
+pub type Callback = Box<dyn FnMut(CallbackContext<'_>) + Send + Sync>;
+pub type CleanupCallback =
+  Box<dyn FnMut(CallbackContext<'_>, &mut String) + Send + Sync>;
 pub type Partial = Box<dyn FnMut(RouteContext<'_>) -> String + Send + Sync>;
-- 
cgit v1.2.3