views.py

import csv
import json
from typing import Any

from django.conf import settings
from django.contrib.auth.decorators import permission_required
from django.contrib.auth.mixins import PermissionRequiredMixin
from django.db.models.query import QuerySet
from django.http import Http404, HttpResponse
from django.shortcuts import get_object_or_404
from django.template.response import TemplateResponse
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from django.views.generic import CreateView, DetailView, ListView, UpdateView
from filer.models import Image

from .forms import (
    DrawingCreateForm,
    DrawingManualForm,
    DrawingParentForm,
    DrawingUpdateForm,
    LayerUpdateForm,
)
from .models import Drawing, Entity, Layer


class HxTemplateMixin:
    """Switches template depending on request.htmx"""

    def get_template_names(self) -> list[str]:
        if self.request.htmx:
            return [self.template_name + "#htmx-partial"]
        return [self.template_name]


class HxSetupMixin:
    """Restricts to HTMX requests"""

    def setup(self, request, *args, **kwargs):
        if not request.htmx:
            raise Http404("Request without HTMX headers")
        super().setup(request, *args, **kwargs)


class BaseListView(HxTemplateMixin, ListView):
    model = Drawing
    context_object_name = "drawings"
    template_name = "djeocadengine/base_list.html"

    def get_queryset(self) -> QuerySet[Any]:
        qs = Drawing.objects.exclude(epsg=None)
        return qs

    def get_context_data(self, **kwargs) -> dict[str, Any]:
        context = super().get_context_data(**kwargs)
        context["unreferenced"] = Drawing.objects.filter(epsg=None)
        context["leaflet_config"] = settings.LEAFLET_CONFIG
        return context

    def dispatch(self, request, *args, **kwargs):
        response = super().dispatch(request, *args, **kwargs)
        if request.htmx:
            dict = {"refreshCollections": True}
            response["HX-Trigger-After-Swap"] = json.dumps(dict)
        return response


class DrawingCreateView(PermissionRequiredMixin, HxSetupMixin, CreateView):
    model = Drawing
    permission_required = "djeocadengine.add_drawing"
    form_class = DrawingCreateForm
    template_name = "djeocadengine/htmx/drawing_create.html"

    def form_valid(self, form):
        if form.cleaned_data["temp_image"]:
            img = Image.objects.create(
                owner=self.request.user,
                original_filename=form.cleaned_data["title"],
                file=form.cleaned_data["temp_image"],
            )
            form.instance.image = img
            form.instance.temp_image = None
        return super().form_valid(form)

    def get_success_url(self):
        if not self.object.epsg:
            return reverse(
                "djeocadengine:drawing_geodata",
                kwargs={"pk": self.object.id},
            )
        return reverse(
            "djeocadengine:drawing_detail",
            kwargs={"pk": self.object.id},
        )


class DrawingGeodataView(PermissionRequiredMixin, HxSetupMixin, UpdateView):
    model = Drawing
    permission_required = "djeocadengine.change_drawing"
    template_name = "djeocadengine/htmx/drawing_geodata.html"
    form_class = DrawingParentForm

    def get_success_url(self):
        return reverse(
            "djeocadengine:drawing_detail",
            kwargs={"pk": self.object.id},
        )


class DrawingManualView(PermissionRequiredMixin, HxSetupMixin, UpdateView):
    model = Drawing
    permission_required = "djeocadengine.change_drawing"
    template_name = "djeocadengine/htmx/drawing_manual.html"
    form_class = DrawingManualForm

    def get_context_data(self, **kwargs) -> dict[str, Any]:
        context = super().get_context_data(**kwargs)
        context["drawings"] = Drawing.objects.none()
        context["leaflet_config"] = settings.LEAFLET_CONFIG
        context["map_status"] = {"map_on_click": True}
        return context

    def get_initial(self):
        initial = super().get_initial()
        initial["lat"] = settings.LEAFLET_CONFIG["DEFAULT_CENTER"][0]
        initial["long"] = settings.LEAFLET_CONFIG["DEFAULT_CENTER"][1]
        return initial

    def form_valid(self, form):
        form.instance.geom = {
            "type": "Point",
            "coordinates": [form.cleaned_data["long"], form.cleaned_data["lat"]],
        }
        return super().form_valid(form)

    def get_success_url(self):
        return reverse(
            "djeocadengine:drawing_detail",
            kwargs={"pk": self.object.id},
        )

    def dispatch(self, request, *args, **kwargs):
        response = super().dispatch(request, *args, **kwargs)
        dict = {"refreshCollections": True}
        response["HX-Trigger-After-Swap"] = json.dumps(dict)
        return response


