COSCUP-ap-demo/app.py

#!/usr/bin/env python3
"""Creatr App"""
import sys

from flask import Flask, Response, request, Request, abort, jsonify

from loguru import logger

from demo.utils.checker import inbox_prechecker
from demo.activitypub import ME
from demo import config


logger.remove()
logger.add(sys.stdout, level="DEBUG")
logger.add("output.log", level="DEBUG")

app = Flask(__name__,
            static_folder="demo/static",)

def is_ap_requested(ap_request: Request) -> bool:
    """Check request accept headers."""
    accept_str = ap_request.headers.get("accept")
    if accept_str is None:
        return False
    for i in [
        "application/activity+json",
        "application/ld+json",
    ]:
        if accept_str.startswith(i):
            return True
    return False


@app.route('/')
def index():
    """Show index page"""
    return "Hello Fediverse!"


@app.route("/inbox", methods=["POST"])
def inbox():
    """Process inbox request"""
    is_verify = inbox_prechecker(request)
    if is_verify:
        return "STUB"
    return "STUB"


@app.route("/.well-known/nodeinfo")
def well_known_nodeinfo() -> Response:
    """Return nodeinfo path."""
    return jsonify(
        {
            "links": [
                {
                    "rel": "http://nodeinfo.diaspora.software/ns/schema/2.0",
                    "href": f"{config.BASE_URL}/nodeinfo/2.0",
                }
            ]
        }
    )


@app.get("/nodeinfo/2.0")
def nodeinfo() -> Response:
    """Return nodeinfo."""
    return jsonify(
        {
            "version": "2.0",
            "software": {
                "name": "COSCUP-demo",
                "version": "0.0.1",
            },
            "protocols": ["activitypub"],
            "services": {"inbound": [], "outbound": []},
            "usage": {"users": {"total": 1}},
            "openRegistrations": False,
            "metadata": {},
        },
    )


@app.route("/.well-known/webfinger")
def wellknown_webfinger() -> Response:
    """Exposes servers WebFinger data."""
    resource = request.args.get("resource")

    if resource not in [f"acct:{config.USERNAME}@{config.DOMAIN}", config.ID]:
        abort(404)

    resp = jsonify(
        {
            "subject": f"acct:{config.USERNAME}@{config.DOMAIN}",
            "aliases": [config.ID],
            "links": [
                {
                    "rel": "http://webfinger.net/rel/profile-page",
                    "type": "text/html",
                    "href": config.ID,
                },
                {
                    "rel": "self",
                    "type": "application/activity+json",
                    "href": config.ID,
                },
            ],
        },
    )
    resp.headers["Access-Control-Allow-Origin"]  = "*"
    resp.headers["Content-Type"] = "application/jrd+json; charset=utf-8"

    return resp


@app.route(f"/user/{config.USERNAME}")
def locate_user() -> Response:
    """Return user ActivityPub response."""
    resp = jsonify(ME)
    resp.headers["Content-Type"] = "application/activity+json"

    return resp