forked from mirrors/bookwyrm
273 lines
9 KiB
Python
273 lines
9 KiB
Python
""" what are we here for if not for posting """
|
|
import re
|
|
import logging
|
|
from urllib.parse import urlparse
|
|
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.core.validators import URLValidator
|
|
from django.core.exceptions import ValidationError
|
|
from django.http import HttpResponse, HttpResponseBadRequest, Http404
|
|
from django.shortcuts import get_object_or_404, redirect
|
|
from django.template.response import TemplateResponse
|
|
from django.utils import timezone
|
|
from django.utils.decorators import method_decorator
|
|
from django.views import View
|
|
from django.views.decorators.http import require_POST
|
|
|
|
from markdown import markdown
|
|
from bookwyrm import forms, models
|
|
from bookwyrm.sanitize_html import InputHtmlParser
|
|
from bookwyrm.settings import DOMAIN
|
|
from bookwyrm.utils import regex
|
|
from .helpers import handle_remote_webfinger, is_api_request
|
|
from .helpers import load_date_in_user_tz_as_utc
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# pylint: disable= no-self-use
|
|
@method_decorator(login_required, name="dispatch")
|
|
class EditStatus(View):
|
|
"""the view for *posting*"""
|
|
|
|
def get(self, request, status_id): # pylint: disable=unused-argument
|
|
"""load the edit panel"""
|
|
status = get_object_or_404(
|
|
models.Status.objects.select_subclasses(), id=status_id
|
|
)
|
|
status.raise_not_editable(request.user)
|
|
|
|
status_type = "reply" if status.reply_parent else status.status_type.lower()
|
|
data = {
|
|
"type": status_type,
|
|
"book": getattr(status, "book", None),
|
|
"draft": status,
|
|
}
|
|
return TemplateResponse(request, "compose.html", data)
|
|
|
|
|
|
# pylint: disable= no-self-use
|
|
@method_decorator(login_required, name="dispatch")
|
|
class CreateStatus(View):
|
|
"""the view for *posting*"""
|
|
|
|
def get(self, request, status_type): # pylint: disable=unused-argument
|
|
"""compose view (...not used?)"""
|
|
book = get_object_or_404(models.Edition, id=request.GET.get("book"))
|
|
data = {"book": book}
|
|
return TemplateResponse(request, "compose.html", data)
|
|
|
|
# pylint: disable=too-many-branches
|
|
def post(self, request, status_type, existing_status_id=None):
|
|
"""create status of whatever type"""
|
|
created = not existing_status_id
|
|
existing_status = None
|
|
if existing_status_id:
|
|
existing_status = get_object_or_404(
|
|
models.Status.objects.select_subclasses(), id=existing_status_id
|
|
)
|
|
existing_status.raise_not_editable(request.user)
|
|
existing_status.edited_date = timezone.now()
|
|
|
|
status_type = status_type[0].upper() + status_type[1:]
|
|
|
|
try:
|
|
form = getattr(forms, f"{status_type}Form")(
|
|
request.POST, instance=existing_status
|
|
)
|
|
except AttributeError as err:
|
|
logger.exception(err)
|
|
return HttpResponseBadRequest()
|
|
|
|
if not form.is_valid():
|
|
if is_api_request(request):
|
|
logger.exception(form.errors)
|
|
return HttpResponseBadRequest()
|
|
return redirect(request.headers.get("Referer", "/"))
|
|
|
|
status = form.save(commit=False)
|
|
# save the plain, unformatted version of the status for future editing
|
|
status.raw_content = status.content
|
|
if hasattr(status, "quote"):
|
|
status.raw_quote = status.quote
|
|
|
|
status.sensitive = status.content_warning not in [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(
|
|
rf"{mention_text}([^@]|$)",
|
|
rf'<a href="{mention_user.remote_id}">{mention_text}</a>\g<1>',
|
|
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=created)
|
|
|
|
# update a readthrough, if needed
|
|
if bool(request.POST.get("id")):
|
|
try:
|
|
edit_readthrough(request)
|
|
except Http404:
|
|
pass
|
|
|
|
if is_api_request(request):
|
|
return HttpResponse()
|
|
return redirect("/")
|
|
|
|
|
|
@method_decorator(login_required, name="dispatch")
|
|
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
|
|
status.raise_not_deletable(request.user)
|
|
|
|
# perform deletion
|
|
status.delete()
|
|
return redirect(request.headers.get("Referer", "/"))
|
|
|
|
|
|
@login_required
|
|
@require_POST
|
|
def update_progress(request, book_id): # pylint: disable=unused-argument
|
|
"""Either it's just a progress update, or it's a comment with a progress update"""
|
|
if request.POST.get("post-status"):
|
|
return CreateStatus.as_view()(request, "comment")
|
|
return edit_readthrough(request)
|
|
|
|
|
|
@login_required
|
|
@require_POST
|
|
def edit_readthrough(request):
|
|
"""can't use the form because the dates are too finnicky"""
|
|
# TODO: remove this, it duplicates the code in the ReadThrough view
|
|
readthrough = get_object_or_404(models.ReadThrough, id=request.POST.get("id"))
|
|
readthrough.raise_not_editable(request.user)
|
|
|
|
readthrough.start_date = load_date_in_user_tz_as_utc(
|
|
request.POST.get("start_date"), request.user
|
|
)
|
|
readthrough.finish_date = load_date_in_user_tz_as_utc(
|
|
request.POST.get("finish_date"), request.user
|
|
)
|
|
|
|
progress = request.POST.get("progress")
|
|
try:
|
|
progress = int(progress)
|
|
readthrough.progress = progress
|
|
except (ValueError, TypeError):
|
|
pass
|
|
|
|
progress_mode = request.POST.get("progress_mode")
|
|
try:
|
|
progress_mode = models.ProgressMode(progress_mode)
|
|
readthrough.progress_mode = progress_mode
|
|
except ValueError:
|
|
pass
|
|
|
|
readthrough.save()
|
|
|
|
# record the progress update individually
|
|
# use default now for date field
|
|
readthrough.create_update()
|
|
|
|
if is_api_request(request):
|
|
return HttpResponse()
|
|
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"""
|
|
validator = URLValidator()
|
|
formatted_content = ""
|
|
split_content = re.split(r"(\s+)", content)
|
|
|
|
for potential_link in split_content:
|
|
if not potential_link:
|
|
continue
|
|
wrapped = _wrapped(potential_link)
|
|
if wrapped:
|
|
wrapper_close = potential_link[-1]
|
|
formatted_content += potential_link[0]
|
|
potential_link = potential_link[1:-1]
|
|
|
|
try:
|
|
# raises an error on anything that's not a valid link
|
|
validator(potential_link)
|
|
|
|
# use everything but the scheme in the presentation of the link
|
|
url = urlparse(potential_link)
|
|
link = url.netloc + url.path + url.params
|
|
if url.query != "":
|
|
link += "?" + url.query
|
|
if url.fragment != "":
|
|
link += "#" + url.fragment
|
|
|
|
formatted_content += f'<a href="{potential_link}">{link}</a>'
|
|
except (ValidationError, UnicodeError):
|
|
formatted_content += potential_link
|
|
|
|
if wrapped:
|
|
formatted_content += wrapper_close
|
|
|
|
return formatted_content
|
|
|
|
|
|
def _wrapped(text):
|
|
"""check if a line of text is wrapped"""
|
|
wrappers = [("(", ")"), ("[", "]"), ("{", "}")]
|
|
for wrapper in wrappers:
|
|
if text[0] == wrapper[0] and text[-1] == wrapper[-1]:
|
|
return True
|
|
return False
|
|
|
|
|
|
def to_markdown(content):
|
|
"""catch links and convert to markdown"""
|
|
content = format_links(content)
|
|
content = markdown(content)
|
|
# sanitize resulting html
|
|
sanitizer = InputHtmlParser()
|
|
sanitizer.feed(content)
|
|
return sanitizer.get_output()
|