''' what are we here for if not for posting ''' import re from django.contrib.auth.decorators import login_required from django.http import HttpResponseBadRequest from django.shortcuts import get_object_or_404, redirect from django.utils.decorators import method_decorator from django.views import View from markdown import markdown from bookwyrm import forms, models from bookwyrm.sanitize_html import InputHtmlParser from bookwyrm.settings import DOMAIN from bookwyrm.status import delete_status from bookwyrm.utils import regex from .helpers import handle_remote_webfinger # pylint: disable= no-self-use @method_decorator(login_required, name='dispatch') class CreateStatus(View): ''' the view for *posting* ''' def post(self, request, status_type): ''' create status of whatever type ''' status_type = status_type[0].upper() + status_type[1:] try: form = getattr(forms, '%sForm' % status_type)(request.POST) except AttributeError: return HttpResponseBadRequest() if not form.is_valid(): return redirect(request.headers.get('Referer', '/')) status = form.save(commit=False) if not status.sensitive and status.content_warning: # the cw text field remains populated when you click "remove" status.content_warning = None status.save(broadcast=False) # inspect the text for user tags content = status.content for (mention_text, mention_user) in find_mentions(content): # add them to status mentions fk status.mention_users.add(mention_user) # turn the mention into a link content = re.sub( r'%s([^@]|$)' % mention_text, r'%s\g<1>' % \ (mention_user.remote_id, mention_text), content) # add reply parent to mentions if status.reply_parent: status.mention_users.add(status.reply_parent.user) # deduplicate mentions status.mention_users.set(set(status.mention_users.all())) # don't apply formatting to generated notes if not isinstance(status, models.GeneratedNote): status.content = to_markdown(content) # do apply formatting to quotes if hasattr(status, 'quote'): status.quote = to_markdown(status.quote) status.save(created=True) return redirect(request.headers.get('Referer', '/')) class DeleteStatus(View): ''' tombstone that bad boy ''' def post(self, request, status_id): ''' delete and tombstone a status ''' status = get_object_or_404(models.Status, id=status_id) # don't let people delete other people's statuses if status.user != request.user: return HttpResponseBadRequest() # perform deletion delete_status(status) return redirect(request.headers.get('Referer', '/')) def find_mentions(content): ''' detect @mentions in raw status content ''' for match in re.finditer(regex.strict_username, content): username = match.group().strip().split('@')[1:] if len(username) == 1: # this looks like a local user (@user), fill in the domain username.append(DOMAIN) username = '@'.join(username) mention_user = handle_remote_webfinger(username) if not mention_user: # we can ignore users we don't know about continue yield (match.group(), mention_user) def format_links(content): ''' detect and format links ''' return re.sub( r'([^(href=")]|^|\()(https?:\/\/(%s([\w\.\-_\/+&\?=:;,])*))' % \ regex.domain, r'\g<1>\g<3>', content) def to_markdown(content): ''' catch links and convert to markdown ''' content = markdown(content) content = format_links(content) # sanitize resulting html sanitizer = InputHtmlParser() sanitizer.feed(content) return sanitizer.get_output()