Source code for web.views.daemon_views.DaemonDetailWithDeleteView

# SPDX-License-Identifier: AGPL-3.0-or-later
#
# Eonvelope - a open-source self-hostable email archiving server
# Copyright (C) 2024 David Aderbauer & The Eonvelope Contributors
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.

"""Module with the :class:`web.views.DaemonDetailWithDeleteView` view."""

from __future__ import annotations

from typing import TYPE_CHECKING, Any, override

from django.contrib import messages
from django.contrib.auth.mixins import LoginRequiredMixin
from django.urls import reverse_lazy
from django.utils.translation import gettext_lazy as _
from django.views.generic.edit import DeletionMixin

from core.models import Daemon
from web.mixins.CustomActionMixin import CustomActionMixin
from web.mixins.TestActionMixin import TestActionMixin
from web.views.base import DetailWithDeleteView

from .DaemonFilterView import DaemonFilterView

if TYPE_CHECKING:
    from django.db.models import QuerySet
    from django.http import HttpRequest, HttpResponse




class DaemonDetailWithDeleteView(
    LoginRequiredMixin,
    DetailWithDeleteView,
    CustomActionMixin,
    TestActionMixin,
):
    """View for a single :class:`core.models.Daemon` instance."""

    URL_NAME = Daemon.get_detail_web_url_name()
    model = Daemon
    template_name = "web/daemon/daemon_detail.html"
    success_url = reverse_lazy("web:" + DaemonFilterView.URL_NAME)



    @override
    def get_queryset(self) -> QuerySet[Daemon]:
        """Restricts the queryset to objects owned by the requesting user."""
        return (
            super()
            .get_queryset()
            .filter(mailbox__account__user=self.request.user)
            .select_related("mailbox", "mailbox__account", "interval", "celery_task")
        )




    @override
    def post(self, request: HttpRequest, *args: Any, **kwargs: Any) -> HttpResponse:
        if "delete" in request.POST:
            return DeletionMixin.post(self, request)
        return CustomActionMixin.post(self, request)




    def handle_start_daemon(self, request: HttpRequest) -> HttpResponse:
        """Handler function for the `start-daemon` action.

        Args:
            request: The action request to handle.

        Returns:
            A template response with the updated view after the action.
        """
        self.object = self.get_object()
        result = self.object.start()
        if result:
            messages.success(request, _("Daemon started successfully"))
        else:
            messages.warning(request, _("Daemon was already running"))
        self.object.refresh_from_db()
        return self.get(request)




    def handle_stop_daemon(self, request: HttpRequest) -> HttpResponse:
        """Handler function for the `stop-daemon` action.

        Args:
            request: The action request to handle.

        Returns:
            A template response with the updated view after the action.
        """
        self.object = self.get_object()
        result = self.object.stop()
        if result:
            messages.success(request, _("Daemon stopped successfully"))
        else:
            messages.warning(request, _("Daemon was not running"))
        self.object.refresh_from_db()
        return self.get(request)