Something went wrong. Try again.
A privacy-first, self-hosted, fully open source personal knowledge management software, written in typescript and golang. (PERSONAL FORK)
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317package sql
import ( "bytes" "fmt" "net/url" "strings"
"github.com/88250/lute" "github.com/88250/lute/ast" "github.com/88250/lute/html" "github.com/88250/lute/parse" "github.com/88250/lute/render" "github.com/siyuan-note/logging" "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, depth *int, cachedAttrViews map[string]*av.AttributeView) (ret *av.Gallery) { viewable := attrView.RenderedViewables[view.ID] if nil != viewable { ret = viewable.(*av.Gallery) return }
ret = &av.Gallery{ BaseInstance: av.NewViewBaseInstance(view), CoverFrom: view.Gallery.CoverFrom, CoverFromAssetKeyID: view.Gallery.CoverFromAssetKeyID, CardAspectRatio: view.Gallery.CardAspectRatio, CardSize: view.Gallery.CardSize, FitImage: view.Gallery.FitImage, DisplayFieldName: view.Gallery.DisplayFieldName, 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, Wrap: field.Wrap, Hidden: field.Hidden, Desc: key.Desc, Calc: field.Calc, Options: key.Options, NumberFormat: key.NumberFormat, Template: key.Template, Relation: key.Relation, Rollup: key.Rollup, Date: key.Date, Created: key.Created, Updated: key.Updated, }, }) }
cardsValues := generateAttrViewItems(attrView, view) // 生成卡片 filterNotFoundAttrViewItems(cardsValues) // 过滤掉不存在的卡片
// 批量加载绑定块对应的树 var ialIDs []string for _, keyValues := range cardsValues { for _, kValues := range keyValues { blockVal := kValues.GetBlockValue() if nil != blockVal && !blockVal.IsDetached { ialIDs = append(ialIDs, blockVal.Block.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: cardID[:14] + ast.NewNodeID()[14:], ValueType: field.Type, }, } } galleryCard.ID = cardID
filedDateIsTime := false if nil != field.Date { filedDateIsTime = field.Date.FillSpecificTime } fillAttributeViewBaseValue(fieldValue.BaseValue, field.ID, cardID, field.NumberFormat, field.Template, filedDateIsTime) galleryCard.Values = append(galleryCard.Values, fieldValue) }
fillAttributeViewGalleryCardCover(attrView, view, cardValues, &galleryCard, cardID, luteEngine, boundTrees) ret.Cards = append(ret.Cards, &galleryCard) }
// 回填补全数据 fillAttributeViewKeyValues(attrView, ret)
// 批量获取块属性以提升性能 ials := BatchGetBlockAttrsWitTrees(ialIDs, boundTrees)
// 渲染自动生成的字段值，比如关联、汇总、创建时间和更新时间 fillAttributeViewAutoGeneratedValues(attrView, ret, ials, depth, cachedAttrViews)
// 最后渲染模板字段，这样模板就可以使用汇总、关联、创建时间和更新时间的值了 renderTemplateErr := fillAttributeViewTemplateValues(attrView, view, ret, ials) if nil != renderTemplateErr { util.PushErrMsg(fmt.Sprintf(util.Langs[util.Lang][44], util.EscapeHTML(renderTemplateErr.Error())), 30000) }
filterByQuery(query, ret) manualSort(view, 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.Block.ID] if nil == tree { break } node := treenode.GetNodeInTree(tree, blockValue.Block.ID) 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 { galleryCard.CoverContent = renderCoverContentBlock(node, luteEngine) return } case av.CoverFromAssetField: if "" == view.Gallery.CoverFromAssetKeyID { break }
assetValue := attrView.GetValue(view.Gallery.CoverFromAssetKeyID, cardID) if nil == assetValue || 1 > len(assetValue.MAsset) { break }
p := assetValue.MAsset[0].Content if util.IsAssetsImage(p) { galleryCard.CoverURL = p } return case av.CoverFromContentBlock: blockValue := getBlockValue(cardValues) if blockValue.IsDetached { break }
tree := trees[blockValue.Block.ID] if nil == tree { break } node := treenode.GetNodeInTree(tree, blockValue.Block.ID) if nil == node { break } galleryCard.CoverContent = renderCoverContentBlock(node, luteEngine) }}
func renderCoverContentBlock(node *ast.Node, luteEngine *lute.Lute) string { isDoc := ast.NodeDocument == node.Type if isDoc { node = node.FirstChild } heading := node isHeading := ast.NodeHeading == node.Type headingLevel := node.HeadingLevel if isHeading { node = node.Next }
buf := bytes.Buffer{} for c := node; nil != c; c = c.Next { if isHeading && ast.NodeHeading == c.Type && c.HeadingLevel <= headingLevel { if 1 > buf.Len() { buf.WriteString(renderBlockDOMByNode(heading, luteEngine)) } break } buf.WriteString(renderBlockDOMByNode(c, luteEngine)) if (!isDoc && !isHeading) || 1024*4 < buf.Len() { break } } return buf.String()}
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, luteEngine.ParseOptions) blockRenderer.Options.ProtyleContenteditable = false resetIDs := map[string]string{} ast.Walk(node, func(n *ast.Node, entering bool) ast.WalkStatus { if entering { if n.IsBlock() { // 内容图中不需要渲染数据库角标 https://github.com/siyuan-note/siyuan/issues/15057 ial := parse.IAL2Map(n.KramdownIAL) delete(ial, av.NodeAttrNameAvs)
// 重置 data-node-id 的值，避免触发前端绑定的事件 https://github.com/siyuan-note/siyuan/issues/15088 newID := ast.NewNodeID() resetIDs[newID] = n.ID n.ID, ial["id"] = newID, newID n.KramdownIAL = parse.Map2IAL(ial)
if ast.NodeIFrame == n.Type || ast.NodeAudio == n.Type || ast.NodeVideo == n.Type { // 禁止自动播放 Disable automatic video playback in database card view https://github.com/siyuan-note/siyuan/issues/15212 dest := treenode.GetNodeSrcTokens(n) oldDest := dest if (strings.HasPrefix(dest, "http://") || strings.HasPrefix(dest, "https://")) && !strings.Contains(dest, "autoplay") { dest = html.UnescapeHTMLStr(dest) destURL, err := url.Parse(dest) if nil == err { q := destURL.Query() q.Set("autoplay", "0") destURL.RawQuery = q.Encode() dest = destURL.String() dest = html.EscapeHTMLStr(dest) n.Tokens = bytes.ReplaceAll(n.Tokens, []byte(oldDest), []byte(dest)) } else { logging.LogWarnf("parse url [%s] failed: %v", dest, err) } } } } } rendererFunc := blockRenderer.RendererFuncs[n.Type] return rendererFunc(n, entering) }) ret := strings.TrimSpace(blockRenderer.Writer.String()) if strings.HasPrefix(ret, "<li") { ret = "<ul>" + ret + "</ul>" } ast.Walk(node, func(n *ast.Node, entering bool) ast.WalkStatus { if entering { if n.IsBlock() { // 还原上面重置 data-node-id 的值 ial := parse.IAL2Map(n.KramdownIAL) oldID := resetIDs[n.ID] n.ID, ial["id"] = oldID, oldID n.KramdownIAL = parse.Map2IAL(ial) } } return ast.WalkContinue }) return ret}