From face8951d231f71313f007185f1e020cb49b145e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gina=20H=C3=A4u=C3=9Fge?= <gina@octoprint.org>
Date: Wed, 6 May 2026 11:40:29 +0200
Subject: [PATCH 1/2] fix: quit bpm if connection to printer is lost

Should properly reset things into disconnection state for good.
---
 octoprint_bambu_connector/connector.py | 46 +++++++++++++++++++++-----
 1 file changed, 38 insertions(+), 8 deletions(-)

diff --git a/octoprint_bambu_connector/connector.py b/octoprint_bambu_connector/connector.py
index 22e32eb..c2acbae 100644
--- a/octoprint_bambu_connector/connector.py
+++ b/octoprint_bambu_connector/connector.py
@@ -36,6 +36,7 @@
 )
 from octoprint.printer.job import PrintJob
 from octoprint.schema import BaseModel
+from octoprint.util.version import is_version_compatible, safe_get_package_version
 from octoprint.util.tz import LOCAL_TZ
 
 GCODE_STATE_LOOKUP = {
@@ -238,6 +239,8 @@ def __init__(self, *args, **kwargs):
         self._job_stage = JobStage.UNKNOWN
 
         self._error = None
+        self._disconnecting = False
+        self._disconnect_thread = None
 
         self._progress: JobProgress = None
         self._old_progress: int = None
@@ -256,6 +259,8 @@ def __init__(self, *args, **kwargs):
                     f"Cannot load configured printer timezone {timezone_str}, falling back to server timezone"
                 )
 
+        self._bpm_version = safe_get_package_version("bambu-printer-manager")
+
     @property
     def connection_parameters(self):
         parameters = super().connection_parameters
@@ -405,9 +410,23 @@ def connect(self, *args, **kwargs):
     def disconnect(self, *args, **kwargs):
         if self._client is None:
             return
+
+        if self._disconnecting:
+            return
+        self._disconnecting = True
+
         eventManager().fire(Events.DISCONNECTING)
+        if is_version_compatible(self._bpm_version, "<=1.0.2"):
+            # workaround until synman/bambu-printer-manager#54 is merged
+            self._client._client.disconnect()
         self._client.quit()
-        self.set_state(ConnectedPrinterState.CLOSED)
+        self.set_state(
+            ConnectedPrinterState.CLOSED
+        )  # should already have been set through self._client.quit(), but better safe than sorry
+
+        # clean up references
+        self._client.on_update = None
+        self._client = None
 
     def emergency_stop(self, *args, **kwargs):
         self.commands("M112", tags=kwargs.get("tags", set()))
@@ -991,13 +1010,24 @@ def _update_state_from_state(self, printer: bpm.bambuprinter.BambuPrinter):
             elif self._gcode_state == GcodeState.OFFLINE:
                 new_state = ConnectedPrinterState.CLOSED
 
-        else:
-            internal_error = printer.internalException
-            if internal_error:
-                new_state = ConnectedPrinterState.CLOSED_WITH_ERROR
-                error = str(internal_error)
-            else:
-                new_state = ConnectedPrinterState.CLOSED
+        elif self._connection_state == bpm.bambuprinter.ServiceState.DISCONNECTED:
+            if not self._disconnecting and not self._disconnect_thread:
+                self._logger.warning("Lost connection to printer")
+                self._disconnect_thread = threading.Thread(
+                    target=self.disconnect
+                ).start()  # decouple this call from the status update thread or bpm will run into an issue on thread join in `quit`
+
+        elif self._connection_state == bpm.bambuprinter.ServiceState.QUIT:
+            if self.state not in (
+                ConnectedPrinterState.CLOSED,
+                ConnectedPrinterState.CLOSED_WITH_ERROR,
+            ):
+                internal_error = printer.internalException
+                if internal_error:
+                    new_state = ConnectedPrinterState.CLOSED_WITH_ERROR
+                    error = str(internal_error)
+                else:
+                    new_state = ConnectedPrinterState.CLOSED
 
         if new_state:
             self._state_context = (new_state, printer.active_job_info.stage_name)

From 8d61727979083fb51fb04dc55a78773dcd0e0fb0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gina=20H=C3=A4u=C3=9Fge?= <gina@octoprint.org>
Date: Wed, 6 May 2026 11:59:33 +0200
Subject: [PATCH 2/2] fix(ux): report connection loss in terminal

---
 octoprint_bambu_connector/connector.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/octoprint_bambu_connector/connector.py b/octoprint_bambu_connector/connector.py
index c2acbae..30cffdb 100644
--- a/octoprint_bambu_connector/connector.py
+++ b/octoprint_bambu_connector/connector.py
@@ -1012,7 +1012,9 @@ def _update_state_from_state(self, printer: bpm.bambuprinter.BambuPrinter):
 
         elif self._connection_state == bpm.bambuprinter.ServiceState.DISCONNECTED:
             if not self._disconnecting and not self._disconnect_thread:
-                self._logger.warning("Lost connection to printer")
+                message = "Lost connection to printer"
+                self._logger.warning(message)
+                self._listener.on_printer_logs(message)
                 self._disconnect_thread = threading.Thread(
                     target=self.disconnect
                 ).start()  # decouple this call from the status update thread or bpm will run into an issue on thread join in `quit`