Dinu Blanovschi
2024-05-29 0b9f79e1b779d727588cb9aa475a1cad1c8f237d
quartz/components/scripts/graph.inline.ts
@@ -1,4 +1,4 @@
import type { ContentDetails } from "../../plugins/emitters/contentIndex"
import type { ContentDetails, ContentIndex } from "../../plugins/emitters/contentIndex"
import * as d3 from "d3"
import { registerEscapeHandler, removeAllChildren } from "./util"
import { FullSlug, SimpleSlug, getFullSlug, resolveRelative, simplifySlug } from "../../util/path"
@@ -42,17 +42,39 @@
    linkDistance,
    fontSize,
    opacityScale,
    removeTags,
    showTags,
    focusOnHover,
  } = JSON.parse(graph.dataset["cfg"]!)
  const data = await fetchData
  const data: Map<SimpleSlug, ContentDetails> = new Map(
    Object.entries<ContentDetails>(await fetchData).map(([k, v]) => [
      simplifySlug(k as FullSlug),
      v,
    ]),
  )
  const links: LinkData[] = []
  for (const [src, details] of Object.entries<ContentDetails>(data)) {
    const source = simplifySlug(src as FullSlug)
  const tags: SimpleSlug[] = []
  const validLinks = new Set(data.keys())
  for (const [source, details] of data.entries()) {
    const outgoing = details.links ?? []
    for (const dest of outgoing) {
      if (dest in data) {
        links.push({ source, target: dest })
      if (validLinks.has(dest)) {
        links.push({ source: source, target: dest })
      }
    }
    if (showTags) {
      const localTags = details.tags
        .filter((tag) => !removeTags.includes(tag))
        .map((tag) => simplifySlug(("tags/" + tag) as FullSlug))
      tags.push(...localTags.filter((tag) => !tags.includes(tag)))
      for (const tag of localTags) {
        links.push({ source: source, target: tag })
      }
    }
  }
@@ -74,15 +96,19 @@
      }
    }
  } else {
    Object.keys(data).forEach((id) => neighbourhood.add(simplifySlug(id as FullSlug)))
    validLinks.forEach((id) => neighbourhood.add(id))
    if (showTags) tags.forEach((tag) => neighbourhood.add(tag))
  }
  const graphData: { nodes: NodeData[]; links: LinkData[] } = {
    nodes: [...neighbourhood].map((url) => ({
      id: url,
      text: data[url]?.title ?? url,
      tags: data[url]?.tags ?? [],
    })),
    nodes: [...neighbourhood].map((url) => {
      const text = url.startsWith("tags/") ? "#" + url.substring(5) : data.get(url)?.title ?? url
      return {
        id: url,
        text: text,
        tags: data.get(url)?.tags ?? [],
      }
    }),
    links: links.filter((l) => neighbourhood.has(l.source) && neighbourhood.has(l.target)),
  }
@@ -126,7 +152,7 @@
    const isCurrent = d.id === slug
    if (isCurrent) {
      return "var(--secondary)"
    } else if (visited.has(d.id)) {
    } else if (visited.has(d.id) || d.id.startsWith("tags/")) {
      return "var(--tertiary)"
    } else {
      return "var(--gray)"
@@ -164,6 +190,8 @@
    return 2 + Math.sqrt(numLinks)
  }
  let connectedNodes: SimpleSlug[] = []
  // draw individual nodes
  const node = graphNode
    .append("circle")
@@ -177,17 +205,37 @@
      window.spaNavigate(new URL(targ, window.location.toString()))
    })
    .on("mouseover", function (_, d) {
      const neighbours: SimpleSlug[] = data[fullSlug].links ?? []
      const neighbourNodes = d3
        .selectAll<HTMLElement, NodeData>(".node")
        .filter((d) => neighbours.includes(d.id))
      const currentId = d.id
      const linkNodes = d3
        .selectAll(".link")
        .filter((d: any) => d.source.id === currentId || d.target.id === currentId)
      // highlight neighbour nodes
      neighbourNodes.transition().duration(200).attr("fill", color)
      if (focusOnHover) {
        // fade out non-neighbour nodes
        connectedNodes = linkNodes.data().flatMap((d: any) => [d.source.id, d.target.id])
        d3.selectAll<HTMLElement, NodeData>(".link")
          .transition()
          .duration(200)
          .style("opacity", 0.2)
        d3.selectAll<HTMLElement, NodeData>(".node")
          .filter((d) => !connectedNodes.includes(d.id))
          .transition()
          .duration(200)
          .style("opacity", 0.2)
        d3.selectAll<HTMLElement, NodeData>(".node")
          .filter((d) => !connectedNodes.includes(d.id))
          .nodes()
          .map((it) => d3.select(it.parentNode as HTMLElement).select("text"))
          .forEach((it) => {
            let opacity = parseFloat(it.style("opacity"))
            it.transition()
              .duration(200)
              .attr("opacityOld", opacity)
              .style("opacity", Math.min(opacity, 0.2))
          })
      }
      // highlight links
      linkNodes.transition().duration(200).attr("stroke", "var(--gray)").attr("stroke-width", 1)
@@ -206,6 +254,16 @@
        .style("font-size", bigFont + "em")
    })
    .on("mouseleave", function (_, d) {
      if (focusOnHover) {
        d3.selectAll<HTMLElement, NodeData>(".link").transition().duration(200).style("opacity", 1)
        d3.selectAll<HTMLElement, NodeData>(".node").transition().duration(200).style("opacity", 1)
        d3.selectAll<HTMLElement, NodeData>(".node")
          .filter((d) => !connectedNodes.includes(d.id))
          .nodes()
          .map((it) => d3.select(it.parentNode as HTMLElement).select("text"))
          .forEach((it) => it.transition().duration(200).style("opacity", it.attr("opacityOld")))
      }
      const currentId = d.id
      const linkNodes = d3
        .selectAll(".link")
@@ -230,11 +288,7 @@
    .attr("dx", 0)
    .attr("dy", (d) => -nodeRadius(d) + "px")
    .attr("text-anchor", "middle")
    .text(
      (d) =>
        data[d.id]?.title ||
        (d.id.charAt(0).toUpperCase() + d.id.slice(1, d.id.length - 1)).replace("-", " "),
    )
    .text((d) => d.text)
    .style("opacity", (opacityScale - 1) / 3.75)
    .style("pointer-events", "none")
    .style("font-size", fontSize + "em")
@@ -298,12 +352,12 @@
  registerEscapeHandler(container, hideGlobalGraph)
}
document.addEventListener("nav", async (e: unknown) => {
  const slug = (e as CustomEventMap["nav"]).detail.url
document.addEventListener("nav", async (e: CustomEventMap["nav"]) => {
  const slug = e.detail.url
  addToVisited(slug)
  await renderGraph("graph-container", slug)
  const containerIcon = document.getElementById("global-graph-icon")
  containerIcon?.removeEventListener("click", renderGlobalGraph)
  containerIcon?.addEventListener("click", renderGlobalGraph)
  window.addCleanup(() => containerIcon?.removeEventListener("click", renderGlobalGraph))
})