"""Pydantic schemas for workspace and invitation endpoints."""

from datetime import datetime
from uuid import UUID

from pydantic import BaseModel, Field

from app.entities.workspace import Workspace
from app.entities.workspace_invitation import WorkspaceInvitation


class WorkspaceCreate(BaseModel):
    name: str = Field(min_length=1, max_length=100)


class WorkspaceUpdate(BaseModel):
    name: str | None = Field(default=None, min_length=1, max_length=100)
    planning_goal: str | None = None
    planning_mode: str | None = None


class WorkspaceResponse(BaseModel):
    id: UUID
    name: str
    planning_goal: str
    planning_mode: str
    created_at: datetime | None

    @classmethod
    def from_entity(cls, ws: Workspace) -> "WorkspaceResponse":
        return cls(
            id=ws.id,
            name=ws.name,
            planning_goal=ws.planning_goal,
            planning_mode=ws.planning_mode,
            created_at=ws.created_at,
        )


class PersonalDataResponse(BaseModel):
    uuid: UUID
    workspace_id: UUID
    name: str
    salary_type: str
    employment_status: str
    created_at: datetime | None


class PersonalDataCreate(BaseModel):
    """Request body for both adding a placeholder parent (POST) and renaming

    one (PATCH) — both take exactly one field, so one schema serves both.
    """

    display_name: str = Field(min_length=1, max_length=100)


class InviteResponse(BaseModel):
    token: str  # raw one-time token — share out-of-band; not stored on backend


class RedeemInviteRequest(BaseModel):
    token: str


class InvitationResponse(BaseModel):
    id: UUID
    workspace_id: UUID
    status: str
    expires_at: datetime
    created_at: datetime

    @classmethod
    def from_entity(cls, inv: WorkspaceInvitation) -> "InvitationResponse":
        return cls(
            id=inv.id,
            workspace_id=inv.workspace_id,
            status=inv.status,
            expires_at=inv.expires_at,
            created_at=inv.created_at,
        )