class DrawingDetailView(HxTemplateMixin, DetailView):
    model = Drawing
    template_name = "djeocadengine/drawing_detail.html"

    def get_context_data(self, **kwargs) -> dict[str, Any]:
        context = super().get_context_data(**kwargs)
        layers = self.object.related_layers.filter(is_block=False)
        id_list = layers.values_list("id", flat=True)
        context["lines"] = Entity.objects.filter(
            layer_id__in=id_list
        ).prefetch_related()
        context["drawings"] = self.object
        name_list = layers.values_list("name", flat=True)
        context["layer_list"] = list(dict.fromkeys(name_list))
        context["layer_list"] = [_("Layer - ") + s for s in context["layer_list"]]
        return context

    def dispatch(self, request, *args, **kwargs):
        response = super(DrawingDetailView, self).dispatch(request, *args, **kwargs)
        if request.htmx:
            dict = {"refreshCollections": True}
            response["HX-Trigger-After-Swap"] = json.dumps(dict)
            response["HX-Push-Url"] = self.object.get_absolute_url()
        return response


class DrawingUpdateView(PermissionRequiredMixin, HxSetupMixin, UpdateView):
    permission_required = "djeocadengine.change_drawing"
    model = Drawing
    form_class = DrawingUpdateForm
    template_name = "djeocadengine/htmx/drawing_update.html"

    def get_initial(self):
        initial = super().get_initial()
        initial["lat"] = self.object.geom["coordinates"][1]
        initial["long"] = self.object.geom["coordinates"][0]
        return initial

    def get_context_data(self, **kwargs) -> dict[str, Any]:
        context = super().get_context_data(**kwargs)
        context["drawings"] = self.object
        context["layers"] = self.object.related_layers.filter(is_block=False)
        context["blocks"] = self.object.related_layers.filter(is_block=True)
        context["map_status"] = {"map_on_click": True}
        return context

    def form_valid(self, form):
        if form.cleaned_data["temp_image"]:
            img = Image.objects.create(
                owner=self.request.user,
                original_filename=form.cleaned_data["title"],
                file=form.cleaned_data["temp_image"],
            )
            form.instance.image = img
            form.instance.temp_image = None
        form.instance.geom = {
            "type": "Point",
            "coordinates": [form.cleaned_data["long"], form.cleaned_data["lat"]],
        }
        return super().form_valid(form)

    def get_success_url(self):
        return reverse(
            "djeocadengine:drawing_detail",
            kwargs={"pk": self.object.id},
        )

    def dispatch(self, request, *args, **kwargs):
        response = super().dispatch(request, *args, **kwargs)
        dict = {"refreshCollections": True}
        response["HX-Trigger-After-Swap"] = json.dumps(dict)
        return response


@permission_required("djeocadengine.delete_drawing")
def drawing_delete_view(request, pk):
    if not request.htmx:
        raise Http404("Request without HTMX headers")
    drawing = get_object_or_404(Drawing, id=pk)
    drawing.delete()
    return TemplateResponse(
        request,
        "djeocadengine/htmx/drawing_delete.html",
        {},
    )


class LayerDetailView(HxSetupMixin, DetailView):
    model = Layer
    template_name = "djeocadengine/htmx/layer_inline.html"
    context_object_name = "layer"


class LayerUpdateView(PermissionRequiredMixin, HxSetupMixin, UpdateView):
    permission_required = "djeocadengine.change_layer"
    model = Layer
    form_class = LayerUpdateForm
    template_name = "djeocadengine/htmx/layer_update.html"

    def get_success_url(self):
        return reverse(
            "djeocadengine:layer_detail",
            kwargs={"pk": self.object.id},
        )


@permission_required("djeocadengine.delete_layer")
def layer_delete_view(request, pk):
    layer = get_object_or_404(Layer, id=pk)
    if not request.htmx or layer.name == "0":
        raise Http404("Request without HTMX headers")
    layer.delete()
    return TemplateResponse(
        request,
        "djeocadengine/htmx/layer_delete.html",
        {},
    )


def csv_download(request, pk):
    drawing = get_object_or_404(Drawing, id=pk)
    # Create the HttpResponse object with the appropriate CSV header.
    response = HttpResponse(content_type="text/csv")
    response["Content-Disposition"] = f'attachment; filename="{drawing.title}.csv"'
    writer = csv.writer(response)
    writer = drawing.write_csv(writer)

    return response


def drawing_download(request, pk):
    drawing = get_object_or_404(Drawing, id=pk)
    response = HttpResponse(drawing.dxf, content_type="text/plain")
    response["Content-Disposition"] = "attachment; filename=%s.dxf" % drawing.title

    return response