107 lines
3.8 KiB
Python
107 lines
3.8 KiB
Python
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?")
|