NDOJ/judge/views/ticket.py

import json

from itertools import chain

from django import forms
from django.contrib.auth.mixins import LoginRequiredMixin
from django.core.exceptions import (
    ImproperlyConfigured,
    PermissionDenied,
    ValidationError,
)
from django.http import (
    Http404,
    HttpResponse,
    HttpResponseBadRequest,
    HttpResponseRedirect,
    JsonResponse,
)
from django.shortcuts import get_object_or_404
from django.template.defaultfilters import truncatechars
from django.template.loader import get_template
from django.urls import reverse, reverse_lazy
from django.utils.functional import cached_property
from django.utils.html import escape, format_html, linebreaks
from django.utils.safestring import mark_safe
from django.utils.translation import gettext as _, gettext_lazy
from django.views import View
from django.views.generic import ListView
from django.views.generic.detail import SingleObjectMixin

from judge import event_poster as event
from judge.models import Problem, Profile, Ticket, TicketMessage, Notification
from judge.utils.diggpaginator import DiggPaginator
from judge.utils.tickets import filter_visible_tickets, own_ticket_filter
from judge.utils.views import SingleObjectFormView, TitleMixin, paginate_query_context
from judge.views.problem import ProblemMixin
from judge.widgets import HeavyPreviewPageDownWidget

ticket_widget = (
    forms.Textarea()
    if HeavyPreviewPageDownWidget is None
    else HeavyPreviewPageDownWidget(
        preview=reverse_lazy("ticket_preview"),
        preview_timeout=1000,
        hide_preview_button=True,
    )
)


def add_ticket_notifications(users, author, link, ticket):
    html = f'<a href="{link}">{ticket.linked_item}</a>'

    users = set(users)
    if author in users:
        users.remove(author)

    for user in users:
        notification = Notification(
            owner=user, html_link=html, category="Ticket", author=author
        )
        notification.save()


class TicketForm(forms.Form):
    title = forms.CharField(max_length=100, label=gettext_lazy("Ticket title"))
    body = forms.CharField(widget=ticket_widget)

    def __init__(self, request, *args, **kwargs):
        self.request = request
        super(TicketForm, self).__init__(*args, **kwargs)
        self.fields["title"].widget.attrs.update({"placeholder": _("Ticket title")})
        self.fields["body"].widget.attrs.update({"placeholder": _("Issue description")})

    def clean(self):
        if self.request is not None and self.request.user.is_authenticated:
            profile = self.request.profile
            if profile.mute:
                raise ValidationError(_("Your part is silent, little toad."))
        return super(TicketForm, self).clean()


class NewTicketView(LoginRequiredMixin, SingleObjectFormView):
    form_class = TicketForm
    template_name = "ticket/new.html"

    def get_assignees(self):
        return []

    def get_form_kwargs(self):
        kwargs = super(NewTicketView, self).get_form_kwargs()
        kwargs["request"] = self.request
        return kwargs

    def form_valid(self, form):
        ticket = Ticket(user=self.request.profile, title=form.cleaned_data["title"])
        ticket.linked_item = self.object
        ticket.save()
        message = TicketMessage(
            ticket=ticket, user=ticket.user, body=form.cleaned_data["body"]
        )
        message.save()
        ticket.assignees.set(self.get_assignees())

        link = reverse("ticket", args=[ticket.id])

        add_ticket_notifications(ticket.assignees.all(), ticket.user, link, ticket)

        if event.real:
            event.post(
                "tickets",
                {
                    "type": "new-ticket",
                    "id": ticket.id,
                    "message": message.id,
                    "user": ticket.user_id,
                    "assignees": list(ticket.assignees.values_list("id", flat=True)),
                },
            )
        return HttpResponseRedirect(link)


class NewProblemTicketView(ProblemMixin, TitleMixin, NewTicketView):
    template_name = "ticket/new_problem.html"

    def get_assignees(self):
        return self.object.authors.all()

    def get_title(self):
        return _("New ticket for %s") % self.object.name

    def get_content_title(self):
        return mark_safe(
            escape(_("New ticket for %s"))
            % format_html(
                '<a href="{0}">{1}</a>',
                reverse("problem_detail", args=[self.object.code]),
                self.object.translated_name(self.request.LANGUAGE_CODE),
            )
        )

    def form_valid(self, form):
        if not self.object.is_accessible_by(self.request.user):
            raise Http404()
        return super().form_valid(form)


