Nathanwoodburn.github.io/server.py

140 lines
4.9 KiB
Python
Raw Normal View History

2023-11-02 20:26:15 +11:00
from flask import Flask, make_response, redirect, request, jsonify, render_template, send_from_directory
import os
import dotenv
2023-11-02 21:51:49 +11:00
import requests
2023-11-02 20:26:15 +11:00
app = Flask(__name__)
dotenv.load_dotenv()
2023-11-02 21:06:43 +11:00
# Custom header
def add_custom_header(response):
response.headers['Onion-Location'] = 'http://wdbrncwefot4hd7bdrz5rzb74mefay7zvrjn2vmkpdm44l7fwnih5ryd.onion/'
return response
app.after_request(add_custom_header)
2023-11-02 20:45:36 +11:00
#Assets routes
2023-11-02 20:26:15 +11:00
@app.route('/assets/<path:path>')
def send_report(path):
return send_from_directory('templates/assets', path)
2023-11-02 20:45:36 +11:00
# 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])
2023-11-02 21:37:18 +11:00
@app.route('/generator/')
def removeTrailingSlash():
return render_template(request.path.split('/')[-2] + '.html')
2023-11-02 20:45:36 +11:00
2023-11-02 21:51:49 +11:00
@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')
2023-11-02 21:51:49 +11:00
# Set cookie
resp = make_response(address.text, 200, {'Content-Type': 'text/plain'})
2023-11-02 21:53:23 +11:00
# Cookie should last 1 week
resp.set_cookie('HNS', address.text, max_age=604800)
2023-11-02 21:51:49 +11:00
return resp
return send_from_directory('.well-known/wallets', path, mimetype='text/plain')
2023-11-02 20:45:36 +11:00
# Main routes
2023-11-02 20:26:15 +11:00
@app.route('/')
def index():
2023-11-03 11:51:10 +11:00
git=requests.get('https://git.woodburn.au/api/v1/users/nathanwoodburn/activities/feeds?only-performed-by=true&limit=1&token=' + os.getenv('git_token'))
git = git.json()
git = git[0]
repo_name=git['repo']['name']
repo_name=repo_name.lower()
repo_description=git['repo']['description']
# Special names
if repo_name == "nathanwoodburn.github.io":
repo_name = "Nathan.Woodburn/"
html_url=git['repo']['html_url']
repo = "<a href=\"" + html_url + "\" target=\"_blank\">" + repo_name + "</a>"
2023-11-02 20:26:15 +11:00
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
2023-11-02 21:37:18 +11:00
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":
2023-11-02 20:26:15 +11:00
handshake_scripts = ""
2023-11-03 11:51:10 +11:00
2023-11-02 22:12:15 +11:00
if request.cookies.get('HNS'):
2023-11-03 11:51:10 +11:00
return render_template('index.html', handshake_scripts=handshake_scripts, HNS=request.cookies.get('HNS'), repo=repo, repo_description=repo_description)
if handshake_scripts == "":
address = "hs1............example"
else:
address = requests.get('http://hip02-server:3000').text
2023-11-02 22:12:15 +11:00
# Set cookie
2023-11-03 11:51:10 +11:00
resp = make_response(render_template('index.html', handshake_scripts=handshake_scripts, HNS=address, repo=repo, repo_description=repo_description), 200, {'Content-Type': 'text/html'})
2023-11-02 22:12:15 +11:00
# Cookie should last 1 week
2023-11-03 11:51:10 +11:00
resp.set_cookie('HNS', address, max_age=604800)
2023-11-02 22:12:15 +11:00
return resp
2023-11-02 20:26:15 +11:00
2023-11-02 20:45:36 +11:00
2023-11-02 20:26:15 +11:00
@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
2023-11-02 21:37:18 +11:00
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":
2023-11-02 20:26:15 +11:00
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')