1 files changed, 23 insertions, 0 deletions
diff --git a/util/getLinkNodeColor.ts b/util/getLinkNodeColor.ts
new file mode 100644
index 0000000..86cfa4b
--- /dev/null
+++ b/util/getLinkNodeColor.ts
@@ -0,0 +1,23 @@
+import { initialColoring, initialVisuals } from '../components/config'
+import { LinksByNodeId } from '../pages'
+import { getNodeColorById } from './getNodeColorById'
+
+export const getLinkNodeColor = ({
+  sourceId,
+  targetId,
+  linksByNodeId,
+  visuals,
+  coloring,
+  cluster,
+}: {
+  sourceId: string
+  targetId: string
+  linksByNodeId: LinksByNodeId
+  visuals: typeof initialVisuals
+  coloring: typeof initialColoring
+  cluster: any
+}) => {
+  return linksByNodeId[sourceId]!.length > linksByNodeId[targetId]!.length
+    ? getNodeColorById({ id: sourceId, linksByNodeId, visuals, cluster, coloring })
+    : getNodeColorById({ id: targetId, visuals, linksByNodeId, cluster, coloring })
+}