from datetime import datetime
from typing import Dict, Optional

from pydantic import BaseModel, field_validator

from junglescout.models.parameters import Attributes, Marketplace, Params, Sort
from junglescout.models.requests.method import Method
from junglescout.models.requests.request import Request
from junglescout.session import Session


class HistoricalSearchVolumeArgs(BaseModel):
    keyword: str
    start_date: str
    end_date: str
    sort_option: Optional[Sort]
    marketplace: Marketplace


class HistoricalSearchVolumeParams(Params):
    keyword: str
    start_date: str
    end_date: str

    @field_validator("start_date")
    @classmethod
    def _validate_start_date(cls, v: str) -> str:
        return cls.__validate_date(v)

    @field_validator("end_date")
    @classmethod
    def _validate_end_date(cls, v: str) -> str:
        return cls.__validate_date(v)

    @staticmethod
    def __validate_date(date: str) -> str:
        valid_date_length = 10
        assert len(date) == valid_date_length, "Date must be in the format YYYY-MM-DD"
        try:
            datetime.strptime(date, "%Y-%m-%d")  # noqa: DTZ007
        except ValueError as exc:
            msg = "Incorrect data format, should be YYYY-MM-DD"
            raise ValueError(msg) from exc
        return date


class HistoricalSearchVolumeAttributes(Attributes):
    pass


class HistoricalSearchVolumeRequest(
    Request[HistoricalSearchVolumeArgs, HistoricalSearchVolumeParams, HistoricalSearchVolumeAttributes]
):
    @property
    def url(self) -> str:
        return self.session.build_url("keywords", "historical_search_volume", params=self.params_serialized)

    @property
    def method(self) -> Method:
        return Method.GET

    def serialize_params(self) -> Dict:
        return self.params.model_dump(by_alias=True, exclude_none=True)

    @classmethod
    def from_args(cls, args: HistoricalSearchVolumeArgs, session: Session) -> "HistoricalSearchVolumeRequest":
        params = HistoricalSearchVolumeParams(
            marketplace=args.marketplace,
            sort=args.sort_option,
            keyword=args.keyword,
            start_date=args.start_date,
            end_date=args.end_date,
        )
        attributes = HistoricalSearchVolumeAttributes()
        return cls(session=session, params=params, attributes=attributes)