diff --git a/BETA_TEST_PLAN.md b/BETA_TEST_PLAN.md new file mode 100644 index 000000000..62e679345 --- /dev/null +++ b/BETA_TEST_PLAN.md @@ -0,0 +1,139 @@ +# Beta Test Plan — Spool Inventory & Related Features + +## Prerequisites + +- At least one printer connected with AMS +- At least one Bambu Lab spool (RFID) loaded in AMS +- At least one spool without RFID tag +- At least one empty AMS slot (or slot with non-BL filament) +- A 3MF file ready to print (small/fast test print recommended) + +--- + +## 1. Filament Tracking Mode Switching + +**Location:** Settings > Filament + +| # | Test | Steps | Expected | +|---|------|-------|----------| +| 1.1 | Default mode | Open Settings > Filament | "Built-in Inventory" card is selected (green border), info panel shows RFID/usage/catalog bullet points | +| 1.2 | Switch to Spoolman | Click "Spoolman" card | Spoolman config appears (URL input, Sync Mode, connection status). Built-in info panel disappears | +| 1.3 | Switch back | Click "Built-in Inventory" card | Spoolman config disappears, built-in info panel reappears | +| 1.4 | Persistence | Switch mode, reload page | Selected mode persists after reload | +| 1.5 | Spoolman disabled state | Select Spoolman without URL, check Connect button | Connect button should be disabled when URL is empty | + +--- + +## 2. Spool Management + +**Location:** Inventory page (sidebar) + +| # | Test | Steps | Expected | +|---|------|-------|----------| +| 2.1 | Add spool | Click "+ Add Spool", fill material/brand/color/weight, save | Spool appears in table with correct info | +| 2.2 | Edit spool | Click a spool row, change fields, save | Changes reflected in table | +| 2.3 | Remaining weight | Edit spool > Additional section > adjust remaining weight | Remaining weight = label_weight - weight_used, slider/input updates correctly | +| 2.4 | Spool catalog | Edit spool > Additional > Empty Spool Weight dropdown | Pre-defined spool weights shown, selecting one updates the field | +| 2.5 | Color picker | Edit spool > Color section | Recent colors, brand palettes, and hex input all work | +| 2.6 | PA profile tab | Edit spool > PA Profile tab | Matching K-profiles shown grouped by printer/nozzle (requires calibration data) | +| 2.7 | Archive spool | Archive a spool from context menu or edit | Spool moves to "Archived" tab | +| 2.8 | Delete spool | Delete a spool | Spool removed from all views | +| 2.9 | Summary cards | Check top of inventory page | Total Inventory, Total Consumed, By Material, In Printer, Low Stock cards show correct values | +| 2.10 | Filters | Try Active/Archived/All tabs, material/brand dropdowns, search | Filtering works correctly | +| 2.11 | View modes | Toggle between Table and Card view | Both views show correct spool data | + +--- + +## 3. AMS Slot Assignment + +**Location:** Printers page, AMS hover cards + +| # | Test | Steps | Expected | +|---|------|-------|----------| +| 3.1 | Assign spool | Hover over an empty/non-BL AMS slot > click "Assign Spool" | Modal opens showing only manual (non-BL) spools | +| 3.2 | BL spools filtered | Open assign modal | Bambu Lab spools (with RFID tags) are NOT in the list | +| 3.3 | Assigned spools filtered | Assign spool A to slot 1, then open assign modal for slot 2 | Spool A is NOT in the list (already assigned) | +| 3.4 | Current slot spool visible | Open assign for slot that already has a spool | The currently assigned spool IS still shown (for reassignment) | +| 3.5 | Confirm assignment | Select spool, click "Assign Spool" | Slot shows the assigned spool info on hover | +| 3.6 | Unassign spool | Hover over assigned slot > click "Unassign" | Spool removed from slot, slot shows default info | +| 3.7 | BL slot — no buttons | Hover over a slot with a Bambu Lab spool (RFID) | No "Assign Spool" or "Unassign" buttons shown | +| 3.8 | Empty slot display | Check an empty AMS slot | Shows type from AMS data or "Empty" (localized) | + +--- + +## 4. Auto-Unlink on BL Spool Insertion + +| # | Test | Steps | Expected | +|---|------|-------|----------| +| 4.1 | BL spool replaces manual | Assign a manual spool to a slot, then physically insert a BL spool into that slot | Assignment automatically removed, slot now shows BL spool info | +| 4.2 | Log message | Check backend logs after 4.1 | Log shows "Auto-unlink: spool X AMSY-TZ — Bambu Lab spool detected" | + +--- + +## 5. Spool Tag Linking + +**Location:** Printers page, AMS hover cards for BL spools + +| # | Test | Steps | Expected | +|---|------|-------|----------| +| 5.1 | Link modal | Hover over BL spool slot > click "Link to Spool" (if available) | Modal opens showing only untagged spools | +| 5.2 | Search filter | Type in search box | Spools filtered by material/brand/color | +| 5.3 | Link spool | Click a spool in the list | Success toast, modal closes, spool now linked to tag | +| 5.4 | Tagged spools hidden | After linking spool A, open link modal again | Spool A is no longer in the list | + +--- + +## 6. Usage Tracking — BL Spools (AMS Remain%) + +| # | Test | Steps | Expected | +|---|------|-------|----------| +| 6.1 | Session capture | Start a print with a BL spool | Backend log shows "Captured start remain% for printer X (N trays)" | +| 6.2 | Completed print | Let print finish | Spool's weight_used increases by (delta% * label_weight). Check inventory page remaining weight | +| 6.3 | Failed/aborted print | Start and cancel a print mid-way | Usage still tracked based on remain% delta at time of stop | +| 6.4 | No double-tracking | Complete a print with BL spool | Only AMS delta tracking applies (no 3MF fallback for this spool) | + +--- + +## 7. Usage Tracking — Non-BL Spools (3MF Estimates) + +| # | Test | Steps | Expected | +|---|------|-------|----------| +| 7.1 | Assign and print | Assign a manual spool to AMS slot, start a print using that slot | Backend log shows "3MF fallback available" at print start | +| 7.2 | Completed print | Let print finish | Spool's weight_used increases by the 3MF estimated used_g. Check inventory page | +| 7.3 | Failed print scaling | Start a print, cancel at ~50% | Usage = 3MF estimate * (progress/100). E.g., 100g estimate at 50% = ~50g tracked | +| 7.4 | Multi-slot print | Print using multiple filament slots (some BL, some manual) | BL spools tracked via remain% delta, manual spools via 3MF. No double-counting | +| 7.5 | Slot mapping | Use a spool in AMS slot 5 (second AMS unit) | Correctly maps to AMS 1, Tray 0. Check usage history shows correct ams_id/tray_id | + +--- + +## 8. Edge Cases + +| # | Test | Steps | Expected | +|---|------|-------|----------| +| 8.1 | No AMS data | Print from external spool (no AMS) | No crash, usage tracking gracefully skipped | +| 8.2 | No archive | Print without 3MF archive available | AMS delta path still works for BL spools, 3MF path skipped gracefully | +| 8.3 | Spool refilled | If remain% goes UP between start and end (spool swapped/refilled) | Negative delta skipped, no negative usage recorded | +| 8.4 | Rapid print start/stop | Start and immediately cancel a print | No errors, minimal or zero usage tracked | +| 8.5 | Concurrent printers | Print on two printers simultaneously | Each printer tracked independently, no cross-contamination | + +--- + +## 9. UI/UX Checks + +| # | Test | Steps | Expected | +|---|------|-------|----------| +| 9.1 | Mobile layout | Open on mobile or narrow browser | Inventory page, assign modal, and settings all responsive | +| 9.2 | Dark/light mode | Toggle theme | All inventory UI elements properly themed | +| 9.3 | Language switching | Switch to German and Japanese | All inventory strings translated (no hardcoded English) | +| 9.4 | Keyboard nav | Use keyboard shortcuts on printers page | AMS slot interactions accessible | + +--- + +## Reporting Issues + +When reporting a bug, please include: +- Test case number (e.g., "Beta 9.1 failed") +- Browser + version +- Screenshot or screen recording +- Backend logs (if relevant — check Docker logs) +- Steps to reproduce if different from above diff --git a/CHANGELOG.md b/CHANGELOG.md index 9eae7b17f..ed605fd90 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -2,6 +2,29 @@ All notable changes to Bambuddy will be documented in this file. +## [0.2.0b] - Not released + +### New Features +- **Spool Inventory — AMS Slot Assignment** — Assign inventory spools to AMS slots for filament tracking. Hover over any non-Bambu-Lab AMS slot to assign or unassign spools. The assign modal filters out Bambu Lab spools (tracked via RFID) and spools already assigned to other slots. Bambu Lab spool slots automatically hide assign/unassign UI since they are managed by the AMS. When a Bambu Lab spool is inserted into a slot with a manual assignment, the assignment is automatically unlinked. +- **Spool Inventory — Remaining Weight Editing** — Edit the remaining filament weight when adding or editing a spool. The new "Remaining Weight" field in the Additional section shows current weight (label weight minus consumed) with a max reference. Edits are stored as `weight_used` internally. +- **Spool Inventory — 3MF-Based Usage Tracking for Non-BL Spools** — Non-Bambu-Lab spools (no RFID) cannot use AMS remain% for usage tracking. Now falls back to per-filament weight estimates from the archived 3MF file (`used_g` per filament slot). For completed prints, uses the full slicer estimate. For failed or aborted prints, scales by print progress percentage. Bambu Lab spools continue using AMS remain% delta tracking as before. + +### Fixed +- **External Camera Not Used for Snapshot + Stream Dropping** ([#325](https://github.com/maziggy/bambuddy/issues/325)) — The snapshot endpoint (`/camera/snapshot`) always used the internal printer camera even when an external camera was configured. Now checks for external camera first, matching the existing stream endpoint behavior. Also fixed external MJPEG and RTSP streams silently dropping every ~60 seconds due to missing reconnect logic — the underlying stream generators exit on read timeout, and the caller now retries up to 3 times with a 2-second delay instead of ending the stream. +- **H2C Nozzle Rack Text Unreadable on Light Filament Colors** ([#300](https://github.com/maziggy/bambuddy/issues/300)) — Nozzle rack slots use the loaded filament color as background, but white/light filaments made the white "0.4" text nearly invisible. Now uses a luminance check to switch to dark text on light backgrounds. +- **File Downloads Show Generic Filenames** ([#334](https://github.com/maziggy/bambuddy/issues/334)) — Downloaded files with special characters in their names (spaces, umlauts, parentheses) were saved as generic `file_1`, `file_2` instead of the original filename. The `Content-Disposition` header parser now handles RFC 5987 percent-encoded filenames (`filename*=utf-8''...`) used by FastAPI for non-ASCII characters. Fix applied to all download endpoints (library files, archives, source files, F3D files, project exports, support bundles, printer files). +- **Printer Card Cover Image Not Updating Between Prints** — The cover image on the printer card only refreshed on page reload. The `` URL was always the same (`/printers/{id}/cover`) regardless of which print was active, so the browser served its cached image. Now appends the print name as a cache-busting query parameter so the browser fetches the new cover when a different print starts. +- **Telegram Bold Title Broken by Underscores in Message** ([#332](https://github.com/maziggy/bambuddy/issues/332)) — Telegram notifications showed literal `*Title*` asterisks instead of bold text when the message body contained underscores (e.g. job name `A1_plate_8`, error code `0300_0001`). The code was disabling Markdown parsing entirely when underscores were detected. Now escapes underscores in the body with `\_` so Markdown rendering stays enabled. +- **Queued Jobs Incorrectly Archived After Duplicate Execution Detection** ([#341](https://github.com/maziggy/bambuddy/issues/341)) — When the same file was added to the print queue multiple times, only the first job executed. All subsequent jobs were automatically skipped with "already printed X hours ago" because they shared the same archive reference, and a safety check incorrectly treated them as phantom reprints. The same issue also affected single queue items created from recently completed archives. Removed the overly broad 4-hour duplicate detection check — the crash recovery scenario it guarded against is already handled by the queue item status lifecycle. + +### New Features +- **External Links: Open in New Tab** ([#338](https://github.com/maziggy/bambuddy/issues/338)) — External sidebar links can now optionally open in a new browser tab instead of an iframe. Sites behind reverse proxies (Traefik, nginx) that send `X-Frame-Options: SAMEORIGIN` or CSP `frame-ancestors` headers block iframe embedding, causing "refused to connect" errors. A new "Open in new tab" toggle in the add/edit link modal lets users choose per-link. Keyboard shortcuts (number keys) also respect the setting. Defaults to iframe (existing behavior) for backward compatibility. +- **Print Queue: Clear Plate Confirmation** — When a print finishes or fails and more items are queued, the printer card now shows a "Clear Plate & Start Next" button. The scheduler no longer auto-starts the next print while the printer is in FINISH or FAILED state — the user must confirm the build plate has been cleared first. This prevents prints from starting on a dirty plate. The button respects the `printers:control` permission and is available in all supported languages (en/de/ja). + +### Improved +- **Additional Currency Options** ([#329](https://github.com/maziggy/bambuddy/issues/329), [#333](https://github.com/maziggy/bambuddy/issues/333)) — Added 17 additional currencies to the cost tracking dropdown: HKD, INR, KRW, SEK, NOK, DKK, PLN, BRL, TWD, SGD, NZD, MXN, CZK, THB, ZAR, RUB. +- **Move Email Settings Under Authentication Tab** — Renamed the settings "Users" tab to "Authentication" and moved the standalone "Global Email" tab into it as an "Email Authentication" sub-tab. Groups email/SMTP configuration with user management where it logically belongs. Legacy `?tab=email` URLs are handled automatically. + ## [0.1.9] - 2026-02-10 ### New Features @@ -37,6 +60,7 @@ All notable changes to Bambuddy will be documented in this file. - **Virtual Printer IP Override for Server Mode** ([#52](https://github.com/maziggy/bambuddy/issues/52)) — The `remote_interface_ip` setting (network interface override) was only used in proxy mode, but users with multiple network interfaces (LAN + Tailscale, Docker bridges) also needed it in server modes (immediate/review/print_queue). Auto-detected IP from `_get_local_ip()` followed the OS default route, causing wrong IP in TLS certificate SAN (handshake failures) and SSDP broadcasts (slicer can't discover printer). Now the interface override applies to all modes: included in certificate SAN, passed to SSDP server as advertise IP, and triggers service restart on change. UI dropdown shown for all modes when enabled (not just proxy). - **Wrong Thumbnail When Reprinting Same Project** ([#314](https://github.com/maziggy/bambuddy/issues/314)) — Reprinting a project with the same name but a different bed layout showed the old thumbnail during printing. The cover image cache was keyed by `subtask_name` and never invalidated between prints, so a cache hit returned the stale first-print thumbnail. Now the cover cache is cleared on every print start. - **Wrong Timelapse Attached to Archive** ([#315](https://github.com/maziggy/bambuddy/issues/315)) — After a print, the archive could receive a timelapse from a previous print instead of the just-completed one. The auto-scan sorted MP4 files by mtime and grabbed the "most recent," but in LAN-only mode (no NTP) the printer's clock is wrong, making mtime unreliable. Replaced with a snapshot-diff approach: baseline existing files before waiting, then detect the new file that appears after encoding. Falls back to print-name matching if no new file is found after retries. +- **Timelapse Not Attached — Baseline Race Condition** ([#315](https://github.com/maziggy/bambuddy/issues/315)) — Follow-up to the snapshot-diff timelapse fix: the baseline of existing MP4 files was captured at print completion time inside a background task, but fast-encoding printers could finish writing the timelapse before the baseline was taken, causing the new file to appear in the baseline and never be detected as "new." Moved baseline capture to print start time, when the timelapse file cannot possibly exist yet. Falls back to completion-time baseline if the app was restarted mid-print. - **Calibration Prints Archived** ([#315](https://github.com/maziggy/bambuddy/issues/315)) — Standalone calibration prints (flow, vibration, bed leveling) were being archived as regular prints. The calibration gcode (`/usr/etc/print/auto_cali_for_user.gcode`) and other internal printer files under `/usr/` are now detected and skipped during print start. - **Camera Stop 401 When Auth Enabled** — Camera stop requests (`sendBeacon`) failed with 401 Unauthorized when authentication was enabled because `sendBeacon` cannot send auth headers. Replaced with `fetch` + `keepalive: true` which supports Authorization headers while remaining reliable during page unload. - **Spoolman Creates Duplicate Spools on Startup** ([#295](https://github.com/maziggy/bambuddy/pull/295)) — Each AMS tray independently fetched all spools from Spoolman, causing redundant API calls and duplicate spool creation with large databases (300+ spools). Now fetches spools once and reuses cached data across all tray operations. Added retry logic (3 attempts, 500ms delay) with connection recreation for transient network errors. diff --git a/DOCKERHUB.md b/DOCKERHUB.md index bcaef6d61..fe3770ffc 100644 --- a/DOCKERHUB.md +++ b/DOCKERHUB.md @@ -92,7 +92,7 @@ docker compose pull && docker compose up -d | Series | Models | Status | |---|---|---| -| H2 | H2D | Tested | +| H2 | H2C, H2D, H2D Pro, H2S | Tested | | X1 | X1 Carbon, X1E | Tested | | P1 | P1P, P1S | Compatible | | P2 | P2S | Compatible | diff --git a/README.md b/README.md index f9eb6d842..de9befc33 100644 --- a/README.md +++ b/README.md @@ -103,6 +103,7 @@ Perfect for remote print farms, traveling makers, or accessing your home printer - Per-printer AMS mapping (individual slot configuration for print farms) - Scheduled prints (date/time) - Queue Only mode (stage without auto-start) +- Clear plate confirmation between queued prints - Smart plug integration (Tasmota, Home Assistant, MQTT) - MQTT smart plugs: Subscribe to Zigbee2MQTT, Shelly, or any MQTT topic for energy monitoring - Energy consumption tracking (per-print kWh and cost) @@ -145,6 +146,11 @@ Perfect for remote print farms, traveling makers, or accessing your home printer - Build plate detection alerts - Queue events (waiting, skipped, failed) +### 🧵 Spool Inventory +- Built-in spool inventory with AMS slot assignment, usage tracking, and remaining weight management +- Automatic filament consumption tracking: AMS RFID for Bambu Lab spools, 3MF estimates for third-party spools +- Spool catalog, color catalog, PA profile matching, and low-stock alerts + ### 🔧 Integrations - [Spoolman](https://github.com/Donkie/Spoolman) filament sync with per-filament usage tracking and fill level display - MQTT publishing for Home Assistant, Node-RED, etc. diff --git a/backend/app/api/routes/camera.py b/backend/app/api/routes/camera.py index 1fae52eb2..265bc0942 100644 --- a/backend/app/api/routes/camera.py +++ b/backend/app/api/routes/camera.py @@ -547,6 +547,25 @@ async def camera_snapshot( printer = await get_printer_or_404(printer_id, db) + # Check for external camera first + if printer.external_camera_enabled and printer.external_camera_url: + from backend.app.services.external_camera import capture_frame + + frame_data = await capture_frame(printer.external_camera_url, printer.external_camera_type, timeout=15) + if not frame_data: + raise HTTPException( + status_code=503, + detail="Failed to capture frame from external camera.", + ) + return Response( + content=frame_data, + media_type="image/jpeg", + headers={ + "Cache-Control": "no-cache, no-store, must-revalidate", + "Content-Disposition": f'inline; filename="snapshot_{printer_id}.jpg"', + }, + ) + # Create temporary file for the snapshot with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as f: temp_path = Path(f.name) diff --git a/backend/app/api/routes/cloud.py b/backend/app/api/routes/cloud.py index f3dbbaf28..898b731f2 100644 --- a/backend/app/api/routes/cloud.py +++ b/backend/app/api/routes/cloud.py @@ -246,11 +246,12 @@ async def get_slicer_settings( for api_key, our_type in type_mapping.items(): type_data = data.get(api_key, {}) - # Combine public and private presets, private (user's own) first - all_settings = type_data.get("private", []) + type_data.get("public", []) + private_settings = type_data.get("private", []) + public_settings = type_data.get("public", []) parsed = [] - for s in all_settings: + # Private (custom) presets first + for s in private_settings: parsed.append( SlicerSetting( setting_id=s.get("setting_id", s.get("id", "")), @@ -259,6 +260,20 @@ async def get_slicer_settings( version=s.get("version"), user_id=s.get("user_id"), updated_time=s.get("updated_time"), + is_custom=True, + ) + ) + # Public (default) presets + for s in public_settings: + parsed.append( + SlicerSetting( + setting_id=s.get("setting_id", s.get("id", "")), + name=s.get("name", "Unknown"), + type=our_type, + version=s.get("version"), + user_id=s.get("user_id"), + updated_time=s.get("updated_time"), + is_custom=False, ) ) setattr(result, our_type, parsed) @@ -302,6 +317,22 @@ async def get_setting_detail( raise HTTPException(status_code=500, detail=str(e)) +@router.get("/filaments", response_model=list[SlicerSetting]) +async def get_filament_presets( + version: str = "02.04.00.70", + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.FILAMENTS_READ), +): + """ + Get just filament presets (convenience endpoint). + + Returns all filament presets with custom presets first. + Uses the same cache as get_slicer_settings. + """ + settings = await get_slicer_settings(version=version, db=db) + return settings.filament + + # Cache for filament preset info (setting_id -> {name, k}) _filament_cache: dict[str, dict] = {} _filament_cache_time: float = 0 @@ -844,6 +875,20 @@ def _load_fields(preset_type: str) -> dict: return data +@router.get("/builtin-filaments") +async def get_builtin_filaments( + _: User | None = RequirePermissionIfAuthEnabled(Permission.FILAMENTS_READ), +): + """ + Get built-in filament names as a fallback source. + + Returns the static _BUILTIN_FILAMENT_NAMES table as a list of + {filament_id, name} objects. Used by the frontend when cloud + and local profiles are unavailable. + """ + return [{"filament_id": fid, "name": name} for fid, name in _BUILTIN_FILAMENT_NAMES.items()] + + @router.get("/fields/{preset_type}") async def get_preset_fields( preset_type: Literal["filament", "print", "process", "printer"], diff --git a/backend/app/api/routes/inventory.py b/backend/app/api/routes/inventory.py new file mode 100644 index 000000000..813a57f12 --- /dev/null +++ b/backend/app/api/routes/inventory.py @@ -0,0 +1,946 @@ +import json +import logging + +import httpx +from fastapi import APIRouter, Depends, HTTPException +from fastapi.responses import StreamingResponse +from pydantic import BaseModel +from sqlalchemy import func, select +from sqlalchemy.ext.asyncio import AsyncSession +from sqlalchemy.orm import selectinload + +from backend.app.core.auth import RequirePermissionIfAuthEnabled +from backend.app.core.catalog_defaults import DEFAULT_COLOR_CATALOG, DEFAULT_SPOOL_CATALOG +from backend.app.core.database import get_db +from backend.app.core.permissions import Permission +from backend.app.models.color_catalog import ColorCatalogEntry +from backend.app.models.spool import Spool +from backend.app.models.spool_assignment import SpoolAssignment +from backend.app.models.spool_catalog import SpoolCatalogEntry +from backend.app.models.spool_k_profile import SpoolKProfile +from backend.app.models.user import User +from backend.app.schemas.spool import ( + SpoolAssignmentCreate, + SpoolAssignmentResponse, + SpoolCreate, + SpoolKProfileBase, + SpoolKProfileResponse, + SpoolResponse, + SpoolUpdate, +) +from backend.app.schemas.spool_usage import SpoolUsageHistoryResponse + +logger = logging.getLogger(__name__) + +router = APIRouter(prefix="/inventory", tags=["inventory"]) + +# Material temperature defaults (nozzle min/max) +MATERIAL_TEMPS: dict[str, tuple[int, int]] = { + "PLA": (190, 230), + "PETG": (220, 260), + "ABS": (240, 270), + "ASA": (240, 270), + "TPU": (200, 240), + "PA": (260, 290), + "PC": (250, 280), + "PVA": (190, 210), + "PLA-CF": (210, 240), + "PETG-CF": (240, 270), + "PA-CF": (270, 300), +} + +# FilamentColors.xyz API +FILAMENT_COLORS_API = "https://filamentcolors.xyz/api" + + +# ── Spool Catalog Schemas ────────────────────────────────────────────────── + + +class CatalogEntryResponse(BaseModel): + id: int + name: str + weight: int + is_default: bool + + class Config: + from_attributes = True + + +class CatalogEntryCreate(BaseModel): + name: str + weight: int + + +class CatalogEntryUpdate(BaseModel): + name: str + weight: int + + +# ── Color Catalog Schemas ────────────────────────────────────────────────── + + +class ColorEntryResponse(BaseModel): + id: int + manufacturer: str + color_name: str + hex_color: str + material: str | None + is_default: bool + + class Config: + from_attributes = True + + +class ColorEntryCreate(BaseModel): + manufacturer: str + color_name: str + hex_color: str + material: str | None = None + + +class ColorEntryUpdate(BaseModel): + manufacturer: str + color_name: str + hex_color: str + material: str | None = None + + +class ColorLookupResult(BaseModel): + found: bool + hex_color: str | None = None + material: str | None = None + + +# ── Spool Catalog CRUD ───────────────────────────────────────────────────── + + +@router.get("/catalog", response_model=list[CatalogEntryResponse]) +async def get_spool_catalog( + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """Get all spool catalog entries.""" + result = await db.execute(select(SpoolCatalogEntry).order_by(SpoolCatalogEntry.name)) + return list(result.scalars().all()) + + +@router.post("/catalog", response_model=CatalogEntryResponse) +async def add_catalog_entry( + entry: CatalogEntryCreate, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Add a new spool catalog entry.""" + row = SpoolCatalogEntry(name=entry.name, weight=entry.weight, is_default=False) + db.add(row) + await db.commit() + await db.refresh(row) + return row + + +@router.put("/catalog/{entry_id}", response_model=CatalogEntryResponse) +async def update_catalog_entry( + entry_id: int, + entry: CatalogEntryUpdate, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Update a spool catalog entry.""" + result = await db.execute(select(SpoolCatalogEntry).where(SpoolCatalogEntry.id == entry_id)) + row = result.scalar_one_or_none() + if not row: + raise HTTPException(404, "Entry not found") + row.name = entry.name + row.weight = entry.weight + await db.commit() + await db.refresh(row) + return row + + +@router.delete("/catalog/{entry_id}") +async def delete_catalog_entry( + entry_id: int, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Delete a spool catalog entry.""" + result = await db.execute(select(SpoolCatalogEntry).where(SpoolCatalogEntry.id == entry_id)) + row = result.scalar_one_or_none() + if not row: + raise HTTPException(404, "Entry not found") + await db.delete(row) + await db.commit() + return {"status": "deleted"} + + +@router.post("/catalog/reset") +async def reset_spool_catalog( + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Reset spool catalog to defaults.""" + await db.execute(select(SpoolCatalogEntry)) # ensure table loaded + # Delete all + result = await db.execute(select(SpoolCatalogEntry)) + for row in result.scalars().all(): + await db.delete(row) + # Re-seed defaults + for name, weight in DEFAULT_SPOOL_CATALOG: + db.add(SpoolCatalogEntry(name=name, weight=weight, is_default=True)) + await db.commit() + return {"status": "reset"} + + +# ── Color Catalog CRUD ───────────────────────────────────────────────────── + + +@router.get("/colors", response_model=list[ColorEntryResponse]) +async def get_color_catalog( + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """Get all color catalog entries.""" + result = await db.execute( + select(ColorCatalogEntry).order_by( + ColorCatalogEntry.manufacturer, ColorCatalogEntry.material, ColorCatalogEntry.color_name + ) + ) + return list(result.scalars().all()) + + +@router.post("/colors", response_model=ColorEntryResponse) +async def add_color_entry( + entry: ColorEntryCreate, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Add a new color catalog entry.""" + row = ColorCatalogEntry( + manufacturer=entry.manufacturer, + color_name=entry.color_name, + hex_color=entry.hex_color, + material=entry.material, + is_default=False, + ) + db.add(row) + await db.commit() + await db.refresh(row) + return row + + +@router.put("/colors/{entry_id}", response_model=ColorEntryResponse) +async def update_color_entry( + entry_id: int, + entry: ColorEntryUpdate, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Update a color catalog entry.""" + result = await db.execute(select(ColorCatalogEntry).where(ColorCatalogEntry.id == entry_id)) + row = result.scalar_one_or_none() + if not row: + raise HTTPException(404, "Entry not found") + row.manufacturer = entry.manufacturer + row.color_name = entry.color_name + row.hex_color = entry.hex_color + row.material = entry.material + await db.commit() + await db.refresh(row) + return row + + +@router.delete("/colors/{entry_id}") +async def delete_color_entry( + entry_id: int, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Delete a color catalog entry.""" + result = await db.execute(select(ColorCatalogEntry).where(ColorCatalogEntry.id == entry_id)) + row = result.scalar_one_or_none() + if not row: + raise HTTPException(404, "Entry not found") + await db.delete(row) + await db.commit() + return {"status": "deleted"} + + +@router.post("/colors/reset") +async def reset_color_catalog( + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Reset color catalog to defaults.""" + result = await db.execute(select(ColorCatalogEntry)) + for row in result.scalars().all(): + await db.delete(row) + for manufacturer, color_name, hex_color, material in DEFAULT_COLOR_CATALOG: + db.add( + ColorCatalogEntry( + manufacturer=manufacturer, + color_name=color_name, + hex_color=hex_color, + material=material, + is_default=True, + ) + ) + await db.commit() + return {"status": "reset"} + + +@router.get("/colors/lookup", response_model=ColorLookupResult) +async def lookup_color( + manufacturer: str, + color_name: str, + material: str | None = None, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """Look up a color by manufacturer and color name.""" + query = select(ColorCatalogEntry).where( + ColorCatalogEntry.manufacturer == manufacturer, + ColorCatalogEntry.color_name == color_name, + ) + if material: + query = query.where(ColorCatalogEntry.material == material) + query = query.limit(1) + result = await db.execute(query) + row = result.scalar_one_or_none() + if row: + return ColorLookupResult(found=True, hex_color=row.hex_color, material=row.material) + return ColorLookupResult(found=False) + + +@router.get("/colors/search", response_model=list[ColorEntryResponse]) +async def search_colors( + manufacturer: str | None = None, + material: str | None = None, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """Search colors by manufacturer and/or material.""" + query = select(ColorCatalogEntry) + if manufacturer: + query = query.where(func.lower(ColorCatalogEntry.manufacturer).contains(manufacturer.lower())) + if material: + query = query.where(func.lower(ColorCatalogEntry.material).contains(material.lower())) + query = query.order_by(ColorCatalogEntry.manufacturer, ColorCatalogEntry.color_name).limit(100) + result = await db.execute(query) + return list(result.scalars().all()) + + +@router.post("/colors/sync") +async def sync_from_filamentcolors( + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Sync colors from FilamentColors.xyz API with progress streaming.""" + + async def generate(): + from backend.app.core.database import async_session + + added = 0 + skipped = 0 + total_fetched = 0 + total_available = 0 + + try: + async with httpx.AsyncClient(timeout=120.0) as client: + page = 1 + while True: + response = await client.get( + f"{FILAMENT_COLORS_API}/swatch/", + params={"page": page}, + ) + response.raise_for_status() + data = response.json() + total_available = data.get("count", total_available) + results = data.get("results", []) + if not results: + break + + async with async_session() as db: + for swatch in results: + total_fetched += 1 + manufacturer_data = swatch.get("manufacturer") + manufacturer_name = ( + manufacturer_data.get("name", "") if isinstance(manufacturer_data, dict) else "" + ) + filament_type_data = swatch.get("filament_type") + mat = filament_type_data.get("name", "") if isinstance(filament_type_data, dict) else None + color_name_val = swatch.get("color_name", "") + hex_color_val = swatch.get("hex_color", "") + + if not manufacturer_name or not color_name_val or not hex_color_val: + skipped += 1 + continue + + if not hex_color_val.startswith("#"): + hex_color_val = f"#{hex_color_val}" + + # Check if entry already exists + existing = await db.execute( + select(ColorCatalogEntry) + .where( + ColorCatalogEntry.manufacturer == manufacturer_name, + ColorCatalogEntry.color_name == color_name_val, + ColorCatalogEntry.material == mat, + ) + .limit(1) + ) + if existing.scalar_one_or_none(): + skipped += 1 + else: + db.add( + ColorCatalogEntry( + manufacturer=manufacturer_name, + color_name=color_name_val, + hex_color=hex_color_val.upper(), + material=mat, + is_default=False, + ) + ) + added += 1 + + await db.commit() + + progress = { + "type": "progress", + "added": added, + "skipped": skipped, + "total_fetched": total_fetched, + "total_available": total_available, + } + yield f"data: {json.dumps(progress)}\n\n" + + if not data.get("next") or total_fetched >= total_available: + break + page += 1 + + result = { + "type": "complete", + "added": added, + "skipped": skipped, + "total_fetched": total_fetched, + "total_available": total_available, + } + yield f"data: {json.dumps(result)}\n\n" + + except httpx.HTTPError as e: + logger.error("HTTP error syncing from FilamentColors.xyz: %s", e) + yield f"data: {json.dumps({'type': 'error', 'error': str(e)})}\n\n" + except Exception as e: + logger.error("Error syncing from FilamentColors.xyz: %s", e) + yield f"data: {json.dumps({'type': 'error', 'error': 'Unexpected error during sync'})}\n\n" + + return StreamingResponse(generate(), media_type="text/event-stream") + + +# ── Spool CRUD ─────────────────────────────────────────────────────────────── + + +@router.get("/spools", response_model=list[SpoolResponse]) +async def list_spools( + include_archived: bool = False, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """List all spools, excluding archived by default.""" + query = select(Spool).options(selectinload(Spool.k_profiles)) + if not include_archived: + query = query.where(Spool.archived_at.is_(None)) + query = query.order_by(Spool.material, Spool.brand, Spool.color_name) + result = await db.execute(query) + return list(result.scalars().all()) + + +@router.get("/spools/{spool_id}", response_model=SpoolResponse) +async def get_spool( + spool_id: int, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """Get a single spool with k_profiles.""" + result = await db.execute(select(Spool).options(selectinload(Spool.k_profiles)).where(Spool.id == spool_id)) + spool = result.scalar_one_or_none() + if not spool: + raise HTTPException(404, "Spool not found") + return spool + + +@router.post("/spools", response_model=SpoolResponse) +async def create_spool( + spool_data: SpoolCreate, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Create a new spool.""" + spool = Spool(**spool_data.model_dump()) + db.add(spool) + await db.commit() + await db.refresh(spool) + result = await db.execute(select(Spool).options(selectinload(Spool.k_profiles)).where(Spool.id == spool.id)) + return result.scalar_one() + + +@router.patch("/spools/{spool_id}", response_model=SpoolResponse) +async def update_spool( + spool_id: int, + spool_data: SpoolUpdate, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Update a spool.""" + result = await db.execute(select(Spool).where(Spool.id == spool_id)) + spool = result.scalar_one_or_none() + if not spool: + raise HTTPException(404, "Spool not found") + + for field, value in spool_data.model_dump(exclude_unset=True).items(): + setattr(spool, field, value) + + await db.commit() + result = await db.execute(select(Spool).options(selectinload(Spool.k_profiles)).where(Spool.id == spool_id)) + return result.scalar_one() + + +@router.delete("/spools/{spool_id}") +async def delete_spool( + spool_id: int, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Hard delete a spool.""" + result = await db.execute(select(Spool).where(Spool.id == spool_id)) + spool = result.scalar_one_or_none() + if not spool: + raise HTTPException(404, "Spool not found") + + await db.delete(spool) + await db.commit() + return {"status": "deleted"} + + +@router.post("/spools/{spool_id}/archive", response_model=SpoolResponse) +async def archive_spool( + spool_id: int, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Soft-delete a spool by setting archived_at.""" + from datetime import datetime, timezone + + result = await db.execute(select(Spool).where(Spool.id == spool_id)) + spool = result.scalar_one_or_none() + if not spool: + raise HTTPException(404, "Spool not found") + + spool.archived_at = datetime.now(timezone.utc) + await db.commit() + result = await db.execute(select(Spool).options(selectinload(Spool.k_profiles)).where(Spool.id == spool_id)) + return result.scalar_one() + + +@router.post("/spools/{spool_id}/restore", response_model=SpoolResponse) +async def restore_spool( + spool_id: int, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Restore an archived spool.""" + result = await db.execute(select(Spool).where(Spool.id == spool_id)) + spool = result.scalar_one_or_none() + if not spool: + raise HTTPException(404, "Spool not found") + + spool.archived_at = None + await db.commit() + result = await db.execute(select(Spool).options(selectinload(Spool.k_profiles)).where(Spool.id == spool_id)) + return result.scalar_one() + + +# ── K-Profiles ─────────────────────────────────────────────────────────────── + + +@router.get("/spools/{spool_id}/k-profiles", response_model=list[SpoolKProfileResponse]) +async def list_k_profiles( + spool_id: int, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """List K-profiles for a spool.""" + result = await db.execute(select(SpoolKProfile).where(SpoolKProfile.spool_id == spool_id)) + return list(result.scalars().all()) + + +@router.put("/spools/{spool_id}/k-profiles", response_model=list[SpoolKProfileResponse]) +async def replace_k_profiles( + spool_id: int, + profiles: list[SpoolKProfileBase], + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Replace all K-profiles for a spool (batch save).""" + # Verify spool exists + result = await db.execute(select(Spool).where(Spool.id == spool_id)) + if not result.scalar_one_or_none(): + raise HTTPException(404, "Spool not found") + + # Delete existing + existing = await db.execute(select(SpoolKProfile).where(SpoolKProfile.spool_id == spool_id)) + for old in existing.scalars().all(): + await db.delete(old) + + # Create new + new_profiles = [] + for p in profiles: + kp = SpoolKProfile(spool_id=spool_id, **p.model_dump()) + db.add(kp) + new_profiles.append(kp) + + await db.commit() + for kp in new_profiles: + await db.refresh(kp) + return new_profiles + + +# ── Spool Assignments ──────────────────────────────────────────────────────── + + +@router.get("/assignments", response_model=list[SpoolAssignmentResponse]) +async def list_assignments( + printer_id: int | None = None, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """List spool assignments, optionally filtered by printer.""" + query = select(SpoolAssignment).options( + selectinload(SpoolAssignment.spool).selectinload(Spool.k_profiles), + selectinload(SpoolAssignment.printer), + ) + if printer_id is not None: + query = query.where(SpoolAssignment.printer_id == printer_id) + result = await db.execute(query) + return list(result.scalars().all()) + + +@router.post("/assignments", response_model=SpoolAssignmentResponse) +async def assign_spool( + data: SpoolAssignmentCreate, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Assign a spool to an AMS slot and auto-configure via MQTT.""" + from backend.app.services.printer_manager import printer_manager + + # 1. Validate spool exists and is not archived + result = await db.execute(select(Spool).options(selectinload(Spool.k_profiles)).where(Spool.id == data.spool_id)) + spool = result.scalar_one_or_none() + if not spool: + raise HTTPException(404, "Spool not found") + if spool.archived_at: + raise HTTPException(400, "Cannot assign an archived spool") + + # 2. Get current AMS tray state for fingerprint + fingerprint_color = None + fingerprint_type = None + state = printer_manager.get_status(data.printer_id) + if state and state.raw_data: + ams_data = state.raw_data.get("ams", {}) + ams_list = ( + ams_data.get("ams", []) if isinstance(ams_data, dict) else ams_data if isinstance(ams_data, list) else [] + ) + tray = _find_tray_in_ams_data( + ams_list, + data.ams_id, + data.tray_id, + ) + if tray: + fingerprint_color = tray.get("tray_color", "") + fingerprint_type = tray.get("tray_type", "") + + # 3. Upsert assignment (replace if same printer+ams+tray) + existing = await db.execute( + select(SpoolAssignment).where( + SpoolAssignment.printer_id == data.printer_id, + SpoolAssignment.ams_id == data.ams_id, + SpoolAssignment.tray_id == data.tray_id, + ) + ) + old = existing.scalar_one_or_none() + if old: + await db.delete(old) + await db.flush() + + assignment = SpoolAssignment( + spool_id=data.spool_id, + printer_id=data.printer_id, + ams_id=data.ams_id, + tray_id=data.tray_id, + fingerprint_color=fingerprint_color, + fingerprint_type=fingerprint_type, + ) + db.add(assignment) + await db.commit() + await db.refresh(assignment) + + # 4. Auto-configure AMS slot via MQTT + configured = False + try: + client = printer_manager.get_client(data.printer_id) + if client: + # Build filament setting from spool data + tray_type = spool.material + tray_sub_brands = f"{spool.material} {spool.subtype}" if spool.subtype else spool.material + tray_color = spool.rgba or "FFFFFFFF" + tray_info_idx = spool.slicer_filament or "" + setting_id = "" + + # Temperature: use spool overrides if set, else material defaults + temp_min, temp_max = MATERIAL_TEMPS.get(spool.material.upper(), (200, 240)) + if spool.nozzle_temp_min is not None: + temp_min = spool.nozzle_temp_min + if spool.nozzle_temp_max is not None: + temp_max = spool.nozzle_temp_max + + # a. Set filament setting + client.ams_set_filament_setting( + ams_id=data.ams_id, + tray_id=data.tray_id, + tray_info_idx=tray_info_idx, + tray_type=tray_type, + tray_sub_brands=tray_sub_brands, + tray_color=tray_color, + nozzle_temp_min=temp_min, + nozzle_temp_max=temp_max, + setting_id=setting_id, + ) + + # b. Look up K-profile for this spool + printer + nozzle + nozzle_diameter = "0.4" + if state and state.nozzles: + nd = state.nozzles[0].nozzle_diameter + if nd: + nozzle_diameter = nd + + matching_kp = None + for kp in spool.k_profiles: + if kp.printer_id == data.printer_id and kp.nozzle_diameter == nozzle_diameter: + matching_kp = kp + break + + if matching_kp and matching_kp.cali_idx is not None: + client.extrusion_cali_sel( + ams_id=data.ams_id, + tray_id=data.tray_id, + cali_idx=matching_kp.cali_idx, + filament_id=tray_info_idx, + nozzle_diameter=nozzle_diameter, + setting_id=matching_kp.setting_id, + ) + + configured = True + logger.info( + "Auto-configured AMS slot ams=%d tray=%d for spool %d on printer %d", + data.ams_id, + data.tray_id, + spool.id, + data.printer_id, + ) + except Exception as e: + logger.warning("MQTT auto-configure failed for spool %d: %s", spool.id, e) + + # Return assignment with spool data + result = await db.execute( + select(SpoolAssignment) + .options( + selectinload(SpoolAssignment.spool).selectinload(Spool.k_profiles), + selectinload(SpoolAssignment.printer), + ) + .where(SpoolAssignment.id == assignment.id) + ) + resp = result.scalar_one() + response = SpoolAssignmentResponse.model_validate(resp) + response.configured = configured + return response + + +@router.delete("/assignments/{printer_id}/{ams_id}/{tray_id}") +async def unassign_spool( + printer_id: int, + ams_id: int, + tray_id: int, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Unassign a spool from an AMS slot.""" + result = await db.execute( + select(SpoolAssignment).where( + SpoolAssignment.printer_id == printer_id, + SpoolAssignment.ams_id == ams_id, + SpoolAssignment.tray_id == tray_id, + ) + ) + assignment = result.scalar_one_or_none() + if not assignment: + raise HTTPException(404, "Assignment not found") + + await db.delete(assignment) + await db.commit() + return {"status": "deleted"} + + +# ── Tag Linking ─────────────────────────────────────────────────────────────── + + +class LinkTagRequest(BaseModel): + tag_uid: str | None = None + tray_uuid: str | None = None + tag_type: str | None = None + data_origin: str | None = "nfc_link" + + +@router.patch("/spools/{spool_id}/link-tag", response_model=SpoolResponse) +async def link_tag_to_spool( + spool_id: int, + data: LinkTagRequest, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Link an RFID tag_uid/tray_uuid to an existing spool.""" + result = await db.execute(select(Spool).options(selectinload(Spool.k_profiles)).where(Spool.id == spool_id)) + spool = result.scalar_one_or_none() + if not spool: + raise HTTPException(404, "Spool not found") + if spool.archived_at: + raise HTTPException(400, "Cannot link tag to archived spool") + + # Check for conflicts: tag already linked to another active spool + if data.tag_uid: + conflict = await db.execute( + select(Spool).where( + Spool.tag_uid == data.tag_uid, + Spool.id != spool_id, + Spool.archived_at.is_(None), + ) + ) + if conflict.scalar_one_or_none(): + raise HTTPException(409, "Tag UID already linked to another active spool") + # Auto-clear from archived spools (tag recycling) + archived_with_tag = await db.execute( + select(Spool).where( + Spool.tag_uid == data.tag_uid, + Spool.id != spool_id, + Spool.archived_at.is_not(None), + ) + ) + for old_spool in archived_with_tag.scalars().all(): + old_spool.tag_uid = None + + if data.tray_uuid: + conflict = await db.execute( + select(Spool).where( + Spool.tray_uuid == data.tray_uuid, + Spool.id != spool_id, + Spool.archived_at.is_(None), + ) + ) + if conflict.scalar_one_or_none(): + raise HTTPException(409, "Tray UUID already linked to another active spool") + archived_with_uuid = await db.execute( + select(Spool).where( + Spool.tray_uuid == data.tray_uuid, + Spool.id != spool_id, + Spool.archived_at.is_not(None), + ) + ) + for old_spool in archived_with_uuid.scalars().all(): + old_spool.tray_uuid = None + + if data.tag_uid is not None: + spool.tag_uid = data.tag_uid + if data.tray_uuid is not None: + spool.tray_uuid = data.tray_uuid + if data.tag_type is not None: + spool.tag_type = data.tag_type + if data.data_origin is not None: + spool.data_origin = data.data_origin + + await db.commit() + result = await db.execute(select(Spool).options(selectinload(Spool.k_profiles)).where(Spool.id == spool_id)) + return result.scalar_one() + + +# ── Usage History ───────────────────────────────────────────────────────────── + + +@router.get("/spools/{spool_id}/usage", response_model=list[SpoolUsageHistoryResponse]) +async def get_spool_usage_history( + spool_id: int, + limit: int = 50, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """Get usage history for a specific spool.""" + from backend.app.models.spool_usage_history import SpoolUsageHistory + + # Verify spool exists + spool_result = await db.execute(select(Spool).where(Spool.id == spool_id)) + if not spool_result.scalar_one_or_none(): + raise HTTPException(404, "Spool not found") + + result = await db.execute( + select(SpoolUsageHistory) + .where(SpoolUsageHistory.spool_id == spool_id) + .order_by(SpoolUsageHistory.created_at.desc()) + .limit(limit) + ) + return list(result.scalars().all()) + + +@router.get("/usage", response_model=list[SpoolUsageHistoryResponse]) +async def get_all_usage_history( + limit: int = 100, + printer_id: int | None = None, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_READ), +): + """Get global usage history, optionally filtered by printer.""" + from backend.app.models.spool_usage_history import SpoolUsageHistory + + query = select(SpoolUsageHistory).order_by(SpoolUsageHistory.created_at.desc()).limit(limit) + if printer_id is not None: + query = query.where(SpoolUsageHistory.printer_id == printer_id) + result = await db.execute(query) + return list(result.scalars().all()) + + +@router.delete("/spools/{spool_id}/usage") +async def clear_spool_usage_history( + spool_id: int, + db: AsyncSession = Depends(get_db), + _: User | None = RequirePermissionIfAuthEnabled(Permission.INVENTORY_UPDATE), +): + """Clear usage history for a spool.""" + from backend.app.models.spool_usage_history import SpoolUsageHistory + + result = await db.execute(select(SpoolUsageHistory).where(SpoolUsageHistory.spool_id == spool_id)) + for row in result.scalars().all(): + await db.delete(row) + await db.commit() + return {"status": "cleared"} + + +# ── Helpers ────────────────────────────────────────────────────────────────── + + +def _find_tray_in_ams_data(ams_data: list, ams_id: int, tray_id: int) -> dict | None: + """Find a specific tray in the AMS data structure.""" + if not ams_data: + return None + for ams_unit in ams_data: + if int(ams_unit.get("id", -1)) != ams_id: + continue + for tray in ams_unit.get("tray", []): + if int(tray.get("id", -1)) == tray_id: + return tray + return None diff --git a/backend/app/api/routes/printers.py b/backend/app/api/routes/printers.py index b54072c96..b2d50bca9 100644 --- a/backend/app/api/routes/printers.py +++ b/backend/app/api/routes/printers.py @@ -1821,6 +1821,37 @@ async def stop_print( return {"success": True, "message": "Print stop command sent"} +@router.post("/{printer_id}/clear-plate") +async def clear_plate( + printer_id: int, + _=RequirePermissionIfAuthEnabled(Permission.PRINTERS_CONTROL), + db: AsyncSession = Depends(get_db), +): + """Acknowledge that the build plate has been cleared after a finished/failed print. + + Sets a plate-cleared flag so the scheduler can start the next queued print. + No MQTT command is sent to the printer — the scheduler's start_print command + will override the FINISH/FAILED state when it sends the next job. + """ + result = await db.execute(select(Printer).where(Printer.id == printer_id)) + printer = result.scalar_one_or_none() + if not printer: + raise HTTPException(404, "Printer not found") + + if not printer_manager.is_connected(printer_id): + raise HTTPException(400, "Printer not connected") + + state = printer_manager.get_status(printer_id) + if not state or state.state not in ("FINISH", "FAILED"): + raise HTTPException( + 400, f"Printer is not in FINISH or FAILED state (current: {state.state if state else 'unknown'})" + ) + + printer_manager.set_plate_cleared(printer_id) + + return {"success": True, "message": "Plate cleared, next print will start shortly"} + + @router.post("/{printer_id}/print/pause") async def pause_print( printer_id: int, diff --git a/backend/app/api/routes/spoolman.py b/backend/app/api/routes/spoolman.py index bf98e0ac1..0a59ea53b 100644 --- a/backend/app/api/routes/spoolman.py +++ b/backend/app/api/routes/spoolman.py @@ -6,12 +6,14 @@ from fastapi import APIRouter, Depends, HTTPException from pydantic import BaseModel from sqlalchemy import select from sqlalchemy.ext.asyncio import AsyncSession +from sqlalchemy.orm import selectinload from backend.app.core.auth import RequirePermissionIfAuthEnabled from backend.app.core.database import get_db from backend.app.core.permissions import Permission from backend.app.models.printer import Printer from backend.app.models.settings import Settings +from backend.app.models.spool_assignment import SpoolAssignment from backend.app.models.user import User from backend.app.services.printer_manager import printer_manager from backend.app.services.spoolman import ( @@ -230,6 +232,22 @@ async def sync_printer_ams( detail=f"Failed to connect to Spoolman after multiple retries: {str(e)}", ) + # Load inventory weights as fallback (when AMS MQTT data lacks remain values) + inv_weights: dict[tuple[int, int], float] = {} + try: + assign_result = await db.execute( + select(SpoolAssignment) + .options(selectinload(SpoolAssignment.spool)) + .where(SpoolAssignment.printer_id == printer_id) + ) + for assignment in assign_result.scalars().all(): + spool = assignment.spool + if spool and spool.label_weight > 0: + remaining = max(0.0, spool.label_weight - (spool.weight_used or 0)) + inv_weights[(assignment.ams_id, assignment.tray_id)] = remaining + except Exception as e: + logger.debug("Could not load inventory weights for printer %s: %s", printer_id, e) + for ams_unit in ams_units: if not isinstance(ams_unit, dict): continue @@ -270,11 +288,13 @@ async def sync_printer_ams( current_tray_uuids.add(spool_tag.upper()) try: + inv_remaining = inv_weights.get((ams_id, tray.tray_id)) sync_result = await client.sync_ams_tray( tray, printer.name, disable_weight_sync=disable_weight_sync, cached_spools=cached_spools, + inventory_remaining=inv_remaining, ) if sync_result: synced += 1 @@ -345,6 +365,8 @@ async def sync_all_printers( all_errors = [] # Track tray UUIDs per printer (for clearing removed spools) printer_tray_uuids: dict[str, set[str]] = {} + # Track synced spool IDs per printer (for location-based cleanup when no UUIDs available) + printer_synced_ids: dict[str, set[int]] = {} # OPTIMIZATION: Fetch all spools once before processing ALL printers/trays # This eliminates redundant API calls across all printers @@ -359,6 +381,19 @@ async def sync_all_printers( detail=f"Failed to connect to Spoolman after multiple retries: {str(e)}", ) + # Load inventory assignments for weight fallback (when AMS MQTT data lacks remain values) + # Key: (printer_id, ams_id, tray_id) → remaining_weight in grams + inventory_weights: dict[tuple[int, int, int], float] = {} + try: + assign_result = await db.execute(select(SpoolAssignment).options(selectinload(SpoolAssignment.spool))) + for assignment in assign_result.scalars().all(): + spool = assignment.spool + if spool and spool.label_weight > 0: + remaining = max(0.0, spool.label_weight - (spool.weight_used or 0)) + inventory_weights[(assignment.printer_id, assignment.ams_id, assignment.tray_id)] = remaining + except Exception as e: + logger.debug("Could not load inventory assignments for weight fallback: %s", e) + for printer in printers: state = printer_manager.get_status(printer.id) if not state or not state.raw_data: @@ -368,8 +403,9 @@ async def sync_all_printers( if not ams_data: continue - # Initialize tray UUID set for this printer + # Initialize tracking sets for this printer printer_tray_uuids[printer.name] = set() + printer_synced_ids[printer.name] = set() # Handle different AMS data structures # Traditional AMS: list of {"id": N, "tray": [...]} dicts @@ -432,16 +468,21 @@ async def sync_all_printers( printer_tray_uuids[printer.name].add(spool_tag.upper()) try: + # Look up inventory weight as fallback when AMS data is invalid + inv_remaining = inventory_weights.get((printer.id, ams_id, tray.tray_id)) sync_result = await client.sync_ams_tray( tray, printer.name, disable_weight_sync=disable_weight_sync, cached_spools=cached_spools, + inventory_remaining=inv_remaining, ) if sync_result: total_synced += 1 - # Add newly created spool to cache + # Track synced spool ID for cleanup if sync_result.get("id"): + printer_synced_ids[printer.name].add(sync_result["id"]) + # Add newly created spool to cache spool_exists = any(s.get("id") == sync_result["id"] for s in cached_spools) if not spool_exists: cached_spools.append(sync_result) @@ -453,7 +494,10 @@ async def sync_all_printers( for printer_name, current_tray_uuids in printer_tray_uuids.items(): try: cleared = await client.clear_location_for_removed_spools( - printer_name, current_tray_uuids, cached_spools=cached_spools + printer_name, + current_tray_uuids, + cached_spools=cached_spools, + synced_spool_ids=printer_synced_ids.get(printer_name, set()), ) if cleared > 0: logger.info("Cleared location for %s spools removed from %s", cleared, printer_name) diff --git a/backend/app/core/bambu_colors.py b/backend/app/core/bambu_colors.py new file mode 100644 index 000000000..61a945a55 --- /dev/null +++ b/backend/app/core/bambu_colors.py @@ -0,0 +1,317 @@ +"""Bambu Lab filament color code to color name mapping. + +Source: https://github.com/queengooborg/Bambu-Lab-RFID-Library + +Maps tray_id_name codes (e.g. "A06-D0") to human-readable color names (e.g. "Titan Gray"). +""" + +# Full color code → name mapping by material prefix +BAMBU_FILAMENT_COLORS: dict[str, str] = { + # PLA Basic (A00) + "A00-W1": "Jade White", + "A00-P0": "Beige", + "A00-D2": "Light Gray", + "A00-Y0": "Yellow", + "A00-Y2": "Sunflower Yellow", + "A00-A1": "Pumpkin Orange", + "A00-A0": "Orange", + "A00-Y4": "Gold", + "A00-G3": "Bright Green", + "A00-G1": "Bambu Green", + "A00-G2": "Mistletoe Green", + "A00-R3": "Hot Pink", + "A00-P6": "Magenta", + "A00-R0": "Red", + "A00-R2": "Maroon Red", + "A00-P5": "Purple", + "A00-P2": "Indigo Purple", + "A00-B5": "Turquoise", + "A00-B8": "Cyan", + "A00-B3": "Cobalt Blue", + "A00-N0": "Brown", + "A00-N1": "Cocoa Brown", + "A00-Y3": "Bronze", + "A00-D0": "Gray", + "A00-D1": "Silver", + "A00-B1": "Blue Grey", + "A00-D3": "Dark Gray", + "A00-K0": "Black", + # PLA Basic Gradient (A00-M*) + "A00-M3": "Pink Citrus", + "A00-M6": "Dusk Glare", + "A00-M0": "Arctic Whisper", + "A00-M1": "Solar Breeze", + "A00-M5": "Blueberry Bubblegum", + "A00-M4": "Mint Lime", + "A00-M2": "Ocean to Meadow", + "A00-M7": "Cotton Candy Cloud", + # PLA Lite (A18) + "A18-K0": "Black", + "A18-D0": "Gray", + "A18-W0": "White", + "A18-R0": "Red", + "A18-Y0": "Yellow", + "A18-B0": "Cyan", + "A18-B1": "Blue", + "A18-P0": "Matte Beige", + # PLA Matte (A01) + "A01-W2": "Ivory White", + "A01-W3": "Bone White", + "A01-Y2": "Lemon Yellow", + "A01-A2": "Mandarin Orange", + "A01-P3": "Sakura Pink", + "A01-P4": "Lilac Purple", + "A01-R3": "Plum", + "A01-R1": "Scarlet Red", + "A01-R4": "Dark Red", + "A01-G0": "Apple Green", + "A01-G1": "Grass Green", + "A01-G7": "Dark Green", + "A01-B4": "Ice Blue", + "A01-B0": "Sky Blue", + "A01-B3": "Marine Blue", + "A01-B6": "Dark Blue", + "A01-Y3": "Desert Tan", + "A01-N1": "Latte Brown", + "A01-N3": "Caramel", + "A01-R2": "Terracotta", + "A01-N2": "Dark Brown", + "A01-N0": "Dark Chocolate", + "A01-D3": "Ash Gray", + "A01-D0": "Nardo Gray", + "A01-K1": "Charcoal", + # PLA Glow (A12) + "A12-G0": "Green", + "A12-R0": "Pink", + "A12-A0": "Orange", + "A12-Y0": "Yellow", + "A12-B0": "Blue", + # PLA Marble (A07) + "A07-R5": "Red Granite", + "A07-D4": "White Marble", + # PLA Aero (A11) + "A11-W0": "White", + "A11-K0": "Black", + # PLA Sparkle (A08) + "A08-G3": "Alpine Green Sparkle", + "A08-D5": "Slate Gray Sparkle", + "A08-B7": "Royal Purple Sparkle", + "A08-R2": "Crimson Red Sparkle", + "A08-K2": "Onyx Black Sparkle", + "A08-Y1": "Classic Gold Sparkle", + # PLA Metal (A02) + "A02-B2": "Cobalt Blue Metallic", + "A02-G2": "Oxide Green Metallic", + "A02-Y1": "Iridium Gold Metallic", + "A02-D2": "Iron Gray Metallic", + # PLA Translucent (A17) + "A17-B1": "Blue", + "A17-A0": "Orange", + "A17-P0": "Purple", + # PLA Silk+ (A06) + "A06-Y1": "Gold", + "A06-D0": "Titan Gray", + "A06-D1": "Silver", + "A06-W0": "White", + "A06-R0": "Candy Red", + "A06-G0": "Candy Green", + "A06-G1": "Mint", + "A06-B1": "Blue", + "A06-B0": "Baby Blue", + "A06-P0": "Purple", + "A06-R1": "Rose Gold", + "A06-R2": "Pink", + "A06-Y0": "Champagne", + # PLA Silk Multi-Color (A05) + "A05-M8": "Dawn Radiance", + "A05-M4": "Aurora Purple", + "A05-M1": "South Beach", + "A05-T3": "Neon City", + "A05-T2": "Midnight Blaze", + "A05-T1": "Gilded Rose", + "A05-T4": "Blue Hawaii", + "A05-T5": "Velvet Eclipse", + # PLA Galaxy (A15) + "A15-B0": "Purple", + "A15-G0": "Green", + "A15-G1": "Nebulae", + "A15-R0": "Brown", + # PLA Wood (A16) + "A16-K0": "Black Walnut", + "A16-R0": "Rosewood", + "A16-N0": "Clay Brown", + "A16-G0": "Classic Birch", + "A16-W0": "White Oak", + "A16-Y0": "Ochre Yellow", + # PLA-CF (A50) + "A50-D6": "Lava Gray", + "A50-K0": "Black", + "A50-B6": "Royal Blue", + # PLA Tough+ (A10) + "A10-W0": "White", + "A10-D0": "Gray", + # PLA Tough (A09) + "A09-B5": "Lavender Blue", + "A09-B4": "Light Blue", + "A09-A0": "Orange", + "A09-D1": "Silver", + "A09-R3": "Vermilion Red", + "A09-Y0": "Yellow", + # PETG HF (G02) + "G02-K0": "Black", + "G02-W0": "White", + "G02-R0": "Red", + "G02-D0": "Gray", + "G02-D1": "Dark Gray", + "G02-Y1": "Cream", + "G02-Y0": "Yellow", + "G02-A0": "Orange", + "G02-N1": "Peanut Brown", + "G02-G1": "Lime Green", + "G02-G0": "Green", + "G02-G2": "Forest Green", + "G02-B1": "Lake Blue", + "G02-B0": "Blue", + # PETG Translucent (G01) + "G01-G1": "Translucent Teal", + "G01-B0": "Translucent Light Blue", + "G01-C0": "Clear", + "G01-D0": "Translucent Gray", + "G01-G0": "Translucent Olive", + "G01-N0": "Translucent Brown", + "G01-A0": "Translucent Orange", + "G01-P1": "Translucent Pink", + "G01-P0": "Translucent Purple", + # PETG-CF (G50) + "G50-P7": "Violet Purple", + "G50-K0": "Black", + # ABS (B00) + "B00-D1": "Silver", + "B00-K0": "Black", + "B00-W0": "White", + "B00-G6": "Bambu Green", + "B00-G7": "Olive", + "B00-Y1": "Tangerine Yellow", + "B00-A0": "Orange", + "B00-R0": "Red", + "B00-B4": "Azure", + "B00-B0": "Blue", + "B00-B6": "Navy Blue", + # ABS-GF (B50) + "B50-A0": "Orange", + "B50-K0": "Black", + # ASA (B01) + "B01-W0": "White", + "B01-K0": "Black", + "B01-D0": "Gray", + # ASA Aero (B02) + "B02-W0": "White", + # PC (C00) + "C00-C1": "Transparent", + "C00-C0": "Clear Black", + "C00-K0": "Black", + "C00-W0": "White", + # PC FR (C01) + "C01-K0": "Black", + # TPU for AMS (U02) + "U02-B0": "Blue", + "U02-D0": "Gray", + "U02-K0": "Black", + # PAHT-CF (N04) + "N04-K0": "Black", + # PA6-GF (N08) + "N08-K0": "Black", + # Support for PLA/PETG (S02, S05) + "S02-W0": "Nature", + "S02-W1": "White", + "S05-C0": "Black", + # Support for ABS (S06) + "S06-W0": "White", + # Support for PA/PET (S03) + "S03-G1": "Green", + # PVA (S04) + "S04-Y0": "Clear", +} + +# Fallback: color code suffix → name (for unknown material prefixes) +BAMBU_COLOR_CODE_FALLBACK: dict[str, str] = { + "W0": "White", + "W1": "Jade White", + "W2": "Ivory White", + "W3": "Bone White", + "Y0": "Yellow", + "Y1": "Gold", + "Y2": "Sunflower Yellow", + "Y3": "Bronze", + "Y4": "Gold", + "A0": "Orange", + "A1": "Pumpkin Orange", + "A2": "Mandarin Orange", + "R0": "Red", + "R1": "Scarlet Red", + "R2": "Maroon Red", + "R3": "Hot Pink", + "R4": "Dark Red", + "R5": "Red Granite", + "P0": "Beige", + "P1": "Pink", + "P2": "Indigo Purple", + "P3": "Sakura Pink", + "P4": "Lilac Purple", + "P5": "Purple", + "P6": "Magenta", + "P7": "Violet Purple", + "B0": "Blue", + "B1": "Blue Grey", + "B2": "Cobalt Blue", + "B3": "Cobalt Blue", + "B4": "Ice Blue", + "B5": "Turquoise", + "B6": "Navy Blue", + "B7": "Royal Purple", + "B8": "Cyan", + "G0": "Green", + "G1": "Grass Green", + "G2": "Mistletoe Green", + "G3": "Bright Green", + "G6": "Bambu Green", + "G7": "Dark Green", + "N0": "Brown", + "N1": "Peanut Brown", + "N2": "Dark Brown", + "N3": "Caramel", + "D0": "Gray", + "D1": "Silver", + "D2": "Light Gray", + "D3": "Dark Gray", + "D4": "White Marble", + "D5": "Slate Gray", + "D6": "Lava Gray", + "K0": "Black", + "K1": "Charcoal", + "K2": "Onyx Black", + "C0": "Clear Black", + "C1": "Transparent", +} + + +def resolve_bambu_color_name(tray_id_name: str) -> str | None: + """Resolve a Bambu Lab tray_id_name code to a human-readable color name. + + Tries exact match first, then falls back to color code suffix lookup. + Returns None if the code cannot be resolved. + """ + if not tray_id_name: + return None + + # Exact match + name = BAMBU_FILAMENT_COLORS.get(tray_id_name) + if name: + return name + + # Fallback: use color code suffix (e.g. "D0" from "A06-D0") + parts = tray_id_name.split("-") + if len(parts) >= 2: + return BAMBU_COLOR_CODE_FALLBACK.get(parts[1]) + + return None diff --git a/backend/app/core/catalog_defaults.py b/backend/app/core/catalog_defaults.py new file mode 100644 index 000000000..18625a60e --- /dev/null +++ b/backend/app/core/catalog_defaults.py @@ -0,0 +1,357 @@ +"""Default spool and color catalog entries.""" + +# (name, weight_in_grams) +DEFAULT_SPOOL_CATALOG: list[tuple[str, int]] = [ + ("3D FilaPrint - Cardboard", 210), + ("3D FilaPrint - Plastic", 238), + ("3D Fuel - Plastic", 264), + ("3D Power - Plastic", 220), + ("3D Solutech - Plastic", 173), + ("3DE - Cardboard", 136), + ("3DE - Plastic", 181), + ("3DHOJOR - Cardboard", 157), + ("3DJake - Cardboard", 209), + ("3DJake - Plastic", 232), + ("3DJake 250g - Plastic", 91), + ("3DJake ecoPLA - Plastic", 210), + ("3DXTech - Plastic", 258), + ("Acccreate - Plastic", 161), + ("Amazon Basics - Plastic", 234), + ("Amolen - Plastic", 150), + ("AMZ3D - Plastic", 233), + ("Anycubic - Cardboard", 125), + ("Anycubic - Plastic", 127), + ("Atomic Filament - Plastic", 272), + ("Aurapol - Plastic", 220), + ("Azure Film - Plastic", 163), + ("Bambu Lab - Plastic High Temp", 216), + ("Bambu Lab - Plastic Low Temp", 250), + ("Bambu Lab - Plastic White", 253), + ("BQ - Plastic", 218), + ("Colorfabb - Plastic", 236), + ("Colorfabb 750g - Cardboard", 152), + ("Colorfabb 750g - Plastic", 254), + ("Comgrow - Cardboard", 166), + ("Creality - Cardboard", 180), + ("Creality - Plastic", 135), + ("Das Filament - Plastic", 211), + ("Devil Design - Plastic", 256), + ("Duramic 3D - Cardboard", 136), + ("Elegoo - Cardboard", 153), + ("Elegoo - Plastic", 111), + ("Eryone - Cardboard", 156), + ("Eryone - Plastic", 187), + ("eSUN - Cardboard", 147), + ("eSUN - Plastic", 240), + ("eSUN 2.5kg - Plastic", 634), + ("Extrudr - Plastic", 244), + ("Fiberlogy - Plastic", 260), + ("Filament PM - Plastic", 224), + ("Fillamentum - Plastic", 230), + ("Flashforge - Plastic", 167), + ("FormFutura - Cardboard", 155), + ("FormFutura 750g - Plastic", 212), + ("Geeetech - Plastic", 178), + ("Gembird - Cardboard", 143), + ("Hatchbox - Plastic", 225), + ("Inland - Cardboard", 142), + ("Inland - Plastic", 210), + ("Jayo - Cardboard", 120), + ("Jayo - Plastic", 126), + ("Jayo 250g - Plastic", 58), + ("Kingroon - Cardboard", 155), + ("Kingroon - Plastic", 156), + ("KVP - Plastic", 263), + ("Matter Hackers - Plastic", 215), + ("MG Chemicals - Cardboard", 150), + ("MG Chemicals - Plastic", 239), + ("Mika3D - Plastic", 175), + ("MonoPrice - Plastic", 221), + ("Overture - Cardboard", 150), + ("Overture - Plastic", 237), + ("PolyMaker - Cardboard", 137), + ("PolyMaker - Plastic", 220), + ("PolyMaker 3kg - Cardboard", 418), + ("PolyTerra PLA - Cardboard", 147), + ("PrimaSelect - Plastic", 222), + ("ProtoPasta - Cardboard", 80), + ("Prusament - Plastic", 201), + ("Prusament - Plastic w/ Cardboard Core", 196), + ("Rosa3D - Plastic", 245), + ("Sakata3D - Plastic", 205), + ("Snapmaker - Cardboard", 148), + ("Sovol - Cardboard", 145), + ("Spectrum - Cardboard", 180), + ("Spectrum - Plastic", 257), + ("Sunlu - Plastic", 117), + ("Sunlu - Plastic V2", 165), + ("Sunlu - Plastic V3", 179), + ("Sunlu 250g - Plastic", 55), + ("UltiMaker - Plastic", 235), + ("Voolt3D - Plastic", 190), + ("Voxelab - Plastic", 171), + ("Wanhao - Plastic", 267), + ("Ziro - Plastic", 166), + ("ZYLtech - Plastic", 179), +] + +# (manufacturer, color_name, hex_color, material) +DEFAULT_COLOR_CATALOG: list[tuple[str, str, str, str]] = [ + # Bambu Lab PLA Basic (from official hex code PDF) + ("Bambu Lab", "Jade White", "#FFFFFF", "PLA Basic"), + ("Bambu Lab", "Black", "#000000", "PLA Basic"), + ("Bambu Lab", "Silver", "#A6A9AA", "PLA Basic"), + ("Bambu Lab", "Light Gray", "#D1D3D5", "PLA Basic"), + ("Bambu Lab", "Gray", "#8E9089", "PLA Basic"), + ("Bambu Lab", "Dark Gray", "#545454", "PLA Basic"), + ("Bambu Lab", "Red", "#C12E1F", "PLA Basic"), + ("Bambu Lab", "Maroon Red", "#9D2235", "PLA Basic"), + ("Bambu Lab", "Magenta", "#EC008C", "PLA Basic"), + ("Bambu Lab", "Hot Pink", "#F5547C", "PLA Basic"), + ("Bambu Lab", "Pink", "#F55A74", "PLA Basic"), + ("Bambu Lab", "Beige", "#F7E6DE", "PLA Basic"), + ("Bambu Lab", "Yellow", "#F4EE2A", "PLA Basic"), + ("Bambu Lab", "Sunflower Yellow", "#FEC600", "PLA Basic"), + ("Bambu Lab", "Gold", "#E4BD68", "PLA Basic"), + ("Bambu Lab", "Orange", "#FF6A13", "PLA Basic"), + ("Bambu Lab", "Pumpkin Orange", "#FF9016", "PLA Basic"), + ("Bambu Lab", "Bright Green", "#BECF00", "PLA Basic"), + ("Bambu Lab", "Bambu Green", "#00AE42", "PLA Basic"), + ("Bambu Lab", "Mistletoe Green", "#3F8E43", "PLA Basic"), + ("Bambu Lab", "Turquoise", "#00B1B7", "PLA Basic"), + ("Bambu Lab", "Cyan", "#0086D6", "PLA Basic"), + ("Bambu Lab", "Blue", "#0A2989", "PLA Basic"), + ("Bambu Lab", "Blue Grey", "#5B6579", "PLA Basic"), + ("Bambu Lab", "Cobalt Blue", "#0056B8", "PLA Basic"), + ("Bambu Lab", "Purple", "#5E43B7", "PLA Basic"), + ("Bambu Lab", "Indigo Purple", "#482960", "PLA Basic"), + ("Bambu Lab", "Brown", "#9D432C", "PLA Basic"), + ("Bambu Lab", "Cocoa Brown", "#6F5034", "PLA Basic"), + ("Bambu Lab", "Bronze", "#847D48", "PLA Basic"), + # Bambu Lab PLA Matte (from official hex code PDF) + ("Bambu Lab", "Ivory White", "#FFFFFF", "PLA Matte"), + ("Bambu Lab", "Bone White", "#CBC6B8", "PLA Matte"), + ("Bambu Lab", "Desert Tan", "#E8DBB7", "PLA Matte"), + ("Bambu Lab", "Latte Brown", "#D3B7A7", "PLA Matte"), + ("Bambu Lab", "Caramel", "#AE835B", "PLA Matte"), + ("Bambu Lab", "Terracotta", "#B15533", "PLA Matte"), + ("Bambu Lab", "Dark Brown", "#7D6556", "PLA Matte"), + ("Bambu Lab", "Dark Chocolate", "#4D3324", "PLA Matte"), + ("Bambu Lab", "Lilac Purple", "#AE96D4", "PLA Matte"), + ("Bambu Lab", "Sakura Pink", "#E8AFCF", "PLA Matte"), + ("Bambu Lab", "Mandarin Orange", "#F99963", "PLA Matte"), + ("Bambu Lab", "Lemon Yellow", "#F7D959", "PLA Matte"), + ("Bambu Lab", "Plum", "#950051", "PLA Matte"), + ("Bambu Lab", "Scarlet Red", "#DE4343", "PLA Matte"), + ("Bambu Lab", "Dark Red", "#BB3D43", "PLA Matte"), + ("Bambu Lab", "Dark Green", "#68724D", "PLA Matte"), + ("Bambu Lab", "Grass Green", "#61C680", "PLA Matte"), + ("Bambu Lab", "Apple Green", "#C2E189", "PLA Matte"), + ("Bambu Lab", "Ice Blue", "#A3D8E1", "PLA Matte"), + ("Bambu Lab", "Sky Blue", "#56B7E6", "PLA Matte"), + ("Bambu Lab", "Marine Blue", "#0078BF", "PLA Matte"), + ("Bambu Lab", "Dark Blue", "#042F56", "PLA Matte"), + ("Bambu Lab", "Ash Gray", "#9B9EA0", "PLA Matte"), + ("Bambu Lab", "Nardo Gray", "#757575", "PLA Matte"), + ("Bambu Lab", "Charcoal", "#000000", "PLA Matte"), + # Bambu Lab PLA Silk+ (from store page) + ("Bambu Lab", "Gold", "#F4A925", "PLA Silk"), + ("Bambu Lab", "Silver", "#C8C8C8", "PLA Silk"), + ("Bambu Lab", "Titan Gray", "#5F6367", "PLA Silk"), + ("Bambu Lab", "Blue", "#008BDA", "PLA Silk"), + ("Bambu Lab", "Purple", "#8671CB", "PLA Silk"), + ("Bambu Lab", "Candy Red", "#D02727", "PLA Silk"), + ("Bambu Lab", "Candy Green", "#018814", "PLA Silk"), + ("Bambu Lab", "Rose Gold", "#BA9594", "PLA Silk"), + ("Bambu Lab", "Baby Blue", "#A8C6EE", "PLA Silk"), + ("Bambu Lab", "Pink", "#F7ADA6", "PLA Silk"), + ("Bambu Lab", "Mint", "#96DCB9", "PLA Silk"), + ("Bambu Lab", "Champagne", "#F3CFB2", "PLA Silk"), + ("Bambu Lab", "White", "#FFFFFF", "PLA Silk"), + # Bambu Lab PLA Sparkle (from store page) + ("Bambu Lab", "Classic Gold Sparkle", "#CEA629", "PLA Sparkle"), + ("Bambu Lab", "Slate Gray Sparkle", "#8E9089", "PLA Sparkle"), + ("Bambu Lab", "Crimson Red Sparkle", "#792B36", "PLA Sparkle"), + ("Bambu Lab", "Royal Purple Sparkle", "#483D8B", "PLA Sparkle"), + ("Bambu Lab", "Alpine Green Sparkle", "#3F5443", "PLA Sparkle"), + ("Bambu Lab", "Onyx Black Sparkle", "#2D2B28", "PLA Sparkle"), + # Bambu Lab PLA Translucent (from official hex code PDF) + ("Bambu Lab", "Teal", "#009FA1", "PLA Translucent"), + ("Bambu Lab", "Light Jade", "#96D8AF", "PLA Translucent"), + ("Bambu Lab", "Blue", "#0047BB", "PLA Translucent"), + ("Bambu Lab", "Mellow Yellow", "#F5DBAB", "PLA Translucent"), + ("Bambu Lab", "Purple", "#8344B0", "PLA Translucent"), + ("Bambu Lab", "Cherry Pink", "#F5B6CD", "PLA Translucent"), + ("Bambu Lab", "Orange", "#F74E02", "PLA Translucent"), + ("Bambu Lab", "Ice Blue", "#B8CDE9", "PLA Translucent"), + ("Bambu Lab", "Red", "#B50011", "PLA Translucent"), + ("Bambu Lab", "Lavender", "#B8ACD6", "PLA Translucent"), + # Bambu Lab PLA Glow (from store page) + ("Bambu Lab", "Glow Green", "#A1FFAC", "PLA Glow"), + ("Bambu Lab", "Glow Yellow", "#F8FF80", "PLA Glow"), + ("Bambu Lab", "Glow Pink", "#F17B8F", "PLA Glow"), + ("Bambu Lab", "Glow Blue", "#7AC0E9", "PLA Glow"), + ("Bambu Lab", "Glow Orange", "#FF9D5B", "PLA Glow"), + # Bambu Lab PLA Galaxy (from store page) + ("Bambu Lab", "Brown", "#684A43", "PLA Galaxy"), + ("Bambu Lab", "Green", "#3B665E", "PLA Galaxy"), + ("Bambu Lab", "Nebulae", "#424379", "PLA Galaxy"), + ("Bambu Lab", "Purple", "#594177", "PLA Galaxy"), + # Bambu Lab PLA Metal (from store page) + ("Bambu Lab", "Iridium Gold Metallic", "#B39B84", "PLA Metal"), + ("Bambu Lab", "Copper Brown Metallic", "#AA6443", "PLA Metal"), + ("Bambu Lab", "Oxide Green Metallic", "#1D7C6A", "PLA Metal"), + ("Bambu Lab", "Cobalt Blue Metallic", "#39699E", "PLA Metal"), + ("Bambu Lab", "Iron Gray Metallic", "#43403D", "PLA Metal"), + # Bambu Lab PLA Marble (from store page) + ("Bambu Lab", "White Marble", "#F7F3F0", "PLA Marble"), + ("Bambu Lab", "Red Granite", "#AD4E38", "PLA Marble"), + # Bambu Lab PLA Wood (from store page) + ("Bambu Lab", "Black Walnut", "#4F3F24", "PLA Wood"), + ("Bambu Lab", "Rosewood", "#4C241C", "PLA Wood"), + ("Bambu Lab", "Clay Brown", "#995F11", "PLA Wood"), + ("Bambu Lab", "Classic Birch", "#918669", "PLA Wood"), + ("Bambu Lab", "White Oak", "#D6CCA3", "PLA Wood"), + ("Bambu Lab", "Ochre Yellow", "#C98935", "PLA Wood"), + # Bambu Lab PLA Tough+ (from official hex code PDF) + ("Bambu Lab", "White", "#FFFFFF", "PLA Tough"), + ("Bambu Lab", "Gray", "#AFB1AE", "PLA Tough"), + ("Bambu Lab", "Black", "#000000", "PLA Tough"), + ("Bambu Lab", "Silver", "#959698", "PLA Tough"), + ("Bambu Lab", "Yellow", "#F4D53F", "PLA Tough"), + ("Bambu Lab", "Cyan", "#009BD8", "PLA Tough"), + ("Bambu Lab", "Orange", "#DC3A27", "PLA Tough"), + # Bambu Lab PLA-CF (from official hex code PDF) + ("Bambu Lab", "Burgundy Red", "#951E23", "PLA-CF"), + ("Bambu Lab", "Iris Purple", "#69398E", "PLA-CF"), + ("Bambu Lab", "Matcha Green", "#5C9748", "PLA-CF"), + ("Bambu Lab", "Jeans Blue", "#6E88BC", "PLA-CF"), + ("Bambu Lab", "Royal Blue", "#2842AD", "PLA-CF"), + ("Bambu Lab", "Lava Gray", "#4D5054", "PLA-CF"), + ("Bambu Lab", "Black", "#000000", "PLA-CF"), + # Bambu Lab ABS (from official hex code PDF) + ("Bambu Lab", "White", "#FFFFFF", "ABS"), + ("Bambu Lab", "Desert Tan", "#E8DBB7", "ABS"), + ("Bambu Lab", "Olive", "#789D4A", "ABS"), + ("Bambu Lab", "Azure", "#489FDF", "ABS"), + ("Bambu Lab", "Navy Blue", "#0C2340", "ABS"), + ("Bambu Lab", "Blue", "#0A2CA5", "ABS"), + ("Bambu Lab", "Tangerine Yellow", "#FFC72C", "ABS"), + ("Bambu Lab", "Orange", "#FF6A13", "ABS"), + ("Bambu Lab", "Red", "#D32941", "ABS"), + ("Bambu Lab", "Purple", "#AF1685", "ABS"), + ("Bambu Lab", "Silver", "#87909A", "ABS"), + ("Bambu Lab", "Black", "#000000", "ABS"), + # Bambu Lab ASA (from store page) + ("Bambu Lab", "White", "#FFFAF2", "ASA"), + ("Bambu Lab", "Gray", "#8A949E", "ASA"), + ("Bambu Lab", "Red", "#E02928", "ASA"), + ("Bambu Lab", "Green", "#00A6A0", "ASA"), + ("Bambu Lab", "Blue", "#2140B4", "ASA"), + ("Bambu Lab", "Black", "#000000", "ASA"), + # Bambu Lab PETG HF (from store page) + ("Bambu Lab", "Yellow", "#FFD00B", "PETG HF"), + ("Bambu Lab", "Orange", "#F75403", "PETG HF"), + ("Bambu Lab", "Green", "#00AE42", "PETG HF"), + ("Bambu Lab", "Red", "#EB3A3A", "PETG HF"), + ("Bambu Lab", "Blue", "#002E96", "PETG HF"), + ("Bambu Lab", "Black", "#000000", "PETG HF"), + ("Bambu Lab", "White", "#FFFFFF", "PETG HF"), + ("Bambu Lab", "Cream", "#F9DFB9", "PETG HF"), + ("Bambu Lab", "Lime Green", "#6EE53C", "PETG HF"), + ("Bambu Lab", "Forest Green", "#39541A", "PETG HF"), + ("Bambu Lab", "Lake Blue", "#1F79E5", "PETG HF"), + ("Bambu Lab", "Peanut Brown", "#875718", "PETG HF"), + ("Bambu Lab", "Gray", "#ADB1B2", "PETG HF"), + ("Bambu Lab", "Dark Gray", "#515151", "PETG HF"), + # Bambu Lab PETG Translucent (from store page) + ("Bambu Lab", "Translucent Gray", "#8E8E8E", "PETG Translucent"), + ("Bambu Lab", "Translucent Light Blue", "#61B0FF", "PETG Translucent"), + ("Bambu Lab", "Translucent Olive", "#748C45", "PETG Translucent"), + ("Bambu Lab", "Translucent Brown", "#C9A381", "PETG Translucent"), + ("Bambu Lab", "Translucent Teal", "#77EDD7", "PETG Translucent"), + ("Bambu Lab", "Translucent Orange", "#FF911A", "PETG Translucent"), + ("Bambu Lab", "Translucent Purple", "#D6ABFF", "PETG Translucent"), + ("Bambu Lab", "Translucent Pink", "#F9C1BD", "PETG Translucent"), + # Bambu Lab PETG-CF (from official hex code PDF) + ("Bambu Lab", "Brick Red", "#9F332A", "PETG-CF"), + ("Bambu Lab", "Violet Purple", "#583061", "PETG-CF"), + ("Bambu Lab", "Indigo Blue", "#324585", "PETG-CF"), + ("Bambu Lab", "Malachite Green", "#16B08E", "PETG-CF"), + ("Bambu Lab", "Black", "#000000", "PETG-CF"), + ("Bambu Lab", "Titan Gray", "#565656", "PETG-CF"), + # Bambu Lab TPU 95A HF (from store page) + ("Bambu Lab", "White", "#FFFFFF", "TPU 95A"), + ("Bambu Lab", "Yellow", "#F3E600", "TPU 95A"), + ("Bambu Lab", "Blue", "#0072CE", "TPU 95A"), + ("Bambu Lab", "Red", "#C8102E", "TPU 95A"), + ("Bambu Lab", "Gray", "#898D8D", "TPU 95A"), + ("Bambu Lab", "Black", "#101820", "TPU 95A"), + # Bambu Lab TPU 90A (from official hex code PDF) + ("Bambu Lab", "Black", "#000000", "TPU 90A"), + ("Bambu Lab", "White", "#FFFFFF", "TPU 90A"), + ("Bambu Lab", "Grape Jelly", "#D6ABFF", "TPU 90A"), + ("Bambu Lab", "Crystal Blue", "#7EB4E1", "TPU 90A"), + ("Bambu Lab", "Cocoa Brown", "#5C4738", "TPU 90A"), + # Bambu Lab PAHT-CF + ("Bambu Lab", "Black", "#1A1A1A", "PAHT-CF"), + # Bambu Lab Support Materials + ("Bambu Lab", "Natural", "#F5F5DC", "PLA Support"), + ("Bambu Lab", "Natural", "#F5F5DC", "PVA Support"), + # Polymaker PolyTerra PLA + ("Polymaker", "Cotton White", "#F5F5F5", "PolyTerra PLA"), + ("Polymaker", "Charcoal Black", "#2B2B2B", "PolyTerra PLA"), + ("Polymaker", "Marble White", "#E8E8E8", "PolyTerra PLA"), + ("Polymaker", "Fossil Grey", "#6B6B6B", "PolyTerra PLA"), + ("Polymaker", "Shadow Black", "#1A1A1A", "PolyTerra PLA"), + ("Polymaker", "Army Red", "#8B0000", "PolyTerra PLA"), + ("Polymaker", "Lava Red", "#CF1020", "PolyTerra PLA"), + ("Polymaker", "Sakura Pink", "#FFB7C5", "PolyTerra PLA"), + ("Polymaker", "Rose", "#FF007F", "PolyTerra PLA"), + ("Polymaker", "Peach", "#FFCBA4", "PolyTerra PLA"), + ("Polymaker", "Banana", "#FFE135", "PolyTerra PLA"), + ("Polymaker", "Savannah Yellow", "#F4C430", "PolyTerra PLA"), + ("Polymaker", "Sunrise Orange", "#FF6600", "PolyTerra PLA"), + ("Polymaker", "Muted Green", "#4F7942", "PolyTerra PLA"), + ("Polymaker", "Forest Green", "#228B22", "PolyTerra PLA"), + ("Polymaker", "Mint", "#98FF98", "PolyTerra PLA"), + ("Polymaker", "Lavender Purple", "#B57EDC", "PolyTerra PLA"), + ("Polymaker", "Sapphire Blue", "#0F52BA", "PolyTerra PLA"), + ("Polymaker", "Ice", "#D6ECEF", "PolyTerra PLA"), + # Prusament PLA + ("Prusament", "Jet Black", "#1A1A1A", "PLA"), + ("Prusament", "Galaxy Black", "#1F1F1F", "PLA"), + ("Prusament", "Pristine White", "#FFFFFF", "PLA"), + ("Prusament", "Gentleman's Grey", "#5A5A5A", "PLA"), + ("Prusament", "Lipstick Red", "#C21E1E", "PLA"), + ("Prusament", "Orange", "#FF6600", "PLA"), + ("Prusament", "Pineapple Yellow", "#FFD700", "PLA"), + ("Prusament", "Jungle Green", "#29AB87", "PLA"), + ("Prusament", "Azure Blue", "#007FFF", "PLA"), + ("Prusament", "Royal Blue", "#4169E1", "PLA"), + ("Prusament", "Mystic Purple", "#7B68EE", "PLA"), + # eSUN PLA+ + ("eSUN", "White", "#FFFFFF", "PLA+"), + ("eSUN", "Black", "#000000", "PLA+"), + ("eSUN", "Grey", "#808080", "PLA+"), + ("eSUN", "Red", "#FF0000", "PLA+"), + ("eSUN", "Blue", "#0000FF", "PLA+"), + ("eSUN", "Green", "#00FF00", "PLA+"), + ("eSUN", "Yellow", "#FFFF00", "PLA+"), + ("eSUN", "Orange", "#FFA500", "PLA+"), + ("eSUN", "Purple", "#800080", "PLA+"), + ("eSUN", "Pink", "#FFC0CB", "PLA+"), + # Hatchbox PLA + ("Hatchbox", "White", "#FFFFFF", "PLA"), + ("Hatchbox", "Black", "#000000", "PLA"), + ("Hatchbox", "Gray", "#808080", "PLA"), + ("Hatchbox", "Red", "#FF0000", "PLA"), + ("Hatchbox", "Blue", "#0000FF", "PLA"), + ("Hatchbox", "Green", "#00FF00", "PLA"), + ("Hatchbox", "Yellow", "#FFFF00", "PLA"), + ("Hatchbox", "Orange", "#FFA500", "PLA"), + ("Hatchbox", "Purple", "#800080", "PLA"), + ("Hatchbox", "Pink", "#FFC0CB", "PLA"), + ("Hatchbox", "True Blue", "#0073CF", "PLA"), + ("Hatchbox", "True Green", "#008000", "PLA"), +] diff --git a/backend/app/core/config.py b/backend/app/core/config.py index b4f012b99..9a792c20d 100644 --- a/backend/app/core/config.py +++ b/backend/app/core/config.py @@ -5,7 +5,7 @@ from pathlib import Path from pydantic_settings import BaseSettings # Application version - single source of truth -APP_VERSION = "0.1.9" +APP_VERSION = "0.2.0b" GITHUB_REPO = "maziggy/bambuddy" # App directory - where the application is installed (for static files) diff --git a/backend/app/core/database.py b/backend/app/core/database.py index c3808010e..065796e3d 100644 --- a/backend/app/core/database.py +++ b/backend/app/core/database.py @@ -59,6 +59,7 @@ async def init_db(): ams_history, api_key, archive, + color_catalog, external_link, filament, github_backup, @@ -78,6 +79,11 @@ async def init_db(): settings, slot_preset, smart_plug, + spool, + spool_assignment, + spool_catalog, + spool_k_profile, + spool_usage_history, user, ) @@ -93,6 +99,10 @@ async def init_db(): # Seed default groups and migrate existing users await seed_default_groups() + # Seed default catalog entries + await seed_spool_catalog() + await seed_color_catalog() + async def run_migrations(conn): """Add new columns to existing tables if they don't exist.""" @@ -1118,6 +1128,63 @@ async def run_migrations(conn): except OperationalError: pass # Already applied + # Migration: Add inventory spool tracking columns + try: + await conn.execute(text("ALTER TABLE spool ADD COLUMN added_full BOOLEAN")) + except OperationalError: + pass # Already applied + try: + await conn.execute(text("ALTER TABLE spool ADD COLUMN last_used DATETIME")) + except OperationalError: + pass # Already applied + try: + await conn.execute(text("ALTER TABLE spool ADD COLUMN encode_time DATETIME")) + except OperationalError: + pass # Already applied + + # Migration: Add RFID tag matching columns to spool + try: + await conn.execute(text("ALTER TABLE spool ADD COLUMN tag_uid VARCHAR(16)")) + except OperationalError: + pass # Already applied + try: + await conn.execute(text("ALTER TABLE spool ADD COLUMN tray_uuid VARCHAR(32)")) + except OperationalError: + pass # Already applied + try: + await conn.execute(text("ALTER TABLE spool ADD COLUMN data_origin VARCHAR(20)")) + except OperationalError: + pass # Already applied + try: + await conn.execute(text("ALTER TABLE spool ADD COLUMN tag_type VARCHAR(20)")) + except OperationalError: + pass # Already applied + + # Migration: Create spool_usage_history table for filament consumption tracking + try: + await conn.execute( + text(""" + CREATE TABLE IF NOT EXISTS spool_usage_history ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + spool_id INTEGER NOT NULL REFERENCES spool(id) ON DELETE CASCADE, + printer_id INTEGER REFERENCES printers(id) ON DELETE SET NULL, + print_name VARCHAR(500), + weight_used REAL NOT NULL DEFAULT 0, + percent_used INTEGER NOT NULL DEFAULT 0, + status VARCHAR(20) NOT NULL DEFAULT 'completed', + created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP + ) + """) + ) + except OperationalError: + pass # Already applied + + # Migration: Add open_in_new_tab column to external_links + try: + await conn.execute(text("ALTER TABLE external_links ADD COLUMN open_in_new_tab BOOLEAN DEFAULT 0")) + except OperationalError: + pass # Already applied + async def seed_notification_templates(): """Seed default notification templates if they don't exist.""" @@ -1286,3 +1353,57 @@ async def seed_default_groups(): logger.info("Migrated user '%s' to Operators group", user.username) await session.commit() + + +async def seed_spool_catalog(): + """Seed the spool catalog with default entries if empty.""" + import logging + + from sqlalchemy import func, select + + from backend.app.core.catalog_defaults import DEFAULT_SPOOL_CATALOG + from backend.app.models.spool_catalog import SpoolCatalogEntry + + logger = logging.getLogger(__name__) + + async with async_session() as session: + result = await session.execute(select(func.count()).select_from(SpoolCatalogEntry)) + count = result.scalar() or 0 + if count > 0: + return # Already seeded + + for name, weight in DEFAULT_SPOOL_CATALOG: + session.add(SpoolCatalogEntry(name=name, weight=weight, is_default=True)) + await session.commit() + logger.info("Seeded %d default spool catalog entries", len(DEFAULT_SPOOL_CATALOG)) + + +async def seed_color_catalog(): + """Seed the color catalog with default entries if empty.""" + import logging + + from sqlalchemy import func, select + + from backend.app.core.catalog_defaults import DEFAULT_COLOR_CATALOG + from backend.app.models.color_catalog import ColorCatalogEntry + + logger = logging.getLogger(__name__) + + async with async_session() as session: + result = await session.execute(select(func.count()).select_from(ColorCatalogEntry)) + count = result.scalar() or 0 + if count > 0: + return # Already seeded + + for manufacturer, color_name, hex_color, material in DEFAULT_COLOR_CATALOG: + session.add( + ColorCatalogEntry( + manufacturer=manufacturer, + color_name=color_name, + hex_color=hex_color, + material=material, + is_default=True, + ) + ) + await session.commit() + logger.info("Seeded %d default color catalog entries", len(DEFAULT_COLOR_CATALOG)) diff --git a/backend/app/core/permissions.py b/backend/app/core/permissions.py index a6c56d409..ff2ac4fd8 100644 --- a/backend/app/core/permissions.py +++ b/backend/app/core/permissions.py @@ -62,6 +62,12 @@ class Permission(StrEnum): FILAMENTS_UPDATE = "filaments:update" FILAMENTS_DELETE = "filaments:delete" + # Inventory (Spool Inventory, Spool Catalog, Color Catalog) + INVENTORY_READ = "inventory:read" + INVENTORY_CREATE = "inventory:create" + INVENTORY_UPDATE = "inventory:update" + INVENTORY_DELETE = "inventory:delete" + # Smart Plugs SMART_PLUGS_READ = "smart_plugs:read" SMART_PLUGS_CREATE = "smart_plugs:create" @@ -201,6 +207,12 @@ PERMISSION_CATEGORIES = { Permission.FILAMENTS_UPDATE, Permission.FILAMENTS_DELETE, ], + "Inventory": [ + Permission.INVENTORY_READ, + Permission.INVENTORY_CREATE, + Permission.INVENTORY_UPDATE, + Permission.INVENTORY_DELETE, + ], "Smart Plugs": [ Permission.SMART_PLUGS_READ, Permission.SMART_PLUGS_CREATE, @@ -335,6 +347,11 @@ DEFAULT_GROUPS = { Permission.FILAMENTS_CREATE.value, Permission.FILAMENTS_UPDATE.value, Permission.FILAMENTS_DELETE.value, + # Inventory - full access + Permission.INVENTORY_READ.value, + Permission.INVENTORY_CREATE.value, + Permission.INVENTORY_UPDATE.value, + Permission.INVENTORY_DELETE.value, # Smart Plugs - full access Permission.SMART_PLUGS_READ.value, Permission.SMART_PLUGS_CREATE.value, @@ -390,6 +407,7 @@ DEFAULT_GROUPS = { Permission.LIBRARY_READ.value, Permission.PROJECTS_READ.value, Permission.FILAMENTS_READ.value, + Permission.INVENTORY_READ.value, Permission.SMART_PLUGS_READ.value, Permission.CAMERA_VIEW.value, Permission.MAINTENANCE_READ.value, diff --git a/backend/app/main.py b/backend/app/main.py index b9f6e88a2..d55686522 100644 --- a/backend/app/main.py +++ b/backend/app/main.py @@ -184,6 +184,7 @@ from backend.app.api.routes import ( firmware, github_backup, groups, + inventory, kprofiles, library, local_presets, @@ -216,6 +217,7 @@ from backend.app.services.bambu_mqtt import PrinterState from backend.app.services.github_backup import github_backup_service from backend.app.services.homeassistant import homeassistant_service from backend.app.services.mqtt_relay import mqtt_relay +from backend.app.services.mqtt_smart_plug import mqtt_smart_plug_service from backend.app.services.notification_service import notification_service from backend.app.services.print_scheduler import scheduler as print_scheduler from backend.app.services.printer_manager import ( @@ -253,6 +255,10 @@ _last_progress_milestone: dict[int, int] = {} # This prevents sending duplicate notifications for the same error _notified_hms_errors: dict[int, set[str]] = {} +# Track timelapse file baselines at print start: {printer_id: set of MP4 filenames} +# Used for snapshot-diff detection at print completion +_timelapse_baselines: dict[int, set[str]] = {} + async def _get_plug_energy(plug, db) -> dict | None: """Get energy from plug regardless of type (Tasmota, Home Assistant, or MQTT). @@ -410,6 +416,9 @@ async def on_printer_status_change(printer_id: int, state: PrinterState): # Find new errors that haven't been notified yet new_error_codes = current_error_codes - previously_notified + # Update tracking immediately to prevent duplicate notifications from concurrent callbacks + _notified_hms_errors[printer_id] = current_error_codes + if new_error_codes: # Get the actual new errors for the notification # Filter to severity >= 2 (skip informational/status messages like H2D sends) @@ -480,8 +489,6 @@ async def on_printer_status_change(printer_id: int, state: PrinterState): except Exception as e: logging.getLogger(__name__).warning(f"HMS error notification failed: {e}") - # Update tracking with all current errors - _notified_hms_errors[printer_id] = current_error_codes else: # No HMS errors - clear tracking so future errors get notified if printer_id in _notified_hms_errors: @@ -493,6 +500,11 @@ async def on_printer_status_change(printer_id: int, state: PrinterState): ) +def _is_bambu_uuid(tray_uuid: str) -> bool: + """Check if a tray UUID looks like a valid Bambu Lab RFID UUID (non-empty, non-zero).""" + return bool(tray_uuid) and tray_uuid not in ("", "0" * len(tray_uuid)) + + async def on_ams_change(printer_id: int, ams_data: list): """Handle AMS data changes - sync to Spoolman if enabled and auto mode.""" logger = logging.getLogger(__name__) @@ -518,6 +530,196 @@ async def on_ams_change(printer_id: int, ams_data: list): except Exception as e: logger.warning("Failed to broadcast AMS change for printer %s: %s", printer_id, e) + # Auto-unlink spool assignments with stale fingerprints + try: + async with async_session() as db: + from sqlalchemy.orm import selectinload + + from backend.app.api.routes.inventory import _find_tray_in_ams_data + from backend.app.models.spool_assignment import SpoolAssignment as SA + + result = await db.execute(select(SA).where(SA.printer_id == printer_id).options(selectinload(SA.spool))) + stale = [] + for assignment in result.scalars().all(): + current_tray = _find_tray_in_ams_data(ams_data, assignment.ams_id, assignment.tray_id) + if not current_tray: + logger.info( + "Auto-unlink: spool %d AMS%d-T%d — tray not found in AMS data (slot empty?)", + assignment.spool_id, + assignment.ams_id, + assignment.tray_id, + ) + stale.append(assignment) # Slot empty + elif _is_bambu_uuid(current_tray.get("tray_uuid", "")): + # A Bambu Lab spool was inserted — always unlink manual assignments + logger.info( + "Auto-unlink: spool %d AMS%d-T%d — Bambu Lab spool detected (uuid=%s)", + assignment.spool_id, + assignment.ams_id, + assignment.tray_id, + current_tray.get("tray_uuid", ""), + ) + stale.append(assignment) + else: + cur_color = current_tray.get("tray_color", "") + cur_type = current_tray.get("tray_type", "") + fp_color = assignment.fingerprint_color or "" + fp_type = assignment.fingerprint_type or "" + if cur_color.upper() != fp_color.upper() or cur_type.upper() != fp_type.upper(): + # Fingerprint mismatch — but check if tray now matches the + # assigned spool (e.g. auto-configure changed the tray). + spool = assignment.spool + if spool: + spool_color = (spool.rgba or "FFFFFFFF").upper() + spool_type = (spool.material or "").upper() + if cur_color.upper() == spool_color and cur_type.upper() == spool_type: + # Tray was reconfigured to match the spool — update fingerprint + logger.info( + "Auto-unlink: spool %d AMS%d-T%d — fingerprint mismatch but tray matches spool, updating fp", + assignment.spool_id, + assignment.ams_id, + assignment.tray_id, + ) + assignment.fingerprint_color = cur_color + assignment.fingerprint_type = cur_type + continue + logger.info( + "Auto-unlink: spool %d AMS%d-T%d — fingerprint mismatch (cur=%s/%s fp=%s/%s spool=%s/%s)", + assignment.spool_id, + assignment.ams_id, + assignment.tray_id, + cur_color, + cur_type, + fp_color, + fp_type, + spool.rgba if spool else "?", + spool.material if spool else "?", + ) + stale.append(assignment) # Spool changed + for a in stale: + await db.delete(a) + if stale: + logger.info("Auto-unlinked %d stale spool assignments for printer %d", len(stale), printer_id) + # Commit any changes (stale deletions and/or fingerprint updates) + await db.commit() + except Exception as e: + logger.warning("Spool assignment cleanup failed: %s", e) + + # Auto-manage inventory spools from AMS tray data (skip if Spoolman manages AMS) + try: + async with async_session() as db: + from backend.app.api.routes.settings import get_setting + from backend.app.models.spool_assignment import SpoolAssignment as SA + from backend.app.services.spool_tag_matcher import ( + auto_assign_spool, + create_spool_from_tray, + get_spool_by_tag, + is_bambu_tag, + is_valid_tag, + ) + + _spoolman_on = await get_setting(db, "spoolman_enabled") + if not _spoolman_on or _spoolman_on.lower() != "true": + for ams_unit in ams_data: + if not isinstance(ams_unit, dict): + continue + ams_id = int(ams_unit.get("id", 0)) + for tray in ams_unit.get("tray", []): + if not isinstance(tray, dict): + continue + tray_id = int(tray.get("id", 0)) + tag_uid = tray.get("tag_uid", "") + tray_uuid = tray.get("tray_uuid", "") + tray_info_idx = tray.get("tray_info_idx", "") + if not tray.get("tray_type"): + continue # Empty slot + # Check if assignment already exists for this slot + existing = await db.execute( + select(SA) + .options(selectinload(SA.spool)) + .where(SA.printer_id == printer_id, SA.ams_id == ams_id, SA.tray_id == tray_id) + ) + existing_assignment = existing.scalar_one_or_none() + if existing_assignment: + # Sync spool weight_used from AMS remain if valid + remain_raw = tray.get("remain") + if remain_raw is not None and existing_assignment.spool: + try: + remain_val = int(remain_raw) + except (TypeError, ValueError): + remain_val = -1 + if 0 <= remain_val <= 100: + lw = existing_assignment.spool.label_weight or 1000 + new_used = round(lw * (100 - remain_val) / 100.0, 1) + if abs((existing_assignment.spool.weight_used or 0) - new_used) > 1: + logger.info( + "Weight sync: spool %d weight_used %s -> %s (remain=%d)", + existing_assignment.spool_id, + existing_assignment.spool.weight_used, + new_used, + remain_val, + ) + existing_assignment.spool.weight_used = new_used + await db.commit() + continue + + if is_bambu_tag(tag_uid, tray_uuid, tray_info_idx): + # BL spool with RFID tag: auto-match or auto-create + spool = await get_spool_by_tag(db, tag_uid, tray_uuid) + if not spool: + spool = await create_spool_from_tray(db, tray) + await auto_assign_spool( + printer_id, + ams_id, + tray_id, + spool, + printer_manager, + db, + ) + await db.commit() + await ws_manager.broadcast( + { + "type": "spool_auto_assigned", + "printer_id": printer_id, + "ams_id": ams_id, + "tray_id": tray_id, + "spool_id": spool.id, + } + ) + logger.info( + "RFID auto-assigned spool %d to printer %d AMS%d-T%d", + spool.id, + printer_id, + ams_id, + tray_id, + ) + elif is_valid_tag(tag_uid, tray_uuid): + # Non-BL spool with some tag — let user choose + await ws_manager.broadcast( + { + "type": "unknown_tag", + "printer_id": printer_id, + "ams_id": ams_id, + "tray_id": tray_id, + "tag_uid": tag_uid, + "tray_uuid": tray_uuid, + } + ) + else: + # No tag at all — let user choose from inventory + await ws_manager.broadcast( + { + "type": "unknown_tag", + "printer_id": printer_id, + "ams_id": ams_id, + "tray_id": tray_id, + "tag_uid": "", + "tray_uuid": "", + } + ) + except Exception as e: + logger.warning("RFID spool auto-assign failed: %s", e) + try: async with async_session() as db: from backend.app.api.routes.settings import get_setting @@ -571,6 +773,26 @@ async def on_ams_change(printer_id: int, ams_data: list): ) return + # Load inventory weights as fallback (when AMS MQTT data lacks remain values) + from sqlalchemy.orm import selectinload + + from backend.app.models.spool_assignment import SpoolAssignment + + inventory_weights: dict[tuple[int, int], float] = {} + try: + assign_result = await db.execute( + select(SpoolAssignment) + .options(selectinload(SpoolAssignment.spool)) + .where(SpoolAssignment.printer_id == printer_id) + ) + for assignment in assign_result.scalars().all(): + spool = assignment.spool + if spool and spool.label_weight > 0: + remaining = max(0.0, spool.label_weight - (spool.weight_used or 0)) + inventory_weights[(assignment.ams_id, assignment.tray_id)] = remaining + except Exception as e: + logger.debug("Could not load inventory weights for printer %s: %s", printer_id, e) + # Sync each AMS tray synced = 0 for ams_unit in ams_data: @@ -583,11 +805,13 @@ async def on_ams_change(printer_id: int, ams_data: list): continue # Empty tray try: + inv_remaining = inventory_weights.get((ams_id, tray.tray_id)) result = await client.sync_ams_tray( tray, printer_name, disable_weight_sync=disable_weight_sync, cached_spools=cached_spools, + inventory_remaining=inv_remaining, ) if result: synced += 1 @@ -749,6 +973,19 @@ async def on_print_start(printer_id: int, data: dict): except Exception: pass # Don't fail print start callback if MQTT fails + # Capture AMS tray remain% for filament consumption tracking (skip if Spoolman handles usage) + try: + async with async_session() as db: + from backend.app.api.routes.settings import get_setting + + _spoolman_on = await get_setting(db, "spoolman_enabled") + if not _spoolman_on or _spoolman_on.lower() != "true": + from backend.app.services.usage_tracker import on_print_start as usage_on_print_start + + await usage_on_print_start(printer_id, data, printer_manager) + except Exception as e: + logger.warning("Usage tracker on_print_start failed: %s", e) + # Track if notification was sent (to avoid sending twice) notification_sent = False @@ -1404,6 +1641,18 @@ async def on_print_start(printer_id: int, data: dict): await _store_spoolman_print_data(printer_id, archive.id, archive.file_path, db, printer_manager) except Exception as e: logger.warning("[SPOOLMAN] Failed to store tracking data: %s", e) + + # Capture timelapse file baseline for snapshot-diff on completion + try: + baseline_files, _ = await _list_timelapse_mp4s(printer) + _timelapse_baselines[printer_id] = {f.get("name", "") for f in baseline_files} + logger.info( + "[TIMELAPSE] Baseline at print start: %s MP4 files for printer %s", + len(_timelapse_baselines[printer_id]), + printer_id, + ) + except Exception as e: + logger.warning("[TIMELAPSE] Failed to capture baseline at print start: %s", e) finally: if temp_path and temp_path.exists(): temp_path.unlink() @@ -1435,7 +1684,7 @@ async def _list_timelapse_mp4s(printer) -> tuple[list[dict], str | None]: return [], None -async def _scan_for_timelapse_with_retries(archive_id: int): +async def _scan_for_timelapse_with_retries(archive_id: int, baseline_names: set[str] | None = None): """ Scan for timelapse with retries using a snapshot-diff approach. @@ -1443,6 +1692,10 @@ async def _scan_for_timelapse_with_retries(archive_id: int): clock is wrong in LAN-only mode), we snapshot existing MP4 filenames BEFORE waiting, then look for any NEW filename that appears after each delay. + If baseline_names is provided (captured at print start), it is used directly. + Otherwise falls back to taking a baseline at completion time (best-effort + for prints started before app restart). + Falls back to name-matching (print name contained in MP4 filename) if no new file appears after all retries. """ @@ -1468,18 +1721,28 @@ async def _scan_for_timelapse_with_retries(archive_id: int): logger.warning("[TIMELAPSE] Archive %s has no printer, aborting", archive_id) return - result = await db.execute(select(Printer).where(Printer.id == archive.printer_id)) - printer = result.scalar_one_or_none() - if not printer: - logger.warning("[TIMELAPSE] Printer not found for archive %s, aborting", archive_id) - return + if baseline_names is not None: + # Use pre-captured baseline from print start (no race condition) + logger.info( + "[TIMELAPSE] Using print-start baseline: %s existing MP4 files for archive %s", + len(baseline_names), + archive_id, + ) + else: + # Fallback: take baseline now (e.g. app restarted mid-print) + result = await db.execute(select(Printer).where(Printer.id == archive.printer_id)) + printer = result.scalar_one_or_none() + if not printer: + logger.warning("[TIMELAPSE] Printer not found for archive %s, aborting", archive_id) + return - # Snapshot current MP4 filenames as baseline - baseline_files, _ = await _list_timelapse_mp4s(printer) - baseline_names: set[str] = {f.get("name", "") for f in baseline_files} - logger.info( - "[TIMELAPSE] Baseline snapshot: %s existing MP4 files for archive %s", len(baseline_names), archive_id - ) + baseline_files, _ = await _list_timelapse_mp4s(printer) + baseline_names = {f.get("name", "") for f in baseline_files} + logger.info( + "[TIMELAPSE] Baseline snapshot (fallback): %s existing MP4 files for archive %s", + len(baseline_names), + archive_id, + ) # Derive base_name for name-matching fallback base_name = Path(archive.filename).stem if archive.filename else "" @@ -1837,6 +2100,31 @@ async def on_print_complete(printer_id: int, data: dict): log_timing("Archive status update") + # Track filament consumption from AMS remain% deltas (skip if Spoolman handles usage) + try: + async with async_session() as db: + from backend.app.api.routes.settings import get_setting + + _spoolman_on = await get_setting(db, "spoolman_enabled") + if not _spoolman_on or _spoolman_on.lower() != "true": + from backend.app.services.usage_tracker import on_print_complete as usage_on_print_complete + + async with async_session() as db: + usage_results = await usage_on_print_complete( + printer_id, data, printer_manager, db, archive_id=archive_id + ) + if usage_results: + await ws_manager.broadcast( + { + "type": "spool_usage_logged", + "printer_id": printer_id, + "usage": usage_results, + } + ) + log_timing("Usage tracker") + except Exception as e: + logger.warning("Usage tracker on_print_complete failed: %s", e) + # Report filament usage to Spoolman if print completed successfully if data.get("status") == "completed": try: @@ -2179,7 +2467,8 @@ async def on_print_complete(printer_id: int, data: dict): logger.info("[TIMELAPSE] Timelapse was active during print, scheduling auto-scan for archive %s", archive_id) # Schedule timelapse scan as background task with retries # The printer needs time to encode the video after print completion - asyncio.create_task(_scan_for_timelapse_with_retries(archive_id)) + baseline = _timelapse_baselines.pop(printer_id, None) + asyncio.create_task(_scan_for_timelapse_with_retries(archive_id, baseline)) log_timing("Timelapse scan scheduled") # Update queue item if this was a scheduled print @@ -2746,6 +3035,10 @@ async def lifespan(app: FastAPI): if virtual_printer_manager.is_enabled: await virtual_printer_manager.configure(enabled=False) + await mqtt_smart_plug_service.disconnect(timeout=2) + + await mqtt_relay.disconnect(timeout=2) + app = FastAPI( title=app_settings.app_name, @@ -2905,6 +3198,7 @@ app.include_router(groups.router, prefix=app_settings.api_prefix) app.include_router(printers.router, prefix=app_settings.api_prefix) app.include_router(archives.router, prefix=app_settings.api_prefix) app.include_router(filaments.router, prefix=app_settings.api_prefix) +app.include_router(inventory.router, prefix=app_settings.api_prefix) app.include_router(settings_routes.router, prefix=app_settings.api_prefix) app.include_router(cloud.router, prefix=app_settings.api_prefix) app.include_router(local_presets.router, prefix=app_settings.api_prefix) diff --git a/backend/app/models/__init__.py b/backend/app/models/__init__.py index 30b5f09c1..1a53df1c8 100644 --- a/backend/app/models/__init__.py +++ b/backend/app/models/__init__.py @@ -1,6 +1,7 @@ from backend.app.models.ams_history import AMSSensorHistory from backend.app.models.api_key import APIKey from backend.app.models.archive import PrintArchive +from backend.app.models.color_catalog import ColorCatalogEntry from backend.app.models.filament import Filament from backend.app.models.github_backup import GitHubBackupConfig, GitHubBackupLog from backend.app.models.group import Group, user_groups @@ -16,6 +17,11 @@ from backend.app.models.printer import Printer from backend.app.models.project import Project from backend.app.models.settings import Settings from backend.app.models.smart_plug import SmartPlug +from backend.app.models.spool import Spool +from backend.app.models.spool_assignment import SpoolAssignment +from backend.app.models.spool_catalog import SpoolCatalogEntry +from backend.app.models.spool_k_profile import SpoolKProfile +from backend.app.models.spool_usage_history import SpoolUsageHistory from backend.app.models.user import User __all__ = [ @@ -43,4 +49,10 @@ __all__ = [ "GitHubBackupLog", "LocalPreset", "OrcaBaseProfile", + "Spool", + "SpoolKProfile", + "SpoolAssignment", + "SpoolCatalogEntry", + "SpoolUsageHistory", + "ColorCatalogEntry", ] diff --git a/backend/app/models/color_catalog.py b/backend/app/models/color_catalog.py new file mode 100644 index 000000000..fa2a66bdf --- /dev/null +++ b/backend/app/models/color_catalog.py @@ -0,0 +1,20 @@ +from datetime import datetime + +from sqlalchemy import Boolean, DateTime, String, func +from sqlalchemy.orm import Mapped, mapped_column + +from backend.app.core.database import Base + + +class ColorCatalogEntry(Base): + """Color catalog entry for automatic color lookup when adding spools.""" + + __tablename__ = "color_catalog" + + id: Mapped[int] = mapped_column(primary_key=True) + manufacturer: Mapped[str] = mapped_column(String(200)) + color_name: Mapped[str] = mapped_column(String(200)) + hex_color: Mapped[str] = mapped_column(String(7)) # #RRGGBB + material: Mapped[str | None] = mapped_column(String(100)) + is_default: Mapped[bool] = mapped_column(Boolean, default=False) + created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now()) diff --git a/backend/app/models/external_link.py b/backend/app/models/external_link.py index f0d1b3bea..7cdeff0c2 100644 --- a/backend/app/models/external_link.py +++ b/backend/app/models/external_link.py @@ -1,6 +1,6 @@ from datetime import datetime -from sqlalchemy import DateTime, Integer, String, func +from sqlalchemy import Boolean, DateTime, Integer, String, func from sqlalchemy.orm import Mapped, mapped_column from backend.app.core.database import Base @@ -16,6 +16,7 @@ class ExternalLink(Base): url: Mapped[str] = mapped_column(String(500)) icon: Mapped[str] = mapped_column(String(50), default="link") custom_icon: Mapped[str | None] = mapped_column(String(255), nullable=True) # Filename of uploaded icon + open_in_new_tab: Mapped[bool] = mapped_column(Boolean, default=False) sort_order: Mapped[int] = mapped_column(Integer, default=0) created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now()) updated_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now()) diff --git a/backend/app/models/spool.py b/backend/app/models/spool.py new file mode 100644 index 000000000..893e4db66 --- /dev/null +++ b/backend/app/models/spool.py @@ -0,0 +1,44 @@ +from datetime import datetime + +from sqlalchemy import DateTime, Float, Integer, String, func +from sqlalchemy.orm import Mapped, mapped_column, relationship + +from backend.app.core.database import Base + + +class Spool(Base): + """Spool inventory item for tracking filament spools and their properties.""" + + __tablename__ = "spool" + + id: Mapped[int] = mapped_column(primary_key=True) + material: Mapped[str] = mapped_column(String(50)) # PLA, PETG, ABS, etc. + subtype: Mapped[str | None] = mapped_column(String(50)) # Basic, Matte, Silk, etc. + color_name: Mapped[str | None] = mapped_column(String(100)) # "Jade White" + rgba: Mapped[str | None] = mapped_column(String(8)) # RRGGBBAA hex + brand: Mapped[str | None] = mapped_column(String(100)) # "Polymaker" + label_weight: Mapped[int] = mapped_column(Integer, default=1000) # Advertised net weight (g) + core_weight: Mapped[int] = mapped_column(Integer, default=250) # Empty spool weight (g) + weight_used: Mapped[float] = mapped_column(Float, default=0) # Consumed grams + slicer_filament: Mapped[str | None] = mapped_column(String(50)) # Preset ID (e.g. "GFL99") + slicer_filament_name: Mapped[str | None] = mapped_column(String(100)) # Preset name for slicer + nozzle_temp_min: Mapped[int | None] = mapped_column() # Override min temp + nozzle_temp_max: Mapped[int | None] = mapped_column() # Override max temp + note: Mapped[str | None] = mapped_column(String(500)) + added_full: Mapped[bool | None] = mapped_column() # Whether spool was added as full (unused) + last_used: Mapped[datetime | None] = mapped_column(DateTime) # Last time this spool was used in a print + encode_time: Mapped[datetime | None] = mapped_column(DateTime) # When spool was encoded/written to tag + tag_uid: Mapped[str | None] = mapped_column(String(16)) # RFID tag UID (16 hex chars) + tray_uuid: Mapped[str | None] = mapped_column(String(32)) # Bambu Lab spool UUID (32 hex chars) + data_origin: Mapped[str | None] = mapped_column(String(20)) # How data was populated: manual, rfid_auto, nfc_link + tag_type: Mapped[str | None] = mapped_column(String(20)) # Tag vendor: bambulab, generic, etc. + archived_at: Mapped[datetime | None] = mapped_column(DateTime) # NULL = active + created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now()) + updated_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now()) + + k_profiles: Mapped[list["SpoolKProfile"]] = relationship(back_populates="spool", cascade="all, delete-orphan") + assignments: Mapped[list["SpoolAssignment"]] = relationship(back_populates="spool", cascade="all, delete-orphan") + + +from backend.app.models.spool_assignment import SpoolAssignment # noqa: E402 +from backend.app.models.spool_k_profile import SpoolKProfile # noqa: E402 diff --git a/backend/app/models/spool_assignment.py b/backend/app/models/spool_assignment.py new file mode 100644 index 000000000..23cbee47c --- /dev/null +++ b/backend/app/models/spool_assignment.py @@ -0,0 +1,35 @@ +from datetime import datetime + +from sqlalchemy import DateTime, ForeignKey, Integer, String, UniqueConstraint, func +from sqlalchemy.orm import Mapped, mapped_column, relationship + +from backend.app.core.database import Base + + +class SpoolAssignment(Base): + """Assignment of a spool to a specific AMS slot on a printer.""" + + __tablename__ = "spool_assignment" + + id: Mapped[int] = mapped_column(primary_key=True) + spool_id: Mapped[int] = mapped_column(ForeignKey("spool.id", ondelete="CASCADE")) + printer_id: Mapped[int] = mapped_column(ForeignKey("printers.id", ondelete="CASCADE")) + ams_id: Mapped[int] = mapped_column(Integer) # 0-3, 128+ (HT), 254/255 (ext) + tray_id: Mapped[int] = mapped_column(Integer) # 0-3 + fingerprint_color: Mapped[str | None] = mapped_column(String(8)) # tray_color snapshot + fingerprint_type: Mapped[str | None] = mapped_column(String(50)) # tray_type snapshot + created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now()) + + spool: Mapped["Spool"] = relationship(back_populates="assignments") + printer: Mapped["Printer"] = relationship() + + __table_args__ = (UniqueConstraint("printer_id", "ams_id", "tray_id"),) + + @property + def printer_name(self) -> str | None: + """Get printer name from loaded relationship.""" + return self.printer.name if self.printer else None + + +from backend.app.models.printer import Printer # noqa: E402, F401 +from backend.app.models.spool import Spool # noqa: E402, F401 diff --git a/backend/app/models/spool_catalog.py b/backend/app/models/spool_catalog.py new file mode 100644 index 000000000..d2370b982 --- /dev/null +++ b/backend/app/models/spool_catalog.py @@ -0,0 +1,18 @@ +from datetime import datetime + +from sqlalchemy import Boolean, DateTime, Integer, String, func +from sqlalchemy.orm import Mapped, mapped_column + +from backend.app.core.database import Base + + +class SpoolCatalogEntry(Base): + """Spool weight catalog entry for weight lookup when adding spools.""" + + __tablename__ = "spool_catalog" + + id: Mapped[int] = mapped_column(primary_key=True) + name: Mapped[str] = mapped_column(String(200)) + weight: Mapped[int] = mapped_column(Integer) + is_default: Mapped[bool] = mapped_column(Boolean, default=False) + created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now()) diff --git a/backend/app/models/spool_k_profile.py b/backend/app/models/spool_k_profile.py new file mode 100644 index 000000000..165b86a54 --- /dev/null +++ b/backend/app/models/spool_k_profile.py @@ -0,0 +1,31 @@ +from datetime import datetime + +from sqlalchemy import DateTime, Float, ForeignKey, Integer, String, func +from sqlalchemy.orm import Mapped, mapped_column, relationship + +from backend.app.core.database import Base + + +class SpoolKProfile(Base): + """K-value calibration profile for a spool on a specific printer/nozzle combo.""" + + __tablename__ = "spool_k_profile" + + id: Mapped[int] = mapped_column(primary_key=True) + spool_id: Mapped[int] = mapped_column(ForeignKey("spool.id", ondelete="CASCADE")) + printer_id: Mapped[int] = mapped_column(ForeignKey("printers.id", ondelete="CASCADE")) + extruder: Mapped[int] = mapped_column(Integer, default=0) # 0 or 1 (H2D) + nozzle_diameter: Mapped[str] = mapped_column(String(10), default="0.4") # "0.4", "0.6" + nozzle_type: Mapped[str | None] = mapped_column(String(50)) + k_value: Mapped[float] = mapped_column(Float) # e.g. 0.020 + name: Mapped[str | None] = mapped_column(String(100)) # Profile display name + cali_idx: Mapped[int | None] = mapped_column(Integer) # Calibration index on printer + setting_id: Mapped[str | None] = mapped_column(String(50)) # Full setting ID + created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now()) + + spool: Mapped["Spool"] = relationship(back_populates="k_profiles") + printer: Mapped["Printer"] = relationship() + + +from backend.app.models.printer import Printer # noqa: E402, F401 +from backend.app.models.spool import Spool # noqa: E402, F401 diff --git a/backend/app/models/spool_usage_history.py b/backend/app/models/spool_usage_history.py new file mode 100644 index 000000000..15fcb0543 --- /dev/null +++ b/backend/app/models/spool_usage_history.py @@ -0,0 +1,21 @@ +from datetime import datetime + +from sqlalchemy import DateTime, Float, ForeignKey, Integer, String, func +from sqlalchemy.orm import Mapped, mapped_column + +from backend.app.core.database import Base + + +class SpoolUsageHistory(Base): + """Record of filament consumption for a spool during a print.""" + + __tablename__ = "spool_usage_history" + + id: Mapped[int] = mapped_column(primary_key=True) + spool_id: Mapped[int] = mapped_column(ForeignKey("spool.id", ondelete="CASCADE")) + printer_id: Mapped[int | None] = mapped_column(ForeignKey("printers.id", ondelete="SET NULL")) + print_name: Mapped[str | None] = mapped_column(String(500)) + weight_used: Mapped[float] = mapped_column(Float, default=0) + percent_used: Mapped[int] = mapped_column(Integer, default=0) + status: Mapped[str] = mapped_column(String(20), default="completed") # completed/failed/aborted + created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now()) diff --git a/backend/app/schemas/cloud.py b/backend/app/schemas/cloud.py index a4f8c80f7..f315b4000 100644 --- a/backend/app/schemas/cloud.py +++ b/backend/app/schemas/cloud.py @@ -49,6 +49,7 @@ class SlicerSetting(BaseModel): version: str | None = None user_id: str | None = None updated_time: str | None = None + is_custom: bool = False class SlicerSettingsResponse(BaseModel): diff --git a/backend/app/schemas/external_link.py b/backend/app/schemas/external_link.py index 2c5facba1..2a5676561 100644 --- a/backend/app/schemas/external_link.py +++ b/backend/app/schemas/external_link.py @@ -9,6 +9,7 @@ class ExternalLinkBase(BaseModel): name: str = Field(..., min_length=1, max_length=50, description="Display name for the link") url: str = Field(..., min_length=1, max_length=500, description="External URL") icon: str = Field(default="link", max_length=50, description="Lucide icon name") + open_in_new_tab: bool = False @field_validator("url") @classmethod @@ -31,6 +32,7 @@ class ExternalLinkUpdate(BaseModel): name: str | None = Field(default=None, min_length=1, max_length=50) url: str | None = Field(default=None, min_length=1, max_length=500) icon: str | None = Field(default=None, max_length=50) + open_in_new_tab: bool | None = None @field_validator("url") @classmethod @@ -45,6 +47,7 @@ class ExternalLinkResponse(ExternalLinkBase): """Response schema for external links.""" id: int + open_in_new_tab: bool custom_icon: str | None = None sort_order: int created_at: datetime diff --git a/backend/app/schemas/spool.py b/backend/app/schemas/spool.py new file mode 100644 index 000000000..6e86af4c6 --- /dev/null +++ b/backend/app/schemas/spool.py @@ -0,0 +1,109 @@ +from datetime import datetime + +from pydantic import BaseModel, Field + + +class SpoolBase(BaseModel): + material: str = Field(..., min_length=1, max_length=50) + subtype: str | None = None + color_name: str | None = None + rgba: str | None = Field(None, pattern=r"^[0-9A-Fa-f]{8}$") + brand: str | None = None + label_weight: int = 1000 + core_weight: int = 250 + weight_used: float = 0 + slicer_filament: str | None = None + slicer_filament_name: str | None = None + nozzle_temp_min: int | None = None + nozzle_temp_max: int | None = None + note: str | None = None + tag_uid: str | None = None + tray_uuid: str | None = None + data_origin: str | None = None + tag_type: str | None = None + + +class SpoolCreate(SpoolBase): + pass + + +class SpoolUpdate(BaseModel): + material: str | None = None + subtype: str | None = None + color_name: str | None = None + rgba: str | None = None + brand: str | None = None + label_weight: int | None = None + core_weight: int | None = None + weight_used: float | None = None + slicer_filament: str | None = None + slicer_filament_name: str | None = None + nozzle_temp_min: int | None = None + nozzle_temp_max: int | None = None + note: str | None = None + tag_uid: str | None = None + tray_uuid: str | None = None + data_origin: str | None = None + tag_type: str | None = None + + +class SpoolKProfileBase(BaseModel): + printer_id: int + extruder: int = 0 + nozzle_diameter: str = "0.4" + nozzle_type: str | None = None + k_value: float + name: str | None = None + cali_idx: int | None = None + setting_id: str | None = None + + +class SpoolKProfileResponse(SpoolKProfileBase): + id: int + spool_id: int + created_at: datetime + + class Config: + from_attributes = True + + +class SpoolResponse(SpoolBase): + id: int + added_full: bool | None = None + last_used: datetime | None = None + encode_time: datetime | None = None + tag_uid: str | None = None + tray_uuid: str | None = None + data_origin: str | None = None + tag_type: str | None = None + archived_at: datetime | None = None + created_at: datetime + updated_at: datetime + k_profiles: list[SpoolKProfileResponse] = [] + + class Config: + from_attributes = True + + +class SpoolAssignmentCreate(BaseModel): + spool_id: int + printer_id: int + ams_id: int + tray_id: int + + +class SpoolAssignmentResponse(BaseModel): + id: int + spool_id: int + printer_id: int + printer_name: str | None = None + ams_id: int + tray_id: int + fingerprint_color: str | None = None + fingerprint_type: str | None = None + created_at: datetime + spool: SpoolResponse | None = None + configured: bool = False + + class Config: + from_attributes = True diff --git a/backend/app/schemas/spool_usage.py b/backend/app/schemas/spool_usage.py new file mode 100644 index 000000000..2ba82ac05 --- /dev/null +++ b/backend/app/schemas/spool_usage.py @@ -0,0 +1,17 @@ +from datetime import datetime + +from pydantic import BaseModel + + +class SpoolUsageHistoryResponse(BaseModel): + id: int + spool_id: int + printer_id: int | None = None + print_name: str | None = None + weight_used: float + percent_used: int + status: str + created_at: datetime + + class Config: + from_attributes = True diff --git a/backend/app/services/bambu_mqtt.py b/backend/app/services/bambu_mqtt.py index dfad91417..b8e5e774d 100644 --- a/backend/app/services/bambu_mqtt.py +++ b/backend/app/services/bambu_mqtt.py @@ -11,6 +11,7 @@ import asyncio import json import logging import ssl +import threading import time from collections import deque from collections.abc import Callable @@ -279,6 +280,7 @@ class BambuMQTTClient: self._message_log: deque[MQTTLogEntry] = deque(maxlen=100) self._logging_enabled: bool = False self._last_message_time: float = 0.0 # Track when we last received a message + self._disconnection_event: threading.Event | None = None self._previous_ams_hash: str | None = None # Track AMS changes # K-profile command tracking @@ -357,6 +359,8 @@ class BambuMQTTClient: self.state.connected = False if self.on_state_change: self.on_state_change(self.state) + if self._disconnection_event: + self._disconnection_event.set() def _on_message(self, client, userdata, msg): try: @@ -941,8 +945,9 @@ class BambuMQTTClient: # Fields that should always be updated (even with empty/zero values): # - remain, k, id, cali_idx: status indicators where 0 is valid # - tray_type, tray_sub_brands, tag_uid, tray_uuid, tray_info_idx, - # tray_color, tray_id_name: slot content indicators that must be - # cleared when a spool is removed (fixes #147 - old AMS empty slot) + # tray_color, tray_id_name: slot content indicators that must + # be cleared when a spool is removed (fixes #147 - old AMS + # empty slot) always_update_fields = ( "remain", "k", @@ -1062,7 +1067,9 @@ class BambuMQTTClient: self._previous_ams_hash = ams_hash if self.on_ams_change: logger.info("[%s] AMS data changed, triggering sync callback", self.serial_number) - self.on_ams_change(ams_list) + # Pass merged AMS data (not raw ams_list) — partial MQTT updates + # may lack fields like 'remain' that the merged state preserves + self.on_ams_change(merged_ams) def _update_state(self, data: dict): """Update printer state from message data.""" @@ -2371,11 +2378,13 @@ class BambuMQTTClient: return True - def disconnect(self): + def disconnect(self, timeout: float = 0): """Disconnect from the printer.""" if self._client: - self._client.loop_stop() + self._disconnection_event = threading.Event() self._client.disconnect() + self._disconnection_event.wait(timeout=timeout) + self._client.loop_stop() self._client = None self.state.connected = False diff --git a/backend/app/services/external_camera.py b/backend/app/services/external_camera.py index 1ffa43782..d891ac9ff 100644 --- a/backend/app/services/external_camera.py +++ b/backend/app/services/external_camera.py @@ -362,7 +362,7 @@ async def _capture_rtsp_frame(url: str, timeout: int) -> bytes | None: ] try: - print(f"[EXT-CAM] Running ffmpeg command: {' '.join(cmd[:6])}...") + logger.debug(f"Running ffmpeg command: {' '.join(cmd[:6])}...") process = await asyncio.create_subprocess_exec( *cmd, stdout=asyncio.subprocess.PIPE, @@ -370,13 +370,12 @@ async def _capture_rtsp_frame(url: str, timeout: int) -> bytes | None: ) stdout, stderr = await asyncio.wait_for(process.communicate(), timeout=timeout) - print( - f"[EXT-CAM] ffmpeg returned: code={process.returncode}, stdout={len(stdout)} bytes, stderr={len(stderr)} bytes" + logger.debug( + f"ffmpeg returned: code={process.returncode}, stdout={len(stdout)} bytes, stderr={len(stderr)} bytes" ) if process.returncode != 0: logger.error("ffmpeg RTSP capture failed: %s", stderr.decode()[:200]) - print(f"[EXT-CAM] ffmpeg error: {stderr.decode()[:300]}") return None if not stdout or len(stdout) < 100: @@ -440,11 +439,9 @@ async def test_connection(url: str, camera_type: str) -> dict: Returns: Dict with {success: bool, error?: str, resolution?: str} """ - print(f"[EXT-CAM] Testing camera connection: type={camera_type}, url={url[:50]}...") logger.info("Testing camera connection: type=%s, url=%s...", camera_type, url[:50]) try: frame = await capture_frame(url, camera_type, timeout=10) - print(f"[EXT-CAM] Capture result: {len(frame) if frame else 0} bytes") logger.info("Capture result: %s bytes", len(frame) if frame else 0) if frame: @@ -490,17 +487,41 @@ async def generate_mjpeg_stream(url: str, camera_type: str, fps: int = 10) -> As last_frame_time = 0.0 if camera_type == "mjpeg": - # Proxy MJPEG stream directly - async for frame in _stream_mjpeg(url): - current_time = asyncio.get_event_loop().time() - if current_time - last_frame_time >= frame_interval: - last_frame_time = current_time - yield _format_mjpeg_frame(frame) + # Proxy MJPEG stream directly, with reconnect on timeout + max_retries = 3 + for attempt in range(max_retries + 1): + frame_yielded = False + async for frame in _stream_mjpeg(url): + frame_yielded = True + current_time = asyncio.get_event_loop().time() + if current_time - last_frame_time >= frame_interval: + last_frame_time = current_time + yield _format_mjpeg_frame(frame) + if not frame_yielded or attempt == max_retries: + break + logger.warning( + "External MJPEG stream ended, reconnecting (attempt %d/%d)...", + attempt + 1, + max_retries, + ) + await asyncio.sleep(2) elif camera_type == "rtsp": - # Use ffmpeg to convert RTSP to MJPEG - async for frame in _stream_rtsp(url, fps): - yield _format_mjpeg_frame(frame) + # Use ffmpeg to convert RTSP to MJPEG, with reconnect on timeout + max_retries = 3 + for attempt in range(max_retries + 1): + frame_yielded = False + async for frame in _stream_rtsp(url, fps): + frame_yielded = True + yield _format_mjpeg_frame(frame) + if not frame_yielded or attempt == max_retries: + break + logger.warning( + "External RTSP stream ended, reconnecting (attempt %d/%d)...", + attempt + 1, + max_retries, + ) + await asyncio.sleep(2) elif camera_type == "usb": # Use ffmpeg to stream from USB camera diff --git a/backend/app/services/firmware_check.py b/backend/app/services/firmware_check.py index 62e09c6ca..9d8328900 100644 --- a/backend/app/services/firmware_check.py +++ b/backend/app/services/firmware_check.py @@ -1,8 +1,9 @@ """ Firmware Check Service -Checks for firmware updates by fetching from Bambu Lab's official firmware download page. -Also provides firmware download functionality for offline updates. +Checks for firmware updates by fetching from Bambu Lab's official wiki and firmware +download page. The wiki is used as the primary version source (always up-to-date), +while the download page provides firmware file URLs for offline updates. """ import logging @@ -18,10 +19,13 @@ from backend.app.core.config import _data_dir logger = logging.getLogger(__name__) -# Bambu Lab firmware download page +# Bambu Lab firmware download page (for download URLs) BAMBU_FIRMWARE_BASE = "https://bambulab.com" FIRMWARE_PAGE = "/en/support/firmware-download/all" +# Bambu Lab wiki (primary source for latest version detection) +BAMBU_WIKI_BASE = "https://wiki.bambulab.com" + # Cache TTL in seconds (1 hour) CACHE_TTL = 3600 @@ -61,6 +65,20 @@ API_KEY_TO_DEV_MODEL = { "h2d-pro": "O1E", } +# Wiki firmware release history pages (primary version source) +API_KEY_TO_WIKI_PATH = { + "x1": "/en/x1/manual/X1-X1C-firmware-release-history", + "x1e": "/en/x1/manual/X1E-firmware-release-history", + "p1": "/en/p1/manual/p1p-firmware-release-history", + "a1": "/en/a1/manual/a1-firmware-release-history", + "a1-mini": "/en/a1-mini/manual/a1-mini-firmware-release-history", + "h2d": "/en/h2d/manual/h2d-firmware-release-history", + "h2c": "/en/h2c/manual/h2c-firmware-release-history", + "h2s": "/en/h2s/manual/h2s-firmware-release-history", + "p2s": "/en/p2s/manual/p2s-firmware-release-history", + "h2d-pro": "/en/h2d-pro/manual/firmware-release-history", +} + @dataclass class FirmwareVersion: @@ -109,11 +127,34 @@ class FirmwareCheckService: return self._build_id # Return cached value if available - async def _fetch_firmware_versions(self, api_key: str) -> FirmwareVersion | None: - """Fetch firmware versions for a specific printer from Bambu Lab API.""" + async def _fetch_version_from_wiki(self, api_key: str) -> str | None: + """Fetch the latest firmware version from Bambu Lab's wiki release history page.""" + wiki_path = API_KEY_TO_WIKI_PATH.get(api_key) + if not wiki_path: + return None + + try: + url = f"{BAMBU_WIKI_BASE}{wiki_path}" + response = await self._client.get(url, follow_redirects=True) + + if response.status_code == 200: + # Extract version strings (format: XX.XX.XX.XX), first match is the latest + versions = re.findall(r"(\d{2}\.\d{2}\.\d{2}\.\d{2})", response.text) + if versions: + logger.debug("Wiki firmware for %s: %s", api_key, versions[0]) + return versions[0] + else: + logger.debug("Wiki firmware page for %s returned %s", api_key, response.status_code) + + except Exception as e: + logger.debug("Error fetching wiki firmware for %s: %s", api_key, e) + + return None + + async def _fetch_from_download_page(self, api_key: str) -> FirmwareVersion | None: + """Fetch firmware info from Bambu Lab's download page (has download URLs).""" build_id = await self._get_build_id() if not build_id: - logger.warning("No build ID available, cannot fetch firmware versions") return None try: @@ -135,16 +176,39 @@ class FirmwareCheckService: release_notes=latest.get("release_notes_en"), release_time=latest.get("release_time"), ) - else: - # api_key is a printer model identifier (e.g. "x1", "p1"), not a secret - logger.warning("Failed to fetch firmware for %s: %s", api_key, response.status_code) except Exception as e: - # api_key is a printer model identifier (e.g. "x1", "p1"), not a secret - logger.error("Error fetching firmware for %s: %s", api_key, e) + logger.debug("Error fetching download page firmware for %s: %s", api_key, e) return None + async def _fetch_firmware_versions(self, api_key: str) -> FirmwareVersion | None: + """Fetch firmware version info, using wiki as primary source and download page as fallback.""" + # Try wiki first (always has the latest version) + wiki_version = await self._fetch_version_from_wiki(api_key) + + # Try download page (has download URLs, may lag behind wiki) + download_info = await self._fetch_from_download_page(api_key) + + if wiki_version: + # Wiki has the latest version — use it, attach download URL if available + download_url = "" + release_notes = None + if download_info and download_info.version == wiki_version: + download_url = download_info.download_url + release_notes = download_info.release_notes + return FirmwareVersion( + version=wiki_version, + download_url=download_url, + release_notes=release_notes, + ) + + if download_info: + return download_info + + logger.warning("Could not fetch firmware info for %s from wiki or download page", api_key) + return None + async def get_latest_version(self, model: str) -> FirmwareVersion | None: """ Get the latest firmware version for a printer model. diff --git a/backend/app/services/mqtt_relay.py b/backend/app/services/mqtt_relay.py index c2d72d673..9987c970e 100644 --- a/backend/app/services/mqtt_relay.py +++ b/backend/app/services/mqtt_relay.py @@ -36,6 +36,7 @@ class MQTTRelayService: self._last_printer_status: dict[int, float] = {} # printer_id -> last publish timestamp self._smart_plug_service = None # Lazy import to avoid circular dependency self._settings: dict = {} # Store settings for smart plug service + self._disconnection_event: threading.Event | None = None async def configure(self, settings: dict) -> bool: """Configure MQTT connection from settings. @@ -187,15 +188,19 @@ class MQTTRelayService: logger.warning("MQTT relay disconnected: %s", rc) else: logger.info("MQTT relay disconnected cleanly") + if self._disconnection_event: + self._disconnection_event.set() - async def disconnect(self): + async def disconnect(self, timeout: float = 0): """Disconnect from MQTT broker.""" if self.client: try: # Publish offline status before disconnecting self._publish_status("offline") - self.client.loop_stop() + self._disconnection_event = threading.Event() self.client.disconnect() + await asyncio.to_thread(self._disconnection_event.wait, timeout=timeout) + self.client.loop_stop() except Exception as e: logger.debug("MQTT disconnect error (ignored): %s", e) finally: diff --git a/backend/app/services/mqtt_smart_plug.py b/backend/app/services/mqtt_smart_plug.py index 87ec7ab77..525ebec55 100644 --- a/backend/app/services/mqtt_smart_plug.py +++ b/backend/app/services/mqtt_smart_plug.py @@ -3,6 +3,7 @@ This service enables integration with Shelly, Zigbee2MQTT, and other MQTT-based energy monitoring devices. """ +import asyncio import json import logging import threading @@ -52,6 +53,7 @@ class MQTTSmartPlugService: self.plug_configs: dict[int, dict[str, MQTTDataSourceConfig]] = {} # plug_id -> latest data self.plug_data: dict[int, SmartPlugMQTTData] = {} + self._disconnection_event: threading.Event | None = None self._configured = False self._broker = "" self._port = 1883 @@ -209,6 +211,8 @@ class MQTTSmartPlugService: logger.warning("MQTT smart plug service disconnected: %s", rc) else: logger.info("MQTT smart plug service disconnected cleanly") + if self._disconnection_event: + self._disconnection_event.set() def _on_message(self, client: mqtt.Client, userdata: Any, msg: mqtt.MQTTMessage): """Handle incoming MQTT message, extract data using JSON path.""" @@ -471,12 +475,14 @@ class MQTTSmartPlugService: timeout = timedelta(minutes=self.REACHABLE_TIMEOUT_MINUTES) return datetime.utcnow() - data.last_seen < timeout - async def disconnect(self): + async def disconnect(self, timeout: float = 0): """Disconnect from MQTT broker.""" if self.client: try: - self.client.loop_stop() + self._disconnection_event = threading.Event() self.client.disconnect() + await asyncio.to_thread(self._disconnection_event.wait, timeout=timeout) + self.client.loop_stop() except Exception as e: logger.debug("MQTT smart plug disconnect error (ignored): %s", e) finally: diff --git a/backend/app/services/notification_service.py b/backend/app/services/notification_service.py index 4b229030a..0111c352e 100644 --- a/backend/app/services/notification_service.py +++ b/backend/app/services/notification_service.py @@ -267,19 +267,20 @@ class NotificationService: url = f"https://api.telegram.org/bot{bot_token}/sendMessage" - # Check if message contains characters that break Markdown parsing - # URLs and error codes with underscores cause issues - has_url = "http://" in message or "https://" in message - # Check for underscores outside of the bold title (odd number of _ breaks markdown) - body_part = message.split("\n", 1)[1] if "\n" in message else "" - has_problematic_underscore = "_" in body_part + # Escape underscores in the message body so Telegram Markdown + # parsing doesn't break on job names like "A1_plate_8" or error + # codes like "0300_0001". The title is already wrapped in *bold* + # markers, so only escape after the first newline. + if "\n" in message: + title_part, body_part = message.split("\n", 1) + body_part = body_part.replace("_", "\\_") + message = f"{title_part}\n{body_part}" data = { "chat_id": chat_id, "text": message, + "parse_mode": "Markdown", } - if not has_url and not has_problematic_underscore: - data["parse_mode"] = "Markdown" client = await self._get_client() response = await client.post(url, json=data) @@ -344,7 +345,9 @@ class NotificationService: except Exception as e: return False, f"Email error: {str(e)}" - async def _send_discord(self, config: dict, title: str, message: str) -> tuple[bool, str]: + async def _send_discord( + self, config: dict, title: str, message: str, image_data: bytes | None = None + ) -> tuple[bool, str]: """Send notification via Discord webhook.""" webhook_url = config.get("webhook_url", "").strip() @@ -355,18 +358,25 @@ class NotificationService: return False, "Invalid Discord webhook URL" # Discord embed format for nicer messages - data = { - "embeds": [ - { - "title": title, - "description": message, - "color": 0x00AE42, # Bambu green - } - ] + embed = { + "title": title, + "description": message, + "color": 0x00AE42, # Bambu green } client = await self._get_client() - response = await client.post(webhook_url, json=data) + + if image_data: + # Attach image via multipart form-data and reference in embed + embed["image"] = {"url": "attachment://photo.jpg"} + payload = {"embeds": [embed]} + response = await client.post( + webhook_url, + data={"payload_json": json.dumps(payload)}, + files={"files[0]": ("photo.jpg", image_data, "image/jpeg")}, + ) + else: + response = await client.post(webhook_url, json={"embeds": [embed]}) if response.status_code in (200, 204): return True, "Message sent successfully" @@ -449,7 +459,7 @@ class NotificationService: elif provider.provider_type == "email": return await self._send_email(config, title, message) elif provider.provider_type == "discord": - return await self._send_discord(config, title, message) + return await self._send_discord(config, title, message, image_data=image_data) elif provider.provider_type == "webhook": return await self._send_webhook(config, title, message) else: diff --git a/backend/app/services/print_scheduler.py b/backend/app/services/print_scheduler.py index cc838a5f4..edc60c7e4 100644 --- a/backend/app/services/print_scheduler.py +++ b/backend/app/services/print_scheduler.py @@ -4,7 +4,7 @@ import asyncio import json import logging import zipfile -from datetime import datetime, timedelta +from datetime import datetime from pathlib import Path import defusedxml.ElementTree as ET @@ -694,9 +694,11 @@ class PrintScheduler: if not state: return False - # Printer is idle if state is IDLE, FINISH, FAILED, or unknown - # FAILED means previous print failed, printer is ready for new print - return state.state in ("IDLE", "FINISH", "FAILED", "unknown") + # IDLE = ready for next print + # FINISH/FAILED = ready only if user confirmed plate is cleared + return state.state == "IDLE" or ( + state.state in ("FINISH", "FAILED") and printer_manager.is_plate_cleared(printer_id) + ) async def _get_smart_plug(self, db: AsyncSession, printer_id: int) -> SmartPlug | None: """Get the smart plug associated with a printer.""" @@ -860,27 +862,6 @@ class PrintScheduler: await self._power_off_if_needed(db, item) return - # Safety: Check if this archive was printed recently (within 4 hours) - # This prevents phantom reprints if a queue item got stuck in "pending" - # after its print already started due to a crash/restart - if archive.status == "completed" and archive.completed_at: - completed_at = ( - archive.completed_at.replace(tzinfo=None) if archive.completed_at.tzinfo else archive.completed_at - ) - time_since_completed = datetime.utcnow() - completed_at - if time_since_completed < timedelta(hours=4): - logger.warning( - f"Queue item {item.id}: Archive {item.archive_id} was already printed " - f"{time_since_completed.total_seconds() / 3600:.1f} hours ago, skipping to prevent duplicate" - ) - item.status = "skipped" - item.error_message = ( - f"Archive was already printed {time_since_completed.total_seconds() / 3600:.1f} hours ago" - ) - item.completed_at = datetime.utcnow() - await db.commit() - return - file_path = settings.base_dir / archive.file_path filename = archive.filename @@ -1034,6 +1015,9 @@ class PrintScheduler: item.status = "printing" item.started_at = datetime.utcnow() await db.commit() + + # Consume the plate-cleared flag now that we're starting a print + printer_manager.consume_plate_cleared(item.printer_id) logger.info("Queue item %s: Status set to 'printing', sending print command...", item.id) # Start the print with AMS mapping, plate_id and print options diff --git a/backend/app/services/printer_manager.py b/backend/app/services/printer_manager.py index 65b5e2998..9c3a5d00f 100644 --- a/backend/app/services/printer_manager.py +++ b/backend/app/services/printer_manager.py @@ -100,6 +100,8 @@ class PrinterManager: self._loop: asyncio.AbstractEventLoop | None = None # Track who started the current print (Issue #206) self._current_print_user: dict[int, dict] = {} # {printer_id: {"user_id": int, "username": str}} + # Track plate-cleared acknowledgments for queue flow + self._plate_cleared: set[int] = set() # printer_ids where user confirmed plate is cleared def get_printer(self, printer_id: int) -> PrinterInfo | None: """Get printer info by ID.""" @@ -117,6 +119,18 @@ class PrinterManager: """Clear the current print user when print completes (Issue #206).""" self._current_print_user.pop(printer_id, None) + def set_plate_cleared(self, printer_id: int): + """Mark that user has cleared the build plate for this printer.""" + self._plate_cleared.add(printer_id) + + def is_plate_cleared(self, printer_id: int) -> bool: + """Check if user has confirmed the plate is cleared.""" + return printer_id in self._plate_cleared + + def consume_plate_cleared(self, printer_id: int): + """Clear the plate-cleared flag (called when scheduler starts next print).""" + self._plate_cleared.discard(printer_id) + def set_event_loop(self, loop: asyncio.AbstractEventLoop): """Set the event loop for async callbacks.""" self._loop = loop @@ -209,18 +223,18 @@ class PrinterManager: await asyncio.sleep(1) return client.state.connected - def disconnect_printer(self, printer_id: int): + def disconnect_printer(self, printer_id: int, timeout: float = 0): """Disconnect from a printer.""" if printer_id in self._clients: - self._clients[printer_id].disconnect() + self._clients[printer_id].disconnect(timeout=timeout) del self._clients[printer_id] self._models.pop(printer_id, None) # Clean up model cache self._printer_info.pop(printer_id, None) # Clean up printer info cache - def disconnect_all(self): + def disconnect_all(self, timeout: float = 0): """Disconnect from all printers.""" for printer_id in list(self._clients.keys()): - self.disconnect_printer(printer_id) + self.disconnect_printer(printer_id, timeout=timeout) def get_status(self, printer_id: int) -> PrinterState | None: """Get the current status of a printer (checks for stale connections).""" @@ -519,7 +533,7 @@ def printer_state_to_dict(state: PrinterState, printer_id: int | None = None, mo trays.append( { - "id": tray.get("id", 0), + "id": int(tray.get("id", 0)), "tray_color": tray.get("tray_color"), "tray_type": tray.get("tray_type"), "tray_sub_brands": tray.get("tray_sub_brands"), @@ -556,7 +570,7 @@ def printer_state_to_dict(state: PrinterState, printer_id: int | None = None, mo ams_units.append( { - "id": ams_data.get("id", 0), + "id": int(ams_data.get("id", 0)), "humidity": humidity_value, "temp": ams_data.get("temp"), "is_ams_ht": is_ams_ht, diff --git a/backend/app/services/spool_tag_matcher.py b/backend/app/services/spool_tag_matcher.py new file mode 100644 index 000000000..e8ba3013c --- /dev/null +++ b/backend/app/services/spool_tag_matcher.py @@ -0,0 +1,310 @@ +"""RFID tag matching and auto-assignment for spool inventory.""" + +import logging + +from sqlalchemy import func, select +from sqlalchemy.ext.asyncio import AsyncSession +from sqlalchemy.orm import selectinload + +from backend.app.models.spool import Spool +from backend.app.models.spool_assignment import SpoolAssignment + +logger = logging.getLogger(__name__) + +# Zero-value constants for tag validation +ZERO_TAG_UID = "0000000000000000" +ZERO_TRAY_UUID = "00000000000000000000000000000000" + + +def is_valid_tag(tag_uid: str, tray_uuid: str) -> bool: + """Check if a tag/UUID pair contains a non-zero, non-empty value.""" + uid_valid = bool(tag_uid) and tag_uid != ZERO_TAG_UID and tag_uid != "0" * len(tag_uid) + uuid_valid = bool(tray_uuid) and tray_uuid != ZERO_TRAY_UUID and tray_uuid != "0" * len(tray_uuid) + return uid_valid or uuid_valid + + +def is_bambu_tag(tag_uid: str, tray_uuid: str, tray_info_idx: str) -> bool: + """Check if an AMS tray contains a Bambu Lab RFID spool (has valid UUID or slicer preset).""" + uuid_valid = bool(tray_uuid) and tray_uuid != ZERO_TRAY_UUID and tray_uuid != "0" * len(tray_uuid) + has_preset = bool(tray_info_idx) + return uuid_valid or (is_valid_tag(tag_uid, tray_uuid) and has_preset) + + +async def create_spool_from_tray(db: AsyncSession, tray_data: dict) -> Spool: + """Create a new Spool inventory entry from AMS tray MQTT data. + + Extracts material, subtype, color, temps, and tag info from the tray dict. + Looks up core_weight from the spool catalog if a Bambu Lab entry matches. + """ + from backend.app.models.color_catalog import ColorCatalogEntry + from backend.app.models.spool_catalog import SpoolCatalogEntry + + tray_type = tray_data.get("tray_type", "") # "PLA" + tray_sub_brands = tray_data.get("tray_sub_brands", "") # "PLA Basic" + tray_color = tray_data.get("tray_color", "FFFFFFFF") # RRGGBBAA + tray_id_name = tray_data.get("tray_id_name", "") # Color name e.g. "Jade White" + tag_uid = tray_data.get("tag_uid", "") + tray_uuid = tray_data.get("tray_uuid", "") + tray_info_idx = tray_data.get("tray_info_idx", "") + nozzle_min = tray_data.get("nozzle_temp_min", 0) + nozzle_max = tray_data.get("nozzle_temp_max", 0) + label_weight = int(tray_data.get("tray_weight", 1000)) + + # Parse material and subtype from tray_sub_brands ("PLA Basic" → material="PLA", subtype="Basic") + material = tray_type or "PLA" + subtype = None + if tray_sub_brands and " " in tray_sub_brands: + parts = tray_sub_brands.split(" ", 1) + if parts[0].upper() == material.upper(): + subtype = parts[1] + else: + # tray_sub_brands is the full material name (e.g. "PETG-HF") + material = tray_sub_brands + elif tray_sub_brands and tray_sub_brands.upper() != material.upper(): + material = tray_sub_brands + + # Resolve color name from tray_id_name code, hex catalog, or raw tray_id_name + from backend.app.core.bambu_colors import resolve_bambu_color_name + + rgba = tray_color if tray_color else None + color_name = None + + # 1. Try Bambu color code mapping (e.g. "A06-D0" → "Titan Gray") + if tray_id_name: + color_name = resolve_bambu_color_name(tray_id_name) + logger.info("Color resolve: tray_id_name=%r → resolved=%r", tray_id_name, color_name) + # If not a known code, use tray_id_name directly (it may be a readable name) + if not color_name and "-" not in tray_id_name: + color_name = tray_id_name + else: + logger.info("Color resolve: tray_id_name is empty, rgba=%r", rgba) + + # 2. Try color catalog lookup by hex color + if not color_name and rgba and len(rgba) >= 6: + hex_prefix = f"#{rgba[:6].upper()}" + cat_result = await db.execute( + select(ColorCatalogEntry) + .where(func.upper(ColorCatalogEntry.hex_color) == hex_prefix) + .where(func.upper(ColorCatalogEntry.manufacturer) == "BAMBU LAB") + .limit(1) + ) + entry = cat_result.scalar_one_or_none() + if entry: + color_name = entry.color_name + + # Look up core weight from spool catalog + core_weight = 250 # Default for Bambu Lab plastic spools + cat_result = await db.execute(select(SpoolCatalogEntry).where(SpoolCatalogEntry.name.ilike("Bambu Lab%")).limit(10)) + for entry in cat_result.scalars().all(): + # Pick the best match (prefer exact, fallback to first Bambu Lab entry) + core_weight = entry.weight + break + + # Resolve slicer filament name from builtin table + slicer_filament_name = None + if tray_info_idx: + try: + from backend.app.api.routes.cloud import _BUILTIN_FILAMENT_NAMES + + slicer_filament_name = _BUILTIN_FILAMENT_NAMES.get(tray_info_idx) + except Exception: + pass + # Fallback: use tray_sub_brands as the display name + if not slicer_filament_name and tray_sub_brands: + slicer_filament_name = tray_sub_brands + + # Calculate initial weight_used from AMS remain percentage + remain_raw = tray_data.get("remain") + try: + remain_pct = int(remain_raw) if remain_raw is not None else 100 + except (TypeError, ValueError): + remain_pct = 100 + # Clamp to valid range: negative means unknown, >100 is invalid + if remain_pct < 0 or remain_pct > 100: + remain_pct = 100 # Unknown → assume full + weight_used = round(label_weight * (100 - remain_pct) / 100.0, 1) + + spool = Spool( + material=material, + subtype=subtype, + color_name=color_name, + rgba=rgba, + brand="Bambu Lab", + label_weight=label_weight, + core_weight=core_weight, + weight_used=weight_used, + slicer_filament=tray_info_idx or None, + slicer_filament_name=slicer_filament_name, + nozzle_temp_min=int(nozzle_min) if nozzle_min else None, + nozzle_temp_max=int(nozzle_max) if nozzle_max else None, + tag_uid=tag_uid if tag_uid and tag_uid != ZERO_TAG_UID else None, + tray_uuid=tray_uuid if tray_uuid and tray_uuid != ZERO_TRAY_UUID else None, + data_origin="rfid_auto", + tag_type="bambulab", + ) + db.add(spool) + await db.flush() + + logger.info( + "Auto-created spool %d from AMS tray data: %s %s %s (tag=%s uuid=%s)", + spool.id, + material, + subtype or "", + color_name or "", + tag_uid, + tray_uuid, + ) + return spool + + +async def get_spool_by_tag(db: AsyncSession, tag_uid: str, tray_uuid: str) -> Spool | None: + """Look up an active spool by RFID tag UID or Bambu Lab tray UUID. + + Prefers tray_uuid match over tag_uid (more reliable). + """ + # Try tray_uuid first (Bambu Lab spools — more reliable) + if tray_uuid and tray_uuid != ZERO_TRAY_UUID and tray_uuid != "0" * len(tray_uuid): + result = await db.execute( + select(Spool) + .options(selectinload(Spool.k_profiles)) + .where(Spool.tray_uuid == tray_uuid, Spool.archived_at.is_(None)) + .limit(1) + ) + spool = result.scalar_one_or_none() + if spool: + return spool + + # Fall back to tag_uid + if tag_uid and tag_uid != ZERO_TAG_UID and tag_uid != "0" * len(tag_uid): + result = await db.execute( + select(Spool) + .options(selectinload(Spool.k_profiles)) + .where(Spool.tag_uid == tag_uid, Spool.archived_at.is_(None)) + .limit(1) + ) + spool = result.scalar_one_or_none() + if spool: + return spool + + return None + + +async def auto_assign_spool( + printer_id: int, + ams_id: int, + tray_id: int, + spool: Spool, + printer_manager, + db: AsyncSession, +) -> SpoolAssignment: + """Create a SpoolAssignment and auto-configure the AMS slot via MQTT. + + Reuses the same MQTT configuration logic as the manual assign endpoint. + """ + from backend.app.api.routes.inventory import MATERIAL_TEMPS + + # Get current tray state for fingerprint + fingerprint_color = None + fingerprint_type = None + state = printer_manager.get_status(printer_id) + if state and state.raw_data: + from backend.app.api.routes.inventory import _find_tray_in_ams_data + + ams = state.raw_data.get("ams", []) + if isinstance(ams, dict): + ams = ams.get("ams", []) + tray = _find_tray_in_ams_data( + ams, + ams_id, + tray_id, + ) + if tray: + fingerprint_color = tray.get("tray_color", "") + fingerprint_type = tray.get("tray_type", "") + + # Upsert: remove old assignment for this slot + existing = await db.execute( + select(SpoolAssignment).where( + SpoolAssignment.printer_id == printer_id, + SpoolAssignment.ams_id == ams_id, + SpoolAssignment.tray_id == tray_id, + ) + ) + old = existing.scalar_one_or_none() + if old: + await db.delete(old) + await db.flush() + + assignment = SpoolAssignment( + spool_id=spool.id, + printer_id=printer_id, + ams_id=ams_id, + tray_id=tray_id, + fingerprint_color=fingerprint_color, + fingerprint_type=fingerprint_type, + ) + db.add(assignment) + await db.flush() + + # Auto-configure AMS slot via MQTT + try: + client = printer_manager.get_client(printer_id) + if client: + tray_type = spool.material + tray_sub_brands = f"{spool.material} {spool.subtype}" if spool.subtype else spool.material + tray_color = spool.rgba or "FFFFFFFF" + tray_info_idx = spool.slicer_filament or "" + setting_id = "" + + temp_min, temp_max = MATERIAL_TEMPS.get(spool.material.upper(), (200, 240)) + if spool.nozzle_temp_min is not None: + temp_min = spool.nozzle_temp_min + if spool.nozzle_temp_max is not None: + temp_max = spool.nozzle_temp_max + + client.ams_set_filament_setting( + ams_id=ams_id, + tray_id=tray_id, + tray_info_idx=tray_info_idx, + tray_type=tray_type, + tray_sub_brands=tray_sub_brands, + tray_color=tray_color, + nozzle_temp_min=temp_min, + nozzle_temp_max=temp_max, + setting_id=setting_id, + ) + + # Apply K-profile if available + nozzle_diameter = "0.4" + if state and state.nozzles: + nd = state.nozzles[0].nozzle_diameter + if nd: + nozzle_diameter = nd + + matching_kp = None + for kp in spool.k_profiles: + if kp.printer_id == printer_id and kp.nozzle_diameter == nozzle_diameter: + matching_kp = kp + break + + if matching_kp and matching_kp.cali_idx is not None: + client.extrusion_cali_sel( + ams_id=ams_id, + tray_id=tray_id, + cali_idx=matching_kp.cali_idx, + filament_id=tray_info_idx, + nozzle_diameter=nozzle_diameter, + setting_id=matching_kp.setting_id, + ) + + logger.info( + "Auto-configured AMS slot ams=%d tray=%d for spool %d on printer %d (RFID match)", + ams_id, + tray_id, + spool.id, + printer_id, + ) + except Exception as e: + logger.warning("MQTT auto-configure failed for spool %d (RFID match): %s", spool.id, e) + + return assignment diff --git a/backend/app/services/spoolman.py b/backend/app/services/spoolman.py index 2c9acb4c5..b7f54a8bf 100644 --- a/backend/app/services/spoolman.py +++ b/backend/app/services/spoolman.py @@ -468,6 +468,26 @@ class SpoolmanClient: return spool return None + def _find_spool_by_location(self, location: str, cached_spools: list[dict] | None) -> dict | None: + """Find a spool by exact location match. + + Used as fallback when RFID tag data is unavailable (e.g., newer firmware + that doesn't expose tray_uuid/tag_uid via MQTT). + + Args: + location: Exact location string (e.g., "H2D-1 - AMS A1") + cached_spools: Pre-fetched list of spools to search + + Returns: + Spool dictionary or None if not found. + """ + if not cached_spools: + return None + for spool in cached_spools: + if spool.get("location") == location: + return spool + return None + async def find_spools_by_location_prefix( self, location_prefix: str, cached_spools: list[dict] | None = None ) -> list[dict]: @@ -494,17 +514,21 @@ class SpoolmanClient: printer_name: str, current_tray_uuids: set[str], cached_spools: list[dict] | None = None, + synced_spool_ids: set[int] | None = None, ) -> int: """Clear location for spools that are no longer in the AMS. When a spool is removed from the AMS, its location should be cleared in Spoolman. This method finds all spools with locations for this printer - and clears the location for any that are not in the current_tray_uuids set. + and clears the location for any that are not in the current_tray_uuids set + and were not synced in this cycle (synced_spool_ids). Args: printer_name: The printer name used as location prefix current_tray_uuids: Set of tray_uuids currently in the AMS cached_spools: Optional pre-fetched list of spools to search (avoids API call) + synced_spool_ids: Set of spool IDs that were synced in this cycle + (protects location-matched spools when RFID data is unavailable) Returns: Number of spools whose location was cleared. @@ -514,6 +538,12 @@ class SpoolmanClient: cleared_count = 0 for spool in spools_at_printer: + spool_id = spool.get("id") + + # Skip spools that were just synced (matched by location or tag) + if synced_spool_ids and spool_id in synced_spool_ids: + continue + # Get the tray_uuid (stored as "tag" in extra field) extra = spool.get("extra", {}) or {} stored_tag = extra.get("tag", "") @@ -526,10 +556,10 @@ class SpoolmanClient: # If this spool's UUID is not in the current AMS, clear its location if spool_uuid not in current_tray_uuids: logger.info( - f"Clearing location for spool {spool['id']} " + f"Clearing location for spool {spool_id} " f"(was: {spool.get('location')}, uuid: {spool_uuid[:16] if spool_uuid else 'none'}...)" ) - result = await self.update_spool(spool_id=spool["id"], clear_location=True) + result = await self.update_spool(spool_id=spool_id, clear_location=True) if result: cleared_count += 1 @@ -628,8 +658,8 @@ class SpoolmanClient: # Get tray_info_idx (Bambu filament preset ID like "GFA00") tray_info_idx = tray_data.get("tray_info_idx", "") or "" - # Get remaining percentage, ensure non-negative - remain = max(0, int(tray_data.get("remain", 0))) + # Get remaining percentage (-1 means unknown/not read by AMS) + remain = int(tray_data.get("remain", -1)) return AMSTray( ams_id=ams_id, @@ -730,6 +760,7 @@ class SpoolmanClient: printer_name: str, disable_weight_sync: bool = False, cached_spools: list[dict] | None = None, + inventory_remaining: float | None = None, ) -> dict | None: """Sync a single AMS tray to Spoolman. @@ -747,6 +778,8 @@ class SpoolmanClient: cached_spools: Optional pre-fetched list of spools to search (avoids API calls). When provided, this cache is passed to find_spool_by_tag to avoid redundant API calls during batch sync operations. + inventory_remaining: Optional fallback remaining weight (grams) from the built-in + inventory when AMS MQTT data has invalid remain/tray_weight values. Returns: Synced spool dictionary or None if skipped or failed. @@ -770,53 +803,82 @@ class SpoolmanClient: return None # Determine which identifier to use for Spoolman (prefer tray_uuid, fallback to tag_uid) - spool_tag = ( - tray.tray_uuid if tray.tray_uuid and tray.tray_uuid != "00000000000000000000000000000000" else tray.tag_uid - ) - - # If no unique identifier available, we can't sync even if it's a Bambu Lab spool - if not spool_tag: - logger.warning( - f"Bambu Lab spool detected but no unique identifier for Spoolman: " - f"{printer_name} AMS {tray.ams_id} tray {tray.tray_id} (tray_info_idx={tray.tray_info_idx})" - ) - return None + # Zero-filled values mean the AMS hasn't read the RFID tag — treat as no tag + zero_uuid = "00000000000000000000000000000000" + zero_tag = "0000000000000000" + spool_tag = None + if tray.tray_uuid and tray.tray_uuid != zero_uuid: + spool_tag = tray.tray_uuid + elif tray.tag_uid and tray.tag_uid != zero_tag: + spool_tag = tray.tag_uid # Calculate remaining weight - remaining = self.calculate_remaining_weight(tray.remain, tray.tray_weight) + # Primary: AMS MQTT data (remain percentage + tray_weight) + # Fallback: Built-in inventory tracked weight (when firmware sends invalid remain/tray_weight) + if tray.remain >= 0 and tray.tray_weight > 0: + remaining = self.calculate_remaining_weight(tray.remain, tray.tray_weight) + elif inventory_remaining is not None: + remaining = inventory_remaining + logger.debug( + "Using inventory weight fallback for %s AMS %s tray %s: %.1fg", + printer_name, + tray.ams_id, + tray.tray_id, + remaining, + ) + else: + remaining = None location = f"{printer_name} - {self.convert_ams_slot_to_location(tray.ams_id, tray.tray_id)}" - # Find existing spool by tag (tray_uuid or tag_uid, stored as "tag" in Spoolman) - existing = await self.find_spool_by_tag(spool_tag, cached_spools=cached_spools) + if spool_tag: + # Primary path: match by RFID tag + existing = await self.find_spool_by_tag(spool_tag, cached_spools=cached_spools) + if existing: + logger.info("Updating existing spool %s for tag %s...", existing["id"], spool_tag[:16]) + return await self.update_spool( + spool_id=existing["id"], + remaining_weight=None if disable_weight_sync else remaining, + location=location, + ) + + # Spool not found by tag - auto-create it + logger.info("Creating new spool in Spoolman for %s (tag: %s...)", tray.tray_sub_brands, spool_tag[:16]) + filament = await self._find_or_create_filament(tray) + if not filament: + logger.error("Failed to find or create filament for %s", tray.tray_sub_brands) + return None + + import json + + return await self.create_spool( + filament_id=filament["id"], + remaining_weight=remaining, + location=location, + comment="Created by Bambuddy", + extra={"tag": json.dumps(spool_tag)}, + ) + + # Fallback path: no RFID tag available (newer firmware may not expose UUIDs) + # Only update existing spools matched by location — never create new ones without a tag + # to avoid duplicates when old spools exist from previous RFID-based syncs + existing = self._find_spool_by_location(location, cached_spools) if existing: - # Update existing spool - logger.info("Updating existing spool %s for tag %s...", existing["id"], spool_tag[:16]) + logger.info( + "Updating spool %s by location match '%s' (no RFID tag available)", + existing["id"], + location, + ) return await self.update_spool( spool_id=existing["id"], remaining_weight=None if disable_weight_sync else remaining, location=location, ) - # Spool not found - auto-create it - logger.info("Creating new spool in Spoolman for %s (tag: %s...)", tray.tray_sub_brands, spool_tag[:16]) - - # First find or create the filament type - filament = await self._find_or_create_filament(tray) - if not filament: - logger.error("Failed to find or create filament for %s", tray.tray_sub_brands) - return None - - # Create the spool with identifier stored as "tag" in extra field - # Note: Spoolman extra field values must be valid JSON, so we encode the string - import json - - return await self.create_spool( - filament_id=filament["id"], - remaining_weight=remaining, - location=location, - comment="Created by Bambuddy", - extra={"tag": json.dumps(spool_tag)}, + logger.info( + "No existing spool found at '%s' — skipping (no RFID tag to create with)", + location, ) + return None async def _find_or_create_filament(self, tray: AMSTray) -> dict | None: """Find existing filament or create new one. diff --git a/backend/app/services/usage_tracker.py b/backend/app/services/usage_tracker.py new file mode 100644 index 000000000..f5e36a1e7 --- /dev/null +++ b/backend/app/services/usage_tracker.py @@ -0,0 +1,328 @@ +"""Automatic filament consumption tracking. + +Captures AMS tray remain% at print start, then computes consumption +deltas at print complete to update spool weight_used and last_used. + +For non-BL spools (no RFID, AMS reports remain=-1), falls back to +per-filament usage estimates from the archived 3MF file. +""" + +import logging +from dataclasses import dataclass, field +from datetime import datetime, timezone + +from sqlalchemy import select +from sqlalchemy.ext.asyncio import AsyncSession + +from backend.app.models.spool import Spool +from backend.app.models.spool_assignment import SpoolAssignment +from backend.app.models.spool_usage_history import SpoolUsageHistory + +logger = logging.getLogger(__name__) + + +@dataclass +class PrintSession: + printer_id: int + print_name: str + started_at: datetime + tray_remain_start: dict[tuple[int, int], int] = field(default_factory=dict) + + +# Module-level storage, keyed by printer_id +_active_sessions: dict[int, PrintSession] = {} + + +async def on_print_start(printer_id: int, data: dict, printer_manager) -> None: + """Capture AMS tray remain% at print start.""" + state = printer_manager.get_status(printer_id) + if not state or not state.raw_data: + logger.debug("[UsageTracker] No state for printer %d, skipping", printer_id) + return + + ams_raw = state.raw_data.get("ams", []) + ams_data = ams_raw.get("ams", []) if isinstance(ams_raw, dict) else ams_raw if isinstance(ams_raw, list) else [] + if not ams_data: + logger.debug("[UsageTracker] No AMS data for printer %d, skipping", printer_id) + return + + tray_remain_start: dict[tuple[int, int], int] = {} + for ams_unit in ams_data: + ams_id = int(ams_unit.get("id", 0)) + for tray in ams_unit.get("tray", []): + tray_id = int(tray.get("id", 0)) + remain = tray.get("remain", -1) + if isinstance(remain, int) and 0 <= remain <= 100: + tray_remain_start[(ams_id, tray_id)] = remain + + print_name = data.get("subtask_name", "") or data.get("filename", "unknown") + + # Always create session (even without valid remain data) so print_name + # is available at completion for 3MF-based tracking + session = PrintSession( + printer_id=printer_id, + print_name=print_name, + started_at=datetime.now(timezone.utc), + tray_remain_start=tray_remain_start, + ) + _active_sessions[printer_id] = session + + if tray_remain_start: + logger.info( + "[UsageTracker] Captured start remain%% for printer %d (%d trays): %s", + printer_id, + len(tray_remain_start), + {f"{k[0]}-{k[1]}": v for k, v in tray_remain_start.items()}, + ) + else: + logger.debug("[UsageTracker] No valid remain%% for printer %d, 3MF fallback available", printer_id) + + +async def on_print_complete( + printer_id: int, + data: dict, + printer_manager, + db: AsyncSession, + archive_id: int | None = None, +) -> list[dict]: + """Compute consumption deltas and update spool weight_used/last_used. + + Uses two tracking strategies: + 1. AMS remain% delta — for BL spools with valid RFID remain data + 2. 3MF per-filament estimates — for non-BL spools without remain data + + Returns a list of dicts describing what was logged (for WebSocket broadcast). + """ + session = _active_sessions.pop(printer_id, None) + status = data.get("status", "completed") + results = [] + handled_trays: set[tuple[int, int]] = set() + + # --- Path 1: AMS remain% delta (for spools with valid RFID remain data) --- + if session and session.tray_remain_start: + state = printer_manager.get_status(printer_id) + if state and state.raw_data: + ams_raw = state.raw_data.get("ams", []) + ams_data = ( + ams_raw.get("ams", []) if isinstance(ams_raw, dict) else ams_raw if isinstance(ams_raw, list) else [] + ) + + for ams_unit in ams_data: + ams_id = int(ams_unit.get("id", 0)) + for tray in ams_unit.get("tray", []): + tray_id = int(tray.get("id", 0)) + key = (ams_id, tray_id) + + if key not in session.tray_remain_start: + continue + + current_remain = tray.get("remain", -1) + if not isinstance(current_remain, int) or current_remain < 0 or current_remain > 100: + continue + + start_remain = session.tray_remain_start[key] + delta_pct = start_remain - current_remain + + if delta_pct <= 0: + continue # No consumption or tray was refilled + + # Look up SpoolAssignment for this slot + result = await db.execute( + select(SpoolAssignment).where( + SpoolAssignment.printer_id == printer_id, + SpoolAssignment.ams_id == ams_id, + SpoolAssignment.tray_id == tray_id, + ) + ) + assignment = result.scalar_one_or_none() + if not assignment: + continue + + # Load spool + spool_result = await db.execute(select(Spool).where(Spool.id == assignment.spool_id)) + spool = spool_result.scalar_one_or_none() + if not spool: + continue + + # Compute weight consumed + weight_grams = (delta_pct / 100.0) * spool.label_weight + + # Update spool + spool.weight_used = (spool.weight_used or 0) + weight_grams + spool.last_used = datetime.now(timezone.utc) + + # Insert usage history record + history = SpoolUsageHistory( + spool_id=spool.id, + printer_id=printer_id, + print_name=session.print_name, + weight_used=round(weight_grams, 1), + percent_used=delta_pct, + status=status, + ) + db.add(history) + + handled_trays.add(key) + results.append( + { + "spool_id": spool.id, + "weight_used": round(weight_grams, 1), + "percent_used": delta_pct, + "ams_id": ams_id, + "tray_id": tray_id, + } + ) + + logger.info( + "[UsageTracker] Spool %d consumed %.1fg (%d%%) on printer %d AMS%d-T%d (%s)", + spool.id, + weight_grams, + delta_pct, + printer_id, + ams_id, + tray_id, + status, + ) + + # --- Path 2: 3MF per-filament estimates (for non-BL spools without remain data) --- + if archive_id: + print_name = ( + (session.print_name if session else None) or data.get("subtask_name", "") or data.get("filename", "unknown") + ) + threemf_results = await _track_from_3mf( + printer_id, archive_id, status, print_name, handled_trays, printer_manager, db + ) + results.extend(threemf_results) + + if results: + await db.commit() + + return results + + +async def _track_from_3mf( + printer_id: int, + archive_id: int, + status: str, + print_name: str, + handled_trays: set[tuple[int, int]], + printer_manager, + db: AsyncSession, +) -> list[dict]: + """Track usage from 3MF per-filament data for non-BL spools. + + Falls back to slicer-estimated filament weight when AMS remain% is + unavailable (non-RFID spools). For partial prints (failed/aborted), + scales the estimate by print progress. + """ + from backend.app.core.config import settings as app_settings + from backend.app.models.archive import PrintArchive + from backend.app.utils.threemf_tools import extract_filament_usage_from_3mf + + result = await db.execute(select(PrintArchive).where(PrintArchive.id == archive_id)) + archive = result.scalar_one_or_none() + if not archive or not archive.file_path: + return [] + + file_path = app_settings.base_dir / archive.file_path + if not file_path.exists(): + return [] + + filament_usage = extract_filament_usage_from_3mf(file_path) + if not filament_usage: + return [] + + # Scale factor for partial prints (failed/aborted) + if status == "completed": + scale = 1.0 + else: + state = printer_manager.get_status(printer_id) + progress = state.progress if state else 0 + scale = max(0.0, min(progress / 100.0, 1.0)) + + results = [] + + for usage in filament_usage: + slot_id = usage.get("slot_id", 0) + used_g = usage.get("used_g", 0) + if used_g <= 0: + continue + + # Map 3MF slot_id (1-based) to (ams_id, tray_id) + global_tray_id = slot_id - 1 + if global_tray_id >= 128: + ams_id = global_tray_id + tray_id = 0 + else: + ams_id = global_tray_id // 4 + tray_id = global_tray_id % 4 + + key = (ams_id, tray_id) + if key in handled_trays: + continue # Already tracked via AMS remain% delta + + # Find spool assignment for this tray + assign_result = await db.execute( + select(SpoolAssignment).where( + SpoolAssignment.printer_id == printer_id, + SpoolAssignment.ams_id == ams_id, + SpoolAssignment.tray_id == tray_id, + ) + ) + assignment = assign_result.scalar_one_or_none() + if not assignment: + continue + + # Load spool + spool_result = await db.execute(select(Spool).where(Spool.id == assignment.spool_id)) + spool = spool_result.scalar_one_or_none() + if not spool: + continue + + # Only use 3MF tracking for non-BL spools (BL spools use AMS remain%) + if spool.tag_uid or spool.tray_uuid: + continue + + weight_grams = used_g * scale + if weight_grams <= 0: + continue + + # Update spool + spool.weight_used = (spool.weight_used or 0) + weight_grams + spool.last_used = datetime.now(timezone.utc) + + percent = round(weight_grams / (spool.label_weight or 1000) * 100) + + # Insert usage history record + history = SpoolUsageHistory( + spool_id=spool.id, + printer_id=printer_id, + print_name=print_name, + weight_used=round(weight_grams, 1), + percent_used=percent, + status=status, + ) + db.add(history) + + results.append( + { + "spool_id": spool.id, + "weight_used": round(weight_grams, 1), + "percent_used": percent, + "ams_id": ams_id, + "tray_id": tray_id, + } + ) + + logger.info( + "[UsageTracker] Spool %d consumed %.1fg (3MF estimate%s) on printer %d AMS%d-T%d (%s)", + spool.id, + weight_grams, + f" scaled to {scale:.0%}" if scale < 1 else "", + printer_id, + ams_id, + tray_id, + status, + ) + + return results diff --git a/backend/tests/integration/test_camera_api.py b/backend/tests/integration/test_camera_api.py index 773b06dca..e963b1ffb 100644 --- a/backend/tests/integration/test_camera_api.py +++ b/backend/tests/integration/test_camera_api.py @@ -192,6 +192,49 @@ class TestCameraAPI: assert response.status_code == 503 assert "Failed to capture" in response.json()["detail"] + @pytest.mark.asyncio + @pytest.mark.integration + async def test_camera_snapshot_external_camera_success(self, async_client: AsyncClient, printer_factory): + """Verify snapshot uses external camera when configured.""" + printer = await printer_factory( + external_camera_enabled=True, + external_camera_url="http://192.168.1.50/mjpeg", + external_camera_type="mjpeg", + ) + + fake_jpeg = b"\xff\xd8\xff\xe0\x00\x10JFIF\x00\x01\x01\x00\x00\x01\x00\x01\x00\x00" + + with patch( + "backend.app.services.external_camera.capture_frame", + new_callable=AsyncMock, + return_value=fake_jpeg, + ): + response = await async_client.get(f"/api/v1/printers/{printer.id}/camera/snapshot") + + assert response.status_code == 200 + assert response.headers["content-type"] == "image/jpeg" + assert response.content == fake_jpeg + + @pytest.mark.asyncio + @pytest.mark.integration + async def test_camera_snapshot_external_camera_failure(self, async_client: AsyncClient, printer_factory): + """Verify 503 when external camera capture fails.""" + printer = await printer_factory( + external_camera_enabled=True, + external_camera_url="http://192.168.1.50/mjpeg", + external_camera_type="mjpeg", + ) + + with patch( + "backend.app.services.external_camera.capture_frame", + new_callable=AsyncMock, + return_value=None, + ): + response = await async_client.get(f"/api/v1/printers/{printer.id}/camera/snapshot") + + assert response.status_code == 503 + assert "external camera" in response.json()["detail"].lower() + # ======================================================================== # Camera Stream Endpoint # ======================================================================== diff --git a/backend/tests/unit/services/test_usage_tracker.py b/backend/tests/unit/services/test_usage_tracker.py new file mode 100644 index 000000000..64fb0296e --- /dev/null +++ b/backend/tests/unit/services/test_usage_tracker.py @@ -0,0 +1,388 @@ +"""Unit tests for the filament usage tracker. + +Tests both AMS remain% delta tracking (Path 1) and 3MF per-filament +fallback tracking (Path 2) for non-BL spools. +""" + +from datetime import datetime, timezone +from unittest.mock import AsyncMock, MagicMock, patch + +import pytest + +from backend.app.services.usage_tracker import ( + PrintSession, + _active_sessions, + _track_from_3mf, + on_print_complete, + on_print_start, +) + + +def _make_spool(*, id=1, label_weight=1000, weight_used=0, tag_uid=None, tray_uuid=None): + """Create a mock Spool object.""" + spool = MagicMock() + spool.id = id + spool.label_weight = label_weight + spool.weight_used = weight_used + spool.tag_uid = tag_uid + spool.tray_uuid = tray_uuid + spool.last_used = None + return spool + + +def _make_assignment(*, spool_id=1, printer_id=1, ams_id=0, tray_id=0): + """Create a mock SpoolAssignment object.""" + assignment = MagicMock() + assignment.spool_id = spool_id + assignment.printer_id = printer_id + assignment.ams_id = ams_id + assignment.tray_id = tray_id + return assignment + + +def _make_printer_state(ams_data, progress=0): + """Create a mock printer state with AMS data.""" + state = MagicMock() + state.raw_data = {"ams": ams_data} + state.progress = progress + return state + + +def _make_printer_manager(state=None): + """Create a mock printer manager.""" + pm = MagicMock() + pm.get_status.return_value = state + return pm + + +class TestOnPrintStart: + """Tests for on_print_start — capturing AMS remain%.""" + + @pytest.fixture(autouse=True) + def _clear_sessions(self): + _active_sessions.clear() + yield + _active_sessions.clear() + + @pytest.mark.asyncio + async def test_creates_session_with_valid_remain(self): + """Session created with remain% data for trays reporting 0-100.""" + ams_data = [{"id": 0, "tray": [{"id": 0, "remain": 80}]}] + pm = _make_printer_manager(_make_printer_state(ams_data)) + + await on_print_start(1, {"subtask_name": "test_print"}, pm) + + assert 1 in _active_sessions + session = _active_sessions[1] + assert session.print_name == "test_print" + assert session.tray_remain_start == {(0, 0): 80} + + @pytest.mark.asyncio + async def test_creates_session_even_without_valid_remain(self): + """Session still created when remain=-1 (for 3MF fallback path).""" + ams_data = [{"id": 0, "tray": [{"id": 0, "remain": -1}]}] + pm = _make_printer_manager(_make_printer_state(ams_data)) + + await on_print_start(1, {"subtask_name": "test_print"}, pm) + + assert 1 in _active_sessions + session = _active_sessions[1] + assert session.tray_remain_start == {} # Empty, no valid remain + + @pytest.mark.asyncio + async def test_skips_without_ams_data(self): + """No session created when no AMS data available.""" + state = MagicMock() + state.raw_data = {"ams": []} + pm = _make_printer_manager(state) + + await on_print_start(1, {"subtask_name": "test"}, pm) + + assert 1 not in _active_sessions + + +class TestOnPrintCompleteAMSDelta: + """Tests for Path 1: AMS remain% delta tracking.""" + + @pytest.fixture(autouse=True) + def _clear_sessions(self): + _active_sessions.clear() + yield + _active_sessions.clear() + + @pytest.mark.asyncio + async def test_computes_delta_and_updates_spool(self): + """Spool weight_used updated by remain% delta * label_weight.""" + # Set up session with start remain = 80% + _active_sessions[1] = PrintSession( + printer_id=1, + print_name="test", + started_at=datetime.now(timezone.utc), + tray_remain_start={(0, 0): 80}, + ) + + # Current remain = 70% → 10% consumed → 100g on 1000g spool + ams_data = [{"id": 0, "tray": [{"id": 0, "remain": 70}]}] + pm = _make_printer_manager(_make_printer_state(ams_data)) + + spool = _make_spool(label_weight=1000, weight_used=50) + assignment = _make_assignment() + + db = AsyncMock() + # First execute → assignment, second → spool + db.execute = AsyncMock( + side_effect=[ + MagicMock(scalar_one_or_none=MagicMock(return_value=assignment)), + MagicMock(scalar_one_or_none=MagicMock(return_value=spool)), + ] + ) + + results = await on_print_complete(1, {"status": "completed"}, pm, db) + + assert len(results) == 1 + assert results[0]["weight_used"] == 100.0 + assert results[0]["percent_used"] == 10 + # weight_used should be old (50) + delta (100) + assert spool.weight_used == 150.0 + db.commit.assert_called_once() + + @pytest.mark.asyncio + async def test_skips_negative_delta(self): + """No tracking when remain increased (spool refilled).""" + _active_sessions[1] = PrintSession( + printer_id=1, + print_name="test", + started_at=datetime.now(timezone.utc), + tray_remain_start={(0, 0): 50}, + ) + + # Remain went UP: 50 → 80 (refilled) + ams_data = [{"id": 0, "tray": [{"id": 0, "remain": 80}]}] + pm = _make_printer_manager(_make_printer_state(ams_data)) + db = AsyncMock() + + results = await on_print_complete(1, {"status": "completed"}, pm, db) + + assert results == [] + db.commit.assert_not_called() + + @pytest.mark.asyncio + async def test_no_session_falls_through_to_3mf(self): + """When no session exists, AMS delta path skipped (3MF may still run).""" + pm = _make_printer_manager() + db = AsyncMock() + + results = await on_print_complete(1, {"status": "completed"}, pm, db) + + assert results == [] + + +class TestTrackFrom3MF: + """Tests for Path 2: 3MF per-filament fallback tracking.""" + + @pytest.mark.asyncio + async def test_updates_non_bl_spool_from_3mf(self): + """Non-BL spool gets weight_used from 3MF used_g for completed print.""" + spool = _make_spool(id=5, label_weight=1000, weight_used=100) + assignment = _make_assignment(spool_id=5) + archive = MagicMock() + archive.file_path = "archives/test.3mf" + + db = AsyncMock() + # First execute → archive, second → assignment, third → spool + db.execute = AsyncMock( + side_effect=[ + MagicMock(scalar_one_or_none=MagicMock(return_value=archive)), + MagicMock(scalar_one_or_none=MagicMock(return_value=assignment)), + MagicMock(scalar_one_or_none=MagicMock(return_value=spool)), + ] + ) + + pm = _make_printer_manager() + filament_usage = [{"slot_id": 1, "used_g": 25.5, "type": "PLA", "color": "#FF0000"}] + + with ( + patch("backend.app.core.config.settings") as mock_settings, + patch("backend.app.utils.threemf_tools.extract_filament_usage_from_3mf", return_value=filament_usage), + ): + mock_path = MagicMock() + mock_path.exists.return_value = True + mock_settings.base_dir.__truediv__ = MagicMock(return_value=mock_path) + + results = await _track_from_3mf( + printer_id=1, + archive_id=10, + status="completed", + print_name="test_print", + handled_trays=set(), + printer_manager=pm, + db=db, + ) + + assert len(results) == 1 + assert results[0]["spool_id"] == 5 + assert results[0]["weight_used"] == 25.5 + # weight_used = old (100) + 3MF (25.5) + assert spool.weight_used == 125.5 + + @pytest.mark.asyncio + async def test_scales_by_progress_for_failed_print(self): + """Failed print scales 3MF estimate by progress percentage.""" + spool = _make_spool(id=1, label_weight=1000, weight_used=0) + assignment = _make_assignment() + archive = MagicMock() + archive.file_path = "archives/test.3mf" + + db = AsyncMock() + db.execute = AsyncMock( + side_effect=[ + MagicMock(scalar_one_or_none=MagicMock(return_value=archive)), + MagicMock(scalar_one_or_none=MagicMock(return_value=assignment)), + MagicMock(scalar_one_or_none=MagicMock(return_value=spool)), + ] + ) + + # Print failed at 50% progress → 50g consumed from 100g estimate + pm = _make_printer_manager(_make_printer_state([], progress=50)) + filament_usage = [{"slot_id": 1, "used_g": 100.0, "type": "PLA", "color": ""}] + + with ( + patch("backend.app.core.config.settings") as mock_settings, + patch("backend.app.utils.threemf_tools.extract_filament_usage_from_3mf", return_value=filament_usage), + ): + mock_path = MagicMock() + mock_path.exists.return_value = True + mock_settings.base_dir.__truediv__ = MagicMock(return_value=mock_path) + + results = await _track_from_3mf( + printer_id=1, + archive_id=10, + status="failed", + print_name="test", + handled_trays=set(), + printer_manager=pm, + db=db, + ) + + assert len(results) == 1 + assert results[0]["weight_used"] == 50.0 + assert spool.weight_used == 50.0 + + @pytest.mark.asyncio + async def test_skips_bl_spools(self): + """BL spools (with tag_uid) are NOT tracked via 3MF — they use AMS remain%.""" + spool = _make_spool(tag_uid="ABCD1234", tray_uuid="A1B2C3D4E5F6A1B2C3D4E5F6A1B2C3D4") + assignment = _make_assignment() + archive = MagicMock() + archive.file_path = "archives/test.3mf" + + db = AsyncMock() + db.execute = AsyncMock( + side_effect=[ + MagicMock(scalar_one_or_none=MagicMock(return_value=archive)), + MagicMock(scalar_one_or_none=MagicMock(return_value=assignment)), + MagicMock(scalar_one_or_none=MagicMock(return_value=spool)), + ] + ) + + pm = _make_printer_manager() + filament_usage = [{"slot_id": 1, "used_g": 50.0, "type": "PLA", "color": ""}] + + with ( + patch("backend.app.core.config.settings") as mock_settings, + patch("backend.app.utils.threemf_tools.extract_filament_usage_from_3mf", return_value=filament_usage), + ): + mock_path = MagicMock() + mock_path.exists.return_value = True + mock_settings.base_dir.__truediv__ = MagicMock(return_value=mock_path) + + results = await _track_from_3mf( + printer_id=1, + archive_id=10, + status="completed", + print_name="test", + handled_trays=set(), + printer_manager=pm, + db=db, + ) + + assert results == [] + + @pytest.mark.asyncio + async def test_skips_already_handled_trays(self): + """Trays handled by AMS remain% delta are not double-tracked via 3MF.""" + archive = MagicMock() + archive.file_path = "archives/test.3mf" + + db = AsyncMock() + db.execute = AsyncMock( + side_effect=[ + MagicMock(scalar_one_or_none=MagicMock(return_value=archive)), + ] + ) + + pm = _make_printer_manager() + filament_usage = [{"slot_id": 1, "used_g": 50.0, "type": "PLA", "color": ""}] + + with ( + patch("backend.app.core.config.settings") as mock_settings, + patch("backend.app.utils.threemf_tools.extract_filament_usage_from_3mf", return_value=filament_usage), + ): + mock_path = MagicMock() + mock_path.exists.return_value = True + mock_settings.base_dir.__truediv__ = MagicMock(return_value=mock_path) + + results = await _track_from_3mf( + printer_id=1, + archive_id=10, + status="completed", + print_name="test", + handled_trays={(0, 0)}, # slot_id=1 → ams_id=0, tray_id=0 + printer_manager=pm, + db=db, + ) + + assert results == [] + + @pytest.mark.asyncio + async def test_slot_to_tray_mapping(self): + """3MF slot_id maps correctly to (ams_id, tray_id).""" + # slot 5 → global_tray_id 4 → ams_id=1, tray_id=0 + spool = _make_spool(id=9) + assignment = _make_assignment(spool_id=9, ams_id=1, tray_id=0) + archive = MagicMock() + archive.file_path = "archives/test.3mf" + + db = AsyncMock() + db.execute = AsyncMock( + side_effect=[ + MagicMock(scalar_one_or_none=MagicMock(return_value=archive)), + MagicMock(scalar_one_or_none=MagicMock(return_value=assignment)), + MagicMock(scalar_one_or_none=MagicMock(return_value=spool)), + ] + ) + + pm = _make_printer_manager() + filament_usage = [{"slot_id": 5, "used_g": 30.0, "type": "PETG", "color": ""}] + + with ( + patch("backend.app.core.config.settings") as mock_settings, + patch("backend.app.utils.threemf_tools.extract_filament_usage_from_3mf", return_value=filament_usage), + ): + mock_path = MagicMock() + mock_path.exists.return_value = True + mock_settings.base_dir.__truediv__ = MagicMock(return_value=mock_path) + + results = await _track_from_3mf( + printer_id=1, + archive_id=10, + status="completed", + print_name="test", + handled_trays=set(), + printer_manager=pm, + db=db, + ) + + assert len(results) == 1 + assert results[0]["ams_id"] == 1 + assert results[0]["tray_id"] == 0 diff --git a/backend/tests/unit/test_scheduler_clear_plate.py b/backend/tests/unit/test_scheduler_clear_plate.py new file mode 100644 index 000000000..ba72026aa --- /dev/null +++ b/backend/tests/unit/test_scheduler_clear_plate.py @@ -0,0 +1,122 @@ +"""Tests for the clear plate queue flow in the print scheduler.""" + +from unittest.mock import MagicMock, patch + +import pytest + +from backend.app.services.print_scheduler import PrintScheduler +from backend.app.services.printer_manager import PrinterManager + + +class TestPrinterManagerPlateCleared: + """Test the plate-cleared flag management in PrinterManager.""" + + @pytest.fixture + def manager(self): + return PrinterManager() + + def test_plate_cleared_initially_false(self, manager): + """No printers should have plate cleared by default.""" + assert not manager.is_plate_cleared(1) + assert not manager.is_plate_cleared(999) + + def test_set_plate_cleared(self, manager): + """Setting plate cleared should make is_plate_cleared return True.""" + manager.set_plate_cleared(1) + assert manager.is_plate_cleared(1) + assert not manager.is_plate_cleared(2) + + def test_consume_plate_cleared(self, manager): + """Consuming plate cleared should reset the flag.""" + manager.set_plate_cleared(1) + assert manager.is_plate_cleared(1) + manager.consume_plate_cleared(1) + assert not manager.is_plate_cleared(1) + + def test_consume_plate_cleared_idempotent(self, manager): + """Consuming when not set should not raise.""" + manager.consume_plate_cleared(1) # Should not raise + assert not manager.is_plate_cleared(1) + + def test_set_plate_cleared_multiple_printers(self, manager): + """Plate cleared should be tracked per printer.""" + manager.set_plate_cleared(1) + manager.set_plate_cleared(3) + assert manager.is_plate_cleared(1) + assert not manager.is_plate_cleared(2) + assert manager.is_plate_cleared(3) + + def test_consume_only_affects_target_printer(self, manager): + """Consuming plate cleared for one printer should not affect others.""" + manager.set_plate_cleared(1) + manager.set_plate_cleared(2) + manager.consume_plate_cleared(1) + assert not manager.is_plate_cleared(1) + assert manager.is_plate_cleared(2) + + +class TestSchedulerIdleCheckWithPlateCleared: + """Test _is_printer_idle with plate-cleared flag interactions.""" + + @pytest.fixture + def scheduler(self): + return PrintScheduler() + + @patch("backend.app.services.print_scheduler.printer_manager") + def test_idle_state_is_idle(self, mock_pm, scheduler): + """Printer in IDLE state should be considered idle.""" + mock_pm.is_connected.return_value = True + mock_pm.get_status.return_value = MagicMock(state="IDLE") + assert scheduler._is_printer_idle(1) is True + + @patch("backend.app.services.print_scheduler.printer_manager") + def test_running_state_not_idle(self, mock_pm, scheduler): + """Printer in RUNNING state should not be idle.""" + mock_pm.is_connected.return_value = True + mock_pm.get_status.return_value = MagicMock(state="RUNNING") + assert scheduler._is_printer_idle(1) is False + + @patch("backend.app.services.print_scheduler.printer_manager") + def test_finish_state_not_idle_without_plate_cleared(self, mock_pm, scheduler): + """Printer in FINISH state should NOT be idle without plate cleared.""" + mock_pm.is_connected.return_value = True + mock_pm.get_status.return_value = MagicMock(state="FINISH") + mock_pm.is_plate_cleared.return_value = False + assert scheduler._is_printer_idle(1) is False + + @patch("backend.app.services.print_scheduler.printer_manager") + def test_finish_state_idle_with_plate_cleared(self, mock_pm, scheduler): + """Printer in FINISH state should be idle when plate is cleared.""" + mock_pm.is_connected.return_value = True + mock_pm.get_status.return_value = MagicMock(state="FINISH") + mock_pm.is_plate_cleared.return_value = True + assert scheduler._is_printer_idle(1) is True + + @patch("backend.app.services.print_scheduler.printer_manager") + def test_failed_state_not_idle_without_plate_cleared(self, mock_pm, scheduler): + """Printer in FAILED state should NOT be idle without plate cleared.""" + mock_pm.is_connected.return_value = True + mock_pm.get_status.return_value = MagicMock(state="FAILED") + mock_pm.is_plate_cleared.return_value = False + assert scheduler._is_printer_idle(1) is False + + @patch("backend.app.services.print_scheduler.printer_manager") + def test_failed_state_idle_with_plate_cleared(self, mock_pm, scheduler): + """Printer in FAILED state should be idle when plate is cleared.""" + mock_pm.is_connected.return_value = True + mock_pm.get_status.return_value = MagicMock(state="FAILED") + mock_pm.is_plate_cleared.return_value = True + assert scheduler._is_printer_idle(1) is True + + @patch("backend.app.services.print_scheduler.printer_manager") + def test_disconnected_printer_not_idle(self, mock_pm, scheduler): + """Disconnected printer should never be idle.""" + mock_pm.is_connected.return_value = False + assert scheduler._is_printer_idle(1) is False + + @patch("backend.app.services.print_scheduler.printer_manager") + def test_no_status_not_idle(self, mock_pm, scheduler): + """Printer with no status should not be idle.""" + mock_pm.is_connected.return_value = True + mock_pm.get_status.return_value = None + assert scheduler._is_printer_idle(1) is False diff --git a/docker-publish-beta.sh b/docker-publish-beta.sh new file mode 100755 index 000000000..9e9f6a576 --- /dev/null +++ b/docker-publish-beta.sh @@ -0,0 +1,193 @@ +#!/bin/bash +# Build and push multi-architecture Docker image to GHCR (private beta) +# +# Usage: +# ./docker-publish-beta.sh [version] [--parallel] +# +# Examples: +# ./docker-publish-beta.sh 0.2.0b # Sequential build +# ./docker-publish-beta.sh 0.2.0b --parallel # Build both archs simultaneously +# +# All versions are also tagged as 'beta' +# +# Prerequisites: +# 1. Log in to ghcr.io: +# echo $GITHUB_TOKEN | docker login ghcr.io -u YOUR_USERNAME --password-stdin +# +# 2. Create a GitHub Personal Access Token with 'write:packages' scope: +# https://github.com/settings/tokens/new?scopes=write:packages +# +# 3. After first push, set package to Private in GitHub → Packages → Settings +# and add beta testers via Manage Access +# +# Supported architectures: +# - linux/amd64 (x86_64, most servers/desktops) +# - linux/arm64 (Raspberry Pi 4/5, Apple Silicon via emulation) + +set -e + +# Configuration +GHCR_REGISTRY="ghcr.io" +IMAGE_NAME="maziggy/bambuddy-beta" +GHCR_IMAGE="${GHCR_REGISTRY}/${IMAGE_NAME}" +PLATFORMS="linux/amd64,linux/arm64" +BUILDER_NAME="bambuddy-builder" + +# Colors for output +RED='\033[0;31m' +GREEN='\033[0;32m' +YELLOW='\033[1;33m' +BLUE='\033[0;34m' +NC='\033[0m' # No Color + +# Parse arguments +VERSION="" +PARALLEL=false +for arg in "$@"; do + case $arg in + --parallel) + PARALLEL=true + ;; + *) + if [ -z "$VERSION" ]; then + VERSION="$arg" + fi + ;; + esac +done + +if [ -z "$VERSION" ]; then + echo -e "${YELLOW}Usage: $0 [--parallel]${NC}" + echo "Example: $0 0.2.0b" + echo " $0 0.2.0b --parallel # Build both architectures simultaneously" + exit 1 +fi + +# Get CPU count +CPU_COUNT=$(nproc 2>/dev/null || sysctl -n hw.ncpu 2>/dev/null || echo 4) + +echo -e "${GREEN}================================================${NC}" +echo -e "${GREEN} Building multi-arch BETA image${NC}" +echo -e "${GREEN} Version: ${VERSION}${NC}" +echo -e "${GREEN} Platforms: ${PLATFORMS}${NC}" +echo -e "${GREEN} CPU cores: ${CPU_COUNT}${NC}" +if [ "$PARALLEL" = true ]; then + echo -e "${GREEN} Mode: PARALLEL (both archs simultaneously)${NC}" +else + echo -e "${GREEN} Mode: Sequential (amd64 → arm64)${NC}" +fi +echo -e "${GREEN} Registry: ${GHCR_IMAGE}${NC}" +echo -e "${GREEN}================================================${NC}" +echo "" + +# Check registry login +if ! grep -q "ghcr.io" ~/.docker/config.json 2>/dev/null; then + echo -e "${YELLOW}Warning: You may not be logged in to ghcr.io${NC}" + echo "Run: echo \$GITHUB_TOKEN | docker login ghcr.io -u YOUR_USERNAME --password-stdin" + echo "" +fi + +# Setup buildx builder if not exists +echo -e "${BLUE}[1/4] Setting up Docker Buildx...${NC}" +if ! docker buildx inspect "$BUILDER_NAME" >/dev/null 2>&1; then + echo "Creating new buildx builder: $BUILDER_NAME (optimized for ${CPU_COUNT} cores)" + docker buildx create \ + --name "$BUILDER_NAME" \ + --driver docker-container \ + --driver-opt network=host \ + --driver-opt "env.BUILDKIT_STEP_LOG_MAX_SIZE=10000000" \ + --buildkitd-flags "--allow-insecure-entitlement network.host --oci-worker-gc=false" \ + --config /dev/stdin <&1 | sed 's/^/[amd64] /' + echo -e "${GREEN}[amd64] Complete!${NC}" + ) & + PID_AMD64=$! + + # Build arm64 in background + ( + echo -e "${BLUE}[arm64] Starting build...${NC}" + docker buildx build \ + --platform linux/arm64 \ + -t "${GHCR_IMAGE}:${VERSION}-arm64" \ + ${BUILD_ARGS} \ + --push \ + . 2>&1 | sed 's/^/[arm64] /' + echo -e "${GREEN}[arm64] Complete!${NC}" + ) & + PID_ARM64=$! + + # Wait for both builds + echo "Waiting for parallel builds to complete..." + wait $PID_AMD64 + wait $PID_ARM64 + + # Create multi-arch manifest + echo -e "${BLUE}Creating multi-arch manifest...${NC}" + docker buildx imagetools create \ + -t "${GHCR_IMAGE}:${VERSION}" -t "${GHCR_IMAGE}:beta" \ + "${GHCR_IMAGE}:${VERSION}-amd64" \ + "${GHCR_IMAGE}:${VERSION}-arm64" +else + # Sequential build (default): Build both platforms in one command + echo -e "${YELLOW}Building sequentially with ${CPU_COUNT} cores (no cache)...${NC}" + DOCKER_BUILDKIT=1 docker buildx build \ + --platform "$PLATFORMS" \ + ${BUILD_ARGS} \ + $TAGS \ + --push \ + . +fi + +echo -e "${BLUE}[4/4] Verifying manifest...${NC}" +docker buildx imagetools inspect "${GHCR_IMAGE}:${VERSION}" + +echo "" +echo -e "${GREEN}================================================${NC}" +echo -e "${GREEN} Successfully pushed multi-arch BETA image:${NC}" +echo -e "${GREEN}================================================${NC}" +echo " ${GHCR_IMAGE}:${VERSION}" +echo " ${GHCR_IMAGE}:beta" +echo "" +echo -e "${BLUE}Supported platforms:${NC}" +echo " - linux/amd64 (Intel/AMD servers, desktops)" +echo " - linux/arm64 (Raspberry Pi 4/5, Apple Silicon)" +echo "" +echo -e "${GREEN}Beta testers can run:${NC}" +echo " docker pull ${GHCR_IMAGE}:${VERSION}" +echo " docker pull ${GHCR_IMAGE}:beta" +echo "" +echo -e "${YELLOW}Reminder: Set package to Private in GitHub → Packages → Settings${NC}" diff --git a/frontend/src/App.tsx b/frontend/src/App.tsx index d62d353d7..c3a072159 100644 --- a/frontend/src/App.tsx +++ b/frontend/src/App.tsx @@ -14,6 +14,7 @@ import { FileManagerPage } from './pages/FileManagerPage'; import { CameraPage } from './pages/CameraPage'; import { StreamOverlayPage } from './pages/StreamOverlayPage'; import { ExternalLinkPage } from './pages/ExternalLinkPage'; +import InventoryPage from './pages/InventoryPage'; import { SystemInfoPage } from './pages/SystemInfoPage'; import { LoginPage } from './pages/LoginPage'; import { SetupPage } from './pages/SetupPage'; @@ -122,6 +123,7 @@ function App() { } /> } /> } /> + } /> } /> } /> } /> diff --git a/frontend/src/__tests__/components/AssignSpoolModal.test.tsx b/frontend/src/__tests__/components/AssignSpoolModal.test.tsx new file mode 100644 index 000000000..4b8d9cf20 --- /dev/null +++ b/frontend/src/__tests__/components/AssignSpoolModal.test.tsx @@ -0,0 +1,134 @@ +import { describe, it, expect, vi, beforeEach } from 'vitest'; +import { screen, waitFor } from '@testing-library/react'; +import { render } from '../utils'; +import { AssignSpoolModal } from '../../components/AssignSpoolModal'; +import { api } from '../../api/client'; + +vi.mock('../../api/client', () => ({ + api: { + getSpools: vi.fn(), + getAssignments: vi.fn(), + assignSpool: vi.fn(), + getSettings: vi.fn().mockResolvedValue({}), + getAuthStatus: vi.fn().mockResolvedValue({ auth_enabled: false }), + }, +})); + +const defaultProps = { + isOpen: true, + onClose: vi.fn(), + printerId: 1, + amsId: 0, + trayId: 0, + trayInfo: { type: 'PLA', color: 'FF0000', location: 'AMS 1 - Slot 1' }, +}; + +const manualSpool = { + id: 1, + material: 'PLA', + subtype: 'Basic', + brand: 'Polymaker', + color_name: 'Red', + rgba: 'FF0000FF', + label_weight: 1000, + weight_used: 0, + tag_uid: null, + tray_uuid: null, +}; + +const blSpool = { + id: 2, + material: 'PLA', + subtype: 'Basic', + brand: 'Bambu', + color_name: 'Jade White', + rgba: 'FFFFFFFE', + label_weight: 1000, + weight_used: 50, + tag_uid: '05CC1E0F00000100', + tray_uuid: 'A1B2C3D4E5F6A1B2C3D4E5F6A1B2C3D4', +}; + +const anotherManualSpool = { + id: 3, + material: 'PETG', + subtype: 'HF', + brand: 'Overture', + color_name: 'Black', + rgba: '000000FF', + label_weight: 1000, + weight_used: 200, + tag_uid: null, + tray_uuid: null, +}; + +describe('AssignSpoolModal', () => { + beforeEach(() => { + vi.clearAllMocks(); + (api.getSpools as ReturnType).mockResolvedValue([manualSpool, blSpool, anotherManualSpool]); + (api.getAssignments as ReturnType).mockResolvedValue([]); + }); + + it('renders nothing when closed', () => { + render(); + expect(screen.queryByText('Assign Spool')).not.toBeInTheDocument(); + }); + + it('filters out Bambu Lab spools (with tag_uid/tray_uuid)', async () => { + render(); + + await waitFor(() => { + expect(screen.getByText(/Polymaker/)).toBeInTheDocument(); + }); + + // Manual spools should be visible + expect(screen.getByText(/Polymaker/)).toBeInTheDocument(); + expect(screen.getByText(/Overture/)).toBeInTheDocument(); + + // BL spool should NOT be visible + expect(screen.queryByText(/Jade White/)).not.toBeInTheDocument(); + }); + + it('filters out spools already assigned to other slots', async () => { + (api.getAssignments as ReturnType).mockResolvedValue([ + { id: 1, spool_id: 3, printer_id: 1, ams_id: 0, tray_id: 1 }, // spool 3 assigned to different slot + ]); + + render(); + + await waitFor(() => { + expect(screen.getByText(/Polymaker/)).toBeInTheDocument(); + }); + + // Spool 1 (not assigned) should be visible + expect(screen.getByText(/Polymaker/)).toBeInTheDocument(); + + // Spool 3 (assigned to another slot) should NOT be visible + expect(screen.queryByText(/Overture/)).not.toBeInTheDocument(); + }); + + it('keeps spool visible if assigned to the current slot', async () => { + (api.getAssignments as ReturnType).mockResolvedValue([ + { id: 1, spool_id: 1, printer_id: 1, ams_id: 0, tray_id: 0 }, // spool 1 assigned to THIS slot + ]); + + render(); + + await waitFor(() => { + expect(screen.getByText(/Polymaker/)).toBeInTheDocument(); + }); + + // Spool 1 (assigned to current slot) should still be visible for re-assignment + expect(screen.getByText(/Polymaker/)).toBeInTheDocument(); + }); + + it('shows noManualSpools message when all spools are BL or assigned', async () => { + (api.getSpools as ReturnType).mockResolvedValue([blSpool]); + + render(); + + await waitFor(() => { + expect(screen.getByText(/No manually added spools/i)).toBeInTheDocument(); + }); + }); +}); diff --git a/frontend/src/__tests__/components/LinkSpoolModal.test.tsx b/frontend/src/__tests__/components/LinkSpoolModal.test.tsx index d7738e24a..26997929c 100644 --- a/frontend/src/__tests__/components/LinkSpoolModal.test.tsx +++ b/frontend/src/__tests__/components/LinkSpoolModal.test.tsx @@ -1,11 +1,11 @@ /** * Tests for the LinkSpoolModal component. * - * Tests the Spoolman link spool modal including: - * - Displaying unlinked spools - * - Selecting a spool to link - * - Link success with toast notification - * - Link error with toast notification + * Tests the inventory link-to-spool modal including: + * - Rendering modal with tag/tray info + * - Displaying untagged spools + * - Linking a spool via click + * - Search filtering */ import { describe, it, expect, vi, beforeEach } from 'vitest'; @@ -16,10 +16,10 @@ import { LinkSpoolModal } from '../../components/LinkSpoolModal'; // Mock the API client vi.mock('../../api/client', () => ({ api: { - getUnlinkedSpools: vi.fn(), - linkSpool: vi.fn(), + getSpools: vi.fn(), + linkTagToSpool: vi.fn(), getSettings: vi.fn().mockResolvedValue({}), - getAuthStatus: vi.fn().mockResolvedValue({ enabled: false, configured: false }), + getAuthStatus: vi.fn().mockResolvedValue({ auth_enabled: false }), }, })); @@ -40,37 +40,56 @@ describe('LinkSpoolModal', () => { const defaultProps = { isOpen: true, onClose: vi.fn(), + tagUid: 'ABCD1234', trayUuid: 'A1B2C3D4E5F6A1B2C3D4E5F6A1B2C3D4', - trayInfo: { - type: 'PLA Basic', - color: 'FF0000', - location: 'AMS A1', - }, + printerId: 1, + amsId: 0, + trayId: 0, }; - const mockUnlinkedSpools = [ + const mockSpools = [ { id: 1, - filament_name: 'PLA Red', - filament_material: 'PLA', - filament_color_hex: 'FF0000', - remaining_weight: 800, - location: 'Shelf A', + material: 'PLA', + brand: 'Generic', + subtype: '', + color_name: 'Red', + rgba: 'FF0000FF', + label_weight: 1000, + weight_used: 200, + tag_uid: null, + tray_uuid: null, }, { id: 2, - filament_name: 'PETG Blue', - filament_material: 'PETG', - filament_color_hex: '0000FF', - remaining_weight: 500, - location: null, + material: 'PETG', + brand: 'Bambu', + subtype: 'Basic', + color_name: 'Blue', + rgba: '0000FFFF', + label_weight: 1000, + weight_used: 500, + tag_uid: null, + tray_uuid: null, + }, + { + id: 3, + material: 'ABS', + brand: 'Brand', + subtype: '', + color_name: 'White', + rgba: 'FFFFFFFF', + label_weight: 1000, + weight_used: 0, + tag_uid: 'EXISTING_TAG', + tray_uuid: 'EXISTING_UUID', }, ]; beforeEach(() => { vi.clearAllMocks(); - vi.mocked(api.getUnlinkedSpools).mockResolvedValue(mockUnlinkedSpools); - vi.mocked(api.linkSpool).mockResolvedValue({ success: true, message: 'Linked' }); + vi.mocked(api.getSpools).mockResolvedValue(mockSpools); + vi.mocked(api.linkTagToSpool).mockResolvedValue({}); }); describe('rendering', () => { @@ -78,33 +97,21 @@ describe('LinkSpoolModal', () => { render(); await waitFor(() => { - // Look for the title in h2 element - expect(screen.getByRole('heading', { name: /link to spoolman/i })).toBeInTheDocument(); + expect(screen.getByRole('heading', { name: /link to spool/i })).toBeInTheDocument(); }); }); - it('displays tray info', async () => { + it('displays printer and tray info', async () => { render(); await waitFor(() => { - expect(screen.getByText('PLA Basic')).toBeInTheDocument(); - expect(screen.getByText('(AMS A1)')).toBeInTheDocument(); - }); - }); - - it('displays tray UUID', async () => { - render(); - - await waitFor(() => { - expect(screen.getByText(defaultProps.trayUuid)).toBeInTheDocument(); + expect(screen.getByText(/AMS 0 T0/)).toBeInTheDocument(); + expect(screen.getByText(/Printer #1/)).toBeInTheDocument(); }); }); it('shows loading state while fetching spools', async () => { - // Delay the response - vi.mocked(api.getUnlinkedSpools).mockImplementation( - () => new Promise(() => {}) - ); + vi.mocked(api.getSpools).mockImplementation(() => new Promise(() => {})); render(); @@ -113,132 +120,74 @@ describe('LinkSpoolModal', () => { }); }); - it('displays unlinked spools list', async () => { + it('displays untagged spools only', async () => { render(); await waitFor(() => { - expect(screen.getByText('PLA Red')).toBeInTheDocument(); - expect(screen.getByText('PETG Blue')).toBeInTheDocument(); + // Spools 1 and 2 have no tag_uid/tray_uuid — should be shown + expect(screen.getByText(/Generic PLA/)).toBeInTheDocument(); + expect(screen.getByText(/Bambu PETG/)).toBeInTheDocument(); }); - }); - it('shows message when no unlinked spools', async () => { - vi.mocked(api.getUnlinkedSpools).mockResolvedValue([]); - - render(); - - await waitFor(() => { - expect(screen.getByText('No unlinked spools available')).toBeInTheDocument(); - }); + // Spool 3 has tag_uid — should be filtered out + expect(screen.queryByText(/Brand ABS/)).not.toBeInTheDocument(); }); it('does not render when isOpen is false', () => { render(); - expect(screen.queryByRole('heading', { name: /link to spoolman/i })).not.toBeInTheDocument(); - }); - }); - - describe('spool selection', () => { - it('allows selecting a spool', async () => { - render(); - - await waitFor(() => { - expect(screen.getByText('PLA Red')).toBeInTheDocument(); - }); - - // Click to select spool - fireEvent.click(screen.getByText('PLA Red')); - - // Should show check mark (via visual styling) - const selectedButton = screen.getByText('PLA Red').closest('button'); - expect(selectedButton).toHaveClass('border-bambu-green'); - }); - - it('link button is disabled until spool is selected', async () => { - render(); - - await waitFor(() => { - expect(screen.getByText('PLA Red')).toBeInTheDocument(); - }); - - const linkButton = screen.getByRole('button', { name: /link to spoolman/i }); - expect(linkButton).toBeDisabled(); - - // Select a spool - fireEvent.click(screen.getByText('PLA Red')); - - expect(linkButton).not.toBeDisabled(); + expect(screen.queryByRole('heading', { name: /link to spool/i })).not.toBeInTheDocument(); }); }); describe('linking', () => { - it('calls linkSpool API on submit', async () => { + it('calls linkTagToSpool on spool click', async () => { render(); await waitFor(() => { - expect(screen.getByText('PLA Red')).toBeInTheDocument(); + expect(screen.getByText(/Generic PLA/)).toBeInTheDocument(); }); - // Select a spool - fireEvent.click(screen.getByText('PLA Red')); - - // Click link button - fireEvent.click(screen.getByRole('button', { name: /link to spoolman/i })); + fireEvent.click(screen.getByText(/Generic PLA/).closest('button')!); await waitFor(() => { - expect(api.linkSpool).toHaveBeenCalledWith(1, defaultProps.trayUuid); + expect(api.linkTagToSpool).toHaveBeenCalledWith(1, { + tag_uid: 'ABCD1234', + tray_uuid: 'A1B2C3D4E5F6A1B2C3D4E5F6A1B2C3D4', + tag_type: 'bambulab', + data_origin: 'nfc_link', + }); }); }); - it('shows success toast on successful link', async () => { + it('shows success toast and calls onClose', async () => { render(); await waitFor(() => { - expect(screen.getByText('PLA Red')).toBeInTheDocument(); + expect(screen.getByText(/Generic PLA/)).toBeInTheDocument(); }); - fireEvent.click(screen.getByText('PLA Red')); - fireEvent.click(screen.getByRole('button', { name: /link to spoolman/i })); - - await waitFor(() => { - expect(mockShowToast).toHaveBeenCalledWith( - 'Spool linked to Spoolman successfully', - 'success' - ); - }); - }); - - it('calls onClose after successful link', async () => { - render(); - - await waitFor(() => { - expect(screen.getByText('PLA Red')).toBeInTheDocument(); - }); - - fireEvent.click(screen.getByText('PLA Red')); - fireEvent.click(screen.getByRole('button', { name: /link to spoolman/i })); + fireEvent.click(screen.getByText(/Generic PLA/).closest('button')!); await waitFor(() => { + expect(mockShowToast).toHaveBeenCalled(); expect(defaultProps.onClose).toHaveBeenCalled(); }); }); - it('shows error toast on link failure', async () => { - const errorMessage = 'Failed to update spool'; - vi.mocked(api.linkSpool).mockRejectedValue(new Error(errorMessage)); + it('shows error toast on failure', async () => { + vi.mocked(api.linkTagToSpool).mockRejectedValue(new Error('Link failed')); render(); await waitFor(() => { - expect(screen.getByText('PLA Red')).toBeInTheDocument(); + expect(screen.getByText(/Generic PLA/)).toBeInTheDocument(); }); - fireEvent.click(screen.getByText('PLA Red')); - fireEvent.click(screen.getByRole('button', { name: /link to spoolman/i })); + fireEvent.click(screen.getByText(/Generic PLA/).closest('button')!); await waitFor(() => { expect(mockShowToast).toHaveBeenCalledWith( - `Failed to link spool: ${errorMessage}`, + expect.stringContaining('Link failed'), 'error' ); }); @@ -246,25 +195,13 @@ describe('LinkSpoolModal', () => { }); describe('modal actions', () => { - it('calls onClose when cancel button is clicked', async () => { - render(); - - await waitFor(() => { - expect(screen.getByText('Cancel')).toBeInTheDocument(); - }); - - fireEvent.click(screen.getByText('Cancel')); - expect(defaultProps.onClose).toHaveBeenCalled(); - }); - it('calls onClose when backdrop is clicked', async () => { render(); await waitFor(() => { - expect(screen.getByRole('heading', { name: /link to spoolman/i })).toBeInTheDocument(); + expect(screen.getByRole('heading', { name: /link to spool/i })).toBeInTheDocument(); }); - // Click the backdrop (the element with bg-black/60) const backdrop = document.querySelector('.bg-black\\/60'); if (backdrop) { fireEvent.click(backdrop); @@ -276,10 +213,9 @@ describe('LinkSpoolModal', () => { render(); await waitFor(() => { - expect(screen.getByRole('heading', { name: /link to spoolman/i })).toBeInTheDocument(); + expect(screen.getByRole('heading', { name: /link to spool/i })).toBeInTheDocument(); }); - // Find and click the X button in the header const closeButtons = screen.getAllByRole('button'); const xButton = closeButtons.find(btn => btn.querySelector('svg.lucide-x')); if (xButton) { diff --git a/frontend/src/__tests__/components/PrinterQueueWidgetClearPlate.test.tsx b/frontend/src/__tests__/components/PrinterQueueWidgetClearPlate.test.tsx new file mode 100644 index 000000000..7f41f845b --- /dev/null +++ b/frontend/src/__tests__/components/PrinterQueueWidgetClearPlate.test.tsx @@ -0,0 +1,177 @@ +/** + * Tests for the PrinterQueueWidget clear plate behavior. + * + * When the printer is in FINISH or FAILED state and has pending queue items, + * the widget shows a "Clear Plate & Start Next" button instead of the + * passive queue link. After clicking, it shows a confirmation state. + */ + +import { describe, it, expect, beforeEach } from 'vitest'; +import { screen, waitFor } from '@testing-library/react'; +import userEvent from '@testing-library/user-event'; +import { render } from '../utils'; +import { PrinterQueueWidget } from '../../components/PrinterQueueWidget'; +import { http, HttpResponse } from 'msw'; +import { server } from '../mocks/server'; + +const mockQueueItems = [ + { + id: 1, + printer_id: 1, + archive_id: 1, + position: 1, + status: 'pending', + archive_name: 'First Print', + printer_name: 'X1 Carbon', + print_time_seconds: 3600, + scheduled_time: null, + }, + { + id: 2, + printer_id: 1, + archive_id: 2, + position: 2, + status: 'pending', + archive_name: 'Second Print', + printer_name: 'X1 Carbon', + print_time_seconds: 7200, + scheduled_time: null, + }, +]; + +describe('PrinterQueueWidget - Clear Plate', () => { + beforeEach(() => { + server.use( + http.get('/api/v1/queue/', ({ request }) => { + const url = new URL(request.url); + const printerId = url.searchParams.get('printer_id'); + if (printerId === '1') { + return HttpResponse.json(mockQueueItems); + } + return HttpResponse.json([]); + }), + http.post('/api/v1/printers/:id/clear-plate', () => { + return HttpResponse.json({ success: true, message: 'Plate cleared' }); + }) + ); + }); + + describe('clear plate button visibility', () => { + it('shows clear plate button when printer state is FINISH', async () => { + render(); + + await waitFor(() => { + expect(screen.getByText('Clear Plate & Start Next')).toBeInTheDocument(); + }); + }); + + it('shows clear plate button when printer state is FAILED', async () => { + render(); + + await waitFor(() => { + expect(screen.getByText('Clear Plate & Start Next')).toBeInTheDocument(); + }); + }); + + it('shows passive link when printer state is IDLE', async () => { + render(); + + await waitFor(() => { + const link = screen.getByRole('link'); + expect(link).toHaveAttribute('href', '/queue'); + }); + + expect(screen.queryByText('Clear Plate & Start Next')).not.toBeInTheDocument(); + }); + + it('shows passive link when printer state is RUNNING', async () => { + render(); + + await waitFor(() => { + const link = screen.getByRole('link'); + expect(link).toHaveAttribute('href', '/queue'); + }); + }); + + it('shows passive link when printerState is not provided', async () => { + render(); + + await waitFor(() => { + const link = screen.getByRole('link'); + expect(link).toHaveAttribute('href', '/queue'); + }); + }); + }); + + describe('clear plate button shows queue info', () => { + it('shows next item name in clear plate mode', async () => { + render(); + + await waitFor(() => { + expect(screen.getByText('First Print')).toBeInTheDocument(); + }); + }); + + it('shows additional items badge in clear plate mode', async () => { + render(); + + await waitFor(() => { + expect(screen.getByText('+1')).toBeInTheDocument(); + }); + }); + }); + + describe('clear plate action', () => { + it('shows confirmation state after clicking clear plate', async () => { + const user = userEvent.setup(); + render(); + + await waitFor(() => { + expect(screen.getByText('Clear Plate & Start Next')).toBeInTheDocument(); + }); + + await user.click(screen.getByText('Clear Plate & Start Next')); + + await waitFor(() => { + // Both the widget confirmation and the toast show this text + const elements = screen.getAllByText('Plate cleared — ready for next print'); + expect(elements.length).toBeGreaterThanOrEqual(1); + }); + }); + + it('shows error toast on API failure', async () => { + server.use( + http.post('/api/v1/printers/:id/clear-plate', () => { + return HttpResponse.json( + { detail: 'Printer not connected' }, + { status: 400 } + ); + }) + ); + + const user = userEvent.setup(); + render(); + + await waitFor(() => { + expect(screen.getByText('Clear Plate & Start Next')).toBeInTheDocument(); + }); + + await user.click(screen.getByText('Clear Plate & Start Next')); + + // Button should remain visible (not transition to success state) + await waitFor(() => { + expect(screen.getByText('Clear Plate & Start Next')).toBeInTheDocument(); + }); + }); + }); + + describe('empty queue', () => { + it('renders nothing in FINISH state with no queue items', async () => { + const { container } = render(); + + await waitFor(() => { + expect(container.querySelector('button')).not.toBeInTheDocument(); + }); + }); + }); +}); diff --git a/frontend/src/__tests__/components/SpoolmanSettings.test.tsx b/frontend/src/__tests__/components/SpoolmanSettings.test.tsx index 254399929..7e3b591e1 100644 --- a/frontend/src/__tests__/components/SpoolmanSettings.test.tsx +++ b/frontend/src/__tests__/components/SpoolmanSettings.test.tsx @@ -1,15 +1,16 @@ /** * Tests for the SpoolmanSettings component. * - * Tests the Spoolman integration UI including: - * - Enable/disable toggle - * - URL configuration - * - Connection status - * - Sync functionality + * Tests the filament tracking mode selector and Spoolman integration UI: + * - Mode selector (Built-in Inventory vs Spoolman) + * - Built-in Inventory info panel + * - Spoolman URL, sync mode, connection status + * - Weight sync and partial usage toggles */ import { describe, it, expect, vi, beforeEach } from 'vitest'; import { screen, waitFor } from '@testing-library/react'; +import userEvent from '@testing-library/user-event'; import { render } from '../utils'; import { SpoolmanSettings } from '../../components/SpoolmanSettings'; @@ -26,6 +27,7 @@ vi.mock('../../api/client', () => ({ syncAllPrintersAms: vi.fn(), syncPrinterAms: vi.fn(), getPrinters: vi.fn(), + getAuthStatus: vi.fn().mockResolvedValue({ auth_enabled: false }), }, })); @@ -36,7 +38,7 @@ describe('SpoolmanSettings', () => { beforeEach(() => { vi.clearAllMocks(); - // Default API mocks + // Default API mocks — Spoolman disabled (Built-in Inventory mode) vi.mocked(api.getSpoolmanSettings).mockResolvedValue({ spoolman_enabled: 'false', spoolman_url: '', @@ -70,90 +72,61 @@ describe('SpoolmanSettings', () => { describe('rendering', () => { it('renders loading state initially', () => { - // Delay the API response to catch loading state vi.mocked(api.getSpoolmanSettings).mockImplementation(() => new Promise(() => {})); render(); - // Should show loading spinner expect(document.querySelector('.animate-spin')).toBeInTheDocument(); }); - it('renders component title', async () => { + it('renders filament tracking title', async () => { render(); await waitFor(() => { - expect(screen.getByText('Spoolman Integration')).toBeInTheDocument(); + expect(screen.getByText('Filament Tracking')).toBeInTheDocument(); }); }); - it('renders enable toggle', async () => { + it('renders mode selector cards', async () => { render(); await waitFor(() => { - expect(screen.getByText('Enable Spoolman')).toBeInTheDocument(); - }); - }); - - it('renders URL input', async () => { - render(); - - await waitFor(() => { - expect(screen.getByText('Spoolman URL')).toBeInTheDocument(); - expect(screen.getByPlaceholderText('http://192.168.1.100:7912')).toBeInTheDocument(); - }); - }); - - it('renders sync mode selector', async () => { - render(); - - await waitFor(() => { - expect(screen.getByText('Sync Mode')).toBeInTheDocument(); - }); - }); - - it('renders info banner about sync', async () => { - render(); - - await waitFor(() => { - expect(screen.getByText('How Sync Works')).toBeInTheDocument(); - expect(screen.getByText(/Only official Bambu Lab spools/)).toBeInTheDocument(); + expect(screen.getByText('Built-in Inventory')).toBeInTheDocument(); + expect(screen.getByText('Spoolman')).toBeInTheDocument(); }); }); }); - describe('disabled state', () => { - it('URL input is disabled when Spoolman is disabled', async () => { + describe('built-in inventory mode (default)', () => { + it('shows built-in inventory as selected by default', async () => { render(); await waitFor(() => { - const urlInput = screen.getByPlaceholderText('http://192.168.1.100:7912'); - expect(urlInput).toBeDisabled(); + // Built-in Inventory card should have the active border + const builtInBtn = screen.getByText('Built-in Inventory').closest('button'); + expect(builtInBtn).toHaveClass('border-bambu-green'); }); }); - it('sync mode selector is disabled when Spoolman is disabled', async () => { + it('shows built-in info panel when selected', async () => { render(); await waitFor(() => { - // Find the select by its display value - const selectElement = screen.getByDisplayValue('Automatic'); - expect(selectElement).toBeDisabled(); + expect(screen.getByText(/Automatically detects Bambu Lab RFID spools/)).toBeInTheDocument(); }); }); - it('does not show connection status when disabled', async () => { + it('does not show Spoolman URL input', async () => { render(); await waitFor(() => { - expect(screen.getByText('Spoolman Integration')).toBeInTheDocument(); + expect(screen.getByText('Filament Tracking')).toBeInTheDocument(); }); - // Status section should not be visible when disabled - expect(screen.queryByText('Status:')).not.toBeInTheDocument(); + expect(screen.queryByPlaceholderText('http://192.168.1.100:7912')).not.toBeInTheDocument(); }); }); - describe('enabled state', () => { + describe('spoolman mode', () => { beforeEach(() => { vi.mocked(api.getSpoolmanSettings).mockResolvedValue({ spoolman_enabled: 'true', @@ -171,16 +144,40 @@ describe('SpoolmanSettings', () => { }); }); - it('URL input is enabled when Spoolman is enabled', async () => { + it('shows Spoolman card as selected', async () => { render(); await waitFor(() => { - const urlInput = screen.getByPlaceholderText('http://192.168.1.100:7912'); - expect(urlInput).not.toBeDisabled(); + const spoolmanBtn = screen.getByText('Spoolman').closest('button'); + expect(spoolmanBtn).toHaveClass('border-bambu-green'); }); }); - it('shows connection status section when enabled', async () => { + it('shows URL input when Spoolman is selected', async () => { + render(); + + await waitFor(() => { + expect(screen.getByPlaceholderText('http://192.168.1.100:7912')).toBeInTheDocument(); + }); + }); + + it('shows sync mode selector', async () => { + render(); + + await waitFor(() => { + expect(screen.getByText('Sync Mode')).toBeInTheDocument(); + }); + }); + + it('shows how sync works info', async () => { + render(); + + await waitFor(() => { + expect(screen.getByText('How Sync Works')).toBeInTheDocument(); + }); + }); + + it('shows connection status section', async () => { render(); await waitFor(() => { @@ -202,20 +199,6 @@ describe('SpoolmanSettings', () => { }); }); - it('shows Connect button when disconnected', async () => { - vi.mocked(api.getSpoolmanStatus).mockResolvedValue({ - enabled: true, - connected: false, - url: 'http://localhost:7912', - }); - - render(); - - await waitFor(() => { - expect(screen.getByText('Connect')).toBeInTheDocument(); - }); - }); - it('shows Connected and Disconnect button when connected', async () => { vi.mocked(api.getSpoolmanStatus).mockResolvedValue({ enabled: true, @@ -242,27 +225,12 @@ describe('SpoolmanSettings', () => { await waitFor(() => { expect(screen.getByText('Sync AMS Data')).toBeInTheDocument(); - expect(screen.getByText('Sync')).toBeInTheDocument(); - }); - }); - - it('shows All Printers option in sync dropdown', async () => { - vi.mocked(api.getSpoolmanStatus).mockResolvedValue({ - enabled: true, - connected: true, - url: 'http://localhost:7912', - }); - - render(); - - await waitFor(() => { - expect(screen.getByRole('option', { name: 'All Printers' })).toBeInTheDocument(); }); }); }); describe('weight sync toggle', () => { - it('shows weight sync toggle when sync mode is auto and enabled', async () => { + it('shows weight sync toggle when Spoolman enabled and sync mode is auto', async () => { vi.mocked(api.getSpoolmanSettings).mockResolvedValue({ spoolman_enabled: 'true', spoolman_url: 'http://localhost:7912', @@ -290,20 +258,11 @@ describe('SpoolmanSettings', () => { render(); await waitFor(() => { - expect(screen.getByText('Spoolman Integration')).toBeInTheDocument(); + expect(screen.getByText('Filament Tracking')).toBeInTheDocument(); }); expect(screen.queryByText('Disable AMS Estimated Weight Sync')).not.toBeInTheDocument(); }); - - it('shows weight sync toggle in disabled state when sync mode is auto', async () => { - render(); - - await waitFor(() => { - // Toggle label is visible since sync mode defaults to auto - expect(screen.getByText('Disable AMS Estimated Weight Sync')).toBeInTheDocument(); - }); - }); }); describe('partial usage toggle', () => { @@ -335,49 +294,28 @@ describe('SpoolmanSettings', () => { render(); await waitFor(() => { - expect(screen.getByText('Spoolman Integration')).toBeInTheDocument(); + expect(screen.getByText('Filament Tracking')).toBeInTheDocument(); }); expect(screen.queryByText('Report Partial Usage for Failed Prints')).not.toBeInTheDocument(); }); }); - describe('sync mode options', () => { - it('shows Automatic option', async () => { + describe('mode switching', () => { + it('can switch to Spoolman mode', async () => { + const user = userEvent.setup(); render(); await waitFor(() => { - expect(screen.getByRole('option', { name: 'Automatic' })).toBeInTheDocument(); + expect(screen.getByText('Built-in Inventory')).toBeInTheDocument(); }); - }); - it('shows Manual Only option', async () => { - render(); + // Click Spoolman card + await user.click(screen.getByText('Spoolman').closest('button')!); + // Spoolman settings should now be visible await waitFor(() => { - expect(screen.getByRole('option', { name: 'Manual Only' })).toBeInTheDocument(); - }); - }); - }); - - describe('info text', () => { - it('shows URL help text', async () => { - render(); - - await waitFor(() => { - expect( - screen.getByText('URL of your Spoolman server (e.g., http://localhost:7912)') - ).toBeInTheDocument(); - }); - }); - - it('shows sync mode description for auto mode', async () => { - render(); - - await waitFor(() => { - expect( - screen.getByText('AMS data syncs automatically when changes are detected') - ).toBeInTheDocument(); + expect(screen.getByPlaceholderText('http://192.168.1.100:7912')).toBeInTheDocument(); }); }); }); diff --git a/frontend/src/__tests__/pages/SettingsPage.test.tsx b/frontend/src/__tests__/pages/SettingsPage.test.tsx index 4eeeb8de6..c1856377c 100644 --- a/frontend/src/__tests__/pages/SettingsPage.test.tsx +++ b/frontend/src/__tests__/pages/SettingsPage.test.tsx @@ -86,7 +86,7 @@ describe('SettingsPage', () => { expect(screen.getAllByText('General').length).toBeGreaterThan(0); expect(screen.getByText('Smart Plugs')).toBeInTheDocument(); expect(screen.getByText('Notifications')).toBeInTheDocument(); - expect(screen.getByText('Filament')).toBeInTheDocument(); + expect(screen.getAllByText('Filament').length).toBeGreaterThan(0); expect(screen.getByText('Network')).toBeInTheDocument(); expect(screen.getByText('API Keys')).toBeInTheDocument(); }); @@ -207,10 +207,10 @@ describe('SettingsPage', () => { render(); await waitFor(() => { - expect(screen.getByText('Filament')).toBeInTheDocument(); + expect(screen.getAllByText('Filament').length).toBeGreaterThan(0); }); - await user.click(screen.getByText('Filament')); + await user.click(screen.getAllByText('Filament')[0]); await waitFor(() => { expect(screen.getByText('AMS Display Thresholds')).toBeInTheDocument(); diff --git a/frontend/src/__tests__/pages/StatsPage.test.tsx b/frontend/src/__tests__/pages/StatsPage.test.tsx index d1c193be0..fdeaf501e 100644 --- a/frontend/src/__tests__/pages/StatsPage.test.tsx +++ b/frontend/src/__tests__/pages/StatsPage.test.tsx @@ -46,7 +46,7 @@ const mockArchives = [ ]; const mockSettings = { - currency: '$', + currency: 'USD', check_updates: false, check_printer_firmware: false, }; diff --git a/frontend/src/__tests__/utils/currency.test.ts b/frontend/src/__tests__/utils/currency.test.ts new file mode 100644 index 000000000..2a5b0fc61 --- /dev/null +++ b/frontend/src/__tests__/utils/currency.test.ts @@ -0,0 +1,43 @@ +import { describe, it, expect } from 'vitest'; +import { getCurrencySymbol, SUPPORTED_CURRENCIES } from '../../utils/currency'; + +describe('getCurrencySymbol', () => { + it('returns $ for USD', () => { + expect(getCurrencySymbol('USD')).toBe('$'); + }); + + it('returns € for EUR', () => { + expect(getCurrencySymbol('EUR')).toBe('€'); + }); + + it('returns £ for GBP', () => { + expect(getCurrencySymbol('GBP')).toBe('£'); + }); + + it('returns ₹ for INR', () => { + expect(getCurrencySymbol('INR')).toBe('₹'); + }); + + it('returns HK$ for HKD', () => { + expect(getCurrencySymbol('HKD')).toBe('HK$'); + }); + + it('returns the code itself for unknown currencies', () => { + expect(getCurrencySymbol('XYZ')).toBe('XYZ'); + }); + + it('is case-insensitive', () => { + expect(getCurrencySymbol('usd')).toBe('$'); + expect(getCurrencySymbol('eur')).toBe('€'); + }); +}); + +describe('SUPPORTED_CURRENCIES', () => { + it('contains INR', () => { + expect(SUPPORTED_CURRENCIES.find((c) => c.code === 'INR')).toBeDefined(); + }); + + it('has 25 entries', () => { + expect(SUPPORTED_CURRENCIES).toHaveLength(25); + }); +}); diff --git a/frontend/src/api/client.ts b/frontend/src/api/client.ts index 1bec8600e..ddaf728d7 100644 --- a/frontend/src/api/client.ts +++ b/frontend/src/api/client.ts @@ -18,6 +18,18 @@ export function getAuthToken(): string | null { return authToken; } +function parseContentDispositionFilename(header: string | null): string | null { + if (!header) return null; + // RFC 5987: filename*=utf-8''percent-encoded-name + const rfc5987Match = header.match(/filename\*=(?:UTF-8|utf-8)''(.+?)(?:;|$)/); + if (rfc5987Match) { + try { return decodeURIComponent(rfc5987Match[1]); } catch { /* fall through */ } + } + // Standard: filename="name" or filename=name + const standardMatch = header.match(/filename="?([^";\n]+)"?/); + return standardMatch?.[1] || null; +} + async function request( endpoint: string, options: RequestInit = {} @@ -812,6 +824,29 @@ export interface SlicerSetting { version: string | null; user_id: string | null; updated_time: string | null; + is_custom: boolean; +} + +export interface SpoolCatalogEntry { + id: number; + name: string; + weight: number; + is_default: boolean; +} + +export interface ColorCatalogEntry { + id: number; + manufacturer: string; + color_name: string; + hex_color: string; + material: string | null; + is_default: boolean; +} + +export interface ColorLookupResult { + found: boolean; + hex_color: string | null; + material: string | null; } export interface SlicerSettingsResponse { @@ -853,6 +888,12 @@ export interface SlicerSettingDeleteResponse { message: string; } +// Built-in filament fallback (static table from backend) +export interface BuiltinFilament { + filament_id: string; + name: string; +} + // Local preset types (OrcaSlicer imports) export interface LocalPreset { id: number; @@ -1686,6 +1727,85 @@ export interface LinkedSpoolsMap { linked: Record; // tag (uppercase) -> spool info } +// Inventory types +export interface InventorySpool { + id: number; + material: string; + subtype: string | null; + color_name: string | null; + rgba: string | null; + brand: string | null; + label_weight: number; + core_weight: number; + weight_used: number; + slicer_filament: string | null; + slicer_filament_name: string | null; + nozzle_temp_min: number | null; + nozzle_temp_max: number | null; + note: string | null; + added_full: boolean | null; + last_used: string | null; + encode_time: string | null; + tag_uid: string | null; + tray_uuid: string | null; + data_origin: string | null; + tag_type: string | null; + archived_at: string | null; + created_at: string; + updated_at: string; + k_profiles?: SpoolKProfile[]; +} + +export interface SpoolUsageRecord { + id: number; + spool_id: number; + printer_id: number | null; + print_name: string | null; + weight_used: number; + percent_used: number; + status: string; + created_at: string; +} + +export interface SpoolKProfile { + id: number; + spool_id: number; + printer_id: number; + extruder: number; + nozzle_diameter: string; + nozzle_type: string | null; + k_value: number; + name: string | null; + cali_idx: number | null; + setting_id: string | null; + created_at: string; +} + +export interface SpoolKProfileInput { + printer_id: number; + extruder?: number; + nozzle_diameter?: string; + nozzle_type?: string | null; + k_value: number; + name?: string | null; + cali_idx?: number | null; + setting_id?: string | null; +} + +export interface SpoolAssignment { + id: number; + spool_id: number; + printer_id: number; + printer_name: string | null; + ams_id: number; + tray_id: number; + fingerprint_color: string | null; + fingerprint_type: string | null; + spool?: InventorySpool | null; + configured: boolean; + created_at: string; +} + // Update types export interface VersionInfo { version: string; @@ -1792,6 +1912,7 @@ export interface ExternalLink { name: string; url: string; icon: string; + open_in_new_tab: boolean; custom_icon: string | null; sort_order: number; created_at: string; @@ -1802,12 +1923,14 @@ export interface ExternalLinkCreate { name: string; url: string; icon: string; + open_in_new_tab?: boolean; } export interface ExternalLinkUpdate { name?: string; url?: string; icon?: string; + open_in_new_tab?: boolean; } // Permission type - all available permissions @@ -2019,7 +2142,7 @@ export const api = { request<{ message: string; auth_enabled: boolean }>('/auth/disable', { method: 'POST', }), - + // Advanced Authentication testSMTP: (data: TestSMTPRequest) => request('/auth/smtp/test', { @@ -2155,6 +2278,10 @@ export const api = { request<{ success: boolean; message: string }>(`/printers/${printerId}/print/resume`, { method: 'POST', }), + clearPlate: (printerId: number) => + request<{ success: boolean; message: string }>(`/printers/${printerId}/clear-plate`, { + method: 'POST', + }), // Get current print user (for reprint tracking - Issue #206) getCurrentPrintUser: (printerId: number) => @@ -2263,8 +2390,7 @@ export const api = { throw new Error(error.detail || `HTTP ${response.status}`); } const disposition = response.headers.get('Content-Disposition'); - const filenameMatch = disposition?.match(/filename="?([^";\n]+)"?/); - const filename = filenameMatch?.[1] || path.split('/').pop() || 'download'; + const filename = parseContentDispositionFilename(disposition) || path.split('/').pop() || 'download'; const blob = await response.blob(); const url = window.URL.createObjectURL(blob); const a = document.createElement('a'); @@ -2464,8 +2590,7 @@ export const api = { throw new Error(error.detail || `HTTP ${response.status}`); } const disposition = response.headers.get('Content-Disposition'); - const filenameMatch = disposition?.match(/filename="?([^";\n]+)"?/); - const downloadFilename = filenameMatch?.[1] || filename || `archive_${id}.3mf`; + const downloadFilename = parseContentDispositionFilename(disposition) || filename || `archive_${id}.3mf`; const blob = await response.blob(); const url = window.URL.createObjectURL(blob); const a = document.createElement('a'); @@ -2605,8 +2730,7 @@ export const api = { throw new Error(error.detail || `HTTP ${response.status}`); } const disposition = response.headers.get('Content-Disposition'); - const filenameMatch = disposition?.match(/filename="?([^";\n]+)"?/); - const filename = filenameMatch?.[1] || `source_${archiveId}.3mf`; + const filename = parseContentDispositionFilename(disposition) || `source_${archiveId}.3mf`; const blob = await response.blob(); const url = window.URL.createObjectURL(blob); const a = document.createElement('a'); @@ -2655,8 +2779,7 @@ export const api = { throw new Error(error.detail || `HTTP ${response.status}`); } const disposition = response.headers.get('Content-Disposition'); - const filenameMatch = disposition?.match(/filename="?([^";\n]+)"?/); - const filename = filenameMatch?.[1] || `archive_${archiveId}.f3d`; + const filename = parseContentDispositionFilename(disposition) || `archive_${archiveId}.f3d`; const blob = await response.blob(); const url = window.URL.createObjectURL(blob); const a = document.createElement('a'); @@ -2903,6 +3026,8 @@ export const api = { request<{ success: boolean }>('/cloud/logout', { method: 'POST' }), getCloudSettings: (version = '02.04.00.70') => request(`/cloud/settings?version=${version}`), + getBuiltinFilaments: () => + request('/cloud/builtin-filaments'), getCloudSettingDetail: (settingId: string) => request(`/cloud/settings/${settingId}`), createCloudSetting: (data: SlicerSettingCreate) => @@ -3242,6 +3367,78 @@ export const api = { body: JSON.stringify(data), }), + // Inventory + getSpools: (includeArchived = false) => + request(`/inventory/spools?include_archived=${includeArchived}`), + getSpool: (id: number) => request(`/inventory/spools/${id}`), + createSpool: (data: Omit) => + request('/inventory/spools', { + method: 'POST', + body: JSON.stringify(data), + }), + updateSpool: (id: number, data: Partial>) => + request(`/inventory/spools/${id}`, { + method: 'PATCH', + body: JSON.stringify(data), + }), + deleteSpool: (id: number) => + request<{ status: string }>(`/inventory/spools/${id}`, { method: 'DELETE' }), + archiveSpool: (id: number) => + request(`/inventory/spools/${id}/archive`, { method: 'POST' }), + restoreSpool: (id: number) => + request(`/inventory/spools/${id}/restore`, { method: 'POST' }), + getSpoolKProfiles: (spoolId: number) => + request(`/inventory/spools/${spoolId}/k-profiles`), + saveSpoolKProfiles: (spoolId: number, profiles: SpoolKProfileInput[]) => + request(`/inventory/spools/${spoolId}/k-profiles`, { + method: 'PUT', + body: JSON.stringify(profiles), + }), + getAssignments: (printerId?: number) => + request(`/inventory/assignments${printerId ? `?printer_id=${printerId}` : ''}`), + assignSpool: (data: { spool_id: number; printer_id: number; ams_id: number; tray_id: number }) => + request('/inventory/assignments', { + method: 'POST', + body: JSON.stringify(data), + }), + unassignSpool: (printerId: number, amsId: number, trayId: number) => + request<{ status: string }>(`/inventory/assignments/${printerId}/${amsId}/${trayId}`, { method: 'DELETE' }), + getSpoolCatalog: () => + request('/inventory/catalog'), + addCatalogEntry: (data: { name: string; weight: number }) => + request('/inventory/catalog', { method: 'POST', body: JSON.stringify(data) }), + updateCatalogEntry: (id: number, data: { name: string; weight: number }) => + request(`/inventory/catalog/${id}`, { method: 'PUT', body: JSON.stringify(data) }), + deleteCatalogEntry: (id: number) => + request<{ status: string }>(`/inventory/catalog/${id}`, { method: 'DELETE' }), + resetSpoolCatalog: () => + request<{ status: string }>('/inventory/catalog/reset', { method: 'POST' }), + getColorCatalog: () => + request('/inventory/colors'), + addColorEntry: (data: { manufacturer: string; color_name: string; hex_color: string; material: string | null }) => + request('/inventory/colors', { method: 'POST', body: JSON.stringify(data) }), + updateColorEntry: (id: number, data: { manufacturer: string; color_name: string; hex_color: string; material: string | null }) => + request(`/inventory/colors/${id}`, { method: 'PUT', body: JSON.stringify(data) }), + deleteColorEntry: (id: number) => + request<{ status: string }>(`/inventory/colors/${id}`, { method: 'DELETE' }), + resetColorCatalog: () => + request<{ status: string }>('/inventory/colors/reset', { method: 'POST' }), + lookupColor: (manufacturer: string, colorName: string, material?: string) => + request(`/inventory/colors/lookup?manufacturer=${encodeURIComponent(manufacturer)}&color_name=${encodeURIComponent(colorName)}${material ? `&material=${encodeURIComponent(material)}` : ''}`), + linkTagToSpool: (spoolId: number, data: { tag_uid?: string; tray_uuid?: string; tag_type?: string; data_origin?: string }) => + request(`/inventory/spools/${spoolId}/link-tag`, { + method: 'PATCH', + body: JSON.stringify(data), + }), + getSpoolUsageHistory: (spoolId: number, limit = 50) => + request(`/inventory/spools/${spoolId}/usage?limit=${limit}`), + getAllUsageHistory: (limit = 100, printerId?: number) => + request(`/inventory/usage?limit=${limit}${printerId ? `&printer_id=${printerId}` : ''}`), + clearSpoolUsageHistory: (spoolId: number) => + request<{ status: string }>(`/inventory/spools/${spoolId}/usage`, { method: 'DELETE' }), + getFilamentPresets: () => + request('/cloud/filaments'), + // Updates getVersion: () => request('/updates/version'), checkForUpdates: () => request('/updates/check'), @@ -3539,8 +3736,7 @@ export const api = { throw new Error(error.detail || `HTTP ${response.status}`); } const contentDisposition = response.headers.get('Content-Disposition'); - const filenameMatch = contentDisposition?.match(/filename="(.+)"/); - const filename = filenameMatch?.[1] || `project_${projectId}.zip`; + const filename = parseContentDispositionFilename(contentDisposition) || `project_${projectId}.zip`; const blob = await response.blob(); return { blob, filename }; }, @@ -3668,8 +3864,7 @@ export const api = { throw new Error(error.detail || `HTTP ${response.status}`); } const disposition = response.headers.get('Content-Disposition'); - const filenameMatch = disposition?.match(/filename="?([^";\n]+)"?/); - const downloadFilename = filenameMatch?.[1] || filename || `file_${id}`; + const downloadFilename = parseContentDispositionFilename(disposition) || filename || `file_${id}`; const blob = await response.blob(); const url = window.URL.createObjectURL(blob); const a = document.createElement('a'); @@ -4348,8 +4543,7 @@ export const supportApi = { } // Get filename from Content-Disposition header or use default const disposition = response.headers.get('Content-Disposition'); - const filenameMatch = disposition?.match(/filename=(.+)/); - const filename = filenameMatch ? filenameMatch[1] : 'bambuddy-support.zip'; + const filename = parseContentDispositionFilename(disposition) || 'bambuddy-support.zip'; // Download the blob const blob = await response.blob(); diff --git a/frontend/src/components/AddExternalLinkModal.tsx b/frontend/src/components/AddExternalLinkModal.tsx index d4c29a4ca..ea477e1e2 100644 --- a/frontend/src/components/AddExternalLinkModal.tsx +++ b/frontend/src/components/AddExternalLinkModal.tsx @@ -1,6 +1,7 @@ import { useState, useEffect, useRef } from 'react'; import { useMutation, useQueryClient } from '@tanstack/react-query'; import { X, Save, Loader2, Upload, Trash2 } from 'lucide-react'; +import { useTranslation } from 'react-i18next'; import { api } from '../api/client'; import type { ExternalLink, ExternalLinkCreate, ExternalLinkUpdate } from '../api/client'; import { Button } from './Button'; @@ -11,6 +12,7 @@ interface AddExternalLinkModalProps { } export function AddExternalLinkModal({ link, onClose }: AddExternalLinkModalProps) { + const { t } = useTranslation(); const queryClient = useQueryClient(); const isEditing = !!link; const fileInputRef = useRef(null); @@ -18,6 +20,7 @@ export function AddExternalLinkModal({ link, onClose }: AddExternalLinkModalProp const [name, setName] = useState(link?.name || ''); const [url, setUrl] = useState(link?.url || ''); const [icon, setIcon] = useState(link?.icon || 'link'); + const [openInNewTab, setOpenInNewTab] = useState(link?.open_in_new_tab || false); const [useCustomIcon, setUseCustomIcon] = useState(!!link?.custom_icon); const [customIconPreview, setCustomIconPreview] = useState( link?.custom_icon ? api.getExternalLinkIconUrl(link.id) : null @@ -137,6 +140,7 @@ export function AddExternalLinkModal({ link, onClose }: AddExternalLinkModalProp name: name.trim(), url: url.trim(), icon: useCustomIcon ? icon : icon, // Keep preset icon as fallback + open_in_new_tab: openInNewTab, }; if (isEditing) { @@ -213,6 +217,24 @@ export function AddExternalLinkModal({ link, onClose }: AddExternalLinkModalProp /> + {/* Open in New Tab */} +
+ + +
+ {/* Icon Section */}
diff --git a/frontend/src/components/AssignSpoolModal.tsx b/frontend/src/components/AssignSpoolModal.tsx new file mode 100644 index 000000000..6ec9b0ca2 --- /dev/null +++ b/frontend/src/components/AssignSpoolModal.tsx @@ -0,0 +1,231 @@ +import { useState } from 'react'; +import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query'; +import { useTranslation } from 'react-i18next'; +import { X, Loader2, Package, Check, Search } from 'lucide-react'; +import { api } from '../api/client'; +import type { InventorySpool, SpoolAssignment } from '../api/client'; +import { Button } from './Button'; +import { useToast } from '../contexts/ToastContext'; + +interface AssignSpoolModalProps { + isOpen: boolean; + onClose: () => void; + printerId: number; + amsId: number; + trayId: number; + trayInfo?: { + type: string; + color: string; + location: string; + }; +} + +export function AssignSpoolModal({ isOpen, onClose, printerId, amsId, trayId, trayInfo }: AssignSpoolModalProps) { + const { t } = useTranslation(); + const queryClient = useQueryClient(); + const { showToast } = useToast(); + const [selectedSpoolId, setSelectedSpoolId] = useState(null); + const [searchFilter, setSearchFilter] = useState(''); + + const { data: spools, isLoading } = useQuery({ + queryKey: ['inventory-spools'], + queryFn: () => api.getSpools(), + enabled: isOpen, + }); + + const { data: assignments } = useQuery({ + queryKey: ['spool-assignments'], + queryFn: () => api.getAssignments(), + enabled: isOpen, + }); + + const assignMutation = useMutation({ + mutationFn: (spoolId: number) => + api.assignSpool({ spool_id: spoolId, printer_id: printerId, ams_id: amsId, tray_id: trayId }), + onSuccess: (newAssignment) => { + // Immediately update cache so UI reflects the new assignment without waiting for refetch + queryClient.setQueryData(['spool-assignments'], (old) => { + const filtered = (old || []).filter(a => + !(a.printer_id === printerId && a.ams_id === amsId && a.tray_id === trayId) + ); + filtered.push(newAssignment); + return filtered; + }); + queryClient.invalidateQueries({ queryKey: ['spool-assignments'] }); + showToast(t('inventory.assignSuccess'), 'success'); + onClose(); + }, + onError: (error: Error) => { + showToast(`${t('inventory.assignFailed')}: ${error.message}`, 'error'); + }, + }); + + if (!isOpen) return null; + + // Filter out Bambu Lab spools (identified by RFID tag_uid or tray_uuid) + // and spools already assigned to other slots + const assignedSpoolIds = new Set( + (assignments || []) + .filter(a => !(a.printer_id === printerId && a.ams_id === amsId && a.tray_id === trayId)) + .map(a => a.spool_id) + ); + const manualSpools = spools?.filter((spool: InventorySpool) => + !spool.tag_uid && !spool.tray_uuid && !assignedSpoolIds.has(spool.id) + ); + + const filteredSpools = manualSpools?.filter((spool: InventorySpool) => { + if (!searchFilter) return true; + const q = searchFilter.toLowerCase(); + return ( + spool.material.toLowerCase().includes(q) || + (spool.brand?.toLowerCase().includes(q) ?? false) || + (spool.color_name?.toLowerCase().includes(q) ?? false) || + (spool.subtype?.toLowerCase().includes(q) ?? false) + ); + }); + + const handleAssign = () => { + if (selectedSpoolId) { + assignMutation.mutate(selectedSpoolId); + } + }; + + return ( +
+
+ +
+ {/* Header */} +
+
+ +

{t('inventory.assignSpool')}

+
+ +
+ + {/* Content */} +
+ {/* Tray info */} + {trayInfo && ( +
+

{t('inventory.selectSpool')}:

+
+ {trayInfo.color && ( + + )} + {trayInfo.type || t('ams.emptySlot')} + ({trayInfo.location}) +
+
+ )} + + {/* Search filter */} +
+ + setSearchFilter(e.target.value)} + placeholder={t('inventory.searchSpools')} + className="w-full pl-9 pr-3 py-2 bg-bambu-dark border border-bambu-dark-tertiary rounded-lg text-white text-sm placeholder:text-bambu-gray focus:outline-none focus:border-bambu-green" + /> +
+ + {/* Spool list */} +
+ {isLoading ? ( +
+ +
+ ) : filteredSpools && filteredSpools.length > 0 ? ( +
+ {filteredSpools.map((spool: InventorySpool) => ( + + ))} +
+ ) : manualSpools && manualSpools.length === 0 ? ( +
+

{t('inventory.noManualSpools')}

+
+ ) : ( +
+

{t('inventory.noSpoolsMatch')}

+
+ )} +
+
+ + {/* Footer */} +
+ + +
+ + {assignMutation.isError && ( +
+ {(assignMutation.error as Error).message} +
+ )} +
+
+ ); +} diff --git a/frontend/src/components/ColorCatalogSettings.tsx b/frontend/src/components/ColorCatalogSettings.tsx new file mode 100644 index 000000000..710245ec0 --- /dev/null +++ b/frontend/src/components/ColorCatalogSettings.tsx @@ -0,0 +1,583 @@ +import { useState, useEffect, useCallback, useRef } from 'react'; +import { useTranslation } from 'react-i18next'; +import { Palette, Plus, Trash2, RotateCcw, Loader2, Pencil, Check, X, Search, Download, Upload, Cloud } from 'lucide-react'; +import { api, getAuthToken } from '../api/client'; +import type { ColorCatalogEntry } from '../api/client'; +import { useToast } from '../contexts/ToastContext'; +import { Card, CardHeader, CardContent } from './Card'; +import { ConfirmModal } from './ConfirmModal'; + +export function ColorCatalogSettings() { + const { t } = useTranslation(); + const { showToast } = useToast(); + const [catalog, setCatalog] = useState([]); + const [loading, setLoading] = useState(true); + const [search, setSearch] = useState(''); + const [filterManufacturer, setFilterManufacturer] = useState('Bambu Lab'); + const fileInputRef = useRef(null); + + // Add/Edit form state + const [showAddForm, setShowAddForm] = useState(false); + const [editingId, setEditingId] = useState(null); + const [formManufacturer, setFormManufacturer] = useState(''); + const [formColorName, setFormColorName] = useState(''); + const [formHexColor, setFormHexColor] = useState('#FFFFFF'); + const [formMaterial, setFormMaterial] = useState(''); + const [saving, setSaving] = useState(false); + + // Sync state + const [syncing, setSyncing] = useState(false); + const [syncProgress, setSyncProgress] = useState<{ fetched: number; total: number } | null>(null); + + // Confirmation modals + const [deleteEntry, setDeleteEntry] = useState(null); + const [showResetConfirm, setShowResetConfirm] = useState(false); + + const loadCatalog = useCallback(async () => { + try { + const entries = await api.getColorCatalog(); + setCatalog(entries); + } catch { + showToast(t('settings.colorCatalog.loadFailed'), 'error'); + } finally { + setLoading(false); + } + }, [showToast, t]); + + useEffect(() => { + loadCatalog(); + }, [loadCatalog]); + + const manufacturers = [...new Set(catalog.map(e => e.manufacturer))].sort(); + + const filteredCatalog = catalog.filter(entry => { + const matchesSearch = search === '' || + entry.manufacturer.toLowerCase().includes(search.toLowerCase()) || + entry.color_name.toLowerCase().includes(search.toLowerCase()) || + (entry.material?.toLowerCase().includes(search.toLowerCase()) ?? false); + const matchesManufacturer = filterManufacturer === '' || entry.manufacturer === filterManufacturer; + return matchesSearch && matchesManufacturer; + }); + + const resetForm = () => { + setFormManufacturer(''); + setFormColorName(''); + setFormHexColor('#FFFFFF'); + setFormMaterial(''); + }; + + const handleAdd = async () => { + if (!formManufacturer.trim() || !formColorName.trim() || !formHexColor) { + showToast(t('settings.colorCatalog.fieldsRequired'), 'error'); + return; + } + setSaving(true); + try { + const entry = await api.addColorEntry({ + manufacturer: formManufacturer.trim(), + color_name: formColorName.trim(), + hex_color: formHexColor, + material: formMaterial.trim() || null, + }); + setCatalog(prev => [...prev, entry].sort((a, b) => + a.manufacturer.localeCompare(b.manufacturer) || + (a.material || '').localeCompare(b.material || '') || + a.color_name.localeCompare(b.color_name) + )); + setShowAddForm(false); + resetForm(); + showToast(t('settings.colorCatalog.colorAdded'), 'success'); + } catch { + showToast(t('settings.colorCatalog.addFailed'), 'error'); + } finally { + setSaving(false); + } + }; + + const startEdit = (entry: ColorCatalogEntry) => { + setEditingId(entry.id); + setFormManufacturer(entry.manufacturer); + setFormColorName(entry.color_name); + setFormHexColor(entry.hex_color); + setFormMaterial(entry.material || ''); + }; + + const cancelEdit = () => { + setEditingId(null); + resetForm(); + }; + + const handleUpdate = async (id: number) => { + if (!formManufacturer.trim() || !formColorName.trim() || !formHexColor) { + showToast(t('settings.colorCatalog.fieldsRequired'), 'error'); + return; + } + setSaving(true); + try { + const updated = await api.updateColorEntry(id, { + manufacturer: formManufacturer.trim(), + color_name: formColorName.trim(), + hex_color: formHexColor, + material: formMaterial.trim() || null, + }); + setCatalog(prev => + prev.map(e => e.id === id ? updated : e).sort((a, b) => + a.manufacturer.localeCompare(b.manufacturer) || + (a.material || '').localeCompare(b.material || '') || + a.color_name.localeCompare(b.color_name) + ) + ); + setEditingId(null); + resetForm(); + showToast(t('settings.colorCatalog.colorUpdated'), 'success'); + } catch { + showToast(t('settings.colorCatalog.updateFailed'), 'error'); + } finally { + setSaving(false); + } + }; + + const handleDelete = async () => { + if (!deleteEntry) return; + try { + await api.deleteColorEntry(deleteEntry.id); + setCatalog(prev => prev.filter(e => e.id !== deleteEntry.id)); + showToast(t('settings.colorCatalog.colorDeleted'), 'success'); + } catch { + showToast(t('settings.colorCatalog.deleteFailed'), 'error'); + } finally { + setDeleteEntry(null); + } + }; + + const handleReset = async () => { + setShowResetConfirm(false); + setLoading(true); + try { + await api.resetColorCatalog(); + await loadCatalog(); + showToast(t('settings.colorCatalog.resetSuccess'), 'success'); + } catch { + showToast(t('settings.colorCatalog.resetFailed'), 'error'); + setLoading(false); + } + }; + + const handleSync = async () => { + setSyncing(true); + setSyncProgress(null); + try { + const headers: Record = {}; + const token = getAuthToken(); + if (token) { + headers['Authorization'] = `Bearer ${token}`; + } + const response = await fetch('/api/v1/inventory/colors/sync', { method: 'POST', headers }); + if (!response.ok) throw new Error('Failed to start sync'); + + const reader = response.body?.getReader(); + if (!reader) throw new Error('No response body'); + + const decoder = new TextDecoder(); + let buffer = ''; + + while (true) { + const { done, value } = await reader.read(); + if (done) break; + + buffer += decoder.decode(value, { stream: true }); + const lines = buffer.split('\n'); + buffer = lines.pop() || ''; + + for (const line of lines) { + if (line.startsWith('data: ')) { + try { + const data = JSON.parse(line.slice(6)); + if (data.type === 'progress') { + setSyncProgress({ fetched: data.total_fetched, total: data.total_available }); + } else if (data.type === 'complete') { + if (data.added === 0) { + showToast(t('settings.colorCatalog.syncUpToDate', { count: data.total_fetched }), 'success'); + } else { + showToast(t('settings.colorCatalog.syncComplete', { added: data.added, skipped: data.skipped }), 'success'); + } + } else if (data.type === 'error') { + showToast(`${t('settings.colorCatalog.syncError')}: ${data.error}`, 'error'); + } + } catch { + // Ignore parse errors + } + } + } + } + await loadCatalog(); + } catch { + showToast(t('settings.colorCatalog.syncFailed'), 'error'); + } finally { + setSyncing(false); + setSyncProgress(null); + } + }; + + const handleExport = () => { + const exportData = catalog.map(({ manufacturer, color_name, hex_color, material }) => ({ + manufacturer, color_name, hex_color, material, + })); + const blob = new Blob([JSON.stringify(exportData, null, 2)], { type: 'application/json' }); + const url = URL.createObjectURL(blob); + const a = document.createElement('a'); + a.href = url; + a.download = 'color-catalog.json'; + document.body.appendChild(a); + a.click(); + document.body.removeChild(a); + URL.revokeObjectURL(url); + showToast(t('settings.colorCatalog.exported', { count: catalog.length }), 'success'); + }; + + const handleImport = async (e: React.ChangeEvent) => { + const file = e.target.files?.[0]; + if (!file) return; + try { + const text = await file.text(); + const data = JSON.parse(text) as Array<{ + manufacturer: string; color_name: string; hex_color: string; material?: string | null; + }>; + if (!Array.isArray(data)) throw new Error('Invalid format'); + + let added = 0; + let skipped = 0; + for (const item of data) { + if (!item.manufacturer || !item.color_name || !item.hex_color) { skipped++; continue; } + const exists = catalog.some(c => + c.manufacturer.toLowerCase() === item.manufacturer.toLowerCase() && + c.color_name.toLowerCase() === item.color_name.toLowerCase() && + (c.material || '').toLowerCase() === (item.material || '').toLowerCase() + ); + if (exists) { skipped++; continue; } + try { + const entry = await api.addColorEntry({ + manufacturer: item.manufacturer, + color_name: item.color_name, + hex_color: item.hex_color, + material: item.material || null, + }); + setCatalog(prev => [...prev, entry].sort((a, b) => + a.manufacturer.localeCompare(b.manufacturer) || + (a.material || '').localeCompare(b.material || '') || + a.color_name.localeCompare(b.color_name) + )); + added++; + } catch { skipped++; } + } + showToast(t('settings.colorCatalog.imported', { added, skipped }), 'success'); + } catch { + showToast(t('settings.colorCatalog.importFailed'), 'error'); + } + if (fileInputRef.current) fileInputRef.current.value = ''; + }; + + return ( + + +
+ +

{t('settings.colorCatalog.title')}

+ ({catalog.length}) +
+
+ + + + + + +
+
+ +

+ {t('settings.colorCatalog.description')} +

+ + {/* Search and filter */} +
+
+ + setSearch(e.target.value)} + /> +
+ +
+ + {/* Add form */} + {showAddForm && ( +
+

{t('settings.colorCatalog.addNewColor')}

+
+ setFormManufacturer(e.target.value)} + /> + setFormColorName(e.target.value)} + /> +
+ setFormHexColor(e.target.value)} + /> + setFormHexColor(e.target.value)} + /> +
+ setFormMaterial(e.target.value)} + /> +
+ + +
+
+
+ )} + + {/* Filter info */} + {(search || filterManufacturer) && ( +
+ {t('settings.colorCatalog.showing', { filtered: filteredCatalog.length, total: catalog.length })} +
+ )} + + {/* Catalog list */} + {loading ? ( +
+ + {t('common.loading')} +
+ ) : ( +
+ + + + + + + + + + + + + {filteredCatalog.length === 0 ? ( + + + + ) : ( + filteredCatalog.map(entry => ( + + {editingId === entry.id ? ( + <> + + + + + + + + ) : ( + <> + + + + + + + + )} + + )) + )} + +
{t('settings.colorCatalog.manufacturer')}{t('inventory.material')}{t('settings.colorCatalog.colorName')}{t('settings.colorCatalog.hex')}
+ {search || filterManufacturer ? t('settings.colorCatalog.noMatch') : t('settings.colorCatalog.empty')} +
+ setFormHexColor(e.target.value)} + /> + + setFormManufacturer(e.target.value)} + /> + + setFormMaterial(e.target.value)} + /> + + setFormColorName(e.target.value)} + /> + + setFormHexColor(e.target.value)} + /> + +
+ + +
+
+
+
{entry.manufacturer}{entry.material || '-'}{entry.color_name}{entry.hex_color} +
+ + +
+
+
+ )} +
+ + {/* Delete confirmation */} + {deleteEntry && ( + setDeleteEntry(null)} + /> + )} + + {/* Reset confirmation */} + {showResetConfirm && ( + setShowResetConfirm(false)} + /> + )} +
+ ); +} diff --git a/frontend/src/components/ColumnConfigModal.tsx b/frontend/src/components/ColumnConfigModal.tsx new file mode 100644 index 000000000..9d6d3b0df --- /dev/null +++ b/frontend/src/components/ColumnConfigModal.tsx @@ -0,0 +1,187 @@ +import { useState, useEffect, useRef } from 'react'; +import { useTranslation } from 'react-i18next'; +import { GripVertical, Eye, EyeOff, ChevronUp, ChevronDown, RotateCcw } from 'lucide-react'; +import { Card, CardContent } from './Card'; +import { Button } from './Button'; + +export interface ColumnConfig { + id: string; + label: string; + visible: boolean; +} + +interface ColumnConfigModalProps { + isOpen: boolean; + onClose: () => void; + columns: ColumnConfig[]; + defaultColumns: ColumnConfig[]; + onSave: (columns: ColumnConfig[]) => void; +} + +export function ColumnConfigModal({ isOpen, onClose, columns, defaultColumns, onSave }: ColumnConfigModalProps) { + const { t } = useTranslation(); + const [localColumns, setLocalColumns] = useState(columns); + const [draggedIndex, setDraggedIndex] = useState(null); + const draggedIndexRef = useRef(null); + + useEffect(() => { + if (isOpen) { + setLocalColumns(columns.map((c) => ({ ...c }))); + } + }, [isOpen, columns]); + + useEffect(() => { + if (!isOpen) return; + const handleKeyDown = (e: KeyboardEvent) => { + if (e.key === 'Escape') onClose(); + }; + window.addEventListener('keydown', handleKeyDown); + return () => window.removeEventListener('keydown', handleKeyDown); + }, [isOpen, onClose]); + + if (!isOpen) return null; + + const toggleVisibility = (index: number) => { + setLocalColumns((prev) => + prev.map((col, i) => (i === index ? { ...col, visible: !col.visible } : col)) + ); + }; + + const moveColumn = (fromIndex: number, toIndex: number) => { + if (toIndex < 0 || toIndex >= localColumns.length) return; + setLocalColumns((prev) => { + const newColumns = [...prev]; + const [moved] = newColumns.splice(fromIndex, 1); + newColumns.splice(toIndex, 0, moved); + return newColumns; + }); + }; + + const handleDragStart = (e: React.DragEvent, index: number) => { + draggedIndexRef.current = index; + setDraggedIndex(index); + e.dataTransfer.effectAllowed = 'move'; + }; + + const handleDragOver = (e: React.DragEvent, index: number) => { + e.preventDefault(); + e.dataTransfer.dropEffect = 'move'; + const from = draggedIndexRef.current; + if (from !== null && from !== index) { + moveColumn(from, index); + draggedIndexRef.current = index; + setDraggedIndex(index); + } + }; + + const handleDrop = (e: React.DragEvent) => { + e.preventDefault(); + }; + + const handleDragEnd = () => { + draggedIndexRef.current = null; + setDraggedIndex(null); + }; + + const resetToDefaults = () => { + setLocalColumns(defaultColumns.map((c) => ({ ...c }))); + }; + + const handleSave = () => { + onSave(localColumns); + onClose(); + }; + + const visibleCount = localColumns.filter((c) => c.visible).length; + + return ( +
+ e.stopPropagation()}> + + {/* Header */} +

{t('inventory.configureColumns')}

+

+ {t('inventory.configureColumnsDesc')} + + ({visibleCount} {t('inventory.of')} {localColumns.length} {t('inventory.visible')}) + +

+ + {/* Column list */} +
+ {localColumns.map((column, index) => ( +
handleDragStart(e, index)} + onDragOver={(e) => handleDragOver(e, index)} + onDrop={handleDrop} + onDragEnd={handleDragEnd} + > + {/* Drag Handle */} +
+ +
+ + {/* Column Name */} + {column.label} + + {/* Move Buttons */} +
+ + +
+ + {/* Visibility Toggle */} + +
+ ))} +
+ + {/* Footer */} +
+ + + +
+
+
+
+ ); +} diff --git a/frontend/src/components/ConfigureAmsSlotModal.tsx b/frontend/src/components/ConfigureAmsSlotModal.tsx index 7adffa5cf..9cdc7248a 100644 --- a/frontend/src/components/ConfigureAmsSlotModal.tsx +++ b/frontend/src/components/ConfigureAmsSlotModal.tsx @@ -226,11 +226,12 @@ export function ConfigureAmsSlotModal({ const [showSuccess, setShowSuccess] = useState(false); const [showExtendedColors, setShowExtendedColors] = useState(false); - // Fetch cloud settings - const { data: cloudSettings, isLoading: settingsLoading } = useQuery({ + // Fetch cloud settings (gracefully handle 401 when logged out) + const { data: cloudSettings, isLoading: settingsLoading, isError: cloudError } = useQuery({ queryKey: ['cloudSettings'], queryFn: () => api.getCloudSettings(), enabled: isOpen, + retry: false, }); // Fetch local presets @@ -240,6 +241,14 @@ export function ConfigureAmsSlotModal({ enabled: isOpen, }); + // Fetch built-in filament names (static fallback) + const { data: builtinFilaments, isLoading: builtinLoading } = useQuery({ + queryKey: ['builtinFilaments'], + queryFn: () => api.getBuiltinFilaments(), + enabled: isOpen, + staleTime: Infinity, + }); + // Fetch K profiles const { data: kprofilesData, isLoading: kprofilesLoading } = useQuery({ queryKey: ['kprofiles', printerId, nozzleDiameter], @@ -252,23 +261,29 @@ export function ConfigureAmsSlotModal({ mutationFn: async () => { if (!selectedPresetId) throw new Error('No filament preset selected'); - // Check if this is a local preset + // Determine preset source const isLocal = selectedPresetId.startsWith('local_'); + const isBuiltin = selectedPresetId.startsWith('builtin_'); const localId = isLocal ? parseInt(selectedPresetId.replace('local_', ''), 10) : null; + const builtinFilamentId = isBuiltin ? selectedPresetId.replace('builtin_', '') : null; const localPreset = isLocal ? localPresets?.filament.find(p => p.id === localId) : null; + const builtinPreset = isBuiltin + ? builtinFilaments?.find(b => b.filament_id === builtinFilamentId) + : null; - // Get the selected cloud preset details (null for local presets) - const selectedPreset = !isLocal + // Get the selected cloud preset details (null for local/builtin presets) + const selectedPreset = (!isLocal && !isBuiltin) ? cloudSettings?.filament.find(p => p.setting_id === selectedPresetId) : null; - if (!isLocal && !selectedPreset) throw new Error('Selected preset not found'); + if (!isLocal && !isBuiltin && !selectedPreset) throw new Error('Selected preset not found'); if (isLocal && !localPreset) throw new Error('Selected local preset not found'); + if (isBuiltin && !builtinPreset) throw new Error('Selected builtin preset not found'); // Parse the preset name for filament info - const presetName = isLocal ? localPreset!.name : selectedPreset!.name; + const presetName = isLocal ? localPreset!.name : isBuiltin ? builtinPreset!.name : selectedPreset!.name; const parsed = parsePresetName(presetName); // Get cali_idx from selected K profile's slot_id (-1 = use default 0.020) @@ -287,6 +302,10 @@ export function ConfigureAmsSlotModal({ // Local presets have no Bambu Cloud mapping trayInfoIdx = ''; settingId = ''; + } else if (isBuiltin) { + // Built-in presets use the filament_id directly as tray_info_idx + trayInfoIdx = builtinFilamentId!; + settingId = ''; } else { // Get tray_info_idx: for user presets, fetch detail to get filament_id or derive from base_id trayInfoIdx = convertToTrayInfoIdx(selectedPresetId); @@ -312,7 +331,7 @@ export function ConfigureAmsSlotModal({ let tempMin = isLocal && localPreset?.nozzle_temp_min ? localPreset.nozzle_temp_min : 190; let tempMax = isLocal && localPreset?.nozzle_temp_max ? localPreset.nozzle_temp_max : 230; - if (!isLocal || (!localPreset?.nozzle_temp_min && !localPreset?.nozzle_temp_max)) { + if (!isLocal || isBuiltin || (!localPreset?.nozzle_temp_min && !localPreset?.nozzle_temp_max)) { // Fall back to material-based defaults const material = (isLocal ? (localPreset?.filament_type || parsed.material) : parsed.material).toUpperCase(); if (material.includes('PLA')) { @@ -368,8 +387,8 @@ export function ConfigureAmsSlotModal({ // Save the preset mapping so we can display the correct name in the UI // This is needed because user presets use filament_id (e.g., P285e239) as tray_info_idx, // which can't be resolved to a name via the filamentInfo API - const mappingPresetId = isLocal ? `local_${localId}` : selectedPresetId; - const mappingSource = isLocal ? 'local' : 'cloud'; + const mappingPresetId = isLocal ? `local_${localId}` : isBuiltin ? `builtin_${builtinFilamentId}` : selectedPresetId; + const mappingSource = isLocal ? 'local' : isBuiltin ? 'builtin' : 'cloud'; try { await api.saveSlotPreset(printerId, slotInfo.amsId, slotInfo.trayId, mappingPresetId, traySubBrands, mappingSource); } catch (e) { @@ -405,15 +424,28 @@ export function ConfigureAmsSlotModal({ }, }); - // Unified preset item for the list (cloud + local) - type PresetItem = { id: string; name: string; source: 'cloud' | 'local'; isUser: boolean }; + // Unified preset item for the list (cloud + local + builtin fallback) + type PresetItem = { id: string; name: string; source: 'cloud' | 'local' | 'builtin'; isUser: boolean }; - // Filter filament presets based on search (merged cloud + local) + // Filter filament presets based on search (merged cloud + local + builtin) const filteredPresets = useMemo(() => { const query = searchQuery.toLowerCase(); const items: PresetItem[] = []; - // Add local presets first + // Collect IDs already covered by cloud and local to avoid duplicates in fallback + const coveredIds = new Set(); + + // 1. Cloud presets + if (cloudSettings?.filament) { + for (const cp of cloudSettings.filament) { + coveredIds.add(cp.setting_id); + if (!query || cp.name.toLowerCase().includes(query)) { + items.push({ id: cp.setting_id, name: cp.name, source: 'cloud', isUser: isUserPreset(cp.setting_id) }); + } + } + } + + // 2. Local presets if (localPresets?.filament) { for (const lp of localPresets.filament) { if (!query || lp.name.toLowerCase().includes(query)) { @@ -422,35 +454,46 @@ export function ConfigureAmsSlotModal({ } } - // Add cloud presets - if (cloudSettings?.filament) { - for (const cp of cloudSettings.filament) { - if (!query || cp.name.toLowerCase().includes(query)) { - items.push({ id: cp.setting_id, name: cp.name, source: 'cloud', isUser: isUserPreset(cp.setting_id) }); + // 3. Built-in filament names (fallback — only add entries not already covered) + if (builtinFilaments) { + for (const bf of builtinFilaments) { + if (coveredIds.has(bf.filament_id)) continue; + // Convert filament_id to setting_id format for cloud compatibility + // e.g. "GFA00" → cloud setting_id would be "GFSA00" (insert S after GF) + const settingId = bf.filament_id.startsWith('GF') + ? 'GFS' + bf.filament_id.slice(2) + : bf.filament_id; + if (coveredIds.has(settingId)) continue; + if (!query || bf.name.toLowerCase().includes(query)) { + items.push({ id: `builtin_${bf.filament_id}`, name: bf.name, source: 'builtin', isUser: false }); } } } - // Sort: local first, then user cloud presets, then built-in, alphabetically within groups + // Sort: cloud user presets first, then cloud built-in, then local, then builtin fallback return items.sort((a, b) => { - if (a.source === 'local' && b.source !== 'local') return -1; - if (a.source !== 'local' && b.source === 'local') return 1; + const sourceOrder = { cloud: 0, local: 1, builtin: 2 }; + if (a.source !== b.source) return sourceOrder[a.source] - sourceOrder[b.source]; if (a.isUser && !b.isUser) return -1; if (!a.isUser && b.isUser) return 1; return a.name.localeCompare(b.name); }); - }, [cloudSettings?.filament, localPresets?.filament, searchQuery]); + }, [cloudSettings?.filament, localPresets?.filament, builtinFilaments, searchQuery]); // Get full preset name for K profile filtering (brand + material, without printer suffix) const selectedPresetInfo = useMemo(() => { if (!selectedPresetId) return null; - // Resolve the name from either local or cloud presets + // Resolve the name from cloud, local, or builtin presets let presetName: string | null = null; if (selectedPresetId.startsWith('local_')) { const localId = parseInt(selectedPresetId.replace('local_', ''), 10); const lp = localPresets?.filament.find(p => p.id === localId); presetName = lp?.name || null; + } else if (selectedPresetId.startsWith('builtin_')) { + const filamentId = selectedPresetId.replace('builtin_', ''); + const bf = builtinFilaments?.find(b => b.filament_id === filamentId); + presetName = bf?.name || null; } else if (cloudSettings?.filament) { const cp = cloudSettings.filament.find(p => p.setting_id === selectedPresetId); presetName = cp?.name || null; @@ -470,7 +513,7 @@ export function ConfigureAmsSlotModal({ material: parsed.material, brand: parsed.brand, }; - }, [selectedPresetId, cloudSettings?.filament, localPresets?.filament]); + }, [selectedPresetId, cloudSettings?.filament, localPresets?.filament, builtinFilaments]); // For backwards compatibility with the label const selectedMaterial = selectedPresetInfo?.fullName || ''; @@ -596,7 +639,7 @@ export function ConfigureAmsSlotModal({ if (!isOpen) return null; - const isLoading = settingsLoading || localLoading || kprofilesLoading; + const isLoading = (settingsLoading && !cloudError) || localLoading || builtinLoading || kprofilesLoading; const canSave = selectedPresetId && !configureMutation.isPending; // Get display color (custom or slot default) @@ -703,6 +746,11 @@ export function ConfigureAmsSlotModal({ {t('profiles.localProfiles.badge')} )} + {preset.source === 'builtin' && ( + + {t('configureAmsSlot.builtin')} + + )} {preset.isUser && ( {t('configureAmsSlot.custom')} diff --git a/frontend/src/components/FilamentHoverCard.tsx b/frontend/src/components/FilamentHoverCard.tsx index 9b074245a..f1adb160c 100644 --- a/frontend/src/components/FilamentHoverCard.tsx +++ b/frontend/src/components/FilamentHoverCard.tsx @@ -1,6 +1,6 @@ import { useState, useRef, useEffect, type ReactNode } from 'react'; import { useTranslation } from 'react-i18next'; -import { Droplets, Link2, Copy, Check, Settings2, ExternalLink } from 'lucide-react'; +import { Droplets, Link2, Copy, Check, Settings2, ExternalLink, Package, Unlink } from 'lucide-react'; interface FilamentData { vendor: 'Bambu Lab' | 'Generic'; @@ -10,7 +10,7 @@ interface FilamentData { kFactor: string; fillLevel: number | null; // null = unknown trayUuid?: string | null; // Bambu Lab spool UUID for Spoolman linking - fillSource?: 'ams' | 'spoolman'; // Source of fill level data + fillSource?: 'ams' | 'spoolman' | 'inventory'; // Source of fill level data } interface SpoolmanConfig { @@ -21,6 +21,12 @@ interface SpoolmanConfig { spoolmanUrl?: string | null; // Base URL for Spoolman (for "Open in Spoolman" link) } +interface InventoryConfig { + onAssignSpool?: () => void; + onUnassignSpool?: () => void; + assignedSpool?: { id: number; material: string; brand: string | null; color_name: string | null } | null; +} + interface ConfigureSlotConfig { enabled: boolean; onConfigure?: () => void; @@ -32,6 +38,7 @@ interface FilamentHoverCardProps { disabled?: boolean; className?: string; spoolman?: SpoolmanConfig; + inventory?: InventoryConfig; configureSlot?: ConfigureSlotConfig; } @@ -39,7 +46,7 @@ interface FilamentHoverCardProps { * A hover card that displays filament details when hovering over AMS slots. * Replaces the basic browser tooltip with a styled popover. */ -export function FilamentHoverCard({ data, children, disabled, className = '', spoolman, configureSlot }: FilamentHoverCardProps) { +export function FilamentHoverCard({ data, children, disabled, className = '', spoolman, inventory, configureSlot }: FilamentHoverCardProps) { const { t } = useTranslation(); const [isVisible, setIsVisible] = useState(false); const [position, setPosition] = useState<'top' | 'bottom'>('top'); @@ -235,6 +242,9 @@ export function FilamentHoverCard({ data, children, disabled, className = '', sp {data.fillSource === 'spoolman' && data.fillLevel !== null && ( {t('spoolman.fillSourceLabel')} )} + {data.fillSource === 'inventory' && data.fillLevel !== null && ( + {t('inventory.fillSourceLabel')} + )}
{/* Fill bar */} @@ -319,6 +329,50 @@ export function FilamentHoverCard({ data, children, disabled, className = '', sp
)} + {/* Inventory section - only for non-Bambu spools */} + {inventory && data.vendor !== 'Bambu Lab' && ( +
+ {inventory.assignedSpool ? ( + <> +
+ + + {t('inventory.assigned')} + +
+

+ {inventory.assignedSpool.brand ? `${inventory.assignedSpool.brand} ` : ''} + {inventory.assignedSpool.material} + {inventory.assignedSpool.color_name ? ` - ${inventory.assignedSpool.color_name}` : ''} +

+ {inventory.onUnassignSpool && ( + + )} + + ) : inventory.onAssignSpool ? ( + + ) : null} +
+ )} + {/* Configure slot section - always show if enabled */} {configureSlot?.enabled && (
diff --git a/frontend/src/components/Layout.tsx b/frontend/src/components/Layout.tsx index 477b87355..4ede5f234 100644 --- a/frontend/src/components/Layout.tsx +++ b/frontend/src/components/Layout.tsx @@ -1,6 +1,6 @@ import { useState, useEffect, useCallback, useRef, useMemo } from 'react'; import { NavLink, Outlet, useNavigate, useLocation } from 'react-router-dom'; -import { Printer, Archive, Calendar, BarChart3, Cloud, Settings, Sun, Moon, ChevronLeft, ChevronRight, Keyboard, Github, GripVertical, ArrowUpCircle, Wrench, FolderKanban, FolderOpen, X, Menu, Info, Plug, Bug, LogOut, Key, Loader2, type LucideIcon } from 'lucide-react'; +import { Printer, Archive, Calendar, BarChart3, Cloud, Settings, Sun, Moon, ChevronLeft, ChevronRight, Keyboard, Github, GripVertical, ArrowUpCircle, Wrench, FolderKanban, FolderOpen, X, Menu, Info, Plug, Bug, LogOut, Key, Loader2, Package, type LucideIcon } from 'lucide-react'; import { useTranslation } from 'react-i18next'; import { useTheme } from '../contexts/ThemeContext'; import { KeyboardShortcutsModal } from './KeyboardShortcutsModal'; @@ -29,6 +29,7 @@ export const defaultNavItems: NavItem[] = [ { id: 'profiles', to: '/profiles', icon: Cloud, labelKey: 'nav.profiles' }, { id: 'maintenance', to: '/maintenance', icon: Wrench, labelKey: 'nav.maintenance' }, { id: 'projects', to: '/projects', icon: FolderKanban, labelKey: 'nav.projects' }, + { id: 'inventory', to: '/inventory', icon: Package, labelKey: 'nav.inventory' }, { id: 'files', to: '/files', icon: FolderOpen, labelKey: 'nav.files' }, { id: 'settings', to: '/settings', icon: Settings, labelKey: 'nav.settings' }, ]; @@ -118,6 +119,13 @@ export function Layout() { refetchInterval: 60 * 60 * 1000, // Check every hour }); + // Fetch Spoolman settings to determine if inventory should be hidden + const { data: spoolmanSettings } = useQuery({ + queryKey: ['spoolman-settings'], + queryFn: api.getSpoolmanSettings, + staleTime: 5 * 60 * 1000, + }); + // Fetch external links for sidebar const { data: externalLinks } = useQuery({ queryKey: ['external-links'], @@ -189,13 +197,13 @@ export function Layout() { // Determine if settings should be hidden (user role and auth enabled) const hideSettings = authEnabled && user?.role === 'user'; + // Hide inventory when Spoolman mode is active + const hideInventory = spoolmanSettings?.spoolman_enabled === 'true'; // Add items in stored order for (const id of sidebarOrder) { - // Skip settings if user is not admin - if (hideSettings && id === 'settings') { - continue; - } + if (hideSettings && id === 'settings') continue; + if (hideInventory && id === 'inventory') continue; if (navItemsMap.has(id) || extLinksMap.has(id)) { result.push(id); seen.add(id); @@ -204,10 +212,8 @@ export function Layout() { // Add any new internal nav items not in stored order for (const item of defaultNavItems) { - // Skip settings if user is not admin - if (hideSettings && item.id === 'settings') { - continue; - } + if (hideSettings && item.id === 'settings') continue; + if (hideInventory && item.id === 'inventory') continue; if (!seen.has(item.id)) { result.push(item.id); seen.add(item.id); @@ -347,9 +353,14 @@ export function Layout() { e.preventDefault(); if (isExternalLinkId(id)) { - // External link - navigate to iframe page - const linkId = id.replace('ext-', ''); - navigate(`/external/${linkId}`); + // External link + const extLink = extLinksMap.get(id); + if (extLink?.open_in_new_tab) { + window.open(extLink.url, '_blank', 'noopener,noreferrer'); + } else { + const linkId = id.replace('ext-', ''); + navigate(`/external/${linkId}`); + } } else { // Internal nav item const navItem = navItemsMap.get(id); @@ -370,7 +381,7 @@ export function Layout() { break; } } - }, [navigate, orderedSidebarIds, navItemsMap]); + }, [navigate, orderedSidebarIds, navItemsMap, extLinksMap]); useEffect(() => { document.addEventListener('keydown', handleKeyDown); @@ -451,31 +462,55 @@ export function Layout() { : '' }`} > - - `flex items-center ${isMobile || sidebarExpanded ? 'gap-3 px-4' : 'justify-center px-2'} py-3 rounded-lg transition-colors group ${ - isActive - ? 'bg-bambu-green text-white' - : 'text-bambu-gray-light hover:bg-bambu-dark-tertiary hover:text-white' - }` - } - title={!isMobile && !sidebarExpanded ? link.name : undefined} - > - {sidebarExpanded && !isMobile && ( - - )} - {link.custom_icon ? ( - - ) : ( - LinkIcon && - )} - {(isMobile || sidebarExpanded) && {link.name}} - + {link.open_in_new_tab ? ( + + {sidebarExpanded && !isMobile && ( + + )} + {link.custom_icon ? ( + + ) : ( + LinkIcon && + )} + {(isMobile || sidebarExpanded) && {link.name}} + + ) : ( + + `flex items-center ${isMobile || sidebarExpanded ? 'gap-3 px-4' : 'justify-center px-2'} py-3 rounded-lg transition-colors group ${ + isActive + ? 'bg-bambu-green text-white' + : 'text-bambu-gray-light hover:bg-bambu-dark-tertiary hover:text-white' + }` + } + title={!isMobile && !sidebarExpanded ? link.name : undefined} + > + {sidebarExpanded && !isMobile && ( + + )} + {link.custom_icon ? ( + + ) : ( + LinkIcon && + )} + {(isMobile || sidebarExpanded) && {link.name}} + + )} ); } else { diff --git a/frontend/src/components/LinkSpoolModal.tsx b/frontend/src/components/LinkSpoolModal.tsx index 4be13430d..cd5eaa4d9 100644 --- a/frontend/src/components/LinkSpoolModal.tsx +++ b/frontend/src/components/LinkSpoolModal.tsx @@ -1,190 +1,154 @@ -import { useState } from 'react'; +import { useState, useMemo } from 'react'; import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query'; import { useTranslation } from 'react-i18next'; -import { X, Loader2, Link2, Check } from 'lucide-react'; +import { X, Loader2, Search, Link } from 'lucide-react'; import { api } from '../api/client'; +import type { InventorySpool } from '../api/client'; import { Button } from './Button'; import { useToast } from '../contexts/ToastContext'; interface LinkSpoolModalProps { isOpen: boolean; onClose: () => void; + tagUid: string; trayUuid: string; - trayInfo?: { - type: string; - color: string; - location: string; - }; + printerId: number; + amsId: number; + trayId: number; } -export function LinkSpoolModal({ isOpen, onClose, trayUuid, trayInfo }: LinkSpoolModalProps) { +export function LinkSpoolModal({ isOpen, onClose, tagUid, trayUuid, printerId, amsId, trayId }: LinkSpoolModalProps) { const { t } = useTranslation(); const queryClient = useQueryClient(); const { showToast } = useToast(); - const [selectedSpoolId, setSelectedSpoolId] = useState(null); + const [search, setSearch] = useState(''); - // Fetch unlinked spools - const { data: unlinkedSpools, isLoading } = useQuery({ - queryKey: ['unlinked-spools'], - queryFn: api.getUnlinkedSpools, + const { data: spools, isLoading } = useQuery({ + queryKey: ['inventory-spools'], + queryFn: () => api.getSpools(false), enabled: isOpen, }); - // Link mutation + // Filter to untagged spools matching search + const filteredSpools = useMemo(() => { + if (!spools) return []; + return spools.filter((s: InventorySpool) => { + if (s.tag_uid || s.tray_uuid) return false; // Already tagged + if (!search) return true; + const q = search.toLowerCase(); + return ( + s.material.toLowerCase().includes(q) || + (s.brand && s.brand.toLowerCase().includes(q)) || + (s.color_name && s.color_name.toLowerCase().includes(q)) + ); + }); + }, [spools, search]); + const linkMutation = useMutation({ - mutationFn: (spoolId: number) => api.linkSpool(spoolId, trayUuid), + mutationFn: (spoolId: number) => + api.linkTagToSpool(spoolId, { + tag_uid: tagUid || undefined, + tray_uuid: trayUuid || undefined, + tag_type: trayUuid ? 'bambulab' : 'generic', + data_origin: 'nfc_link', + }), onSuccess: () => { - queryClient.invalidateQueries({ queryKey: ['unlinked-spools'] }); - queryClient.invalidateQueries({ queryKey: ['linked-spools'] }); - queryClient.invalidateQueries({ queryKey: ['spoolman-status'] }); - showToast(t('spoolman.linkSuccess'), 'success'); + queryClient.invalidateQueries({ queryKey: ['inventory-spools'] }); + queryClient.invalidateQueries({ queryKey: ['inventory-assignments'] }); + showToast(t('inventory.tagLinked'), 'success'); onClose(); }, - onError: (error: Error) => { - showToast(`${t('spoolman.linkFailed')}: ${error.message}`, 'error'); + onError: (err: Error) => { + showToast(err.message || t('inventory.tagLinkFailed'), 'error'); }, }); if (!isOpen) return null; - const handleLink = () => { - if (selectedSpoolId) { - linkMutation.mutate(selectedSpoolId); - } - }; - return (
- {/* Backdrop */} -
- - {/* Modal */} -
+
+
{/* Header */} -
-
- -

{t('spoolman.linkToSpoolman')}

+
+
+

+ + {t('inventory.linkToSpool')} +

+

+ AMS {amsId} T{trayId} · Printer #{printerId} +

-
- {/* Content */} -
- {/* Tray info */} - {trayInfo && ( -
-

Linking AMS tray:

-
- {trayInfo.color && ( - - )} - {trayInfo.type} - ({trayInfo.location}) -
-
- )} - - {/* Spool UUID */} -
-

{t('spoolman.spoolId')}:

- {trayUuid} + {/* Search */} +
+
+ + setSearch(e.target.value)} + placeholder={t('inventory.searchSpools')} + className="w-full pl-9 pr-3 py-2 bg-bambu-dark rounded-lg border border-white/10 text-white text-sm placeholder:text-bambu-gray focus:outline-none focus:border-bambu-green" + />
- - {/* Spool list */} -
-

- {t('spoolman.selectSpool')}: + {(tagUid || trayUuid) && ( +

+ Tag: {tagUid || trayUuid}

+ )} +
- {isLoading ? ( -
- -
- ) : unlinkedSpools && unlinkedSpools.length > 0 ? ( -
- {unlinkedSpools.map((spool) => ( - - ))} -
- ) : ( -
-

{t('spoolman.noUnlinkedSpools')}

-
- )} -
+ {/* Spool List */} +
+ {isLoading ? ( +
+ +
+ ) : filteredSpools.length === 0 ? ( +

+ {t('inventory.noSpoolsMatch')} +

+ ) : ( + filteredSpools.map((spool: InventorySpool) => ( + + )) + )}
{/* Footer */} -
- -
- - {/* Error */} - {linkMutation.isError && ( -
- {(linkMutation.error as Error).message} -
- )}
); diff --git a/frontend/src/components/PrinterQueueWidget.tsx b/frontend/src/components/PrinterQueueWidget.tsx index 687adb8cd..d7b34d240 100644 --- a/frontend/src/components/PrinterQueueWidget.tsx +++ b/frontend/src/components/PrinterQueueWidget.tsx @@ -1,11 +1,15 @@ -import { useQuery } from '@tanstack/react-query'; -import { Clock, Calendar, ChevronRight } from 'lucide-react'; +import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query'; +import { Clock, Calendar, ChevronRight, Loader2, CircleCheck } from 'lucide-react'; import { Link } from 'react-router-dom'; +import { useTranslation } from 'react-i18next'; import { api } from '../api/client'; +import { useAuth } from '../contexts/AuthContext'; +import { useToast } from '../contexts/ToastContext'; import { parseUTCDate } from '../utils/date'; interface PrinterQueueWidgetProps { printerId: number; + printerState?: string | null; } function formatRelativeTime(dateString: string | null): string { @@ -22,13 +26,29 @@ function formatRelativeTime(dateString: string | null): string { return date.toLocaleDateString(); } -export function PrinterQueueWidget({ printerId }: PrinterQueueWidgetProps) { +export function PrinterQueueWidget({ printerId, printerState }: PrinterQueueWidgetProps) { + const { t } = useTranslation(); + const queryClient = useQueryClient(); + const { showToast } = useToast(); + const { hasPermission } = useAuth(); const { data: queue } = useQuery({ queryKey: ['queue', printerId, 'pending'], queryFn: () => api.getQueue(printerId, 'pending'), refetchInterval: 30000, }); + const clearPlateMutation = useMutation({ + mutationFn: () => api.clearPlate(printerId), + onSuccess: () => { + queryClient.invalidateQueries({ queryKey: ['queue', printerId] }); + queryClient.invalidateQueries({ queryKey: ['printerStatus', printerId] }); + showToast(t('queue.clearPlateSuccess'), 'success'); + }, + onError: (err: Error) => { + showToast(err.message, 'error'); + }, + }); + const nextItem = queue?.[0]; const totalPending = queue?.length || 0; @@ -36,6 +56,48 @@ export function PrinterQueueWidget({ printerId }: PrinterQueueWidgetProps) { return null; } + const needsClearPlate = printerState === 'FINISH' || printerState === 'FAILED'; + + if (needsClearPlate) { + return ( +
+
+ +
+

{t('queue.nextInQueue')}

+

+ {nextItem?.archive_name || `Archive #${nextItem?.archive_id}`} +

+
+ {totalPending > 1 && ( + + +{totalPending - 1} + + )} +
+ {clearPlateMutation.isSuccess ? ( +
+ + {t('queue.plateReady')} +
+ ) : ( + + )} +
+ ); + } + return (
-

Next in queue

+

{t('queue.nextInQueue')}

{nextItem?.archive_name || `Archive #${nextItem?.archive_id}`}

diff --git a/frontend/src/components/SpoolCatalogSettings.tsx b/frontend/src/components/SpoolCatalogSettings.tsx new file mode 100644 index 000000000..9cecee779 --- /dev/null +++ b/frontend/src/components/SpoolCatalogSettings.tsx @@ -0,0 +1,397 @@ +import { useState, useEffect, useCallback, useRef } from 'react'; +import { useTranslation } from 'react-i18next'; +import { Database, Plus, Trash2, RotateCcw, Loader2, Pencil, Check, X, Search, Download, Upload } from 'lucide-react'; +import { api } from '../api/client'; +import type { SpoolCatalogEntry } from '../api/client'; +import { useToast } from '../contexts/ToastContext'; +import { Card, CardHeader, CardContent } from './Card'; +import { ConfirmModal } from './ConfirmModal'; + +export function SpoolCatalogSettings() { + const { t } = useTranslation(); + const { showToast } = useToast(); + const [catalog, setCatalog] = useState([]); + const [loading, setLoading] = useState(true); + const [search, setSearch] = useState(''); + const fileInputRef = useRef(null); + + // Add/Edit form state + const [showAddForm, setShowAddForm] = useState(false); + const [editingId, setEditingId] = useState(null); + const [formName, setFormName] = useState(''); + const [formWeight, setFormWeight] = useState(''); + const [saving, setSaving] = useState(false); + + // Confirmation modals + const [deleteEntry, setDeleteEntry] = useState(null); + const [showResetConfirm, setShowResetConfirm] = useState(false); + + const loadCatalog = useCallback(async () => { + try { + const entries = await api.getSpoolCatalog(); + setCatalog(entries); + } catch { + showToast(t('settings.catalog.loadFailed'), 'error'); + } finally { + setLoading(false); + } + }, [showToast, t]); + + useEffect(() => { + loadCatalog(); + }, [loadCatalog]); + + const filteredCatalog = catalog.filter(entry => + entry.name.toLowerCase().includes(search.toLowerCase()) + ); + + const handleAdd = async () => { + if (!formName.trim() || !formWeight) { + showToast(t('settings.catalog.nameWeightRequired'), 'error'); + return; + } + setSaving(true); + try { + const entry = await api.addCatalogEntry({ name: formName.trim(), weight: parseInt(formWeight) }); + setCatalog(prev => [...prev, entry].sort((a, b) => a.name.localeCompare(b.name))); + setShowAddForm(false); + setFormName(''); + setFormWeight(''); + showToast(t('settings.catalog.entryAdded'), 'success'); + } catch { + showToast(t('settings.catalog.addFailed'), 'error'); + } finally { + setSaving(false); + } + }; + + const startEdit = (entry: SpoolCatalogEntry) => { + setEditingId(entry.id); + setFormName(entry.name); + setFormWeight(entry.weight.toString()); + }; + + const cancelEdit = () => { + setEditingId(null); + setFormName(''); + setFormWeight(''); + }; + + const handleUpdate = async (id: number) => { + if (!formName.trim() || !formWeight) { + showToast(t('settings.catalog.nameWeightRequired'), 'error'); + return; + } + setSaving(true); + try { + const updated = await api.updateCatalogEntry(id, { name: formName.trim(), weight: parseInt(formWeight) }); + setCatalog(prev => prev.map(e => e.id === id ? updated : e).sort((a, b) => a.name.localeCompare(b.name))); + setEditingId(null); + setFormName(''); + setFormWeight(''); + showToast(t('settings.catalog.entryUpdated'), 'success'); + } catch { + showToast(t('settings.catalog.updateFailed'), 'error'); + } finally { + setSaving(false); + } + }; + + const handleDelete = async () => { + if (!deleteEntry) return; + try { + await api.deleteCatalogEntry(deleteEntry.id); + setCatalog(prev => prev.filter(e => e.id !== deleteEntry.id)); + showToast(t('settings.catalog.entryDeleted'), 'success'); + } catch { + showToast(t('settings.catalog.deleteFailed'), 'error'); + } finally { + setDeleteEntry(null); + } + }; + + const handleReset = async () => { + setShowResetConfirm(false); + setLoading(true); + try { + await api.resetSpoolCatalog(); + await loadCatalog(); + showToast(t('settings.catalog.resetSuccess'), 'success'); + } catch { + showToast(t('settings.catalog.resetFailed'), 'error'); + setLoading(false); + } + }; + + const handleExport = () => { + const exportData = catalog.map(({ name, weight }) => ({ name, weight })); + const blob = new Blob([JSON.stringify(exportData, null, 2)], { type: 'application/json' }); + const url = URL.createObjectURL(blob); + const a = document.createElement('a'); + a.href = url; + a.download = 'spool-catalog.json'; + document.body.appendChild(a); + a.click(); + document.body.removeChild(a); + URL.revokeObjectURL(url); + showToast(t('settings.catalog.exported', { count: catalog.length }), 'success'); + }; + + const handleImport = async (e: React.ChangeEvent) => { + const file = e.target.files?.[0]; + if (!file) return; + try { + const text = await file.text(); + const data = JSON.parse(text) as Array<{ name: string; weight: number }>; + if (!Array.isArray(data)) throw new Error('Invalid format'); + + let added = 0; + let skipped = 0; + for (const item of data) { + if (!item.name || typeof item.weight !== 'number') { skipped++; continue; } + const exists = catalog.some(c => c.name.toLowerCase() === item.name.toLowerCase()); + if (exists) { skipped++; continue; } + try { + const entry = await api.addCatalogEntry({ name: item.name, weight: item.weight }); + setCatalog(prev => [...prev, entry].sort((a, b) => a.name.localeCompare(b.name))); + added++; + } catch { skipped++; } + } + showToast(t('settings.catalog.imported', { added, skipped }), 'success'); + } catch { + showToast(t('settings.catalog.importFailed'), 'error'); + } + if (fileInputRef.current) fileInputRef.current.value = ''; + }; + + return ( + + +
+ +

{t('settings.catalog.spoolCatalog')}

+ ({catalog.length}) +
+
+ + + + + +
+
+ +

+ {t('settings.catalog.spoolCatalogDescription')} +

+ + {/* Search */} +
+ + setSearch(e.target.value)} + /> +
+ + {/* Add form */} + {showAddForm && ( +
+

{t('settings.catalog.addNewEntry')}

+
+
+ setFormName(e.target.value)} + /> +
+ setFormWeight(e.target.value)} + /> + g + + +
+
+ )} + + {/* Catalog list */} + {loading ? ( +
+ + {t('common.loading')} +
+ ) : ( +
+ + + + + + + + + + + {filteredCatalog.length === 0 ? ( + + + + ) : ( + filteredCatalog.map(entry => ( + + {editingId === entry.id ? ( + <> + + + + + + ) : ( + <> + + + + + + )} + + )) + )} + +
{t('common.name')}{t('settings.catalog.weight')}{t('settings.catalog.type')}
+ {search ? t('settings.catalog.noMatch') : t('settings.catalog.empty')} +
+ setFormName(e.target.value)} + /> + + setFormWeight(e.target.value)} + /> + + - + +
+ + +
+
{entry.name}{entry.weight}g + {entry.is_default ? ( + + {t('settings.catalog.default')} + + ) : ( + + {t('settings.catalog.custom')} + + )} + +
+ + +
+
+
+ )} +
+ + {/* Delete confirmation */} + {deleteEntry && ( + setDeleteEntry(null)} + /> + )} + + {/* Reset confirmation */} + {showResetConfirm && ( + setShowResetConfirm(false)} + /> + )} +
+ ); +} diff --git a/frontend/src/components/SpoolFormModal.tsx b/frontend/src/components/SpoolFormModal.tsx new file mode 100644 index 000000000..b2d5b9710 --- /dev/null +++ b/frontend/src/components/SpoolFormModal.tsx @@ -0,0 +1,504 @@ +import { useState, useEffect, useMemo } from 'react'; +import { useMutation, useQueryClient } from '@tanstack/react-query'; +import { useTranslation } from 'react-i18next'; +import { X, Loader2, Save, Beaker, Palette } from 'lucide-react'; +import { api } from '../api/client'; +import type { InventorySpool, SlicerSetting, SpoolCatalogEntry, LocalPreset } from '../api/client'; +import { Button } from './Button'; +import { useToast } from '../contexts/ToastContext'; +import type { SpoolFormData, PrinterWithCalibrations, ColorPreset } from './spool-form/types'; +import { defaultFormData, validateForm } from './spool-form/types'; +import { buildFilamentOptions, extractBrandsFromPresets, findPresetOption, loadRecentColors, saveRecentColor } from './spool-form/utils'; +import { FilamentSection } from './spool-form/FilamentSection'; +import { ColorSection } from './spool-form/ColorSection'; +import { AdditionalSection } from './spool-form/AdditionalSection'; +import { PAProfileSection } from './spool-form/PAProfileSection'; +import { SpoolUsageHistory } from './SpoolUsageHistory'; + +type TabId = 'filament' | 'pa-profile'; + +interface SpoolFormModalProps { + isOpen: boolean; + onClose: () => void; + spool?: InventorySpool | null; + printersWithCalibrations?: PrinterWithCalibrations[]; +} + +export function SpoolFormModal({ isOpen, onClose, spool, printersWithCalibrations = [] }: SpoolFormModalProps) { + const { t } = useTranslation(); + const queryClient = useQueryClient(); + const { showToast } = useToast(); + + const isEditing = !!spool; + + // Form state + const [formData, setFormData] = useState(defaultFormData); + const [errors, setErrors] = useState>>({}); + const [activeTab, setActiveTab] = useState('filament'); + + // Cloud presets + const [cloudAuthenticated, setCloudAuthenticated] = useState(false); + const [loadingCloudPresets, setLoadingCloudPresets] = useState(false); + const [cloudPresets, setCloudPresets] = useState([]); + const [presetInputValue, setPresetInputValue] = useState(''); + + // Spool catalog + const [spoolCatalog, setSpoolCatalog] = useState([]); + + // Local presets (OrcaSlicer imports) + const [localPresets, setLocalPresets] = useState([]); + + // Color catalog + const [colorCatalog, setColorCatalog] = useState<{ manufacturer: string; color_name: string; hex_color: string; material: string | null }[]>([]); + + // Color state + const [recentColors, setRecentColors] = useState([]); + + // PA Profile state + const [fetchedCalibrations, setFetchedCalibrations] = useState([]); + const [selectedProfiles, setSelectedProfiles] = useState>(new Set()); + const [expandedPrinters, setExpandedPrinters] = useState>(new Set()); + + // Use prop if provided, otherwise use self-fetched data + const resolvedCalibrations = printersWithCalibrations.length > 0 + ? printersWithCalibrations + : fetchedCalibrations; + + // Count selected PA profiles for tab badge + const selectedProfileCount = useMemo(() => { + return selectedProfiles.size; + }, [selectedProfiles]); + + // Load recent colors on mount + useEffect(() => { + setRecentColors(loadRecentColors()); + }, []); + + // Fetch cloud presets and catalog when modal opens + useEffect(() => { + if (isOpen) { + const fetchData = async () => { + setLoadingCloudPresets(true); + try { + const status = await api.getCloudStatus(); + setCloudAuthenticated(status.is_authenticated); + if (status.is_authenticated) { + const presets = await api.getFilamentPresets(); + setCloudPresets(presets); + } + } catch (e) { + console.error('Failed to fetch cloud presets:', e); + setCloudAuthenticated(false); + } finally { + setLoadingCloudPresets(false); + } + }; + fetchData(); + api.getSpoolCatalog().then(setSpoolCatalog).catch(console.error); + api.getColorCatalog().then(setColorCatalog).catch(console.error); + api.getLocalPresets().then(r => setLocalPresets(r.filament)).catch(console.error); + + // Fetch printer calibrations if not provided via props + if (printersWithCalibrations.length === 0) { + (async () => { + try { + const printers = await api.getPrinters(); + const statuses = await Promise.all( + printers.map(p => api.getPrinterStatus(p.id).catch(() => null)), + ); + const results: PrinterWithCalibrations[] = []; + for (let i = 0; i < printers.length; i++) { + const printer = printers[i]; + const status = statuses[i]; + const connected = status?.connected ?? false; + let calibrations: PrinterWithCalibrations['calibrations'] = []; + if (connected) { + try { + const kRes = await api.getKProfiles(printer.id); + calibrations = kRes.profiles.map(p => ({ + cali_idx: p.slot_id, + filament_id: p.filament_id, + setting_id: p.setting_id || '', + name: p.name, + k_value: parseFloat(p.k_value) || 0, + n_coef: parseFloat(p.n_coef) || 0, + extruder_id: p.extruder_id, + nozzle_diameter: p.nozzle_diameter, + })); + } catch { + // Printer may not support K-profiles + } + } + results.push({ printer: { ...printer, connected }, calibrations }); + } + setFetchedCalibrations(results); + } catch (e) { + console.error('Failed to fetch printer calibrations:', e); + } + })(); + } + } + }, [isOpen, printersWithCalibrations.length]); + + // Build filament options: cloud → local → fallback + const filamentOptions = useMemo( + () => buildFilamentOptions(cloudPresets, new Set(), localPresets), + [cloudPresets, localPresets], + ); + + // Extract brands from presets + const availableBrands = useMemo( + () => extractBrandsFromPresets(cloudPresets, localPresets), + [cloudPresets, localPresets], + ); + + // Find selected preset option + const selectedPresetOption = useMemo( + () => findPresetOption(formData.slicer_filament, filamentOptions), + [formData.slicer_filament, filamentOptions], + ); + + // Reset form when modal opens/closes or spool changes + useEffect(() => { + if (isOpen) { + if (spool) { + setFormData({ + material: spool.material || '', + subtype: spool.subtype || '', + brand: spool.brand || '', + color_name: spool.color_name || '', + rgba: spool.rgba || '808080FF', + label_weight: spool.label_weight || 1000, + core_weight: spool.core_weight || 250, + weight_used: spool.weight_used || 0, + slicer_filament: spool.slicer_filament || '', + note: spool.note || '', + }); + setPresetInputValue(spool.slicer_filament_name || spool.slicer_filament || ''); + + // Load K-profiles for this spool + if (spool.k_profiles && spool.k_profiles.length > 0) { + const profileKeys = new Set(); + for (const p of spool.k_profiles) { + if (p.cali_idx !== null && p.cali_idx !== undefined) { + profileKeys.add(`${p.printer_id}:${p.cali_idx}:${p.extruder ?? 'null'}`); + } + } + setSelectedProfiles(profileKeys); + } else { + setSelectedProfiles(new Set()); + } + } else { + setFormData(defaultFormData); + setPresetInputValue(''); + setSelectedProfiles(new Set()); + } + setErrors({}); + setActiveTab('filament'); + } + }, [isOpen, spool]); + + // Expand all printers in PA profile section when calibrations are available + useEffect(() => { + if (isOpen && resolvedCalibrations.length > 0) { + setExpandedPrinters(new Set(resolvedCalibrations.map(p => String(p.printer.id)))); + } + }, [isOpen, resolvedCalibrations]); + + // Update field helper + const updateField = (key: K, value: SpoolFormData[K]) => { + setFormData(prev => ({ ...prev, [key]: value })); + if (errors[key]) { + setErrors(prev => ({ ...prev, [key]: undefined })); + } + }; + + // Handle color selection + const handleColorUsed = (color: ColorPreset) => { + setRecentColors(prev => saveRecentColor(color, prev)); + }; + + // Mutations + const createMutation = useMutation({ + mutationFn: (data: Record) => + api.createSpool(data as Parameters[0]), + onSuccess: async (newSpool) => { + // Save K-profiles if any selected + if (selectedProfiles.size > 0 && newSpool?.id) { + await saveKProfiles(newSpool.id); + } + await queryClient.invalidateQueries({ queryKey: ['inventory-spools'] }); + showToast(t('inventory.spoolCreated'), 'success'); + onClose(); + }, + onError: (error: Error) => { + showToast(error.message, 'error'); + }, + }); + + const updateMutation = useMutation({ + mutationFn: (data: Record) => + api.updateSpool(spool!.id, data as Parameters[1]), + onSuccess: async () => { + // Save K-profiles + if (spool?.id) { + await saveKProfiles(spool.id); + } + await queryClient.invalidateQueries({ queryKey: ['inventory-spools'] }); + showToast(t('inventory.spoolUpdated'), 'success'); + onClose(); + }, + onError: (error: Error) => { + showToast(error.message, 'error'); + }, + }); + + // Save K-profiles for selected calibrations + const saveKProfiles = async (spoolId: number) => { + if (selectedProfiles.size === 0) { + // Clear existing K-profiles + try { + await api.saveSpoolKProfiles(spoolId, []); + } catch { + // Ignore + } + return; + } + + const profiles = []; + for (const key of selectedProfiles) { + const [printerIdStr, caliIdxStr, extruderStr] = key.split(':'); + const printerId = parseInt(printerIdStr); + const caliIdx = parseInt(caliIdxStr); + const extruder = extruderStr === 'null' ? 0 : parseInt(extruderStr); + + // Find the matching calibration + const pc = resolvedCalibrations.find(p => p.printer.id === printerId); + if (pc) { + const cal = pc.calibrations.find(c => c.cali_idx === caliIdx); + if (cal) { + profiles.push({ + printer_id: printerId, + extruder, + nozzle_diameter: cal.nozzle_diameter || '0.4', + k_value: cal.k_value, + name: cal.name || null, + cali_idx: cal.cali_idx, + setting_id: cal.setting_id || null, + }); + } + } + } + + if (profiles.length > 0) { + try { + await api.saveSpoolKProfiles(spoolId, profiles); + } catch (e) { + console.error('Failed to save K-profiles:', e); + } + } + }; + + // Close on Escape key + useEffect(() => { + if (!isOpen) return; + const handleKeyDown = (e: KeyboardEvent) => { + if (e.key === 'Escape') onClose(); + }; + document.addEventListener('keydown', handleKeyDown); + return () => document.removeEventListener('keydown', handleKeyDown); + }, [isOpen, onClose]); + + if (!isOpen) return null; + + const handleSubmit = () => { + const validation = validateForm(formData); + if (!validation.isValid) { + setErrors(validation.errors); + // Switch to filament tab if there are errors there + if (validation.errors.slicer_filament || validation.errors.material) { + setActiveTab('filament'); + } + return; + } + + // Find preset name from selected option + const presetName = selectedPresetOption?.displayName || presetInputValue || null; + + const data: Record = { + material: formData.material, + subtype: formData.subtype || null, + brand: formData.brand || null, + color_name: formData.color_name || null, + rgba: formData.rgba || null, + label_weight: formData.label_weight, + core_weight: formData.core_weight, + weight_used: formData.weight_used, + slicer_filament: formData.slicer_filament || null, + slicer_filament_name: presetName, + nozzle_temp_min: null, + nozzle_temp_max: null, + note: formData.note || null, + }; + + if (isEditing) { + updateMutation.mutate(data); + } else { + createMutation.mutate(data); + } + }; + + const isPending = createMutation.isPending || updateMutation.isPending; + + return ( +
+
+ +
+ {/* Header */} +
+

+ {isEditing ? t('inventory.editSpool') : t('inventory.addSpool')} +

+ +
+ + {/* Tabs */} +
+ + +
+ + {/* Content */} +
+ {activeTab === 'filament' ? ( +
+ {/* Filament Info Section */} +
+

+ {t('inventory.filamentInfo')} +

+ + {errors.slicer_filament && ( +

{errors.slicer_filament}

+ )} + {errors.material && ( +

{errors.material}

+ )} +
+ + {/* Color Section */} +
+

+ {t('inventory.color')} +

+ +
+ + {/* Additional Section */} +
+

+ {t('inventory.additional')} +

+ +
+ + {/* Usage History (only when editing) */} + {isEditing && spool && ( +
+ +
+ )} +
+ ) : ( + + )} +
+ + {/* Footer */} +
+ + +
+
+
+ ); +} diff --git a/frontend/src/components/SpoolUsageHistory.tsx b/frontend/src/components/SpoolUsageHistory.tsx new file mode 100644 index 000000000..594338538 --- /dev/null +++ b/frontend/src/components/SpoolUsageHistory.tsx @@ -0,0 +1,101 @@ +import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query'; +import { useTranslation } from 'react-i18next'; +import { Loader2, Trash2, Clock } from 'lucide-react'; +import { api } from '../api/client'; +import type { SpoolUsageRecord } from '../api/client'; +import { Button } from './Button'; +import { useToast } from '../contexts/ToastContext'; + +interface SpoolUsageHistoryProps { + spoolId: number; +} + +function formatDate(dateStr: string): string { + const date = new Date(dateStr); + return date.toLocaleDateString('en-GB', { day: '2-digit', month: '2-digit', year: '2-digit' }) + + ' ' + date.toLocaleTimeString('en-GB', { hour: '2-digit', minute: '2-digit' }); +} + +const STATUS_COLORS: Record = { + completed: 'text-bambu-green', + failed: 'text-red-400', + aborted: 'text-yellow-400', +}; + +export function SpoolUsageHistory({ spoolId }: SpoolUsageHistoryProps) { + const { t } = useTranslation(); + const queryClient = useQueryClient(); + const { showToast } = useToast(); + + const { data: history, isLoading } = useQuery({ + queryKey: ['spool-usage', spoolId], + queryFn: () => api.getSpoolUsageHistory(spoolId), + }); + + const clearMutation = useMutation({ + mutationFn: () => api.clearSpoolUsageHistory(spoolId), + onSuccess: () => { + queryClient.invalidateQueries({ queryKey: ['spool-usage', spoolId] }); + showToast(t('inventory.historyCleared'), 'success'); + }, + }); + + if (isLoading) { + return ( +
+ +
+ ); + } + + if (!history || history.length === 0) { + return ( +
+ + {t('inventory.noUsageHistory')} +
+ ); + } + + return ( +
+
+

{t('inventory.usageHistory')}

+ +
+
+ {history.map((record: SpoolUsageRecord) => ( +
+
+ {formatDate(record.created_at)} + {record.print_name && ( + + {record.print_name} + + )} +
+
+ {record.weight_used.toFixed(1)}g + ({record.percent_used}%) + + {record.status} + +
+
+ ))} +
+
+ ); +} diff --git a/frontend/src/components/SpoolmanSettings.tsx b/frontend/src/components/SpoolmanSettings.tsx index c6a7748d1..6a77f9074 100644 --- a/frontend/src/components/SpoolmanSettings.tsx +++ b/frontend/src/components/SpoolmanSettings.tsx @@ -1,7 +1,7 @@ import { useState, useEffect } from 'react'; import { useTranslation } from 'react-i18next'; import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query'; -import { Loader2, Check, X, RefreshCw, Link2, Link2Off, Database, ChevronDown, Info, AlertTriangle } from 'lucide-react'; +import { Loader2, Check, X, RefreshCw, Link2, Link2Off, Database, ChevronDown, Info, AlertTriangle, Package, ExternalLink } from 'lucide-react'; import { api } from '../api/client'; import type { SpoolmanSyncResult, Printer } from '../api/client'; import { Card, CardContent, CardHeader } from './Card'; @@ -146,7 +146,7 @@ export function SpoolmanSettings() {
-

Spoolman Integration

+

{t('settings.filamentTracking')}

@@ -164,7 +164,7 @@ export function SpoolmanSettings() {
-

Spoolman Integration

+

{t('settings.filamentTracking')}

{saveMutation.isPending && ( @@ -173,303 +173,356 @@ export function SpoolmanSettings() {

- Connect to Spoolman for filament inventory tracking. AMS data will sync automatically. + {t('settings.filamentTrackingDesc')}

- {/* Info banner about sync requirements */} -
-
- -
-

How Sync Works

-
    -
  • Only official Bambu Lab spools with RFID are synced
  • -
  • New spools are auto-created in Spoolman on first sync
  • -
  • Non-Bambu Lab spools (third-party, refilled) are skipped
  • -
-

Linking Existing Spools

-

- To link existing Spoolman spools to your AMS, hover over an AMS slot and click "Link to Spoolman". -

-
-
-
- - {/* Enable toggle */} -
-
-

Enable Spoolman

-

- Sync filament data with Spoolman server -

-
- -
- - {/* URL input */} -
- - setLocalUrl(e.target.value)} - disabled={!localEnabled} - className="w-full px-3 py-2 bg-bambu-dark border border-bambu-dark-tertiary rounded-lg text-white placeholder-bambu-gray/50 focus:border-bambu-green focus:outline-none disabled:opacity-50" - /> -

- URL of your Spoolman server (e.g., http://localhost:7912) -

-
- - {/* Sync mode */} -
- - -

- {localSyncMode === 'auto' - ? 'AMS data syncs automatically when changes are detected' - : 'Only sync when manually triggered'} -

-
- - {/* Disable Weight Sync toggle - only show when sync mode is auto */} - {localSyncMode === 'auto' && ( -
-
-

{t('spoolman.disableWeightSync')}

-

- {t('spoolman.disableWeightSyncDesc')} -

+
+ + + {t('settings.trackingModeBuiltIn')} +
- -
- )} - - {/* Report Partial Usage toggle - only show when weight sync is disabled */} - {localDisableWeightSync && ( -
-
-

{t('spoolman.reportPartialUsage')}

-

- {t('spoolman.reportPartialUsageDesc')} -

-
- -
- )} - - {/* Connection status */} - {localEnabled && ( -
-
-
- Status: - {statusLoading ? ( - - ) : status?.connected ? ( - - - Connected - - ) : ( - - - Disconnected - - )} -
-
- {status?.connected ? ( - - ) : ( - - )} -
-
- - {/* Error display */} - {connectMutation.isError && ( -
- {(connectMutation.error as Error).message} +

+ {t('settings.trackingModeBuiltInDesc')} +

+ {!localEnabled && ( +
+ + {t('common.enabled')}
)} + - {/* Manual sync section */} - {status?.connected && ( -
-
-

Sync AMS Data

-

- Manually sync printer AMS data to Spoolman + {/* Spoolman */} + +

+ + {/* Built-in Inventory details */} + {!localEnabled && ( +
+
+ +
+
    +
  • {t('settings.builtInFeatureRfid')}
  • +
  • {t('settings.builtInFeatureUsage')}
  • +
  • {t('settings.builtInFeatureCatalog')}
  • +
+
+
+
+ )} + + {/* Spoolman settings - only shown when Spoolman mode is selected */} + {localEnabled && ( +
+ {/* Info banner about sync requirements */} +
+
+ +
+

{t('settings.howSyncWorks')}

+
    +
  • {t('settings.syncInfoRfidOnly')}
  • +
  • {t('settings.syncInfoAutoCreate')}
  • +
  • {t('settings.syncInfoThirdPartySkipped')}
  • +
+

{t('settings.linkingExistingSpools')}

+

+ {t('settings.linkingExistingSpoolsDesc')}

+
+
+ + {/* URL input */} +
+ + setLocalUrl(e.target.value)} + className="w-full px-3 py-2 bg-bambu-dark border border-bambu-dark-tertiary rounded-lg text-white placeholder-bambu-gray/50 focus:border-bambu-green focus:outline-none" + /> +

+ {t('settings.spoolmanUrlHint')} +

+
+ + {/* Sync mode */} +
+ + +

+ {localSyncMode === 'auto' + ? t('settings.syncModeAutoDesc') + : t('settings.syncModeManualDesc')} +

+
+ + {/* Disable Weight Sync toggle - only show when sync mode is auto */} + {localSyncMode === 'auto' && ( +
+
+

{t('spoolman.disableWeightSync')}

+

+ {t('spoolman.disableWeightSyncDesc')} +

+
+ +
+ )} + + {/* Report Partial Usage toggle - only show when weight sync is disabled */} + {localDisableWeightSync && ( +
+
+

{t('spoolman.reportPartialUsage')}

+

+ {t('spoolman.reportPartialUsageDesc')} +

+
+ +
+ )} + + {/* Connection status */} +
+
- {/* Printer selector */} -
- - -
- {/* Sync button */} - + {disconnectMutation.isPending ? ( + + ) : ( + + )} + {t('settings.disconnect')} + + ) : ( + + )}
- )} - {/* Sync result */} - {syncSuccess && syncResult && ( -
- {/* Main result */} -
- {syncResult.success - ? `Synced ${syncResult.synced_count} spool${syncResult.synced_count !== 1 ? 's' : ''} successfully` - : `Synced ${syncResult.synced_count} spool${syncResult.synced_count !== 1 ? 's' : ''} with ${syncResult.errors.length} error${syncResult.errors.length !== 1 ? 's' : ''}`} + {/* Error display */} + {connectMutation.isError && ( +
+ {(connectMutation.error as Error).message}
+ )} - {/* Skipped spools */} - {syncResult.skipped_count > 0 && ( -
-
-
- - - {syncResult.skipped_count} spool{syncResult.skipped_count !== 1 ? 's' : ''} skipped - -
- {syncResult.skipped_count > 5 && ( - - )} + {/* Manual sync section */} + {status?.connected && ( +
+
+

{t('settings.syncAmsData')}

+

+ {t('settings.syncAmsDataDesc')} +

+
+
+ {/* Printer selector */} +
+ +
-
    - {(showAllSkipped ? syncResult.skipped : syncResult.skipped.slice(0, 5)).map((s, i) => ( -
  • - {s.color && ( - - )} - {s.location} - - {s.reason} -
  • - ))} - {!showAllSkipped && syncResult.skipped_count > 5 && ( -
  • - ...and {syncResult.skipped_count - 5} more -
  • + {/* Sync button */} +
+ {t('spoolman.sync')} +
- )} +
+ )} - {/* Errors */} - {syncResult.errors.length > 0 && ( -
-
Errors:
-
    - {syncResult.errors.map((err, i) => ( -
  • {err}
  • - ))} -
+ {/* Sync result */} + {syncSuccess && syncResult && ( +
+ {/* Main result */} +
+ {syncResult.success + ? `Synced ${syncResult.synced_count} spool${syncResult.synced_count !== 1 ? 's' : ''} successfully` + : `Synced ${syncResult.synced_count} spool${syncResult.synced_count !== 1 ? 's' : ''} with ${syncResult.errors.length} error${syncResult.errors.length !== 1 ? 's' : ''}`}
- )} -
- )} + + {/* Skipped spools */} + {syncResult.skipped_count > 0 && ( +
+
+
+ + + {syncResult.skipped_count} spool{syncResult.skipped_count !== 1 ? 's' : ''} skipped + +
+ {syncResult.skipped_count > 5 && ( + + )} +
+
    + {(showAllSkipped ? syncResult.skipped : syncResult.skipped.slice(0, 5)).map((s, i) => ( +
  • + {s.color && ( + + )} + {s.location} + - {s.reason} +
  • + ))} + {!showAllSkipped && syncResult.skipped_count > 5 && ( +
  • + ...and {syncResult.skipped_count - 5} more +
  • + )} +
+
+ )} + + {/* Errors */} + {syncResult.errors.length > 0 && ( +
+
Errors:
+
    + {syncResult.errors.map((err, i) => ( +
  • {err}
  • + ))} +
+
+ )} +
+ )} +
)} diff --git a/frontend/src/components/spool-form/AdditionalSection.tsx b/frontend/src/components/spool-form/AdditionalSection.tsx new file mode 100644 index 000000000..e8e4233f9 --- /dev/null +++ b/frontend/src/components/spool-form/AdditionalSection.tsx @@ -0,0 +1,176 @@ +import { useState, useRef, useEffect, useMemo } from 'react'; +import { Scale } from 'lucide-react'; +import { useTranslation } from 'react-i18next'; +import type { AdditionalSectionProps } from './types'; + +function SpoolWeightPicker({ + catalog, + value, + onChange, +}: { + catalog: { id: number; name: string; weight: number }[]; + value: number; + onChange: (weight: number) => void; +}) { + const { t } = useTranslation(); + const [isOpen, setIsOpen] = useState(false); + const [search, setSearch] = useState(''); + const [selectedId, setSelectedId] = useState(null); + const dropdownRef = useRef(null); + const inputRef = useRef(null); + + useEffect(() => { + const handleClick = (e: MouseEvent) => { + if (dropdownRef.current && !dropdownRef.current.contains(e.target as Node)) { + setIsOpen(false); + } + }; + document.addEventListener('mousedown', handleClick); + return () => document.removeEventListener('mousedown', handleClick); + }, []); + + const filtered = useMemo(() => { + if (!search) return catalog; + const s = search.toLowerCase(); + return catalog.filter(e => + e.name.toLowerCase().includes(s) || + e.weight.toString().includes(s), + ); + }, [catalog, search]); + + // Display value: show catalog name if selected, or the weight + const displayValue = useMemo(() => { + if (isOpen) return search; + if (selectedId) { + const entry = catalog.find(e => e.id === selectedId); + if (entry) return entry.name; + } + const match = catalog.find(e => e.weight === value); + if (match) return match.name; + return ''; + }, [isOpen, search, selectedId, catalog, value]); + + return ( +
+ +
+
+ { + setIsOpen(true); + setSearch(''); + }} + onChange={(e) => { + setSearch(e.target.value); + setIsOpen(true); + }} + /> + {isOpen && ( +
+ {filtered.length === 0 ? ( +
{t('inventory.noResults')}
+ ) : ( + filtered.map(entry => ( + + )) + )} +
+ )} +
+
+ { + const val = parseInt(e.target.value); + if (!isNaN(val) && val >= 0) onChange(val); + }} + /> + g +
+
+
+ ); +} + +export function AdditionalSection({ + formData, + updateField, + spoolCatalog, +}: AdditionalSectionProps) { + const { t } = useTranslation(); + + return ( +
+ {/* Empty Spool Weight */} + updateField('core_weight', weight)} + /> + + {/* Current Weight (remaining filament) */} +
+ +
+
+ { + const remaining = parseInt(e.target.value) || 0; + updateField('weight_used', Math.max(0, formData.label_weight - remaining)); + }} + className="w-full px-3 py-2 pr-7 bg-bambu-dark border border-bambu-dark-tertiary rounded-lg text-white text-sm focus:outline-none focus:border-bambu-green" + /> + g +
+ / {formData.label_weight}g +
+
+ + {/* Note */} +
+ +