create security node and adjust network modification node creation rules (#802)

* create security node and adjust network modification node creation rules
 
---------
Signed-off-by: SOUISSI Maissa (Externe) <[email protected]>

Author: souissimai

src/main/java/org/gridsuite/study/server/service/NetworkModificationTreeService.java

@@ -603,6 +603,42 @@ public void assertIsRootOrConstructionNode(UUID nodeUuid) {
         }
     }
 
+    private NetworkModificationNodeType getReferenceNodeType(NodeEntity referenceNode) {
+        return referenceNode.getType().equals(NodeType.ROOT)
+                ? null
+                : getNetworkModificationNodeInfoEntity(referenceNode.getIdNode()).getNodeType();
+    }
+
+    private boolean isConstructionUnderSecurityNode(NetworkModificationNodeType newNodeType, NetworkModificationNodeType referenceNodeType) {
+        return newNodeType == NetworkModificationNodeType.CONSTRUCTION &&
+                referenceNodeType == NetworkModificationNodeType.SECURITY;
+    }
+
+    private boolean isInvalidSecurityNodeInsertion(NetworkModificationNodeType newNodeType, InsertMode insertMode, NetworkModificationNodeType referenceNodeType) {
+        return newNodeType == NetworkModificationNodeType.SECURITY &&
+                insertMode != InsertMode.CHILD &&
+                referenceNodeType != NetworkModificationNodeType.SECURITY;
+    }
+
+    private void assertIsNetworkModificationInsertionAllowed(
+            NodeEntity nodeEntity,
+            NetworkModificationNodeType newNodeType,
+            InsertMode insertMode
+    ) {
+        NetworkModificationNodeType referenceNodeType = getReferenceNodeType(nodeEntity);
+
+        if (isConstructionUnderSecurityNode(newNodeType, referenceNodeType) ||
+                isInvalidSecurityNodeInsertion(newNodeType, insertMode, referenceNodeType)) {
+            throw new StudyException(NOT_ALLOWED);
+        }
+    }
+
+    public void assertIsNetworkModificationNodeCreationAllowed(UUID nodeId, NetworkModificationNode nodeInfo, InsertMode insertMode) {
+        NetworkModificationNodeType newNodeType = nodeInfo.getNodeType();
+        NodeEntity nodeEntity = getNodeEntity(nodeId);
+        assertIsNetworkModificationInsertionAllowed(nodeEntity, newNodeType, insertMode);
+    }
+
     public boolean isConstructionNode(UUID nodeUuid) {
         return getNetworkModificationNodeInfoEntity(nodeUuid).getNodeType() == NetworkModificationNodeType.CONSTRUCTION;
     }

src/main/java/org/gridsuite/study/server/service/StudyService.java

@@ -3253,7 +3253,7 @@ public void createOrUpdateStateEstimationParameters(StudyEntity studyEntity, Str
     @Transactional
     public NetworkModificationNode createNode(UUID studyUuid, UUID nodeId, NetworkModificationNode nodeInfo, InsertMode insertMode, String userId) {
         StudyEntity study = studyRepository.findById(studyUuid).orElseThrow(() -> new StudyException(STUDY_NOT_FOUND));
-        networkModificationTreeService.assertIsRootOrConstructionNode(nodeId);
+        networkModificationTreeService.assertIsNetworkModificationNodeCreationAllowed(nodeId, nodeInfo, insertMode);
         NetworkModificationNode newNode = networkModificationTreeService.createNode(study, nodeId, nodeInfo, insertMode, userId);
 
         UUID parentUuid = networkModificationTreeService.getParentNodeUuid(newNode.getId()).orElse(null);