Add disconnect handler to messaging blueprint

Author: angusjfw

pitop/labs/web/blueprints/messaging/__init__.py

@@ -64,6 +64,10 @@ def send(response_message):
                 if message:
                     handle_message(message, send)
 
+            disconnect_handler = message_handlers.get("disconnect")
+            if disconnect_handler:
+                disconnect_handler()
+
             del self.sockets[id]
 
     def register(self, app, options, *args):

pitop/labs/web/blueprints/rover/__init__.py

@@ -49,6 +49,13 @@ def right_joystick(data):
 
             message_handlers["right_joystick"] = right_joystick
 
+        if message_handlers.get("disconnect") is None:
+
+            def disconnect():
+                drive.robot_move(0, 0)
+
+            message_handlers["disconnect"] = disconnect
+
         self.controller_blueprint = ControllerBlueprint(
             get_frame=get_frame, message_handlers=message_handlers
         )