from flask import Flask, make_response, redirect, request import os import dotenv import requests import hip02 app = Flask(__name__) dotenv.load_dotenv() HSD_IP = '10.2.1.15' HSD_PORT = 5350 if os.getenv('HSD_IP'): HSD_IP = os.getenv('HSD_IP') if os.getenv('HSD_PORT'): HSD_PORT = os.getenv('HSD_PORT') @app.route('/') def index(): return redirect("https://nathan.woodburn.au") # Special routes @app.route('/.well-known/wallets/') def send_wallet(token): address = requests.get('https://nathan.woodburn.au/.well-known/wallets/'+token).text return make_response(address, 200, {'Content-Type': 'text/plain'}) @app.route('/favicon.ico') def favicon(): return redirect('https://nathan.woodburn.au/favicon.ico') @app.route('/.json') def jsonlookup(path): TLSA = hip02.TLSA_check(HSD_IP,HSD_PORT,path) if not TLSA: return make_response({"success":False,"error":TLSA}, 200, {'Content-Type': 'application/json'}) token = "HNS" if 'token' in request.args: token = request.args['token'].upper() hip2 = hip02.resolve(HSD_IP, HSD_PORT, path,token) if not hip2: return make_response({"success":False,"error":hip2}, 200, {'Content-Type': 'application/json'}) return make_response({"success":True,"address":hip2}, 200, {'Content-Type': 'application/json'}) @app.route('/') def lookup(path): TLSA = hip02.TLSA_check(HSD_IP,HSD_PORT,path) if not TLSA: return make_response(TLSA, 200, {'Content-Type': 'text/plain'}) token = "HNS" if 'token' in request.args: token = request.args['token'].upper() return make_response(hip02.resolve(HSD_IP, HSD_PORT, path,token), 200, {'Content-Type': 'text/plain'}) # 404 catch all @app.errorhandler(404) def not_found(e): return redirect('/') if __name__ == '__main__': app.run(debug=False, port=5000, host='0.0.0.0')