Add support for conditional credential preservation

core/models/session/SessionProfile.py

@@ -6,6 +6,7 @@ from core.models.session.ProxyConfiguration import ProxyConfiguration
 from core.models.session.SessionConnection import SessionConnection
 from dataclasses import dataclass
 from json import JSONDecodeError
+from pathlib import Path
 from typing import Optional
 import json
 import os
@@ -31,6 +32,11 @@ class SessionProfile(BaseProfile):
             if os.path.isdir(persistent_state_path):
                 shutil.rmtree(persistent_state_path, ignore_errors=True)
 
+        if 'location' in self._get_dirty_keys():
+
+            self.__delete_proxy_configuration()
+            self.__delete_wireguard_configuration()
+
         super().save()
 
     def attach_proxy_configuration(self, proxy_configuration):
@@ -122,3 +128,9 @@ class SessionProfile(BaseProfile):
 
             if match:
                 return re.sub(r'[^a-zA-Z0-9+\-_ /]', '', match.group(1).strip())
+
+    def __delete_proxy_configuration(self):
+        Path(self.get_proxy_configuration_path()).unlink(missing_ok=True)
+
+    def __delete_wireguard_configuration(self):
+        Path(self.get_wireguard_configuration_path()).unlink(missing_ok=True)

core/models/system/SystemProfile.py

@@ -16,6 +16,13 @@ class SystemProfile(BaseProfile):
     def get_system_config_path(self):
         return self.__get_system_config_path(self.id)
 
+    def save(self):
+
+        if 'location' in self._get_dirty_keys():
+            self.__delete_wireguard_configuration()
+
+        super().save()
+
     def attach_wireguard_configuration(self, wireguard_configuration):
 
         if shutil.which('pkexec') is None:
@@ -40,7 +47,7 @@ class SystemProfile(BaseProfile):
                 failed_attempt_count += 1
 
         if not wireguard_configuration_is_attached:
-            raise ProfileModificationError('WireGuard configuration could not be attached.')
+            raise ProfileModificationError('The WireGuard configuration could not be attached.')
 
     def get_wireguard_configuration_path(self):
         return f'{self.get_system_config_path()}/wg.conf'
@@ -50,12 +57,15 @@ class SystemProfile(BaseProfile):
 
     def delete(self):
 
-        if self.has_wireguard_configuration():
+        try:
+            self.__delete_wireguard_configuration()
+        except ProfileModificationError:
+            raise ProfileDeletionError('The WireGuard configuration could not be deleted.')
 
         if shutil.which('pkexec') is None:
             raise CommandNotFoundError('pkexec')
 
-            process = subprocess.Popen(('pkexec', 'rm', '-d', self.get_wireguard_configuration_path(), self.get_system_config_path()))
+        process = subprocess.Popen(('pkexec', 'rm', '-d', self.get_system_config_path()))
         completed_successfully = not bool(os.waitpid(process.pid, 0)[1] >> 8)
 
         if not completed_successfully:
@@ -63,6 +73,19 @@ class SystemProfile(BaseProfile):
 
         super().delete()
 
+    def __delete_wireguard_configuration(self):
+
+        if self.has_wireguard_configuration():
+
+            if shutil.which('pkexec') is None:
+                raise CommandNotFoundError('pkexec')
+
+            process = subprocess.Popen(('pkexec', 'rm', '-d', self.get_wireguard_configuration_path()))
+            completed_successfully = not bool(os.waitpid(process.pid, 0)[1] >> 8)
+
+            if not completed_successfully:
+                raise ProfileModificationError('The WireGuard configuration could not be deleted.')
+
     @staticmethod
     def __get_system_config_path(id: int):
         return f'{Constants.HV_SYSTEM_PROFILE_CONFIG_PATH}/{str(id)}'