lemoncurry/micropub/views/create.py

from django.urls import reverse
from django.http import HttpResponse
from urllib.parse import urljoin

from entries.jobs import ping_hub, send_mentions
from entries.models import Cat, Entry
from entries.kinds import Article, Note, Reply, Like, Repost
from lemoncurry import utils

from .. import error


def form_to_mf2(request):
    properties = {}
    post = request.POST
    for key in post.keys():
        if key.endswith("[]"):
            key = key[:-2]
        if key == "access_token":
            continue
        properties[key] = post.getlist(key) + post.getlist(key + "[]")

    type = []
    if "h" in properties:
        type = ["h-" + p for p in properties["h"]]
        del properties["h"]
    return {"type": type, "properties": properties}


def create(request):
    normalise = {
        "application/json": lambda r: r.json,
        "application/x-www-form-urlencoded": form_to_mf2,
    }
    if "create" not in request.token:
        raise error.bad_scope("create")
    if request.content_type not in normalise:
        raise error.unsupported_type(request.content_type)
    body = normalise[request.content_type](request)
    if "type" not in body:
        raise error.bad_req("mf2 object type required")
    if body["type"] != ["h-entry"]:
        raise error.bad_req("only h-entry supported")

    entry = Entry(author=request.token.user)
    props = body.get("properties", {})
    kind = Note
    if "name" in props:
        entry.name = "\n".join(props["name"])
        kind = Article
    if "content" in props:
        entry.content = "\n".join(
            c if isinstance(c, str) else c["html"] for c in props["content"]
        )
    if "in-reply-to" in props:
        entry.in_reply_to = props["in-reply-to"]
        kind = Reply
    if "like-of" in props:
        entry.like_of = props["like-of"]
        kind = Like
    if "repost-of" in props:
        entry.repost_of = props["repost-of"]
        kind = Repost

    cats = [Cat.objects.from_name(c) for c in props.get("category", [])]

    entry.kind = kind.id
    entry.save()
    entry.cats.set(cats)
    entry.save()

    for url in props.get("syndication", []):
        entry.syndications.create(url=url)

    base = utils.origin(request)
    perma = urljoin(base, entry.url)
    short = urljoin(base, entry.short_url)
    ping_hub.delay(*entry.affected_urls)
    send_mentions.delay(perma)

    res = HttpResponse(status=201)
    res["Location"] = perma
    res["Link"] = '<{}>; rel="shortlink"'.format(short)
    return res