from http import HTTPStatus
from typing import Optional

from fastapi import HTTPException, Request, Response

from ..base_lora_api_transform import BaseLoRAApiTransform
from ..constants import RequestField, ResponseMessage
from ..models import BaseLoRATransformRequestOutput
from ..utils import get_adapter_name_from_request_path


def validate_sagemaker_unregister_request(raw_request: Request):
    adapter_name = get_adapter_name_from_request_path(raw_request)
    if not adapter_name:
        raise HTTPException(
            status_code=HTTPStatus.FAILED_DEPENDENCY.value,
            detail=f"Malformed request path; missing path parameter: {RequestField.ADAPTER_NAME}",
        )
    return adapter_name


class UnregisterLoRAApiTransform(BaseLoRAApiTransform):
    async def transform_request(
        self, raw_request: Request
    ) -> BaseLoRATransformRequestOutput:
        """
        :param Optional[pydantic.BaseModel] request: Not used because the Unregister LoRA API does not take a request body.
        :param fastapi.Request raw_request:
        """
        adapter_name = validate_sagemaker_unregister_request(raw_request)
        transformed_request = self._transform_request(None, raw_request)
        return BaseLoRATransformRequestOutput(
            request=transformed_request,
            raw_request=raw_request,
            adapter_name=adapter_name,
        )

    def _transform_ok_response(self, response: Response, **kwargs):
        adapter_name: Optional[str] = kwargs.get("adapter_name")
        adapter_alias: Optional[str] = kwargs.get("adapter_alias")
        return Response(
            status_code=HTTPStatus.OK.value,
            content=ResponseMessage.ADAPTER_UNREGISTERED.format(
                alias=adapter_alias or adapter_name
            ),
        )

    def _transform_error_response(self, response: Response, **kwargs):
        # TODO: add error handling
        return response