From 66acd69bec254c500a4dc7fbcca198fad7f3db27 Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Thu, 2 Nov 2023 21:51:49 +1100
Subject: [PATCH] feat: Add wallet addresses

---
 requirements.txt |  3 ++-
 server.py        | 23 +++++++++++++++++++++++
 2 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 886c3db..f640785 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
 flask
 python-dotenv
-gunicorn
\ No newline at end of file
+gunicorn
+requests
\ No newline at end of file
diff --git a/server.py b/server.py
index 0abf763..280043c 100644
--- a/server.py
+++ b/server.py
@@ -1,6 +1,7 @@
 from flask import Flask, make_response, redirect, request, jsonify, render_template, send_from_directory
 import os
 import dotenv
+import requests
 
 app = Flask(__name__)
 dotenv.load_dotenv()
@@ -56,6 +57,28 @@ def handshake():
 def removeTrailingSlash():
     return render_template(request.path.split('/')[-2] + '.html')
 
+@app.route('/.well-known/wallets/<path:path>')
+def wallet(path):
+    # If HNS, redirect to HNS wallet
+    if path == "HNS":
+        # Get from 100.66.107.77:8080 then return result
+        # Check for cookie
+        if request.cookies.get('HNS'):
+            print(request.cookies.get('HNS'))
+            return make_response(request.cookies.get('HNS'), 200, {'Content-Type': 'text/plain'})
+        
+        address = requests.get('http://100.66.107.77:8080')
+        # Set cookie
+        resp = make_response(address.text, 200, {'Content-Type': 'text/plain'})
+        resp.set_cookie('HNS', address.text)
+        return resp
+
+
+
+    return send_from_directory('.well-known/wallets', path, mimetype='text/plain')
+        
+
+
 # Main routes
 @app.route('/')
 def index():