From 7e3f2363936abc46719f272af5ebe50b1739b4a0 Mon Sep 17 00:00:00 2001
From: Arksine <arksine.code@gmail.com>
Date: Mon, 28 Sep 2020 14:23:10 -0400
Subject: [PATCH] data_store:  add "/server/gcode_store" endpoint

Clients may use this API to fetch up to 1000 lines of cached gocde responses.

Signed-off-by:  Eric Callahan <arksine.code@gmail.com>
---
 moonraker/plugins/data_store.py | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/moonraker/plugins/data_store.py b/moonraker/plugins/data_store.py
index 02c2fb7..8687c4e 100644
--- a/moonraker/plugins/data_store.py
+++ b/moonraker/plugins/data_store.py
@@ -10,12 +10,15 @@ from tornado.ioloop import IOLoop, PeriodicCallback
 TEMPERATURE_UPDATE_MS = 1000
 TEMPERATURE_STORE_SIZE = 20 * 60
 
+MAX_GCODE_LINES = 1000
+
 class DataStore:
     def __init__(self, config):
         self.server = config.get_server()
 
         # Temperature Store Tracking
         self.last_temps = {}
+        self.gcode_queue = deque(maxlen=MAX_GCODE_LINES)
         self.temperature_store = {}
         self.temp_update_cb = PeriodicCallback(
             self._update_temperature_store, TEMPERATURE_UPDATE_MS)
@@ -23,13 +26,18 @@ class DataStore:
         # Register status update event
         self.server.register_event_handler(
             "server:status_update", self._set_current_temps)
+        self.server.register_event_handler(
+            "server:gcode_response", self._update_gcode_store)
         self.server.register_event_handler(
             "server:klippy_ready", self._init_sensors)
 
-        # Register endpoint
+        # Register endpoints
         self.server.register_endpoint(
             "/server/temperature_store", ['GET'],
             self._handle_temp_store_request)
+        self.server.register_endpoint(
+            "/server/gcode_store", ['GET'],
+            self._handle_gcode_store_request)
 
     async def _init_sensors(self):
         klippy_apis = self.server.lookup_plugin('klippy_apis')
@@ -98,5 +106,23 @@ class DataStore:
     async def close(self):
         self.temp_update_cb.stop()
 
+    def _update_gcode_store(self, response):
+        lines = response.strip().split("\n")
+        self.gcode_queue.extend(lines)
+
+    async def _handle_gcode_store_request(self, path, method, args):
+        count = args.get("count", None)
+        if count is not None:
+            try:
+                count = int(count)
+            except Exception:
+                raise self.server.error(
+                    "Parameter <count> must be an integer value, "
+                    f"received: {count}")
+            res = "\n".join(list(self.gcode_queue)[-count:])
+        else:
+            res = "\n".join(list(self.gcode_queue))
+        return {'gcode_store': res}
+
 def load_plugin(config):
     return DataStore(config)