From 09ff35a60264d7d6eba3cace282a44ea3e008d24 Mon Sep 17 00:00:00 2001
From: Raphael C <greefine@hotmail.fr>
Date: Wed, 24 Apr 2024 20:56:43 +0200
Subject: [PATCH] changes: guard

---
 actix-web/CHANGES.md       |  1 +
 actix-web/src/guard/mod.rs | 20 ++++++++++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/actix-web/CHANGES.md b/actix-web/CHANGES.md
index c6454ed65..f370f38a4 100644
--- a/actix-web/CHANGES.md
+++ b/actix-web/CHANGES.md
@@ -5,6 +5,7 @@
 ### Added
 
 - Add `unicode` crate feature (on-by-default) to switch between `regex` and `regex-lite` as a trade-off between full unicode support and binary size.
+- Support for app_data method in a GuardContext
 
 ### Changed
 
diff --git a/actix-web/src/guard/mod.rs b/actix-web/src/guard/mod.rs
index 9451a60f9..f2b13c59c 100644
--- a/actix-web/src/guard/mod.rs
+++ b/actix-web/src/guard/mod.rs
@@ -110,6 +110,12 @@ impl<'a> GuardContext<'a> {
     pub fn header<H: Header>(&self) -> Option<H> {
         H::parse(self.req).ok()
     }
+
+    /// Counterpart to [`HttpRequest::app_data`].
+    #[inline]
+    pub fn app_data<T: 'static>(&self) -> Option<&T> {
+        self.req.app_data()
+    }
 }
 
 /// Interface for routing guards.
@@ -512,4 +518,18 @@ mod tests {
             .to_srv_request();
         assert!(guard.check(&req.guard_ctx()));
     }
+
+    #[test]
+    fn app_data() {
+        const TEST_VALUE: u32 = 42;
+        let guard = fn_guard(|ctx| dbg!(ctx.app_data::<u32>()) == Some(&TEST_VALUE));
+
+        let req = TestRequest::default().app_data(TEST_VALUE).to_srv_request();
+        assert!(guard.check(&req.guard_ctx()));
+
+        let req = TestRequest::default()
+            .app_data(TEST_VALUE * 2)
+            .to_srv_request();
+        assert!(!guard.check(&req.guard_ctx()));
+    }
 }