class TicketCommentForm(forms.Form):
    body = forms.CharField(widget=ticket_widget)


class TicketMixin(object):
    model = Ticket

    def get_object(self, queryset=None):
        ticket = super(TicketMixin, self).get_object(queryset)
        profile_id = self.request.profile.id
        if self.request.user.has_perm("judge.change_ticket"):
            return ticket
        if ticket.user_id == profile_id:
            return ticket
        if ticket.assignees.filter(id=profile_id).exists():
            return ticket
        linked = ticket.linked_item
        if isinstance(linked, Problem) and linked.is_editable_by(self.request.user):
            return ticket
        raise PermissionDenied()


class TicketView(TitleMixin, LoginRequiredMixin, TicketMixin, SingleObjectFormView):
    form_class = TicketCommentForm
    template_name = "ticket/ticket.html"
    context_object_name = "ticket"

    def form_valid(self, form):
        message = TicketMessage(
            user=self.request.profile,
            body=form.cleaned_data["body"],
            ticket=self.object,
        )
        message.save()

        link = "%s#message-%d" % (reverse("ticket", args=[self.object.id]), message.id)

        notify_list = list(chain(self.object.assignees.all(), [self.object.user]))
        add_ticket_notifications(notify_list, message.user, link, self.object)

        if event.real:
            event.post(
                "tickets",
                {
                    "type": "ticket-message",
                    "id": self.object.id,
                    "message": message.id,
                    "user": self.object.user_id,
                    "assignees": list(
                        self.object.assignees.values_list("id", flat=True)
                    ),
                },
            )
            event.post(
                "ticket-%d" % self.object.id,
                {
                    "type": "ticket-message",
                    "message": message.id,
                },
            )
        return HttpResponseRedirect(link)

    def get_title(self):
        return _("%(title)s - Ticket %(id)d") % {
            "title": self.object.title,
            "id": self.object.id,
        }

    def get_context_data(self, **kwargs):
        context = super(TicketView, self).get_context_data(**kwargs)
        context["ticket_messages"] = self.object.messages.select_related("user__user")
        context["assignees"] = self.object.assignees.select_related("user")
        context["last_msg"] = event.last()
        return context


class TicketStatusChangeView(LoginRequiredMixin, TicketMixin, SingleObjectMixin, View):
    open = None

    def post(self, request, *args, **kwargs):
        if self.open is None:
            raise ImproperlyConfigured("Need to define open")
        ticket = self.get_object()
        if ticket.is_open != self.open:
            ticket.is_open = self.open
            ticket.save()
            if event.real:
                event.post(
                    "tickets",
                    {
                        "type": "ticket-status",
                        "id": ticket.id,
                        "open": self.open,
                        "user": ticket.user_id,
                        "assignees": list(
                            ticket.assignees.values_list("id", flat=True)
                        ),
                        "title": ticket.title,
                    },
                )
                event.post(
                    "ticket-%d" % ticket.id,
                    {
                        "type": "ticket-status",
                        "open": self.open,
                    },
                )
        return HttpResponse(status=204)


class TicketNotesForm(forms.Form):
    notes = forms.CharField(widget=forms.Textarea(), required=False)


class TicketNotesEditView(LoginRequiredMixin, TicketMixin, SingleObjectFormView):
    template_name = "ticket/edit-notes.html"
    form_class = TicketNotesForm
    context_object_name = "ticket"

    def get_initial(self):
        return {"notes": self.get_object().notes}

    def form_valid(self, form):
        ticket = self.get_object()
        ticket.notes = notes = form.cleaned_data["notes"]
        ticket.save()
        if notes:
            return HttpResponse(linebreaks(notes, autoescape=True))
        else:
            return HttpResponse()

    def form_invalid(self, form):
        return HttpResponseBadRequest()


