Add klipper service restart button

Author: TheSin-

octoprint_moonraker_connector/__init__.py

@@ -3,6 +3,7 @@
 from urllib.parse import urljoin
 
 import requests
+import subprocess
 from flask import jsonify
 from flask_babel import gettext
 
@@ -72,6 +73,7 @@ def _configure_json_rpc_logging(self):
     def get_settings_defaults(self):
         return {
             "emergency_stop_on_cancel": False,
+            "restart_klipper_service_command": "sudo systemctl restart klipper.service",
         }
 
     ##~~ SimpleApiPlugin
@@ -93,6 +95,24 @@ def on_api_get(self, request):
         )
         return jsonify(response.model_dump(by_alias=True))
 
+    def get_api_commands(self):
+        return dict(
+            restart_klipper_service=[]
+        )
+
+    def on_api_command(self, command, data):
+        if command == 'restart_klipper_service':
+            configured_command = self._settings.get(["restart_klipper_service_command"])
+
+            try:
+                # Use subprocess.Popen for a non-blocking execution
+                subprocess.Popen(configured_command, shell=True) 
+                return jsonify({"status": "executed", "command": configured_command})
+            except Exception as e:
+                return jsonify({"status": "error", "message": str(e)}), 500
+
+        return jsonify({"status": "error", "message": "Invalid command"}), 400
+
     def is_api_protected(self):
         return True
 

octoprint_moonraker_connector/static/clientjs/moonraker_connector.js

@@ -13,6 +13,10 @@
         return this.base.simpleApiGet("moonraker_connector", opts);
     };
 
+    OctoPrintMoonrakerConnectorClient.prototype.command = function (command, payload) {
+        return this.base.simpleApiCommand("moonraker_connector", command, payload);
+    };
+
     OctoPrintClient.registerPluginComponent(
         "moonraker_connector",
         OctoPrintMoonrakerConnectorClient

octoprint_moonraker_connector/static/js/moonraker_connector.js

@@ -15,6 +15,30 @@ $(function () {
             OctoPrint.control.sendGcode('FIRMWARE_RESTART');
         }
 
+        self.btnKlipperRestartClick = function() {
+            showConfirmationDialog({
+                message: gettext("<strong>This will restart the Klipper service.</strong></p><p>This might disrupt any ongoing operations related to Klipper."),
+                onproceed: function() {
+                    // Execute the registered system command
+                    OctoPrint.plugins.moonraker_connector.command({
+                        command: 'restart_klipper_service'
+                    }).done(function(response) {
+                        new PNotify({
+                            title: gettext("Success"),
+                            text: gettext("System command sent: ") + response.command,
+                            type: "success"
+                        });
+                    }).fail(function(jqXHR, textStatus, errorThrown) {
+                        new PNotify({
+                            title: gettext("Error"),
+                            text: gettext("Failed to execute command: ") + jqXHR.responseJSON.message,
+                            type: "error"
+                        });
+                    });
+                }
+            });
+        }
+
         self.initializeButton = function() {
             var buttonContainer = $('#job_print')[0].parentElement;
             var container = document.createElement("div");
@@ -24,39 +48,66 @@ $(function () {
 
             var btnRestart = document.createElement("button");
             btnRestart.id = "job_restart";
-            btnRestart.title = "Reload configuration file and performs an internal reset of the host software. It does not clear the error state from the micro-controller.";
+            btnRestart.title = gettext("Reload configuration file and performs an internal reset of the host software. It does not clear the error state from the micro-controller.");
             btnRestart.classList.add("btn");
             btnRestart.classList.add("span6");
             btnRestart.addEventListener("click", self.btnRestartClick);
 
             var btnRestartIcon = document.createElement("i");
-            btnRestartIcon.classList.add("fa", "fa-redo");
+            btnRestartIcon.classList.add("fas", "fa-sync-alt");
+            btnRestartIcon.style.marginRight = "5px";
             btnRestart.appendChild(btnRestartIcon);
 
             var btnRestartText = document.createElement("span");
-            btnRestartText.textContent = " Restart";
+            btnRestartText.textContent = gettext("Restart");
             btnRestart.appendChild(btnRestartText);
 
             container.appendChild(btnRestart);
 
             var btnFirmwareRestart = document.createElement("button");
             btnFirmwareRestart.id = "job_firmware_restart";
-            btnFirmwareRestart.title = "Reload configuration file and performs an internal reset of the host software, but it also clears any error states from the micro-controller.";
+            btnFirmwareRestart.title = gettext("Reload configuration file and performs an internal reset of the host software, but it also clears any error states from the micro-controller.");
             btnFirmwareRestart.classList.add("btn");
             btnFirmwareRestart.classList.add("span6");
             btnFirmwareRestart.addEventListener("click", self.btnFirmwareRestartClick);
 
             var btnFirmwareRestartIcon = document.createElement("i");
-            btnFirmwareRestartIcon.classList.add("fa", "fa-sync");
+            btnFirmwareRestartIcon.classList.add("fas", "fa-microchip");
+            btnFirmwareRestartIcon.style.marginRight = "5px";
             btnFirmwareRestart.appendChild(btnFirmwareRestartIcon);
 
             var btnFirmwareRestartText = document.createElement("span");
-            btnFirmwareRestartText.textContent = " Firmware Restart";
+            btnFirmwareRestartText.textContent = gettext("Firmware Restart");
             btnFirmwareRestart.appendChild(btnFirmwareRestartText);
 
             container.appendChild(btnFirmwareRestart);
 
             buttonContainer.after(container);
+
+            var container2 = document.createElement("div");
+            container2.classList.add("row-fluid", "print-control");
+            container2.style.marginTop = "10px";
+            container2.setAttribute("data-bind", "visible: $root.loginState.hasPermissionKo($root.access.permissions.PRINT)");
+
+            var btnKlipperRestart = document.createElement("button");
+            btnKlipperRestart.id = "job_klipper_restart";
+            btnKlipperRestart.title = gettext("Restart klipper process.");
+            btnKlipperRestart.classList.add("btn");
+            btnKlipperRestart.classList.add("span12");
+            btnKlipperRestart.addEventListener("click", self.btnKlipperRestartClick);
+
+            var btnKlipperRestartIcon = document.createElement("i");
+            btnKlipperRestartIcon.classList.add("fas", "fa-power-off");
+            btnKlipperRestartIcon.style.marginRight = "5px";
+            btnKlipperRestart.appendChild(btnKlipperRestartIcon);
+
+            var btnKlipperRestartText = document.createElement("span");
+            btnKlipperRestartText.textContent = gettext("Restart Klipper");
+            btnKlipperRestart.appendChild(btnKlipperRestartText);
+
+            container2.appendChild(btnKlipperRestart);
+
+            container.after(container2);
         };
 
         self.webcams = ko.observableArray([]);