| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785 |
- """Virtual Printer Manager - coordinates SSDP, MQTT, and FTP services.
- Each virtual printer runs its own independent services (FTP, MQTT, SSDP, Bind)
- bound to its dedicated IP address, regardless of mode.
- """
- import asyncio
- import logging
- from collections.abc import Callable
- from datetime import datetime, timezone
- from pathlib import Path
- from backend.app.core.config import settings as app_settings
- from backend.app.services.virtual_printer.bind_server import BindServer
- from backend.app.services.virtual_printer.certificate import CertificateService
- from backend.app.services.virtual_printer.ftp_server import VirtualPrinterFTPServer
- from backend.app.services.virtual_printer.mqtt_server import SimpleMQTTServer
- from backend.app.services.virtual_printer.ssdp_server import SSDPProxy, VirtualPrinterSSDPServer
- from backend.app.services.virtual_printer.tcp_proxy import SlicerProxyManager
- logger = logging.getLogger(__name__)
- # Mapping of SSDP model codes to display names
- # These are the codes that slicers expect during discovery
- # Sources:
- # - https://gist.github.com/Alex-Schaefer/72a9e2491a42da2ef99fb87601955cc3
- # - https://github.com/psychoticbeef/BambuLabOrcaSlicerDiscovery
- VIRTUAL_PRINTER_MODELS = {
- # X1 Series
- "3DPrinter-X1-Carbon": "X1C", # X1 Carbon
- "3DPrinter-X1": "X1", # X1
- "C13": "X1E", # X1E
- # P Series
- "C11": "P1P", # P1P
- "C12": "P1S", # P1S
- "N7": "P2S", # P2S
- # A1 Series
- "N2S": "A1", # A1
- "N1": "A1 Mini", # A1 Mini
- # H2 Series
- "O1D": "H2D", # H2D
- "O1C": "H2C", # H2C
- "O1C2": "H2C", # H2C (dual nozzle variant)
- "O1S": "H2S", # H2S
- }
- # Serial number prefixes for each model (based on Bambu Lab serial number format)
- # Format: MMM??RYMDDUUUUU (15 chars total)
- # MMM = Model prefix (3 chars)
- # ?? = Unknown/revision code (2 chars)
- # R = Revision letter (1 char)
- # Y = Year digit (1 char)
- # M = Month (1 char, hex: 1-9, A=Oct, B=Nov, C=Dec)
- # DD = Day (2 chars)
- # UUUUU = Unit number (5 chars)
- MODEL_SERIAL_PREFIXES = {
- # X1 Series
- "3DPrinter-X1-Carbon": "00M00A", # X1C
- "3DPrinter-X1": "00M00A", # X1
- "C13": "03W00A", # X1E
- # P Series
- "C11": "01S00A", # P1P
- "C12": "01P00A", # P1S
- "N7": "22E00A", # P2S
- # A1 Series
- "N2S": "03900A", # A1
- "N1": "03000A", # A1 Mini
- # H2 Series
- "O1D": "09400A", # H2D
- "O1C": "09400A", # H2C
- "O1C2": "09400A", # H2C (dual nozzle variant)
- "O1S": "09400A", # H2S
- }
- # Default model
- DEFAULT_VIRTUAL_PRINTER_MODEL = "3DPrinter-X1-Carbon" # X1C
- def _get_serial_for_model(model: str, serial_suffix: str) -> str:
- """Get serial number for the given model and suffix."""
- prefix = MODEL_SERIAL_PREFIXES.get(model, "00M09A")
- return f"{prefix}{serial_suffix}"
- class VirtualPrinterInstance:
- """Per-printer state and file handling logic.
- Each instance represents one virtual printer with its own config,
- upload directory, certificates, and file handling mode.
- """
- def __init__(
- self,
- *,
- vp_id: int,
- name: str,
- mode: str,
- model: str,
- access_code: str,
- serial_suffix: str,
- target_printer_ip: str = "",
- target_printer_serial: str = "",
- target_printer_id: int | None = None,
- bind_ip: str = "",
- remote_interface_ip: str = "",
- base_dir: Path,
- session_factory: Callable | None = None,
- ):
- self.id = vp_id
- self.name = name
- self.mode = mode
- self.model = model
- self.access_code = access_code
- self.serial_suffix = serial_suffix
- self.target_printer_ip = target_printer_ip
- self.target_printer_serial = target_printer_serial
- self.target_printer_id = target_printer_id
- self.bind_ip = bind_ip
- self.remote_interface_ip = remote_interface_ip
- self._session_factory = session_factory
- # Directories
- self.upload_dir = base_dir / "uploads" / str(vp_id)
- self.cert_dir = base_dir / "certs" / str(vp_id)
- shared_ca_dir = base_dir / "certs"
- # Ensure directories exist
- self.upload_dir.mkdir(parents=True, exist_ok=True)
- (self.upload_dir / "cache").mkdir(exist_ok=True)
- self.cert_dir.mkdir(parents=True, exist_ok=True)
- # Certificate service (shared CA, per-instance printer cert)
- self._cert_service = CertificateService(
- cert_dir=self.cert_dir,
- serial=self.serial,
- shared_ca_dir=shared_ca_dir,
- )
- # Pending files for MQTT correlation
- self._pending_files: dict[str, Path] = {}
- # Per-instance services
- self._proxy: SlicerProxyManager | None = None
- self._ftp: VirtualPrinterFTPServer | None = None
- self._mqtt: SimpleMQTTServer | None = None
- self._bind: BindServer | None = None
- self._ssdp: VirtualPrinterSSDPServer | None = None
- self._ssdp_proxy: SSDPProxy | None = None
- self._tasks: list[asyncio.Task] = []
- @property
- def serial(self) -> str:
- """Full serial number for this virtual printer."""
- return _get_serial_for_model(self.model or DEFAULT_VIRTUAL_PRINTER_MODEL, self.serial_suffix)
- @property
- def cert_path(self) -> Path:
- return self._cert_service.cert_path
- @property
- def key_path(self) -> Path:
- return self._cert_service.key_path
- @property
- def is_proxy(self) -> bool:
- return self.mode == "proxy"
- @property
- def is_running(self) -> bool:
- return len(self._tasks) > 0 and all(not t.done() for t in self._tasks)
- def generate_certificates(self) -> tuple[Path, Path]:
- """Generate certificates for this instance."""
- self._cert_service.serial = self.serial if not self.is_proxy else (self.target_printer_serial or self.serial)
- additional_ips = [self.remote_interface_ip] if self.remote_interface_ip else None
- if self.bind_ip:
- additional_ips = additional_ips or []
- additional_ips.append(self.bind_ip)
- self._cert_service.delete_printer_certificate()
- return self._cert_service.generate_certificates(additional_ips=additional_ips)
- # -- File handling callbacks --
- async def on_file_received(self, file_path: Path, source_ip: str) -> None:
- """Handle file upload completion from FTP."""
- logger.info("[VP %s] Received file: %s from %s", self.name, file_path.name, source_ip)
- self._pending_files[file_path.name] = file_path
- if self.mode == "immediate":
- await self._archive_file(file_path, source_ip)
- elif self.mode == "print_queue":
- await self._add_to_print_queue(file_path, source_ip)
- else:
- await self._queue_file(file_path, source_ip)
- # Reset MQTT status back to IDLE
- if self._mqtt and file_path.suffix.lower() == ".3mf":
- self._mqtt.set_gcode_state("IDLE")
- async def on_print_command(self, filename: str, data: dict) -> None:
- """Handle print command from MQTT."""
- logger.info("[VP %s] Print command for: %s", self.name, filename)
- async def _archive_file(self, file_path: Path, source_ip: str) -> None:
- """Archive file immediately."""
- if not self._session_factory:
- logger.error("Cannot archive: no database session factory configured")
- return
- if file_path.suffix.lower() != ".3mf":
- logger.debug("Skipping non-3MF file: %s", file_path.name)
- self._pending_files.pop(file_path.name, None)
- try:
- file_path.unlink()
- except OSError:
- pass
- return
- try:
- from backend.app.services.archive import ArchiveService
- async with self._session_factory() as db:
- service = ArchiveService(db)
- archive = await service.archive_print(
- printer_id=None,
- source_file=file_path,
- print_data={
- "status": "archived",
- "source": "virtual_printer",
- "source_ip": source_ip,
- },
- )
- if archive:
- logger.info("[VP %s] Archived: %s - %s", self.name, archive.id, archive.print_name)
- try:
- file_path.unlink()
- except OSError:
- pass
- self._pending_files.pop(file_path.name, None)
- else:
- logger.error("Failed to archive file: %s", file_path.name)
- except Exception as e:
- logger.error("Error archiving file: %s", e)
- async def _queue_file(self, file_path: Path, source_ip: str) -> None:
- """Queue file for user review."""
- if not self._session_factory:
- logger.error("Cannot queue: no database session factory configured")
- return
- if file_path.suffix.lower() != ".3mf":
- self._pending_files.pop(file_path.name, None)
- try:
- file_path.unlink()
- except OSError:
- pass
- return
- try:
- from backend.app.models.pending_upload import PendingUpload
- async with self._session_factory() as db:
- pending = PendingUpload(
- filename=file_path.name,
- file_path=str(file_path),
- file_size=file_path.stat().st_size,
- source_ip=source_ip,
- status="pending",
- uploaded_at=datetime.now(timezone.utc),
- )
- db.add(pending)
- await db.commit()
- logger.info("[VP %s] Queued: %s - %s", self.name, pending.id, file_path.name)
- self._pending_files.pop(file_path.name, None)
- except Exception as e:
- logger.error("Error queueing file: %s", e)
- async def _add_to_print_queue(self, file_path: Path, source_ip: str) -> None:
- """Archive file and add to print queue, assigned to target printer or model."""
- if not self._session_factory:
- logger.error("Cannot add to print queue: no database session factory configured")
- return
- if file_path.suffix.lower() != ".3mf":
- self._pending_files.pop(file_path.name, None)
- try:
- file_path.unlink()
- except OSError:
- pass
- return
- try:
- from backend.app.models.print_queue import PrintQueueItem
- from backend.app.services.archive import ArchiveService
- async with self._session_factory() as db:
- service = ArchiveService(db)
- archive = await service.archive_print(
- printer_id=None,
- source_file=file_path,
- print_data={
- "status": "archived",
- "source": "virtual_printer",
- "source_ip": source_ip,
- },
- )
- if archive:
- logger.info("[VP %s] Archived: %s - %s", self.name, archive.id, archive.print_name)
- # Assign to specific printer if configured, otherwise use model for "Any X" scheduling
- target_model = None
- if not self.target_printer_id and self.model:
- target_model = VIRTUAL_PRINTER_MODELS.get(self.model)
- plate_id = self._extract_plate_id(file_path)
- queue_item = PrintQueueItem(
- printer_id=self.target_printer_id,
- target_model=target_model,
- archive_id=archive.id,
- plate_id=plate_id,
- position=1,
- status="pending",
- )
- db.add(queue_item)
- await db.commit()
- logger.info("[VP %s] Added to queue: %s", self.name, queue_item.id)
- try:
- file_path.unlink()
- except OSError:
- pass
- self._pending_files.pop(file_path.name, None)
- else:
- logger.error("Failed to archive file: %s", file_path.name)
- except Exception as e:
- logger.error("Error adding to print queue: %s", e)
- @staticmethod
- def _extract_plate_id(file_path: Path) -> int | None:
- """Extract plate index from 3MF slice_info.config."""
- try:
- import xml.etree.ElementTree as ET
- import zipfile
- with zipfile.ZipFile(file_path, "r") as zf:
- if "Metadata/slice_info.config" in zf.namelist():
- content = zf.read("Metadata/slice_info.config").decode()
- root = ET.fromstring(content) # noqa: S314 # nosec B314
- plate = root.find(".//plate")
- if plate is not None:
- for meta in plate.findall("metadata"):
- if meta.get("key") == "index" and meta.get("value"):
- return int(meta.get("value"))
- except Exception:
- return None
- return None
- # -- Service lifecycle --
- async def start_server(self) -> None:
- """Start server-mode services (FTP, MQTT, SSDP, Bind) on this VP's bind_ip."""
- logger.info("[VP %s] Starting server-mode services on %s", self.name, self.bind_ip)
- cert_path, key_path = self.generate_certificates()
- bind_addr = self.bind_ip or "0.0.0.0" # nosec B104
- async def run_with_logging(coro, svc_name):
- try:
- await coro
- except Exception as e:
- logger.error("[VP %s] %s failed: %s", self.name, svc_name, e)
- self._tasks = []
- # FTP server
- self._ftp = VirtualPrinterFTPServer(
- upload_dir=self.upload_dir,
- access_code=self.access_code,
- cert_path=cert_path,
- key_path=key_path,
- on_file_received=self.on_file_received,
- bind_address=bind_addr,
- )
- self._tasks.append(
- asyncio.create_task(
- run_with_logging(self._ftp.start(), "FTP"),
- name=f"vp_{self.id}_ftp",
- )
- )
- # MQTT server
- self._mqtt = SimpleMQTTServer(
- serial=self.serial,
- access_code=self.access_code,
- cert_path=cert_path,
- key_path=key_path,
- on_print_command=self.on_print_command,
- model=self.model or DEFAULT_VIRTUAL_PRINTER_MODEL,
- bind_address=bind_addr,
- )
- self._tasks.append(
- asyncio.create_task(
- run_with_logging(self._mqtt.start(), "MQTT"),
- name=f"vp_{self.id}_mqtt",
- )
- )
- # Bind server
- self._bind = BindServer(
- serial=self.serial,
- model=self.model or DEFAULT_VIRTUAL_PRINTER_MODEL,
- name=self.name,
- bind_address=bind_addr,
- cert_path=cert_path,
- key_path=key_path,
- )
- self._tasks.append(
- asyncio.create_task(
- run_with_logging(self._bind.start(), "Bind"),
- name=f"vp_{self.id}_bind",
- )
- )
- # SSDP server
- self._ssdp = VirtualPrinterSSDPServer(
- name=self.name,
- serial=self.serial,
- model=self.model or DEFAULT_VIRTUAL_PRINTER_MODEL,
- advertise_ip=self.remote_interface_ip or self.bind_ip or "",
- bind_ip=bind_addr,
- )
- self._tasks.append(
- asyncio.create_task(
- run_with_logging(self._ssdp.start(), "SSDP"),
- name=f"vp_{self.id}_ssdp",
- )
- )
- logger.info("[VP %s] Server-mode services started on %s", self.name, bind_addr)
- async def stop_server(self) -> None:
- """Stop server-mode services."""
- if self._ftp:
- await self._ftp.stop()
- self._ftp = None
- if self._mqtt:
- await self._mqtt.stop()
- self._mqtt = None
- if self._bind:
- await self._bind.stop()
- self._bind = None
- if self._ssdp:
- await self._ssdp.stop()
- self._ssdp = None
- await self._cancel_tasks()
- async def start_proxy(self) -> None:
- """Start proxy mode services for this instance."""
- logger.info("[VP %s] Starting proxy mode to %s", self.name, self.target_printer_ip)
- cert_path, key_path = self.generate_certificates()
- self._proxy = SlicerProxyManager(
- target_host=self.target_printer_ip,
- cert_path=cert_path,
- key_path=key_path,
- on_activity=lambda n, m: logger.info("[VP %s] Proxy %s: %s", self.name, n, m),
- bind_address=self.bind_ip or "0.0.0.0", # nosec B104
- )
- async def run_with_logging(coro, svc_name):
- try:
- await coro
- except Exception as e:
- logger.error("[VP %s] %s failed: %s", self.name, svc_name, e)
- self._tasks = []
- # SSDP for proxy
- proxy_serial = self.target_printer_serial or self.serial
- if self.remote_interface_ip:
- from backend.app.services.network_utils import find_interface_for_ip
- local_iface = find_interface_for_ip(self.target_printer_ip)
- if local_iface:
- self._ssdp_proxy = SSDPProxy(
- local_interface_ip=local_iface["ip"],
- remote_interface_ip=self.remote_interface_ip,
- target_printer_ip=self.target_printer_ip,
- )
- self._tasks.append(
- asyncio.create_task(
- run_with_logging(self._ssdp_proxy.start(), "SSDP Proxy"),
- name=f"vp_{self.id}_ssdp_proxy",
- )
- )
- else:
- self._start_fallback_ssdp(proxy_serial, run_with_logging)
- else:
- self._start_fallback_ssdp(proxy_serial, run_with_logging)
- self._tasks.append(
- asyncio.create_task(
- run_with_logging(self._proxy.start(), "Proxy"),
- name=f"vp_{self.id}_proxy",
- )
- )
- def _start_fallback_ssdp(self, proxy_serial: str, run_with_logging) -> None:
- """Start single-interface SSDP server as fallback for proxy mode."""
- self._ssdp = VirtualPrinterSSDPServer(
- name=f"{self.name} (Proxy)",
- serial=proxy_serial,
- model=self.model or DEFAULT_VIRTUAL_PRINTER_MODEL,
- advertise_ip=self.bind_ip or "",
- bind_ip=self.bind_ip or "",
- )
- self._tasks.append(
- asyncio.create_task(
- run_with_logging(self._ssdp.start(), "SSDP"),
- name=f"vp_{self.id}_ssdp",
- )
- )
- async def stop_proxy(self) -> None:
- """Stop proxy mode services for this instance."""
- if self._proxy:
- await self._proxy.stop()
- self._proxy = None
- if self._ssdp:
- await self._ssdp.stop()
- self._ssdp = None
- if self._ssdp_proxy:
- await self._ssdp_proxy.stop()
- self._ssdp_proxy = None
- await self._cancel_tasks()
- async def _cancel_tasks(self) -> None:
- """Cancel all running tasks and wait for cleanup."""
- for task in self._tasks:
- task.cancel()
- if self._tasks:
- try:
- await asyncio.wait_for(asyncio.gather(*self._tasks, return_exceptions=True), timeout=1.0)
- except TimeoutError:
- pass
- self._tasks = []
- def get_status(self) -> dict:
- """Get status for this instance."""
- status: dict = {
- "running": self.is_running,
- "pending_files": len(self._pending_files),
- }
- if self.is_proxy and self._proxy:
- status["proxy"] = self._proxy.get_status()
- return status
- class VirtualPrinterManager:
- """Multi-instance virtual printer registry and orchestrator.
- Every VP runs its own independent services on a dedicated bind IP.
- """
- def __init__(self):
- self._session_factory: Callable | None = None
- self._instances: dict[int, VirtualPrinterInstance] = {}
- # Directories
- self._base_dir = app_settings.base_dir / "virtual_printer"
- # Ensure base directories exist
- self._ensure_base_directories()
- def _ensure_base_directories(self) -> None:
- """Create base directories at startup."""
- for dir_path in [self._base_dir, self._base_dir / "uploads", self._base_dir / "certs"]:
- try:
- dir_path.mkdir(parents=True, exist_ok=True)
- except PermissionError:
- logger.error(
- f"Cannot create directory {dir_path}: Permission denied. "
- f"For Docker: ensure the data volume is writable by the container user. "
- f"For bare metal: run 'sudo chown -R $(whoami) {self._base_dir}'"
- )
- def set_session_factory(self, session_factory: Callable) -> None:
- """Set the database session factory."""
- self._session_factory = session_factory
- @property
- def is_enabled(self) -> bool:
- """Check if any virtual printer is running."""
- return len(self._instances) > 0
- async def sync_from_db(self) -> None:
- """Load all VPs from DB, reconcile running state."""
- if not self._session_factory:
- logger.warning("Cannot sync virtual printers: no session factory")
- return
- from sqlalchemy import select
- from backend.app.models.printer import Printer
- from backend.app.models.virtual_printer import VirtualPrinter
- async with self._session_factory() as db:
- result = await db.execute(
- select(VirtualPrinter).where(VirtualPrinter.enabled == True).order_by(VirtualPrinter.position) # noqa: E712
- )
- enabled_vps = result.scalars().all()
- # Stop instances that are no longer enabled or changed mode
- enabled_ids = {vp.id for vp in enabled_vps}
- for vp_id in list(self._instances.keys()):
- if vp_id not in enabled_ids:
- await self.remove_instance(vp_id)
- # Look up printer IPs for proxy VPs
- proxy_vps = [vp for vp in enabled_vps if vp.mode == "proxy"]
- proxy_ips: dict[int, tuple[str, str]] = {}
- if proxy_vps:
- async with self._session_factory() as db:
- for pvp in proxy_vps:
- if pvp.target_printer_id:
- result = await db.execute(select(Printer).where(Printer.id == pvp.target_printer_id))
- printer = result.scalar_one_or_none()
- if printer:
- proxy_ips[pvp.id] = (printer.ip_address, printer.serial_number)
- # Detect config changes on running instances and restart if needed
- for vp in enabled_vps:
- instance = self._instances.get(vp.id)
- if not instance:
- continue
- changed = (
- instance.mode != vp.mode
- or instance.model != (vp.model or DEFAULT_VIRTUAL_PRINTER_MODEL)
- or instance.access_code != (vp.access_code or "")
- or instance.bind_ip != (vp.bind_ip or "")
- or instance.remote_interface_ip != (vp.remote_interface_ip or "")
- or instance.target_printer_id != vp.target_printer_id
- )
- if changed:
- logger.info(
- "VP %s config changed (mode: %s→%s), restarting",
- instance.name,
- instance.mode,
- vp.mode,
- )
- await self.remove_instance(vp.id)
- # Start instances for all enabled VPs (skip already running)
- for vp in enabled_vps:
- if vp.id in self._instances:
- continue
- if vp.mode == "proxy":
- ip_info = proxy_ips.get(vp.id)
- if not ip_info:
- logger.warning("Proxy VP %s: target printer not found, skipping", vp.name)
- continue
- target_ip, target_serial = ip_info
- instance = VirtualPrinterInstance(
- vp_id=vp.id,
- name=vp.name,
- mode=vp.mode,
- model=vp.model or DEFAULT_VIRTUAL_PRINTER_MODEL,
- access_code=vp.access_code or "",
- serial_suffix=vp.serial_suffix,
- target_printer_ip=target_ip,
- target_printer_serial=target_serial,
- bind_ip=vp.bind_ip or "",
- remote_interface_ip=vp.remote_interface_ip or "",
- base_dir=self._base_dir,
- session_factory=self._session_factory,
- )
- self._instances[vp.id] = instance
- await instance.start_proxy()
- logger.info("Started proxy VP: %s → %s", instance.name, target_ip)
- else:
- instance = VirtualPrinterInstance(
- vp_id=vp.id,
- name=vp.name,
- mode=vp.mode,
- model=vp.model or DEFAULT_VIRTUAL_PRINTER_MODEL,
- access_code=vp.access_code or "",
- serial_suffix=vp.serial_suffix,
- target_printer_id=vp.target_printer_id,
- bind_ip=vp.bind_ip or "",
- remote_interface_ip=vp.remote_interface_ip or "",
- base_dir=self._base_dir,
- session_factory=self._session_factory,
- )
- self._instances[vp.id] = instance
- await instance.start_server()
- logger.info("Started server-mode VP: %s on %s", instance.name, vp.bind_ip)
- async def remove_instance(self, vp_id: int) -> None:
- """Stop and remove a single VP instance."""
- instance = self._instances.pop(vp_id, None)
- if instance:
- if instance.is_proxy:
- await instance.stop_proxy()
- else:
- await instance.stop_server()
- logger.info("Removed VP instance: %s", instance.name)
- async def stop_all(self) -> None:
- """Shutdown all virtual printer services."""
- logger.info("Stopping all virtual printer services...")
- for vp_id in list(self._instances.keys()):
- await self.remove_instance(vp_id)
- logger.info("All virtual printer services stopped")
- def get_instance(self, vp_id: int) -> VirtualPrinterInstance | None:
- """Get a running instance by ID."""
- return self._instances.get(vp_id)
- def get_all_status(self) -> list[dict]:
- """Get status for all running instances."""
- return [
- {
- "id": inst.id,
- "name": inst.name,
- "mode": inst.mode,
- **inst.get_status(),
- }
- for inst in self._instances.values()
- ]
- # -- Legacy single-printer compat --
- def get_status(self) -> dict:
- """Get status for first virtual printer (backward compat)."""
- if self._instances:
- first = next(iter(self._instances.values()))
- return {
- "enabled": True,
- "running": first.is_running,
- "mode": first.mode,
- "name": first.name,
- "serial": first.serial,
- "model": first.model or DEFAULT_VIRTUAL_PRINTER_MODEL,
- "model_name": VIRTUAL_PRINTER_MODELS.get(
- first.model or DEFAULT_VIRTUAL_PRINTER_MODEL,
- first.model or DEFAULT_VIRTUAL_PRINTER_MODEL,
- ),
- "pending_files": first.get_status().get("pending_files", 0),
- **({"target_printer_ip": first.target_printer_ip} if first.is_proxy else {}),
- **({"proxy": first.get_status().get("proxy", {})} if first.is_proxy else {}),
- }
- return {
- "enabled": False,
- "running": False,
- "mode": "immediate",
- "name": "Bambuddy",
- "serial": "",
- "model": DEFAULT_VIRTUAL_PRINTER_MODEL,
- "model_name": VIRTUAL_PRINTER_MODELS[DEFAULT_VIRTUAL_PRINTER_MODEL],
- "pending_files": 0,
- }
- async def configure(
- self,
- enabled: bool,
- access_code: str = "",
- mode: str = "immediate",
- model: str = "",
- target_printer_ip: str = "",
- target_printer_serial: str = "",
- remote_interface_ip: str = "",
- ) -> None:
- """Legacy single-printer configure. Delegates to sync_from_db()."""
- # This method is kept for backward compat with the settings endpoint.
- # The actual work is done by sync_from_db() which reads from the DB.
- await self.sync_from_db()
- # Global instance
- virtual_printer_manager = VirtualPrinterManager()
|