settings.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770
  1. import io
  2. import logging
  3. import zipfile
  4. from datetime import datetime
  5. from pathlib import Path
  6. from fastapi import APIRouter, Depends, File, UploadFile
  7. from fastapi.responses import JSONResponse, StreamingResponse
  8. from sqlalchemy import delete, select
  9. from sqlalchemy.ext.asyncio import AsyncSession
  10. from backend.app.core.auth import RequirePermissionIfAuthEnabled
  11. from backend.app.core.config import settings as app_settings
  12. from backend.app.core.database import get_db
  13. from backend.app.core.permissions import Permission
  14. from backend.app.models.settings import Settings
  15. from backend.app.models.user import User
  16. from backend.app.schemas.settings import AppSettings, AppSettingsUpdate
  17. logger = logging.getLogger(__name__)
  18. router = APIRouter(prefix="/settings", tags=["settings"])
  19. # Default settings
  20. DEFAULT_SETTINGS = AppSettings()
  21. async def get_setting(db: AsyncSession, key: str) -> str | None:
  22. """Get a single setting value by key."""
  23. result = await db.execute(select(Settings).where(Settings.key == key))
  24. setting = result.scalar_one_or_none()
  25. return setting.value if setting else None
  26. async def get_external_login_url(db: AsyncSession) -> str:
  27. """Get the external URL for the login page.
  28. Uses external_url from settings if available, otherwise falls back to APP_URL env var.
  29. Args:
  30. db: Database session
  31. Returns:
  32. Full URL to the login page
  33. """
  34. import os
  35. external_url = await get_setting(db, "external_url")
  36. if external_url:
  37. external_url = external_url.rstrip("/")
  38. else:
  39. external_url = os.environ.get("APP_URL", "http://localhost:5173")
  40. return external_url + "/login"
  41. async def set_setting(db: AsyncSession, key: str, value: str) -> None:
  42. """Set a single setting value."""
  43. from sqlalchemy import func
  44. from sqlalchemy.dialects.sqlite import insert as sqlite_insert
  45. # Use upsert (INSERT ... ON CONFLICT UPDATE) for reliability
  46. stmt = sqlite_insert(Settings).values(key=key, value=value)
  47. stmt = stmt.on_conflict_do_update(index_elements=["key"], set_={"value": value, "updated_at": func.now()})
  48. await db.execute(stmt)
  49. @router.get("", response_model=AppSettings)
  50. @router.get("/", response_model=AppSettings)
  51. async def get_settings(
  52. db: AsyncSession = Depends(get_db),
  53. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_READ),
  54. ):
  55. """Get all application settings."""
  56. settings_dict = DEFAULT_SETTINGS.model_dump()
  57. # Load saved settings from database
  58. result = await db.execute(select(Settings))
  59. db_settings = result.scalars().all()
  60. for setting in db_settings:
  61. if setting.key in settings_dict:
  62. # Parse the value based on the expected type
  63. if setting.key in [
  64. "auto_archive",
  65. "save_thumbnails",
  66. "capture_finish_photo",
  67. "spoolman_enabled",
  68. "spoolman_disable_weight_sync",
  69. "spoolman_report_partial_usage",
  70. "disable_filament_warnings",
  71. "prefer_lowest_filament",
  72. "check_updates",
  73. "check_printer_firmware",
  74. "include_beta_updates",
  75. "virtual_printer_enabled",
  76. "ftp_retry_enabled",
  77. "mqtt_enabled",
  78. "mqtt_use_tls",
  79. "ha_enabled",
  80. "per_printer_mapping_expanded",
  81. "prometheus_enabled",
  82. "user_notifications_enabled",
  83. "queue_drying_enabled",
  84. "queue_drying_block",
  85. "ambient_drying_enabled",
  86. "require_plate_clear",
  87. "default_bed_levelling",
  88. "default_flow_cali",
  89. "default_vibration_cali",
  90. "default_layer_inspect",
  91. "default_timelapse",
  92. ]:
  93. settings_dict[setting.key] = setting.value.lower() == "true"
  94. elif setting.key in [
  95. "default_filament_cost",
  96. "energy_cost_per_kwh",
  97. "ams_temp_good",
  98. "ams_temp_fair",
  99. "library_disk_warning_gb",
  100. "low_stock_threshold",
  101. ]:
  102. settings_dict[setting.key] = float(setting.value)
  103. elif setting.key in [
  104. "ams_humidity_good",
  105. "ams_humidity_fair",
  106. "ams_history_retention_days",
  107. "ftp_retry_count",
  108. "ftp_retry_delay",
  109. "ftp_timeout",
  110. "mqtt_port",
  111. "stagger_group_size",
  112. "stagger_interval_minutes",
  113. ]:
  114. settings_dict[setting.key] = int(setting.value)
  115. elif setting.key == "default_printer_id":
  116. # Handle nullable integer
  117. settings_dict[setting.key] = int(setting.value) if setting.value and setting.value != "None" else None
  118. else:
  119. settings_dict[setting.key] = setting.value
  120. # Get Home Assistant settings (with environment variable overrides)
  121. ha_settings = await get_homeassistant_settings(db)
  122. settings_dict.update(ha_settings)
  123. return AppSettings(**settings_dict)
  124. @router.put("/", response_model=AppSettings)
  125. async def update_settings(
  126. settings_update: AppSettingsUpdate,
  127. db: AsyncSession = Depends(get_db),
  128. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_UPDATE),
  129. ):
  130. """Update application settings."""
  131. update_data = settings_update.model_dump(exclude_unset=True)
  132. # Check if any MQTT settings are being updated
  133. mqtt_keys = {
  134. "mqtt_enabled",
  135. "mqtt_broker",
  136. "mqtt_port",
  137. "mqtt_username",
  138. "mqtt_password",
  139. "mqtt_topic_prefix",
  140. "mqtt_use_tls",
  141. }
  142. mqtt_updated = bool(mqtt_keys & set(update_data.keys()))
  143. for key, value in update_data.items():
  144. # Convert value to string for storage
  145. if isinstance(value, bool):
  146. str_value = "true" if value else "false"
  147. elif value is None:
  148. str_value = "None"
  149. else:
  150. str_value = str(value)
  151. await set_setting(db, key, str_value)
  152. await db.commit()
  153. # Expire all objects to ensure fresh reads after commit
  154. db.expire_all()
  155. # Reconfigure MQTT relay if any MQTT settings changed
  156. if mqtt_updated:
  157. try:
  158. from backend.app.services.mqtt_relay import mqtt_relay
  159. mqtt_settings = {
  160. "mqtt_enabled": (await get_setting(db, "mqtt_enabled") or "false") == "true",
  161. "mqtt_broker": await get_setting(db, "mqtt_broker") or "",
  162. "mqtt_port": int(await get_setting(db, "mqtt_port") or "1883"),
  163. "mqtt_username": await get_setting(db, "mqtt_username") or "",
  164. "mqtt_password": await get_setting(db, "mqtt_password") or "",
  165. "mqtt_topic_prefix": await get_setting(db, "mqtt_topic_prefix") or "bambuddy",
  166. "mqtt_use_tls": (await get_setting(db, "mqtt_use_tls") or "false") == "true",
  167. }
  168. await mqtt_relay.configure(mqtt_settings)
  169. except Exception:
  170. pass # Don't fail the settings update if MQTT reconfiguration fails
  171. # Return updated settings
  172. return await get_settings(db)
  173. @router.patch("/", response_model=AppSettings)
  174. @router.patch("", response_model=AppSettings)
  175. async def patch_settings(
  176. settings_update: AppSettingsUpdate,
  177. db: AsyncSession = Depends(get_db),
  178. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_UPDATE),
  179. ):
  180. """Partially update application settings (same as PUT, for REST compatibility)."""
  181. return await update_settings(settings_update, db, _)
  182. @router.post("/reset", response_model=AppSettings)
  183. async def reset_settings(
  184. db: AsyncSession = Depends(get_db),
  185. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_UPDATE),
  186. ):
  187. """Reset all settings to defaults."""
  188. # Delete all settings
  189. result = await db.execute(select(Settings))
  190. for setting in result.scalars().all():
  191. await db.delete(setting)
  192. await db.commit()
  193. return DEFAULT_SETTINGS
  194. @router.get("/default-sidebar-order")
  195. async def get_default_sidebar_order(
  196. db: AsyncSession = Depends(get_db),
  197. ):
  198. """Get the admin-set default sidebar order.
  199. Intentionally unauthenticated: non-admin users need to read this value to apply
  200. the default sidebar order, but may lack SETTINGS_READ permission.
  201. The value is non-sensitive (sidebar item IDs only).
  202. """
  203. value = await get_setting(db, "default_sidebar_order")
  204. return {"default_sidebar_order": value or ""}
  205. @router.get("/check-ffmpeg")
  206. async def check_ffmpeg():
  207. """Check if ffmpeg is installed and available."""
  208. from backend.app.services.camera import get_ffmpeg_path
  209. ffmpeg_path = get_ffmpeg_path()
  210. return {
  211. "installed": ffmpeg_path is not None,
  212. "path": ffmpeg_path,
  213. }
  214. @router.get("/spoolman")
  215. async def get_spoolman_settings(
  216. db: AsyncSession = Depends(get_db),
  217. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_READ),
  218. ):
  219. """Get Spoolman integration settings."""
  220. spoolman_enabled = await get_setting(db, "spoolman_enabled") or "false"
  221. spoolman_url = await get_setting(db, "spoolman_url") or ""
  222. spoolman_sync_mode = await get_setting(db, "spoolman_sync_mode") or "auto"
  223. spoolman_disable_weight_sync = await get_setting(db, "spoolman_disable_weight_sync") or "false"
  224. spoolman_report_partial_usage = await get_setting(db, "spoolman_report_partial_usage") or "true"
  225. return {
  226. "spoolman_enabled": spoolman_enabled,
  227. "spoolman_url": spoolman_url,
  228. "spoolman_sync_mode": spoolman_sync_mode,
  229. "spoolman_disable_weight_sync": spoolman_disable_weight_sync,
  230. "spoolman_report_partial_usage": spoolman_report_partial_usage,
  231. }
  232. @router.put("/spoolman")
  233. async def update_spoolman_settings(
  234. settings: dict,
  235. db: AsyncSession = Depends(get_db),
  236. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_UPDATE),
  237. ):
  238. """Update Spoolman integration settings."""
  239. if "spoolman_enabled" in settings:
  240. old_val = await get_setting(db, "spoolman_enabled") or "false"
  241. new_val = settings["spoolman_enabled"]
  242. await set_setting(db, "spoolman_enabled", new_val)
  243. # Switching to Spoolman: clear built-in inventory slot assignments
  244. if old_val.lower() != "true" and new_val.lower() == "true":
  245. from backend.app.models.spool_assignment import SpoolAssignment
  246. result = await db.execute(delete(SpoolAssignment))
  247. logger.info("Cleared %d spool assignments on switch to Spoolman mode", result.rowcount)
  248. if "spoolman_url" in settings:
  249. await set_setting(db, "spoolman_url", settings["spoolman_url"])
  250. if "spoolman_sync_mode" in settings:
  251. await set_setting(db, "spoolman_sync_mode", settings["spoolman_sync_mode"])
  252. if "spoolman_disable_weight_sync" in settings:
  253. await set_setting(db, "spoolman_disable_weight_sync", settings["spoolman_disable_weight_sync"])
  254. if "spoolman_report_partial_usage" in settings:
  255. await set_setting(db, "spoolman_report_partial_usage", settings["spoolman_report_partial_usage"])
  256. await db.commit()
  257. db.expire_all()
  258. # Return updated settings
  259. return await get_spoolman_settings(db)
  260. async def get_homeassistant_settings(db: AsyncSession) -> dict:
  261. """
  262. Get Home Assistant integration settings.
  263. Environment variables (HA_URL, HA_TOKEN) take precedence over database settings.
  264. """
  265. import os
  266. # Check environment variables first
  267. ha_url_env = os.environ.get("HA_URL")
  268. ha_token_env = os.environ.get("HA_TOKEN")
  269. # Fall back to database values
  270. ha_url = ha_url_env or await get_setting(db, "ha_url") or ""
  271. ha_token = ha_token_env or await get_setting(db, "ha_token") or ""
  272. ha_enabled_db = await get_setting(db, "ha_enabled") or "false"
  273. # Track which settings come from environment
  274. ha_url_from_env = bool(ha_url_env)
  275. ha_token_from_env = bool(ha_token_env)
  276. ha_env_managed = ha_url_from_env and ha_token_from_env
  277. # Auto-enable when both env vars are set, otherwise use database value
  278. if ha_url_env and ha_token_env:
  279. ha_enabled = True
  280. else:
  281. ha_enabled = ha_enabled_db.lower() == "true"
  282. return {
  283. "ha_enabled": ha_enabled,
  284. "ha_url": ha_url,
  285. "ha_token": ha_token,
  286. "ha_url_from_env": ha_url_from_env,
  287. "ha_token_from_env": ha_token_from_env,
  288. "ha_env_managed": ha_env_managed,
  289. }
  290. @router.get("/backup")
  291. async def create_backup(
  292. db: AsyncSession = Depends(get_db),
  293. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_BACKUP),
  294. ):
  295. """Create a complete backup (database + all files) as a ZIP.
  296. This is a simplified backup that includes the entire SQLite database
  297. and all data directories. It is complete by definition and cannot miss data.
  298. """
  299. import shutil
  300. import tempfile
  301. from sqlalchemy import text
  302. from backend.app.core.database import engine
  303. try:
  304. base_dir = app_settings.base_dir
  305. db_path = Path(app_settings.database_url.replace("sqlite+aiosqlite:///", ""))
  306. with tempfile.TemporaryDirectory() as temp_dir:
  307. temp_path = Path(temp_dir)
  308. # 1. Checkpoint WAL to ensure all data is in main db file
  309. async with engine.begin() as conn:
  310. await conn.execute(text("PRAGMA wal_checkpoint(TRUNCATE)"))
  311. # 2. Copy database file
  312. shutil.copy2(db_path, temp_path / "bambuddy.db")
  313. # 3. Copy data directories (if they exist)
  314. dirs_to_backup = [
  315. ("archive", base_dir / "archive"),
  316. ("virtual_printer", base_dir / "virtual_printer"),
  317. ("plate_calibration", app_settings.plate_calibration_dir),
  318. ("icons", base_dir / "icons"),
  319. ("projects", base_dir / "projects"),
  320. ]
  321. for name, src_dir in dirs_to_backup:
  322. if src_dir.exists() and any(src_dir.iterdir()):
  323. try:
  324. shutil.copytree(src_dir, temp_path / name)
  325. except shutil.Error as e:
  326. # Some files may have restricted permissions (e.g., SSL keys)
  327. # Log the error but continue with partial backup
  328. logger.warning("Some files in %s could not be copied: %s", name, e)
  329. except PermissionError as e:
  330. logger.warning("Permission denied copying %s: %s", name, e)
  331. # 4. Create ZIP
  332. zip_buffer = io.BytesIO()
  333. with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zf:
  334. for file_path in temp_path.rglob("*"):
  335. if file_path.is_file():
  336. arcname = file_path.relative_to(temp_path)
  337. zf.write(file_path, arcname)
  338. zip_buffer.seek(0)
  339. filename = f"bambuddy-backup-{datetime.now().strftime('%Y%m%d-%H%M%S')}.zip"
  340. return StreamingResponse(
  341. zip_buffer,
  342. media_type="application/zip",
  343. headers={"Content-Disposition": f"attachment; filename={filename}"},
  344. )
  345. except Exception as e:
  346. logger.error("Backup failed: %s", e, exc_info=True)
  347. return JSONResponse(
  348. status_code=500,
  349. content={"success": False, "message": "Backup failed. Check server logs for details."},
  350. )
  351. @router.post("/restore")
  352. async def restore_backup(
  353. file: UploadFile = File(...),
  354. db: AsyncSession = Depends(get_db),
  355. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_RESTORE),
  356. ):
  357. """Restore from a complete backup ZIP.
  358. This is a simplified restore that replaces the database and all data directories
  359. from the backup ZIP. Requires a restart after restore.
  360. """
  361. import shutil
  362. import tempfile
  363. from fastapi import HTTPException
  364. from backend.app.core.database import close_all_connections, init_db, reinitialize_database
  365. from backend.app.services.virtual_printer import virtual_printer_manager
  366. base_dir = app_settings.base_dir
  367. db_path = Path(app_settings.database_url.replace("sqlite+aiosqlite:///", ""))
  368. with tempfile.TemporaryDirectory() as temp_dir:
  369. temp_path = Path(temp_dir)
  370. # 1. Read and extract ZIP
  371. content = await file.read()
  372. # Check if it's a valid ZIP
  373. if not file.filename or not file.filename.endswith(".zip"):
  374. raise HTTPException(400, "Invalid backup file: must be a .zip file")
  375. try:
  376. with zipfile.ZipFile(io.BytesIO(content), "r") as zf:
  377. zf.extractall(temp_path)
  378. except zipfile.BadZipFile:
  379. raise HTTPException(400, "Invalid backup file: not a valid ZIP")
  380. # 2. Validate backup (must have database)
  381. backup_db = temp_path / "bambuddy.db"
  382. if not backup_db.exists():
  383. raise HTTPException(400, "Invalid backup: missing bambuddy.db")
  384. try:
  385. import asyncio
  386. # 3. Stop virtual printer if running (releases file locks)
  387. try:
  388. if virtual_printer_manager.is_enabled:
  389. logger.info("Stopping virtual printer for restore...")
  390. await virtual_printer_manager.configure(enabled=False)
  391. # Give it time to fully release file handles
  392. await asyncio.sleep(1)
  393. except Exception as e:
  394. logger.warning("Failed to stop virtual printer: %s", e)
  395. # 4. Close current database connections
  396. logger.info("Closing database connections...")
  397. await close_all_connections()
  398. # 5. Replace database
  399. logger.info("Restoring database from backup...")
  400. shutil.copy2(backup_db, db_path)
  401. # 6. Replace data directories
  402. # For Docker compatibility: clear contents then copy (don't delete mount points)
  403. dirs_to_restore = [
  404. ("archive", base_dir / "archive"),
  405. ("virtual_printer", base_dir / "virtual_printer"),
  406. ("plate_calibration", app_settings.plate_calibration_dir),
  407. ("icons", base_dir / "icons"),
  408. ("projects", base_dir / "projects"),
  409. ]
  410. skipped_dirs = []
  411. for name, dest_dir in dirs_to_restore:
  412. src_dir = temp_path / name
  413. if src_dir.exists():
  414. logger.info("Restoring %s directory...", name)
  415. try:
  416. # Clear destination contents (not the dir itself - may be Docker mount)
  417. if dest_dir.exists():
  418. for item in dest_dir.iterdir():
  419. try:
  420. if item.is_dir():
  421. shutil.rmtree(item)
  422. else:
  423. item.unlink()
  424. except OSError as e:
  425. logger.warning("Could not delete %s: %s", item, e)
  426. else:
  427. dest_dir.mkdir(parents=True, exist_ok=True)
  428. # Copy contents from backup
  429. for item in src_dir.iterdir():
  430. dest_item = dest_dir / item.name
  431. if item.is_dir():
  432. shutil.copytree(item, dest_item)
  433. else:
  434. shutil.copy2(item, dest_item)
  435. except OSError as e:
  436. logger.warning("Could not restore %s directory: %s", name, e)
  437. skipped_dirs.append(name)
  438. # 7. Reinitialize the database engine and apply schema migrations so that
  439. # tables added after the backup was created (e.g. ams_labels) exist
  440. # immediately, without requiring a manual restart.
  441. await reinitialize_database()
  442. await init_db()
  443. logger.info("Restore complete - restart required")
  444. message = "Backup restored successfully. Please restart Bambuddy for changes to take effect."
  445. if skipped_dirs:
  446. message += f" Note: Some directories could not be restored ({', '.join(skipped_dirs)})."
  447. return {
  448. "success": True,
  449. "message": message,
  450. }
  451. except Exception as e:
  452. logger.error("Restore failed: %s", e, exc_info=True)
  453. return JSONResponse(
  454. status_code=500,
  455. content={"success": False, "message": "Restore failed. Check server logs for details."},
  456. )
  457. @router.get("/network-interfaces")
  458. async def get_network_interfaces(
  459. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_READ),
  460. ):
  461. """Get available network interfaces with all IPs (primary + aliases)."""
  462. from backend.app.services.network_utils import get_all_interface_ips
  463. interfaces = get_all_interface_ips()
  464. return {"interfaces": interfaces}
  465. @router.get("/virtual-printer/models")
  466. async def get_virtual_printer_models(
  467. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_READ),
  468. ):
  469. """Get available virtual printer models."""
  470. from backend.app.services.virtual_printer import (
  471. DEFAULT_VIRTUAL_PRINTER_MODEL,
  472. VIRTUAL_PRINTER_MODELS,
  473. )
  474. return {
  475. "models": VIRTUAL_PRINTER_MODELS,
  476. "default": DEFAULT_VIRTUAL_PRINTER_MODEL,
  477. }
  478. @router.get("/virtual-printer")
  479. async def get_virtual_printer_settings(
  480. db: AsyncSession = Depends(get_db),
  481. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_READ),
  482. ):
  483. """Get virtual printer settings and status."""
  484. from backend.app.services.virtual_printer import (
  485. DEFAULT_VIRTUAL_PRINTER_MODEL,
  486. virtual_printer_manager,
  487. )
  488. enabled = await get_setting(db, "virtual_printer_enabled")
  489. access_code = await get_setting(db, "virtual_printer_access_code")
  490. mode = await get_setting(db, "virtual_printer_mode")
  491. model = await get_setting(db, "virtual_printer_model")
  492. target_printer_id = await get_setting(db, "virtual_printer_target_printer_id")
  493. remote_interface_ip = await get_setting(db, "virtual_printer_remote_interface_ip")
  494. return {
  495. "enabled": enabled == "true" if enabled else False,
  496. "access_code_set": bool(access_code),
  497. "mode": mode or "immediate",
  498. "model": model or DEFAULT_VIRTUAL_PRINTER_MODEL,
  499. "target_printer_id": int(target_printer_id) if target_printer_id else None,
  500. "remote_interface_ip": remote_interface_ip or "",
  501. "status": virtual_printer_manager.get_status(),
  502. }
  503. @router.put("/virtual-printer")
  504. async def update_virtual_printer_settings(
  505. enabled: bool = None,
  506. access_code: str = None,
  507. mode: str = None,
  508. model: str = None,
  509. target_printer_id: int = None,
  510. remote_interface_ip: str = None,
  511. db: AsyncSession = Depends(get_db),
  512. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_UPDATE),
  513. ):
  514. """Update virtual printer settings and restart services if needed.
  515. For proxy mode with SSDP proxy (dual-homed setup):
  516. - remote_interface_ip: IP of interface on slicer's network (LAN B)
  517. - Local interface is auto-detected based on target printer IP
  518. """
  519. from sqlalchemy import select
  520. from backend.app.models.printer import Printer
  521. from backend.app.services.virtual_printer import (
  522. DEFAULT_VIRTUAL_PRINTER_MODEL,
  523. VIRTUAL_PRINTER_MODELS,
  524. virtual_printer_manager,
  525. )
  526. # Get current values
  527. current_enabled = await get_setting(db, "virtual_printer_enabled") == "true"
  528. current_access_code = await get_setting(db, "virtual_printer_access_code") or ""
  529. current_mode = await get_setting(db, "virtual_printer_mode") or "immediate"
  530. current_model = await get_setting(db, "virtual_printer_model") or DEFAULT_VIRTUAL_PRINTER_MODEL
  531. current_target_id_str = await get_setting(db, "virtual_printer_target_printer_id")
  532. current_target_id = int(current_target_id_str) if current_target_id_str else None
  533. current_remote_iface = await get_setting(db, "virtual_printer_remote_interface_ip") or ""
  534. # Apply updates
  535. new_enabled = enabled if enabled is not None else current_enabled
  536. new_access_code = access_code if access_code is not None else current_access_code
  537. new_mode = mode if mode is not None else current_mode
  538. new_model = model if model is not None else current_model
  539. new_target_id = target_printer_id if target_printer_id is not None else current_target_id
  540. new_remote_iface = remote_interface_ip if remote_interface_ip is not None else current_remote_iface
  541. # Validate mode
  542. # "review" is the new name for "queue" (pending review before archiving)
  543. # "print_queue" archives and adds to print queue (unassigned)
  544. # "proxy" is transparent TCP proxy to a real printer
  545. if new_mode not in ("immediate", "queue", "review", "print_queue", "proxy"):
  546. return JSONResponse(
  547. status_code=400,
  548. content={"detail": "Mode must be 'immediate', 'review', 'print_queue', or 'proxy'"},
  549. )
  550. # Normalize legacy "queue" to "review" for storage
  551. if new_mode == "queue":
  552. new_mode = "review"
  553. # Validate model
  554. if model is not None and model not in VIRTUAL_PRINTER_MODELS:
  555. return JSONResponse(
  556. status_code=400,
  557. content={"detail": f"Invalid model. Must be one of: {', '.join(VIRTUAL_PRINTER_MODELS.keys())}"},
  558. )
  559. # Mode-specific validation and printer lookup
  560. target_printer_ip = ""
  561. target_printer_serial = ""
  562. if new_mode == "proxy":
  563. # Proxy mode requires target printer when enabling
  564. if new_enabled and not new_target_id:
  565. # If just switching to proxy mode (not explicitly enabling), auto-disable
  566. if enabled is None:
  567. new_enabled = False
  568. else:
  569. return JSONResponse(
  570. status_code=400,
  571. content={"detail": "Target printer is required for proxy mode"},
  572. )
  573. # Look up printer IP and serial if we have a target
  574. if new_target_id:
  575. result = await db.execute(select(Printer).where(Printer.id == new_target_id))
  576. printer = result.scalar_one_or_none()
  577. if not printer:
  578. return JSONResponse(
  579. status_code=400,
  580. content={"detail": f"Printer with ID {new_target_id} not found"},
  581. )
  582. target_printer_ip = printer.ip_address
  583. target_printer_serial = printer.serial_number
  584. # Access code not required for proxy mode
  585. else:
  586. # Non-proxy modes require access code when enabling
  587. if new_enabled and not new_access_code:
  588. # If just switching modes (not explicitly enabling), auto-disable
  589. if enabled is None:
  590. new_enabled = False
  591. else:
  592. return JSONResponse(
  593. status_code=400,
  594. content={"detail": "Access code is required when enabling virtual printer"},
  595. )
  596. # Validate access code length (Bambu Studio requires exactly 8 characters)
  597. if access_code is not None and access_code and len(access_code) != 8:
  598. return JSONResponse(
  599. status_code=400,
  600. content={"detail": "Access code must be exactly 8 characters"},
  601. )
  602. # Save settings
  603. await set_setting(db, "virtual_printer_enabled", "true" if new_enabled else "false")
  604. if access_code is not None:
  605. await set_setting(db, "virtual_printer_access_code", access_code)
  606. await set_setting(db, "virtual_printer_mode", new_mode)
  607. if model is not None:
  608. await set_setting(db, "virtual_printer_model", model)
  609. if target_printer_id is not None:
  610. await set_setting(db, "virtual_printer_target_printer_id", str(target_printer_id))
  611. if remote_interface_ip is not None:
  612. await set_setting(db, "virtual_printer_remote_interface_ip", remote_interface_ip)
  613. await db.commit()
  614. db.expire_all()
  615. # Reconfigure virtual printer
  616. try:
  617. await virtual_printer_manager.configure(
  618. enabled=new_enabled,
  619. access_code=new_access_code,
  620. mode=new_mode,
  621. model=new_model,
  622. target_printer_ip=target_printer_ip,
  623. target_printer_serial=target_printer_serial,
  624. remote_interface_ip=new_remote_iface,
  625. )
  626. except ValueError as e:
  627. logger.warning("Virtual printer configuration validation error: %s", e)
  628. return JSONResponse(
  629. status_code=400,
  630. content={"detail": "Invalid virtual printer configuration. Check the provided values."},
  631. )
  632. except Exception as e:
  633. logger.error("Failed to configure virtual printer: %s", e, exc_info=True)
  634. return JSONResponse(
  635. status_code=500,
  636. content={"detail": "Failed to configure virtual printer. Check server logs for details."},
  637. )
  638. return await get_virtual_printer_settings(db)
  639. # =============================================================================
  640. # MQTT Relay Settings
  641. # =============================================================================
  642. @router.get("/mqtt/status")
  643. async def get_mqtt_status(
  644. _: User | None = RequirePermissionIfAuthEnabled(Permission.SETTINGS_READ),
  645. ):
  646. """Get MQTT relay connection status."""
  647. from backend.app.services.mqtt_relay import mqtt_relay
  648. return mqtt_relay.get_status()