bambuddy/backend/app/schemas/archive.py

from datetime import datetime
from typing import Annotated

from pydantic import BaseModel, BeforeValidator, Field, model_validator

from backend.app.schemas.not_null import IgnoresNullForRequiredColumns
from backend.app.utils.filename import clean_display_name

# Free text, punctuation and all -- only control characters are taken out, and
# only on the way in (#2832). Anything that turns a name into a path sanitises
# it there instead, where the budget and the fallback are known.
DisplayName = Annotated[str | None, BeforeValidator(clean_display_name)]


class ArchiveBase(BaseModel):
    print_name: DisplayName = None
    is_favorite: bool | None = None
    tags: str | None = None
    notes: str | None = None
    # Added to every total: a negative (or infinite) cost would wreck them
    cost: Annotated[float | None, Field(ge=0, le=1_000_000)] = None
    failure_reason: str | None = None
    # Number of items printed. 0 is a legal answer -- a plate that jammed and
    # came off ruined produced nothing, and the project's completed-items count
    # sums this column (#3051). Bounded for the same reason as the grams below:
    # it feeds project totals, and a negative would subtract from them.
    quantity: Annotated[int | None, Field(ge=0, le=10_000)] = None
    # User-defined link (Printables, Thingiverse, etc.)
    external_url: str | None = None


class ArchiveUpdate(IgnoresNullForRequiredColumns, ArchiveBase):
    __orm_table__ = "print_archives"
    printer_id: int | None = None
    project_id: int | None = None
    # A finished outcome only. 'printing' would be closed again at the next
    # reconnect as a missed completion, with notifications and a run of its own.
    status: str | None = Field(default=None, pattern="^(completed|failed|aborted|cancelled)$")
    # Post-print quality verdict (#1898): 'good' / 'reject'; null clears it.
    user_verdict: str | None = Field(default=None, pattern="^(good|reject)$")
    # Which surface the verdict came from, for the "recorded when the plate was
    # cleared" hint. Only the sources a client can honestly claim are accepted;
    # 'link', 'plate_clear' and 'reaction' are stamped server-side by the paths
    # that own them and must not be forgeable over this route. Omitted means
    # 'api' — some script or integration did it.
    user_verdict_source: str | None = Field(default=None, pattern="^(dialog|printer_card|api)$")
    # Editable because a print archived without its 3MF has no figure at all,
    # and nothing else can supply one after the fact -- the figure is read from
    # a 3MF this archive does not have (#1820). Bounded because it feeds the filament
    # totals: 100 kg is far past any single print and well short of a value
    # that would swamp a chart.
    filament_used_grams: Annotated[float | None, Field(ge=0, le=100_000)] = None
    # The run (print log entry) a verdict, status or failure reason is for:
    # plates of one archive are runs of their own. Omitted means the
    # archive's latest run.
    run_id: int | None = None


class ArchiveDuplicate(BaseModel):
    """Reference to a duplicate archive."""

    id: int
    print_name: str | None
    created_at: datetime | None
    match_type: str  # "exact" (hash match) or "similar" (name match)


