1 files changed, 104 insertions, 0 deletions
diff --git a/components/Graph/drawLabels.ts b/components/Graph/drawLabels.ts
new file mode 100644
index 0000000..d2d84f8
--- /dev/null
+++ b/components/Graph/drawLabels.ts
@@ -0,0 +1,104 @@
+import { OrgRoamNode } from '../../api'
+import { NodeObject } from 'force-graph'
+import { initialVisuals } from '../config'
+import { hexToRGBA } from '../../pages'
+import wrap from 'word-wrap'
+
+export interface drawLabelsProps {
+  labelBackgroundColor: string
+  labelTextColor: string
+  node: NodeObject
+  ctx: any
+  globalScale: number
+  highlightedNodes: { [id: string]: {} }
+  previouslyHighlightedNodes: { [id: string]: {} }
+  visuals: typeof initialVisuals
+  opacity: number
+  nodeSize: (node: NodeObject) => number
+}
+export function drawLabels(props: drawLabelsProps) {
+  const {
+    labelBackgroundColor,
+    labelTextColor,
+    node,
+    ctx,
+    globalScale,
+    highlightedNodes,
+    previouslyHighlightedNodes,
+    visuals,
+    opacity,
+    nodeSize,
+  } = props
+
+  if (!node) {
+    return
+  }
+  //if (dragging) {
+  //   return
+  //}
+
+  if (!visuals.labels) {
+    return
+  }
+  const wasHighlightedNode = previouslyHighlightedNodes[node.id!]
+
+  if (
+    (globalScale <= visuals.labelScale || visuals.labels === 1) &&
+    !highlightedNodes[node.id!] &&
+    !wasHighlightedNode
+  ) {
+    return
+  }
+
+  const nodeTitle = (node as OrgRoamNode).title!
+  const label = nodeTitle.substring(0, visuals.labelLength)
+  const fontSize = visuals.labelFontSize / (0.75 * Math.min(Math.max(0.5, globalScale), 3))
+  const textWidth = ctx.measureText(label).width
+  const bckgDimensions = [textWidth * 1.1, fontSize].map((n) => n + fontSize * 0.5) as [
+    number,
+    number,
+  ] // some padding
+
+  const fadeFactor = Math.min((3 * (globalScale - visuals.labelScale)) / visuals.labelScale, 1)
+
+  // draw label background
+  const getLabelOpacity = () => {
+    if (visuals.labels === 1) {
+      return opacity
+    }
+    if (globalScale <= visuals.labelScale) {
+      return opacity
+    }
+    return highlightedNodes[node.id!] || previouslyHighlightedNodes[node.id!]
+      ? Math.max(fadeFactor, opacity)
+      : 1 * fadeFactor * (-1 * (visuals.highlightFade * opacity - 1))
+  }
+  const nodeS = 8 * Math.cbrt(nodeSize(node) * visuals.nodeRel)
+  if (visuals.labelBackgroundColor && visuals.labelBackgroundOpacity) {
+    const backgroundOpacity = getLabelOpacity() * visuals.labelBackgroundOpacity
+    const labelBackground = hexToRGBA(labelBackgroundColor, backgroundOpacity)
+    ctx.fillStyle = labelBackground
+    ctx.fillRect(
+      node.x! - bckgDimensions[0] / 2,
+      node.y! - bckgDimensions[1] / 2 + nodeS,
+      ...bckgDimensions,
+    )
+  }
+
+  // draw label text
+  const textOpacity = getLabelOpacity()
+  ctx.textAlign = 'center'
+  ctx.textBaseline = 'middle'
+  const labelText = hexToRGBA(labelTextColor, textOpacity)
+  ctx.fillStyle = labelText
+  ctx.font = `${fontSize}px Sans-Serif`
+  const wordsArray = wrap(label, { width: visuals.labelWordWrap }).split('\n')
+
+  const truncatedWords =
+    nodeTitle.length > visuals.labelLength
+      ? [...wordsArray.slice(0, -1), `${wordsArray.slice(-1)}...`]
+      : wordsArray
+  truncatedWords.forEach((word, index) => {
+    ctx.fillText(word, node.x!, node.y! + nodeS + visuals.labelLineSpace * fontSize * index)
+  })
+}