import { ContentDetails } from "../../plugins/emitters/contentIndex" import * as d3 from 'd3' type NodeData = { id: string, text: string, tags: string[] } & d3.SimulationNodeDatum type LinkData = { source: string, target: string } function relative(from: string, to: string) { const pieces = [location.protocol, '//', location.host, location.pathname] const url = pieces.join('').slice(0, -from.length) + to return url } function removeAllChildren(node: HTMLElement) { while (node.firstChild) { node.removeChild(node.firstChild) } } async function renderGraph(container: string, slug: string) { const graph = document.getElementById(container)! removeAllChildren(graph) let { drag: enableDrag, zoom: enableZoom, depth, scale, repelForce, centerForce, linkDistance, fontSize, opacityScale } = JSON.parse(graph.dataset["cfg"]!) const data = await fetchData const links: LinkData[] = [] for (const [src, details] of Object.entries(data)) { const outgoing = details.links ?? [] for (const dest of outgoing) { if (src in data && dest in data) { links.push({ source: src, target: dest }) } } } const neighbourhood = new Set() const wl = [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 { links.flatMap(l => [l.source, l.target]).forEach((id) => neighbourhood.add(id)) } const graphData: { nodes: NodeData[], links: LinkData[] } = { nodes: Object.keys(data).filter(id => neighbourhood.has(id)).map(url => ({ id: url, text: data[url]?.title ?? url, 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", 2) // svg groups const graphNode = svg.append("g").selectAll("g").data(graphData.nodes).enter().append("g") // calculate radius const color = (d: NodeData) => { // TODO: does this handle the index page const isCurrent = d.id === slug return isCurrent ? "var(--secondary)" : "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 = relative(slug, d.id) window.spaNavigate(new URL(targ)) }) .on("mouseover", function(_, d) { const neighbours: string[] = data[slug].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)") 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) + 8 + "px") .attr("text-anchor", "middle") .text((d) => data[d.id]?.title || (d.id.charAt(1).toUpperCase() + d.id.slice(2)).replace("-", " ")) .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 = document.body.dataset["slug"]! renderGraph("global-graph-container", slug) const container = document.getElementById("global-graph-outer") container?.classList.add("active") function hideGlobalGraph(this: HTMLElement, e: HTMLElementEventMap["click"]) { if (e.target !== this) return container?.classList.remove("active") const graph = document.getElementById("global-graph-container")! removeAllChildren(graph) } container?.removeEventListener("click", hideGlobalGraph) container?.addEventListener("click", hideGlobalGraph) } document.addEventListener("nav", async (e: unknown) => { const slug = (e as CustomEventMap["nav"]).detail.url await renderGraph("graph-container", slug) const containerIcon = document.getElementById("global-graph-icon") containerIcon?.removeEventListener("click", renderGlobalGraph) containerIcon?.addEventListener("click", renderGlobalGraph) })