diff --git a/website/routes.py b/website/routes.py index 1b40806..23e8652 100644 --- a/website/routes.py +++ b/website/routes.py @@ -1,6 +1,6 @@ import time from .varo_auth import flask_login as varo_auth_flask_login -from flask import Blueprint, request, session, url_for +from flask import Blueprint, request, session, url_for, make_response from flask import render_template, redirect, jsonify, send_from_directory from werkzeug.security import gen_salt from authlib.integrations.flask_oauth2 import current_token @@ -40,6 +40,14 @@ def split_by_crlf(s): @bp.route("/", methods=("GET", "POST")) def home(): next_page = request.args.get("next") + + # Check if session exists + if "uuid" not in session: + session["uuid"] = gen_salt(24) + session.permanent = True + + uuid = session["uuid"] + if request.method == "POST": auth = varo_auth_flask_login(request) if auth == False: @@ -83,9 +91,15 @@ def home(): if openseaInfo.status_code == 200: hnsid = openseaInfo.json() + domains = [] + if 'domains' in session: + domains = session['domains'] + - return render_template("home.html", user=user, clients=clients, address=address, hnsid=hnsid, users=users) + return render_template("home.html", user=user, clients=clients, + address=address, hnsid=hnsid, users=users, + uuid=uuid, next=next_page, domains=domains) @bp.route("/hnsid", methods=["POST"]) def hnsid(): @@ -142,6 +156,129 @@ def hnsid_domain(domain): return jsonify({"success": False, "error": "Domain not found"}) + +@bp.route("/txt", methods=["POST"]) +def txtLogin(): + # Get domain from form + domain = request.form.get("domain") + # Get uuid + uuid = session["uuid"] + + query = dns.message.make_query(domain, dns.rdatatype.TXT) + dns_request = query.to_wire() + + # Send the DNS query over HTTPS + response = requests.post('https://hnsdoh.com/dns-query', data=dns_request, headers={'Content-Type': 'application/dns-message'}) + + # Parse the DNS response + dns_response = dns.message.from_wire(response.content) + + # Loop over TXT records and look for profile avatar + idns_records = [] + for record in dns_response.answer: + if record.rdtype == dns.rdatatype.TXT: + for txt in record: + txt_value:str = txt.to_text().strip('"') + if txt_value.startswith("IDNS1"): + print(txt_value) + idns = txt_value.removeprefix("IDNS1 ") + idns = idns.split(" ") + for r in idns: + idns_records.append(r) + + for record in idns_records: + print(record) + type = record.split(":")[0] + content = record.split(":")[1] + key = content.split("=")[0] + value = content.split("=")[1] + print(f"Type: {type}, Key: {key}, Value: {value}") + if type == "auth" and key == "login.hns.au": + if value == uuid: + # Add domain to user + user = User.query.filter_by(username=domain).first() + if not user: + # Create user with username and profile picture + user = User(username=domain) + db.session.add(user) + db.session.commit() + session["id"] = user.id + session.permanent = True + if "domains" not in session: + session["domains"] = [] + + if domain not in session["domains"]: + session["domains"].append(domain) + print("User logged in with TXT") + + # Check if next page is specified + next_page = request.args.get("next") + print(next_page) + if next_page and next_page != "None": + return redirect(next_page) + + return redirect("/") + +@bp.route("/txt/") +def txtLoginDomain(domain): + # Get uuid + uuid = session["uuid"] + + query = dns.message.make_query(domain, dns.rdatatype.TXT) + dns_request = query.to_wire() + + # Send the DNS query over HTTPS + response = requests.post('https://hnsdoh.com/dns-query', data=dns_request, headers={'Content-Type': 'application/dns-message'}) + + # Parse the DNS response + dns_response = dns.message.from_wire(response.content) + + # Loop over TXT records and look for profile avatar + idns_records = [] + for record in dns_response.answer: + if record.rdtype == dns.rdatatype.TXT: + for txt in record: + txt_value:str = txt.to_text().strip('"') + if txt_value.startswith("IDNS1"): + print(txt_value) + idns = txt_value.removeprefix("IDNS1 ") + idns = idns.split(" ") + for r in idns: + idns_records.append(r) + + for record in idns_records: + print(record) + type = record.split(":")[0] + content = record.split(":")[1] + key = content.split("=")[0] + value = content.split("=")[1] + print(f"Type: {type}, Key: {key}, Value: {value}") + if type == "auth" and key == "login.hns.au": + if value == uuid: + # Add domain to user + user = User.query.filter_by(username=domain).first() + if not user: + # Create user with username and profile picture + user = User(username=domain) + db.session.add(user) + db.session.commit() + session["id"] = user.id + session.permanent = True + if "domains" not in session: + session["domains"] = [] + + if domain not in session["domains"]: + session["domains"].append(domain) + print("User logged in with TXT") + + # Check if next page is specified + next_page = request.args.get("next") + print(next_page) + if next_page and next_page != "None": + return redirect(next_page) + + return redirect("/") + @bp.route("/logout") def logout(): del session["id"] @@ -355,11 +492,13 @@ def avatar(username): return send_from_directory("templates", "favicon.png", mimetype="image/png") - - - - - @bp.route("/favicon.png") def favicon(): return send_from_directory("templates", "favicon.png", mimetype="image/png") + + +@bp.errorhandler(404) +@bp.app_errorhandler(404) +def page_not_found(e): + print(f'404 error: {e}') + return render_template("404.html"), 404 diff --git a/website/templates/404.html b/website/templates/404.html new file mode 100644 index 0000000..b26dd72 --- /dev/null +++ b/website/templates/404.html @@ -0,0 +1,125 @@ + + + + + + + HNS Login + + + + + + +

HNS Login

+ +

404 - Page Not Found

+ Return to Home + + + +
+ Powered by Varo Auth, HNS.ID and Nathan.Woodburn/ +
+
+ + + \ No newline at end of file diff --git a/website/templates/home.html b/website/templates/home.html index 8f654ab..2646eee 100644 --- a/website/templates/home.html +++ b/website/templates/home.html @@ -62,6 +62,7 @@ text-align: center; text-decoration: none; } + a { color: white; } @@ -77,17 +78,33 @@ .login-option { margin-top: 20px; } + select { - padding: 10px 20px; - font-size: 16px; - background-color: #333; - color: #fff; - border: none; - cursor: pointer; - margin-right: 25px; + padding: 10px 20px; + font-size: 16px; + background-color: #333; + color: #fff; + border: none; + cursor: pointer; + margin-right: 25px; } + .centre { + display: block; + text-align: center; + } + + input { + padding: 10px 20px; + font-size: 16px; + background-color: #333; + color: #fff; + border: none; + cursor: pointer; + margin-right: 25px; + } + @@ -118,16 +135,16 @@ {% endfor %}
{% if user.id == 1 %} - {% for user_tmp in users %} - - 
+  {% for user_tmp in users %}
+
+  
 User Info
     {%- for key in user_tmp %}
     {{ key }}: {{ user_tmp[key] }}
     {%- endfor %}
 

-    

-    {% endfor %}
+  

+  {% endfor %}
   {% endif %}
 
   


@@ -135,17 +152,49 @@
     Contact Nathan.Woodburn/ on any social media platform

 
   {% else %}
+
   
Login with your Handshake domain

-
-
-
-
-  
+  
 
   
+
+    

+      

+        
+        
+    

+  
+
+
+  
+  
+  
-  
+
 
   
 
+  
 
   {% endif %}
 
 
 
   

-    Powered by Varo Auth, HNS.ID and Nathan.Woodburn/
+    Powered by Varo Auth, HNS.ID and Nathan.Woodburn/