From 6ae24d4ee3557d503cab2e8bf130f49d1aad62fe Mon Sep 17 00:00:00 2001
From: Liam Chen <liamchzh@gmail.com>
Date: Wed, 11 May 2022 15:47:17 -0400
Subject: [PATCH] Add :compojure/route-context to the request map

Contains the combined route context for the current route.
---
 src/compojure/core.clj       | 12 +++++++-----
 test/compojure/core_test.clj | 17 +++++++++++++++++
 2 files changed, 24 insertions(+), 5 deletions(-)

diff --git a/src/compojure/core.clj b/src/compojure/core.clj
index 32b25de9..4216cbea 100644
--- a/src/compojure/core.clj
+++ b/src/compojure/core.clj
@@ -251,7 +251,7 @@
 (defn- remove-suffix [path suffix]
   (subs path 0 (- (count path) (count suffix))))
 
-(defn- context-request [request route]
+(defn- context-request [request route route-context]
   (if-let [params (clout/route-matches route request)]
     (let [uri     (:uri request)
           path    (:path-info request uri)
@@ -261,7 +261,8 @@
       (-> request
           (assoc-route-params (decode-route-params params))
           (assoc :path-info (if (= subpath "") "/" subpath)
-                 :context   (remove-suffix uri subpath))))))
+                 :context   (remove-suffix uri subpath))
+          (update :compojure/route-context str route-context)))))
 
 (defn- context-route [route]
   (let [re-context {:__path-info #"|/.*"}]
@@ -279,7 +280,7 @@
       :else
        `(clout/route-compile (str ~route ":__path-info") ~re-context))))
 
-(defn ^:no-doc make-context [route make-handler]
+(defn ^:no-doc make-context [route path make-handler]
   (letfn [(handler
             ([request]
              (when-let [context-handler (make-handler request)]
@@ -292,10 +293,10 @@
       handler
       (fn
         ([request]
-         (if-let [request (context-request request route)]
+         (if-let [request (context-request request route path)]
            (handler request)))
         ([request respond raise]
-         (if-let [request (context-request request route)]
+         (if-let [request (context-request request route path)]
            (handler request respond raise)
            (respond nil)))))))
 
@@ -311,6 +312,7 @@
   [path args & routes]
   `(make-context
     ~(context-route path)
+    ~path
     (fn [request#]
       (let-request [~args request#]
         (routes ~@routes)))))
diff --git a/test/compojure/core_test.clj b/test/compojure/core_test.clj
index f5430801..7576ad73 100644
--- a/test/compojure/core_test.clj
+++ b/test/compojure/core_test.clj
@@ -196,6 +196,23 @@
         "/foo/10/b%20r" "/b%20r"
         "/bar/10"       nil)))
 
+  (testing ":compojure/route-context key"
+    (let [handler (context "/foo/:id" [_] :compojure/route-context)]
+      (are [url route-context] (= (handler (mock/request :get url)) route-context)
+        "/foo/10"       "/foo/:id"
+        "/foo/10/bar"   "/foo/:id"
+        "/foo/10/b%20r" "/foo/:id"
+        "/bar/10"       nil)))
+
+  (testing ":compojure/route-context key in nested context"
+    (let [handler (context "/foo/:foo-id" [_]
+                    (context "/bar/:bar-id" [_] :compojure/route-context))]
+      (are [url route-context] (= (handler (mock/request :get url)) route-context)
+        "/foo/1/bar/2"       "/foo/:foo-id/bar/:bar-id"
+        "/foo/1/bar/2/baz"   "/foo/:foo-id/bar/:bar-id"
+        "/foo/1/bar/2/b%20r" "/foo/:foo-id/bar/:bar-id"
+        "/bar/10"       nil)))
+
   (testing "routes"
     (let [handler (context "/foo/:id" [id]
                     (GET "/" [] "root")