Merge pull request #16 from YouweGit/feature/set-before-2fa-url

set before auth url
YouweGit · Dec 29, 2023 · 06be826 · 06be826
2 parents eb54132 + e166372
commit 06be826
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/Plugin/AccountRedirectPlugin.php b/Plugin/AccountRedirectPlugin.php
@@ -13,9 +13,19 @@ public function __construct(private readonly CustomerSession $customerSession)
     {
     }
 
+    public function beforeGetRedirect(Redirect $subject): array
+    {
+        $beforeAuthUrl = $this->session->getBeforeAuthUrl(false);
+        if ($beforeAuthUrl) {
+            $this->customerSession->setBefore2faUrl($beforeAuthUrl);
+        }
+
+        return [];
+    }
+
     public function afterGetRedirectCookie(Redirect $subject, ?string $result): ?string
     {
-        if ($result && !$this->customerSession->getBefore2faUrl(false)) {
+        if ($result) {
             $this->customerSession->setBefore2faUrl($result);
         }
 

diff --git a/Plugin/LoginPostPlugin.php b/Plugin/LoginPostPlugin.php
@@ -31,7 +31,7 @@ public function __construct(
     public function afterExecute(LoginPost $subject, Redirect $result): Redirect
     {
         if (!$this->config->isEnabled()) {
-            return $this;
+            return $result;
         }
 
         if ($this->customerSession->get2faSuccessful()) {