From d2ec2ba7af65a7a620e61c85e5a3724aa4c7d0a1 Mon Sep 17 00:00:00 2001
From: titanism <101466223+titanism@users.noreply.github.com>
Date: Thu, 19 Dec 2024 04:28:44 -0600
Subject: [PATCH] fix: added ?return_to for ensureLoggedIn middleware

---
 index.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/index.js b/index.js
index 6d14b78..7ac7c2e 100644
--- a/index.js
+++ b/index.js
@@ -189,11 +189,14 @@ class Policies {
         }
       }
 
-      const redirectTo =
+      let redirectTo =
         typeof ctx.state.l === 'function'
           ? ctx.state.l(this.config.loginRoute)
           : this.config.loginRoute;
 
+      if (ctx.url && ctx.url !== '/')
+        redirectTo += `?return_to=${encodeURIComponent(ctx.url)}`;
+
       if (ctx.accepts('html')) ctx.redirect(redirectTo);
       else ctx.body = { message, redirectTo };