from __future__ import annotations

from datetime import datetime

from pydantic import BaseModel, EmailStr, Field

class Token(BaseModel):

    access_token: str

    token_type: str = "bearer"

class UserCreate(BaseModel):

    email: EmailStr

    password: str = Field(min_length=8)

    full_name: str | None = None

    role: str = "analyst"

class UserOut(BaseModel):

    id: str

    email: str

    full_name: str | None

    role: str

    is_active: bool

    class Config:

        from_attributes = True

class InvestigationCreate(BaseModel):

    title: str

    summary: str | None = None

    priority: str = "normal"

    organization: str | None = None

class InvestigationOut(BaseModel):

    id: str

    title: str

    status: str

    priority: str

    summary: str | None

    organization: str | None

    created_at: datetime

    class Config:

        from_attributes = True

class NoteCreate(BaseModel):

    body: str

    pinned: bool = False

class SearchQuery(BaseModel):

    plate: str | None = None

    partial_plate: str | None = None

    make: str | None = None

    color: str | None = None

    body_type: str | None = None

    camera_id: str | None = None

    from_ts: datetime | None = None

    to_ts: datetime | None = None

    bbox: list[float] | None = None

    incident_ref: str | None = None

    limit: int = 200

class WatchlistCreate(BaseModel):

    name: str

    kind: str = "plate"

    organization: str | None = None

class WatchlistEntryCreate(BaseModel):

    plate_norm: str | None = None

    partial_plate: str | None = None

    vehicle_desc: dict | None = None

    reason: str | None = None

    expires_at: datetime | None = None

class EnrichRunRequest(BaseModel):

    enricher: str

    target: str

    investigation_id: str | None = None

class ReportRequest(BaseModel):

    title: str | None = None

    sections: list[str] | None = None

    format: str = "markdown"

class SourceCreate(BaseModel):

    name: str

    system_type: str | None = None

    organization: str | None = None

    jurisdiction: str | None = None

    trust_level: float = 0.8