ardupilot_manager: allow updating master endpoint

Author: Williangalvani

core/services/ardupilot_manager/autopilot_manager.py

@@ -674,7 +674,7 @@ async def restore_default_firmware(self, board: FlightController) -> None:
     async def set_manual_board_master_endpoint(self, endpoint: Endpoint) -> bool:
         self.configuration["manual_board_master_endpoint"] = endpoint.as_dict()
         self.settings.save(self.configuration)
-        self._save_current_endpoints()
+        self.mavlink_manager.master_endpoint = endpoint
         await self.mavlink_manager.restart()
         return True
 

core/services/ardupilot_manager/mavlink_proxy/AbstractRouter.py

@@ -94,6 +94,10 @@ def version(self) -> Optional[str]:
     def master_endpoint(self) -> Optional[Endpoint]:
         return self._master_endpoint
 
+    @master_endpoint.setter
+    def master_endpoint(self, master_endpoint: Endpoint) -> None:
+        self._master_endpoint = master_endpoint
+
     async def start(self, master_endpoint: Endpoint) -> None:
         self._master_endpoint = master_endpoint
         command = self.assemble_command(self._master_endpoint)

core/services/ardupilot_manager/mavlink_proxy/Manager.py

@@ -114,6 +114,10 @@ def _reset_endpoints(self) -> None:
     def master_endpoint(self) -> Optional[Endpoint]:
         return self.tool.master_endpoint
 
+    @master_endpoint.setter
+    def master_endpoint(self, master_endpoint: Endpoint) -> None:
+        self.tool.master_endpoint = master_endpoint
+
     async def start(self, master_endpoint: Endpoint) -> None:
         # If the tool is already running, don't start it again to avoid port conflicts
         if self.should_be_running and await self.is_running():
@@ -151,8 +155,10 @@ async def stop(self) -> None:
         self.should_be_running = False
         await self.tool.exit()
 
-    async def restart(self) -> None:
+    async def restart(self, master_endpoint: Optional[Endpoint] = None) -> None:
         self.should_be_running = False
+        if master_endpoint:
+            self.tool.master_endpoint = master_endpoint
         await self.tool.restart()
         self._last_valid_endpoints = self.endpoints()
         self.should_be_running = True