'#2011: render group nodes in graph using a multiple people icon

sepinf-inc · Jul 26, 2024 · d33361c · d33361c
1 parent da55663
commit d33361c
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/iped-engine/src/main/java/iped/engine/graph/GraphTask.java b/iped-engine/src/main/java/iped/engine/graph/GraphTask.java
@@ -354,6 +354,10 @@ private NodeValues getPhoneNodeValues(String value) {
         return null;
     }
 
+    private NodeValues getGroupNodeValues(String value) {
+        return new NodeValues(DynLabel.label(GraphConfiguration.CONTACT_GROUP_LABEL), BasicProps.NAME, value.trim().toLowerCase());
+    }
+
     private NodeValues getGenericNodeValues(String value) {
         return new NodeValues(DynLabel.label("GENERIC"), "entity", value.trim().toLowerCase());
     }
@@ -469,7 +473,12 @@ private void processCommunicationMetadata(IItem evidence) throws IOException {
         recipients.addAll(Arrays.asList(metadata.getValues(Message.MESSAGE_BCC)));
 
         for (String recipient : recipients) {
-            NodeValues nv2 = getNodeValues(recipient, evidence.getMetadata(), detectPhones);
+            NodeValues nv2;
+            if (Boolean.valueOf(metadata.get(ExtraProperties.IS_GROUP_MESSAGE))) {
+                nv2 = getGroupNodeValues(recipient);
+            } else {
+                nv2 = getNodeValues(recipient, metadata, detectPhones);
+            }
             graphFileWriter.writeNode(nv2.label, nv2.propertyName, nv2.propertyValue, nv2.props);
             graphFileWriter.writeRelationship(nv1.label, nv1.propertyName, nv1.propertyValue, nv2.label,
                     nv2.propertyName, nv2.propertyValue, relationshipType, relProps);