From 232bd2f016f7d57602fe39703407100f5d26f278 Mon Sep 17 00:00:00 2001
From: Jacky Zhao <j.zhao2k19@gmail.com>
Date: Fri, 27 May 2022 18:01:20 +0000
Subject: [PATCH] Merge pull request #134 from aidenybai/add-prefetching-within-graph

---
 assets/js/router.js        |   14 ++-
 assets/js/search.js        |    2 
 assets/js/graph.js         |  205 +++++++++++++++++++++------------------------------
 layouts/partials/head.html |    2 
 4 files changed, 96 insertions(+), 127 deletions(-)

diff --git a/assets/js/graph.js b/assets/js/graph.js
index f71e44d..c7634cf 100644
--- a/assets/js/graph.js
+++ b/assets/js/graph.js
@@ -1,12 +1,5 @@
-async function drawGraph(
-  baseUrl,
-  pathColors,
-  depth,
-  enableDrag,
-  enableLegend,
-  enableZoom
-) {
-  const container = document.getElementById('graph-container')
+async function drawGraph(baseUrl, pathColors, depth, enableDrag, enableLegend, enableZoom) {
+  const container = document.getElementById("graph-container")
   const { index, links, content } = await fetchData
 
   // Use .pathname to remove hashes / searchParams / text fragments
@@ -23,22 +16,19 @@
   const copyLinks = JSON.parse(JSON.stringify(links))
 
   const neighbours = new Set()
-  const wl = [curPage || '/', '__SENTINEL']
+  const wl = [curPage || "/", "__SENTINEL"]
   if (depth >= 0) {
     while (depth >= 0 && wl.length > 0) {
       // compute neighbours
       const cur = wl.shift()
-      if (cur === '__SENTINEL') {
+      if (cur === "__SENTINEL") {
         depth--
-        wl.push('__SENTINEL')
+        wl.push("__SENTINEL")
       } else {
         neighbours.add(cur)
         const outgoing = index.links[cur] || []
         const incoming = index.backlinks[cur] || []
-        wl.push(
-          ...outgoing.map((l) => l.target),
-          ...incoming.map((l) => l.source)
-        )
+        wl.push(...outgoing.map((l) => l.target), ...incoming.map((l) => l.source))
       }
     }
   } else {
@@ -47,14 +37,12 @@
 
   const data = {
     nodes: [...neighbours].map((id) => ({ id })),
-    links: copyLinks.filter(
-      (l) => neighbours.has(l.source) && neighbours.has(l.target)
-    ),
+    links: copyLinks.filter((l) => neighbours.has(l.source) && neighbours.has(l.target)),
   }
 
   const color = (d) => {
-    if (d.id === curPage || (d.id === '/' && curPage === '')) {
-      return 'var(--g-node-active)'
+    if (d.id === curPage || (d.id === "/" && curPage === "")) {
+      return "var(--g-node-active)"
     }
 
     for (const pathColor of pathColors) {
@@ -65,7 +53,7 @@
       }
     }
 
-    return 'var(--g-node)'
+    return "var(--g-node)"
   }
 
   const drag = (simulation) => {
@@ -86,12 +74,12 @@
       d.fy = null
     }
 
-    const noop = () => { }
+    const noop = () => {}
     return d3
       .drag()
-      .on('start', enableDrag ? dragstarted : noop)
-      .on('drag', enableDrag ? dragged : noop)
-      .on('end', enableDrag ? dragended : noop)
+      .on("start", enableDrag ? dragstarted : noop)
+      .on("drag", enableDrag ? dragged : noop)
+      .on("end", enableDrag ? dragended : noop)
   }
 
   const height = Math.max(container.offsetHeight, 250)
@@ -99,67 +87,58 @@
 
   const simulation = d3
     .forceSimulation(data.nodes)
-    .force('charge', d3.forceManyBody().strength(-30))
+    .force("charge", d3.forceManyBody().strength(-30))
     .force(
-      'link',
+      "link",
       d3
         .forceLink(data.links)
         .id((d) => d.id)
-        .distance(40)
+        .distance(40),
     )
-    .force('center', d3.forceCenter())
+    .force("center", d3.forceCenter())
 
   const svg = d3
-    .select('#graph-container')
-    .append('svg')
-    .attr('width', width)
-    .attr('height', height)
-    .attr('viewBox', [-width / 2, -height / 2, width, height])
+    .select("#graph-container")
+    .append("svg")
+    .attr("width", width)
+    .attr("height", height)
+    .attr("viewBox", [-width / 2, -height / 2, width, height])
 
   if (enableLegend) {
-    const legend = [
-      { Current: 'var(--g-node-active)' },
-      { Note: 'var(--g-node)' },
-      ...pathColors,
-    ]
+    const legend = [{ Current: "var(--g-node-active)" }, { Note: "var(--g-node)" }, ...pathColors]
     legend.forEach((legendEntry, i) => {
       const key = Object.keys(legendEntry)[0]
       const colour = legendEntry[key]
       svg
-        .append('circle')
-        .attr('cx', -width / 2 + 20)
-        .attr('cy', height / 2 - 30 * (i + 1))
-        .attr('r', 6)
-        .style('fill', colour)
+        .append("circle")
+        .attr("cx", -width / 2 + 20)
+        .attr("cy", height / 2 - 30 * (i + 1))
+        .attr("r", 6)
+        .style("fill", colour)
       svg
-        .append('text')
-        .attr('x', -width / 2 + 40)
-        .attr('y', height / 2 - 30 * (i + 1))
+        .append("text")
+        .attr("x", -width / 2 + 40)
+        .attr("y", height / 2 - 30 * (i + 1))
         .text(key)
-        .style('font-size', '15px')
-        .attr('alignment-baseline', 'middle')
+        .style("font-size", "15px")
+        .attr("alignment-baseline", "middle")
     })
   }
 
   // draw links between nodes
   const link = svg
-    .append('g')
-    .selectAll('line')
+    .append("g")
+    .selectAll("line")
     .data(data.links)
-    .join('line')
-    .attr('class', 'link')
-    .attr('stroke', 'var(--g-link)')
-    .attr('stroke-width', 2)
-    .attr('data-source', (d) => d.source.id)
-    .attr('data-target', (d) => d.target.id)
+    .join("line")
+    .attr("class", "link")
+    .attr("stroke", "var(--g-link)")
+    .attr("stroke-width", 2)
+    .attr("data-source", (d) => d.source.id)
+    .attr("data-target", (d) => d.target.id)
 
   // svg groups
-  const graphNode = svg
-    .append('g')
-    .selectAll('g')
-    .data(data.nodes)
-    .enter()
-    .append('g')
+  const graphNode = svg.append("g").selectAll("g").data(data.nodes).enter().append("g")
 
   // calculate radius
   const nodeRadius = (d) => {
@@ -170,82 +149,68 @@
 
   // 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) => {
+    .append("circle")
+    .attr("class", "node")
+    .attr("id", (d) => d.id)
+    .attr("r", nodeRadius)
+    .attr("fill", color)
+    .style("cursor", "pointer")
+    .on("click", (_, d) => {
       // SPA navigation
-      window.navigate(
-        new URL(`${baseUrl}${decodeURI(d.id).replace(/\s+/g, '-')}/`),
-        '.singlePage'
-      )
+      window.Million.navigate(new URL(`${baseUrl}${decodeURI(d.id).replace(/\s+/g, "-")}/`), ".singlePage")
     })
-    .on('mouseover', function(_, d) {
-      d3.selectAll('.node')
-        .transition()
-        .duration(100)
-        .attr('fill', 'var(--g-node-inactive)')
+    .on("mouseover", function (_, d) {
+      d3.selectAll(".node").transition().duration(100).attr("fill", "var(--g-node-inactive)")
 
       const neighbours = parseIdsFromLinks([
         ...(index.links[d.id] || []),
         ...(index.backlinks[d.id] || []),
       ])
-      const neighbourNodes = d3
-        .selectAll('.node')
-        .filter((d) => neighbours.includes(d.id))
+      const neighbourNodes = d3.selectAll(".node").filter((d) => neighbours.includes(d.id))
       const currentId = d.id
+      window.Million.prefetch(new URL(`${baseUrl}${decodeURI(d.id).replace(/\s+/g, "-")}/`))
       const linkNodes = d3
-        .selectAll('.link')
+        .selectAll(".link")
         .filter((d) => d.source.id === currentId || d.target.id === currentId)
 
       // highlight neighbour nodes
-      neighbourNodes.transition().duration(200).attr('fill', color)
+      neighbourNodes.transition().duration(200).attr("fill", color)
 
       // highlight links
-      linkNodes
-        .transition()
-        .duration(200)
-        .attr('stroke', 'var(--g-link-active)')
+      linkNodes.transition().duration(200).attr("stroke", "var(--g-link-active)")
 
       // show text for self
       d3.select(this.parentNode)
         .raise()
-        .select('text')
+        .select("text")
         .transition()
         .duration(200)
-        .style('opacity', 1)
+        .style("opacity", 1)
     })
-    .on('mouseleave', function(_, d) {
-      d3.selectAll('.node').transition().duration(200).attr('fill', color)
+    .on("mouseleave", function (_, d) {
+      d3.selectAll(".node").transition().duration(200).attr("fill", color)
 
       const currentId = d.id
       const linkNodes = d3
-        .selectAll('.link')
+        .selectAll(".link")
         .filter((d) => d.source.id === currentId || d.target.id === currentId)
 
-      linkNodes.transition().duration(200).attr('stroke', 'var(--g-link)')
+      linkNodes.transition().duration(200).attr("stroke", "var(--g-link)")
 
-      d3.select(this.parentNode)
-        .select('text')
-        .transition()
-        .duration(200)
-        .style('opacity', 0)
+      d3.select(this.parentNode).select("text").transition().duration(200).style("opacity", 0)
     })
     .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) => content[d.id]?.title || d.id.replace('-', ' '))
-    .style('opacity', 0)
-    .style('pointer-events', 'none')
-    .style('font-size', '0.4em')
+    .append("text")
+    .attr("dx", 0)
+    .attr("dy", (d) => nodeRadius(d) + 8 + "px")
+    .attr("text-anchor", "middle")
+    .text((d) => content[d.id]?.title || d.id.replace("-", " "))
+    .style("opacity", 0)
+    .style("pointer-events", "none")
+    .style("font-size", "0.4em")
     .raise()
     .call(drag(simulation))
 
@@ -260,24 +225,24 @@
           [width, height],
         ])
         .scaleExtent([0.25, 4])
-        .on('zoom', ({ transform }) => {
-          link.attr('transform', transform)
-          node.attr('transform', transform)
+        .on("zoom", ({ transform }) => {
+          link.attr("transform", transform)
+          node.attr("transform", transform)
           const scale = transform.k
           const scaledOpacity = Math.max((scale - 1) / 3.75, 0)
-          labels.attr('transform', transform).style('opacity', scaledOpacity)
-        })
+          labels.attr("transform", transform).style("opacity", scaledOpacity)
+        }),
     )
   }
 
   // progress the simulation
-  simulation.on('tick', () => {
+  simulation.on("tick", () => {
     link
-      .attr('x1', (d) => d.source.x)
-      .attr('y1', (d) => d.source.y)
-      .attr('x2', (d) => d.target.x)
-      .attr('y2', (d) => d.target.y)
-    node.attr('cx', (d) => d.x).attr('cy', (d) => d.y)
-    labels.attr('x', (d) => d.x).attr('y', (d) => d.y)
+      .attr("x1", (d) => d.source.x)
+      .attr("y1", (d) => d.source.y)
+      .attr("x2", (d) => d.target.x)
+      .attr("y2", (d) => d.target.y)
+    node.attr("cx", (d) => d.x).attr("cy", (d) => d.y)
+    labels.attr("x", (d) => d.x).attr("y", (d) => d.y)
   })
 }
diff --git a/assets/js/router.js b/assets/js/router.js
index 81c25ac..3cd3350 100644
--- a/assets/js/router.js
+++ b/assets/js/router.js
@@ -1,12 +1,16 @@
-import { router, navigate } from "https://unpkg.com/million@1.8.9-0/dist/router.mjs"
+import { router, navigate, reload, prefetch } from "https://unpkg.com/million@1.9.4/dist/router.mjs"
 
 export const attachSPARouting = (draw) => {
-  // SPA navigation for access later
-  window.navigate = navigate
-  // We only mutate document.title and content within .singlePage element
+  // Attach SPA functions to the global Million namespace
+  window.Million = {
+    router,
+    navigate,
+    reload,
+    prefetch,
+  };
   router(".singlePage")
   // We need on initial load, then subsequent redirs
   // requestAnimationFrame() delays graph draw until SPA routing is finished
-  window.addEventListener("million:navigate", () => requestAnimationFrame(draw))
+  reload(draw)
   window.addEventListener("DOMContentLoaded", () => requestAnimationFrame(draw))
 }
diff --git a/assets/js/search.js b/assets/js/search.js
index bb94cd3..ee00647 100644
--- a/assets/js/search.js
+++ b/assets/js/search.js
@@ -153,7 +153,7 @@
 
   const redir = (id, term) => {
     // SPA navigation
-    window.navigate(
+    window.Million.navigate(
       new URL(
         `${BASE_URL.replace(/\/$/g, "")}${id}#:~:text=${encodeURIComponent(term)}/`
       ),
diff --git a/layouts/partials/head.html b/layouts/partials/head.html
index a49800b..b65bb84 100644
--- a/layouts/partials/head.html
+++ b/layouts/partials/head.html
@@ -107,7 +107,7 @@
   </script>
   {{else}}
   <script>
-    window.navigate = (url) => window.location.href = url;
+    window.Million.navigate = (url) => window.location.href = url;
     draw();
   </script>
   {{end}}

--
Gitblit v1.10.0