Nathan Woodburn
f1febd7823
All checks were successful
Build Docker / Build Image (push) Successful in 29s
121 lines
4.2 KiB
Python
121 lines
4.2 KiB
Python
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()
|
|
|
|
# Custom header
|
|
def add_custom_header(response):
|
|
response.headers['Onion-Location'] = 'http://wdbrncwefot4hd7bdrz5rzb74mefay7zvrjn2vmkpdm44l7fwnih5ryd.onion/'
|
|
return response
|
|
app.after_request(add_custom_header)
|
|
|
|
|
|
#Assets routes
|
|
@app.route('/assets/<path:path>')
|
|
def send_report(path):
|
|
return send_from_directory('templates/assets', path)
|
|
|
|
|
|
# Special routes
|
|
@app.route('/links')
|
|
def links():
|
|
return render_template('link.html')
|
|
|
|
@app.route('/sitemap')
|
|
@app.route('/sitemap.xml')
|
|
def sitemap():
|
|
# Remove all .html from sitemap
|
|
with open('templates/sitemap.xml') as file:
|
|
sitemap = file.read()
|
|
|
|
sitemap = sitemap.replace('.html', '')
|
|
return make_response(sitemap, 200, {'Content-Type': 'application/xml'})
|
|
|
|
@app.route('/favicon.png')
|
|
def faviconPNG():
|
|
return send_from_directory('templates/assets/img', 'android-chrome-512x512.png')
|
|
|
|
@app.route('/favicon.ico')
|
|
def favicon():
|
|
return send_from_directory('templates/assets/img', 'favicon.ico')
|
|
|
|
@app.route('/favicon.svg')
|
|
def faviconSVG():
|
|
return send_from_directory('templates/assets/img', 'favicon.svg')
|
|
|
|
@app.route('/https.js')
|
|
@app.route('/handshake.js')
|
|
@app.route('/redirect.js')
|
|
def handshake():
|
|
# return request.path
|
|
return send_from_directory('templates/assets/js', request.path.split('/')[-1])
|
|
|
|
@app.route('/generator/')
|
|
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'):
|
|
return make_response(request.cookies.get('HNS'), 200, {'Content-Type': 'text/plain'})
|
|
|
|
address = requests.get('http://hip02-server:3000')
|
|
# Set cookie
|
|
resp = make_response(address.text, 200, {'Content-Type': 'text/plain'})
|
|
# Cookie should last 1 week
|
|
resp.set_cookie('HNS', address.text, max_age=604800)
|
|
return resp
|
|
|
|
return send_from_directory('.well-known/wallets', path, mimetype='text/plain')
|
|
|
|
|
|
|
|
# Main routes
|
|
@app.route('/')
|
|
def index():
|
|
handshake_scripts = "<script src=\"https://nathan.woodburn/handshake.js\" domain=\"nathan.woodburn\"></script><script src=\"https://nathan.woodburn/https.js\"></script>"
|
|
# If localhost, don't load handshake
|
|
if request.host == "localhost:5000" or request.host == "127.0.0.1:5000" or os.getenv('dev') == "true" or request.host == "test.nathan.woodburn.au":
|
|
handshake_scripts = ""
|
|
|
|
if request.cookies.get('HNS'):
|
|
return render_template('index.html', handshake_scripts=handshake_scripts, HNS=request.cookies.get('HNS'))
|
|
|
|
address = requests.get('http://hip02-server:3000')
|
|
# Set cookie
|
|
resp = make_response(render_template('index.html', handshake_scripts=handshake_scripts, HNS=address.text), 200, {'Content-Type': 'text/html'})
|
|
# Cookie should last 1 week
|
|
resp.set_cookie('HNS', address.text, max_age=604800)
|
|
return resp
|
|
|
|
|
|
@app.route('/<path:path>')
|
|
def catch_all(path):
|
|
handshake_scripts = "<script src=\"https://nathan.woodburn/handshake.js\" domain=\"nathan.woodburn\"></script><script src=\"https://nathan.woodburn/https.js\"></script>"
|
|
# If localhost, don't load handshake
|
|
if request.host == "localhost:5000" or request.host == "127.0.0.1:5000" or os.getenv('dev') == "true" or request.host == "test.nathan.woodburn.au":
|
|
handshake_scripts = ""
|
|
# If file exists, load it
|
|
if os.path.isfile('templates/' + path):
|
|
return render_template(path, handshake_scripts=handshake_scripts)
|
|
|
|
# Try with .html
|
|
if os.path.isfile('templates/' + path + '.html'):
|
|
return render_template(path + '.html', handshake_scripts=handshake_scripts)
|
|
|
|
return render_template('404.html'), 404
|
|
|
|
# 404 catch all
|
|
@app.errorhandler(404)
|
|
def not_found(e):
|
|
return render_template('404.html'), 404
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=False, port=5000, host='0.0.0.0') |