#
# 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, HTTPNotFound, Response
from collections.abc import Callable
from typing import ClassVar

from ahriman.core.alpm.remote import AUR
from ahriman.core.types import Comparable
from ahriman.models.aur_package import AURPackage
from ahriman.models.user_access import UserAccess
from ahriman.web.apispec.decorators import apidocs
from ahriman.web.schemas import AURPackageSchema, SearchSchema
from ahriman.web.views.base import BaseView


class SearchView(BaseView):
    """
    AUR search web view

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

    GET_PERMISSION: ClassVar[UserAccess] = UserAccess.Reporter
    ROUTES = ["/api/v1/service/search"]

    @apidocs(
        tags=["Actions"],
        summary="Search for package",
        description="Search for package in AUR",
        permission=GET_PERMISSION,
        error_400_enabled=True,
        error_404_description="Package base is unknown",
        schema=AURPackageSchema(many=True),
        query_schema=SearchSchema,
    )
    async def get(self) -> Response:
        """
        search packages in AUR

        Returns:
            Response: 200 with found package bases and descriptions sorted by base

        Raises:
            HTTPBadRequest: in case if bad data is supplied
            HTTPNotFound: if no packages found
        """
        try:
            search: list[str] = self.get_non_empty(lambda key: self.request.query.getall(key, default=[]), "for")
            packages = AUR.multisearch(*search)
        except Exception as ex:
            raise HTTPBadRequest(reason=str(ex))

        if not packages:
            raise HTTPNotFound(reason=f"No packages found for terms: {search}")

        comparator: Callable[[AURPackage], Comparable] = \
            lambda item: (
                item.package_base not in search,  # inverted because False < True
                not any(item.package_base.startswith(term) for term in search),  # same as above
                item.package_base,
        )
        response = [
            {
                "package": package.package_base,
                "description": package.description,
            } for package in sorted(packages, key=comparator)
        ]
        return self.json_response(response)
