From ab3ab6d7c4f9fbf36d1bd8c8bf505f8be031620a Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Fri, 14 Jun 2024 15:51:28 +1000
Subject: [PATCH] fix: Next path gets lost after selecting hns.id domain

---
 website/routes.py           | 6 ++++++
 website/templates/home.html | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/website/routes.py b/website/routes.py
index a15ee6a..f145750 100644
--- a/website/routes.py
+++ b/website/routes.py
@@ -124,6 +124,12 @@ def hnsid_domain(domain):
                 db.session.commit()
             session["id"] = user.id
             session.permanent = True
+
+            # Check if next page is specified
+            next_page = request.args.get("next")
+            if next_page:
+                return redirect(next_page)
+
             return redirect("/")
 
     return jsonify({"success": False, "error": "Domain not found"})
diff --git a/website/templates/home.html b/website/templates/home.html
index 59a0c1f..ce29af1 100644
--- a/website/templates/home.html
+++ b/website/templates/home.html
@@ -213,7 +213,7 @@
   <script>
       function HNSIDLoginSelect() {
           var selectedNFT = document.getElementById("nftDropdown").value;
-          window.location.href = "/hnsid/" + selectedNFT;
+          window.location.href = "/hnsid/" + selectedNFT + window.location.search;
       }
   </script>