Update the hostname in /boot/firmware/user-data as well as /etc/hostname to ensure it persists properly across reboots

Author: civerachb-cpr

turtlebot4_setup/conf.py

@@ -130,6 +130,7 @@ def __init__(self) -> None:
         self.netplan_wifis_file = os.path.join(self.netplan_dir, '50-wifis.yaml')
         self.discovery_sh_file = os.path.join(self.setup_dir, 'discovery.sh')
         self.hostname_file = '/etc/hostname'
+        self.fw_user_data_file = '/boot/firmware/user-data'
 
         self.system_conf = copy.deepcopy(self.default_system_conf)
         self.wifi_conf = copy.deepcopy(self.default_wifi_conf)
@@ -230,6 +231,11 @@ def write_system(self):
             f.write(self.get(SystemOptions.HOSTNAME))
         subprocess.run(shlex.split('sudo mv /tmp' + self.hostname_file + ' ' + self.hostname_file))
 
+        # update /boot/firmware/user-data with the new hostname
+        subprocess.run(shlex.split(f'cp {self.fw_user_data_file} /tmp/user-data'))
+        subprocess.run(shlex.split(f'sed -i -E "s/^hostname:.+/hostname: {self.get(SystemOptions.HOSTNAME)}/" /tmp/user-data'))
+        subprocess.run(shlex.split(f'sudo mv /tmp/user-data {self.fw_user_data_file}'))
+
     def read_wifi(self):
         try:
             # Try to open the existing wifi configuration, but if it doesn't exist we can carry on