From 47168290b9da41b946f1b2457e385f904993544d Mon Sep 17 00:00:00 2001
From: chasegawa <chasegawa@unicon.net>
Date: Tue, 6 Sep 2022 13:32:44 -0700
Subject: [PATCH] SHIBUI-2270

Added cleanup of settings to the danger controller
---
 .../admin/ui/controller/DangerController.java | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/backend/src/main/java/edu/internet2/tier/shibboleth/admin/ui/controller/DangerController.java b/backend/src/main/java/edu/internet2/tier/shibboleth/admin/ui/controller/DangerController.java
index 2ab357a03..7af217eb2 100644
--- a/backend/src/main/java/edu/internet2/tier/shibboleth/admin/ui/controller/DangerController.java
+++ b/backend/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();
     }
-}
+}
\ No newline at end of file