From 72962c5ca28789789509002fd7535da304b2f56c Mon Sep 17 00:00:00 2001
From: Arksine <arksine.code@gmail.com>
Date: Fri, 19 Jun 2020 06:25:02 -0400
Subject: [PATCH] gcode: add webhooks support

Register "gcode/script", "gcode/help", "gcode/restart", and "gcode/firmware_restart" endpoints.

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
---
 klippy/gcode.py | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/klippy/gcode.py b/klippy/gcode.py
index 1c3b2bcb..c9f8a81b 100644
--- a/klippy/gcode.py
+++ b/klippy/gcode.py
@@ -91,6 +91,16 @@ class GCodeParser:
         self.pending_commands = []
         self.bytes_read = 0
         self.input_log = collections.deque([], 50)
+        # Register webhooks
+        webhooks = self.printer.lookup_object('webhooks')
+        webhooks.register_endpoint(
+            "gcode/help", self._handle_remote_help)
+        webhooks.register_endpoint(
+            "gcode/script", self._handle_remote_script)
+        webhooks.register_endpoint(
+            "gcode/restart", self._handle_remote_restart)
+        webhooks.register_endpoint(
+            "gcode/firmware_restart", self._handle_remote_restart)
         # Command handling
         self.is_printer_ready = False
         self.mutex = self.reactor.mutex()
@@ -349,6 +359,23 @@ class GCodeParser:
         if self.fd_handle is None:
             self.fd_handle = self.reactor.register_fd(self.fd,
                                                       self._process_data)
+    def _handle_remote_help(self, web_request):
+        if web_request.get_method() != 'GET':
+            raise web_request.error("Invalid Request Method")
+        web_request.send(dict(self.gcode_help))
+    def _handle_remote_restart(self, web_request):
+        if web_request.get_method() != 'POST':
+            raise web_request.error("Invalid Request Method")
+        path = web_request.get_path()
+        if path == "gcode/restart":
+            self.run_script('restart')
+        elif path == "gcode/firmware_restart":
+            self.run_script('firmware_restart')
+    def _handle_remote_script(self, web_request):
+        if web_request.get_method() != 'POST':
+            raise web_request.error("Invalid Request Method")
+        script = web_request.get('script')
+        self.run_script(script)
     def run_script_from_command(self, script):
         self._process_commands(script.split('\n'), need_ack=False)
     def run_script(self, script):