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 {translateRoles} from "../../search/EntitiesList";
+import {getWatchlist} from "../../../utils/api";
 import {getLayoutedElements} from "./getLayoutedElements";
+import {watchlistToNodes} from "./WatchlistToNodes";
+import {watchlistToEdges} from "./WatchlistToEdges";
 
-
-function watchlistToNodes(watchlist: Watchlist) {
-    const domains = watchlist.domains.map(d => ({
-        id: d.ldhName,
-        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 type DiagramConfig = {
+    tld?: boolean
+    nameserver?: boolean
+    entities?: boolean
 }
 
 export function ViewDiagramWatchlistButton({token}: { token: string }) {
     const [open, setOpen] = useState(false)
 
     const [loading, setLoading] = useState(false)
-    const [nodes, setNodes, onNodesChange] = useNodesState([]);
-    const [edges, setEdges, onEdgesChange] = useEdgesState([]);
+    const [nodes, setNodes, onNodesChange] = useNodesState([])
+    const [edges, setEdges, onEdgesChange] = useEdgesState([])
 
     useEffect(() => {
         if (!open) return