feat: update entity diagram

assets/components/tracking/diagram/ViewDiagramWatchlistButton.tsx

@@ -2,72 +2,26 @@ import {Button, Flex, Modal, Space, Typography} from "antd"
 import {t} from "ttag"
 import React, {useEffect, useState} from "react"
 import {ApartmentOutlined} from "@ant-design/icons"
-import vCard from "vcf";
 
 import '@xyflow/react/dist/style.css'
 import {Background, Controls, MiniMap, ReactFlow, useEdgesState, useNodesState} from "@xyflow/react";
-import {getWatchlist, Watchlist} from "../../../utils/api";
+import {getWatchlist} from "../../../utils/api";
-import {translateRoles} from "../../search/EntitiesList";
 import {getLayoutedElements} from "./getLayoutedElements";
+import {watchlistToNodes} from "./WatchlistToNodes";
+import {watchlistToEdges} from "./WatchlistToEdges";
 
-
+export type DiagramConfig = {
-function watchlistToNodes(watchlist: Watchlist) {
+    tld?: boolean
-    const domains = watchlist.domains.map(d => ({
+    nameserver?: boolean
-        id: d.ldhName,
+    entities?: boolean
-        data: {label: <b>{d.ldhName}</b>},
-        style: {
-            width: 200
-        }
-    }))
-    const entities = [...new Set(watchlist.domains
-        .map(d => d.entities
-            .filter(e => !e.roles.includes('registrar'))
-            .map(e => e.entity
-            )
-        ).flat())].map(e => {
-        const jCard = vCard.fromJSON(e.jCard)
-        let label = e.handle
-        if (jCard.data.fn !== undefined && !Array.isArray(jCard.data.fn)) label = jCard.data.fn.valueOf()
-
-        return {
-            id: e.handle,
-            data: {label},
-            style: {
-                width: 200
-            }
-        }
-    })
-
-    return [...domains, ...entities]
-}
-
-const rolesToColor = (roles: string[]) => roles.includes('registrant') ? 'green' :
-    roles.includes('administrative') ? 'blue' :
-        roles.includes('technical') ? 'orange' : 'violet'
-
-function watchlistToEdges(watchlist: Watchlist) {
-    const domainRole = translateRoles()
-
-    return watchlist.domains
-        .map(d => d.entities
-            .filter(e => !e.roles.includes('registrar'))
-            .map(e => ({
-                id: `${d.ldhName}-${e.entity.handle}`,
-                source: e.roles.includes('technical') ? d.ldhName : e.entity.handle,
-                target: e.roles.includes('technical') ? e.entity.handle : d.ldhName,
-                style: {stroke: rolesToColor(e.roles), strokeWidth: 3},
-                label: e.roles.map(r => Object.keys(domainRole).includes(r) ? domainRole[r as keyof typeof domainRole] : r).join(', '),
-                animated: e.roles.includes('registrant'),
-            }))
-        ).flat(2)
 }
 
 export function ViewDiagramWatchlistButton({token}: { token: string }) {
     const [open, setOpen] = useState(false)
 
     const [loading, setLoading] = useState(false)
-    const [nodes, setNodes, onNodesChange] = useNodesState([]);
+    const [nodes, setNodes, onNodesChange] = useNodesState([])
-    const [edges, setEdges, onEdgesChange] = useEdgesState([]);
+    const [edges, setEdges, onEdgesChange] = useEdgesState([])
 
     useEffect(() => {
         if (!open) return

assets/components/tracking/diagram/WatchlistToEdges.tsx

@@ -0,0 +1,36 @@
+import {Watchlist} from "../../../utils/api";
+import {translateRoles} from "../../search/EntitiesList";
+
+const rolesToColor = (roles: string[]) => roles.includes('registrant') ? 'green' :
+    roles.includes('administrative') ? 'blue' :
+        roles.includes('technical') ? 'orange' : 'violet'
+
+
+export function watchlistToEdges(watchlist: Watchlist) {
+    const domainRole = translateRoles()
+
+    const entitiesEdges = watchlist.domains
+        .map(d => d.entities
+            .filter(e => !e.roles.includes('registrar')) //
+            .map(e => ({
+                id: `e-${d.ldhName}-${e.entity.handle}`,
+                source: e.roles.includes('registrant') ? e.entity.handle : d.ldhName,
+                target: e.roles.includes('registrant') ? d.ldhName : e.entity.handle,
+                style: {stroke: rolesToColor(e.roles), strokeWidth: 3},
+                label: e.roles.map(r => Object.keys(domainRole).includes(r) ? domainRole[r as keyof typeof domainRole] : r).join(', '),
+                animated: e.roles.includes('registrant'),
+            }))
+        ).flat()
+
+    const nameserversEdges = watchlist.domains
+        .map(d => d.nameservers
+            .map(ns => ({
+                id: `ns-${d.ldhName}-${ns.ldhName}`,
+                source: d.ldhName,
+                target: ns.ldhName,
+                style: {stroke: 'grey', strokeWidth: 3},
+                label: 'DNS'
+            }))).flat()
+
+    return [...entitiesEdges, ...nameserversEdges]
+}

assets/components/tracking/diagram/WatchlistToNodes.tsx

@@ -0,0 +1,52 @@
+import {Watchlist} from "../../../utils/api";
+import vCard from "vcf";
+import React from "react";
+import {t} from 'ttag'
+
+export function watchlistToNodes(watchlist: Watchlist) {
+    const domains = watchlist.domains.map(d => ({
+        id: d.ldhName,
+        data: {label: <b>{d.ldhName}</b>},
+        style: {
+            width: 200
+        },
+        parentId: d.tld.tld,
+        extent: 'parent'
+    }))
+    const entities = [...new Set(watchlist.domains
+        .map(d => d.entities
+            .filter(e => !e.roles.includes('registrar')) //
+            .map(e => {
+                const jCard = vCard.fromJSON(e.entity.jCard)
+                let label = e.entity.handle
+                if (jCard.data.fn !== undefined && !Array.isArray(jCard.data.fn)) label = jCard.data.fn.valueOf()
+
+                return {
+                    id: e.entity.handle,
+                    data: {label},
+                    style: {
+                        width: 200
+                    },
+                    parentId: d.tld.tld,
+                    extent: 'parent'
+                }
+            })).flat())]
+
+    const tlds = [...new Set(watchlist.domains.map(d => d.tld))].map(tld => ({
+        id: tld.tld,
+        data: {label: t`.${tld.tld} Registry`},
+        style: {
+            width: 200
+        }
+    }))
+
+    const nameservers = [...new Set(watchlist.domains.map(d => d.nameservers))].flat().map(ns => ({
+        id: ns.ldhName,
+        data: {label: ns.ldhName},
+        style: {
+            width: 200
+        }
+    }))
+
+    return [...domains, ...entities, ...nameservers]
+}

src/Service/RDAPService.php

@@ -278,6 +278,8 @@ readonly class RDAPService
         }
 
         if (array_key_exists('nameservers', $res) && is_array($res['nameservers'])) {
+            $domain->getNameservers()->clear();
+
             foreach ($res['nameservers'] as $rdapNameserver) {
                 $nameserver = $this->nameserverRepository->findOneBy([
                     'ldhName' => strtolower($rdapNameserver['ldhName']),