mirror of
https://github.com/siyuan-note/siyuan.git
synced 2025-12-16 22:50:13 +01:00
This commit is contained in:
parent
960a60ac4c
commit
fcaddd6966
2 changed files with 50 additions and 4 deletions
|
|
@ -384,8 +384,8 @@ func buildBacklink(refID string, refTree *parse.Tree, luteEngine *lute.Lute) (re
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func BuildTreeBacklink(id, keyword, mentionKeyword string, beforeLen int) (boxID string, backlinks, backmentions []*Block, linkRefsCount, mentionsCount int) {
|
func BuildTreeBacklink(id, keyword, mentionKeyword string, beforeLen int) (boxID string, backlinks, backmentions []*Path, linkRefsCount, mentionsCount int) {
|
||||||
backlinks, backmentions = []*Block{}, []*Block{}
|
backlinks, backmentions = []*Path{}, []*Path{}
|
||||||
|
|
||||||
sqlBlock := sql.GetBlock(id)
|
sqlBlock := sql.GetBlock(id)
|
||||||
if nil == sqlBlock {
|
if nil == sqlBlock {
|
||||||
|
|
@ -396,9 +396,11 @@ func BuildTreeBacklink(id, keyword, mentionKeyword string, beforeLen int) (boxID
|
||||||
refs := sql.QueryRefsByDefID(id, true)
|
refs := sql.QueryRefsByDefID(id, true)
|
||||||
refs = removeDuplicatedRefs(refs) // 同一个块中引用多个相同块时反链去重 https://github.com/siyuan-note/siyuan/issues/3317
|
refs = removeDuplicatedRefs(refs) // 同一个块中引用多个相同块时反链去重 https://github.com/siyuan-note/siyuan/issues/3317
|
||||||
|
|
||||||
backlinks, excludeBacklinkIDs := buildLinkRefs(id, refs)
|
linkRefs, excludeBacklinkIDs := buildLinkRefs(id, refs)
|
||||||
|
backlinks = toSubTree(linkRefs, keyword)
|
||||||
|
|
||||||
backmentions = buildTreeBackmention(sqlBlock, backlinks, mentionKeyword, excludeBacklinkIDs, beforeLen)
|
mentionRefs := buildTreeBackmention(sqlBlock, linkRefs, mentionKeyword, excludeBacklinkIDs, beforeLen)
|
||||||
|
backmentions = toFlatTree(mentionRefs, 0, "backlink")
|
||||||
mentionsCount = len(backmentions)
|
mentionsCount = len(backmentions)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -27,6 +27,7 @@ import (
|
||||||
"github.com/88250/gulu"
|
"github.com/88250/gulu"
|
||||||
"github.com/88250/lute/ast"
|
"github.com/88250/lute/ast"
|
||||||
"github.com/siyuan-note/logging"
|
"github.com/siyuan-note/logging"
|
||||||
|
"github.com/siyuan-note/siyuan/kernel/search"
|
||||||
"github.com/siyuan-note/siyuan/kernel/treenode"
|
"github.com/siyuan-note/siyuan/kernel/treenode"
|
||||||
"github.com/siyuan-note/siyuan/kernel/util"
|
"github.com/siyuan-note/siyuan/kernel/util"
|
||||||
)
|
)
|
||||||
|
|
@ -123,6 +124,49 @@ func toFlatTree(blocks []*Block, baseDepth int, typ string) (ret []*Path) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func toSubTree(blocks []*Block, keyword string) (ret []*Path) {
|
||||||
|
keyword = strings.TrimSpace(keyword)
|
||||||
|
var blockRoots []*Block
|
||||||
|
for _, block := range blocks {
|
||||||
|
root := getBlockIn(blockRoots, block.RootID)
|
||||||
|
if nil == root {
|
||||||
|
root, _ = getBlock(block.RootID)
|
||||||
|
blockRoots = append(blockRoots, root)
|
||||||
|
}
|
||||||
|
block.Depth = 1
|
||||||
|
block.Count = len(block.Children)
|
||||||
|
root.Children = append(root.Children, block)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, root := range blockRoots {
|
||||||
|
treeNode := &Path{
|
||||||
|
ID: root.ID,
|
||||||
|
Box: root.Box,
|
||||||
|
Name: path.Base(root.HPath),
|
||||||
|
Type: "backlink",
|
||||||
|
NodeType: "NodeDocument",
|
||||||
|
SubType: root.SubType,
|
||||||
|
Depth: 0,
|
||||||
|
Count: len(root.Children),
|
||||||
|
}
|
||||||
|
|
||||||
|
rootPos := -1
|
||||||
|
var rootContent string
|
||||||
|
if "" != keyword {
|
||||||
|
rootPos, rootContent = search.MarkText(treeNode.Name, keyword, 12, Conf.Search.CaseSensitive)
|
||||||
|
treeNode.Name = rootContent
|
||||||
|
}
|
||||||
|
if 0 < len(treeNode.Children) || 0 < len(treeNode.Blocks) || (-1 < rootPos && "" != keyword) {
|
||||||
|
ret = append(ret, treeNode)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
sort.Slice(ret, func(i, j int) bool {
|
||||||
|
return ret[i].ID > ret[j].ID
|
||||||
|
})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
func getBlockIn(blocks []*Block, id string) *Block {
|
func getBlockIn(blocks []*Block, id string) *Block {
|
||||||
if "" == id {
|
if "" == id {
|
||||||
return nil
|
return nil
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue