mirror of
https://github.com/maziggy/bambuddy.git
synced 2026-10-03 04:31:28 +02:00
maziggy
8acc91d3cd
- Detect Docker environment automatically via /.dockerenv and cgroup - Show subnet input field in Add Printer dialog when running in Docker - Scan IP range for Bambu printer ports (8883 MQTT, 990 FTPS) - Query SSDP to get printer name/serial/model from discovered IPs - Map raw SSDP model codes to friendly names (BL-P001→X1C, O1D→H2D)
170 lines
4.4 KiB
Python
170 lines
4.4 KiB
Python
"""
|
|
Printer discovery API endpoints.
|
|
|
|
Provides endpoints for discovering Bambu Lab printers on the local network.
|
|
Supports both SSDP discovery (for native installs) and subnet scanning (for Docker).
|
|
"""
|
|
|
|
import logging
|
|
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
|
|
from backend.app.services.discovery import (
|
|
discovery_service,
|
|
is_running_in_docker,
|
|
subnet_scanner,
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
router = APIRouter(prefix="/discovery", tags=["discovery"])
|
|
|
|
|
|
class DiscoveryStatus(BaseModel):
|
|
"""Discovery status response."""
|
|
|
|
running: bool
|
|
|
|
|
|
class DiscoveryInfo(BaseModel):
|
|
"""Discovery environment info."""
|
|
|
|
is_docker: bool
|
|
ssdp_running: bool
|
|
scan_running: bool
|
|
|
|
|
|
class SubnetScanRequest(BaseModel):
|
|
"""Request to scan a subnet."""
|
|
|
|
subnet: str # CIDR notation, e.g., "192.168.1.0/24"
|
|
timeout: float = 1.0 # Connection timeout per host
|
|
|
|
|
|
class SubnetScanStatus(BaseModel):
|
|
"""Subnet scan status response."""
|
|
|
|
running: bool
|
|
scanned: int
|
|
total: int
|
|
|
|
|
|
class DiscoveredPrinterResponse(BaseModel):
|
|
"""Discovered printer response."""
|
|
|
|
serial: str
|
|
name: str
|
|
ip_address: str
|
|
model: str | None = None
|
|
discovered_at: str | None = None
|
|
|
|
|
|
@router.get("/info", response_model=DiscoveryInfo)
|
|
async def get_discovery_info():
|
|
"""Get discovery environment info (Docker detection, etc.)."""
|
|
return DiscoveryInfo(
|
|
is_docker=is_running_in_docker(),
|
|
ssdp_running=discovery_service.is_running,
|
|
scan_running=subnet_scanner.is_running,
|
|
)
|
|
|
|
|
|
@router.get("/status", response_model=DiscoveryStatus)
|
|
async def get_discovery_status():
|
|
"""Get the current SSDP discovery status."""
|
|
return DiscoveryStatus(running=discovery_service.is_running)
|
|
|
|
|
|
@router.post("/start", response_model=DiscoveryStatus)
|
|
async def start_discovery(duration: float = 10.0):
|
|
"""Start SSDP printer discovery.
|
|
|
|
Args:
|
|
duration: Discovery duration in seconds (default 10)
|
|
"""
|
|
await discovery_service.start(duration=duration)
|
|
return DiscoveryStatus(running=discovery_service.is_running)
|
|
|
|
|
|
@router.post("/stop", response_model=DiscoveryStatus)
|
|
async def stop_discovery():
|
|
"""Stop SSDP printer discovery."""
|
|
await discovery_service.stop()
|
|
return DiscoveryStatus(running=discovery_service.is_running)
|
|
|
|
|
|
@router.get("/printers", response_model=list[DiscoveredPrinterResponse])
|
|
async def get_discovered_printers():
|
|
"""Get list of discovered printers (from both SSDP and subnet scan)."""
|
|
# Combine results from both discovery methods
|
|
printers = {}
|
|
|
|
# Add SSDP discovered printers
|
|
for p in discovery_service.discovered_printers:
|
|
printers[p.ip_address] = p
|
|
|
|
# Add subnet scan discovered printers (may override if same IP)
|
|
for p in subnet_scanner.discovered_printers:
|
|
if p.ip_address not in printers:
|
|
printers[p.ip_address] = p
|
|
|
|
return [
|
|
DiscoveredPrinterResponse(
|
|
serial=p.serial,
|
|
name=p.name,
|
|
ip_address=p.ip_address,
|
|
model=p.model,
|
|
discovered_at=p.discovered_at,
|
|
)
|
|
for p in printers.values()
|
|
]
|
|
|
|
|
|
# Subnet scanning endpoints (for Docker environments)
|
|
|
|
|
|
@router.post("/scan", response_model=SubnetScanStatus)
|
|
async def start_subnet_scan(request: SubnetScanRequest):
|
|
"""Start a subnet scan for Bambu printers.
|
|
|
|
Use this when running in Docker where SSDP multicast doesn't work.
|
|
|
|
Args:
|
|
request: Subnet to scan in CIDR notation (e.g., "192.168.1.0/24")
|
|
"""
|
|
# Start scan in background
|
|
import asyncio
|
|
|
|
asyncio.create_task(subnet_scanner.scan_subnet(request.subnet, request.timeout))
|
|
|
|
# Return immediate status
|
|
scanned, total = subnet_scanner.progress
|
|
return SubnetScanStatus(
|
|
running=subnet_scanner.is_running,
|
|
scanned=scanned,
|
|
total=total,
|
|
)
|
|
|
|
|
|
@router.get("/scan/status", response_model=SubnetScanStatus)
|
|
async def get_scan_status():
|
|
"""Get the current subnet scan status."""
|
|
scanned, total = subnet_scanner.progress
|
|
return SubnetScanStatus(
|
|
running=subnet_scanner.is_running,
|
|
scanned=scanned,
|
|
total=total,
|
|
)
|
|
|
|
|
|
@router.post("/scan/stop", response_model=SubnetScanStatus)
|
|
async def stop_subnet_scan():
|
|
"""Stop the current subnet scan."""
|
|
subnet_scanner.stop()
|
|
scanned, total = subnet_scanner.progress
|
|
return SubnetScanStatus(
|
|
running=subnet_scanner.is_running,
|
|
scanned=scanned,
|
|
total=total,
|
|
)
|