class ArchiveResponse(BaseModel):
    id: int
    printer_id: int | None
    project_id: int | None = None
    project_name: str | None = None  # Included for convenience
    filename: str
    file_path: str
    file_size: int
    content_hash: str | None
    thumbnail_path: str | None
    timelapse_path: str | None
    source_3mf_path: str | None = None  # Original project 3MF from slicer
    f3d_path: str | None = None  # Fusion 360 design file

    # Duplicate detection
    duplicates: list[ArchiveDuplicate] | None = None
    duplicate_count: int = 0  # Quick count for list views
    duplicate_sequence: int = 0  # 0 = original, 1+ = nth duplicate
    original_archive_id: int | None = None  # ID of the first/original archive

    # Object count (computed from extra_data.printable_objects)
    object_count: int | None = None

    print_name: str | None
    plate_id: int | None = None  # Selected plate of a multi-plate 3MF (#2603)
    print_time_seconds: int | None  # Estimated time from slicer
    actual_time_seconds: int | None = None  # Computed from started_at/completed_at
    # Percentage: 100 = perfect, >100 = faster than estimated
    time_accuracy: float | None = None
    filament_used_grams: float | None
    filament_type: str | None
    filament_color: str | None
    layer_height: float | None
    total_layers: int | None = None
    nozzle_diameter: float | None
    bed_temperature: int | None
    bed_type: str | None = None  # e.g. "Cool Plate", "Textured PEI Plate" (from 3MF curr_bed_type)
    nozzle_temperature: int | None

    sliced_for_model: str | None = None  # Printer model this file was sliced for

    status: str
    started_at: datetime | None
    completed_at: datetime | None

    extra_data: dict | None

    makerworld_url: str | None
    designer: str | None
    # User-defined link (Printables, Thingiverse, etc.)
    external_url: str | None = None

    is_favorite: bool
    tags: str | None
    notes: str | None
    cost: float | None
    photos: list | None
    failure_reason: str | None
    quantity: int = 1  # Number of items printed

    # Post-print outcome confirmation (#1898)
    user_verdict: str | None = None
    user_verdict_source: str | None = None
    user_verdict_at: datetime | None = None
    confirm_requested: bool = False
    # The run (print log entry) whose outcome this response shows, when it
    # shows one run of the archive rather than its latest (#1898).
    run_id: int | None = None

    # Energy tracking
    energy_kwh: float | None = None
    energy_cost: float | None = None
    wear_cost: float | None = None  # Printer wear, #694

    created_at: datetime | None

    # User tracking (Issue #206)
    created_by_id: int | None = None
    created_by_username: str | None = None

    # Per-archive run aggregates (#1378). Computed from PrintLogEntry — one
    # row per actual print event — so reprints contribute to these counters
    # without overwriting the source archive's first-run data.
    run_count: int = 0
    last_run_at: datetime | None = None
    total_filament_actual_grams: float | None = None
    successful_run_count: int = 0
    failed_run_count: int = 0

    @model_validator(mode="after")
    def compute_object_count(self) -> "ArchiveResponse":
        """Compute object_count from extra_data.printable_objects if not set."""
        if self.object_count is None and self.extra_data:
            printable_objects = self.extra_data.get("printable_objects")
            if printable_objects and isinstance(printable_objects, dict):
                self.object_count = len(printable_objects)
        return self

    class Config:
        from_attributes = True


class ArchiveSlim(BaseModel):
    """Lightweight archive response for stats/dashboard widgets."""

    printer_id: int | None
    print_name: str | None
    print_time_seconds: int | None
    actual_time_seconds: int | None = None
    filament_used_grams: float | None
    filament_type: str | None
    filament_color: str | None
    status: str
    started_at: datetime | None
    completed_at: datetime | None
    cost: float | None
    energy_kwh: float | None = None
    energy_cost: float | None = None
    wear_cost: float | None = None  # Printer wear, #694
    quantity: int = 1
    created_at: datetime | None

    class Config:
        from_attributes = True


class ArchiveStats(BaseModel):
    total_prints: int
    successful_prints: int
    failed_prints: int
    # User/system-stopped prints (PrintLogEntry.status in stopped/cancelled/
    # skipped). Defaulted so older clients that don't send this field still
    # validate against historical fixtures.
    cancelled_prints: int = 0
    total_print_time_hours: float
    total_filament_grams: float
    total_cost: float
    prints_by_filament_type: dict
    prints_by_printer: dict
    # Name each printer id was last recorded under in the print log. Lets the
    # client keep labelling history that belongs to a deleted printer (#2873);
    # a printer that still exists is named from the live record instead.
    printer_names: dict[str, str] = {}
    # Time accuracy stats
    # Average across all prints with data
    average_time_accuracy: float | None = None
    time_accuracy_by_printer: dict | None = None  # Per-printer accuracy
    # Energy stats
    total_energy_kwh: float = 0.0
    total_energy_cost: float = 0.0
    # Printer wear across the runs counted above (#694)
    total_wear_cost: float = 0.0
    # Set when the date-range query in "total consumption" mode is running on
    # incomplete snapshot history — e.g. right after a fresh upgrade before the
    # hourly snapshot loop has built up a baseline. Frontend shows a tooltip.
    energy_data_warming_up: bool = False


class ProjectPageImage(BaseModel):
    """Image embedded in 3MF project page."""

    name: str
    path: str  # Path within 3MF
    url: str  # API URL to fetch image


class ProjectPageResponse(BaseModel):
    """Project page data extracted from 3MF file."""

    # Model info
    title: str | None = None
    description: str | None = None  # HTML content
    designer: str | None = None
    designer_user_id: str | None = None
    license: str | None = None
    copyright: str | None = None
    creation_date: str | None = None
    modification_date: str | None = None
    origin: str | None = None  # "original" or "remix"

    # Profile info
    profile_title: str | None = None
    profile_description: str | None = None
    profile_cover: str | None = None
    profile_user_id: str | None = None
    profile_user_name: str | None = None

    # MakerWorld info
    design_model_id: str | None = None
    design_profile_id: str | None = None
    design_region: str | None = None

    # Images
    model_pictures: list[ProjectPageImage] = []
    profile_pictures: list[ProjectPageImage] = []
    thumbnails: list[ProjectPageImage] = []