🎨 Refresh associated blocks after find-replacing and rolling back doc

https://github.com/siyuan-note/siyuan/issues/12439
https://github.com/siyuan-note/siyuan/issues/12438

kernel/model/push_reload.go

@@ -0,0 +1,215 @@
+// SiYuan - Refactor your thinking
+// Copyright (c) 2020-present, b3log.org
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+package model
+
+import (
+	"strings"
+	"time"
+
+	"github.com/88250/gulu"
+	"github.com/88250/lute/ast"
+	"github.com/88250/lute/parse"
+	"github.com/emirpasic/gods/sets/hashset"
+	"github.com/siyuan-note/siyuan/kernel/av"
+	"github.com/siyuan-note/siyuan/kernel/sql"
+	"github.com/siyuan-note/siyuan/kernel/task"
+	"github.com/siyuan-note/siyuan/kernel/treenode"
+	"github.com/siyuan-note/siyuan/kernel/util"
+)
+
+func refreshProtyle(rootID string) {
+	// 刷新关联的引用
+	defTree, _ := LoadTreeByBlockID(rootID)
+	if nil != defTree {
+		defIDs := sql.QueryChildDefIDsByRootDefID(rootID)
+
+		var defNodes []*ast.Node
+		ast.Walk(defTree.Root, func(n *ast.Node, entering bool) ast.WalkStatus {
+			if !entering || !n.IsBlock() {
+				return ast.WalkContinue
+			}
+
+			if gulu.Str.Contains(n.ID, defIDs) {
+				defNodes = append(defNodes, n)
+			}
+			return ast.WalkContinue
+		})
+
+		for _, def := range defNodes {
+			refreshDynamicRefText(def, defTree)
+		}
+	}
+
+	// 刷新关联的嵌入块
+	refIDs, _ := sql.QueryRefIDsByDefID(rootID, true)
+	var rootIDs []string
+	bts := treenode.GetBlockTrees(refIDs)
+	for _, bt := range bts {
+		rootIDs = append(rootIDs, bt.RootID)
+	}
+	rootIDs = gulu.Str.RemoveDuplicatedElem(rootIDs)
+	for _, id := range rootIDs {
+		task.AppendAsyncTaskWithDelay(task.ReloadProtyle, 200*time.Millisecond, util.PushReloadProtyle, id)
+	}
+}
+
+// refreshRefCount 用于刷新定义块处的引用计数。
+func refreshRefCount(rootID, blockID string) {
+	sql.WaitForWritingDatabase()
+
+	bt := treenode.GetBlockTree(blockID)
+	if nil == bt {
+		return
+	}
+
+	refCounts := sql.QueryRootChildrenRefCount(bt.RootID)
+	refCount := refCounts[blockID]
+	var rootRefCount int
+	for _, count := range refCounts {
+		rootRefCount += count
+	}
+	refIDs, _, _ := GetBlockRefs(blockID)
+	util.PushSetDefRefCount(rootID, blockID, refIDs, refCount, rootRefCount)
+}
+
+// refreshDynamicRefText 用于刷新块引用的动态锚文本。
+// 该实现依赖了数据库缓存，导致外部调用时可能需要阻塞等待数据库写入后才能获取到 refs
+func refreshDynamicRefText(updatedDefNode *ast.Node, updatedTree *parse.Tree) {
+	changedDefs := map[string]*ast.Node{updatedDefNode.ID: updatedDefNode}
+	changedTrees := map[string]*parse.Tree{updatedTree.ID: updatedTree}
+	refreshDynamicRefTexts(changedDefs, changedTrees)
+}
+
+// refreshDynamicRefTexts 用于批量刷新块引用的动态锚文本。
+// 该实现依赖了数据库缓存，导致外部调用时可能需要阻塞等待数据库写入后才能获取到 refs
+func refreshDynamicRefTexts(updatedDefNodes map[string]*ast.Node, updatedTrees map[string]*parse.Tree) {
+	// 1. 更新引用的动态锚文本
+	treeRefNodeIDs := map[string]*hashset.Set{}
+	var changedParentNodes []*ast.Node
+	for _, updateNode := range updatedDefNodes {
+		refs, parentNodes := getRefsCacheByDefNode(updateNode)
+		for _, ref := range refs {
+			if refIDs, ok := treeRefNodeIDs[ref.RootID]; !ok {
+				refIDs = hashset.New()
+				refIDs.Add(ref.BlockID)
+				treeRefNodeIDs[ref.RootID] = refIDs
+			} else {
+				refIDs.Add(ref.BlockID)
+			}
+		}
+		if 0 < len(parentNodes) {
+			changedParentNodes = append(changedParentNodes, parentNodes...)
+		}
+	}
+	if 0 < len(changedParentNodes) {
+		for _, parent := range changedParentNodes {
+			updatedDefNodes[parent.ID] = parent
+		}
+	}
+
+	changedRefTree := map[string]*parse.Tree{}
+
+	for refTreeID, refNodeIDs := range treeRefNodeIDs {
+		refTree, ok := updatedTrees[refTreeID]
+		if !ok {
+			var err error
+			refTree, err = LoadTreeByBlockID(refTreeID)
+			if err != nil {
+				continue
+			}
+		}
+
+		var refTreeChanged bool
+		ast.Walk(refTree.Root, func(n *ast.Node, entering bool) ast.WalkStatus {
+			if !entering {
+				return ast.WalkContinue
+			}
+
+			if n.IsBlock() && refNodeIDs.Contains(n.ID) {
+				changed, changedDefNodes := updateRefText(n, updatedDefNodes)
+				if !refTreeChanged && changed {
+					refTreeChanged = true
+				}
+
+				// 推送动态锚文本节点刷新
+				for _, defNode := range changedDefNodes {
+					if "ref-d" == defNode.refType {
+						task.AppendAsyncTaskWithDelay(task.SetRefDynamicText, 200*time.Millisecond, util.PushSetRefDynamicText, refTreeID, n.ID, defNode.id, defNode.refText)
+					}
+				}
+				return ast.WalkContinue
+			}
+			return ast.WalkContinue
+		})
+
+		if refTreeChanged {
+			changedRefTree[refTreeID] = refTree
+			sql.UpdateRefsTreeQueue(refTree)
+		}
+	}
+
+	// 2. 更新属性视图主键内容
+	for _, updatedDefNode := range updatedDefNodes {
+		avs := updatedDefNode.IALAttr(av.NodeAttrNameAvs)
+		if "" == avs {
+			continue
+		}
+
+		avIDs := strings.Split(avs, ",")
+		for _, avID := range avIDs {
+			attrView, parseErr := av.ParseAttributeView(avID)
+			if nil != parseErr {
+				continue
+			}
+
+			changedAv := false
+			blockValues := attrView.GetBlockKeyValues()
+			if nil == blockValues {
+				continue
+			}
+
+			for _, blockValue := range blockValues.Values {
+				if blockValue.Block.ID == updatedDefNode.ID {
+					newContent := getNodeRefText(updatedDefNode)
+					if newContent != blockValue.Block.Content {
+						blockValue.Block.Content = newContent
+						changedAv = true
+					}
+					break
+				}
+			}
+			if changedAv {
+				av.SaveAttributeView(attrView)
+				ReloadAttrView(avID)
+			}
+		}
+	}
+
+	// 3. 保存变更
+	for _, tree := range changedRefTree {
+		indexWriteTreeUpsertQueue(tree)
+	}
+}
+
+// ReloadAttrView 用于重新加载属性视图。
+func ReloadAttrView(avID string) {
+	task.AppendAsyncTaskWithDelay(task.ReloadAttributeView, 200*time.Millisecond, pushReloadAttrView, avID)
+}
+
+func pushReloadAttrView(avID string) {
+	util.BroadcastByType("protyle", "refreshAttributeView", 0, "", map[string]interface{}{"id": avID})
+}