Make Vona a Python module

2025-10-01 11:06:58 -04:00
parent 0224909a50
commit eb3c2015c3
16 changed files with 52 additions and 27 deletions
--- a/vona/main.py
+++ b/vona/main.py
@@ -0,0 +1,106 @@
+from flask import Flask, jsonify, request, redirect
+import vona.globals as globals
+from datetime import datetime
+import vona.config as config
+import logging
+
+from vona.federation import server
+from vona.custom import custom
+from vona.identity import identity
+from vona.appservice import apps
+from vona.policy import policy
+from vona.client import client
+
+logging.getLogger("werkzeug").disabled = True
+logging.getLogger("flask").disabled = True
+
+app = Flask("vona")
+
+app.register_blueprint(identity)
+app.register_blueprint(policy)
+app.register_blueprint(client)
+app.register_blueprint(custom)
+app.register_blueprint(server)
+app.register_blueprint(apps)
+
+@app.before_request
+async def preflight():
+	if request.method == "OPTIONS":
+		return "", 204
+
+@app.after_request
+async def handle_logging(response):
+	response.headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
+	response.headers["Access-Control-Allow-Origin"] = "*"
+	response.headers["Access-Control-Allow-Headers"] = "*"
+	response.headers["Access-Control-Allow-Methods"] = "GET, HEAD, POST, PUT, DELETE, OPTIONS"
+
+	if request.method == "OPTIONS":
+		# Discard logs for OPTIONS
+		return response
+
+	origin = "unknown"
+
+	try:
+		if "Authorization" in request.headers:
+			if request.headers["Authorization"].split()[0] == "X-Matrix":
+				origin = request.headers["Authorization"].split('origin="')[1].split('"')[0]
+			else:
+				origin = "client"
+	except:
+		pass
+
+	print(f'[{origin}] [{request.remote_addr}] [{datetime.now().strftime("%d/%b/%Y:%H:%M:%S")}] {request.method} {request.full_path} {response.status_code}')
+
+	return response
+
+
+# Landing page
+@app.route("/")
+async def root():
+	return redirect("/_matrix/static/", 308)
+
+@app.route("/_matrix/static/")
+async def matrix_static():
+	return f'<!DOCTYPE html><html lang="en"><head><title>Vona {globals.vona_version} is running</title><style>body {{font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Oxygen-Sans, Ubuntu, Cantarell, "Helvetica Neue", sans-serif;max-width: 40em;margin: auto;text-align: center;}}h1,p {{margin: 1.5em;}}hr {{border: none;background-color: #ccc;color: #ccc;height: 1px;width: 7em;margin-top: 4em;}}.logo {{display: block;width: 12em;height: auto;margin: 4em auto;}}</style></head><body><img src="https://matrix.org/images/matrix-logo.svg" class="logo"><h1>It works! Vona {globals.vona_version} is running</h1><p>Your Vona server is listening on this port and is ready for messages.</p><p>To use this server you\"ll need <a href="https://matrix.org/ecosystem/clients/" target="_blank"rel="noopener noreferrer">a Matrix client</a>.</p><p>Welcome to the Matrix universe :)</p><hr><p><small><a href="https://natribu.org/en/" target="_blank" rel="noopener noreferrer">matrix.org</a></small></p></body></html>'
+
+
+# Error handlers
+@app.errorhandler(404)
+async def not_found(error):
+	return jsonify({"errcode": "M_UNRECOGNIZED", "error": "Unrecognized request"}), 404
+
+@app.errorhandler(405)
+async def invalid_request_method(error):
+	return jsonify({"errcode": "M_UNRECOGNIZED", "error": "Unrecognized request"}), 405
+
+@app.errorhandler(500)
+async def internal_error(error):
+	return jsonify({"errcode": "M_UNKNOWN", "error": "Internal server error"}), 500
+
+
+# Well-known endpoints for federation,
+# clients, and support information.
+@app.route("/.well-known/matrix/server")
+async def server():
+	return jsonify({"m.server": f"{config.server_name}:443"})
+
+@app.route("/.well-known/matrix/support")
+async def support():
+	if config.support:
+		return jsonify(config.support)
+	else:
+		abort(404)
+
+@app.route("/.well-known/matrix/client")
+async def client():
+	return jsonify({
+		"m.homeserver": {"base_url": f"https://{config.server_name}"},
+		"m.identity_server": {"base_url": f"https://{config.server_name}"},
+	})
+
+
+if __name__ == "__main__":
+	app.run(host=config.addr, port=config.port)
+else:
+	print("What the hell are you doing?")