mqtt_server.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843
  1. """MQTT broker for virtual printer.
  2. Implements an MQTT broker that accepts connections from slicers,
  3. authenticates with the configured access code, and logs print commands.
  4. """
  5. import asyncio
  6. import json
  7. import logging
  8. import ssl
  9. from collections.abc import Callable
  10. from pathlib import Path
  11. logger = logging.getLogger(__name__)
  12. # Default MQTT port for Bambu printers (MQTT over TLS)
  13. MQTT_PORT = 8883
  14. # Model code → product_name for version response (must match what slicer expects)
  15. MODEL_PRODUCT_NAMES = {
  16. "BL-P001": "X1 Carbon",
  17. "BL-P002": "X1",
  18. "C13": "X1E",
  19. "C11": "P1P",
  20. "C12": "P1S",
  21. "N7": "P2S",
  22. "N2S": "A1",
  23. "N1": "A1 mini",
  24. "O1D": "H2D",
  25. "O1C": "H2C",
  26. "O1C2": "H2C",
  27. "O1S": "H2S",
  28. }
  29. class VirtualPrinterMQTTServer:
  30. """MQTT broker that accepts connections from slicers.
  31. This is a minimal MQTT broker implementation that:
  32. - Accepts TLS connections on port 8883
  33. - Authenticates with username 'bblp' and the configured access code
  34. - Receives print commands on device/{serial}/request
  35. - Can publish status on device/{serial}/report
  36. """
  37. def __init__(
  38. self,
  39. serial: str,
  40. access_code: str,
  41. cert_path: Path,
  42. key_path: Path,
  43. port: int = MQTT_PORT,
  44. on_print_command: Callable[[str, dict], None] | None = None,
  45. ):
  46. """Initialize the MQTT server.
  47. Args:
  48. serial: Virtual printer serial number
  49. access_code: Password for authentication
  50. cert_path: Path to TLS certificate
  51. key_path: Path to TLS private key
  52. port: Port to listen on (default 8883)
  53. on_print_command: Callback when print command received (filename, data)
  54. """
  55. self.serial = serial
  56. self.access_code = access_code
  57. self.cert_path = cert_path
  58. self.key_path = key_path
  59. self.port = port
  60. self.on_print_command = on_print_command
  61. self._running = False
  62. self._broker = None
  63. self._broker_task = None
  64. async def start(self) -> None:
  65. """Start the MQTT broker."""
  66. if self._running:
  67. return
  68. # Try to import amqtt
  69. try:
  70. from amqtt.broker import Broker
  71. except ImportError:
  72. logger.error("amqtt not installed. Run: pip install amqtt")
  73. return
  74. logger.info("Starting virtual printer MQTT broker on port %s", self.port)
  75. # Build broker configuration
  76. config = {
  77. "listeners": {
  78. "default": {
  79. "type": "tcp",
  80. "bind": f"0.0.0.0:{self.port}",
  81. "ssl": "on",
  82. "certfile": str(self.cert_path),
  83. "keyfile": str(self.key_path),
  84. },
  85. },
  86. "auth": {
  87. "allow-anonymous": False,
  88. "plugins": ["auth_custom"],
  89. },
  90. "topic-check": {
  91. "enabled": False, # Allow any topic
  92. },
  93. }
  94. try:
  95. self._running = True
  96. # Create and start broker
  97. self._broker = Broker(config)
  98. # Register custom auth plugin
  99. self._broker.plugins_manager.plugins_handlers["auth_custom"] = self._authenticate
  100. # Start the broker
  101. await self._broker.start()
  102. logger.info("MQTT broker started on port %s", self.port)
  103. # Keep running
  104. while self._running:
  105. await asyncio.sleep(1)
  106. except OSError as e:
  107. if e.errno == 98: # Address already in use
  108. logger.error("MQTT port %s is already in use", self.port)
  109. else:
  110. logger.error("MQTT broker error: %s", e)
  111. except asyncio.CancelledError:
  112. logger.debug("MQTT broker task cancelled")
  113. except Exception as e:
  114. logger.error("MQTT broker error: %s", e)
  115. finally:
  116. await self.stop()
  117. async def _authenticate(self, session) -> bool:
  118. """Authenticate MQTT connection.
  119. Args:
  120. session: MQTT session with username/password
  121. Returns:
  122. True if authentication successful
  123. """
  124. username = getattr(session, "username", None)
  125. password = getattr(session, "password", None)
  126. # Bambu slicers use 'bblp' as username and access code as password
  127. if username == "bblp" and password == self.access_code:
  128. logger.debug("MQTT client authenticated from %s", session.remote_address)
  129. return True
  130. logger.warning("MQTT auth failed for user '%s' from %s", username, session.remote_address)
  131. return False
  132. async def stop(self) -> None:
  133. """Stop the MQTT broker."""
  134. logger.info("Stopping MQTT broker")
  135. self._running = False
  136. if self._broker:
  137. try:
  138. await self._broker.shutdown()
  139. except OSError as e:
  140. logger.debug("Error shutting down MQTT broker: %s", e)
  141. self._broker = None
  142. class SimpleMQTTServer:
  143. """Simplified MQTT server using raw sockets.
  144. This is a fallback implementation that handles basic MQTT protocol
  145. without requiring the amqtt library. It's less feature-complete but
  146. more lightweight.
  147. """
  148. def __init__(
  149. self,
  150. serial: str,
  151. access_code: str,
  152. cert_path: Path,
  153. key_path: Path,
  154. port: int = MQTT_PORT,
  155. on_print_command: Callable[[str, dict], None] | None = None,
  156. model: str = "",
  157. bind_address: str = "0.0.0.0", # nosec B104
  158. vp_name: str = "",
  159. ):
  160. self.serial = serial
  161. self.access_code = access_code
  162. self.model = model
  163. self.cert_path = cert_path
  164. self.key_path = key_path
  165. self.port = port
  166. self.on_print_command = on_print_command
  167. self.bind_address = bind_address
  168. self.vp_name = vp_name
  169. self._log_prefix = f"[{vp_name}] " if vp_name else ""
  170. self._running = False
  171. self._server = None
  172. self._clients: dict[str, asyncio.StreamWriter] = {}
  173. self._status_push_task: asyncio.Task | None = None
  174. self._sequence_id = 0
  175. # Dynamic state for status reports
  176. self._gcode_state = "IDLE"
  177. self._current_file = ""
  178. self._prepare_percent = "0"
  179. async def start(self) -> None:
  180. """Start the MQTT server."""
  181. if self._running:
  182. return
  183. logger.info("Starting simple MQTT server on port %s", self.port)
  184. # Create SSL context with Bambu-compatible settings
  185. ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
  186. ssl_context.load_cert_chain(str(self.cert_path), str(self.key_path))
  187. # Match Bambu printer behavior - accept any client
  188. ssl_context.verify_mode = ssl.CERT_NONE
  189. # Allow TLS 1.2 for broader compatibility (some slicers may not support 1.3)
  190. ssl_context.minimum_version = ssl.TLSVersion.TLSv1_2
  191. # Disable hostname checking
  192. ssl_context.check_hostname = False
  193. # Log certificate info
  194. import subprocess
  195. try:
  196. result = subprocess.run(
  197. ["openssl", "x509", "-in", str(self.cert_path), "-noout", "-subject", "-issuer"],
  198. capture_output=True,
  199. text=True,
  200. timeout=5,
  201. )
  202. logger.info("MQTT SSL cert info: %s", result.stdout.strip())
  203. except (OSError, subprocess.SubprocessError):
  204. pass # Certificate info is for debug logging only; not critical
  205. logger.info("MQTT SSL context: TLS 1.2+, cert=%s", self.cert_path)
  206. try:
  207. self._running = True
  208. # Wrapper to log ALL connection attempts including SSL errors
  209. async def connection_handler(reader, writer):
  210. try:
  211. addr = writer.get_extra_info("peername")
  212. ssl_obj = writer.get_extra_info("ssl_object")
  213. if ssl_obj:
  214. logger.info(
  215. f"{self._log_prefix}MQTT TLS connection from {addr} - cipher={ssl_obj.cipher()}, version={ssl_obj.version()}"
  216. )
  217. else:
  218. logger.info("%sMQTT connection from %s (no TLS?)", self._log_prefix, addr)
  219. await self._handle_client(reader, writer)
  220. except ssl.SSLError as e:
  221. logger.error("MQTT SSL error: %s", e)
  222. except Exception as e:
  223. logger.error("MQTT connection handler error: %s", e)
  224. # Custom protocol factory to log raw connection attempts
  225. logger.info("Setting up MQTT server with SSL error handling...")
  226. # Add SSL handshake error callback
  227. def handle_ssl_error(loop, context):
  228. exception = context.get("exception")
  229. message = context.get("message", "")
  230. if "ssl" in str(exception).lower() or "ssl" in message.lower():
  231. logger.error("SSL error: %s - %s", message, exception)
  232. else:
  233. logger.debug("Asyncio error: %s", message)
  234. asyncio.get_event_loop().set_exception_handler(handle_ssl_error)
  235. self._server = await asyncio.start_server(
  236. connection_handler,
  237. self.bind_address,
  238. self.port,
  239. ssl=ssl_context,
  240. )
  241. logger.info("Simple MQTT server listening on port %s", self.port)
  242. # Start periodic status push task
  243. self._status_push_task = asyncio.create_task(self._periodic_status_push())
  244. async with self._server:
  245. await self._server.serve_forever()
  246. except OSError as e:
  247. if e.errno == 98: # Address already in use
  248. logger.error("MQTT port %s is already in use", self.port)
  249. else:
  250. logger.error("MQTT server error: %s", e)
  251. except asyncio.CancelledError:
  252. logger.debug("MQTT server task cancelled")
  253. except Exception as e:
  254. logger.error("MQTT server error: %s", e)
  255. finally:
  256. await self.stop()
  257. async def stop(self) -> None:
  258. """Stop the MQTT server."""
  259. logger.info("Stopping simple MQTT server")
  260. self._running = False
  261. # Stop periodic status push
  262. if self._status_push_task:
  263. self._status_push_task.cancel()
  264. try:
  265. await self._status_push_task
  266. except asyncio.CancelledError:
  267. pass # Expected when stopping the periodic status push task
  268. self._status_push_task = None
  269. # Close all client connections (iterate over copy to avoid modification during iteration)
  270. for _client_id, writer in list(self._clients.items()):
  271. try:
  272. writer.close()
  273. await writer.wait_closed()
  274. except OSError:
  275. pass # Best-effort client connection cleanup; client may have disconnected
  276. self._clients.clear()
  277. if self._server:
  278. try:
  279. self._server.close()
  280. await self._server.wait_closed()
  281. except OSError:
  282. pass # Best-effort server shutdown; port may already be released
  283. self._server = None
  284. async def _periodic_status_push(self) -> None:
  285. """Send periodic status updates to all connected clients."""
  286. logger.info("Starting periodic status push task")
  287. while self._running:
  288. try:
  289. await asyncio.sleep(1) # Push every 1 second like real printers
  290. # Send status to all connected clients
  291. disconnected = []
  292. for client_id, writer in list(self._clients.items()):
  293. try:
  294. if writer.is_closing():
  295. disconnected.append(client_id)
  296. continue
  297. await self._send_status_report(writer)
  298. except OSError as e:
  299. logger.debug("Failed to push status to %s: %s", client_id, e)
  300. disconnected.append(client_id)
  301. # Remove disconnected clients
  302. for client_id in disconnected:
  303. self._clients.pop(client_id, None)
  304. except asyncio.CancelledError:
  305. break
  306. except Exception as e:
  307. logger.error("Periodic status push error: %s", e)
  308. logger.info("Periodic status push task stopped")
  309. async def _handle_client(self, reader: asyncio.StreamReader, writer: asyncio.StreamWriter) -> None:
  310. """Handle an MQTT client connection."""
  311. addr = writer.get_extra_info("peername")
  312. client_id = f"{addr[0]}:{addr[1]}" if addr else "unknown"
  313. logger.info("%sMQTT client connected: %s", self._log_prefix, client_id)
  314. authenticated = False
  315. try:
  316. while self._running:
  317. # Read MQTT fixed header
  318. try:
  319. header = await asyncio.wait_for(reader.read(1), timeout=60)
  320. except TimeoutError:
  321. break
  322. if not header:
  323. break
  324. packet_type = (header[0] & 0xF0) >> 4
  325. # Read remaining length
  326. remaining_length = await self._read_remaining_length(reader)
  327. if remaining_length is None:
  328. break
  329. # Read payload
  330. payload = await reader.read(remaining_length) if remaining_length > 0 else b""
  331. # Handle packet types
  332. if packet_type == 1: # CONNECT
  333. authenticated = await self._handle_connect(payload, writer)
  334. if not authenticated:
  335. break
  336. # Register client for periodic status pushes
  337. self._clients[client_id] = writer
  338. elif packet_type == 3: # PUBLISH
  339. if authenticated:
  340. await self._handle_publish(header[0], payload, writer)
  341. elif packet_type == 8: # SUBSCRIBE
  342. if authenticated:
  343. await self._handle_subscribe(payload, writer)
  344. elif packet_type == 12: # PINGREQ
  345. # Send PINGRESP
  346. writer.write(bytes([0xD0, 0x00]))
  347. await writer.drain()
  348. elif packet_type == 14: # DISCONNECT
  349. break
  350. except asyncio.CancelledError:
  351. pass # Expected when server is shutting down and cancels client tasks
  352. except Exception as e:
  353. logger.debug("MQTT client error: %s", e)
  354. finally:
  355. logger.debug("MQTT client disconnected: %s", client_id)
  356. if client_id in self._clients:
  357. del self._clients[client_id]
  358. try:
  359. writer.close()
  360. await writer.wait_closed()
  361. except OSError:
  362. pass # Best-effort socket cleanup on client disconnect
  363. async def _read_remaining_length(self, reader: asyncio.StreamReader) -> int | None:
  364. """Read MQTT remaining length (variable byte integer)."""
  365. multiplier = 1
  366. value = 0
  367. for _ in range(4):
  368. try:
  369. byte = await reader.read(1)
  370. if not byte:
  371. return None
  372. encoded = byte[0]
  373. value += (encoded & 127) * multiplier
  374. if (encoded & 128) == 0:
  375. return value
  376. multiplier *= 128
  377. except OSError:
  378. return None
  379. return None
  380. async def _handle_connect(self, payload: bytes, writer: asyncio.StreamWriter) -> bool:
  381. """Handle MQTT CONNECT packet.
  382. Returns True if authentication successful.
  383. """
  384. try:
  385. # Parse CONNECT packet
  386. # Skip protocol name length and name
  387. idx = 0
  388. proto_len = (payload[idx] << 8) | payload[idx + 1]
  389. idx += 2 + proto_len
  390. # Skip protocol level and connect flags
  391. # connect_flags = payload[idx + 1]
  392. idx += 2
  393. # Skip keepalive
  394. idx += 2
  395. # Read client ID
  396. client_id_len = (payload[idx] << 8) | payload[idx + 1]
  397. idx += 2
  398. # client_id = payload[idx : idx + client_id_len].decode("utf-8")
  399. idx += client_id_len
  400. # Read username
  401. username_len = (payload[idx] << 8) | payload[idx + 1]
  402. idx += 2
  403. username = payload[idx : idx + username_len].decode("utf-8")
  404. idx += username_len
  405. # Read password
  406. password_len = (payload[idx] << 8) | payload[idx + 1]
  407. idx += 2
  408. password = payload[idx : idx + password_len].decode("utf-8")
  409. # Authenticate
  410. if username == "bblp" and password == self.access_code:
  411. # Send CONNACK with success
  412. writer.write(bytes([0x20, 0x02, 0x00, 0x00]))
  413. await writer.drain()
  414. logger.info("%sMQTT client authenticated successfully", self._log_prefix)
  415. # Send immediate status report after auth - slicer expects this
  416. await self._send_status_report(writer)
  417. return True
  418. else:
  419. # Send CONNACK with auth failure
  420. writer.write(bytes([0x20, 0x02, 0x00, 0x05])) # Not authorized
  421. await writer.drain()
  422. logger.warning("%sMQTT auth failed for user '%s' (access code mismatch)", self._log_prefix, username)
  423. return False
  424. except (IndexError, ValueError) as e:
  425. logger.debug("MQTT CONNECT parse error: %s", e)
  426. # Send CONNACK with error
  427. writer.write(bytes([0x20, 0x02, 0x00, 0x02])) # Protocol error
  428. await writer.drain()
  429. return False
  430. async def _handle_subscribe(self, payload: bytes, writer: asyncio.StreamWriter) -> None:
  431. """Handle MQTT SUBSCRIBE packet."""
  432. try:
  433. # Parse packet ID
  434. packet_id = (payload[0] << 8) | payload[1]
  435. # Parse topic filters (just acknowledge them)
  436. idx = 2
  437. granted_qos = []
  438. while idx < len(payload):
  439. topic_len = (payload[idx] << 8) | payload[idx + 1]
  440. idx += 2
  441. topic = payload[idx : idx + topic_len].decode("utf-8")
  442. idx += topic_len
  443. requested_qos = payload[idx]
  444. idx += 1
  445. logger.info("%sMQTT subscribe: %s QoS=%s", self._log_prefix, topic, requested_qos)
  446. granted_qos.append(min(requested_qos, 1)) # Grant up to QoS 1
  447. # Send SUBACK
  448. suback = bytes([0x90, 2 + len(granted_qos), packet_id >> 8, packet_id & 0xFF])
  449. suback += bytes(granted_qos)
  450. writer.write(suback)
  451. await writer.drain()
  452. # Send initial status report after subscribe
  453. await self._send_status_report(writer)
  454. except (IndexError, ValueError, OSError) as e:
  455. logger.debug("MQTT SUBSCRIBE error: %s", e)
  456. async def _send_status_report(self, writer: asyncio.StreamWriter) -> None:
  457. """Send a status report to the slicer after connection."""
  458. try:
  459. # Build status message matching Bambu printer format
  460. self._sequence_id += 1
  461. status = {
  462. "print": {
  463. "sequence_id": str(self._sequence_id),
  464. "command": "push_status",
  465. "msg": 0,
  466. "gcode_state": self._gcode_state,
  467. "gcode_file": self._current_file,
  468. "gcode_file_prepare_percent": self._prepare_percent,
  469. "subtask_name": self._current_file.replace(".3mf", "") if self._current_file else "",
  470. "mc_print_stage": "",
  471. "mc_percent": 0,
  472. "mc_remaining_time": 0,
  473. "wifi_signal": "-44dBm",
  474. "print_error": 0,
  475. "print_type": "",
  476. "bed_temper": 25.0,
  477. "bed_target_temper": 0.0,
  478. "nozzle_temper": 25.0,
  479. "nozzle_target_temper": 0.0,
  480. "chamber_temper": 25.0,
  481. "cooling_fan_speed": "0",
  482. "big_fan1_speed": "0",
  483. "big_fan2_speed": "0",
  484. "heatbreak_fan_speed": "0",
  485. "spd_lvl": 1,
  486. "spd_mag": 100,
  487. "stg": [],
  488. "stg_cur": 0,
  489. "layer_num": 0,
  490. "total_layer_num": 0,
  491. "home_flag": 256, # Bit 8 = SD card present (HAS_SDCARD_NORMAL)
  492. "hw_switch_state": 0,
  493. "online": {"ahb": False, "rfid": False, "version": 7},
  494. "ams_status": 0,
  495. "sdcard": True,
  496. "storage": {"free": 1000000000, "total": 32000000000},
  497. "upgrade_state": {
  498. "sequence_id": 0,
  499. "progress": "",
  500. "status": "",
  501. "consistency_request": False,
  502. "dis_state": 0,
  503. "err_code": 0,
  504. "force_upgrade": False,
  505. "message": "",
  506. "module": "",
  507. "new_version_state": 2,
  508. "new_ver_list": [],
  509. "ota_new_version_number": "",
  510. "ahb_new_version_number": "",
  511. },
  512. "ipcam": {
  513. "ipcam_dev": "1",
  514. "ipcam_record": "enable",
  515. "timelapse": "disable",
  516. "resolution": "1080p",
  517. "mode_bits": 0,
  518. },
  519. "xcam": {
  520. "allow_skip_parts": False,
  521. "buildplate_marker_detector": True,
  522. "first_layer_inspector": True,
  523. "halt_print_sensitivity": "medium",
  524. "print_halt": True,
  525. "printing_monitor": True,
  526. "spaghetti_detector": True,
  527. },
  528. "lights_report": [{"node": "chamber_light", "mode": "on"}],
  529. "nozzle_diameter": "0.4",
  530. "nozzle_type": "hardened_steel",
  531. }
  532. }
  533. await self._publish_to_report(writer, status, self.serial)
  534. except OSError as e:
  535. logger.error("Failed to send status report: %s", e)
  536. async def _send_version_response(self, writer: asyncio.StreamWriter, sequence_id: str) -> None:
  537. """Send version info response to the slicer."""
  538. try:
  539. product_name = MODEL_PRODUCT_NAMES.get(self.model, self.model or "X1 Carbon")
  540. serial = self.serial
  541. # Build version response matching OrcaSlicer expectations
  542. # Required fields per module: name, product_name, sw_ver, sw_new_ver, sn, hw_ver, flag
  543. version_info = {
  544. "info": {
  545. "command": "get_version",
  546. "sequence_id": sequence_id,
  547. "module": [
  548. {
  549. "name": "ota",
  550. "product_name": product_name,
  551. "sw_ver": "01.07.00.00",
  552. "sw_new_ver": "",
  553. "hw_ver": "OTA",
  554. "sn": serial,
  555. "flag": 0,
  556. },
  557. {
  558. "name": "esp32",
  559. "product_name": product_name,
  560. "sw_ver": "01.07.22.25",
  561. "sw_new_ver": "",
  562. "hw_ver": "AP05",
  563. "sn": serial,
  564. "flag": 0,
  565. },
  566. {
  567. "name": "rv1126",
  568. "product_name": product_name,
  569. "sw_ver": "00.00.27.38",
  570. "sw_new_ver": "",
  571. "hw_ver": "AP05",
  572. "sn": serial,
  573. "flag": 0,
  574. },
  575. {
  576. "name": "th",
  577. "product_name": product_name,
  578. "sw_ver": "00.00.04.00",
  579. "sw_new_ver": "",
  580. "hw_ver": "TH07",
  581. "sn": serial,
  582. "flag": 0,
  583. },
  584. {
  585. "name": "mc",
  586. "product_name": product_name,
  587. "sw_ver": "00.00.10.00",
  588. "sw_new_ver": "",
  589. "hw_ver": "MC07",
  590. "sn": serial,
  591. "flag": 0,
  592. },
  593. ],
  594. }
  595. }
  596. await self._publish_to_report(writer, version_info, serial)
  597. logger.info("Sent version response (product_name=%s)", product_name)
  598. except OSError as e:
  599. logger.error("Failed to send version response: %s", e)
  600. def set_gcode_state(self, state: str, filename: str = "", prepare_percent: str = "0") -> None:
  601. """Update the gcode state reported to connected slicers.
  602. Called by the manager to reflect FTP upload progress/completion.
  603. """
  604. self._gcode_state = state
  605. self._current_file = filename
  606. self._prepare_percent = prepare_percent
  607. async def _publish_to_report(self, writer: asyncio.StreamWriter, payload: dict, serial: str = "") -> None:
  608. """Publish a message on the device report topic."""
  609. topic = f"device/{serial or self.serial}/report"
  610. message = json.dumps(payload)
  611. topic_bytes = topic.encode("utf-8")
  612. message_bytes = message.encode("utf-8")
  613. remaining = 2 + len(topic_bytes) + len(message_bytes)
  614. packet = bytes([0x30]) # PUBLISH, QoS 0
  615. while remaining > 0:
  616. byte = remaining % 128
  617. remaining //= 128
  618. if remaining > 0:
  619. byte |= 0x80
  620. packet += bytes([byte])
  621. packet += bytes([len(topic_bytes) >> 8, len(topic_bytes) & 0xFF])
  622. packet += topic_bytes
  623. packet += message_bytes
  624. writer.write(packet)
  625. # Timeout the drain to prevent blocking the event loop if the
  626. # MQTT client stops reading (e.g. slicer busy with FTP upload).
  627. try:
  628. await asyncio.wait_for(writer.drain(), timeout=5)
  629. except TimeoutError:
  630. logger.debug("MQTT drain timeout for %s — client may be busy", topic)
  631. async def _send_print_response(self, writer: asyncio.StreamWriter, sequence_id: str, filename: str) -> None:
  632. """Send project_file acknowledgment matching real Bambu printer behavior."""
  633. # Update state so periodic status pushes reflect preparation
  634. self._gcode_state = "PREPARE"
  635. self._current_file = filename
  636. self._prepare_percent = "0"
  637. try:
  638. # Send command acknowledgment — slicer expects to see
  639. # command: "project_file" echoed back before starting FTP upload
  640. subtask_name = filename.replace(".3mf", "") if filename else ""
  641. response = {
  642. "print": {
  643. "command": "project_file",
  644. "sequence_id": sequence_id,
  645. "param": "Metadata/plate_1.gcode",
  646. "subtask_name": subtask_name,
  647. "gcode_state": "PREPARE",
  648. "gcode_file": filename,
  649. "gcode_file_prepare_percent": "0",
  650. "result": "SUCCESS",
  651. "msg": 0,
  652. }
  653. }
  654. await self._publish_to_report(writer, response)
  655. logger.info("Sent project_file acknowledgment for %s", filename)
  656. except OSError as e:
  657. logger.error("Failed to send print response: %s", e)
  658. async def _handle_publish(self, header: int, payload: bytes, writer: asyncio.StreamWriter) -> None:
  659. """Handle MQTT PUBLISH packet."""
  660. try:
  661. # Parse topic
  662. idx = 0
  663. topic_len = (payload[idx] << 8) | payload[idx + 1]
  664. idx += 2
  665. topic = payload[idx : idx + topic_len].decode("utf-8")
  666. idx += topic_len
  667. # Check for packet ID (QoS > 0)
  668. qos = (header & 0x06) >> 1
  669. if qos > 0:
  670. # packet_id = (payload[idx] << 8) | payload[idx + 1]
  671. idx += 2
  672. # Parse message
  673. message = payload[idx:].decode("utf-8")
  674. logger.info("MQTT publish to %s: %s...", topic, message[:100])
  675. # Handle commands on device request topic
  676. if f"device/{self.serial}/request" in topic:
  677. try:
  678. data = json.loads(message)
  679. # Handle pushing command (status request)
  680. if "pushing" in data:
  681. pushing_data = data["pushing"]
  682. command = pushing_data.get("command", "")
  683. logger.info("MQTT pushing command: %s", command)
  684. if command == "pushall":
  685. # Slicer is requesting full status - send response
  686. logger.info("Sending status report in response to pushall")
  687. await self._send_status_report(writer)
  688. elif command == "start":
  689. # Slicer wants periodic status updates - send one now
  690. logger.info("Starting status push stream")
  691. await self._send_status_report(writer)
  692. # Handle info commands (get_version, etc.)
  693. if "info" in data:
  694. info_data = data["info"]
  695. command = info_data.get("command", "")
  696. sequence_id = info_data.get("sequence_id", "0")
  697. logger.info("MQTT info command: %s", command)
  698. if command == "get_version":
  699. await self._send_version_response(writer, sequence_id)
  700. # Handle print commands
  701. if "print" in data:
  702. print_data = data["print"]
  703. command = print_data.get("command", "")
  704. filename = print_data.get("subtask_name", "")
  705. sequence_id = print_data.get("sequence_id", "0")
  706. logger.info("MQTT print command: %s for %s", command, filename)
  707. if command == "project_file":
  708. # Respond with PREPARE status so slicer proceeds with FTP upload
  709. file_3mf = print_data.get("file", filename)
  710. await self._send_print_response(writer, sequence_id, file_3mf)
  711. if self.on_print_command:
  712. await self._notify_print_command(filename, print_data)
  713. except json.JSONDecodeError:
  714. pass # Non-JSON payloads on request topic are safely ignored
  715. except (IndexError, ValueError, OSError) as e:
  716. logger.debug("MQTT PUBLISH error: %s", e)
  717. async def _notify_print_command(self, filename: str, data: dict) -> None:
  718. """Notify callback of print command."""
  719. if self.on_print_command:
  720. try:
  721. result = self.on_print_command(filename, data)
  722. if asyncio.iscoroutine(result):
  723. await result
  724. except Exception as e:
  725. logger.error("Print command callback error: %s", e)