"""
Pydantic schemas for Session model.

Request and response schemas for work sessions with time tracking.
"""

from datetime import date, datetime
from typing import Optional
from uuid import UUID

from pydantic import BaseModel, Field


class SessionBase(BaseModel):
    """Base schema with shared Session fields."""

    client_id: Optional[str] = Field(None, description="Foreign key to clients table (UUID)")
    project_id: Optional[str] = Field(None, description="Foreign key to projects table (UUID)")
    machine_id: Optional[str] = Field(None, description="Foreign key to machines table (UUID)")
    session_date: date = Field(..., description="Date of the session")
    start_time: Optional[datetime] = Field(None, description="Session start timestamp")
    end_time: Optional[datetime] = Field(None, description="Session end timestamp")
    duration_minutes: Optional[int] = Field(None, description="Duration in minutes (auto-calculated or manual)")
    status: str = Field("completed", description="Session status: completed, in_progress, blocked, pending")
    session_title: str = Field(..., description="Brief title describing the session")
    summary: Optional[str] = Field(None, description="Markdown summary of the session")
    is_billable: bool = Field(False, description="Whether this session is billable")
    billable_hours: Optional[float] = Field(None, description="Billable hours if applicable")
    technician: Optional[str] = Field(None, description="Name of technician who performed the work")
    session_log_file: Optional[str] = Field(None, description="Path to markdown session log file")
    notes: Optional[str] = Field(None, description="Additional notes about the session")


class SessionCreate(SessionBase):
    """Schema for creating a new Session."""
    pass


class SessionUpdate(BaseModel):
    """Schema for updating an existing Session. All fields are optional."""

    client_id: Optional[str] = Field(None, description="Foreign key to clients table (UUID)")
    project_id: Optional[str] = Field(None, description="Foreign key to projects table (UUID)")
    machine_id: Optional[str] = Field(None, description="Foreign key to machines table (UUID)")
    session_date: Optional[date] = Field(None, description="Date of the session")
    start_time: Optional[datetime] = Field(None, description="Session start timestamp")
    end_time: Optional[datetime] = Field(None, description="Session end timestamp")
    duration_minutes: Optional[int] = Field(None, description="Duration in minutes (auto-calculated or manual)")
    status: Optional[str] = Field(None, description="Session status: completed, in_progress, blocked, pending")
    session_title: Optional[str] = Field(None, description="Brief title describing the session")
    summary: Optional[str] = Field(None, description="Markdown summary of the session")
    is_billable: Optional[bool] = Field(None, description="Whether this session is billable")
    billable_hours: Optional[float] = Field(None, description="Billable hours if applicable")
    technician: Optional[str] = Field(None, description="Name of technician who performed the work")
    session_log_file: Optional[str] = Field(None, description="Path to markdown session log file")
    notes: Optional[str] = Field(None, description="Additional notes about the session")


class SessionResponse(SessionBase):
    """Schema for Session responses with ID and timestamps."""

    id: UUID = Field(..., description="Unique identifier for the session")
    created_at: datetime = Field(..., description="Timestamp when the session was created")
    updated_at: datetime = Field(..., description="Timestamp when the session was last updated")

    model_config = {"from_attributes": True}
