forked from mirrors/bookwyrm
116 lines
4 KiB
Python
116 lines
4 KiB
Python
''' 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'<a href="%s">%s</a>\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) and content:
|
|
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 '''
|
|
if not content:
|
|
return
|
|
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><a href="\g<2>">\g<3></a>',
|
|
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()
|