attribute_view.go 92 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402
  1. // SiYuan - Refactor your thinking
  2. // Copyright (c) 2020-present, b3log.org
  3. //
  4. // This program is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Affero General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // This program is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Affero General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Affero General Public License
  15. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  16. package model
  17. import (
  18. "bytes"
  19. "os"
  20. "path/filepath"
  21. "sort"
  22. "strconv"
  23. "strings"
  24. "text/template"
  25. "time"
  26. "github.com/88250/gulu"
  27. "github.com/88250/lute/ast"
  28. "github.com/88250/lute/parse"
  29. "github.com/siyuan-note/dejavu/entity"
  30. "github.com/siyuan-note/filelock"
  31. "github.com/siyuan-note/logging"
  32. "github.com/siyuan-note/siyuan/kernel/av"
  33. "github.com/siyuan-note/siyuan/kernel/cache"
  34. "github.com/siyuan-note/siyuan/kernel/treenode"
  35. "github.com/siyuan-note/siyuan/kernel/util"
  36. "github.com/xrash/smetrics"
  37. )
  38. func SetDatabaseBlockView(blockID, viewID string) (err error) {
  39. node, tree, err := getNodeByBlockID(nil, blockID)
  40. if nil != err {
  41. return
  42. }
  43. attrs := parse.IAL2Map(node.KramdownIAL)
  44. attrs[av.NodeAttrView] = viewID
  45. err = setNodeAttrs(node, tree, attrs)
  46. if nil != err {
  47. logging.LogWarnf("set node [%s] attrs failed: %s", blockID, err)
  48. return
  49. }
  50. return
  51. }
  52. func GetAttributeViewPrimaryKeyValues(avID, keyword string, page, pageSize int) (attributeViewName string, databaseBlockIDs []string, keyValues *av.KeyValues, err error) {
  53. waitForSyncingStorages()
  54. attrView, err := av.ParseAttributeView(avID)
  55. if nil != err {
  56. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  57. return
  58. }
  59. attributeViewName = attrView.Name
  60. if "" == attributeViewName {
  61. attributeViewName = Conf.language(105)
  62. }
  63. databaseBlockIDs = treenode.GetMirrorAttrViewBlockIDs(avID)
  64. keyValues = attrView.GetBlockKeyValues()
  65. // 过滤掉不在视图中的值
  66. tmp := map[string]*av.Value{}
  67. for _, kv := range keyValues.Values {
  68. for _, view := range attrView.Views {
  69. switch view.LayoutType {
  70. case av.LayoutTypeTable:
  71. if !kv.IsDetached {
  72. if nil == treenode.GetBlockTree(kv.BlockID) {
  73. break
  74. }
  75. }
  76. if gulu.Str.Contains(kv.Block.ID, view.Table.RowIDs) {
  77. tmp[kv.Block.ID] = kv
  78. }
  79. }
  80. }
  81. }
  82. keyValues.Values = []*av.Value{}
  83. for _, v := range tmp {
  84. if strings.Contains(strings.ToLower(v.String()), strings.ToLower(keyword)) {
  85. keyValues.Values = append(keyValues.Values, v)
  86. }
  87. }
  88. if 1 > pageSize {
  89. pageSize = 32
  90. }
  91. start := (page - 1) * pageSize
  92. end := start + pageSize
  93. if len(keyValues.Values) < end {
  94. end = len(keyValues.Values)
  95. }
  96. keyValues.Values = keyValues.Values[start:end]
  97. return
  98. }
  99. func GetAttributeViewFilterSort(avID, blockID string) (filters []*av.ViewFilter, sorts []*av.ViewSort) {
  100. waitForSyncingStorages()
  101. attrView, err := av.ParseAttributeView(avID)
  102. if nil != err {
  103. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  104. return
  105. }
  106. view, err := getAttrViewViewByBlockID(attrView, blockID)
  107. if nil == view {
  108. view, err = attrView.GetCurrentView(attrView.ViewID)
  109. if nil != err {
  110. logging.LogErrorf("get current view failed: %s", err)
  111. return
  112. }
  113. }
  114. filters = []*av.ViewFilter{}
  115. sorts = []*av.ViewSort{}
  116. switch view.LayoutType {
  117. case av.LayoutTypeTable:
  118. filters = view.Table.Filters
  119. sorts = view.Table.Sorts
  120. }
  121. return
  122. }
  123. func SearchAttributeViewNonRelationKey(avID, keyword string) (ret []*av.Key) {
  124. waitForSyncingStorages()
  125. ret = []*av.Key{}
  126. attrView, err := av.ParseAttributeView(avID)
  127. if nil != err {
  128. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  129. return
  130. }
  131. for _, keyValues := range attrView.KeyValues {
  132. if av.KeyTypeRelation != keyValues.Key.Type && av.KeyTypeRollup != keyValues.Key.Type && av.KeyTypeTemplate != keyValues.Key.Type && av.KeyTypeCreated != keyValues.Key.Type && av.KeyTypeUpdated != keyValues.Key.Type {
  133. if strings.Contains(strings.ToLower(keyValues.Key.Name), strings.ToLower(keyword)) {
  134. ret = append(ret, keyValues.Key)
  135. }
  136. }
  137. }
  138. return
  139. }
  140. func SearchAttributeViewRelationKey(avID, keyword string) (ret []*av.Key) {
  141. waitForSyncingStorages()
  142. ret = []*av.Key{}
  143. attrView, err := av.ParseAttributeView(avID)
  144. if nil != err {
  145. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  146. return
  147. }
  148. for _, keyValues := range attrView.KeyValues {
  149. if av.KeyTypeRelation == keyValues.Key.Type && nil != keyValues.Key.Relation {
  150. if strings.Contains(strings.ToLower(keyValues.Key.Name), strings.ToLower(keyword)) {
  151. ret = append(ret, keyValues.Key)
  152. }
  153. }
  154. }
  155. return
  156. }
  157. func GetAttributeView(avID string) (ret *av.AttributeView) {
  158. waitForSyncingStorages()
  159. ret, _ = av.ParseAttributeView(avID)
  160. return
  161. }
  162. type SearchAttributeViewResult struct {
  163. AvID string `json:"avID"`
  164. AvName string `json:"avName"`
  165. BlockID string `json:"blockID"`
  166. HPath string `json:"hPath"`
  167. }
  168. func SearchAttributeView(keyword string) (ret []*SearchAttributeViewResult) {
  169. waitForSyncingStorages()
  170. ret = []*SearchAttributeViewResult{}
  171. keyword = strings.TrimSpace(keyword)
  172. type result struct {
  173. AvID string
  174. AvName string
  175. AvUpdated int64
  176. Score float64
  177. }
  178. var avs []*result
  179. avDir := filepath.Join(util.DataDir, "storage", "av")
  180. entries, err := os.ReadDir(avDir)
  181. if nil != err {
  182. logging.LogErrorf("read directory [%s] failed: %s", avDir, err)
  183. return
  184. }
  185. avBlockRels := av.GetBlockRels()
  186. for _, entry := range entries {
  187. if entry.IsDir() {
  188. continue
  189. }
  190. id := strings.TrimSuffix(entry.Name(), ".json")
  191. if !ast.IsNodeIDPattern(id) {
  192. continue
  193. }
  194. if nil == avBlockRels[id] {
  195. continue
  196. }
  197. name, _ := av.GetAttributeViewNameByPath(filepath.Join(avDir, entry.Name()))
  198. info, _ := entry.Info()
  199. if "" != keyword {
  200. if strings.Contains(strings.ToLower(name), strings.ToLower(keyword)) {
  201. score := smetrics.JaroWinkler(name, keyword, 0.7, 4)
  202. a := &result{AvID: id, AvName: name, Score: score}
  203. if nil != info && !info.ModTime().IsZero() {
  204. a.AvUpdated = info.ModTime().UnixMilli()
  205. }
  206. avs = append(avs, a)
  207. }
  208. } else {
  209. a := &result{AvID: id, AvName: name}
  210. if nil != info && !info.ModTime().IsZero() {
  211. a.AvUpdated = info.ModTime().UnixMilli()
  212. }
  213. avs = append(avs, a)
  214. }
  215. }
  216. if "" == keyword {
  217. sort.Slice(avs, func(i, j int) bool { return avs[i].AvUpdated > avs[j].AvUpdated })
  218. } else {
  219. sort.SliceStable(avs, func(i, j int) bool {
  220. if avs[i].Score == avs[j].Score {
  221. return avs[i].AvUpdated > avs[j].AvUpdated
  222. }
  223. return avs[i].Score > avs[j].Score
  224. })
  225. }
  226. if 12 <= len(avs) {
  227. avs = avs[:12]
  228. }
  229. var avIDs []string
  230. for _, a := range avs {
  231. avIDs = append(avIDs, a.AvID)
  232. }
  233. avBlocks := treenode.BatchGetMirrorAttrViewBlocks(avIDs)
  234. var blockIDs []string
  235. for _, avBlock := range avBlocks {
  236. blockIDs = append(blockIDs, avBlock.BlockIDs...)
  237. }
  238. blockIDs = gulu.Str.RemoveDuplicatedElem(blockIDs)
  239. trees := map[string]*parse.Tree{}
  240. for _, blockID := range blockIDs {
  241. bt := treenode.GetBlockTree(blockID)
  242. if nil == bt {
  243. continue
  244. }
  245. tree := trees[bt.RootID]
  246. if nil == tree {
  247. tree, _ = LoadTreeByBlockID(blockID)
  248. if nil != tree {
  249. trees[bt.RootID] = tree
  250. }
  251. }
  252. if nil == tree {
  253. continue
  254. }
  255. node := treenode.GetNodeInTree(tree, blockID)
  256. if nil == node {
  257. continue
  258. }
  259. if "" == node.AttributeViewID {
  260. continue
  261. }
  262. avID := node.AttributeViewID
  263. var existAv *result
  264. for _, av := range avs {
  265. if av.AvID == avID {
  266. existAv = av
  267. break
  268. }
  269. }
  270. if nil == existAv {
  271. continue
  272. }
  273. exist := false
  274. for _, result := range ret {
  275. if result.AvID == avID {
  276. exist = true
  277. break
  278. }
  279. }
  280. var hPath string
  281. baseBlock := treenode.GetBlockTreeRootByPath(node.Box, node.Path)
  282. if nil != baseBlock {
  283. hPath = baseBlock.HPath
  284. }
  285. box := Conf.Box(node.Box)
  286. if nil != box {
  287. hPath = box.Name + hPath
  288. }
  289. if !exist {
  290. ret = append(ret, &SearchAttributeViewResult{
  291. AvID: avID,
  292. AvName: existAv.AvName,
  293. BlockID: blockID,
  294. HPath: hPath,
  295. })
  296. }
  297. }
  298. return
  299. }
  300. type BlockAttributeViewKeys struct {
  301. AvID string `json:"avID"`
  302. AvName string `json:"avName"`
  303. BlockIDs []string `json:"blockIDs"`
  304. KeyValues []*av.KeyValues `json:"keyValues"`
  305. }
  306. func GetBlockAttributeViewKeys(blockID string) (ret []*BlockAttributeViewKeys) {
  307. waitForSyncingStorages()
  308. ret = []*BlockAttributeViewKeys{}
  309. attrs := GetBlockAttrsWithoutWaitWriting(blockID)
  310. avs := attrs[av.NodeAttrNameAvs]
  311. if "" == avs {
  312. return
  313. }
  314. avIDs := strings.Split(avs, ",")
  315. for _, avID := range avIDs {
  316. attrView, err := av.ParseAttributeView(avID)
  317. if nil != err {
  318. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  319. unbindBlockAv(nil, avID, blockID)
  320. return
  321. }
  322. if 1 > len(attrView.Views) {
  323. err = av.ErrViewNotFound
  324. unbindBlockAv(nil, avID, blockID)
  325. return
  326. }
  327. if !attrView.ExistBlock(blockID) {
  328. // 比如剪切后粘贴,块 ID 会变,但是属性还在块上,这里做一次数据订正
  329. // Auto verify the database name when clicking the block superscript icon https://github.com/siyuan-note/siyuan/issues/10861
  330. unbindBlockAv(nil, avID, blockID)
  331. return
  332. }
  333. var keyValues []*av.KeyValues
  334. for _, kv := range attrView.KeyValues {
  335. kValues := &av.KeyValues{Key: kv.Key}
  336. for _, v := range kv.Values {
  337. if v.BlockID == blockID {
  338. kValues.Values = append(kValues.Values, v)
  339. }
  340. }
  341. switch kValues.Key.Type {
  342. case av.KeyTypeRollup:
  343. kValues.Values = append(kValues.Values, &av.Value{ID: ast.NewNodeID(), KeyID: kValues.Key.ID, BlockID: blockID, Type: av.KeyTypeRollup, Rollup: &av.ValueRollup{Contents: []*av.Value{}}})
  344. case av.KeyTypeTemplate:
  345. kValues.Values = append(kValues.Values, &av.Value{ID: ast.NewNodeID(), KeyID: kValues.Key.ID, BlockID: blockID, Type: av.KeyTypeTemplate, Template: &av.ValueTemplate{Content: ""}})
  346. case av.KeyTypeCreated:
  347. kValues.Values = append(kValues.Values, &av.Value{ID: ast.NewNodeID(), KeyID: kValues.Key.ID, BlockID: blockID, Type: av.KeyTypeCreated})
  348. case av.KeyTypeUpdated:
  349. kValues.Values = append(kValues.Values, &av.Value{ID: ast.NewNodeID(), KeyID: kValues.Key.ID, BlockID: blockID, Type: av.KeyTypeUpdated})
  350. }
  351. if 0 < len(kValues.Values) {
  352. keyValues = append(keyValues, kValues)
  353. } else {
  354. // 如果没有值,那么就补一个默认值
  355. kValues.Values = append(kValues.Values, treenode.GetAttributeViewDefaultValue(ast.NewNodeID(), kv.Key.ID, blockID, kv.Key.Type))
  356. keyValues = append(keyValues, kValues)
  357. }
  358. }
  359. // 渲染自动生成的列值,比如模板列、关联列、汇总列、创建时间列和更新时间列
  360. // 先处理关联列、汇总列、创建时间列和更新时间列
  361. for _, kv := range keyValues {
  362. switch kv.Key.Type {
  363. case av.KeyTypeRollup:
  364. if nil == kv.Key.Rollup {
  365. break
  366. }
  367. relKey, _ := attrView.GetKey(kv.Key.Rollup.RelationKeyID)
  368. if nil == relKey {
  369. break
  370. }
  371. relVal := attrView.GetValue(kv.Key.Rollup.RelationKeyID, kv.Values[0].BlockID)
  372. if nil != relVal && nil != relVal.Relation {
  373. destAv, _ := av.ParseAttributeView(relKey.Relation.AvID)
  374. destKey, _ := destAv.GetKey(kv.Key.Rollup.KeyID)
  375. if nil != destAv && nil != destKey {
  376. for _, bID := range relVal.Relation.BlockIDs {
  377. destVal := destAv.GetValue(kv.Key.Rollup.KeyID, bID)
  378. if nil == destVal {
  379. if destAv.ExistBlock(bID) { // 数据库中存在行但是列值不存在是数据未初始化,这里补一个默认值
  380. destVal = treenode.GetAttributeViewDefaultValue(ast.NewNodeID(), kv.Key.Rollup.KeyID, bID, destKey.Type)
  381. }
  382. if nil == destVal {
  383. continue
  384. }
  385. }
  386. if av.KeyTypeNumber == destKey.Type {
  387. destVal.Number.Format = destKey.NumberFormat
  388. destVal.Number.FormatNumber()
  389. }
  390. kv.Values[0].Rollup.Contents = append(kv.Values[0].Rollup.Contents, destVal.Clone())
  391. }
  392. kv.Values[0].Rollup.RenderContents(kv.Key.Rollup.Calc, destKey)
  393. }
  394. }
  395. case av.KeyTypeRelation:
  396. if nil == kv.Key.Relation {
  397. break
  398. }
  399. destAv, _ := av.ParseAttributeView(kv.Key.Relation.AvID)
  400. if nil == destAv {
  401. break
  402. }
  403. blocks := map[string]*av.Value{}
  404. for _, blockValue := range destAv.GetBlockKeyValues().Values {
  405. blocks[blockValue.BlockID] = blockValue
  406. }
  407. kv.Values[0].Relation.Contents = nil // 先清空 https://github.com/siyuan-note/siyuan/issues/10670
  408. for _, bID := range kv.Values[0].Relation.BlockIDs {
  409. kv.Values[0].Relation.Contents = append(kv.Values[0].Relation.Contents, blocks[bID])
  410. }
  411. case av.KeyTypeCreated:
  412. createdStr := blockID[:len("20060102150405")]
  413. created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local)
  414. if nil == parseErr {
  415. kv.Values[0].Created = av.NewFormattedValueCreated(created.UnixMilli(), 0, av.CreatedFormatNone)
  416. kv.Values[0].Created.IsNotEmpty = true
  417. } else {
  418. logging.LogWarnf("parse created [%s] failed: %s", createdStr, parseErr)
  419. kv.Values[0].Created = av.NewFormattedValueCreated(time.Now().UnixMilli(), 0, av.CreatedFormatNone)
  420. }
  421. case av.KeyTypeUpdated:
  422. ial := GetBlockAttrsWithoutWaitWriting(blockID)
  423. updatedStr := ial["updated"]
  424. updated, parseErr := time.ParseInLocation("20060102150405", updatedStr, time.Local)
  425. if nil == parseErr {
  426. kv.Values[0].Updated = av.NewFormattedValueUpdated(updated.UnixMilli(), 0, av.UpdatedFormatNone)
  427. kv.Values[0].Updated.IsNotEmpty = true
  428. } else {
  429. logging.LogWarnf("parse updated [%s] failed: %s", updatedStr, parseErr)
  430. kv.Values[0].Updated = av.NewFormattedValueUpdated(time.Now().UnixMilli(), 0, av.UpdatedFormatNone)
  431. }
  432. }
  433. }
  434. // 再处理模板列
  435. // 获取闪卡信息
  436. // TODO 目前看来使用场景不多,暂时不实现了 https://github.com/siyuan-note/siyuan/issues/10502#issuecomment-1986703280
  437. var flashcard *Flashcard
  438. //deck := Decks[builtinDeckID]
  439. //if nil != deck {
  440. // blockIDs := []string{blockID}
  441. // cards := deck.GetCardsByBlockIDs(blockIDs)
  442. // now := time.Now()
  443. // if 0 < len(cards) {
  444. // flashcard = newFlashcard(cards[0], builtinDeckID, now)
  445. // }
  446. //}
  447. // 渲染模板
  448. for _, kv := range keyValues {
  449. switch kv.Key.Type {
  450. case av.KeyTypeTemplate:
  451. if 0 < len(kv.Values) {
  452. ial := map[string]string{}
  453. block := getRowBlockValue(keyValues)
  454. if nil != block && !block.IsDetached {
  455. ial = GetBlockAttrsWithoutWaitWriting(block.BlockID)
  456. }
  457. kv.Values[0].Template.Content = renderTemplateCol(ial, flashcard, keyValues, kv.Key.Template)
  458. }
  459. }
  460. }
  461. // Attribute Panel - Database sort attributes by view column order https://github.com/siyuan-note/siyuan/issues/9319
  462. viewID := attrs[av.NodeAttrView]
  463. view, _ := attrView.GetCurrentView(viewID)
  464. if nil != view {
  465. sorts := map[string]int{}
  466. for i, col := range view.Table.Columns {
  467. sorts[col.ID] = i
  468. }
  469. sort.Slice(keyValues, func(i, j int) bool {
  470. return sorts[keyValues[i].Key.ID] < sorts[keyValues[j].Key.ID]
  471. })
  472. }
  473. blockIDs := treenode.GetMirrorAttrViewBlockIDs(avID)
  474. if 1 > len(blockIDs) {
  475. // 老数据兼容处理
  476. avBts := treenode.GetBlockTreesByType("av")
  477. for _, avBt := range avBts {
  478. if nil == avBt {
  479. continue
  480. }
  481. tree, _ := LoadTreeByBlockID(avBt.ID)
  482. if nil == tree {
  483. continue
  484. }
  485. node := treenode.GetNodeInTree(tree, avBt.ID)
  486. if nil == node {
  487. continue
  488. }
  489. if avID == node.AttributeViewID {
  490. blockIDs = append(blockIDs, avBt.ID)
  491. }
  492. }
  493. if 1 > len(blockIDs) {
  494. tree, _ := LoadTreeByBlockID(blockID)
  495. if nil != tree {
  496. node := treenode.GetNodeInTree(tree, blockID)
  497. if nil != node {
  498. if removeErr := removeNodeAvID(node, avID, nil, tree); nil != removeErr {
  499. logging.LogErrorf("remove node avID [%s] failed: %s", avID, removeErr)
  500. }
  501. }
  502. }
  503. continue
  504. }
  505. blockIDs = gulu.Str.RemoveDuplicatedElem(blockIDs)
  506. for _, blockID := range blockIDs {
  507. av.UpsertBlockRel(avID, blockID)
  508. }
  509. }
  510. ret = append(ret, &BlockAttributeViewKeys{
  511. AvID: avID,
  512. AvName: attrView.Name,
  513. BlockIDs: blockIDs,
  514. KeyValues: keyValues,
  515. })
  516. }
  517. return
  518. }
  519. func RenderRepoSnapshotAttributeView(indexID, avID string) (viewable av.Viewable, attrView *av.AttributeView, err error) {
  520. repo, err := newRepository()
  521. if nil != err {
  522. return
  523. }
  524. index, err := repo.GetIndex(indexID)
  525. if nil != err {
  526. return
  527. }
  528. files, err := repo.GetFiles(index)
  529. if nil != err {
  530. return
  531. }
  532. var avFile *entity.File
  533. for _, f := range files {
  534. if "/storage/av/"+avID+".json" == f.Path {
  535. avFile = f
  536. break
  537. }
  538. }
  539. if nil == avFile {
  540. attrView = av.NewAttributeView(avID)
  541. } else {
  542. data, readErr := repo.OpenFile(avFile)
  543. if nil != readErr {
  544. logging.LogErrorf("read attribute view [%s] failed: %s", avID, readErr)
  545. return
  546. }
  547. attrView = &av.AttributeView{}
  548. if err = gulu.JSON.UnmarshalJSON(data, attrView); nil != err {
  549. logging.LogErrorf("unmarshal attribute view [%s] failed: %s", avID, err)
  550. return
  551. }
  552. }
  553. viewable, err = renderAttributeView(attrView, "", "", 1, -1)
  554. return
  555. }
  556. func RenderHistoryAttributeView(avID, created string) (viewable av.Viewable, attrView *av.AttributeView, err error) {
  557. createdUnix, parseErr := strconv.ParseInt(created, 10, 64)
  558. if nil != parseErr {
  559. logging.LogErrorf("parse created [%s] failed: %s", created, parseErr)
  560. return
  561. }
  562. dirPrefix := time.Unix(createdUnix, 0).Format("2006-01-02-150405")
  563. globPath := filepath.Join(util.HistoryDir, dirPrefix+"*")
  564. matches, err := filepath.Glob(globPath)
  565. if nil != err {
  566. logging.LogErrorf("glob [%s] failed: %s", globPath, err)
  567. return
  568. }
  569. if 1 > len(matches) {
  570. return
  571. }
  572. historyDir := matches[0]
  573. avJSONPath := filepath.Join(historyDir, "storage", "av", avID+".json")
  574. if !gulu.File.IsExist(avJSONPath) {
  575. avJSONPath = filepath.Join(util.DataDir, "storage", "av", avID+".json")
  576. }
  577. if !gulu.File.IsExist(avJSONPath) {
  578. attrView = av.NewAttributeView(avID)
  579. } else {
  580. data, readErr := os.ReadFile(avJSONPath)
  581. if nil != readErr {
  582. logging.LogErrorf("read attribute view [%s] failed: %s", avID, readErr)
  583. return
  584. }
  585. attrView = &av.AttributeView{}
  586. if err = gulu.JSON.UnmarshalJSON(data, attrView); nil != err {
  587. logging.LogErrorf("unmarshal attribute view [%s] failed: %s", avID, err)
  588. return
  589. }
  590. }
  591. viewable, err = renderAttributeView(attrView, "", "", 1, -1)
  592. return
  593. }
  594. func RenderAttributeView(avID, viewID, query string, page, pageSize int) (viewable av.Viewable, attrView *av.AttributeView, err error) {
  595. waitForSyncingStorages()
  596. if avJSONPath := av.GetAttributeViewDataPath(avID); !filelock.IsExist(avJSONPath) {
  597. attrView = av.NewAttributeView(avID)
  598. if err = av.SaveAttributeView(attrView); nil != err {
  599. logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
  600. return
  601. }
  602. }
  603. attrView, err = av.ParseAttributeView(avID)
  604. if nil != err {
  605. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  606. return
  607. }
  608. viewable, err = renderAttributeView(attrView, viewID, query, page, pageSize)
  609. return
  610. }
  611. func renderAttributeView(attrView *av.AttributeView, viewID, query string, page, pageSize int) (viewable av.Viewable, err error) {
  612. if 1 > len(attrView.Views) {
  613. view, _, _ := av.NewTableViewWithBlockKey(ast.NewNodeID())
  614. attrView.Views = append(attrView.Views, view)
  615. attrView.ViewID = view.ID
  616. if err = av.SaveAttributeView(attrView); nil != err {
  617. logging.LogErrorf("save attribute view [%s] failed: %s", attrView.ID, err)
  618. return
  619. }
  620. }
  621. var view *av.View
  622. if "" != viewID {
  623. view, _ = attrView.GetCurrentView(viewID)
  624. if nil != view && view.ID != attrView.ViewID {
  625. attrView.ViewID = view.ID
  626. if err = av.SaveAttributeView(attrView); nil != err {
  627. logging.LogErrorf("save attribute view [%s] failed: %s", attrView.ID, err)
  628. return
  629. }
  630. }
  631. } else {
  632. view = attrView.GetView(attrView.ViewID)
  633. }
  634. if nil == view {
  635. view = attrView.Views[0]
  636. }
  637. // 做一些数据兼容和订正处理,保存的时候也会做 av.SaveAttributeView()
  638. currentTimeMillis := util.CurrentTimeMillis()
  639. for _, kv := range attrView.KeyValues {
  640. switch kv.Key.Type {
  641. case av.KeyTypeBlock: // 补全 block 的创建时间和更新时间
  642. for _, v := range kv.Values {
  643. if 0 == v.Block.Created {
  644. if "" == v.Block.ID {
  645. v.Block.ID = v.BlockID
  646. if "" == v.Block.ID {
  647. v.Block.ID = ast.NewNodeID()
  648. v.BlockID = v.Block.ID
  649. }
  650. }
  651. createdStr := v.Block.ID[:len("20060102150405")]
  652. created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local)
  653. if nil == parseErr {
  654. v.Block.Created = created.UnixMilli()
  655. } else {
  656. v.Block.Created = currentTimeMillis
  657. }
  658. }
  659. if 0 == v.Block.Updated {
  660. v.Block.Updated = v.Block.Created
  661. }
  662. }
  663. }
  664. for _, v := range kv.Values {
  665. // 校验日期 IsNotEmpty
  666. if av.KeyTypeDate == kv.Key.Type {
  667. if nil != v.Date && 0 != v.Date.Content && !v.Date.IsNotEmpty {
  668. v.Date.IsNotEmpty = true
  669. }
  670. }
  671. // 校验数字 IsNotEmpty
  672. if av.KeyTypeNumber == kv.Key.Type {
  673. if nil != v.Number && 0 != v.Number.Content && !v.Number.IsNotEmpty {
  674. v.Number.IsNotEmpty = true
  675. }
  676. }
  677. // 补全值的创建时间和更新时间
  678. if "" == v.ID {
  679. v.ID = ast.NewNodeID()
  680. }
  681. if 0 == v.CreatedAt {
  682. createdStr := v.ID[:len("20060102150405")]
  683. created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local)
  684. if nil == parseErr {
  685. v.CreatedAt = created.UnixMilli()
  686. } else {
  687. v.CreatedAt = currentTimeMillis
  688. }
  689. }
  690. if 0 == v.UpdatedAt {
  691. v.UpdatedAt = v.CreatedAt + 1000
  692. }
  693. }
  694. }
  695. // 补全过滤器 Value
  696. if nil != view.Table {
  697. for _, f := range view.Table.Filters {
  698. if nil != f.Value {
  699. continue
  700. }
  701. if k, _ := attrView.GetKey(f.Column); nil != k {
  702. f.Value = &av.Value{Type: k.Type}
  703. }
  704. }
  705. }
  706. switch view.LayoutType {
  707. case av.LayoutTypeTable:
  708. // 列删除以后需要删除设置的过滤和排序
  709. tmpFilters := []*av.ViewFilter{}
  710. for _, f := range view.Table.Filters {
  711. if k, _ := attrView.GetKey(f.Column); nil != k {
  712. tmpFilters = append(tmpFilters, f)
  713. }
  714. }
  715. view.Table.Filters = tmpFilters
  716. tmpSorts := []*av.ViewSort{}
  717. for _, s := range view.Table.Sorts {
  718. if k, _ := attrView.GetKey(s.Column); nil != k {
  719. tmpSorts = append(tmpSorts, s)
  720. }
  721. }
  722. view.Table.Sorts = tmpSorts
  723. viewable, err = renderAttributeViewTable(attrView, view, query)
  724. }
  725. viewable.FilterRows(attrView)
  726. viewable.SortRows(attrView)
  727. viewable.CalcCols()
  728. // 分页
  729. switch viewable.GetType() {
  730. case av.LayoutTypeTable:
  731. table := viewable.(*av.Table)
  732. table.RowCount = len(table.Rows)
  733. if 1 > view.Table.PageSize {
  734. view.Table.PageSize = 50
  735. }
  736. table.PageSize = view.Table.PageSize
  737. if 1 > pageSize {
  738. pageSize = table.PageSize
  739. }
  740. start := (page - 1) * pageSize
  741. end := start + pageSize
  742. if len(table.Rows) < end {
  743. end = len(table.Rows)
  744. }
  745. table.Rows = table.Rows[start:end]
  746. }
  747. return
  748. }
  749. func renderTemplateCol(ial map[string]string, flashcard *Flashcard, rowValues []*av.KeyValues, tplContent string) string {
  750. if "" == ial["id"] {
  751. block := getRowBlockValue(rowValues)
  752. if nil != block && nil != block.Block {
  753. ial["id"] = block.Block.ID
  754. }
  755. }
  756. if "" == ial["updated"] {
  757. block := getRowBlockValue(rowValues)
  758. if nil != block && nil != block.Block {
  759. ial["updated"] = time.UnixMilli(block.Block.Updated).Format("20060102150405")
  760. }
  761. }
  762. goTpl := template.New("").Delims(".action{", "}")
  763. tplFuncMap := util.BuiltInTemplateFuncs()
  764. SQLTemplateFuncs(&tplFuncMap)
  765. goTpl = goTpl.Funcs(tplFuncMap)
  766. tpl, tplErr := goTpl.Parse(tplContent)
  767. if nil != tplErr {
  768. logging.LogWarnf("parse template [%s] failed: %s", tplContent, tplErr)
  769. return ""
  770. }
  771. buf := &bytes.Buffer{}
  772. dataModel := map[string]interface{}{} // 复制一份 IAL 以避免修改原始数据
  773. for k, v := range ial {
  774. dataModel[k] = v
  775. // Database template column supports `created` and `updated` built-in variables https://github.com/siyuan-note/siyuan/issues/9364
  776. createdStr := ial["id"]
  777. if "" != createdStr {
  778. createdStr = createdStr[:len("20060102150405")]
  779. }
  780. created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local)
  781. if nil == parseErr {
  782. dataModel["created"] = created
  783. } else {
  784. logging.LogWarnf("parse created [%s] failed: %s", createdStr, parseErr)
  785. dataModel["created"] = time.Now()
  786. }
  787. updatedStr := ial["updated"]
  788. updated, parseErr := time.ParseInLocation("20060102150405", updatedStr, time.Local)
  789. if nil == parseErr {
  790. dataModel["updated"] = updated
  791. } else {
  792. dataModel["updated"] = time.Now()
  793. }
  794. }
  795. if nil != flashcard {
  796. dataModel["flashcard"] = flashcard
  797. }
  798. for _, rowValue := range rowValues {
  799. if 0 < len(rowValue.Values) {
  800. v := rowValue.Values[0]
  801. if av.KeyTypeNumber == v.Type {
  802. if nil != v.Number && v.Number.IsNotEmpty {
  803. dataModel[rowValue.Key.Name] = v.Number.Content
  804. }
  805. } else if av.KeyTypeDate == v.Type {
  806. if nil != v.Date && v.Date.IsNotEmpty {
  807. dataModel[rowValue.Key.Name] = time.UnixMilli(v.Date.Content)
  808. }
  809. } else if av.KeyTypeRollup == v.Type {
  810. if 0 < len(v.Rollup.Contents) && av.KeyTypeNumber == v.Rollup.Contents[0].Type {
  811. // 模板使用汇总时支持数字计算
  812. // Template supports numerical calculations when using rollup https://github.com/siyuan-note/siyuan/issues/10810
  813. // 汇总数字时仅取第一个数字填充模板
  814. dataModel[rowValue.Key.Name] = v.Rollup.Contents[0].Number.Content
  815. }
  816. } else {
  817. dataModel[rowValue.Key.Name] = v.String()
  818. }
  819. }
  820. }
  821. if err := tpl.Execute(buf, dataModel); nil != err {
  822. logging.LogWarnf("execute template [%s] failed: %s", tplContent, err)
  823. }
  824. return buf.String()
  825. }
  826. func renderAttributeViewTable(attrView *av.AttributeView, view *av.View, query string) (ret *av.Table, err error) {
  827. ret = &av.Table{
  828. ID: view.ID,
  829. Icon: view.Icon,
  830. Name: view.Name,
  831. HideAttrViewName: view.HideAttrViewName,
  832. Columns: []*av.TableColumn{},
  833. Rows: []*av.TableRow{},
  834. Filters: view.Table.Filters,
  835. Sorts: view.Table.Sorts,
  836. }
  837. // 组装列
  838. for _, col := range view.Table.Columns {
  839. key, getErr := attrView.GetKey(col.ID)
  840. if nil != getErr {
  841. err = getErr
  842. return
  843. }
  844. ret.Columns = append(ret.Columns, &av.TableColumn{
  845. ID: key.ID,
  846. Name: key.Name,
  847. Type: key.Type,
  848. Icon: key.Icon,
  849. Options: key.Options,
  850. NumberFormat: key.NumberFormat,
  851. Template: key.Template,
  852. Relation: key.Relation,
  853. Rollup: key.Rollup,
  854. Date: key.Date,
  855. Wrap: col.Wrap,
  856. Hidden: col.Hidden,
  857. Width: col.Width,
  858. Pin: col.Pin,
  859. Calc: col.Calc,
  860. })
  861. }
  862. // 生成行
  863. rows := map[string][]*av.KeyValues{}
  864. for _, keyValues := range attrView.KeyValues {
  865. for _, val := range keyValues.Values {
  866. values := rows[val.BlockID]
  867. if nil == values {
  868. values = []*av.KeyValues{{Key: keyValues.Key, Values: []*av.Value{val}}}
  869. } else {
  870. values = append(values, &av.KeyValues{Key: keyValues.Key, Values: []*av.Value{val}})
  871. }
  872. rows[val.BlockID] = values
  873. }
  874. // 数据订正,补全关联
  875. if av.KeyTypeRelation == keyValues.Key.Type && nil != keyValues.Key.Relation {
  876. av.UpsertAvBackRel(attrView.ID, keyValues.Key.Relation.AvID)
  877. if keyValues.Key.Relation.IsTwoWay {
  878. av.UpsertAvBackRel(keyValues.Key.Relation.AvID, attrView.ID)
  879. }
  880. }
  881. }
  882. // 过滤掉不存在的行
  883. var notFound []string
  884. for blockID, keyValues := range rows {
  885. blockValue := getRowBlockValue(keyValues)
  886. if nil == blockValue {
  887. notFound = append(notFound, blockID)
  888. continue
  889. }
  890. if blockValue.IsDetached {
  891. continue
  892. }
  893. if nil != blockValue.Block && "" == blockValue.Block.ID {
  894. notFound = append(notFound, blockID)
  895. continue
  896. }
  897. if nil == treenode.GetBlockTree(blockID) {
  898. notFound = append(notFound, blockID)
  899. }
  900. }
  901. for _, blockID := range notFound {
  902. delete(rows, blockID)
  903. }
  904. // 生成行单元格
  905. for rowID, row := range rows {
  906. var tableRow av.TableRow
  907. for _, col := range ret.Columns {
  908. var tableCell *av.TableCell
  909. for _, keyValues := range row {
  910. if keyValues.Key.ID == col.ID {
  911. tableCell = &av.TableCell{
  912. ID: keyValues.Values[0].ID,
  913. Value: keyValues.Values[0],
  914. ValueType: col.Type,
  915. }
  916. break
  917. }
  918. }
  919. if nil == tableCell {
  920. tableCell = &av.TableCell{
  921. ID: ast.NewNodeID(),
  922. ValueType: col.Type,
  923. }
  924. }
  925. tableRow.ID = rowID
  926. switch tableCell.ValueType {
  927. case av.KeyTypeNumber: // 格式化数字
  928. if nil != tableCell.Value && nil != tableCell.Value.Number && tableCell.Value.Number.IsNotEmpty {
  929. tableCell.Value.Number.Format = col.NumberFormat
  930. tableCell.Value.Number.FormatNumber()
  931. }
  932. case av.KeyTypeTemplate: // 渲染模板列
  933. tableCell.Value = &av.Value{ID: tableCell.ID, KeyID: col.ID, BlockID: rowID, Type: av.KeyTypeTemplate, Template: &av.ValueTemplate{Content: col.Template}}
  934. case av.KeyTypeCreated: // 填充创建时间列值,后面再渲染
  935. tableCell.Value = &av.Value{ID: tableCell.ID, KeyID: col.ID, BlockID: rowID, Type: av.KeyTypeCreated}
  936. case av.KeyTypeUpdated: // 填充更新时间列值,后面再渲染
  937. tableCell.Value = &av.Value{ID: tableCell.ID, KeyID: col.ID, BlockID: rowID, Type: av.KeyTypeUpdated}
  938. case av.KeyTypeRelation: // 清空关联列值,后面再渲染 https://ld246.com/article/1703831044435
  939. if nil != tableCell.Value && nil != tableCell.Value.Relation {
  940. tableCell.Value.Relation.Contents = nil
  941. }
  942. }
  943. treenode.FillAttributeViewTableCellNilValue(tableCell, rowID, col.ID)
  944. tableRow.Cells = append(tableRow.Cells, tableCell)
  945. }
  946. ret.Rows = append(ret.Rows, &tableRow)
  947. }
  948. // 渲染自动生成的列值,比如关联列、汇总列、创建时间列和更新时间列
  949. for _, row := range ret.Rows {
  950. for _, cell := range row.Cells {
  951. switch cell.ValueType {
  952. case av.KeyTypeRollup: // 渲染汇总列
  953. rollupKey, _ := attrView.GetKey(cell.Value.KeyID)
  954. if nil == rollupKey || nil == rollupKey.Rollup {
  955. break
  956. }
  957. relKey, _ := attrView.GetKey(rollupKey.Rollup.RelationKeyID)
  958. if nil == relKey || nil == relKey.Relation {
  959. break
  960. }
  961. relVal := attrView.GetValue(relKey.ID, row.ID)
  962. if nil == relVal || nil == relVal.Relation {
  963. break
  964. }
  965. destAv, _ := av.ParseAttributeView(relKey.Relation.AvID)
  966. if nil == destAv {
  967. break
  968. }
  969. destKey, _ := destAv.GetKey(rollupKey.Rollup.KeyID)
  970. if nil == destKey {
  971. continue
  972. }
  973. for _, blockID := range relVal.Relation.BlockIDs {
  974. destVal := destAv.GetValue(rollupKey.Rollup.KeyID, blockID)
  975. if nil == destVal {
  976. if destAv.ExistBlock(blockID) { // 数据库中存在行但是列值不存在是数据未初始化,这里补一个默认值
  977. destVal = treenode.GetAttributeViewDefaultValue(ast.NewNodeID(), rollupKey.Rollup.KeyID, blockID, destKey.Type)
  978. }
  979. if nil == destVal {
  980. continue
  981. }
  982. }
  983. if av.KeyTypeNumber == destKey.Type {
  984. destVal.Number.Format = destKey.NumberFormat
  985. destVal.Number.FormatNumber()
  986. }
  987. cell.Value.Rollup.Contents = append(cell.Value.Rollup.Contents, destVal.Clone())
  988. }
  989. cell.Value.Rollup.RenderContents(rollupKey.Rollup.Calc, destKey)
  990. // 将汇总列的值保存到 rows 中,后续渲染模板列的时候会用到,下同
  991. // Database table view template columns support reading relation, rollup, created and updated columns https://github.com/siyuan-note/siyuan/issues/10442
  992. keyValues := rows[row.ID]
  993. keyValues = append(keyValues, &av.KeyValues{Key: rollupKey, Values: []*av.Value{{ID: cell.Value.ID, KeyID: rollupKey.ID, BlockID: row.ID, Type: av.KeyTypeRollup, Rollup: cell.Value.Rollup}}})
  994. rows[row.ID] = keyValues
  995. case av.KeyTypeRelation: // 渲染关联列
  996. relKey, _ := attrView.GetKey(cell.Value.KeyID)
  997. if nil != relKey && nil != relKey.Relation {
  998. destAv, _ := av.ParseAttributeView(relKey.Relation.AvID)
  999. if nil != destAv {
  1000. blocks := map[string]*av.Value{}
  1001. for _, blockValue := range destAv.GetBlockKeyValues().Values {
  1002. blocks[blockValue.BlockID] = blockValue
  1003. }
  1004. for _, blockID := range cell.Value.Relation.BlockIDs {
  1005. if val := blocks[blockID]; nil != val {
  1006. cell.Value.Relation.Contents = append(cell.Value.Relation.Contents, val)
  1007. }
  1008. }
  1009. }
  1010. }
  1011. keyValues := rows[row.ID]
  1012. keyValues = append(keyValues, &av.KeyValues{Key: relKey, Values: []*av.Value{{ID: cell.Value.ID, KeyID: relKey.ID, BlockID: row.ID, Type: av.KeyTypeRelation, Relation: cell.Value.Relation}}})
  1013. rows[row.ID] = keyValues
  1014. case av.KeyTypeCreated: // 渲染创建时间
  1015. createdStr := row.ID[:len("20060102150405")]
  1016. created, parseErr := time.ParseInLocation("20060102150405", createdStr, time.Local)
  1017. if nil == parseErr {
  1018. cell.Value.Created = av.NewFormattedValueCreated(created.UnixMilli(), 0, av.CreatedFormatNone)
  1019. cell.Value.Created.IsNotEmpty = true
  1020. } else {
  1021. cell.Value.Created = av.NewFormattedValueCreated(time.Now().UnixMilli(), 0, av.CreatedFormatNone)
  1022. }
  1023. keyValues := rows[row.ID]
  1024. createdKey, _ := attrView.GetKey(cell.Value.KeyID)
  1025. keyValues = append(keyValues, &av.KeyValues{Key: createdKey, Values: []*av.Value{{ID: cell.Value.ID, KeyID: createdKey.ID, BlockID: row.ID, Type: av.KeyTypeCreated, Created: cell.Value.Created}}})
  1026. rows[row.ID] = keyValues
  1027. case av.KeyTypeUpdated: // 渲染更新时间
  1028. ial := map[string]string{}
  1029. block := row.GetBlockValue()
  1030. if nil != block && !block.IsDetached {
  1031. ial = GetBlockAttrsWithoutWaitWriting(row.ID)
  1032. }
  1033. updatedStr := ial["updated"]
  1034. if "" == updatedStr && nil != block {
  1035. cell.Value.Updated = av.NewFormattedValueUpdated(block.Block.Updated, 0, av.UpdatedFormatNone)
  1036. cell.Value.Updated.IsNotEmpty = true
  1037. } else {
  1038. updated, parseErr := time.ParseInLocation("20060102150405", updatedStr, time.Local)
  1039. if nil == parseErr {
  1040. cell.Value.Updated = av.NewFormattedValueUpdated(updated.UnixMilli(), 0, av.UpdatedFormatNone)
  1041. cell.Value.Updated.IsNotEmpty = true
  1042. } else {
  1043. cell.Value.Updated = av.NewFormattedValueUpdated(time.Now().UnixMilli(), 0, av.UpdatedFormatNone)
  1044. }
  1045. }
  1046. keyValues := rows[row.ID]
  1047. updatedKey, _ := attrView.GetKey(cell.Value.KeyID)
  1048. keyValues = append(keyValues, &av.KeyValues{Key: updatedKey, Values: []*av.Value{{ID: cell.Value.ID, KeyID: updatedKey.ID, BlockID: row.ID, Type: av.KeyTypeUpdated, Updated: cell.Value.Updated}}})
  1049. rows[row.ID] = keyValues
  1050. }
  1051. }
  1052. }
  1053. // 最后单独渲染模板列,这样模板列就可以使用汇总、关联、创建时间和更新时间列的值了
  1054. // Database table view template columns support reading relation, rollup, created and updated columns https://github.com/siyuan-note/siyuan/issues/10442
  1055. // 获取闪卡信息
  1056. flashcards := map[string]*Flashcard{}
  1057. //deck := Decks[builtinDeckID]
  1058. //if nil != deck {
  1059. // var blockIDs []string
  1060. // for _, row := range ret.Rows {
  1061. // blockIDs = append(blockIDs, row.ID)
  1062. // }
  1063. // cards := deck.GetCardsByBlockIDs(blockIDs)
  1064. // now := time.Now()
  1065. // for _, card := range cards {
  1066. // flashcards[card.BlockID()] = newFlashcard(card, builtinDeckID, now)
  1067. // }
  1068. //}
  1069. for _, row := range ret.Rows {
  1070. for _, cell := range row.Cells {
  1071. switch cell.ValueType {
  1072. case av.KeyTypeTemplate: // 渲染模板列
  1073. keyValues := rows[row.ID]
  1074. ial := map[string]string{}
  1075. block := row.GetBlockValue()
  1076. if nil != block && !block.IsDetached {
  1077. ial = GetBlockAttrsWithoutWaitWriting(row.ID)
  1078. }
  1079. content := renderTemplateCol(ial, flashcards[row.ID], keyValues, cell.Value.Template.Content)
  1080. cell.Value.Template.Content = content
  1081. }
  1082. }
  1083. }
  1084. // 根据搜索条件过滤
  1085. query = strings.TrimSpace(query)
  1086. if "" != query {
  1087. keywords := strings.Split(query, " ")
  1088. var hitRows []*av.TableRow
  1089. for _, row := range ret.Rows {
  1090. hit := false
  1091. for _, cell := range row.Cells {
  1092. for _, keyword := range keywords {
  1093. if strings.Contains(strings.ToLower(cell.Value.String()), strings.ToLower(keyword)) {
  1094. hit = true
  1095. break
  1096. }
  1097. }
  1098. }
  1099. if hit {
  1100. hitRows = append(hitRows, row)
  1101. }
  1102. }
  1103. ret.Rows = hitRows
  1104. if 1 > len(ret.Rows) {
  1105. ret.Rows = []*av.TableRow{}
  1106. }
  1107. }
  1108. // 自定义排序
  1109. sortRowIDs := map[string]int{}
  1110. if 0 < len(view.Table.RowIDs) {
  1111. for i, rowID := range view.Table.RowIDs {
  1112. sortRowIDs[rowID] = i
  1113. }
  1114. }
  1115. sort.Slice(ret.Rows, func(i, j int) bool {
  1116. iv := sortRowIDs[ret.Rows[i].ID]
  1117. jv := sortRowIDs[ret.Rows[j].ID]
  1118. if iv == jv {
  1119. return ret.Rows[i].ID < ret.Rows[j].ID
  1120. }
  1121. return iv < jv
  1122. })
  1123. return
  1124. }
  1125. func getRowBlockValue(keyValues []*av.KeyValues) (ret *av.Value) {
  1126. for _, kv := range keyValues {
  1127. if av.KeyTypeBlock == kv.Key.Type && 0 < len(kv.Values) {
  1128. ret = kv.Values[0]
  1129. break
  1130. }
  1131. }
  1132. return
  1133. }
  1134. func (tx *Transaction) doSetAttrViewColDate(operation *Operation) (ret *TxErr) {
  1135. err := setAttributeViewColDate(operation)
  1136. if nil != err {
  1137. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1138. }
  1139. return
  1140. }
  1141. func setAttributeViewColDate(operation *Operation) (err error) {
  1142. attrView, err := av.ParseAttributeView(operation.AvID)
  1143. if nil != err {
  1144. return
  1145. }
  1146. keyID := operation.ID
  1147. key, _ := attrView.GetKey(keyID)
  1148. if nil == key || av.KeyTypeDate != key.Type {
  1149. return
  1150. }
  1151. if nil == key.Date {
  1152. key.Date = &av.Date{}
  1153. }
  1154. key.Date.AutoFillNow = operation.Data.(bool)
  1155. err = av.SaveAttributeView(attrView)
  1156. return
  1157. }
  1158. func (tx *Transaction) doHideAttrViewName(operation *Operation) (ret *TxErr) {
  1159. err := hideAttrViewName(operation)
  1160. if nil != err {
  1161. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1162. }
  1163. return
  1164. }
  1165. func hideAttrViewName(operation *Operation) (err error) {
  1166. attrView, err := av.ParseAttributeView(operation.AvID)
  1167. if nil != err {
  1168. logging.LogErrorf("parse attribute view [%s] failed: %s", operation.AvID, err)
  1169. return
  1170. }
  1171. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  1172. if nil == view {
  1173. logging.LogErrorf("get view [%s] failed: %s", operation.BlockID, err)
  1174. return
  1175. }
  1176. view.HideAttrViewName = operation.Data.(bool)
  1177. err = av.SaveAttributeView(attrView)
  1178. return
  1179. }
  1180. func (tx *Transaction) doUpdateAttrViewColRollup(operation *Operation) (ret *TxErr) {
  1181. err := updateAttributeViewColRollup(operation)
  1182. if nil != err {
  1183. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1184. }
  1185. return
  1186. }
  1187. func updateAttributeViewColRollup(operation *Operation) (err error) {
  1188. // operation.AvID 汇总列所在 av
  1189. // operation.ID 汇总列 ID
  1190. // operation.ParentID 汇总列基于的关联列 ID
  1191. // operation.KeyID 目标列 ID
  1192. // operation.Data 计算方式
  1193. attrView, err := av.ParseAttributeView(operation.AvID)
  1194. if nil != err {
  1195. return
  1196. }
  1197. rollUpKey, _ := attrView.GetKey(operation.ID)
  1198. if nil == rollUpKey {
  1199. return
  1200. }
  1201. rollUpKey.Rollup = &av.Rollup{
  1202. RelationKeyID: operation.ParentID,
  1203. KeyID: operation.KeyID,
  1204. }
  1205. if nil != operation.Data {
  1206. data := operation.Data.(map[string]interface{})
  1207. if nil != data["calc"] {
  1208. calcData, jsonErr := gulu.JSON.MarshalJSON(data["calc"])
  1209. if nil != jsonErr {
  1210. err = jsonErr
  1211. return
  1212. }
  1213. if jsonErr = gulu.JSON.UnmarshalJSON(calcData, &rollUpKey.Rollup.Calc); nil != jsonErr {
  1214. err = jsonErr
  1215. return
  1216. }
  1217. }
  1218. }
  1219. err = av.SaveAttributeView(attrView)
  1220. return
  1221. }
  1222. func (tx *Transaction) doUpdateAttrViewColRelation(operation *Operation) (ret *TxErr) {
  1223. err := updateAttributeViewColRelation(operation)
  1224. if nil != err {
  1225. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1226. }
  1227. return
  1228. }
  1229. func updateAttributeViewColRelation(operation *Operation) (err error) {
  1230. // operation.AvID 源 avID
  1231. // operation.ID 目标 avID
  1232. // operation.KeyID 源 av 关联列 ID
  1233. // operation.IsTwoWay 是否双向关联
  1234. // operation.BackRelationKeyID 双向关联的目标关联列 ID
  1235. // operation.Name 双向关联的目标关联列名称
  1236. // operation.Format 源 av 关联列名称
  1237. srcAv, err := av.ParseAttributeView(operation.AvID)
  1238. if nil != err {
  1239. return
  1240. }
  1241. destAv, err := av.ParseAttributeView(operation.ID)
  1242. if nil != err {
  1243. return
  1244. }
  1245. isSameAv := srcAv.ID == destAv.ID
  1246. if isSameAv {
  1247. destAv = srcAv
  1248. }
  1249. for _, keyValues := range srcAv.KeyValues {
  1250. if keyValues.Key.ID != operation.KeyID {
  1251. continue
  1252. }
  1253. srcRel := keyValues.Key.Relation
  1254. // 已经设置过双向关联的话需要先断开双向关联
  1255. if nil != srcRel {
  1256. if srcRel.IsTwoWay {
  1257. oldDestAv, _ := av.ParseAttributeView(srcRel.AvID)
  1258. if nil != oldDestAv {
  1259. isOldSameAv := oldDestAv.ID == destAv.ID
  1260. if isOldSameAv {
  1261. oldDestAv = destAv
  1262. }
  1263. oldDestKey, _ := oldDestAv.GetKey(srcRel.BackKeyID)
  1264. if nil != oldDestKey && nil != oldDestKey.Relation && oldDestKey.Relation.AvID == srcAv.ID && oldDestKey.Relation.IsTwoWay {
  1265. oldDestKey.Relation.IsTwoWay = false
  1266. oldDestKey.Relation.BackKeyID = ""
  1267. }
  1268. if !isOldSameAv {
  1269. err = av.SaveAttributeView(oldDestAv)
  1270. if nil != err {
  1271. return
  1272. }
  1273. }
  1274. }
  1275. }
  1276. av.RemoveAvRel(srcAv.ID, srcRel.AvID)
  1277. }
  1278. srcRel = &av.Relation{
  1279. AvID: operation.ID,
  1280. IsTwoWay: operation.IsTwoWay,
  1281. }
  1282. if operation.IsTwoWay {
  1283. srcRel.BackKeyID = operation.BackRelationKeyID
  1284. } else {
  1285. srcRel.BackKeyID = ""
  1286. }
  1287. keyValues.Key.Relation = srcRel
  1288. keyValues.Key.Name = operation.Format
  1289. break
  1290. }
  1291. destAdded := false
  1292. backRelKey, _ := destAv.GetKey(operation.BackRelationKeyID)
  1293. if nil != backRelKey {
  1294. backRelKey.Relation = &av.Relation{
  1295. AvID: operation.AvID,
  1296. IsTwoWay: operation.IsTwoWay,
  1297. BackKeyID: operation.KeyID,
  1298. }
  1299. destAdded = true
  1300. if operation.IsTwoWay {
  1301. name := strings.TrimSpace(operation.Name)
  1302. if "" == name {
  1303. name = srcAv.Name + " " + operation.Format
  1304. }
  1305. backRelKey.Name = strings.TrimSpace(name)
  1306. }
  1307. }
  1308. if !destAdded {
  1309. if operation.IsTwoWay {
  1310. name := strings.TrimSpace(operation.Name)
  1311. if "" == name {
  1312. name = srcAv.Name + " " + operation.Format
  1313. }
  1314. destAv.KeyValues = append(destAv.KeyValues, &av.KeyValues{
  1315. Key: &av.Key{
  1316. ID: operation.BackRelationKeyID,
  1317. Name: name,
  1318. Type: av.KeyTypeRelation,
  1319. Relation: &av.Relation{AvID: operation.AvID, IsTwoWay: operation.IsTwoWay, BackKeyID: operation.KeyID},
  1320. },
  1321. })
  1322. for _, v := range destAv.Views {
  1323. switch v.LayoutType {
  1324. case av.LayoutTypeTable:
  1325. v.Table.Columns = append(v.Table.Columns, &av.ViewTableColumn{ID: operation.BackRelationKeyID})
  1326. }
  1327. }
  1328. }
  1329. }
  1330. err = av.SaveAttributeView(srcAv)
  1331. if nil != err {
  1332. return
  1333. }
  1334. if !isSameAv {
  1335. err = av.SaveAttributeView(destAv)
  1336. util.BroadcastByType("protyle", "refreshAttributeView", 0, "", map[string]interface{}{"id": destAv.ID})
  1337. }
  1338. av.UpsertAvBackRel(srcAv.ID, destAv.ID)
  1339. return
  1340. }
  1341. func (tx *Transaction) doSortAttrViewView(operation *Operation) (ret *TxErr) {
  1342. avID := operation.AvID
  1343. attrView, err := av.ParseAttributeView(avID)
  1344. if nil != err {
  1345. logging.LogErrorf("parse attribute view [%s] failed: %s", operation.AvID, err)
  1346. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1347. }
  1348. view := attrView.GetView(operation.ID)
  1349. if nil == view {
  1350. logging.LogErrorf("get view failed: %s", operation.BlockID)
  1351. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1352. }
  1353. viewID := view.ID
  1354. previousViewID := operation.PreviousID
  1355. if viewID == previousViewID {
  1356. return
  1357. }
  1358. var index, previousIndex int
  1359. for i, v := range attrView.Views {
  1360. if v.ID == viewID {
  1361. view = v
  1362. index = i
  1363. break
  1364. }
  1365. }
  1366. if nil == view {
  1367. return
  1368. }
  1369. attrView.Views = append(attrView.Views[:index], attrView.Views[index+1:]...)
  1370. for i, v := range attrView.Views {
  1371. if v.ID == previousViewID {
  1372. previousIndex = i + 1
  1373. break
  1374. }
  1375. }
  1376. attrView.Views = util.InsertElem(attrView.Views, previousIndex, view)
  1377. if err = av.SaveAttributeView(attrView); nil != err {
  1378. logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
  1379. return &TxErr{code: TxErrCodeWriteTree, msg: err.Error(), id: avID}
  1380. }
  1381. return
  1382. }
  1383. func (tx *Transaction) doRemoveAttrViewView(operation *Operation) (ret *TxErr) {
  1384. var err error
  1385. avID := operation.AvID
  1386. attrView, err := av.ParseAttributeView(avID)
  1387. if nil != err {
  1388. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  1389. return &TxErr{code: TxErrCodeBlockNotFound, id: avID}
  1390. }
  1391. if 1 >= len(attrView.Views) {
  1392. logging.LogWarnf("can't remove last view [%s] of attribute view [%s]", operation.AvID, avID)
  1393. return
  1394. }
  1395. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  1396. if nil == view {
  1397. logging.LogWarnf("get view failed: %s", operation.BlockID)
  1398. return
  1399. }
  1400. viewID := view.ID
  1401. var index int
  1402. for i, view := range attrView.Views {
  1403. if viewID == view.ID {
  1404. attrView.Views = append(attrView.Views[:i], attrView.Views[i+1:]...)
  1405. index = i - 1
  1406. break
  1407. }
  1408. }
  1409. if 0 > index {
  1410. index = 0
  1411. }
  1412. attrView.ViewID = attrView.Views[index].ID
  1413. if err = av.SaveAttributeView(attrView); nil != err {
  1414. logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
  1415. return &TxErr{code: TxErrCodeWriteTree, msg: err.Error(), id: avID}
  1416. }
  1417. trees, nodes := getMirrorBlocksNodes(avID)
  1418. for _, node := range nodes {
  1419. attrs := parse.IAL2Map(node.KramdownIAL)
  1420. blockViewID := attrs[av.NodeAttrView]
  1421. if blockViewID == viewID {
  1422. attrs[av.NodeAttrView] = attrView.ViewID
  1423. oldAttrs, e := setNodeAttrs0(node, attrs)
  1424. if nil != e {
  1425. logging.LogErrorf("set node attrs failed: %s", e)
  1426. continue
  1427. }
  1428. cache.PutBlockIAL(node.ID, parse.IAL2Map(node.KramdownIAL))
  1429. pushBroadcastAttrTransactions(oldAttrs, node)
  1430. }
  1431. }
  1432. for _, tree := range trees {
  1433. if err = indexWriteJSONQueue(tree); nil != err {
  1434. return
  1435. }
  1436. }
  1437. return
  1438. }
  1439. func getMirrorBlocksNodes(avID string) (trees []*parse.Tree, nodes []*ast.Node) {
  1440. mirrorBlocks := treenode.GetMirrorAttrViewBlockIDs(avID)
  1441. mirrorBlockTree := map[string]*parse.Tree{}
  1442. treeMap := map[string]*parse.Tree{}
  1443. for _, mirrorBlock := range mirrorBlocks {
  1444. bt := treenode.GetBlockTree(mirrorBlock)
  1445. if nil == bt {
  1446. logging.LogErrorf("get block tree by block ID [%s] failed", mirrorBlock)
  1447. continue
  1448. }
  1449. tree := mirrorBlockTree[mirrorBlock]
  1450. if nil == tree {
  1451. tree, _ = LoadTreeByBlockID(mirrorBlock)
  1452. if nil == tree {
  1453. logging.LogErrorf("load tree by block ID [%s] failed", mirrorBlock)
  1454. continue
  1455. }
  1456. treeMap[tree.ID] = tree
  1457. mirrorBlockTree[mirrorBlock] = tree
  1458. }
  1459. }
  1460. for _, mirrorBlock := range mirrorBlocks {
  1461. tree := mirrorBlockTree[mirrorBlock]
  1462. node := treenode.GetNodeInTree(tree, mirrorBlock)
  1463. if nil == node {
  1464. logging.LogErrorf("get node in tree by block ID [%s] failed", mirrorBlock)
  1465. continue
  1466. }
  1467. nodes = append(nodes, node)
  1468. }
  1469. for _, tree := range treeMap {
  1470. trees = append(trees, tree)
  1471. }
  1472. return
  1473. }
  1474. func (tx *Transaction) doDuplicateAttrViewView(operation *Operation) (ret *TxErr) {
  1475. var err error
  1476. avID := operation.AvID
  1477. attrView, err := av.ParseAttributeView(avID)
  1478. if nil != err {
  1479. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  1480. return &TxErr{code: TxErrWriteAttributeView, id: avID}
  1481. }
  1482. masterView := attrView.GetView(operation.PreviousID)
  1483. if nil == masterView {
  1484. logging.LogErrorf("get master view failed: %s", avID)
  1485. return &TxErr{code: TxErrWriteAttributeView, id: avID}
  1486. }
  1487. node, tree, _ := getNodeByBlockID(nil, operation.BlockID)
  1488. if nil == node {
  1489. logging.LogErrorf("get node by block ID [%s] failed", operation.BlockID)
  1490. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID}
  1491. }
  1492. attrs := parse.IAL2Map(node.KramdownIAL)
  1493. attrs[av.NodeAttrView] = operation.ID
  1494. err = setNodeAttrs(node, tree, attrs)
  1495. if nil != err {
  1496. logging.LogWarnf("set node [%s] attrs failed: %s", operation.BlockID, err)
  1497. return
  1498. }
  1499. view := av.NewTableView()
  1500. view.ID = operation.ID
  1501. attrView.Views = append(attrView.Views, view)
  1502. attrView.ViewID = view.ID
  1503. view.Icon = masterView.Icon
  1504. view.Name = attrView.GetDuplicateViewName(masterView.Name)
  1505. view.LayoutType = masterView.LayoutType
  1506. view.HideAttrViewName = masterView.HideAttrViewName
  1507. for _, col := range masterView.Table.Columns {
  1508. view.Table.Columns = append(view.Table.Columns, &av.ViewTableColumn{
  1509. ID: col.ID,
  1510. Wrap: col.Wrap,
  1511. Hidden: col.Hidden,
  1512. Pin: col.Pin,
  1513. Width: col.Width,
  1514. Calc: col.Calc,
  1515. })
  1516. }
  1517. for _, filter := range masterView.Table.Filters {
  1518. view.Table.Filters = append(view.Table.Filters, &av.ViewFilter{
  1519. Column: filter.Column,
  1520. Operator: filter.Operator,
  1521. Value: filter.Value,
  1522. RelativeDate: filter.RelativeDate,
  1523. RelativeDate2: filter.RelativeDate2,
  1524. })
  1525. }
  1526. for _, s := range masterView.Table.Sorts {
  1527. view.Table.Sorts = append(view.Table.Sorts, &av.ViewSort{
  1528. Column: s.Column,
  1529. Order: s.Order,
  1530. })
  1531. }
  1532. view.Table.PageSize = masterView.Table.PageSize
  1533. view.Table.RowIDs = masterView.Table.RowIDs
  1534. if err = av.SaveAttributeView(attrView); nil != err {
  1535. logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
  1536. return &TxErr{code: TxErrWriteAttributeView, msg: err.Error(), id: avID}
  1537. }
  1538. return
  1539. }
  1540. func (tx *Transaction) doAddAttrViewView(operation *Operation) (ret *TxErr) {
  1541. var err error
  1542. avID := operation.AvID
  1543. attrView, err := av.ParseAttributeView(avID)
  1544. if nil != err {
  1545. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  1546. return &TxErr{code: TxErrWriteAttributeView, id: avID}
  1547. }
  1548. if 1 > len(attrView.Views) {
  1549. logging.LogErrorf("no view in attribute view [%s]", avID)
  1550. return &TxErr{code: TxErrWriteAttributeView, id: avID}
  1551. }
  1552. firstView := attrView.Views[0]
  1553. if nil == firstView {
  1554. logging.LogErrorf("get first view failed: %s", avID)
  1555. return &TxErr{code: TxErrWriteAttributeView, id: avID}
  1556. }
  1557. node, tree, _ := getNodeByBlockID(nil, operation.BlockID)
  1558. if nil == node {
  1559. logging.LogErrorf("get node by block ID [%s] failed", operation.BlockID)
  1560. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID}
  1561. }
  1562. attrs := parse.IAL2Map(node.KramdownIAL)
  1563. attrs[av.NodeAttrView] = operation.ID
  1564. err = setNodeAttrs(node, tree, attrs)
  1565. if nil != err {
  1566. logging.LogWarnf("set node [%s] attrs failed: %s", operation.BlockID, err)
  1567. return
  1568. }
  1569. view := av.NewTableView()
  1570. view.ID = operation.ID
  1571. attrView.Views = append(attrView.Views, view)
  1572. attrView.ViewID = view.ID
  1573. for _, col := range firstView.Table.Columns {
  1574. view.Table.Columns = append(view.Table.Columns, &av.ViewTableColumn{ID: col.ID})
  1575. }
  1576. view.Table.RowIDs = firstView.Table.RowIDs
  1577. if err = av.SaveAttributeView(attrView); nil != err {
  1578. logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
  1579. return &TxErr{code: TxErrWriteAttributeView, msg: err.Error(), id: avID}
  1580. }
  1581. return
  1582. }
  1583. func (tx *Transaction) doSetAttrViewViewName(operation *Operation) (ret *TxErr) {
  1584. var err error
  1585. avID := operation.AvID
  1586. attrView, err := av.ParseAttributeView(avID)
  1587. if nil != err {
  1588. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  1589. return &TxErr{code: TxErrWriteAttributeView, id: avID}
  1590. }
  1591. viewID := operation.ID
  1592. view := attrView.GetView(viewID)
  1593. if nil == view {
  1594. logging.LogErrorf("get view [%s] failed: %s", viewID, err)
  1595. return &TxErr{code: TxErrWriteAttributeView, id: viewID}
  1596. }
  1597. view.Name = strings.TrimSpace(operation.Data.(string))
  1598. if err = av.SaveAttributeView(attrView); nil != err {
  1599. logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
  1600. return &TxErr{code: TxErrWriteAttributeView, msg: err.Error(), id: avID}
  1601. }
  1602. return
  1603. }
  1604. func (tx *Transaction) doSetAttrViewViewIcon(operation *Operation) (ret *TxErr) {
  1605. var err error
  1606. avID := operation.AvID
  1607. attrView, err := av.ParseAttributeView(avID)
  1608. if nil != err {
  1609. logging.LogErrorf("parse attribute view [%s] failed: %s", avID, err)
  1610. return &TxErr{code: TxErrWriteAttributeView, id: avID}
  1611. }
  1612. viewID := operation.ID
  1613. view := attrView.GetView(viewID)
  1614. if nil == view {
  1615. logging.LogErrorf("get view [%s] failed: %s", viewID, err)
  1616. return &TxErr{code: TxErrWriteAttributeView, id: viewID}
  1617. }
  1618. view.Icon = operation.Data.(string)
  1619. if err = av.SaveAttributeView(attrView); nil != err {
  1620. logging.LogErrorf("save attribute view [%s] failed: %s", avID, err)
  1621. return &TxErr{code: TxErrWriteAttributeView, msg: err.Error(), id: avID}
  1622. }
  1623. return
  1624. }
  1625. func (tx *Transaction) doSetAttrViewName(operation *Operation) (ret *TxErr) {
  1626. err := setAttributeViewName(operation)
  1627. if nil != err {
  1628. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1629. }
  1630. return
  1631. }
  1632. const attrAvNameTpl = `<span data-av-id="${avID}" data-popover-url="/api/av/getMirrorDatabaseBlocks" class="popover__block">${avName}</span>`
  1633. func setAttributeViewName(operation *Operation) (err error) {
  1634. avID := operation.ID
  1635. attrView, err := av.ParseAttributeView(avID)
  1636. if nil != err {
  1637. return
  1638. }
  1639. attrView.Name = strings.TrimSpace(operation.Data.(string))
  1640. err = av.SaveAttributeView(attrView)
  1641. nodes := getAttrViewBoundNodes(attrView)
  1642. for _, node := range nodes {
  1643. avNames := getAvNames(node.IALAttr(av.NodeAttrNameAvs))
  1644. oldAttrs := parse.IAL2Map(node.KramdownIAL)
  1645. node.SetIALAttr(av.NodeAttrViewNames, avNames)
  1646. pushBroadcastAttrTransactions(oldAttrs, node)
  1647. }
  1648. return
  1649. }
  1650. func getAvNames(avIDs string) (ret string) {
  1651. if "" == avIDs {
  1652. return
  1653. }
  1654. avNames := bytes.Buffer{}
  1655. nodeAvIDs := strings.Split(avIDs, ",")
  1656. for _, nodeAvID := range nodeAvIDs {
  1657. nodeAvName, getErr := av.GetAttributeViewName(nodeAvID)
  1658. if nil != getErr {
  1659. continue
  1660. }
  1661. if "" == nodeAvName {
  1662. nodeAvName = Conf.language(105)
  1663. }
  1664. tpl := strings.ReplaceAll(attrAvNameTpl, "${avID}", nodeAvID)
  1665. tpl = strings.ReplaceAll(tpl, "${avName}", nodeAvName)
  1666. avNames.WriteString(tpl)
  1667. avNames.WriteString("&nbsp;")
  1668. }
  1669. if 0 < avNames.Len() {
  1670. avNames.Truncate(avNames.Len() - 6)
  1671. ret = avNames.String()
  1672. }
  1673. return
  1674. }
  1675. func getAttrViewBoundNodes(attrView *av.AttributeView) (ret []*ast.Node) {
  1676. blockKeyValues := attrView.GetBlockKeyValues()
  1677. treeMap := map[string]*parse.Tree{}
  1678. for _, blockKeyValue := range blockKeyValues.Values {
  1679. if blockKeyValue.IsDetached {
  1680. continue
  1681. }
  1682. var tree *parse.Tree
  1683. tree = treeMap[blockKeyValue.BlockID]
  1684. if nil == tree {
  1685. tree, _ = LoadTreeByBlockID(blockKeyValue.BlockID)
  1686. }
  1687. if nil == tree {
  1688. continue
  1689. }
  1690. treeMap[blockKeyValue.BlockID] = tree
  1691. node := treenode.GetNodeInTree(tree, blockKeyValue.BlockID)
  1692. if nil == node {
  1693. continue
  1694. }
  1695. ret = append(ret, node)
  1696. }
  1697. return
  1698. }
  1699. func (tx *Transaction) doSetAttrViewFilters(operation *Operation) (ret *TxErr) {
  1700. err := setAttributeViewFilters(operation)
  1701. if nil != err {
  1702. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1703. }
  1704. return
  1705. }
  1706. func setAttributeViewFilters(operation *Operation) (err error) {
  1707. attrView, err := av.ParseAttributeView(operation.AvID)
  1708. if nil != err {
  1709. return
  1710. }
  1711. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  1712. if nil != err {
  1713. return
  1714. }
  1715. operationData := operation.Data.([]interface{})
  1716. data, err := gulu.JSON.MarshalJSON(operationData)
  1717. if nil != err {
  1718. return
  1719. }
  1720. switch view.LayoutType {
  1721. case av.LayoutTypeTable:
  1722. if err = gulu.JSON.UnmarshalJSON(data, &view.Table.Filters); nil != err {
  1723. return
  1724. }
  1725. }
  1726. err = av.SaveAttributeView(attrView)
  1727. return
  1728. }
  1729. func (tx *Transaction) doSetAttrViewSorts(operation *Operation) (ret *TxErr) {
  1730. err := setAttributeViewSorts(operation)
  1731. if nil != err {
  1732. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1733. }
  1734. return
  1735. }
  1736. func setAttributeViewSorts(operation *Operation) (err error) {
  1737. attrView, err := av.ParseAttributeView(operation.AvID)
  1738. if nil != err {
  1739. return
  1740. }
  1741. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  1742. if nil != err {
  1743. return
  1744. }
  1745. operationData := operation.Data.([]interface{})
  1746. data, err := gulu.JSON.MarshalJSON(operationData)
  1747. if nil != err {
  1748. return
  1749. }
  1750. switch view.LayoutType {
  1751. case av.LayoutTypeTable:
  1752. if err = gulu.JSON.UnmarshalJSON(data, &view.Table.Sorts); nil != err {
  1753. return
  1754. }
  1755. }
  1756. err = av.SaveAttributeView(attrView)
  1757. return
  1758. }
  1759. func (tx *Transaction) doSetAttrViewPageSize(operation *Operation) (ret *TxErr) {
  1760. err := setAttributeViewPageSize(operation)
  1761. if nil != err {
  1762. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1763. }
  1764. return
  1765. }
  1766. func setAttributeViewPageSize(operation *Operation) (err error) {
  1767. attrView, err := av.ParseAttributeView(operation.AvID)
  1768. if nil != err {
  1769. return
  1770. }
  1771. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  1772. if nil != err {
  1773. return
  1774. }
  1775. switch view.LayoutType {
  1776. case av.LayoutTypeTable:
  1777. view.Table.PageSize = int(operation.Data.(float64))
  1778. }
  1779. err = av.SaveAttributeView(attrView)
  1780. return
  1781. }
  1782. func (tx *Transaction) doSetAttrViewColCalc(operation *Operation) (ret *TxErr) {
  1783. err := setAttributeViewColumnCalc(operation)
  1784. if nil != err {
  1785. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1786. }
  1787. return
  1788. }
  1789. func setAttributeViewColumnCalc(operation *Operation) (err error) {
  1790. attrView, err := av.ParseAttributeView(operation.AvID)
  1791. if nil != err {
  1792. return
  1793. }
  1794. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  1795. if nil != err {
  1796. return
  1797. }
  1798. operationData := operation.Data.(interface{})
  1799. data, err := gulu.JSON.MarshalJSON(operationData)
  1800. if nil != err {
  1801. return
  1802. }
  1803. calc := &av.ColumnCalc{}
  1804. switch view.LayoutType {
  1805. case av.LayoutTypeTable:
  1806. if err = gulu.JSON.UnmarshalJSON(data, calc); nil != err {
  1807. return
  1808. }
  1809. for _, column := range view.Table.Columns {
  1810. if column.ID == operation.ID {
  1811. column.Calc = calc
  1812. break
  1813. }
  1814. }
  1815. }
  1816. err = av.SaveAttributeView(attrView)
  1817. return
  1818. }
  1819. func (tx *Transaction) doInsertAttrViewBlock(operation *Operation) (ret *TxErr) {
  1820. err := AddAttributeViewBlock(tx, operation.SrcIDs, operation.AvID, operation.BlockID, operation.PreviousID, operation.IsDetached, operation.IgnoreFillFilterVal)
  1821. if nil != err {
  1822. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  1823. }
  1824. return
  1825. }
  1826. func AddAttributeViewBlock(tx *Transaction, srcIDs []string, avID, blockID, previousBlockID string, isDetached, ignoreFillFilter bool) (err error) {
  1827. for _, id := range srcIDs {
  1828. var tree *parse.Tree
  1829. if !isDetached {
  1830. var loadErr error
  1831. if nil != tx {
  1832. tree, loadErr = tx.loadTree(id)
  1833. } else {
  1834. tree, loadErr = LoadTreeByBlockID(id)
  1835. }
  1836. if nil != loadErr {
  1837. logging.LogErrorf("load tree [%s] failed: %s", id, err)
  1838. return loadErr
  1839. }
  1840. }
  1841. if avErr := addAttributeViewBlock(avID, blockID, previousBlockID, id, isDetached, ignoreFillFilter, tree, tx); nil != avErr {
  1842. return avErr
  1843. }
  1844. }
  1845. return
  1846. }
  1847. func addAttributeViewBlock(avID, blockID, previousBlockID, addingBlockID string, isDetached, ignoreFillFilter bool, tree *parse.Tree, tx *Transaction) (err error) {
  1848. var node *ast.Node
  1849. if !isDetached {
  1850. node = treenode.GetNodeInTree(tree, addingBlockID)
  1851. if nil == node {
  1852. err = ErrBlockNotFound
  1853. return
  1854. }
  1855. } else {
  1856. if "" == addingBlockID {
  1857. addingBlockID = ast.NewNodeID()
  1858. logging.LogWarnf("detached block id is empty, generate a new one [%s]", addingBlockID)
  1859. }
  1860. }
  1861. attrView, err := av.ParseAttributeView(avID)
  1862. if nil != err {
  1863. return
  1864. }
  1865. var content string
  1866. if !isDetached {
  1867. content = getNodeRefText(node)
  1868. }
  1869. now := time.Now().UnixMilli()
  1870. // 检查是否重复添加相同的块
  1871. blockValues := attrView.GetBlockKeyValues()
  1872. for _, blockValue := range blockValues.Values {
  1873. if blockValue.Block.ID == addingBlockID {
  1874. if !isDetached {
  1875. // 重复绑定一下,比如剪切数据库块、取消绑定块后再次添加的场景需要
  1876. bindBlockAv0(tx, avID, node, tree)
  1877. blockValue.IsDetached = isDetached
  1878. blockValue.Block.Content = content
  1879. blockValue.UpdatedAt = now
  1880. err = av.SaveAttributeView(attrView)
  1881. }
  1882. return
  1883. }
  1884. }
  1885. blockValue := &av.Value{
  1886. ID: ast.NewNodeID(),
  1887. KeyID: blockValues.Key.ID,
  1888. BlockID: addingBlockID,
  1889. Type: av.KeyTypeBlock,
  1890. IsDetached: isDetached,
  1891. CreatedAt: now,
  1892. UpdatedAt: now,
  1893. Block: &av.ValueBlock{ID: addingBlockID, Content: content, Created: now, Updated: now}}
  1894. blockValues.Values = append(blockValues.Values, blockValue)
  1895. // 如果存在过滤条件,则将过滤条件应用到新添加的块上
  1896. view, _ := getAttrViewViewByBlockID(attrView, blockID)
  1897. if nil != view && 0 < len(view.Table.Filters) && !ignoreFillFilter {
  1898. viewable, _ := renderAttributeViewTable(attrView, view, "")
  1899. viewable.FilterRows(attrView)
  1900. viewable.SortRows(attrView)
  1901. var nearRow *av.TableRow
  1902. if 0 < len(viewable.Rows) {
  1903. if "" != previousBlockID {
  1904. for _, row := range viewable.Rows {
  1905. if row.ID == previousBlockID {
  1906. nearRow = row
  1907. break
  1908. }
  1909. }
  1910. } else {
  1911. if 0 < len(viewable.Rows) {
  1912. nearRow = viewable.Rows[0]
  1913. }
  1914. }
  1915. }
  1916. sameKeyFilterSort := false // 是否在同一个字段上同时存在过滤和排序
  1917. if 0 < len(viewable.Sorts) {
  1918. filterKeys, sortKeys := map[string]bool{}, map[string]bool{}
  1919. for _, f := range view.Table.Filters {
  1920. filterKeys[f.Column] = true
  1921. }
  1922. for _, s := range view.Table.Sorts {
  1923. sortKeys[s.Column] = true
  1924. }
  1925. for key := range filterKeys {
  1926. if sortKeys[key] {
  1927. sameKeyFilterSort = true
  1928. break
  1929. }
  1930. }
  1931. }
  1932. if !sameKeyFilterSort {
  1933. // 如果在同一个字段上仅存在过滤条件,则将过滤条件应用到新添加的块上
  1934. for _, filter := range view.Table.Filters {
  1935. for _, keyValues := range attrView.KeyValues {
  1936. if keyValues.Key.ID == filter.Column {
  1937. var defaultVal *av.Value
  1938. if nil != nearRow {
  1939. defaultVal = nearRow.GetValue(filter.Column)
  1940. }
  1941. newValue := filter.GetAffectValue(keyValues.Key, defaultVal)
  1942. if nil == newValue {
  1943. continue
  1944. }
  1945. newValue.ID = ast.NewNodeID()
  1946. newValue.KeyID = keyValues.Key.ID
  1947. newValue.BlockID = addingBlockID
  1948. newValue.IsDetached = isDetached
  1949. keyValues.Values = append(keyValues.Values, newValue)
  1950. break
  1951. }
  1952. }
  1953. }
  1954. }
  1955. }
  1956. // 处理日期字段默认填充当前创建时间
  1957. // The database date field supports filling the current time by default https://github.com/siyuan-note/siyuan/issues/10823
  1958. for _, keyValues := range attrView.KeyValues {
  1959. if av.KeyTypeDate == keyValues.Key.Type && nil != keyValues.Key.Date && keyValues.Key.Date.AutoFillNow {
  1960. dateVal := &av.Value{
  1961. ID: ast.NewNodeID(), KeyID: keyValues.Key.ID, BlockID: addingBlockID, Type: av.KeyTypeDate, IsDetached: isDetached, CreatedAt: now, UpdatedAt: now + 1000,
  1962. Date: &av.ValueDate{Content: now, IsNotEmpty: true},
  1963. }
  1964. keyValues.Values = append(keyValues.Values, dateVal)
  1965. }
  1966. }
  1967. if !isDetached {
  1968. bindBlockAv0(tx, avID, node, tree)
  1969. }
  1970. for _, v := range attrView.Views {
  1971. switch v.LayoutType {
  1972. case av.LayoutTypeTable:
  1973. if "" != previousBlockID {
  1974. changed := false
  1975. for i, id := range v.Table.RowIDs {
  1976. if id == previousBlockID {
  1977. v.Table.RowIDs = append(v.Table.RowIDs[:i+1], append([]string{addingBlockID}, v.Table.RowIDs[i+1:]...)...)
  1978. changed = true
  1979. break
  1980. }
  1981. }
  1982. if !changed {
  1983. v.Table.RowIDs = append(v.Table.RowIDs, addingBlockID)
  1984. }
  1985. } else {
  1986. v.Table.RowIDs = append([]string{addingBlockID}, v.Table.RowIDs...)
  1987. }
  1988. }
  1989. }
  1990. err = av.SaveAttributeView(attrView)
  1991. return
  1992. }
  1993. func (tx *Transaction) doRemoveAttrViewBlock(operation *Operation) (ret *TxErr) {
  1994. err := removeAttributeViewBlock(operation.SrcIDs, operation.AvID, tx)
  1995. if nil != err {
  1996. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID}
  1997. }
  1998. return
  1999. }
  2000. func RemoveAttributeViewBlock(srcIDs []string, avID string) (err error) {
  2001. err = removeAttributeViewBlock(srcIDs, avID, nil)
  2002. return
  2003. }
  2004. func removeAttributeViewBlock(srcIDs []string, avID string, tx *Transaction) (err error) {
  2005. attrView, err := av.ParseAttributeView(avID)
  2006. if nil != err {
  2007. return
  2008. }
  2009. trees := map[string]*parse.Tree{}
  2010. for _, keyValues := range attrView.KeyValues {
  2011. tmp := keyValues.Values[:0]
  2012. for i, values := range keyValues.Values {
  2013. if !gulu.Str.Contains(values.BlockID, srcIDs) {
  2014. tmp = append(tmp, keyValues.Values[i])
  2015. } else {
  2016. // Remove av block also remove node attr https://github.com/siyuan-note/siyuan/issues/9091#issuecomment-1709824006
  2017. if bt := treenode.GetBlockTree(values.BlockID); nil != bt {
  2018. tree := trees[bt.RootID]
  2019. if nil == tree {
  2020. tree, _ = LoadTreeByBlockID(values.BlockID)
  2021. }
  2022. if nil != tree {
  2023. trees[bt.RootID] = tree
  2024. if node := treenode.GetNodeInTree(tree, values.BlockID); nil != node {
  2025. if err = removeNodeAvID(node, avID, tx, tree); nil != err {
  2026. return
  2027. }
  2028. }
  2029. }
  2030. }
  2031. }
  2032. }
  2033. keyValues.Values = tmp
  2034. }
  2035. for _, view := range attrView.Views {
  2036. for _, blockID := range srcIDs {
  2037. view.Table.RowIDs = gulu.Str.RemoveElem(view.Table.RowIDs, blockID)
  2038. }
  2039. }
  2040. relatedAvIDs := av.GetSrcAvIDs(avID)
  2041. for _, relatedAvID := range relatedAvIDs {
  2042. util.BroadcastByType("protyle", "refreshAttributeView", 0, "", map[string]interface{}{"id": relatedAvID})
  2043. }
  2044. err = av.SaveAttributeView(attrView)
  2045. return
  2046. }
  2047. func removeNodeAvID(node *ast.Node, avID string, tx *Transaction, tree *parse.Tree) (err error) {
  2048. attrs := parse.IAL2Map(node.KramdownIAL)
  2049. if ast.NodeDocument == node.Type {
  2050. delete(attrs, "custom-hidden")
  2051. node.RemoveIALAttr("custom-hidden")
  2052. }
  2053. if avs := attrs[av.NodeAttrNameAvs]; "" != avs {
  2054. avIDs := strings.Split(avs, ",")
  2055. avIDs = gulu.Str.RemoveElem(avIDs, avID)
  2056. var existAvIDs []string
  2057. for _, attributeViewID := range avIDs {
  2058. if av.IsAttributeViewExist(attributeViewID) {
  2059. existAvIDs = append(existAvIDs, attributeViewID)
  2060. }
  2061. }
  2062. avIDs = existAvIDs
  2063. if 0 == len(avIDs) {
  2064. attrs[av.NodeAttrNameAvs] = ""
  2065. } else {
  2066. attrs[av.NodeAttrNameAvs] = strings.Join(avIDs, ",")
  2067. node.SetIALAttr(av.NodeAttrNameAvs, strings.Join(avIDs, ","))
  2068. avNames := getAvNames(node.IALAttr(av.NodeAttrNameAvs))
  2069. attrs[av.NodeAttrViewNames] = avNames
  2070. }
  2071. }
  2072. if nil != tx {
  2073. if err = setNodeAttrsWithTx(tx, node, tree, attrs); nil != err {
  2074. return
  2075. }
  2076. } else {
  2077. if err = setNodeAttrs(node, tree, attrs); nil != err {
  2078. return
  2079. }
  2080. }
  2081. return
  2082. }
  2083. func (tx *Transaction) doSetAttrViewColumnWidth(operation *Operation) (ret *TxErr) {
  2084. err := setAttributeViewColWidth(operation)
  2085. if nil != err {
  2086. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2087. }
  2088. return
  2089. }
  2090. func setAttributeViewColWidth(operation *Operation) (err error) {
  2091. attrView, err := av.ParseAttributeView(operation.AvID)
  2092. if nil != err {
  2093. return
  2094. }
  2095. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  2096. if nil != err {
  2097. return
  2098. }
  2099. switch view.LayoutType {
  2100. case av.LayoutTypeTable:
  2101. for _, column := range view.Table.Columns {
  2102. if column.ID == operation.ID {
  2103. column.Width = operation.Data.(string)
  2104. break
  2105. }
  2106. }
  2107. }
  2108. err = av.SaveAttributeView(attrView)
  2109. return
  2110. }
  2111. func (tx *Transaction) doSetAttrViewColumnWrap(operation *Operation) (ret *TxErr) {
  2112. err := setAttributeViewColWrap(operation)
  2113. if nil != err {
  2114. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2115. }
  2116. return
  2117. }
  2118. func setAttributeViewColWrap(operation *Operation) (err error) {
  2119. attrView, err := av.ParseAttributeView(operation.AvID)
  2120. if nil != err {
  2121. return
  2122. }
  2123. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  2124. if nil != err {
  2125. return
  2126. }
  2127. switch view.LayoutType {
  2128. case av.LayoutTypeTable:
  2129. for _, column := range view.Table.Columns {
  2130. if column.ID == operation.ID {
  2131. column.Wrap = operation.Data.(bool)
  2132. break
  2133. }
  2134. }
  2135. }
  2136. err = av.SaveAttributeView(attrView)
  2137. return
  2138. }
  2139. func (tx *Transaction) doSetAttrViewColumnHidden(operation *Operation) (ret *TxErr) {
  2140. err := setAttributeViewColHidden(operation)
  2141. if nil != err {
  2142. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2143. }
  2144. return
  2145. }
  2146. func setAttributeViewColHidden(operation *Operation) (err error) {
  2147. attrView, err := av.ParseAttributeView(operation.AvID)
  2148. if nil != err {
  2149. return
  2150. }
  2151. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  2152. if nil != err {
  2153. return
  2154. }
  2155. switch view.LayoutType {
  2156. case av.LayoutTypeTable:
  2157. for _, column := range view.Table.Columns {
  2158. if column.ID == operation.ID {
  2159. column.Hidden = operation.Data.(bool)
  2160. break
  2161. }
  2162. }
  2163. }
  2164. err = av.SaveAttributeView(attrView)
  2165. return
  2166. }
  2167. func (tx *Transaction) doSetAttrViewColumnPin(operation *Operation) (ret *TxErr) {
  2168. err := setAttributeViewColPin(operation)
  2169. if nil != err {
  2170. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2171. }
  2172. return
  2173. }
  2174. func setAttributeViewColPin(operation *Operation) (err error) {
  2175. attrView, err := av.ParseAttributeView(operation.AvID)
  2176. if nil != err {
  2177. return
  2178. }
  2179. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  2180. if nil != err {
  2181. return
  2182. }
  2183. switch view.LayoutType {
  2184. case av.LayoutTypeTable:
  2185. for _, column := range view.Table.Columns {
  2186. if column.ID == operation.ID {
  2187. column.Pin = operation.Data.(bool)
  2188. break
  2189. }
  2190. }
  2191. }
  2192. err = av.SaveAttributeView(attrView)
  2193. return
  2194. }
  2195. func (tx *Transaction) doSetAttrViewColumnIcon(operation *Operation) (ret *TxErr) {
  2196. err := setAttributeViewColIcon(operation)
  2197. if nil != err {
  2198. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2199. }
  2200. return
  2201. }
  2202. func setAttributeViewColIcon(operation *Operation) (err error) {
  2203. attrView, err := av.ParseAttributeView(operation.AvID)
  2204. if nil != err {
  2205. return
  2206. }
  2207. for _, keyValues := range attrView.KeyValues {
  2208. if keyValues.Key.ID == operation.ID {
  2209. keyValues.Key.Icon = operation.Data.(string)
  2210. break
  2211. }
  2212. }
  2213. err = av.SaveAttributeView(attrView)
  2214. return
  2215. }
  2216. func (tx *Transaction) doSortAttrViewRow(operation *Operation) (ret *TxErr) {
  2217. err := sortAttributeViewRow(operation)
  2218. if nil != err {
  2219. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2220. }
  2221. return
  2222. }
  2223. func sortAttributeViewRow(operation *Operation) (err error) {
  2224. attrView, err := av.ParseAttributeView(operation.AvID)
  2225. if nil != err {
  2226. return
  2227. }
  2228. view, err := getAttrViewViewByBlockID(attrView, operation.BlockID)
  2229. if nil != err {
  2230. return
  2231. }
  2232. var rowID string
  2233. var index, previousIndex int
  2234. for i, r := range view.Table.RowIDs {
  2235. if r == operation.ID {
  2236. rowID = r
  2237. index = i
  2238. break
  2239. }
  2240. }
  2241. if "" == rowID {
  2242. rowID = operation.ID
  2243. view.Table.RowIDs = append(view.Table.RowIDs, rowID)
  2244. index = len(view.Table.RowIDs) - 1
  2245. }
  2246. switch view.LayoutType {
  2247. case av.LayoutTypeTable:
  2248. view.Table.RowIDs = append(view.Table.RowIDs[:index], view.Table.RowIDs[index+1:]...)
  2249. for i, r := range view.Table.RowIDs {
  2250. if r == operation.PreviousID {
  2251. previousIndex = i + 1
  2252. break
  2253. }
  2254. }
  2255. view.Table.RowIDs = util.InsertElem(view.Table.RowIDs, previousIndex, rowID)
  2256. }
  2257. err = av.SaveAttributeView(attrView)
  2258. return
  2259. }
  2260. func (tx *Transaction) doSortAttrViewColumn(operation *Operation) (ret *TxErr) {
  2261. err := SortAttributeViewKey(operation.AvID, operation.BlockID, operation.ID, operation.PreviousID)
  2262. if nil != err {
  2263. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2264. }
  2265. return
  2266. }
  2267. func SortAttributeViewKey(avID, blockID, keyID, previousKeyID string) (err error) {
  2268. attrView, err := av.ParseAttributeView(avID)
  2269. if nil != err {
  2270. return
  2271. }
  2272. view, err := getAttrViewViewByBlockID(attrView, blockID)
  2273. if nil != err {
  2274. return
  2275. }
  2276. switch view.LayoutType {
  2277. case av.LayoutTypeTable:
  2278. var col *av.ViewTableColumn
  2279. var index, previousIndex int
  2280. for i, column := range view.Table.Columns {
  2281. if column.ID == keyID {
  2282. col = column
  2283. index = i
  2284. break
  2285. }
  2286. }
  2287. if nil == col {
  2288. return
  2289. }
  2290. view.Table.Columns = append(view.Table.Columns[:index], view.Table.Columns[index+1:]...)
  2291. for i, column := range view.Table.Columns {
  2292. if column.ID == previousKeyID {
  2293. previousIndex = i + 1
  2294. break
  2295. }
  2296. }
  2297. view.Table.Columns = util.InsertElem(view.Table.Columns, previousIndex, col)
  2298. }
  2299. err = av.SaveAttributeView(attrView)
  2300. return
  2301. }
  2302. func (tx *Transaction) doAddAttrViewColumn(operation *Operation) (ret *TxErr) {
  2303. var icon string
  2304. if nil != operation.Data {
  2305. icon = operation.Data.(string)
  2306. }
  2307. err := AddAttributeViewKey(operation.AvID, operation.ID, operation.Name, operation.Typ, icon, operation.PreviousID)
  2308. if nil != err {
  2309. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2310. }
  2311. return
  2312. }
  2313. func AddAttributeViewKey(avID, keyID, keyName, keyType, keyIcon, previousKeyID string) (err error) {
  2314. attrView, err := av.ParseAttributeView(avID)
  2315. if nil != err {
  2316. return
  2317. }
  2318. keyTyp := av.KeyType(keyType)
  2319. switch keyTyp {
  2320. case av.KeyTypeText, av.KeyTypeNumber, av.KeyTypeDate, av.KeyTypeSelect, av.KeyTypeMSelect, av.KeyTypeURL, av.KeyTypeEmail,
  2321. av.KeyTypePhone, av.KeyTypeMAsset, av.KeyTypeTemplate, av.KeyTypeCreated, av.KeyTypeUpdated, av.KeyTypeCheckbox,
  2322. av.KeyTypeRelation, av.KeyTypeRollup:
  2323. key := av.NewKey(keyID, keyName, keyIcon, keyTyp)
  2324. if av.KeyTypeRollup == keyTyp {
  2325. key.Rollup = &av.Rollup{Calc: &av.RollupCalc{Operator: av.CalcOperatorNone}}
  2326. }
  2327. attrView.KeyValues = append(attrView.KeyValues, &av.KeyValues{Key: key})
  2328. for _, view := range attrView.Views {
  2329. switch view.LayoutType {
  2330. case av.LayoutTypeTable:
  2331. if "" == previousKeyID {
  2332. view.Table.Columns = append([]*av.ViewTableColumn{{ID: key.ID}}, view.Table.Columns...)
  2333. break
  2334. }
  2335. added := false
  2336. for i, column := range view.Table.Columns {
  2337. if column.ID == previousKeyID {
  2338. view.Table.Columns = append(view.Table.Columns[:i+1], append([]*av.ViewTableColumn{{ID: key.ID}}, view.Table.Columns[i+1:]...)...)
  2339. added = true
  2340. break
  2341. }
  2342. }
  2343. if !added {
  2344. view.Table.Columns = append(view.Table.Columns, &av.ViewTableColumn{ID: key.ID})
  2345. }
  2346. }
  2347. }
  2348. }
  2349. err = av.SaveAttributeView(attrView)
  2350. return
  2351. }
  2352. func (tx *Transaction) doUpdateAttrViewColTemplate(operation *Operation) (ret *TxErr) {
  2353. err := updateAttributeViewColTemplate(operation)
  2354. if nil != err {
  2355. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2356. }
  2357. return
  2358. }
  2359. func updateAttributeViewColTemplate(operation *Operation) (err error) {
  2360. attrView, err := av.ParseAttributeView(operation.AvID)
  2361. if nil != err {
  2362. return
  2363. }
  2364. colType := av.KeyType(operation.Typ)
  2365. switch colType {
  2366. case av.KeyTypeTemplate:
  2367. for _, keyValues := range attrView.KeyValues {
  2368. if keyValues.Key.ID == operation.ID && av.KeyTypeTemplate == keyValues.Key.Type {
  2369. keyValues.Key.Template = operation.Data.(string)
  2370. break
  2371. }
  2372. }
  2373. }
  2374. err = av.SaveAttributeView(attrView)
  2375. return
  2376. }
  2377. func (tx *Transaction) doUpdateAttrViewColNumberFormat(operation *Operation) (ret *TxErr) {
  2378. err := updateAttributeViewColNumberFormat(operation)
  2379. if nil != err {
  2380. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2381. }
  2382. return
  2383. }
  2384. func updateAttributeViewColNumberFormat(operation *Operation) (err error) {
  2385. attrView, err := av.ParseAttributeView(operation.AvID)
  2386. if nil != err {
  2387. return
  2388. }
  2389. colType := av.KeyType(operation.Typ)
  2390. switch colType {
  2391. case av.KeyTypeNumber:
  2392. for _, keyValues := range attrView.KeyValues {
  2393. if keyValues.Key.ID == operation.ID && av.KeyTypeNumber == keyValues.Key.Type {
  2394. keyValues.Key.NumberFormat = av.NumberFormat(operation.Format)
  2395. break
  2396. }
  2397. }
  2398. }
  2399. err = av.SaveAttributeView(attrView)
  2400. return
  2401. }
  2402. func (tx *Transaction) doUpdateAttrViewColumn(operation *Operation) (ret *TxErr) {
  2403. err := updateAttributeViewColumn(operation)
  2404. if nil != err {
  2405. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2406. }
  2407. return
  2408. }
  2409. func updateAttributeViewColumn(operation *Operation) (err error) {
  2410. attrView, err := av.ParseAttributeView(operation.AvID)
  2411. if nil != err {
  2412. return
  2413. }
  2414. colType := av.KeyType(operation.Typ)
  2415. switch colType {
  2416. case av.KeyTypeBlock, av.KeyTypeText, av.KeyTypeNumber, av.KeyTypeDate, av.KeyTypeSelect, av.KeyTypeMSelect, av.KeyTypeURL, av.KeyTypeEmail,
  2417. av.KeyTypePhone, av.KeyTypeMAsset, av.KeyTypeTemplate, av.KeyTypeCreated, av.KeyTypeUpdated, av.KeyTypeCheckbox,
  2418. av.KeyTypeRelation, av.KeyTypeRollup:
  2419. for _, keyValues := range attrView.KeyValues {
  2420. if keyValues.Key.ID == operation.ID {
  2421. keyValues.Key.Name = strings.TrimSpace(operation.Name)
  2422. keyValues.Key.Type = colType
  2423. break
  2424. }
  2425. }
  2426. }
  2427. err = av.SaveAttributeView(attrView)
  2428. return
  2429. }
  2430. func (tx *Transaction) doRemoveAttrViewColumn(operation *Operation) (ret *TxErr) {
  2431. err := RemoveAttributeViewKey(operation.AvID, operation.ID)
  2432. if nil != err {
  2433. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2434. }
  2435. return
  2436. }
  2437. func RemoveAttributeViewKey(avID, keyID string) (err error) {
  2438. attrView, err := av.ParseAttributeView(avID)
  2439. if nil != err {
  2440. return
  2441. }
  2442. var removedKey *av.Key
  2443. for i, keyValues := range attrView.KeyValues {
  2444. if keyValues.Key.ID == keyID {
  2445. attrView.KeyValues = append(attrView.KeyValues[:i], attrView.KeyValues[i+1:]...)
  2446. removedKey = keyValues.Key
  2447. break
  2448. }
  2449. }
  2450. if nil != removedKey && av.KeyTypeRelation == removedKey.Type && nil != removedKey.Relation {
  2451. if removedKey.Relation.IsTwoWay {
  2452. // 删除双向关联的目标列
  2453. destAv, _ := av.ParseAttributeView(removedKey.Relation.AvID)
  2454. if nil != destAv {
  2455. destAvRelSrcAv := false
  2456. for i, keyValues := range destAv.KeyValues {
  2457. if keyValues.Key.ID == removedKey.Relation.BackKeyID {
  2458. destAv.KeyValues = append(destAv.KeyValues[:i], destAv.KeyValues[i+1:]...)
  2459. continue
  2460. }
  2461. if av.KeyTypeRelation == keyValues.Key.Type && keyValues.Key.Relation.AvID == attrView.ID {
  2462. destAvRelSrcAv = true
  2463. }
  2464. }
  2465. for _, view := range destAv.Views {
  2466. switch view.LayoutType {
  2467. case av.LayoutTypeTable:
  2468. for i, column := range view.Table.Columns {
  2469. if column.ID == removedKey.Relation.BackKeyID {
  2470. view.Table.Columns = append(view.Table.Columns[:i], view.Table.Columns[i+1:]...)
  2471. break
  2472. }
  2473. }
  2474. }
  2475. }
  2476. av.SaveAttributeView(destAv)
  2477. util.BroadcastByType("protyle", "refreshAttributeView", 0, "", map[string]interface{}{"id": destAv.ID})
  2478. if !destAvRelSrcAv {
  2479. av.RemoveAvRel(destAv.ID, attrView.ID)
  2480. }
  2481. }
  2482. srcAvRelDestAv := false
  2483. for _, keyValues := range attrView.KeyValues {
  2484. if av.KeyTypeRelation == keyValues.Key.Type && nil != keyValues.Key.Relation && keyValues.Key.Relation.AvID == removedKey.Relation.AvID {
  2485. srcAvRelDestAv = true
  2486. }
  2487. }
  2488. if !srcAvRelDestAv {
  2489. av.RemoveAvRel(attrView.ID, removedKey.Relation.AvID)
  2490. }
  2491. }
  2492. }
  2493. for _, view := range attrView.Views {
  2494. switch view.LayoutType {
  2495. case av.LayoutTypeTable:
  2496. for i, column := range view.Table.Columns {
  2497. if column.ID == keyID {
  2498. view.Table.Columns = append(view.Table.Columns[:i], view.Table.Columns[i+1:]...)
  2499. break
  2500. }
  2501. }
  2502. }
  2503. }
  2504. err = av.SaveAttributeView(attrView)
  2505. return
  2506. }
  2507. func (tx *Transaction) doReplaceAttrViewBlock(operation *Operation) (ret *TxErr) {
  2508. err := replaceAttributeViewBlock(operation, tx)
  2509. if nil != err {
  2510. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID}
  2511. }
  2512. return
  2513. }
  2514. func replaceAttributeViewBlock(operation *Operation, tx *Transaction) (err error) {
  2515. attrView, err := av.ParseAttributeView(operation.AvID)
  2516. if nil != err {
  2517. return
  2518. }
  2519. var node *ast.Node
  2520. if !operation.IsDetached {
  2521. node, _, _ = getNodeByBlockID(tx, operation.NextID)
  2522. }
  2523. for _, keyValues := range attrView.KeyValues {
  2524. for _, value := range keyValues.Values {
  2525. if value.BlockID == operation.PreviousID {
  2526. if value.BlockID != operation.NextID {
  2527. // 换绑
  2528. unbindBlockAv(tx, operation.AvID, value.BlockID)
  2529. }
  2530. value.BlockID = operation.NextID
  2531. if nil != value.Block {
  2532. value.Block.ID = operation.NextID
  2533. value.IsDetached = operation.IsDetached
  2534. if !operation.IsDetached {
  2535. value.Block.Content = getNodeRefText(node)
  2536. }
  2537. }
  2538. if !operation.IsDetached {
  2539. bindBlockAv(tx, operation.AvID, operation.NextID)
  2540. }
  2541. }
  2542. }
  2543. }
  2544. replacedRowID := false
  2545. for _, v := range attrView.Views {
  2546. switch v.LayoutType {
  2547. case av.LayoutTypeTable:
  2548. for i, rowID := range v.Table.RowIDs {
  2549. if rowID == operation.PreviousID {
  2550. v.Table.RowIDs[i] = operation.NextID
  2551. replacedRowID = true
  2552. break
  2553. }
  2554. }
  2555. if !replacedRowID {
  2556. v.Table.RowIDs = append(v.Table.RowIDs, operation.NextID)
  2557. }
  2558. }
  2559. }
  2560. err = av.SaveAttributeView(attrView)
  2561. return
  2562. }
  2563. func (tx *Transaction) doUpdateAttrViewCell(operation *Operation) (ret *TxErr) {
  2564. err := updateAttributeViewCell(operation, tx)
  2565. if nil != err {
  2566. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2567. }
  2568. return
  2569. }
  2570. func updateAttributeViewCell(operation *Operation, tx *Transaction) (err error) {
  2571. err = UpdateAttributeViewCell(tx, operation.AvID, operation.KeyID, operation.RowID, operation.ID, operation.Data)
  2572. return
  2573. }
  2574. func UpdateAttributeViewCell(tx *Transaction, avID, keyID, rowID, cellID string, valueData interface{}) (err error) {
  2575. attrView, err := av.ParseAttributeView(avID)
  2576. if nil != err {
  2577. return
  2578. }
  2579. var blockVal *av.Value
  2580. for _, kv := range attrView.KeyValues {
  2581. if av.KeyTypeBlock == kv.Key.Type {
  2582. for _, v := range kv.Values {
  2583. if rowID == v.Block.ID {
  2584. blockVal = v
  2585. break
  2586. }
  2587. }
  2588. break
  2589. }
  2590. }
  2591. now := time.Now().UnixMilli()
  2592. var val *av.Value
  2593. oldIsDetached := true
  2594. if nil != blockVal {
  2595. oldIsDetached = blockVal.IsDetached
  2596. }
  2597. for _, keyValues := range attrView.KeyValues {
  2598. if keyID != keyValues.Key.ID {
  2599. continue
  2600. }
  2601. for _, value := range keyValues.Values {
  2602. if cellID == value.ID {
  2603. val = value
  2604. val.Type = keyValues.Key.Type
  2605. break
  2606. }
  2607. }
  2608. if nil == val {
  2609. val = &av.Value{ID: cellID, KeyID: keyValues.Key.ID, BlockID: rowID, Type: keyValues.Key.Type, CreatedAt: now, UpdatedAt: now}
  2610. keyValues.Values = append(keyValues.Values, val)
  2611. }
  2612. break
  2613. }
  2614. isUpdatingBlockKey := av.KeyTypeBlock == val.Type
  2615. oldBoundBlockID := val.BlockID
  2616. var oldRelationBlockIDs []string
  2617. if av.KeyTypeRelation == val.Type {
  2618. if nil != val.Relation {
  2619. for _, bID := range val.Relation.BlockIDs {
  2620. oldRelationBlockIDs = append(oldRelationBlockIDs, bID)
  2621. }
  2622. }
  2623. }
  2624. data, err := gulu.JSON.MarshalJSON(valueData)
  2625. if nil != err {
  2626. return
  2627. }
  2628. if err = gulu.JSON.UnmarshalJSON(data, &val); nil != err {
  2629. return
  2630. }
  2631. if av.KeyTypeNumber == val.Type {
  2632. if nil != val.Number && !val.Number.IsNotEmpty {
  2633. val.Number.Content = 0
  2634. val.Number.FormattedContent = ""
  2635. }
  2636. } else if av.KeyTypeDate == val.Type {
  2637. if nil != val.Date && !val.Date.IsNotEmpty {
  2638. val.Date.Content = 0
  2639. val.Date.FormattedContent = ""
  2640. }
  2641. }
  2642. relationChangeMode := 0 // 0:不变(仅排序),1:增加,2:减少
  2643. if av.KeyTypeRelation == val.Type {
  2644. // 关联列得 content 是自动渲染的,所以不需要保存
  2645. val.Relation.Contents = nil
  2646. // 计算关联变更模式
  2647. if len(oldRelationBlockIDs) == len(val.Relation.BlockIDs) {
  2648. relationChangeMode = 0
  2649. } else {
  2650. if len(oldRelationBlockIDs) > len(val.Relation.BlockIDs) {
  2651. relationChangeMode = 2
  2652. } else {
  2653. relationChangeMode = 1
  2654. }
  2655. }
  2656. }
  2657. // val.IsDetached 只有更新主键的时候才会传入,所以下面需要结合 isUpdatingBlockKey 来判断
  2658. if oldIsDetached { // 之前是游离行
  2659. if !val.IsDetached { // 现在绑定了块
  2660. // 将游离行绑定到新建的块上
  2661. bindBlockAv(tx, avID, rowID)
  2662. }
  2663. } else { // 之前绑定了块
  2664. if isUpdatingBlockKey { // 正在更新主键
  2665. if val.IsDetached { // 现在是游离行
  2666. // 将绑定的块从属性视图中移除
  2667. unbindBlockAv(tx, avID, rowID)
  2668. } else { // 现在绑定了块
  2669. if oldBoundBlockID != val.BlockID { // 之前绑定的块和现在绑定的块不一样
  2670. // 换绑块
  2671. unbindBlockAv(tx, avID, oldBoundBlockID)
  2672. bindBlockAv(tx, avID, val.BlockID)
  2673. } else { // 之前绑定的块和现在绑定的块一样
  2674. // 直接返回,因为锚文本不允许更改
  2675. return
  2676. }
  2677. }
  2678. }
  2679. }
  2680. if nil != blockVal {
  2681. blockVal.Block.Updated = now
  2682. blockVal.SetUpdatedAt(now)
  2683. if isUpdatingBlockKey {
  2684. blockVal.IsDetached = val.IsDetached
  2685. }
  2686. }
  2687. val.SetUpdatedAt(now)
  2688. key, _ := attrView.GetKey(val.KeyID)
  2689. if nil != key && av.KeyTypeRelation == key.Type && nil != key.Relation {
  2690. destAv, _ := av.ParseAttributeView(key.Relation.AvID)
  2691. if nil != destAv {
  2692. if key.Relation.IsTwoWay {
  2693. // relationChangeMode
  2694. // 0:关联列值不变(仅排序),不影响目标值
  2695. // 1:关联列值增加,增加目标值
  2696. // 2:关联列值减少,减少目标值
  2697. if 1 == relationChangeMode {
  2698. addBlockIDs := val.Relation.BlockIDs
  2699. for _, bID := range oldRelationBlockIDs {
  2700. addBlockIDs = gulu.Str.RemoveElem(addBlockIDs, bID)
  2701. }
  2702. for _, blockID := range addBlockIDs {
  2703. for _, keyValues := range destAv.KeyValues {
  2704. if keyValues.Key.ID != key.Relation.BackKeyID {
  2705. continue
  2706. }
  2707. destVal := keyValues.GetValue(blockID)
  2708. if nil == destVal {
  2709. destVal = &av.Value{ID: ast.NewNodeID(), KeyID: keyValues.Key.ID, BlockID: blockID, Type: keyValues.Key.Type, Relation: &av.ValueRelation{}}
  2710. keyValues.Values = append(keyValues.Values, destVal)
  2711. }
  2712. destVal.Relation.BlockIDs = append(destVal.Relation.BlockIDs, rowID)
  2713. destVal.Relation.BlockIDs = gulu.Str.RemoveDuplicatedElem(destVal.Relation.BlockIDs)
  2714. destVal.SetUpdatedAt(now)
  2715. break
  2716. }
  2717. }
  2718. } else if 2 == relationChangeMode {
  2719. removeBlockIDs := oldRelationBlockIDs
  2720. for _, bID := range val.Relation.BlockIDs {
  2721. removeBlockIDs = gulu.Str.RemoveElem(removeBlockIDs, bID)
  2722. }
  2723. for _, blockID := range removeBlockIDs {
  2724. for _, keyValues := range destAv.KeyValues {
  2725. if keyValues.Key.ID != key.Relation.BackKeyID {
  2726. continue
  2727. }
  2728. for _, value := range keyValues.Values {
  2729. if value.BlockID == blockID {
  2730. value.Relation.BlockIDs = gulu.Str.RemoveElem(value.Relation.BlockIDs, rowID)
  2731. value.SetUpdatedAt(now)
  2732. break
  2733. }
  2734. }
  2735. }
  2736. }
  2737. }
  2738. av.SaveAttributeView(destAv)
  2739. }
  2740. }
  2741. }
  2742. relatedAvIDs := av.GetSrcAvIDs(avID)
  2743. for _, relatedAvID := range relatedAvIDs {
  2744. util.BroadcastByType("protyle", "refreshAttributeView", 0, "", map[string]interface{}{"id": relatedAvID})
  2745. }
  2746. if err = av.SaveAttributeView(attrView); nil != err {
  2747. return
  2748. }
  2749. return
  2750. }
  2751. func unbindBlockAv(tx *Transaction, avID, blockID string) {
  2752. node, tree, err := getNodeByBlockID(tx, blockID)
  2753. if nil != err {
  2754. return
  2755. }
  2756. attrs := parse.IAL2Map(node.KramdownIAL)
  2757. if "" == attrs[av.NodeAttrNameAvs] {
  2758. return
  2759. }
  2760. avIDs := strings.Split(attrs[av.NodeAttrNameAvs], ",")
  2761. avIDs = gulu.Str.RemoveElem(avIDs, avID)
  2762. if 0 == len(avIDs) {
  2763. delete(attrs, av.NodeAttrNameAvs)
  2764. node.RemoveIALAttr(av.NodeAttrNameAvs)
  2765. } else {
  2766. attrs[av.NodeAttrNameAvs] = strings.Join(avIDs, ",")
  2767. node.SetIALAttr(av.NodeAttrNameAvs, strings.Join(avIDs, ","))
  2768. }
  2769. avNames := getAvNames(attrs[av.NodeAttrNameAvs])
  2770. if "" != avNames {
  2771. attrs[av.NodeAttrViewNames] = avNames
  2772. }
  2773. if nil != tx {
  2774. err = setNodeAttrsWithTx(tx, node, tree, attrs)
  2775. } else {
  2776. err = setNodeAttrs(node, tree, attrs)
  2777. }
  2778. if nil != err {
  2779. logging.LogWarnf("set node [%s] attrs failed: %s", blockID, err)
  2780. return
  2781. }
  2782. return
  2783. }
  2784. func bindBlockAv(tx *Transaction, avID, blockID string) {
  2785. node, tree, err := getNodeByBlockID(tx, blockID)
  2786. if nil != err {
  2787. return
  2788. }
  2789. bindBlockAv0(tx, avID, node, tree)
  2790. return
  2791. }
  2792. func bindBlockAv0(tx *Transaction, avID string, node *ast.Node, tree *parse.Tree) {
  2793. attrs := parse.IAL2Map(node.KramdownIAL)
  2794. if "" == attrs[av.NodeAttrNameAvs] {
  2795. attrs[av.NodeAttrNameAvs] = avID
  2796. } else {
  2797. avIDs := strings.Split(attrs[av.NodeAttrNameAvs], ",")
  2798. avIDs = append(avIDs, avID)
  2799. avIDs = gulu.Str.RemoveDuplicatedElem(avIDs)
  2800. attrs[av.NodeAttrNameAvs] = strings.Join(avIDs, ",")
  2801. }
  2802. avNames := getAvNames(attrs[av.NodeAttrNameAvs])
  2803. if "" != avNames {
  2804. attrs[av.NodeAttrViewNames] = avNames
  2805. }
  2806. var err error
  2807. if nil != tx {
  2808. err = setNodeAttrsWithTx(tx, node, tree, attrs)
  2809. } else {
  2810. err = setNodeAttrs(node, tree, attrs)
  2811. }
  2812. if nil != err {
  2813. logging.LogWarnf("set node [%s] attrs failed: %s", node.ID, err)
  2814. return
  2815. }
  2816. return
  2817. }
  2818. func getNodeByBlockID(tx *Transaction, blockID string) (node *ast.Node, tree *parse.Tree, err error) {
  2819. if nil != tx {
  2820. tree, err = tx.loadTree(blockID)
  2821. } else {
  2822. tree, err = LoadTreeByBlockID(blockID)
  2823. }
  2824. if nil != err {
  2825. return
  2826. }
  2827. node = treenode.GetNodeInTree(tree, blockID)
  2828. if nil == node {
  2829. logging.LogWarnf("node [%s] not found in tree [%s]", blockID, tree.ID)
  2830. return
  2831. }
  2832. return
  2833. }
  2834. func (tx *Transaction) doUpdateAttrViewColOptions(operation *Operation) (ret *TxErr) {
  2835. err := updateAttributeViewColumnOptions(operation)
  2836. if nil != err {
  2837. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2838. }
  2839. return
  2840. }
  2841. func updateAttributeViewColumnOptions(operation *Operation) (err error) {
  2842. attrView, err := av.ParseAttributeView(operation.AvID)
  2843. if nil != err {
  2844. return
  2845. }
  2846. jsonData, err := gulu.JSON.MarshalJSON(operation.Data)
  2847. if nil != err {
  2848. return
  2849. }
  2850. options := []*av.SelectOption{}
  2851. if err = gulu.JSON.UnmarshalJSON(jsonData, &options); nil != err {
  2852. return
  2853. }
  2854. for _, keyValues := range attrView.KeyValues {
  2855. if keyValues.Key.ID == operation.ID {
  2856. keyValues.Key.Options = options
  2857. err = av.SaveAttributeView(attrView)
  2858. return
  2859. }
  2860. }
  2861. return
  2862. }
  2863. func (tx *Transaction) doRemoveAttrViewColOption(operation *Operation) (ret *TxErr) {
  2864. err := removeAttributeViewColumnOption(operation)
  2865. if nil != err {
  2866. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2867. }
  2868. return
  2869. }
  2870. func removeAttributeViewColumnOption(operation *Operation) (err error) {
  2871. attrView, err := av.ParseAttributeView(operation.AvID)
  2872. if nil != err {
  2873. return
  2874. }
  2875. optName := operation.Data.(string)
  2876. key, err := attrView.GetKey(operation.ID)
  2877. if nil != err {
  2878. return
  2879. }
  2880. for i, opt := range key.Options {
  2881. if optName == opt.Name {
  2882. key.Options = append(key.Options[:i], key.Options[i+1:]...)
  2883. break
  2884. }
  2885. }
  2886. for _, keyValues := range attrView.KeyValues {
  2887. if keyValues.Key.ID != operation.ID {
  2888. continue
  2889. }
  2890. for _, value := range keyValues.Values {
  2891. if nil == value || nil == value.MSelect {
  2892. continue
  2893. }
  2894. for i, opt := range value.MSelect {
  2895. if optName == opt.Content {
  2896. value.MSelect = append(value.MSelect[:i], value.MSelect[i+1:]...)
  2897. break
  2898. }
  2899. }
  2900. }
  2901. break
  2902. }
  2903. err = av.SaveAttributeView(attrView)
  2904. return
  2905. }
  2906. func (tx *Transaction) doUpdateAttrViewColOption(operation *Operation) (ret *TxErr) {
  2907. err := updateAttributeViewColumnOption(operation)
  2908. if nil != err {
  2909. return &TxErr{code: TxErrWriteAttributeView, id: operation.AvID, msg: err.Error()}
  2910. }
  2911. return
  2912. }
  2913. func updateAttributeViewColumnOption(operation *Operation) (err error) {
  2914. attrView, err := av.ParseAttributeView(operation.AvID)
  2915. if nil != err {
  2916. return
  2917. }
  2918. key, err := attrView.GetKey(operation.ID)
  2919. if nil != err {
  2920. return
  2921. }
  2922. data := operation.Data.(map[string]interface{})
  2923. oldName := data["oldName"].(string)
  2924. newName := data["newName"].(string)
  2925. newColor := data["newColor"].(string)
  2926. for i, opt := range key.Options {
  2927. if oldName == opt.Name {
  2928. key.Options[i].Name = newName
  2929. key.Options[i].Color = newColor
  2930. break
  2931. }
  2932. }
  2933. // 如果存在选项对应的值,需要更新值中的选项
  2934. for _, keyValues := range attrView.KeyValues {
  2935. if keyValues.Key.ID != operation.ID {
  2936. continue
  2937. }
  2938. for _, value := range keyValues.Values {
  2939. if nil == value || nil == value.MSelect {
  2940. continue
  2941. }
  2942. for i, opt := range value.MSelect {
  2943. if oldName == opt.Content {
  2944. value.MSelect[i].Content = newName
  2945. value.MSelect[i].Color = newColor
  2946. break
  2947. }
  2948. }
  2949. }
  2950. break
  2951. }
  2952. // 如果存在选项对应的过滤器,需要更新过滤器中设置的选项值
  2953. // Database select field filters follow option editing changes https://github.com/siyuan-note/siyuan/issues/10881
  2954. for _, view := range attrView.Views {
  2955. switch view.LayoutType {
  2956. case av.LayoutTypeTable:
  2957. table := view.Table
  2958. for _, filter := range table.Filters {
  2959. if filter.Column != key.ID {
  2960. continue
  2961. }
  2962. if nil != filter.Value && (av.KeyTypeSelect == filter.Value.Type || av.KeyTypeMSelect == filter.Value.Type) {
  2963. for i, opt := range filter.Value.MSelect {
  2964. if oldName == opt.Content {
  2965. filter.Value.MSelect[i].Content = newName
  2966. filter.Value.MSelect[i].Color = newColor
  2967. break
  2968. }
  2969. }
  2970. }
  2971. }
  2972. }
  2973. }
  2974. err = av.SaveAttributeView(attrView)
  2975. return
  2976. }
  2977. func getAttrViewViewByBlockID(attrView *av.AttributeView, blockID string) (ret *av.View, err error) {
  2978. node, _, _ := getNodeByBlockID(nil, blockID)
  2979. var viewID string
  2980. if nil != node {
  2981. viewID = node.IALAttr(av.NodeAttrView)
  2982. }
  2983. return attrView.GetCurrentView(viewID)
  2984. }