fix large entity

static/js/graph.js

@@ -270,8 +270,23 @@ class GraphManager {
                 this.initialize();
             }
 
-            // Process nodes with enhanced attributes
-            const processedNodes = graphData.nodes.map(node => this.processNode(node));
+            // Find all aggregated node IDs first
+            const aggregatedNodeIds = new Set();
+            graphData.nodes.forEach(node => {
+                if (node.type === 'large_entity' && node.metadata && Array.isArray(node.metadata.nodes)) {
+                    node.metadata.nodes.forEach(nodeId => aggregatedNodeIds.add(nodeId));
+                }
+            });
+
+            // Process nodes, hiding the ones that are aggregated
+            const processedNodes = graphData.nodes.map(node => {
+                const processed = this.processNode(node);
+                if (aggregatedNodeIds.has(node.id)) {
+                    processed.hidden = true; // Mark node as hidden
+                }
+                return processed;
+            });
+            
             const processedEdges = graphData.edges.map(edge => this.processEdge(edge));
 
             // Update datasets with animation
@@ -441,7 +456,8 @@ class GraphManager {
             'domain': 12,
             'ip': 14,
             'asn': 16,
-            'correlation_object': 8
+            'correlation_object': 8,
+            'large_entity': 12
         };
         return sizes[nodeType] || 12;
     }
@@ -456,7 +472,8 @@ class GraphManager {
             'domain': 'dot',
             'ip': 'square',
             'asn': 'triangle',
-            'correlation_object': 'hexagon'
+            'correlation_object': 'hexagon',
+            'large_entity': 'database'
         };
         return shapes[nodeType] || 'dot';
     }

static/js/main.js

@@ -80,7 +80,6 @@ class DNSReconApp {
             // API Key Modal elements
             apiKeyModal: document.getElementById('api-key-modal'),
             apiKeyModalClose: document.getElementById('api-key-modal-close'),
-            virustotalApiKey: document.getElementById('virustotal-api-key'),
             shodanApiKey: document.getElementById('shodan-api-key'),
             saveApiKeys: document.getElementById('save-api-keys'),
             resetApiKeys: document.getElementById('reset-api-keys'),
@@ -851,13 +850,11 @@ class DNSReconApp {
                 detailsHtml += createDetailRow('Related Domains (SAN)', metadata.related_domains_san);
                 detailsHtml += createDetailRow('Passive DNS', metadata.passive_dns);
                 detailsHtml += createDetailRow('Shodan Data', metadata.shodan);
-                detailsHtml += createDetailRow('VirusTotal Data', metadata.virustotal);
                 break;
             case 'ip':
                 detailsHtml += createDetailRow('Hostnames', metadata.hostnames);
                 detailsHtml += createDetailRow('Passive DNS', metadata.passive_dns);
                 detailsHtml += createDetailRow('Shodan Data', metadata.shodan);
-                detailsHtml += createDetailRow('VirusTotal Data', metadata.virustotal);
                 break;
             case 'correlation_object':
                 detailsHtml += createDetailRow('Correlated Value', metadata.value);
@@ -974,11 +971,9 @@ class DNSReconApp {
      */
     async saveApiKeys() {
         const shodanKey = this.elements.shodanApiKey.value.trim();
-        const virustotalKey = this.elements.virustotalApiKey.value.trim();
 
         const keys = {};
         if (shodanKey) keys.shodan = shodanKey;
-        if (virustotalKey) keys.virustotal = virustotalKey;
 
         if (Object.keys(keys).length === 0) {
             this.showWarning('No API keys were entered.');
@@ -1004,7 +999,6 @@ class DNSReconApp {
      */
     resetApiKeys() {
         this.elements.shodanApiKey.value = '';
-        this.elements.virustotalApiKey.value = '';
     }
     
     /**