add showing and hiding dailynote

shenjinglei · Jan 14, 2024 · 6f0fdd8 · 6f0fdd8
1 parent 22521a8
commit 6f0fdd8
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 3 deletions.
diff --git a/src/dock.ts b/src/dock.ts
@@ -1,5 +1,5 @@
 /* eslint-disable @typescript-eslint/no-non-null-assertion */
-import { Display, initRawGraph, setGraphType, setSourceNode, } from "./graph";
+import { Display, initRawGraph, isDailynote, setGraphType, setIsDailynote, setSourceNode, } from "./graph";
 import { i18n, plugin, rawGraph } from "./utils";
 import { adaptHotkey, fetchSyncPost, getFrontend } from "siyuan";
 
@@ -16,6 +16,7 @@ export function initDock() {
             ${i18n.pluginName}
         </div>
         <span class="fn__flex-1 fn__space"></span>
+        <span id="graph_enhance_dailynote" class="block__icon b3-tooltips b3-tooltips__sw" aria-label="${i18n.dockBtnHideDN}"><svg id="graph_enhance_dailynote_icon"><use xlink:href="#iconCalendar"></use></svg></span>
         <span id="graph_enhance_path" class="block__icon b3-tooltips b3-tooltips__sw" aria-label="${i18n.dockBtnPath}"><svg><use xlink:href="#iconCode"></use></svg></span>
         <span id="graph_enhance_global" class="block__icon b3-tooltips b3-tooltips__sw" aria-label="${i18n.dockBtnGlobal}"><svg><use xlink:href="#iconLanguage"></use></svg></span>
         <span id="graph_enhance_neighbor" class="block__icon b3-tooltips b3-tooltips__sw" aria-label="${i18n.dockBtnNeighbor}"><svg><use xlink:href="#iconWorkspace"></use></svg></span>
@@ -44,6 +45,20 @@ export function initDock() {
         init() {
             this.element.innerHTML = dockHtml;
 
+            document.getElementById("graph_enhance_dailynote")!.onclick = async () => {
+                if (isDailynote) {
+                    setIsDailynote(false);
+                    document.getElementById("graph_enhance_dailynote")?.setAttribute("aria-label", i18n.dockBtnShowDN);
+                    document.getElementById("graph_enhance_dailynote_icon")?.setAttribute("style", "fill: RoyalBlue;");
+                } else {
+                    setIsDailynote(true);
+                    document.getElementById("graph_enhance_dailynote")?.setAttribute("aria-label", i18n.dockBtnHideDN);
+                    document.getElementById("graph_enhance_dailynote_icon")?.setAttribute("style", "fill: #5f6368;");
+                }
+
+                Display();
+            };
+
             document.getElementById("graph_enhance_refresh")!.onclick = async () => {
                 await refreashGraph();
 

diff --git a/src/graph.ts b/src/graph.ts
@@ -319,6 +319,11 @@ function insertEdge2(cur: QueueItem, _rawGraph: dagre.graphlib.Graph<any>, _proc
     }
 }
 
+export let isDailynote = true;
+export function setIsDailynote(_isDailynote: boolean) {
+    isDailynote = _isDailynote;
+}
+
 function searchUp2(cur: QueueItem, q: QueueItem[], _rawGraph: dagre.graphlib.Graph<any>, _processedGraph: dagre.graphlib.Graph<any>, penetrate = 0) {
     const curNodeValue = _processedGraph.node(cur.id);
     if (curNodeValue.state & 2) return; // search is already done
@@ -330,6 +335,7 @@ function searchUp2(cur: QueueItem, q: QueueItem[], _rawGraph: dagre.graphlib.Gra
 
     _rawGraph.inEdges(cur.id)
         ?.filter(e => _processedGraph.node(e.v)?.state !== 3)
+        .filter(e => isDailynote || !/^\d{4}-\d{2}-\d{2}$/.test(_rawGraph.node(e.v).label))
         .filter(e => penetrate || cur.count === 0 || _rawGraph.edge(e.v, e.w)?.state !== "broken")
         .forEach(x => q.push({
             id: x.v,
@@ -350,6 +356,7 @@ function searchDown2(cur: QueueItem, q: QueueItem[], _rawGraph: dagre.graphlib.G
 
     _rawGraph.outEdges(cur.id)
         ?.filter(e => _processedGraph.node(e.w)?.state !== 3)
+        .filter(e => isDailynote || !/^\d{4}-\d{2}-\d{2}$/.test(_rawGraph.node(e.w).label))
         .filter(e => penetrate || cur.count === 0 || _rawGraph.edge(e.v, e.w)?.state !== "broken")
         .forEach(e => q.push({
             id: e.w,

diff --git a/src/i18n/en_US.json b/src/i18n/en_US.json
@@ -49,5 +49,7 @@
   "tailThresholdTitle": "Long Tail Graph Threshold",
   "tailThresholdDescription": "Comma separated, for example: 1,2",
 
-  "dockBtnPath": "Path Graph"
+  "dockBtnPath": "Path Graph",
+  "dockBtnShowDN": "Show DailyNote",
+  "dockBtnHideDN": "Hide DailyNote"
 }
diff --git a/src/i18n/zh_CN.json b/src/i18n/zh_CN.json
@@ -49,5 +49,7 @@
   "tailThresholdTitle": "长尾图阈值",
   "tailThresholdDescription": "最小值与最大值用英文逗号隔开，如：1,2",
 
-  "dockBtnPath": "路径图"
+  "dockBtnPath": "路径图",
+  "dockBtnShowDN": "显示 DailyNote",
+  "dockBtnHideDN": "隐藏 DailyNote"
 }