from datetime import UTC, datetime

from pydantic import BaseModel, field_validator


class ClickUpAuthCredentialsBase(BaseModel):
    access_token: str
    refresh_token: str | None = None
    expires_in: int | None = None
    expires_at: datetime | None = None
    user_id: str | None = None
    user_email: str | None = None
    user_name: str | None = None
    workspace_id: str | None = None
    workspace_name: str | None = None

    @property
    def is_expired(self) -> bool:
        """Check if the credentials have expired."""
        if self.expires_at is None:
            return False  # Long-lived token, treat as not expired
        return self.expires_at <= datetime.now(UTC)

    @property
    def is_refreshable(self) -> bool:
        """Check if the credentials can be refreshed."""
        return self.refresh_token is not None

    def to_dict(self) -> dict:
        """Convert credentials to dictionary for storage."""
        return {
            "access_token": self.access_token,
            "refresh_token": self.refresh_token,
            "expires_in": self.expires_in,
            "expires_at": self.expires_at.isoformat() if self.expires_at else None,
            "user_id": self.user_id,
            "user_email": self.user_email,
            "user_name": self.user_name,
            "workspace_id": self.workspace_id,
            "workspace_name": self.workspace_name,
        }

    @classmethod
    def from_dict(cls, data: dict) -> "ClickUpAuthCredentialsBase":
        """Create credentials from dictionary."""
        expires_at = None
        if data.get("expires_at"):
            expires_at = datetime.fromisoformat(data["expires_at"])

        # Convert user_id to string if it's an integer (for backward compatibility)
        user_id = data.get("user_id")
        if user_id is not None and not isinstance(user_id, str):
            user_id = str(user_id)

        # Convert workspace_id to string if it's an integer (for backward compatibility)
        workspace_id = data.get("workspace_id")
        if workspace_id is not None and not isinstance(workspace_id, str):
            workspace_id = str(workspace_id)

        return cls(
            access_token=data.get("access_token", ""),
            refresh_token=data.get("refresh_token"),
            expires_in=data.get("expires_in"),
            expires_at=expires_at,
            user_id=user_id,
            user_email=data.get("user_email"),
            user_name=data.get("user_name"),
            workspace_id=workspace_id,
            workspace_name=data.get("workspace_name"),
        )

    @field_validator("expires_at", mode="before")
    @classmethod
    def ensure_aware_utc(cls, v):
        # Strings like "2025-08-26T14:46:57.367184"
        if isinstance(v, str):
            # add +00:00 if missing tz info
            if v.endswith("Z"):
                return datetime.fromisoformat(v.replace("Z", "+00:00"))
            dt = datetime.fromisoformat(v)
            return dt if dt.tzinfo else dt.replace(tzinfo=UTC)
        # datetime objects
        if isinstance(v, datetime):
            return v if v.tzinfo else v.replace(tzinfo=UTC)
        return v