oarepo_oaipmh_harvester.oai_record.resource

class AdministrationDetailJSONSerializer(JSONSerializer):
    """JSON serializer for administration API."""

    @override
    def serialize_object(self, obj: Any) -> str:
        """Serialize a single object."""
        obj = self._convert_to_administration_detail(obj)
        return super().serialize_object(obj)

    @override
    def serialize_object_list(self, obj_list: dict[str, Any]) -> str:
        obj_list["hits"]["hits"] = [self._convert_to_administration_detail(obj) for obj in obj_list["hits"]["hits"]]
        return super().serialize_object_list(obj_list)

    def _convert_to_administration_detail(self, serialized_record: dict[str, Any]) -> dict[str, Any]:
        serialized_record = {**serialized_record}
        tr = serialized_record.get("transformed_data", {})
        if tr:
            serialized_record["title"] = tr.get("metadata", {}).get("title", "") or tr.get("title", "")
            if serialized_record["title"]:
                serialized_record["title"] = str(serialized_record["title"])
        record_pid = serialized_record.get("record_pid")
        if record_pid:
            record_link = tr.get("links", {}).get("self_html")
            if record_link:
                serialized_record["record_id_with_link"] = '<a href="{}">{}</a>'.format(
                    record_link, serialized_record["record_pid"]
                )
            else:
                serialized_record["record_id_with_link"] = '<a href="{}">{}</a>'.format(
                    f"/records/{record_pid}", serialized_record["record_pid"]
                )
        serialized_record["errors"] = data_to_html_yaml(
            location_to_list(cast("list[Any]", serialized_record.get("errors")))
        )
        serialized_record["original_data"] = data_to_html_yaml(serialized_record.get("original_data"))
        serialized_record["transformed_data"] = data_to_html_yaml(serialized_record.get("transformed_data"))
        serialized_record["has_errors"] = serialized_record.get("has_errors")
        serialized_record["deleted"] = serialized_record.get("deleted")
        return serialized_record

class OAIRecordResourceConfig(RecordResourceConfig):
    """Harvest record resource configuration."""

    blueprint_name = "oai_records"
    url_prefix = "/oai/harvest/records"
    api_service = "oai-harvest-record"

    routes: Mapping[str, Any] = {
        "list": "",
        "item": "/<path:id>",
        "harvest": "/<path:id>/harvest",
    }

    request_view_args: Mapping[str, Any] = {
        "id": ma.fields.Str(),
    }

    error_handlers: Mapping[
        type[HTTPException | BaseException] | int,
        Callable[[Exception], Response],
    ] = {
        **ErrorHandlersMixin.error_handlers,
    }

    response_handlers: Mapping[str, Any] = {
        "application/vnd.inveniordm.v1+json": RecordResourceConfig.response_handlers["application/json"],
        "application/invenio-administration-detail+json": ResponseHandler(
            serializer=AdministrationDetailJSONSerializer(),
            headers=etag_headers,
        ),
        **RecordResourceConfig.response_handlers,
    }

class OAIRecordResource(RecordResource):
    """OAI Record resource."""

    def p(self, prefix: str, route: str) -> str:
        """Prefix a route with the URL prefix."""
        return f"{prefix}{route}"

    def create_url_rules(self) -> list:
        """Create the URL rules for the harvest record resource."""
        routes = self.config.routes
        return [
            route("GET", routes["list"], self.search),
            route("GET", routes["item"], self.read),
            route("POST", routes["harvest"], self.harvest),
        ]

    @override
    @request_search_args
    @request_view_args
    @response_handler(many=True)
    def search(self) -> tuple[dict[str, Any], int]:
        """Search harvest records."""
        hits = self.service.search(
            identity=g.identity,
            params=resource_requestctx.args,
            search_preference=search_preference(),
        )
        return hits.to_dict(), 200

    @override
    @request_view_args
    @response_handler()
    def read(self) -> tuple[dict[str, Any], int]:
        """Read a single harvest record."""
        item = self.service.read(
            id_=resource_requestctx.view_args["id"],
            identity=g.identity,
        )
        return item.to_dict(), 200

    @request_view_args
    @response_handler()
    def harvest(self) -> tuple[dict[str, Any], int]:
        """Re-harvest the OAI record."""
        item = self.service.harvest(
            id_=resource_requestctx.view_args["id"],
            identity=g.identity,
        )
        return item.to_dict(), 200