siyuan/kernel/sql/av_gallery.go

243 lines
6.9 KiB
Go

package sql
import (
"bytes"
"fmt"
"strings"
"github.com/88250/lute"
"github.com/88250/lute/ast"
"github.com/88250/lute/parse"
"github.com/88250/lute/render"
"github.com/siyuan-note/siyuan/kernel/av"
"github.com/siyuan-note/siyuan/kernel/filesys"
"github.com/siyuan-note/siyuan/kernel/treenode"
"github.com/siyuan-note/siyuan/kernel/util"
)
func RenderAttributeViewGallery(attrView *av.AttributeView, view *av.View, query string) (ret *av.Gallery) {
ret = &av.Gallery{
BaseInstance: &av.BaseInstance{
ID: view.ID,
Icon: view.Icon,
Name: view.Name,
Desc: view.Desc,
HideAttrViewName: view.HideAttrViewName,
Filters: view.Gallery.Filters,
Sorts: view.Gallery.Sorts,
},
CoverFrom: view.Gallery.CoverFrom,
CoverFromAssetKeyID: view.Gallery.CoverFromAssetKeyID,
CardAspectRatio: view.Gallery.CardAspectRatio,
CardSize: view.Gallery.CardSize,
FitImage: view.Gallery.FitImage,
ShowIcon: view.Gallery.ShowIcon,
WrapField: view.Gallery.WrapField,
Fields: []*av.GalleryField{},
Cards: []*av.GalleryCard{},
}
// 组装字段
for _, field := range view.Gallery.CardFields {
key, getErr := attrView.GetKey(field.ID)
if nil != getErr {
// 找不到字段则在视图中删除
removeMissingField(attrView, view, field.ID)
continue
}
ret.Fields = append(ret.Fields, &av.GalleryField{
BaseInstanceField: &av.BaseInstanceField{
ID: key.ID,
Name: key.Name,
Type: key.Type,
Icon: key.Icon,
Hidden: field.Hidden,
Desc: key.Desc,
Options: key.Options,
NumberFormat: key.NumberFormat,
Template: key.Template,
Relation: key.Relation,
Rollup: key.Rollup,
Date: key.Date,
},
})
}
cardsValues := generateAttrViewItems(attrView) // 生成卡片
filterNotFoundAttrViewItems(&cardsValues) // 过滤掉不存在的卡片
// 批量加载绑定块对应的树
var ialIDs []string
for _, card := range ret.Cards {
block := card.GetBlockValue()
if nil != block && !block.IsDetached {
ialIDs = append(ialIDs, card.ID)
}
}
boundTrees := filesys.LoadTrees(ialIDs)
// 生成卡片字段值
for cardID, cardValues := range cardsValues {
var galleryCard av.GalleryCard
for _, field := range ret.Fields {
var fieldValue *av.GalleryFieldValue
for _, keyValues := range cardValues {
if keyValues.Key.ID == field.ID {
fieldValue = &av.GalleryFieldValue{
BaseValue: &av.BaseValue{
ID: keyValues.Values[0].ID,
Value: keyValues.Values[0],
ValueType: field.Type,
},
}
break
}
}
if nil == fieldValue {
fieldValue = &av.GalleryFieldValue{
BaseValue: &av.BaseValue{
ID: ast.NewNodeID(),
ValueType: field.Type,
},
}
}
galleryCard.ID = cardID
fillAttributeViewBaseValue(fieldValue.BaseValue, field.ID, cardID, field.NumberFormat, field.Template)
galleryCard.Values = append(galleryCard.Values, fieldValue)
}
fillAttributeViewGalleryCardCover(attrView, view, cardValues, &galleryCard, cardID, luteEngine, boundTrees)
ret.Cards = append(ret.Cards, &galleryCard)
}
// 批量获取块属性以提升性能
ials := BatchGetBlockAttrsWitTrees(ialIDs, boundTrees)
// 渲染自动生成的字段值,比如关联字段、汇总字段、创建时间字段和更新时间字段
avCache := map[string]*av.AttributeView{}
avCache[attrView.ID] = attrView
for _, card := range ret.Cards {
for _, value := range card.Values {
fillAttributeViewAutoGeneratedValues(attrView, ials, value.Value, card, cardsValues, &avCache)
}
}
// 最后单独渲染模板字段,这样模板字段就可以使用汇总、关联、创建时间和更新时间字段的值了
var renderTemplateErr error
for _, card := range ret.Cards {
for _, value := range card.Values {
err := fillAttributeViewTemplateValue(value.Value, card, attrView, ials, cardsValues)
if nil != err {
renderTemplateErr = err
}
}
}
if nil != renderTemplateErr {
util.PushErrMsg(fmt.Sprintf(util.Langs[util.Lang][44], util.EscapeHTML(renderTemplateErr.Error())), 30000)
}
filterByQuery(query, ret)
manualSort(view.Gallery, ret)
return
}
func fillAttributeViewGalleryCardCover(attrView *av.AttributeView, view *av.View, cardValues []*av.KeyValues, galleryCard *av.GalleryCard, cardID string, luteEngine *lute.Lute, trees map[string]*parse.Tree) {
switch view.Gallery.CoverFrom {
case av.CoverFromNone:
case av.CoverFromContentImage:
blockValue := getBlockValue(cardValues)
if blockValue.IsDetached {
break
}
tree := trees[blockValue.BlockID]
if nil == tree {
break
}
node := treenode.GetNodeInTree(tree, blockValue.BlockID)
if nil == node {
break
}
if ast.NodeDocument == node.Type {
if titleImg := treenode.GetDocTitleImgPath(node); "" != titleImg {
galleryCard.CoverURL = titleImg
break
}
if titleImgCSS := node.IALAttr("title-img"); "" != titleImgCSS {
galleryCard.CoverURL = titleImgCSS
break
}
}
ast.Walk(node, func(n *ast.Node, entering bool) ast.WalkStatus {
if !entering {
return ast.WalkContinue
}
if ast.NodeImage != n.Type {
return ast.WalkContinue
}
dest := n.ChildByType(ast.NodeLinkDest)
if nil == dest {
return ast.WalkContinue
}
galleryCard.CoverURL = dest.TokensStr()
return ast.WalkStop
})
if "" == galleryCard.CoverURL {
isDoc := ast.NodeDocument == node.Type
if isDoc {
node = node.FirstChild
}
buf := bytes.Buffer{}
for c := node; nil != c; c = c.Next {
buf.WriteString(renderBlockDOMByNode(c, luteEngine))
if !isDoc || 1024*4 < buf.Len() {
break
}
}
galleryCard.CoverContent = buf.String()
return
}
case av.CoverFromAssetField:
if "" == view.Gallery.CoverFromAssetKeyID {
break
}
assetValue := attrView.GetValue(view.Gallery.CoverFromAssetKeyID, cardID)
if nil == assetValue || 1 > len(assetValue.MAsset) {
break
}
galleryCard.CoverURL = assetValue.MAsset[0].Content
return
}
}
func renderBlockDOMByNode(node *ast.Node, luteEngine *lute.Lute) string {
tree := &parse.Tree{Root: &ast.Node{Type: ast.NodeDocument}, Context: &parse.Context{ParseOption: luteEngine.ParseOptions}}
blockRenderer := render.NewProtyleRenderer(tree, luteEngine.RenderOptions)
blockRenderer.Options.ProtyleContenteditable = false
ast.Walk(node, func(n *ast.Node, entering bool) ast.WalkStatus {
if entering {
// 内容图中不需要渲染数据库角标 https://github.com/siyuan-note/siyuan/issues/15057
ial := parse.IAL2Map(n.KramdownIAL)
delete(ial, av.NodeAttrNameAvs)
n.KramdownIAL = parse.Map2IAL(ial)
}
rendererFunc := blockRenderer.RendererFuncs[n.Type]
return rendererFunc(n, entering)
})
h := strings.TrimSpace(blockRenderer.Writer.String())
if strings.HasPrefix(h, "<li") {
h = "<ul>" + h + "</ul>"
}
return h
}