forked from mirrors/bookwyrm
118 lines
4 KiB
Python
118 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):
|
|
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><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()
|