lemoncurry/micropub/views.py

from django.http import HttpResponse
from django.urls import reverse
from django.utils.decorators import method_decorator
from django.views import View
from django.views.decorators.csrf import csrf_exempt
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 lemonauth import tokens


def form_to_mf2(post):
    properties = {}
    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}


@method_decorator(csrf_exempt, name='dispatch')
class MicropubView(View):
    def post(self, request):
        token = tokens.auth(request)
        if hasattr(token, 'content'):
            return token

        post = request.POST
        if post.get('h') != 'entry':
            return utils.bad_req('only h=entry supported')
        entry = Entry(author=token.user)
        kind = Note
        if 'name' in post:
            entry.name = post['name']
            kind = Article
        if 'content' in post:
            entry.content = post['content']
        if 'in-reply-to' in post:
            entry.in_reply_to = post['in-reply-to']
            kind = Reply
        if 'like-of' in post:
            entry.like_of = post['like-of']
            kind = Like
        if 'repost-of' in post:
            entry.repost_of = post['repost-of']
            kind = Repost

        cats = [
            Cat.objects.from_name(c) for c in
            post.getlist('category') + post.getlist('category[]')
        ]

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

        base = utils.origin(request)
        perma = urljoin(base, entry.url)
        others = [urljoin(base, url) for url in (
            reverse('home:index'),
            reverse('entries:atom'),
            reverse('entries:rss'),
            reverse('entries:' + kind.index),
            reverse('entries:' + kind.atom),
            reverse('entries:' + kind.rss),
        )] + [urljoin(base, cat.url) for cat in cats]
        ping_hub.delay(perma, *others)
        send_mentions.delay(perma)

        res = HttpResponse(status=201)
        res['Location'] = perma
        return res


micropub = MicropubView.as_view()