From 8f01ef2c024f22bc7b861d915827b8dad562aebb Mon Sep 17 00:00:00 2001
From: Amaury Hamon <contact@amauryhamon.com>
Date: Tue, 12 Sep 2023 23:15:03 +0200
Subject: [PATCH] graph active on nodes n links + click close glob

---
 quartz/components/scripts/graph.inline.ts | 27 ++++++++++++++++-------
 quartz/components/scripts/util.ts         | 26 ++++++++++++++++++++++
 2 files changed, 45 insertions(+), 8 deletions(-)

diff --git a/quartz/components/scripts/graph.inline.ts b/quartz/components/scripts/graph.inline.ts
index dc5c99d..68c5215 100644
--- a/quartz/components/scripts/graph.inline.ts
+++ b/quartz/components/scripts/graph.inline.ts
@@ -1,6 +1,6 @@
 import type { ContentDetails } from "../../plugins/emitters/contentIndex"
 import * as d3 from "d3"
-import { registerEscapeHandler, removeAllChildren } from "./util"
+import { registerCloseHandler, registerEscapeHandler, removeAllChildren } from "./util"
 import { FullSlug, SimpleSlug, getFullSlug, resolveRelative, simplifySlug } from "../../util/path"
 
 type NodeData = {
@@ -186,12 +186,18 @@ async function renderGraph(container: string, fullSlug: FullSlug) {
       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)
+      
+      neighbourNodes
+        .transition().duration(200).attr("fill", color)
+        .attr("class", "node active")
 
       // highlight links
-      linkNodes.transition().duration(200).attr("stroke", "var(--gray)").attr("stroke-width", 1)
+      linkNodes
+        .attr("class", "link active")
+        .transition().duration(200).attr("stroke", "var(--gray)").attr("stroke-width", 1)
+        
 
       const bigFont = fontSize * 1.5
 
@@ -208,11 +214,16 @@ async function renderGraph(container: string, fullSlug: FullSlug) {
     })
     .on("mouseleave", function (_, d) {
       const currentId = d.id
+      const neighbourNodes$ = d3
+        .selectAll(".node.active").attr("class", "node");
       const linkNodes = d3
-        .selectAll(".link")
-        .filter((d: any) => d.source.id === currentId || d.target.id === currentId)
+        .selectAll(".link.active")
+        .filter((d: any) => d.source.id === currentId || d.target.id === currentId);
+        
 
-      linkNodes.transition().duration(200).attr("stroke", "var(--lightgray)")
+      linkNodes
+        .transition().duration(200).attr("stroke", "var(--lightgray)")
+        .attr("class", "link");
 
       const parent = this.parentNode as HTMLElement
       d3.select<HTMLElement, NodeData>(parent)
@@ -296,7 +307,7 @@ function renderGlobalGraph() {
     removeAllChildren(graph)
   }
 
-  registerEscapeHandler(container, hideGlobalGraph)
+  registerCloseHandler(container, hideGlobalGraph)
 }
 
 document.addEventListener("nav", async (e: unknown) => {
diff --git a/quartz/components/scripts/util.ts b/quartz/components/scripts/util.ts
index 5fcabad..b69dd74 100644
--- a/quartz/components/scripts/util.ts
+++ b/quartz/components/scripts/util.ts
@@ -18,6 +18,32 @@ export function registerEscapeHandler(outsideContainer: HTMLElement | null, cb:
   document.addEventListener("keydown", esc)
 }
 
+export function registerCloseHandler(container: HTMLElement | null, cb: () => void) {
+  if (!container) return;
+  function clickInsideOrOnChild(e: MouseEvent) {
+    // Check if the click target is the container itself or a child element
+    if (container.contains(e.target as Node)) {
+      e.preventDefault();
+      // Check if the click is a double-click (button === 0 indicates a left-click)
+      // if (e.detail === 2 && e.button === 0) {
+      //   // Double-click detected; do not close the container
+      //   return;
+      // }
+      cb();
+    }
+  }
+
+  function esc(e: HTMLElementEventMap["keydown"]) {
+    if (!e.key.startsWith("Esc")) return
+    e.preventDefault()
+    cb()
+  }
+  document.removeEventListener("keydown", esc)
+  document.addEventListener("keydown", esc)
+  container.removeEventListener("click", clickInsideOrOnChild);
+  container.addEventListener("click", clickInsideOrOnChild);
+}
+
 export function removeAllChildren(node: HTMLElement) {
   while (node.firstChild) {
     node.removeChild(node.firstChild)