✨ 编辑器式反链面板 #3565

siyuan-note · Oct 2, 2022 · 2155a56 · 2155a56
1 parent 1475fd9
commit 2155a56
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/kernel/api/ref.go b/kernel/api/ref.go
@@ -48,8 +48,9 @@ func getBackmentionDoc(c *gin.Context) {
 	}
 
 	defID := arg["defID"].(string)
+	refTreeID := arg["refTreeID"].(string)
 	keyword := ""
-	backlinks := model.GetBackmentionDoc(defID, keyword)
+	backlinks := model.GetBackmentionDoc(defID, refTreeID, keyword)
 	ret.Data = map[string]interface{}{
 		"backmentions": backlinks,
 	}

diff --git a/kernel/model/backlink.go b/kernel/model/backlink.go
@@ -164,7 +164,7 @@ type Backlink struct {
 	Expand     bool         `json:"expand"`
 }
 
-func GetBackmentionDoc(defID, keyword string) (ret []*Backlink) {
+func GetBackmentionDoc(defID, refTreeID, keyword string) (ret []*Backlink) {
 	ret = []*Backlink{}
 	beforeLen := 12
 	sqlBlock := sql.GetBlock(defID)
@@ -177,9 +177,15 @@ func GetBackmentionDoc(defID, keyword string) (ret []*Backlink) {
 	refs = removeDuplicatedRefs(refs) // 同一个块中引用多个相同块时反链去重 https://github.com/siyuan-note/siyuan/issues/3317
 
 	linkRefs, excludeBacklinkIDs := buildLinkRefs(rootID, refs)
-	mentions := buildTreeBackmention(sqlBlock, linkRefs, keyword, excludeBacklinkIDs, beforeLen)
+	tmpMentions := buildTreeBackmention(sqlBlock, linkRefs, keyword, excludeBacklinkIDs, beforeLen)
 	luteEngine := NewLute()
 	treeCache := map[string]*parse.Tree{}
+	var mentions []*Block
+	for _, mention := range tmpMentions {
+		if mention.RootID == refTreeID {
+			mentions = append(mentions, mention)
+		}
+	}
 	for _, mention := range mentions {
 		refTree := treeCache[mention.RootID]
 		if nil == refTree {