SHIBUI-2270

Added cleanup of settings to the danger controller
TIER · Sep 6, 2022 · 4716829 · 4716829
1 parent cbf8644
commit 4716829
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/...end/src/main/java/edu/internet2/tier/shibboleth/admin/ui/controller/DangerController.java b/...end/src/main/java/edu/internet2/tier/shibboleth/admin/ui/controller/DangerController.java
@@ -7,6 +7,8 @@
 import edu.internet2.tier.shibboleth.admin.ui.repository.FilterRepository;
 import edu.internet2.tier.shibboleth.admin.ui.repository.MetadataResolverRepository;
 import edu.internet2.tier.shibboleth.admin.ui.repository.MetadataResolversPositionOrderContainerRepository;
+import edu.internet2.tier.shibboleth.admin.ui.repository.ShibPropertySetRepository;
+import edu.internet2.tier.shibboleth.admin.ui.repository.ShibPropertySettingRepository;
 import edu.internet2.tier.shibboleth.admin.ui.security.repository.GroupsRepository;
 import edu.internet2.tier.shibboleth.admin.ui.security.repository.OwnershipRepository;
 import edu.internet2.tier.shibboleth.admin.ui.security.repository.UserRepository;
@@ -59,6 +61,12 @@ public class DangerController {
     @Autowired
     private OwnershipRepository ownershipRepository;
 
+    @Autowired
+    private ShibPropertySetRepository shibPropertySetRepository;
+
+    @Autowired
+    private ShibPropertySettingRepository shibPropertySettingRepository;
+
     @Autowired
     UserRepository userRepository;
 
@@ -84,9 +92,18 @@ public ResponseEntity<?> wipeOut() {
 
         clearUsersAndGroups();
 
+        clearShibSettings();
+
         return ResponseEntity.ok("yes, you did it");
     }
 
+    private void clearShibSettings() {
+        shibPropertySetRepository.findAll().forEach(shibPropSet -> {
+            shibPropertySettingRepository.deleteAll(shibPropSet.getProperties());
+            shibPropertySetRepository.delete(shibPropSet);
+        });
+    }
+
     private void clearUsersAndGroups() {
         groupRepository.deleteAll();
         ownershipRepository.clearAllOwnedByGroup();
@@ -99,4 +116,4 @@ private void clearUsersAndGroups() {
         groupService.ensureAdminGroupExists();
         devConfig.createDevUsersAndGroups();
     }
-}
+}