123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526 |
- // SiYuan - Refactor your thinking
- // Copyright (c) 2020-present, b3log.org
- //
- // This program is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Affero General Public License as published by
- // the Free Software Foundation, either version 3 of the License, or
- // (at your option) any later version.
- //
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Affero General Public License for more details.
- //
- // You should have received a copy of the GNU Affero General Public License
- // along with this program. If not, see <https://www.gnu.org/licenses/>.
- package model
- import (
- "bytes"
- "os"
- "path/filepath"
- "sort"
- "strconv"
- "strings"
- "text/template"
- "time"
- "github.com/88250/gulu"
- "github.com/88250/lute/ast"
- "github.com/88250/lute/parse"
- "github.com/siyuan-note/dejavu/entity"
- "github.com/siyuan-note/filelock"
- "github.com/siyuan-note/logging"
- "github.com/siyuan-note/siyuan/kernel/av"
- "github.com/siyuan-note/siyuan/kernel/sql"
- "github.com/siyuan-note/siyuan/kernel/treenode"
- "github.com/siyuan-note/siyuan/kernel/util"
- )
- func GetAttributeView(avID string) (ret *av.AttributeView) {
- waitForSyncingStorages()
- ret, _ = av.ParseAttributeView(avID)
- return
- }
- type SearchAttributeViewResult struct {
- AvID string `json:"avID"`
- AvName string `json:"avName"`
- BlockID string `json:"blockID"`
- HPath string `json:"hPath"`
- }
- func SearchAttributeView(keyword string, page int, pageSize int) (ret []*SearchAttributeViewResult, pageCount int) {
- waitForSyncingStorages()
- ret = []*SearchAttributeViewResult{}
- var blocks []*Block
- keyword = strings.TrimSpace(keyword)
- if "" == keyword {
- sqlBlocks := sql.SelectBlocksRawStmt("SELECT * FROM blocks WHERE type = 'av' ORDER BY updated DESC LIMIT 10", page, pageSize)
- blocks = fromSQLBlocks(&sqlBlocks, "", 36)
- pageCount = 1
- } else {
- var matchedBlockCount int
- blocks, matchedBlockCount, _ = fullTextSearchByKeyword(keyword, "", "", "('av')", "", 36, page, pageSize)
- pageCount = (matchedBlockCount + pageSize - 1) / pageSize
- }
- trees := map[string]*parse.Tree{}
- for _, block := range blocks {
- tree := trees[block.RootID]
- if nil == tree {
- tree, _ = loadTreeByBlockID(block.ID)
- if nil != tree {
- trees[block.RootID] = tree
- }
- }
- if nil == tree {
- continue
- }
- node := treenode.GetNodeInTree(tree, block.ID)
- if nil == node {
- continue
- }
- if "" == node.AttributeViewID {
- continue
- }
- avID := node.AttributeViewID
- attrView, _ := av.ParseAttributeView(avID)
- if nil == attrView {
- continue
- }
- exist := false
- for _, result := range ret {
- if result.AvID == avID {
- exist = true
- break
- }
- }
- var hPath string
- baseBlock := treenode.GetBlockTreeRootByPath(node.Box, node.Path)
- if nil != baseBlock {
- hPath = baseBlock.HPath
- }
- box := Conf.Box(node.Box)
- if nil != box {
- hPath = box.Name + hPath
- }
- if !exist {
- ret = append(ret, &SearchAttributeViewResult{
- AvID: avID,
- AvName: attrView.Name,
- BlockID: block.ID,
- HPath: hPath,
- })
- }
- }
- return
- }
- type BlockAttributeViewKeys struct {
- AvID string `json:"avID"`
- AvName string `json:"avName"`
- BlockIDs []string `json:"blockIDs"`
- KeyValues []*av.KeyValues `json:"keyValues"`
- }
- func GetBlockAttributeViewKeys(blockID string) (ret []*BlockAttributeViewKeys) {
- waitForSyncingStorages()
- ret = []*BlockAttributeViewKeys{}
- attrs := GetBlockAttrsWithoutWaitWriting(blockID)
- avs := attrs[av.NodeAttrNameAvs]
- if "" == avs {
- return
- }
- avIDs := strings.Split(avs, ",")
- for _, avID := range avIDs {
- attrView, err := av.ParseAttributeView(avID)
- if nil != err {
- logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
- return
- }
- if 1 > len(attrView.Views) {
- err = av.ErrViewNotFound
- return
- }
- var keyValues []*av.KeyValues
- for _, kv := range attrView.KeyValues {
- kValues := &av.KeyValues{Key: kv.Key}
- for _, v := range kv.Values {
- if v.BlockID == blockID {
- kValues.Values = append(kValues.Values, v)
- }
- }
- switch kValues.Key.Type {
- case av.KeyTypeTemplate:
- kValues.Values = append(kValues.Values, &av.Value{ID: ast.NewNodeID(), KeyID: kValues.Key.ID, BlockID: blockID, Type: av.KeyTypeTemplate, Template: &av.ValueTemplate{Content: ""}})
- case av.KeyTypeCreated:
- kValues.Values = append(kValues.Values, &av.Value{ID: ast.NewNodeID(), KeyID: kValues.Key.ID, BlockID: blockID, Type: av.KeyTypeCreated})
- case av.KeyTypeUpdated:
- kValues.Values = append(kValues.Values, &av.Value{ID: ast.NewNodeID(), KeyID: kValues.Key.ID, BlockID: blockID, Type: av.KeyTypeUpdated})
- }
- if 0 < len(kValues.Values) {
- keyValues = append(keyValues, kValues)
- }
- }
- // 渲染自动生成的列值,比如模板列、关联列、汇总列、创建时间列和更新时间列
- // 先处理关联列、汇总列、创建时间列和更新时间列
- for _, kv := range keyValues {
- switch kv.Key.Type {
- case av.KeyTypeRollup:
- if nil == kv.Key.Rollup {
- break
- }
- relKey, _ := attrView.GetKey(kv.Key.Rollup.RelationKeyID)
- if nil == relKey {
- break
- }
- var blockIDs []string
- relVal := attrView.GetValue(kv.Key.Rollup.RelationKeyID, kv.Values[0].BlockID)
- if nil != relVal && nil != relVal.Relation {
- blockIDs = relVal.Relation.BlockIDs
- }
- destAv, _ := av.ParseAttributeView(relKey.Relation.AvID)
- if nil != destAv {
- for _, bID := range blockIDs {
- kv.Values[0].Rollup.Contents = append(kv.Values[0].Rollup.Contents, destAv.GetValue(kv.Key.Rollup.KeyID, bID).String())
- }
- }
- case av.KeyTypeRelation:
- if nil == kv.Key.Relation {
- break
- }
- destAv, _ := av.ParseAttributeView(kv.Key.Relation.AvID)
- if nil == destAv {
- break
- }
- blocks := map[string]string{}
- for _, blockValue := range destAv.GetBlockKeyValues().Values {
- blocks[blockValue.BlockID] = blockValue.Block.Content
- }
- for _, bID := range kv.Values[0].Relation.BlockIDs {
- kv.Values[0].Relation.Contents = append(kv.Values[0].Relation.Contents, blocks[bID])
- }
- case av.KeyTypeCreated:
- createdStr := blockID[:len("20060102150405")]
- created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local)
- if nil == parseErr {
- kv.Values[0].Created = av.NewFormattedValueCreated(created.UnixMilli(), 0, av.CreatedFormatNone)
- kv.Values[0].Created.IsNotEmpty = true
- } else {
- logging.LogWarnf("parse created [%s] failed: %s", createdStr, parseErr)
- kv.Values[0].Created = av.NewFormattedValueCreated(time.Now().UnixMilli(), 0, av.CreatedFormatNone)
- }
- case av.KeyTypeUpdated:
- ial := GetBlockAttrsWithoutWaitWriting(blockID)
- updatedStr := ial["updated"]
- updated, parseErr := time.ParseInLocation("20060102150405", updatedStr, time.Local)
- if nil == parseErr {
- kv.Values[0].Updated = av.NewFormattedValueUpdated(updated.UnixMilli(), 0, av.UpdatedFormatNone)
- kv.Values[0].Updated.IsNotEmpty = true
- } else {
- logging.LogWarnf("parse updated [%s] failed: %s", updatedStr, parseErr)
- kv.Values[0].Updated = av.NewFormattedValueUpdated(time.Now().UnixMilli(), 0, av.UpdatedFormatNone)
- }
- }
- }
- // 再处理模板列
- for _, kv := range keyValues {
- switch kv.Key.Type {
- case av.KeyTypeTemplate:
- if 0 < len(kv.Values) {
- ial := map[string]string{}
- block := getRowBlockValue(keyValues)
- if !block.IsDetached {
- ial = GetBlockAttrsWithoutWaitWriting(blockID)
- }
- kv.Values[0].Template.Content = renderTemplateCol(ial, kv.Key.Template, keyValues)
- }
- }
- }
- // Attribute Panel - Database sort attributes by view column order https://github.com/siyuan-note/siyuan/issues/9319
- view, _ := attrView.GetCurrentView()
- if nil != view {
- sorts := map[string]int{}
- for i, col := range view.Table.Columns {
- sorts[col.ID] = i
- }
- sort.Slice(keyValues, func(i, j int) bool {
- return sorts[keyValues[i].Key.ID] < sorts[keyValues[j].Key.ID]
- })
- }
- blockIDs := av.GetMirrorBlockIDs(avID)
- if 1 > len(blockIDs) {
- // 老数据兼容处理
- avBts := treenode.GetBlockTreesByType("av")
- for _, avBt := range avBts {
- if nil == avBt {
- continue
- }
- tree, _ := loadTreeByBlockID(avBt.ID)
- if nil == tree {
- continue
- }
- node := treenode.GetNodeInTree(tree, avBt.ID)
- if nil == node {
- continue
- }
- if avID == node.AttributeViewID {
- blockIDs = append(blockIDs, avBt.ID)
- }
- }
- if 1 > len(blockIDs) {
- continue
- }
- blockIDs = gulu.Str.RemoveDuplicatedElem(blockIDs)
- for _, blockID := range blockIDs {
- av.UpsertBlockRel(avID, blockID)
- }
- }
- ret = append(ret, &BlockAttributeViewKeys{
- AvID: avID,
- AvName: attrView.Name,
- BlockIDs: blockIDs,
- KeyValues: keyValues,
- })
- }
- return
- }
- func RenderRepoSnapshotAttributeView(indexID, avID string) (viewable av.Viewable, attrView *av.AttributeView, err error) {
- repo, err := newRepository()
- if nil != err {
- return
- }
- index, err := repo.GetIndex(indexID)
- if nil != err {
- return
- }
- files, err := repo.GetFiles(index)
- if nil != err {
- return
- }
- var avFile *entity.File
- for _, f := range files {
- if "/storage/av/"+avID+".json" == f.Path {
- avFile = f
- break
- }
- }
- if nil == avFile {
- attrView = av.NewAttributeView(avID)
- } else {
- data, readErr := repo.OpenFile(avFile)
- if nil != readErr {
- logging.LogErrorf("read attribute view [%s] failed: %s", avID, readErr)
- return
- }
- attrView = &av.AttributeView{}
- if err = gulu.JSON.UnmarshalJSON(data, attrView); nil != err {
- logging.LogErrorf("unmarshal attribute view [%s] failed: %s", avID, err)
- return
- }
- }
- viewable, err = renderAttributeView(attrView, "", 1, -1)
- return
- }
- func RenderHistoryAttributeView(avID, created string) (viewable av.Viewable, attrView *av.AttributeView, err error) {
- createdUnix, parseErr := strconv.ParseInt(created, 10, 64)
- if nil != parseErr {
- logging.LogErrorf("parse created [%s] failed: %s", created, parseErr)
- return
- }
- dirPrefix := time.Unix(createdUnix, 0).Format("2006-01-02-150405")
- globPath := filepath.Join(util.HistoryDir, dirPrefix+"*")
- matches, err := filepath.Glob(globPath)
- if nil != err {
- logging.LogErrorf("glob [%s] failed: %s", globPath, err)
- return
- }
- if 1 > len(matches) {
- return
- }
- historyDir := matches[0]
- avJSONPath := filepath.Join(historyDir, "storage", "av", avID+".json")
- if !gulu.File.IsExist(avJSONPath) {
- avJSONPath = filepath.Join(util.DataDir, "storage", "av", avID+".json")
- }
- if !gulu.File.IsExist(avJSONPath) {
- attrView = av.NewAttributeView(avID)
- } else {
- data, readErr := os.ReadFile(avJSONPath)
- if nil != readErr {
- logging.LogErrorf("read attribute view [%s] failed: %s", avID, readErr)
- return
- }
- attrView = &av.AttributeView{}
- if err = gulu.JSON.UnmarshalJSON(data, attrView); nil != err {
- logging.LogErrorf("unmarshal attribute view [%s] failed: %s", avID, err)
- return
- }
- }
- viewable, err = renderAttributeView(attrView, "", 1, -1)
- return
- }
- func RenderAttributeView(avID, viewID string, page, pageSize int) (viewable av.Viewable, attrView *av.AttributeView, err error) {
- waitForSyncingStorages()
- if avJSONPath := av.GetAttributeViewDataPath(avID); !filelock.IsExist(avJSONPath) {
- attrView = av.NewAttributeView(avID)
- if err = av.SaveAttributeView(attrView); nil != err {
- logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
- return
- }
- }
- attrView, err = av.ParseAttributeView(avID)
- if nil != err {
- logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
- return
- }
- viewable, err = renderAttributeView(attrView, viewID, page, pageSize)
- return
- }
- func renderAttributeView(attrView *av.AttributeView, viewID string, page, pageSize int) (viewable av.Viewable, err error) {
- if 1 > len(attrView.Views) {
- view, _ := av.NewTableViewWithBlockKey(ast.NewNodeID())
- attrView.Views = append(attrView.Views, view)
- attrView.ViewID = view.ID
- if err = av.SaveAttributeView(attrView); nil != err {
- logging.LogErrorf("save attribute view [%s] failed: %s", attrView.ID, err)
- return
- }
- }
- var view *av.View
- if "" != viewID {
- view = attrView.GetView(viewID)
- if nil != view && viewID != attrView.ViewID {
- attrView.ViewID = viewID
- if err = av.SaveAttributeView(attrView); nil != err {
- logging.LogErrorf("save attribute view [%s] failed: %s", attrView.ID, err)
- return
- }
- }
- } else {
- if "" != attrView.ViewID {
- view, _ = attrView.GetCurrentView()
- }
- }
- if nil == view {
- view = attrView.Views[0]
- }
- // 做一些数据兼容和订正处理,保存的时候也会做 av.SaveAttributeView()
- currentTimeMillis := util.CurrentTimeMillis()
- for _, kv := range attrView.KeyValues {
- switch kv.Key.Type {
- case av.KeyTypeBlock: // 补全 block 的创建时间和更新时间
- for _, v := range kv.Values {
- if 0 == v.Block.Created {
- if "" == v.Block.ID {
- v.Block.ID = v.BlockID
- if "" == v.Block.ID {
- v.Block.ID = ast.NewNodeID()
- v.BlockID = v.Block.ID
- }
- }
- createdStr := v.Block.ID[:len("20060102150405")]
- created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local)
- if nil == parseErr {
- v.Block.Created = created.UnixMilli()
- } else {
- v.Block.Created = currentTimeMillis
- }
- }
- if 0 == v.Block.Updated {
- v.Block.Updated = currentTimeMillis
- }
- }
- }
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- // 列删除以后需要删除设置的过滤和排序
- tmpFilters := []*av.ViewFilter{}
- for _, f := range view.Table.Filters {
- if k, _ := attrView.GetKey(f.Column); nil != k {
- tmpFilters = append(tmpFilters, f)
- }
- }
- view.Table.Filters = tmpFilters
- tmpSorts := []*av.ViewSort{}
- for _, s := range view.Table.Sorts {
- if k, _ := attrView.GetKey(s.Column); nil != k {
- tmpSorts = append(tmpSorts, s)
- }
- }
- view.Table.Sorts = tmpSorts
- viewable, err = renderAttributeViewTable(attrView, view)
- }
- viewable.FilterRows()
- viewable.SortRows()
- viewable.CalcCols()
- // 分页
- switch viewable.GetType() {
- case av.LayoutTypeTable:
- table := viewable.(*av.Table)
- table.RowCount = len(table.Rows)
- if 1 > view.Table.PageSize {
- view.Table.PageSize = 50
- }
- table.PageSize = view.Table.PageSize
- if 1 > pageSize {
- pageSize = table.PageSize
- }
- start := (page - 1) * pageSize
- end := start + pageSize
- if len(table.Rows) < end {
- end = len(table.Rows)
- }
- table.Rows = table.Rows[start:end]
- }
- return
- }
- func renderTemplateCol(ial map[string]string, tplContent string, rowValues []*av.KeyValues) string {
- if "" == ial["id"] {
- block := getRowBlockValue(rowValues)
- ial["id"] = block.Block.ID
- }
- if "" == ial["updated"] {
- block := getRowBlockValue(rowValues)
- ial["updated"] = time.UnixMilli(block.Block.Updated).Format("20060102150405")
- }
- goTpl := template.New("").Delims(".action{", "}")
- goTpl = goTpl.Funcs(util.BuiltInTemplateFuncs())
- tpl, tplErr := goTpl.Parse(tplContent)
- if nil != tplErr {
- logging.LogWarnf("parse template [%s] failed: %s", tplContent, tplErr)
- return ""
- }
- buf := &bytes.Buffer{}
- dataModel := map[string]interface{}{} // 复制一份 IAL 以避免修改原始数据
- for k, v := range ial {
- dataModel[k] = v
- // Database template column supports `created` and `updated` built-in variables https://github.com/siyuan-note/siyuan/issues/9364
- createdStr := ial["id"]
- if "" != createdStr {
- createdStr = createdStr[:len("20060102150405")]
- }
- created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local)
- if nil == parseErr {
- dataModel["created"] = created
- } else {
- logging.LogWarnf("parse created [%s] failed: %s", createdStr, parseErr)
- dataModel["created"] = time.Now()
- }
- updatedStr := ial["updated"]
- updated, parseErr := time.ParseInLocation("20060102150405", updatedStr, time.Local)
- if nil == parseErr {
- dataModel["updated"] = updated
- } else {
- dataModel["updated"] = time.Now()
- }
- }
- for _, rowValue := range rowValues {
- if 0 < len(rowValue.Values) {
- v := rowValue.Values[0]
- if av.KeyTypeNumber == v.Type {
- dataModel[rowValue.Key.Name] = v.Number.Content
- } else if av.KeyTypeDate == v.Type {
- dataModel[rowValue.Key.Name] = time.UnixMilli(v.Date.Content)
- } else {
- dataModel[rowValue.Key.Name] = v.String()
- }
- }
- }
- if err := tpl.Execute(buf, dataModel); nil != err {
- logging.LogWarnf("execute template [%s] failed: %s", tplContent, err)
- }
- return buf.String()
- }
- func renderAttributeViewTable(attrView *av.AttributeView, view *av.View) (ret *av.Table, err error) {
- ret = &av.Table{
- ID: view.ID,
- Icon: view.Icon,
- Name: view.Name,
- Columns: []*av.TableColumn{},
- Rows: []*av.TableRow{},
- Filters: view.Table.Filters,
- Sorts: view.Table.Sorts,
- }
- // 组装列
- for _, col := range view.Table.Columns {
- key, getErr := attrView.GetKey(col.ID)
- if nil != getErr {
- err = getErr
- return
- }
- ret.Columns = append(ret.Columns, &av.TableColumn{
- ID: key.ID,
- Name: key.Name,
- Type: key.Type,
- Icon: key.Icon,
- Options: key.Options,
- NumberFormat: key.NumberFormat,
- Template: key.Template,
- Relation: key.Relation,
- Rollup: key.Rollup,
- Wrap: col.Wrap,
- Hidden: col.Hidden,
- Width: col.Width,
- Pin: col.Pin,
- Calc: col.Calc,
- })
- }
- // 生成行
- rows := map[string][]*av.KeyValues{}
- for _, keyValues := range attrView.KeyValues {
- for _, val := range keyValues.Values {
- values := rows[val.BlockID]
- if nil == values {
- values = []*av.KeyValues{{Key: keyValues.Key, Values: []*av.Value{val}}}
- } else {
- values = append(values, &av.KeyValues{Key: keyValues.Key, Values: []*av.Value{val}})
- }
- rows[val.BlockID] = values
- }
- }
- // 过滤掉不存在的行
- var notFound []string
- for blockID, keyValues := range rows {
- blockValue := getRowBlockValue(keyValues)
- if nil == blockValue {
- notFound = append(notFound, blockID)
- continue
- }
- if blockValue.IsDetached {
- continue
- }
- if nil != blockValue.Block && "" == blockValue.Block.ID {
- notFound = append(notFound, blockID)
- continue
- }
- if treenode.GetBlockTree(blockID) == nil {
- notFound = append(notFound, blockID)
- }
- }
- for _, blockID := range notFound {
- delete(rows, blockID)
- }
- // 生成行单元格
- for rowID, row := range rows {
- var tableRow av.TableRow
- for _, col := range ret.Columns {
- var tableCell *av.TableCell
- for _, keyValues := range row {
- if keyValues.Key.ID == col.ID {
- tableCell = &av.TableCell{
- ID: keyValues.Values[0].ID,
- Value: keyValues.Values[0],
- ValueType: col.Type,
- }
- break
- }
- }
- if nil == tableCell {
- tableCell = &av.TableCell{
- ID: ast.NewNodeID(),
- ValueType: col.Type,
- }
- }
- tableRow.ID = rowID
- switch tableCell.ValueType {
- case av.KeyTypeNumber: // 格式化数字
- if nil != tableCell.Value && nil != tableCell.Value.Number && tableCell.Value.Number.IsNotEmpty {
- tableCell.Value.Number.Format = col.NumberFormat
- tableCell.Value.Number.FormatNumber()
- }
- case av.KeyTypeTemplate: // 渲染模板列
- tableCell.Value = &av.Value{ID: tableCell.ID, KeyID: col.ID, BlockID: rowID, Type: av.KeyTypeTemplate, Template: &av.ValueTemplate{Content: col.Template}}
- case av.KeyTypeCreated: // 填充创建时间列值,后面再渲染
- tableCell.Value = &av.Value{ID: tableCell.ID, KeyID: col.ID, BlockID: rowID, Type: av.KeyTypeCreated}
- case av.KeyTypeUpdated: // 填充更新时间列值,后面再渲染
- tableCell.Value = &av.Value{ID: tableCell.ID, KeyID: col.ID, BlockID: rowID, Type: av.KeyTypeUpdated}
- }
- treenode.FillAttributeViewTableCellNilValue(tableCell, rowID, col.ID)
- tableRow.Cells = append(tableRow.Cells, tableCell)
- }
- ret.Rows = append(ret.Rows, &tableRow)
- }
- // 渲染自动生成的列值,比如模板列、关联列、汇总列、创建时间列和更新时间列
- for _, row := range ret.Rows {
- for _, cell := range row.Cells {
- switch cell.ValueType {
- case av.KeyTypeTemplate: // 渲染模板列
- keyValues := rows[row.ID]
- ial := map[string]string{}
- block := row.GetBlockValue()
- if nil != block && !block.IsDetached {
- ial = GetBlockAttrsWithoutWaitWriting(row.ID)
- }
- content := renderTemplateCol(ial, cell.Value.Template.Content, keyValues)
- cell.Value.Template.Content = content
- case av.KeyTypeRollup: // 渲染汇总列
- rollupKey, _ := attrView.GetKey(cell.Value.KeyID)
- if nil == rollupKey || nil == rollupKey.Rollup {
- break
- }
- relKey, _ := attrView.GetKey(rollupKey.Rollup.RelationKeyID)
- if nil == relKey || nil == relKey.Relation {
- break
- }
- relVal := attrView.GetValue(relKey.ID, row.ID)
- if nil == relVal || nil == relVal.Relation {
- break
- }
- destAv, _ := av.ParseAttributeView(relKey.Relation.AvID)
- if nil == destAv {
- break
- }
- for _, blockID := range relVal.Relation.BlockIDs {
- cell.Value.Rollup.Contents = append(cell.Value.Rollup.Contents, destAv.GetValue(rollupKey.Rollup.KeyID, blockID).String())
- }
- case av.KeyTypeRelation: // 渲染关联列
- relKey, _ := attrView.GetKey(cell.Value.KeyID)
- if nil != relKey && nil != relKey.Relation {
- destAv, _ := av.ParseAttributeView(relKey.Relation.AvID)
- if nil != destAv {
- blocks := map[string]string{}
- for _, blockValue := range destAv.GetBlockKeyValues().Values {
- blocks[blockValue.BlockID] = blockValue.Block.Content
- }
- for _, blockID := range cell.Value.Relation.BlockIDs {
- cell.Value.Relation.Contents = append(cell.Value.Relation.Contents, blocks[blockID])
- }
- }
- }
- case av.KeyTypeCreated: // 渲染创建时间
- createdStr := row.ID[:len("20060102150405")]
- created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local)
- if nil == parseErr {
- cell.Value.Created = av.NewFormattedValueCreated(created.UnixMilli(), 0, av.CreatedFormatNone)
- cell.Value.Created.IsNotEmpty = true
- } else {
- cell.Value.Created = av.NewFormattedValueCreated(time.Now().UnixMilli(), 0, av.CreatedFormatNone)
- }
- case av.KeyTypeUpdated: // 渲染更新时间
- ial := map[string]string{}
- block := row.GetBlockValue()
- if nil != block && !block.IsDetached {
- ial = GetBlockAttrsWithoutWaitWriting(row.ID)
- }
- updatedStr := ial["updated"]
- if "" == updatedStr && nil != block {
- cell.Value.Updated = av.NewFormattedValueUpdated(block.Block.Updated, 0, av.UpdatedFormatNone)
- cell.Value.Updated.IsNotEmpty = true
- } else {
- updated, parseErr := time.ParseInLocation("20060102150405", updatedStr, time.Local)
- if nil == parseErr {
- cell.Value.Updated = av.NewFormattedValueUpdated(updated.UnixMilli(), 0, av.UpdatedFormatNone)
- cell.Value.Updated.IsNotEmpty = true
- } else {
- cell.Value.Updated = av.NewFormattedValueUpdated(time.Now().UnixMilli(), 0, av.UpdatedFormatNone)
- }
- }
- }
- }
- }
- // 自定义排序
- sortRowIDs := map[string]int{}
- if 0 < len(view.Table.RowIDs) {
- for i, rowID := range view.Table.RowIDs {
- sortRowIDs[rowID] = i
- }
- }
- sort.Slice(ret.Rows, func(i, j int) bool {
- iv := sortRowIDs[ret.Rows[i].ID]
- jv := sortRowIDs[ret.Rows[j].ID]
- if iv == jv {
- return ret.Rows[i].ID < ret.Rows[j].ID
- }
- return iv < jv
- })
- return
- }
- func getRowBlockValue(keyValues []*av.KeyValues) (ret *av.Value) {
- for _, kv := range keyValues {
- if av.KeyTypeBlock == kv.Key.Type && 0 < len(kv.Values) {
- ret = kv.Values[0]
- break
- }
- }
- return
- }
- func (tx *Transaction) doUpdateAttrViewColRollup(operation *Operation) (ret *TxErr) {
- err := updateAttributeViewColRollup(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func updateAttributeViewColRollup(operation *Operation) (err error) {
- // operation.AvID 汇总列所在 av
- // operation.ID 汇总列 ID
- // operation.ParentID 汇总列基于的关联列 ID
- // operation.KeyID 目标列 ID
- // operation.Data 计算方式
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- rollUpKey, _ := attrView.GetKey(operation.ID)
- if nil == rollUpKey {
- return
- }
- rollUpKey.Rollup = &av.Rollup{
- RelationKeyID: operation.ParentID,
- KeyID: operation.KeyID,
- }
- if "" != operation.Data {
- if err = gulu.JSON.UnmarshalJSON([]byte(operation.Data.(string)), &rollUpKey.Rollup.Calc); nil != err {
- return
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doUpdateAttrViewColRelation(operation *Operation) (ret *TxErr) {
- err := updateAttributeViewColRelation(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func updateAttributeViewColRelation(operation *Operation) (err error) {
- // operation.AvID 源 avID
- // operation.ID 目标 avID
- // operation.KeyID 源 av 关联列 ID
- // operation.IsTwoWay 是否双向关联
- // operation.BackRelationKeyID 双向关联的目标关联列 ID
- // operation.Name 双向关联的目标关联列名称
- // operation.Format 源 av 关联列名称
- srcAv, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- destAv, err := av.ParseAttributeView(operation.ID)
- if nil != err {
- return
- }
- isSameAv := srcAv.ID == destAv.ID
- if isSameAv {
- destAv = srcAv
- }
- for _, keyValues := range srcAv.KeyValues {
- if keyValues.Key.ID != operation.KeyID {
- continue
- }
- srcRel := keyValues.Key.Relation
- // 已经设置过双向关联的话需要先断开双向关联
- if nil != srcRel && srcRel.IsTwoWay {
- oldDestAv, _ := av.ParseAttributeView(srcRel.AvID)
- if nil != oldDestAv {
- isOldSameAv := oldDestAv.ID == destAv.ID
- if isOldSameAv {
- oldDestAv = destAv
- }
- oldDestKey, _ := oldDestAv.GetKey(srcRel.BackKeyID)
- if nil != oldDestKey && nil != oldDestKey.Relation && oldDestKey.Relation.AvID == srcAv.ID && oldDestKey.Relation.IsTwoWay {
- oldDestKey.Relation.IsTwoWay = false
- oldDestKey.Relation.BackKeyID = ""
- }
- if !isOldSameAv {
- err = av.SaveAttributeView(oldDestAv)
- if nil != err {
- return
- }
- }
- }
- }
- srcRel = &av.Relation{
- AvID: operation.ID,
- IsTwoWay: operation.IsTwoWay,
- }
- if operation.IsTwoWay {
- srcRel.BackKeyID = operation.BackRelationKeyID
- } else {
- srcRel.BackKeyID = ""
- }
- keyValues.Key.Relation = srcRel
- keyValues.Key.Name = operation.Format
- break
- }
- destAdded := false
- backRelKey, _ := destAv.GetKey(operation.BackRelationKeyID)
- if nil != backRelKey {
- backRelKey.Relation = &av.Relation{
- AvID: operation.AvID,
- IsTwoWay: operation.IsTwoWay,
- BackKeyID: operation.KeyID,
- }
- destAdded = true
- if operation.IsTwoWay {
- name := strings.TrimSpace(operation.Name)
- if "" == name {
- name = srcAv.Name + " " + operation.Format
- }
- backRelKey.Name = strings.TrimSpace(name)
- }
- }
- if !destAdded {
- if operation.IsTwoWay {
- name := strings.TrimSpace(operation.Name)
- if "" == name {
- name = srcAv.Name + " " + operation.Format
- }
- destAv.KeyValues = append(destAv.KeyValues, &av.KeyValues{
- Key: &av.Key{
- ID: operation.BackRelationKeyID,
- Name: name,
- Type: av.KeyTypeRelation,
- Relation: &av.Relation{AvID: operation.AvID, IsTwoWay: operation.IsTwoWay, BackKeyID: operation.KeyID},
- },
- })
- for _, v := range destAv.Views {
- switch v.LayoutType {
- case av.LayoutTypeTable:
- v.Table.Columns = append(v.Table.Columns, &av.ViewTableColumn{ID: operation.BackRelationKeyID})
- }
- }
- }
- }
- err = av.SaveAttributeView(srcAv)
- if nil != err {
- return
- }
- if !isSameAv {
- err = av.SaveAttributeView(destAv)
- util.BroadcastByType("protyle", "refreshAttributeView", 0, "", map[string]interface{}{"id": destAv.ID})
- }
- return
- }
- func (tx *Transaction) doSortAttrViewView(operation *Operation) (ret *TxErr) {
- avID := operation.AvID
- attrView, err := av.ParseAttributeView(avID)
- if nil != err {
- logging.LogErrorf("parse attribute view [%s] failed: %s", operation.AvID, err)
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- viewID := operation.ID
- previewViewID := operation.PreviousID
- if viewID == previewViewID {
- return
- }
- var view *av.View
- var index, previousIndex int
- for i, v := range attrView.Views {
- if v.ID == viewID {
- view = v
- index = i
- break
- }
- }
- if nil == view {
- return
- }
- attrView.Views = append(attrView.Views[:index], attrView.Views[index+1:]...)
- for i, v := range attrView.Views {
- if v.ID == previewViewID {
- previousIndex = i + 1
- break
- }
- }
- attrView.Views = util.InsertElem(attrView.Views, previousIndex, view)
- if err = av.SaveAttributeView(attrView); nil != err {
- logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrCodeWriteTree, msg: err.Error(), id: avID}
- }
- return
- }
- func (tx *Transaction) doRemoveAttrViewView(operation *Operation) (ret *TxErr) {
- var err error
- avID := operation.AvID
- attrView, err := av.ParseAttributeView(avID)
- if nil != err {
- logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrCodeBlockNotFound, id: avID}
- }
- if 1 >= len(attrView.Views) {
- logging.LogWarnf("can't remove last view [%s] of attribute view [%s]", operation.ID, avID)
- return
- }
- viewID := operation.ID
- var index int
- for i, view := range attrView.Views {
- if viewID == view.ID {
- attrView.Views = append(attrView.Views[:i], attrView.Views[i+1:]...)
- index = i - 1
- break
- }
- }
- if 0 > index {
- index = 0
- }
- attrView.ViewID = attrView.Views[index].ID
- if err = av.SaveAttributeView(attrView); nil != err {
- logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrCodeWriteTree, msg: err.Error(), id: avID}
- }
- return
- }
- func (tx *Transaction) doDuplicateAttrViewView(operation *Operation) (ret *TxErr) {
- var err error
- avID := operation.AvID
- attrView, err := av.ParseAttributeView(avID)
- if nil != err {
- logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrWriteAttributeView, id: avID}
- }
- masterView := attrView.GetView(operation.PreviousID)
- if nil == masterView {
- logging.LogErrorf("get master view failed: %s", avID)
- return &TxErr{code: TxErrWriteAttributeView, id: avID}
- }
- view := av.NewTableView()
- view.ID = operation.ID
- attrView.Views = append(attrView.Views, view)
- attrView.ViewID = view.ID
- view.Icon = masterView.Icon
- view.Name = attrView.GetDuplicateViewName(masterView.Name)
- view.LayoutType = masterView.LayoutType
- for _, col := range masterView.Table.Columns {
- view.Table.Columns = append(view.Table.Columns, &av.ViewTableColumn{
- ID: col.ID,
- Wrap: col.Wrap,
- Hidden: col.Hidden,
- Pin: col.Pin,
- Width: col.Width,
- Calc: col.Calc,
- })
- }
- for _, filter := range masterView.Table.Filters {
- view.Table.Filters = append(view.Table.Filters, &av.ViewFilter{
- Column: filter.Column,
- Operator: filter.Operator,
- Value: filter.Value,
- })
- }
- for _, s := range masterView.Table.Sorts {
- view.Table.Sorts = append(view.Table.Sorts, &av.ViewSort{
- Column: s.Column,
- Order: s.Order,
- })
- }
- view.Table.PageSize = masterView.Table.PageSize
- view.Table.RowIDs = masterView.Table.RowIDs
- if err = av.SaveAttributeView(attrView); nil != err {
- logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrWriteAttributeView, msg: err.Error(), id: avID}
- }
- return
- }
- func (tx *Transaction) doAddAttrViewView(operation *Operation) (ret *TxErr) {
- var err error
- avID := operation.AvID
- attrView, err := av.ParseAttributeView(avID)
- if nil != err {
- logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrWriteAttributeView, id: avID}
- }
- firstView := attrView.Views[0]
- if nil == firstView {
- logging.LogErrorf("get first view failed: %s", avID)
- return &TxErr{code: TxErrWriteAttributeView, id: avID}
- }
- view := av.NewTableView()
- view.ID = operation.ID
- attrView.Views = append(attrView.Views, view)
- attrView.ViewID = view.ID
- for _, col := range firstView.Table.Columns {
- view.Table.Columns = append(view.Table.Columns, &av.ViewTableColumn{ID: col.ID})
- }
- view.Table.RowIDs = firstView.Table.RowIDs
- if err = av.SaveAttributeView(attrView); nil != err {
- logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrWriteAttributeView, msg: err.Error(), id: avID}
- }
- return
- }
- func (tx *Transaction) doSetAttrViewViewName(operation *Operation) (ret *TxErr) {
- var err error
- avID := operation.AvID
- attrView, err := av.ParseAttributeView(avID)
- if nil != err {
- logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrWriteAttributeView, id: avID}
- }
- viewID := operation.ID
- view := attrView.GetView(viewID)
- if nil == view {
- logging.LogErrorf("get view [%s] failed: %s", viewID, err)
- return &TxErr{code: TxErrWriteAttributeView, id: viewID}
- }
- view.Name = strings.TrimSpace(operation.Data.(string))
- if err = av.SaveAttributeView(attrView); nil != err {
- logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrWriteAttributeView, msg: err.Error(), id: avID}
- }
- return
- }
- func (tx *Transaction) doSetAttrViewViewIcon(operation *Operation) (ret *TxErr) {
- var err error
- avID := operation.AvID
- attrView, err := av.ParseAttributeView(avID)
- if nil != err {
- logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrWriteAttributeView, id: avID}
- }
- viewID := operation.ID
- view := attrView.GetView(viewID)
- if nil == view {
- logging.LogErrorf("get view [%s] failed: %s", viewID, err)
- return &TxErr{code: TxErrWriteAttributeView, id: viewID}
- }
- view.Icon = operation.Data.(string)
- if err = av.SaveAttributeView(attrView); nil != err {
- logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
- return &TxErr{code: TxErrWriteAttributeView, msg: err.Error(), id: avID}
- }
- return
- }
- func (tx *Transaction) doSetAttrViewName(operation *Operation) (ret *TxErr) {
- err := setAttributeViewName(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewName(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.ID)
- if nil != err {
- return
- }
- attrView.Name = strings.TrimSpace(operation.Data.(string))
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSetAttrViewFilters(operation *Operation) (ret *TxErr) {
- err := setAttributeViewFilters(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewFilters(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- operationData := operation.Data.([]interface{})
- data, err := gulu.JSON.MarshalJSON(operationData)
- if nil != err {
- return
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- if err = gulu.JSON.UnmarshalJSON(data, &view.Table.Filters); nil != err {
- return
- }
- }
- for _, filter := range view.Table.Filters {
- var key *av.Key
- key, err = attrView.GetKey(filter.Column)
- if nil != err {
- return
- }
- filter.Value.Type = key.Type
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSetAttrViewSorts(operation *Operation) (ret *TxErr) {
- err := setAttributeViewSorts(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewSorts(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- operationData := operation.Data.([]interface{})
- data, err := gulu.JSON.MarshalJSON(operationData)
- if nil != err {
- return
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- if err = gulu.JSON.UnmarshalJSON(data, &view.Table.Sorts); nil != err {
- return
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSetAttrViewPageSize(operation *Operation) (ret *TxErr) {
- err := setAttributeViewPageSize(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewPageSize(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- view.Table.PageSize = int(operation.Data.(float64))
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSetAttrViewColCalc(operation *Operation) (ret *TxErr) {
- err := setAttributeViewColumnCalc(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewColumnCalc(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- operationData := operation.Data.(interface{})
- data, err := gulu.JSON.MarshalJSON(operationData)
- if nil != err {
- return
- }
- calc := &av.ColumnCalc{}
- switch view.LayoutType {
- case av.LayoutTypeTable:
- if err = gulu.JSON.UnmarshalJSON(data, calc); nil != err {
- return
- }
- for _, column := range view.Table.Columns {
- if column.ID == operation.ID {
- column.Calc = calc
- break
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doInsertAttrViewBlock(operation *Operation) (ret *TxErr) {
- for _, id := range operation.SrcIDs {
- tree, err := tx.loadTree(id)
- if nil != err && !operation.IsDetached {
- logging.LogErrorf("load tree [%s] failed: %s", id, err)
- return &TxErr{code: TxErrCodeBlockNotFound, id: id, msg: err.Error()}
- }
- var avErr error
- if avErr = addAttributeViewBlock(id, operation, tree, tx); nil != avErr {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: avErr.Error()}
- }
- }
- return
- }
- func addAttributeViewBlock(blockID string, operation *Operation, tree *parse.Tree, tx *Transaction) (err error) {
- var node *ast.Node
- if !operation.IsDetached {
- node = treenode.GetNodeInTree(tree, blockID)
- if nil == node {
- err = ErrBlockNotFound
- return
- }
- if ast.NodeAttributeView == node.Type {
- // 不能将一个属性视图拖拽到另一个属性视图中
- return
- }
- } else {
- if "" == blockID {
- blockID = ast.NewNodeID()
- logging.LogWarnf("detached block id is empty, generate a new one [%s]", blockID)
- }
- }
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- // 不允许重复添加相同的块到属性视图中
- blockValues := attrView.GetBlockKeyValues()
- for _, blockValue := range blockValues.Values {
- if blockValue.Block.ID == blockID {
- return
- }
- }
- var content string
- if !operation.IsDetached {
- content = getNodeRefText(node)
- }
- now := time.Now().UnixMilli()
- blockValue := &av.Value{ID: ast.NewNodeID(), KeyID: blockValues.Key.ID, BlockID: blockID, Type: av.KeyTypeBlock, IsDetached: operation.IsDetached, IsInitialized: false, Block: &av.ValueBlock{ID: blockID, Content: content, Created: now, Updated: now}}
- blockValues.Values = append(blockValues.Values, blockValue)
- // 如果存在过滤条件,则将过滤条件应用到新添加的块上
- view, _ := attrView.GetCurrentView()
- if nil != view && 0 < len(view.Table.Filters) {
- viewable, _ := renderAttributeViewTable(attrView, view)
- viewable.FilterRows()
- viewable.SortRows()
- if 0 < len(viewable.Rows) {
- row := GetLastSortRow(viewable.Rows)
- if nil != row {
- for _, filter := range view.Table.Filters {
- for _, cell := range row.Cells {
- if nil != cell.Value && cell.Value.KeyID == filter.Column {
- if av.KeyTypeBlock == cell.ValueType {
- blockValue.Block.Content = cell.Value.Block.Content
- continue
- }
- newValue := cell.Value.Clone()
- newValue.ID = ast.NewNodeID()
- newValue.BlockID = blockID
- newValue.IsDetached = operation.IsDetached
- newValue.IsInitialized = false
- values, _ := attrView.GetKeyValues(filter.Column)
- values.Values = append(values.Values, newValue)
- }
- }
- }
- }
- }
- }
- if !operation.IsDetached {
- attrs := parse.IAL2Map(node.KramdownIAL)
- if "" == attrs[av.NodeAttrNameAvs] {
- attrs[av.NodeAttrNameAvs] = operation.AvID
- } else {
- avIDs := strings.Split(attrs[av.NodeAttrNameAvs], ",")
- avIDs = append(avIDs, operation.AvID)
- avIDs = gulu.Str.RemoveDuplicatedElem(avIDs)
- attrs[av.NodeAttrNameAvs] = strings.Join(avIDs, ",")
- }
- if err = setNodeAttrsWithTx(tx, node, tree, attrs); nil != err {
- return
- }
- }
- for _, view := range attrView.Views {
- switch view.LayoutType {
- case av.LayoutTypeTable:
- if "" != operation.PreviousID {
- changed := false
- for i, id := range view.Table.RowIDs {
- if id == operation.PreviousID {
- view.Table.RowIDs = append(view.Table.RowIDs[:i+1], append([]string{blockID}, view.Table.RowIDs[i+1:]...)...)
- changed = true
- break
- }
- }
- if !changed {
- view.Table.RowIDs = append(view.Table.RowIDs, blockID)
- }
- } else {
- view.Table.RowIDs = append([]string{blockID}, view.Table.RowIDs...)
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func GetLastSortRow(rows []*av.TableRow) *av.TableRow {
- for i := len(rows) - 1; i >= 0; i-- {
- row := rows[i]
- block := row.GetBlockValue()
- if nil != block && !block.NotAffectFilter() {
- return row
- }
- }
- return nil
- }
- func (tx *Transaction) doRemoveAttrViewBlock(operation *Operation) (ret *TxErr) {
- err := tx.removeAttributeViewBlock(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID}
- }
- return
- }
- func (tx *Transaction) removeAttributeViewBlock(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- trees := map[string]*parse.Tree{}
- for _, keyValues := range attrView.KeyValues {
- tmp := keyValues.Values[:0]
- for i, values := range keyValues.Values {
- if !gulu.Str.Contains(values.BlockID, operation.SrcIDs) {
- tmp = append(tmp, keyValues.Values[i])
- } else {
- // Remove av block also remove node attr https://github.com/siyuan-note/siyuan/issues/9091#issuecomment-1709824006
- if bt := treenode.GetBlockTree(values.BlockID); nil != bt {
- tree := trees[bt.RootID]
- if nil == tree {
- tree, _ = loadTreeByBlockID(values.BlockID)
- }
- if nil != tree {
- trees[bt.RootID] = tree
- if node := treenode.GetNodeInTree(tree, values.BlockID); nil != node {
- attrs := parse.IAL2Map(node.KramdownIAL)
- if ast.NodeDocument == node.Type {
- delete(attrs, "custom-hidden")
- node.RemoveIALAttr("custom-hidden")
- }
- if avs := attrs[av.NodeAttrNameAvs]; "" != avs {
- avIDs := strings.Split(avs, ",")
- avIDs = gulu.Str.RemoveElem(avIDs, operation.AvID)
- if 0 == len(avIDs) {
- delete(attrs, av.NodeAttrNameAvs)
- node.RemoveIALAttr(av.NodeAttrNameAvs)
- } else {
- attrs[av.NodeAttrNameAvs] = strings.Join(avIDs, ",")
- node.SetIALAttr(av.NodeAttrNameAvs, strings.Join(avIDs, ","))
- }
- }
- if err = setNodeAttrsWithTx(tx, node, tree, attrs); nil != err {
- return
- }
- }
- }
- }
- }
- }
- keyValues.Values = tmp
- }
- for _, view := range attrView.Views {
- for _, blockID := range operation.SrcIDs {
- view.Table.RowIDs = gulu.Str.RemoveElem(view.Table.RowIDs, blockID)
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSetAttrViewColumnWidth(operation *Operation) (ret *TxErr) {
- err := setAttributeViewColWidth(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewColWidth(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- for _, column := range view.Table.Columns {
- if column.ID == operation.ID {
- column.Width = operation.Data.(string)
- break
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSetAttrViewColumnWrap(operation *Operation) (ret *TxErr) {
- err := setAttributeViewColWrap(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewColWrap(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- for _, column := range view.Table.Columns {
- if column.ID == operation.ID {
- column.Wrap = operation.Data.(bool)
- break
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSetAttrViewColumnHidden(operation *Operation) (ret *TxErr) {
- err := setAttributeViewColHidden(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewColHidden(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- for _, column := range view.Table.Columns {
- if column.ID == operation.ID {
- column.Hidden = operation.Data.(bool)
- break
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSetAttrViewColumnPin(operation *Operation) (ret *TxErr) {
- err := setAttributeViewColPin(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewColPin(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- for _, column := range view.Table.Columns {
- if column.ID == operation.ID {
- column.Pin = operation.Data.(bool)
- break
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSetAttrViewColumnIcon(operation *Operation) (ret *TxErr) {
- err := setAttributeViewColIcon(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func setAttributeViewColIcon(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- for _, keyValues := range attrView.KeyValues {
- if keyValues.Key.ID == operation.ID {
- keyValues.Key.Icon = operation.Data.(string)
- break
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSortAttrViewRow(operation *Operation) (ret *TxErr) {
- err := sortAttributeViewRow(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func sortAttributeViewRow(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- var rowID string
- var index, previousIndex int
- for i, r := range view.Table.RowIDs {
- if r == operation.ID {
- rowID = r
- index = i
- break
- }
- }
- if "" == rowID {
- rowID = operation.ID
- view.Table.RowIDs = append(view.Table.RowIDs, rowID)
- index = len(view.Table.RowIDs) - 1
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- view.Table.RowIDs = append(view.Table.RowIDs[:index], view.Table.RowIDs[index+1:]...)
- for i, r := range view.Table.RowIDs {
- if r == operation.PreviousID {
- previousIndex = i + 1
- break
- }
- }
- view.Table.RowIDs = util.InsertElem(view.Table.RowIDs, previousIndex, rowID)
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doSortAttrViewColumn(operation *Operation) (ret *TxErr) {
- err := sortAttributeViewColumn(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func sortAttributeViewColumn(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- view, err := attrView.GetCurrentView()
- if nil != err {
- return
- }
- switch view.LayoutType {
- case av.LayoutTypeTable:
- var col *av.ViewTableColumn
- var index, previousIndex int
- for i, column := range view.Table.Columns {
- if column.ID == operation.ID {
- col = column
- index = i
- break
- }
- }
- if nil == col {
- return
- }
- view.Table.Columns = append(view.Table.Columns[:index], view.Table.Columns[index+1:]...)
- for i, column := range view.Table.Columns {
- if column.ID == operation.PreviousID {
- previousIndex = i + 1
- break
- }
- }
- view.Table.Columns = util.InsertElem(view.Table.Columns, previousIndex, col)
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doAddAttrViewColumn(operation *Operation) (ret *TxErr) {
- err := addAttributeViewColumn(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func addAttributeViewColumn(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- keyType := av.KeyType(operation.Typ)
- switch keyType {
- case av.KeyTypeText, av.KeyTypeNumber, av.KeyTypeDate, av.KeyTypeSelect, av.KeyTypeMSelect, av.KeyTypeURL, av.KeyTypeEmail,
- av.KeyTypePhone, av.KeyTypeMAsset, av.KeyTypeTemplate, av.KeyTypeCreated, av.KeyTypeUpdated, av.KeyTypeCheckbox,
- av.KeyTypeRelation, av.KeyTypeRollup:
- var icon string
- if nil != operation.Data {
- icon = operation.Data.(string)
- }
- key := av.NewKey(operation.ID, operation.Name, icon, keyType)
- attrView.KeyValues = append(attrView.KeyValues, &av.KeyValues{Key: key})
- for _, v := range attrView.Views {
- switch v.LayoutType {
- case av.LayoutTypeTable:
- v.Table.Columns = append(v.Table.Columns, &av.ViewTableColumn{ID: key.ID})
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doUpdateAttrViewColTemplate(operation *Operation) (ret *TxErr) {
- err := updateAttributeViewColTemplate(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func updateAttributeViewColTemplate(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- colType := av.KeyType(operation.Typ)
- switch colType {
- case av.KeyTypeTemplate:
- for _, keyValues := range attrView.KeyValues {
- if keyValues.Key.ID == operation.ID && av.KeyTypeTemplate == keyValues.Key.Type {
- keyValues.Key.Template = operation.Data.(string)
- break
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doUpdateAttrViewColNumberFormat(operation *Operation) (ret *TxErr) {
- err := updateAttributeViewColNumberFormat(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func updateAttributeViewColNumberFormat(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- colType := av.KeyType(operation.Typ)
- switch colType {
- case av.KeyTypeNumber:
- for _, keyValues := range attrView.KeyValues {
- if keyValues.Key.ID == operation.ID && av.KeyTypeNumber == keyValues.Key.Type {
- keyValues.Key.NumberFormat = av.NumberFormat(operation.Format)
- break
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doUpdateAttrViewColumn(operation *Operation) (ret *TxErr) {
- err := updateAttributeViewColumn(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func updateAttributeViewColumn(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- colType := av.KeyType(operation.Typ)
- switch colType {
- case av.KeyTypeBlock, av.KeyTypeText, av.KeyTypeNumber, av.KeyTypeDate, av.KeyTypeSelect, av.KeyTypeMSelect, av.KeyTypeURL, av.KeyTypeEmail,
- av.KeyTypePhone, av.KeyTypeMAsset, av.KeyTypeTemplate, av.KeyTypeCreated, av.KeyTypeUpdated, av.KeyTypeCheckbox,
- av.KeyTypeRelation, av.KeyTypeRollup:
- for _, keyValues := range attrView.KeyValues {
- if keyValues.Key.ID == operation.ID {
- keyValues.Key.Name = strings.TrimSpace(operation.Name)
- keyValues.Key.Type = colType
- break
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doRemoveAttrViewColumn(operation *Operation) (ret *TxErr) {
- err := removeAttributeViewColumn(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func removeAttributeViewColumn(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- var removedKey *av.Key
- for i, keyValues := range attrView.KeyValues {
- if keyValues.Key.ID == operation.ID {
- attrView.KeyValues = append(attrView.KeyValues[:i], attrView.KeyValues[i+1:]...)
- removedKey = keyValues.Key
- break
- }
- }
- // 删除双向关联的目标列
- if nil != removedKey && nil != removedKey.Relation && removedKey.Relation.IsTwoWay {
- destAv, _ := av.ParseAttributeView(removedKey.Relation.AvID)
- if nil != destAv {
- for i, keyValues := range destAv.KeyValues {
- if keyValues.Key.ID == removedKey.Relation.BackKeyID {
- destAv.KeyValues = append(destAv.KeyValues[:i], destAv.KeyValues[i+1:]...)
- break
- }
- }
- for _, view := range destAv.Views {
- switch view.LayoutType {
- case av.LayoutTypeTable:
- for i, column := range view.Table.Columns {
- if column.ID == removedKey.Relation.BackKeyID {
- view.Table.Columns = append(view.Table.Columns[:i], view.Table.Columns[i+1:]...)
- break
- }
- }
- }
- }
- av.SaveAttributeView(destAv)
- util.BroadcastByType("protyle", "refreshAttributeView", 0, "", map[string]interface{}{"id": destAv.ID})
- }
- }
- for _, view := range attrView.Views {
- switch view.LayoutType {
- case av.LayoutTypeTable:
- for i, column := range view.Table.Columns {
- if column.ID == operation.ID {
- view.Table.Columns = append(view.Table.Columns[:i], view.Table.Columns[i+1:]...)
- break
- }
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doReplaceAttrViewBlock(operation *Operation) (ret *TxErr) {
- err := replaceAttributeViewBlock(operation, tx)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID}
- }
- return
- }
- func replaceAttributeViewBlock(operation *Operation, tx *Transaction) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- var node *ast.Node
- if !operation.IsDetached {
- node, _, _ = getNodeByBlockID(tx, operation.NextID)
- }
- for _, keyValues := range attrView.KeyValues {
- for _, value := range keyValues.Values {
- if value.BlockID == operation.PreviousID {
- if value.BlockID != operation.NextID {
- // 换绑
- unbindBlockAv(tx, operation.AvID, value.BlockID)
- }
- value.BlockID = operation.NextID
- if nil != value.Block {
- value.Block.ID = operation.NextID
- value.IsDetached = operation.IsDetached
- if !operation.IsDetached {
- value.Block.Content = getNodeRefText(node)
- }
- }
- if !operation.IsDetached {
- bindBlockAv(tx, operation.AvID, operation.NextID)
- }
- }
- }
- }
- replacedRowID := false
- for _, v := range attrView.Views {
- switch v.LayoutType {
- case av.LayoutTypeTable:
- for i, rowID := range v.Table.RowIDs {
- if rowID == operation.PreviousID {
- v.Table.RowIDs[i] = operation.NextID
- replacedRowID = true
- break
- }
- }
- if !replacedRowID {
- v.Table.RowIDs = append(v.Table.RowIDs, operation.NextID)
- }
- }
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doUpdateAttrViewCell(operation *Operation) (ret *TxErr) {
- err := updateAttributeViewCell(operation, tx)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func updateAttributeViewCell(operation *Operation, tx *Transaction) (err error) {
- err = UpdateAttributeViewCell(tx, operation.AvID, operation.KeyID, operation.RowID, operation.ID, operation.Data)
- return
- }
- func UpdateAttributeViewCell(tx *Transaction, avID, keyID, rowID, cellID string, valueData interface{}) (err error) {
- attrView, err := av.ParseAttributeView(avID)
- if nil != err {
- return
- }
- var blockVal *av.Value
- for _, kv := range attrView.KeyValues {
- if av.KeyTypeBlock == kv.Key.Type {
- for _, v := range kv.Values {
- if rowID == v.Block.ID {
- blockVal = v
- break
- }
- }
- break
- }
- }
- var val *av.Value
- oldIsDetached := true
- if nil != blockVal {
- oldIsDetached = blockVal.IsDetached
- }
- for _, keyValues := range attrView.KeyValues {
- if keyID != keyValues.Key.ID {
- continue
- }
- for _, value := range keyValues.Values {
- if cellID == value.ID {
- val = value
- val.Type = keyValues.Key.Type
- break
- }
- }
- if nil == val {
- val = &av.Value{ID: cellID, KeyID: keyValues.Key.ID, BlockID: rowID, Type: keyValues.Key.Type}
- keyValues.Values = append(keyValues.Values, val)
- }
- break
- }
- isUpdatingBlockKey := av.KeyTypeBlock == val.Type
- oldBoundBlockID := val.BlockID
- var oldRelationBlockIDs []string
- if av.KeyTypeRelation == val.Type {
- if nil != val.Relation {
- for _, bID := range val.Relation.BlockIDs {
- oldRelationBlockIDs = append(oldRelationBlockIDs, bID)
- }
- }
- }
- data, err := gulu.JSON.MarshalJSON(valueData)
- if nil != err {
- return
- }
- if err = gulu.JSON.UnmarshalJSON(data, &val); nil != err {
- return
- }
- relationChangeMode := 0 // 0:不变(仅排序),1:增加,2:减少
- if av.KeyTypeRelation == val.Type {
- // 关联列得 content 是自动渲染的,所以不需要保存
- val.Relation.Contents = nil
- // 计算关联排序模式
- if len(oldRelationBlockIDs) == len(val.Relation.BlockIDs) {
- relationChangeMode = 0
- } else {
- if len(oldRelationBlockIDs) > len(val.Relation.BlockIDs) {
- relationChangeMode = 2
- } else {
- relationChangeMode = 1
- }
- }
- }
- // val.IsDetached 只有更新主键的时候才会传入,所以下面需要结合 isUpdatingBlockKey 来判断
- if oldIsDetached { // 之前是游离行
- if !val.IsDetached { // 现在绑定了块
- // 将游离行绑定到新建的块上
- bindBlockAv(tx, avID, rowID)
- }
- } else { // 之前绑定了块
- if isUpdatingBlockKey { // 正在更新主键
- if val.IsDetached { // 现在是游离行
- // 将绑定的块从属性视图中移除
- unbindBlockAv(tx, avID, rowID)
- } else { // 现在绑定了块
- if oldBoundBlockID != val.BlockID { // 之前绑定的块和现在绑定的块不一样
- // 换绑块
- unbindBlockAv(tx, avID, oldBoundBlockID)
- bindBlockAv(tx, avID, val.BlockID)
- } else { // 之前绑定的块和现在绑定的块一样
- // 直接返回,因为锚文本不允许更改
- return
- }
- }
- }
- }
- if nil != blockVal {
- blockVal.Block.Updated = time.Now().UnixMilli()
- blockVal.IsInitialized = true
- if isUpdatingBlockKey {
- blockVal.IsDetached = val.IsDetached
- }
- }
- key, _ := attrView.GetKey(val.KeyID)
- if nil != key && av.KeyTypeRelation == key.Type && nil != key.Relation && key.Relation.IsTwoWay {
- destAv, _ := av.ParseAttributeView(key.Relation.AvID)
- if nil != destAv {
- // relationChangeMode
- // 0:关联列值不变(仅排序),不影响目标值
- // 1:关联列值增加,增加目标值
- // 2:关联列值减少,减少目标值
- if 1 == relationChangeMode {
- addBlockIDs := val.Relation.BlockIDs
- for _, bID := range oldRelationBlockIDs {
- addBlockIDs = gulu.Str.RemoveElem(addBlockIDs, bID)
- }
- for _, blockID := range addBlockIDs {
- for _, keyValues := range destAv.KeyValues {
- if keyValues.Key.ID != key.Relation.BackKeyID {
- continue
- }
- destVal := keyValues.GetValue(blockID)
- if nil == destVal {
- destVal = &av.Value{ID: ast.NewNodeID(), KeyID: keyValues.Key.ID, BlockID: blockID, Type: keyValues.Key.Type, Relation: &av.ValueRelation{}}
- keyValues.Values = append(keyValues.Values, destVal)
- }
- destVal.Relation.BlockIDs = append(destVal.Relation.BlockIDs, rowID)
- destVal.Relation.BlockIDs = gulu.Str.RemoveDuplicatedElem(destVal.Relation.BlockIDs)
- break
- }
- }
- } else if 2 == relationChangeMode {
- removeBlockIDs := oldRelationBlockIDs
- for _, bID := range val.Relation.BlockIDs {
- removeBlockIDs = gulu.Str.RemoveElem(removeBlockIDs, bID)
- }
- for _, blockID := range removeBlockIDs {
- for _, keyValues := range destAv.KeyValues {
- if keyValues.Key.ID != key.Relation.BackKeyID {
- continue
- }
- for _, value := range keyValues.Values {
- if value.BlockID == blockID {
- value.Relation.BlockIDs = gulu.Str.RemoveElem(value.Relation.BlockIDs, rowID)
- break
- }
- }
- }
- }
- }
- av.SaveAttributeView(destAv)
- util.BroadcastByType("protyle", "refreshAttributeView", 0, "", map[string]interface{}{"id": destAv.ID})
- }
- }
- if err = av.SaveAttributeView(attrView); nil != err {
- return
- }
- return
- }
- func unbindBlockAv(tx *Transaction, avID, blockID string) {
- node, tree, err := getNodeByBlockID(tx, blockID)
- if nil != err {
- return
- }
- attrs := parse.IAL2Map(node.KramdownIAL)
- if "" == attrs[av.NodeAttrNameAvs] {
- return
- }
- avIDs := strings.Split(attrs[av.NodeAttrNameAvs], ",")
- avIDs = gulu.Str.RemoveElem(avIDs, avID)
- if 0 == len(avIDs) {
- delete(attrs, av.NodeAttrNameAvs)
- node.RemoveIALAttr(av.NodeAttrNameAvs)
- } else {
- attrs[av.NodeAttrNameAvs] = strings.Join(avIDs, ",")
- node.SetIALAttr(av.NodeAttrNameAvs, strings.Join(avIDs, ","))
- }
- if nil != tx {
- err = setNodeAttrsWithTx(tx, node, tree, attrs)
- } else {
- err = setNodeAttrs(node, tree, attrs)
- }
- if nil != err {
- logging.LogWarnf("set node [%s] attrs failed: %s", blockID, err)
- return
- }
- return
- }
- func bindBlockAv(tx *Transaction, avID, blockID string) {
- node, tree, err := getNodeByBlockID(tx, blockID)
- if nil != err {
- return
- }
- attrs := parse.IAL2Map(node.KramdownIAL)
- if "" == attrs[av.NodeAttrNameAvs] {
- attrs[av.NodeAttrNameAvs] = avID
- } else {
- avIDs := strings.Split(attrs[av.NodeAttrNameAvs], ",")
- avIDs = append(avIDs, avID)
- avIDs = gulu.Str.RemoveDuplicatedElem(avIDs)
- attrs[av.NodeAttrNameAvs] = strings.Join(avIDs, ",")
- }
- if nil != tx {
- err = setNodeAttrsWithTx(tx, node, tree, attrs)
- } else {
- err = setNodeAttrs(node, tree, attrs)
- }
- if nil != err {
- logging.LogWarnf("set node [%s] attrs failed: %s", blockID, err)
- return
- }
- return
- }
- func getNodeByBlockID(tx *Transaction, blockID string) (node *ast.Node, tree *parse.Tree, err error) {
- if nil != tx {
- tree, err = tx.loadTree(blockID)
- } else {
- tree, err = loadTreeByBlockID(blockID)
- }
- if nil != err {
- return
- }
- node = treenode.GetNodeInTree(tree, blockID)
- if nil == node {
- logging.LogWarnf("node [%s] not found in tree [%s]", blockID, tree.ID)
- return
- }
- return
- }
- func (tx *Transaction) doUpdateAttrViewColOptions(operation *Operation) (ret *TxErr) {
- err := updateAttributeViewColumnOptions(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func updateAttributeViewColumnOptions(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- jsonData, err := gulu.JSON.MarshalJSON(operation.Data)
- if nil != err {
- return
- }
- options := []*av.SelectOption{}
- if err = gulu.JSON.UnmarshalJSON(jsonData, &options); nil != err {
- return
- }
- for _, keyValues := range attrView.KeyValues {
- if keyValues.Key.ID == operation.ID {
- keyValues.Key.Options = options
- err = av.SaveAttributeView(attrView)
- return
- }
- }
- return
- }
- func (tx *Transaction) doRemoveAttrViewColOption(operation *Operation) (ret *TxErr) {
- err := removeAttributeViewColumnOption(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func removeAttributeViewColumnOption(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- optName := operation.Data.(string)
- key, err := attrView.GetKey(operation.ID)
- if nil != err {
- return
- }
- for i, opt := range key.Options {
- if optName == opt.Name {
- key.Options = append(key.Options[:i], key.Options[i+1:]...)
- break
- }
- }
- for _, keyValues := range attrView.KeyValues {
- if keyValues.Key.ID != operation.ID {
- continue
- }
- for _, value := range keyValues.Values {
- if nil == value || nil == value.MSelect {
- continue
- }
- for i, opt := range value.MSelect {
- if optName == opt.Content {
- value.MSelect = append(value.MSelect[:i], value.MSelect[i+1:]...)
- break
- }
- }
- }
- break
- }
- err = av.SaveAttributeView(attrView)
- return
- }
- func (tx *Transaction) doUpdateAttrViewColOption(operation *Operation) (ret *TxErr) {
- err := updateAttributeViewColumnOption(operation)
- if nil != err {
- return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
- }
- return
- }
- func updateAttributeViewColumnOption(operation *Operation) (err error) {
- attrView, err := av.ParseAttributeView(operation.AvID)
- if nil != err {
- return
- }
- key, err := attrView.GetKey(operation.ID)
- if nil != err {
- return
- }
- data := operation.Data.(map[string]interface{})
- oldName := data["oldName"].(string)
- newName := data["newName"].(string)
- newColor := data["newColor"].(string)
- for i, opt := range key.Options {
- if oldName == opt.Name {
- key.Options[i].Name = newName
- key.Options[i].Color = newColor
- break
- }
- }
- for _, keyValues := range attrView.KeyValues {
- if keyValues.Key.ID != operation.ID {
- continue
- }
- for _, value := range keyValues.Values {
- if nil == value || nil == value.MSelect {
- continue
- }
- for i, opt := range value.MSelect {
- if oldName == opt.Content {
- value.MSelect[i].Content = newName
- value.MSelect[i].Color = newColor
- break
- }
- }
- }
- break
- }
- err = av.SaveAttributeView(attrView)
- return
- }
|