Add device states to node information on boot

SoonerRobotics · Mar 12, 2024 · a62432f · a62432f
1 parent bae35e6
commit a62432f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/autonav_ws/src/autonav_display/src/display.py b/autonav_ws/src/autonav_display/src/display.py
@@ -155,9 +155,13 @@ async def consumer(self, websocket):
 				nodes = self.get_node_names()
 				for i in range(len(nodes)):
 					nodes[i] = nodes[i].replace("/", "")
+				node_states = {}
+				for identifier in nodes:
+					node_states[identifier] = self.device_states[identifier] if identifier in self.device_states else 0
 				self.pushSendQueue(json.dumps({
 					"op": "get_nodes_callback",
-					"nodes": nodes
+					"nodes": nodes,
+					"states": node_states
 				}), unique_id)
 
 			if obj["op"] == "set_system_state":

diff --git a/display/scripts/main.js b/display/scripts/main.js
@@ -70,6 +70,17 @@ $(document).ready(function () {
                             opcode: 4,
                             iterator: iterate()
                         });
+
+                        const statemap = obj.states;
+                        if (node in statemap) {
+                            deviceStates[node] = statemap[node];
+                            unorderedListElement = $("#element_device_states");
+                            unorderedListElement.empty();
+                            for (const id in deviceStates) {
+                                const state = deviceStates[id];
+                                unorderedListElement.append(`<h5>${id}: <span data-state=\"${state}\">${deviceStateToName(state)}</span></h5>`);
+                            }
+                        }
                     }
                 }
             }