🧑‍💻 Improve kernel API /api/block/updateBlock and /api/block/batchUpdateBlock https://github.com/siyuan-note/siyuan/issues/15301

This commit is contained in:
Daniel 2025-07-16 00:05:51 +08:00
parent 70e31c71cc
commit 56a5f13728
No known key found for this signature in database
GPG key ID: 86211BA83DF03017
5 changed files with 66 additions and 140 deletions

View file

@ -25,7 +25,6 @@ import (
"github.com/88250/lute/ast"
"github.com/88250/lute/parse"
"github.com/gin-gonic/gin"
"github.com/siyuan-note/siyuan/kernel/filesys"
"github.com/siyuan-note/siyuan/kernel/model"
"github.com/siyuan-note/siyuan/kernel/treenode"
"github.com/siyuan-note/siyuan/kernel/util"
@ -568,20 +567,21 @@ func updateBlock(c *gin.Context) {
return
}
block, err := model.GetBlock(id, nil)
oldTree, err := model.LoadTreeByBlockID(id)
if err != nil {
ret.Code = -1
ret.Msg = "get block failed: " + err.Error()
ret.Msg = "load tree failed: " + err.Error()
return
}
if "NodeDocument" == block.Type {
oldTree, err := filesys.LoadTree(block.Box, block.Path, luteEngine)
if err != nil {
ret.Code = -1
ret.Msg = "load tree failed: " + err.Error()
return
}
node := treenode.GetNodeInTree(oldTree, id)
if nil == node {
ret.Code = -1
ret.Msg = "block not found [id=" + id + "]"
return
}
if ast.NodeDocument == node.Type {
var toRemoves []*ast.Node
for n := oldTree.Root.FirstChild; nil != n; n = n.Next {
toRemoves = append(toRemoves, n)
@ -600,36 +600,21 @@ func updateBlock(c *gin.Context) {
model.WriteTreeUpsertQueue(oldTree)
model.ReloadProtyle(id)
return
} else {
if ast.NodeListItem == node.Type && ast.NodeList == tree.Root.FirstChild.Type {
// 使用 API `api/block/updateBlock` 更新列表项时渲染错误 https://github.com/siyuan-note/siyuan/issues/4658
tree.Root.AppendChild(tree.Root.FirstChild.FirstChild) // 将列表下的第一个列表项移到文档结尾,移动以后根下面直接挂列表项,渲染器可以正常工作
tree.Root.FirstChild.Unlink() // 删除列表
tree.Root.FirstChild.Unlink() // 继续删除列表 IAL
}
tree.Root.FirstChild.SetIALAttr("id", id)
tree.Root.FirstChild.ID = id
node.InsertBefore(tree.Root.FirstChild)
node.Unlink()
}
var transactions []*model.Transaction
if "NodeListItem" == block.Type && ast.NodeList == tree.Root.FirstChild.Type {
// 使用 API `api/block/updateBlock` 更新列表项时渲染错误 https://github.com/siyuan-note/siyuan/issues/4658
tree.Root.AppendChild(tree.Root.FirstChild.FirstChild) // 将列表下的第一个列表项移到文档结尾,移动以后根下面直接挂列表项,渲染器可以正常工作
tree.Root.FirstChild.Unlink() // 删除列表
tree.Root.FirstChild.Unlink() // 继续删除列表 IAL
}
tree.Root.FirstChild.SetIALAttr("id", id)
data = luteEngine.Tree2BlockDOM(tree, luteEngine.RenderOptions)
transactions = []*model.Transaction{
{
DoOperations: []*model.Operation{
{
Action: "update",
ID: id,
Data: data,
},
},
},
}
model.PerformTransactions(&transactions)
model.FlushTxQueue()
ret.Data = transactions
broadcastTransactions(transactions)
model.WriteTreeUpsertQueue(oldTree)
model.ReloadProtyle(oldTree.ID)
}
func batchUpdateBlock(c *gin.Context) {
@ -642,16 +627,7 @@ func batchUpdateBlock(c *gin.Context) {
}
blocksArg := arg["blocks"].([]interface{})
type updateBlockArg struct {
ID string
Data string
DataType string
Block *model.Block
Tree *parse.Tree
}
var blocks []*updateBlockArg
blocks := map[string]*parse.Tree{}
luteEngine := util.NewLute()
for _, blockArg := range blocksArg {
blockMap := blockArg.(map[string]interface{})
@ -678,37 +654,26 @@ func batchUpdateBlock(c *gin.Context) {
return
}
block, err := model.GetBlock(id, nil)
blocks[id] = tree
}
trees := map[string]*parse.Tree{}
for id, tree := range blocks {
oldTree, err := model.LoadTreeWithCache(id, &trees)
if err != nil {
ret.Code = -1
ret.Msg = "get block failed: " + err.Error()
ret.Msg = "load tree failed: " + err.Error()
return
}
blocks = append(blocks, &updateBlockArg{
ID: id,
Data: data,
DataType: dataType,
Block: block,
Tree: tree,
})
}
node := treenode.GetNodeInTree(oldTree, id)
if nil == node {
ret.Code = -1
ret.Msg = "block not found [id=" + id + "]"
return
}
var ops []*model.Operation
tx := &model.Transaction{}
transactions := []*model.Transaction{tx}
for _, upBlock := range blocks {
block := upBlock.Block
data := upBlock.Data
tree := upBlock.Tree
id := upBlock.ID
if "NodeDocument" == block.Type {
oldTree, err := filesys.LoadTree(block.Box, block.Path, luteEngine)
if err != nil {
ret.Code = -1
ret.Msg = "load tree failed: " + err.Error()
return
}
if ast.NodeDocument == node.Type {
var toRemoves []*ast.Node
for n := oldTree.Root.FirstChild; nil != n; n = n.Next {
toRemoves = append(toRemoves, n)
@ -723,34 +688,23 @@ func batchUpdateBlock(c *gin.Context) {
for _, n := range toAppends {
oldTree.Root.AppendChild(n)
}
model.WriteTreeUpsertQueue(oldTree)
model.ReloadProtyle(id)
} else {
if "NodeListItem" == block.Type && ast.NodeList == tree.Root.FirstChild.Type {
if ast.NodeListItem == node.Type && ast.NodeList == tree.Root.FirstChild.Type {
// 使用 API `api/block/updateBlock` 更新列表项时渲染错误 https://github.com/siyuan-note/siyuan/issues/4658
tree.Root.AppendChild(tree.Root.FirstChild.FirstChild) // 将列表下的第一个列表项移到文档结尾,移动以后根下面直接挂列表项,渲染器可以正常工作
tree.Root.FirstChild.Unlink() // 删除列表
tree.Root.FirstChild.Unlink() // 继续删除列表 IAL
}
tree.Root.FirstChild.SetIALAttr("id", id)
data = luteEngine.Tree2BlockDOM(tree, luteEngine.RenderOptions)
ops = append(ops, &model.Operation{
Action: "update",
ID: id,
Data: data,
})
tree.Root.FirstChild.ID = id
node.InsertBefore(tree.Root.FirstChild)
node.Unlink()
}
}
if 0 < len(ops) {
tx.DoOperations = ops
model.PerformTransactions(&transactions)
model.FlushTxQueue()
ret.Data = transactions
broadcastTransactions(transactions)
for _, tree := range trees {
model.WriteTreeUpsertQueue(tree)
model.ReloadProtyle(tree.ID)
}
}