import type { ContentDetails } from "../../plugins/emitters/contentIndex" import * as d3 from "d3" import { registerEscapeHandler, removeAllChildren } from "./util" import { FullSlug, SimpleSlug, getFullSlug, resolveRelative, simplifySlug } from "../../util/path" type NodeData = { id: SimpleSlug text: string tags: string[] } & d3.SimulationNodeDatum type LinkData = { source: SimpleSlug target: SimpleSlug } const localStorageKey = "graph-visited" function getVisited(): Set { return new Set(JSON.parse(localStorage.getItem(localStorageKey) ?? "[]")) } function addToVisited(slug: SimpleSlug) { const visited = getVisited() visited.add(slug) localStorage.setItem(localStorageKey, JSON.stringify([...visited])) } async function renderGraph(container: string, fullSlug: FullSlug) { const slug = simplifySlug(fullSlug) const visited = getVisited() const graph = document.getElementById(container) if (!graph) return removeAllChildren(graph) let { drag: enableDrag, zoom: enableZoom, depth, scale, repelForce, centerForce, linkDistance, fontSize, opacityScale, removeTags, showTags, } = JSON.parse(graph.dataset["cfg"]!) const data = await fetchData const links: LinkData[] = [] const tags: SimpleSlug[] = [] const validLinks = new Set(Object.keys(data).map((slug) => simplifySlug(slug as FullSlug))) for (const [src, details] of Object.entries(data)) { const source = simplifySlug(src as FullSlug) const outgoing = details.links ?? [] for (const dest of outgoing) { if (validLinks.has(dest)) { links.push({ 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, target: tag }) } } } const neighbourhood = new Set() const wl: (SimpleSlug | "__SENTINEL")[] = [slug, "__SENTINEL"] if (depth >= 0) { while (depth >= 0 && wl.length > 0) { // compute neighbours const cur = wl.shift()! if (cur === "__SENTINEL") { depth-- wl.push("__SENTINEL") } else { neighbourhood.add(cur) const outgoing = links.filter((l) => l.source === cur) const incoming = links.filter((l) => l.target === cur) wl.push(...outgoing.map((l) => l.target), ...incoming.map((l) => l.source)) } } } else { Object.keys(data).forEach((id) => neighbourhood.add(simplifySlug(id as FullSlug))) if (showTags) tags.forEach((tag) => neighbourhood.add(tag)) } const graphData: { nodes: NodeData[]; links: LinkData[] } = { nodes: [...neighbourhood].map((url) => { const text = url.startsWith("tags/") ? "#" + url.substring(5) : data[url]?.title ?? url return { id: url, text: text, tags: data[url]?.tags ?? [], } }), links: links.filter((l) => neighbourhood.has(l.source) && neighbourhood.has(l.target)), } const simulation: d3.Simulation = d3 .forceSimulation(graphData.nodes) .force("charge", d3.forceManyBody().strength(-100 * repelForce)) .force( "link", d3 .forceLink(graphData.links) .id((d: any) => d.id) .distance(linkDistance), ) .force("center", d3.forceCenter().strength(centerForce)) const height = Math.max(graph.offsetHeight, 250) const width = graph.offsetWidth const svg = d3 .select("#" + container) .append("svg") .attr("width", width) .attr("height", height) .attr("viewBox", [-width / 2 / scale, -height / 2 / scale, width / scale, height / scale]) // draw links between nodes const link = svg .append("g") .selectAll("line") .data(graphData.links) .join("line") .attr("class", "link") .attr("stroke", "var(--lightgray)") .attr("stroke-width", 1) // svg groups const graphNode = svg.append("g").selectAll("g").data(graphData.nodes).enter().append("g") // calculate color const color = (d: NodeData) => { const isCurrent = d.id === slug if (isCurrent) { return "var(--secondary)" } else if (visited.has(d.id) || d.id.startsWith("tags/")) { return "var(--tertiary)" } else { return "var(--gray)" } } const drag = (simulation: d3.Simulation) => { function dragstarted(event: any, d: NodeData) { if (!event.active) simulation.alphaTarget(1).restart() d.fx = d.x d.fy = d.y } function dragged(event: any, d: NodeData) { d.fx = event.x d.fy = event.y } function dragended(event: any, d: NodeData) { if (!event.active) simulation.alphaTarget(0) d.fx = null d.fy = null } const noop = () => {} return d3 .drag() .on("start", enableDrag ? dragstarted : noop) .on("drag", enableDrag ? dragged : noop) .on("end", enableDrag ? dragended : noop) } function nodeRadius(d: NodeData) { const numLinks = links.filter((l: any) => l.source.id === d.id || l.target.id === d.id).length return 2 + Math.sqrt(numLinks) } // draw individual nodes const node = graphNode .append("circle") .attr("class", "node") .attr("id", (d) => d.id) .attr("r", nodeRadius) .attr("fill", color) .style("cursor", "pointer") .on("click", (_, d) => { const targ = resolveRelative(fullSlug, d.id) window.spaNavigate(new URL(targ, window.location.toString())) }) .on("mouseover", function (_, d) { const neighbours: SimpleSlug[] = data[fullSlug].links ?? [] const neighbourNodes = d3 .selectAll(".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) // highlight links linkNodes.transition().duration(200).attr("stroke", "var(--gray)").attr("stroke-width", 1) const bigFont = fontSize * 1.5 // show text for self const parent = this.parentNode as HTMLElement d3.select(parent) .raise() .select("text") .transition() .duration(200) .attr("opacityOld", d3.select(parent).select("text").style("opacity")) .style("opacity", 1) .style("font-size", bigFont + "em") }) .on("mouseleave", function (_, d) { const currentId = d.id const linkNodes = d3 .selectAll(".link") .filter((d: any) => d.source.id === currentId || d.target.id === currentId) linkNodes.transition().duration(200).attr("stroke", "var(--lightgray)") const parent = this.parentNode as HTMLElement d3.select(parent) .select("text") .transition() .duration(200) .style("opacity", d3.select(parent).select("text").attr("opacityOld")) .style("font-size", fontSize + "em") }) // @ts-ignore .call(drag(simulation)) // draw labels const labels = graphNode .append("text") .attr("dx", 0) .attr("dy", (d) => -nodeRadius(d) + "px") .attr("text-anchor", "middle") .text((d) => d.text) .style("opacity", (opacityScale - 1) / 3.75) .style("pointer-events", "none") .style("font-size", fontSize + "em") .raise() // @ts-ignore .call(drag(simulation)) // set panning if (enableZoom) { svg.call( d3 .zoom() .extent([ [0, 0], [width, height], ]) .scaleExtent([0.25, 4]) .on("zoom", ({ transform }) => { link.attr("transform", transform) node.attr("transform", transform) const scale = transform.k * opacityScale const scaledOpacity = Math.max((scale - 1) / 3.75, 0) labels.attr("transform", transform).style("opacity", scaledOpacity) }), ) } // progress the simulation simulation.on("tick", () => { link .attr("x1", (d: any) => d.source.x) .attr("y1", (d: any) => d.source.y) .attr("x2", (d: any) => d.target.x) .attr("y2", (d: any) => d.target.y) node.attr("cx", (d: any) => d.x).attr("cy", (d: any) => d.y) labels.attr("x", (d: any) => d.x).attr("y", (d: any) => d.y) }) } function renderGlobalGraph() { const slug = getFullSlug(window) const container = document.getElementById("global-graph-outer") const sidebar = container?.closest(".sidebar") as HTMLElement container?.classList.add("active") if (sidebar) { sidebar.style.zIndex = "1" } renderGraph("global-graph-container", slug) function hideGlobalGraph() { container?.classList.remove("active") const graph = document.getElementById("global-graph-container") if (sidebar) { sidebar.style.zIndex = "unset" } if (!graph) return removeAllChildren(graph) } registerEscapeHandler(container, hideGlobalGraph) } document.addEventListener("nav", async (e: unknown) => { const slug = (e as CustomEventMap["nav"]).detail.url addToVisited(slug) await renderGraph("graph-container", slug) const containerIcon = document.getElementById("global-graph-icon") containerIcon?.removeEventListener("click", renderGlobalGraph) containerIcon?.addEventListener("click", renderGlobalGraph) })