class TicketList(LoginRequiredMixin, ListView):
    model = Ticket
    template_name = "ticket/list.html"
    context_object_name = "tickets"
    paginate_by = 50
    paginator_class = DiggPaginator

    @cached_property
    def user(self):
        return self.request.user

    @cached_property
    def profile(self):
        return self.user.profile

    @cached_property
    def can_edit_all(self):
        return self.request.user.has_perm("judge.change_ticket")

    @cached_property
    def filter_users(self):
        return self.request.GET.getlist("user")

    @cached_property
    def filter_assignees(self):
        return self.request.GET.getlist("assignee")

    def GET_with_session(self, key):
        if not self.request.GET:
            return self.request.session.get(key, False)
        return self.request.GET.get(key, None) == "1"

    def _get_queryset(self):
        return (
            Ticket.objects.select_related("user__user")
            .prefetch_related("assignees__user")
            .order_by("-id")
        )

    def get_queryset(self):
        queryset = self._get_queryset()
        if self.GET_with_session("own"):
            queryset = queryset.filter(own_ticket_filter(self.profile.id))
        elif not self.can_edit_all:
            queryset = filter_visible_tickets(queryset, self.user, self.profile)
        if self.filter_assignees:
            queryset = queryset.filter(
                assignees__user__username__in=self.filter_assignees
            )
        if self.filter_users:
            queryset = queryset.filter(user__user__username__in=self.filter_users)
        return queryset.distinct()

    def get_context_data(self, **kwargs):
        context = super(TicketList, self).get_context_data(**kwargs)

        page = context["page_obj"]
        context["title"] = _("Tickets - Page %(number)d of %(total)d") % {
            "number": page.number,
            "total": page.paginator.num_pages,
        }
        context["can_edit_all"] = self.can_edit_all
        context["filter_status"] = {
            "own": self.GET_with_session("own"),
            "user": self.filter_users,
            "assignee": self.filter_assignees,
            "user_id": json.dumps(
                list(
                    Profile.objects.filter(
                        user__username__in=self.filter_users
                    ).values_list("id", flat=True)
                )
            ),
            "assignee_id": json.dumps(
                list(
                    Profile.objects.filter(
                        user__username__in=self.filter_assignees
                    ).values_list("id", flat=True)
                )
            ),
            "own_id": self.profile.id if self.GET_with_session("own") else "null",
        }
        context["last_msg"] = event.last()
        context.update(paginate_query_context(self.request))
        return context

    def post(self, request, *args, **kwargs):
        to_update = ("own",)
        for key in to_update:
            if key in request.GET:
                val = request.GET.get(key) == "1"
                request.session[key] = val
            else:
                request.session.pop(key, None)
        return HttpResponseRedirect(request.get_full_path())


class ProblemTicketListView(TicketList):
    def _get_queryset(self):
        problem = get_object_or_404(Problem, code=self.kwargs.get("problem"))
        if problem.is_editable_by(self.request.user):
            return problem.tickets.order_by("-id")
        elif problem.is_accessible_by(self.request.user):
            return problem.tickets.filter(own_ticket_filter(self.profile.id)).order_by(
                "-id"
            )
        raise Http404()


class TicketListDataAjax(TicketMixin, SingleObjectMixin, View):
    def get(self, request, *args, **kwargs):
        try:
            self.kwargs["pk"] = request.GET["id"]
        except KeyError:
            return HttpResponseBadRequest()
        ticket = self.get_object()
        message = ticket.messages.first()
        return JsonResponse(
            {
                "row": get_template("ticket/row.html").render(
                    {"ticket": ticket}, request
                ),
                "notification": {
                    "title": _("New Ticket: %s") % ticket.title,
                    "body": "%s\n%s"
                    % (
                        _("#%(id)d, assigned to: %(users)s")
                        % {
                            "id": ticket.id,
                            "users": (
                                _(", ").join(
                                    ticket.assignees.values_list(
                                        "user__username", flat=True
                                    )
                                )
                                or _("no one")
                            ),
                        },
                        truncatechars(message.body, 200),
                    ),
                },
            }
        )


class TicketMessageDataAjax(TicketMixin, SingleObjectMixin, View):
    def get(self, request, *args, **kwargs):
        try:
            message_id = request.GET["message"]
        except KeyError:
            return HttpResponseBadRequest()
        ticket = self.get_object()
        try:
            message = ticket.messages.get(id=message_id)
        except TicketMessage.DoesNotExist:
            return HttpResponseBadRequest()
        return JsonResponse(
            {
                "message": get_template("ticket/message.html").render(
                    {"message": message}, request
                ),
                "notification": {
                    "title": _("New Ticket Message For: %s") % ticket.title,
                    "body": truncatechars(message.body, 200),
                },
            }
        )