Nathanwoodburn.github.io/server.py

212 lines
7.3 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-12-22 15:07:24 +11:00
import CloudFlare
2023-11-02 20:26:15 +11:00
app = Flask(__name__)
dotenv.load_dotenv()
address = ''
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):
2024-02-18 12:51:59 +11:00
if path.endswith('.json'):
return send_from_directory('templates/assets', path, mimetype='application/json')
2023-11-02 20:26:15 +11:00
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 = getAddress()
2023-11-02 21:51:49 +11:00
# Set cookie
2023-12-05 21:54:18 +11:00
resp = make_response(address, 200, {'Content-Type': 'text/plain'})
2023-11-02 21:53:23 +11:00
# Cookie should last 1 week
2023-12-05 21:54:18 +11:00
resp.set_cookie('HNS', address, 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-12-18 15:02:13 +11:00
global address
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']
2023-11-03 14:11:34 +11:00
custom = ""
# Check for downtime
uptime = requests.get('https://uptime.woodburn.au/api/status-page/main/badge')
uptime = uptime.content.count(b'Up') > 1
if uptime:
custom += "<style>#downtime{display:none !important;}</style>"
else:
custom += "<style>#downtime{opacity:1;}</style>"
2023-11-03 11:51:10 +11:00
# 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 14:11:34 +11:00
return render_template('index.html', handshake_scripts=handshake_scripts, HNS=request.cookies.get('HNS'), repo=repo, repo_description=repo_description, custom=custom)
2023-11-03 11:51:10 +11:00
2023-12-18 15:02:13 +11:00
if address == '':
address = getAddress()
2023-11-02 22:12:15 +11:00
# Set cookie
2023-11-03 14:11:34 +11:00
resp = make_response(render_template('index.html', handshake_scripts=handshake_scripts, HNS=address, repo=repo, repo_description=repo_description, custom=custom), 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
2024-02-18 21:47:35 +11:00
@app.route('/now')
def now():
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 = ""
# Get latest now page
files = os.listdir('templates/now')
# Remove template
files = [file for file in files if file != 'template.html']
files.sort(reverse=True)
return render_template('now/' + files[0], handshake_scripts=handshake_scripts)
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
def getAddress():
global address
if address == '':
address = 'hs1qv3uu4amv87g7p7h49xez2pmzwjf92am0wzpnh4'
# address = requests.get('http://hip02-server:3000').text?
return address
2023-12-22 15:07:24 +11:00
@app.route('/hnsdoh-acme', methods=['POST'])
def hnsdoh_acme():
# Get the TXT record from the request
if not request.json:
return jsonify({'status': 'error', 'error': 'No JSON data provided'})
if 'txt' not in request.json or 'auth' not in request.json:
return jsonify({'status': 'error', 'error': 'Missing required data'})
txt = request.json['txt']
auth = request.json['auth']
if auth != os.getenv('CF_AUTH'):
return jsonify({'status': 'error', 'error': 'Invalid auth'})
cf = CloudFlare.CloudFlare(token=os.getenv('CF_TOKEN'))
zone = cf.zones.get(params={'name': 'hnsdoh.com'})
zone_id = zone[0]['id']
existing_records = cf.zones.dns_records.get(zone_id, params={'type': 'TXT', 'name': '_acme-challenge.hnsdoh.com'})
# Delete existing TXT records
for record in existing_records:
print(record)
record_id = record['id']
cf.zones.dns_records.delete(zone_id, record_id)
record = cf.zones.dns_records.post(zone_id, data={'type': 'TXT', 'name': '_acme-challenge', 'content': txt})
print(record)
return jsonify({'status': 'success'})
2023-11-02 20:26:15 +11:00
# 404 catch all
@app.errorhandler(404)
def not_found(e):
return render_template('404.html'), 404
if __name__ == '__main__':
2023-12-18 15:02:13 +11:00
app.run(debug=True, port=5000, host='0.0.0.0')