FireTax/server.py

from functools import cache
import json
from flask import (
    Flask,
    make_response,
    redirect,
    request,
    jsonify,
    render_template,
    send_from_directory,
    send_file,
)
import os
import json
import requests
from datetime import datetime
import dotenv
import chain_module

dotenv.load_dotenv()

app = Flask(__name__)


def find(name, path):
    for root, dirs, files in os.walk(path):
        if name in files:
            return os.path.join(root, name)

# Assets routes
@app.route("/assets/<path:path>")
def send_assets(path):
    if path.endswith(".json"):
        return send_from_directory(
            "templates/assets", path, mimetype="application/json"
        )

    if os.path.isfile("templates/assets/" + path):
        return send_from_directory("templates/assets", path)

    # Try looking in one of the directories
    filename: str = path.split("/")[-1]
    if (
        filename.endswith(".png")
        or filename.endswith(".jpg")
        or filename.endswith(".jpeg")
        or filename.endswith(".svg")
    ):
        if os.path.isfile("templates/assets/img/" + filename):
            return send_from_directory("templates/assets/img", filename)
        if os.path.isfile("templates/assets/img/favicon/" + filename):
            return send_from_directory("templates/assets/img/favicon", filename)

    return render_template("404.html"), 404


# region Special routes
@app.route("/favicon.png")
def faviconPNG():
    return send_from_directory("templates/assets/img", "favicon.png")


@app.route("/.well-known/<path:path>")
def wellknown(path):
    # Try to proxy to https://nathan.woodburn.au/.well-known/
    req = requests.get(f"https://nathan.woodburn.au/.well-known/{path}")
    return make_response(
        req.content, 200, {"Content-Type": req.headers["Content-Type"]}
    )


# endregion


# region Main routes
@app.route("/")
def index():
    # List chains
    chains = chain_module.listChains()
    addresses = chain_module.getAllAddresses()

    return render_template("index.html",chains=chains,addresses=addresses)

@app.route("/chains/<path:path>")
def chains(path: str):
    path = path.lower()
    # Check if the chain exists
    if not chain_module.chainExists(path):
        return render_template("404.html"), 404
    
    # Get chain info
    chain = chain_module.getChainData(path)
    transactions = chain_module.getTransactionsRender(path)
    addresses = chain_module.getAddresses(path)

    return render_template("chain.html",chain=chain['name'],transactions=transactions,addresses=addresses)

@app.route("/chains/<path:path>", methods=["POST"])
def chainsPost(path: str):
    path = path.lower()
    # Check if the chain exists
    if not chain_module.chainExists(path):
        return jsonify({"error": "Chain not found"}), 404
    
    # Check if the address is valid
    address = request.form['address']
    if not chain_module.validateChainAddress(path, address):
        return jsonify({"error": "Invalid address"}), 400
    
    if not chain_module.importAddress(path, address):
        return jsonify({"error": "Error importing address"}), 400
    return redirect('/chains/'+path)

@app.route("/chains/<path:path>/delete/<address>")
def chainsDelete(path: str, address: str):
    path = path.lower()
    # Check if the chain exists
    if not chain_module.chainExists(path):
        return render_template("404.html"), 404
    
    if not chain_module.deleteAddress(path, address):
        return jsonify({"error": "Error deleting address"}), 400
    return redirect('/chains/'+path)

@app.route("/chains/<path:path>/addAPIKey", methods=["POST"])
def chainsAddAPIKey(path: str):
    path = path.lower()
    # Check if the chain exists
    if not chain_module.chainExists(path):
        return jsonify({"error": "Chain not found"}), 404
    
    apiKey = request.form['apiKey']
    if not chain_module.addAPIKey(path, apiKey):
        return jsonify({"error": "Error adding API key"}), 400
    return redirect('/chains/'+path)
    

@app.route("/chains/sync")
def chainsSync():
    chain_module.syncChains()
    return redirect('/')

@app.route("/chains/sync/<path:path>")
def chainSync(path: str):
    path = path.lower()
    chain_module.syncChain(path)
    return redirect('/chains/'+path)


@app.route("/<path:path>")
def catch_all(path: str):
    if os.path.isfile("templates/" + path):
        return render_template(path)

    # Try with .html
    if os.path.isfile("templates/" + path + ".html"):
        return render_template(path + ".html")

    if os.path.isfile("templates/" + path.strip("/") + ".html"):
        return render_template(path.strip("/") + ".html")

    # Try to find a file matching
    if path.count("/") < 1:
        # Try to find a file matching
        filename = find(path, "templates")
        if filename:
            return send_file(filename)

    return render_template("404.html"), 404


# endregion


# region Error Catching
# 404 catch all
@app.errorhandler(404)
def not_found(e):
    return render_template("404.html"), 404


# endregion
if __name__ == "__main__":
    app.run(debug=True, port=5000, host="0.0.0.0")