#
# Copyright (c) 2021-2026 ahriman team.
#
# This file is part of ahriman
# (see https://github.com/arcan1s/ahriman).
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
from aiohttp.web import HTTPBadRequest, HTTPNoContent, Response
from typing import ClassVar

from ahriman import __version__
from ahriman.models.build_status import BuildStatusEnum
from ahriman.models.counters import Counters
from ahriman.models.internal_status import InternalStatus
from ahriman.models.repository_stats import RepositoryStats
from ahriman.models.user_access import UserAccess
from ahriman.web.apispec.decorators import apidocs
from ahriman.web.schemas import InternalStatusSchema, RepositoryIdSchema, StatusSchema
from ahriman.web.views.base import BaseView
from ahriman.web.views.status_view_guard import StatusViewGuard


class StatusView(StatusViewGuard, BaseView):
    """
    web service status web view

    Attributes:
        GET_PERMISSION(UserAccess): (class attribute) get permissions of self
        POST_PERMISSION(UserAccess): (class attribute) post permissions of self
    """

    GET_PERMISSION: ClassVar[UserAccess] = UserAccess.Read
    POST_PERMISSION: ClassVar[UserAccess] = UserAccess.Full
    ROUTES = ["/api/v1/status"]

    @apidocs(
        tags=["Status"],
        summary="Web service status",
        description="Get web service status counters",
        permission=GET_PERMISSION,
        error_404_description="Repository is unknown",
        schema=InternalStatusSchema,
        query_schema=RepositoryIdSchema,
    )
    async def get(self) -> Response:
        """
        get current service status

        Returns:
            Response: 200 with service status object
        """
        repository_id = self.repository_id()
        packages = await self.service(repository_id).packages()
        counters = Counters.from_packages(packages)
        stats = RepositoryStats.from_packages([package for package, _ in packages])

        status = InternalStatus(
            status=self.service(repository_id).status,
            architecture=repository_id.architecture,
            packages=counters,
            repository=repository_id.name,
            stats=stats,
            version=__version__,
        )

        return self.json_response(status.view())

    @apidocs(
        tags=["Status"],
        summary="Set web service status",
        description="Update web service status. Counters will remain unchanged",
        permission=POST_PERMISSION,
        error_400_enabled=True,
        error_404_description="Repository is unknown",
        query_schema=RepositoryIdSchema,
        body_schema=StatusSchema,
    )
    async def post(self) -> None:
        """
        update service status

        Raises:
            HTTPBadRequest: if bad data is supplied
            HTTPNoContent: in case of success response
        """
        try:
            data = await self.request.json()
            status = BuildStatusEnum(data["status"])
        except Exception as ex:
            raise HTTPBadRequest(reason=str(ex))

        await self.service().status_update(status)

        raise HTTPNoContent
