repository.go 55 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036
  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. "crypto/rand"
  20. "crypto/sha256"
  21. "encoding/base64"
  22. "errors"
  23. "fmt"
  24. "math"
  25. "net/http"
  26. "os"
  27. "path"
  28. "path/filepath"
  29. "sort"
  30. "strings"
  31. "sync"
  32. "sync/atomic"
  33. "time"
  34. "github.com/88250/go-humanize"
  35. "github.com/88250/gulu"
  36. "github.com/88250/lute"
  37. "github.com/88250/lute/ast"
  38. "github.com/88250/lute/html"
  39. "github.com/88250/lute/parse"
  40. "github.com/88250/lute/render"
  41. "github.com/emirpasic/gods/sets/hashset"
  42. "github.com/siyuan-note/dejavu"
  43. "github.com/siyuan-note/dejavu/cloud"
  44. "github.com/siyuan-note/dejavu/entity"
  45. "github.com/siyuan-note/encryption"
  46. "github.com/siyuan-note/eventbus"
  47. "github.com/siyuan-note/httpclient"
  48. "github.com/siyuan-note/logging"
  49. "github.com/siyuan-note/siyuan/kernel/conf"
  50. "github.com/siyuan-note/siyuan/kernel/filesys"
  51. "github.com/siyuan-note/siyuan/kernel/sql"
  52. "github.com/siyuan-note/siyuan/kernel/task"
  53. "github.com/siyuan-note/siyuan/kernel/treenode"
  54. "github.com/siyuan-note/siyuan/kernel/util"
  55. "github.com/studio-b12/gowebdav"
  56. )
  57. func GetRepoFile(fileID string) (ret []byte, p string, err error) {
  58. if 1 > len(Conf.Repo.Key) {
  59. err = errors.New(Conf.Language(26))
  60. return
  61. }
  62. repo, err := newRepository()
  63. if nil != err {
  64. return
  65. }
  66. file, err := repo.GetFile(fileID)
  67. if nil != err {
  68. return
  69. }
  70. ret, err = repo.OpenFile(file)
  71. p = file.Path
  72. return
  73. }
  74. func OpenRepoSnapshotDoc(fileID string) (content string, isProtyleDoc bool, updated int64, err error) {
  75. if 1 > len(Conf.Repo.Key) {
  76. err = errors.New(Conf.Language(26))
  77. return
  78. }
  79. repo, err := newRepository()
  80. if nil != err {
  81. return
  82. }
  83. file, err := repo.GetFile(fileID)
  84. if nil != err {
  85. return
  86. }
  87. data, err := repo.OpenFile(file)
  88. if nil != err {
  89. return
  90. }
  91. updated = file.Updated
  92. if strings.HasSuffix(file.Path, ".sy") {
  93. luteEngine := NewLute()
  94. var snapshotTree *parse.Tree
  95. isProtyleDoc, snapshotTree, err = parseTreeInSnapshot(data, luteEngine)
  96. if nil != err {
  97. logging.LogErrorf("parse tree from snapshot file [%s] failed", fileID)
  98. return
  99. }
  100. if !isProtyleDoc {
  101. renderTree := &parse.Tree{Root: &ast.Node{Type: ast.NodeDocument}}
  102. var unlinks []*ast.Node
  103. ast.Walk(snapshotTree.Root, func(n *ast.Node, entering bool) ast.WalkStatus {
  104. if !entering {
  105. return ast.WalkContinue
  106. }
  107. n.RemoveIALAttr("heading-fold")
  108. n.RemoveIALAttr("fold")
  109. return ast.WalkContinue
  110. })
  111. for _, unlink := range unlinks {
  112. unlink.Unlink()
  113. }
  114. var appends []*ast.Node
  115. for n := snapshotTree.Root.FirstChild; nil != n; n = n.Next {
  116. appends = append(appends, n)
  117. }
  118. for _, n := range appends {
  119. renderTree.Root.AppendChild(n)
  120. }
  121. snapshotTree = renderTree
  122. }
  123. luteEngine.RenderOptions.ProtyleContenteditable = false
  124. if isProtyleDoc {
  125. util.PushMsg(Conf.Language(36), 5000)
  126. formatRenderer := render.NewFormatRenderer(snapshotTree, luteEngine.RenderOptions)
  127. content = gulu.Str.FromBytes(formatRenderer.Render())
  128. } else {
  129. content = luteEngine.Tree2BlockDOM(snapshotTree, luteEngine.RenderOptions)
  130. }
  131. } else {
  132. isProtyleDoc = true
  133. if strings.HasSuffix(file.Path, ".json") {
  134. content = gulu.Str.FromBytes(data)
  135. } else {
  136. if strings.Contains(file.Path, "assets/") { // 剔除笔记本级或者文档级资源文件路径前缀
  137. file.Path = file.Path[strings.Index(file.Path, "assets/"):]
  138. if util.IsDisplayableAsset(file.Path) {
  139. dir, f := filepath.Split(file.Path)
  140. tempRepoDiffDir := filepath.Join(util.TempDir, "repo", "diff", dir)
  141. if mkErr := os.MkdirAll(tempRepoDiffDir, 0755); nil != mkErr {
  142. logging.LogErrorf("mkdir [%s] failed: %v", tempRepoDiffDir, mkErr)
  143. } else {
  144. if wrErr := os.WriteFile(filepath.Join(tempRepoDiffDir, f), data, 0644); nil != wrErr {
  145. logging.LogErrorf("write file [%s] failed: %v", filepath.Join(tempRepoDiffDir, file.Path), wrErr)
  146. }
  147. }
  148. content = path.Join("repo", "diff", file.Path)
  149. }
  150. } else {
  151. content = file.Path
  152. }
  153. }
  154. }
  155. return
  156. }
  157. type LeftRightDiff struct {
  158. LeftIndex *DiffIndex `json:"leftIndex"`
  159. RightIndex *DiffIndex `json:"rightIndex"`
  160. AddsLeft []*DiffFile `json:"addsLeft"`
  161. UpdatesLeft []*DiffFile `json:"updatesLeft"`
  162. UpdatesRight []*DiffFile `json:"updatesRight"`
  163. RemovesRight []*DiffFile `json:"removesRight"`
  164. }
  165. type DiffFile struct {
  166. FileID string `json:"fileID"`
  167. Title string `json:"title"`
  168. Path string `json:"path"`
  169. HSize string `json:"hSize"`
  170. Updated int64 `json:"updated"`
  171. }
  172. type DiffIndex struct {
  173. ID string `json:"id"`
  174. Created int64 `json:"created"`
  175. }
  176. func DiffRepoSnapshots(left, right string) (ret *LeftRightDiff, err error) {
  177. if 1 > len(Conf.Repo.Key) {
  178. err = errors.New(Conf.Language(26))
  179. return
  180. }
  181. repo, err := newRepository()
  182. if nil != err {
  183. return
  184. }
  185. diff, err := repo.DiffIndex(left, right)
  186. if nil != err {
  187. return
  188. }
  189. ret = &LeftRightDiff{
  190. LeftIndex: &DiffIndex{
  191. ID: diff.LeftIndex.ID,
  192. Created: diff.LeftIndex.Created,
  193. },
  194. RightIndex: &DiffIndex{
  195. ID: diff.RightIndex.ID,
  196. Created: diff.RightIndex.Created,
  197. },
  198. }
  199. luteEngine := NewLute()
  200. for _, removeRight := range diff.RemovesRight {
  201. title, parseErr := parseTitleInSnapshot(removeRight.ID, repo, luteEngine)
  202. if "" == title || nil != parseErr {
  203. continue
  204. }
  205. ret.AddsLeft = append(ret.AddsLeft, &DiffFile{
  206. FileID: removeRight.ID,
  207. Title: title,
  208. Path: removeRight.Path,
  209. HSize: humanize.BytesCustomCeil(uint64(removeRight.Size), 2),
  210. Updated: removeRight.Updated,
  211. })
  212. }
  213. if 1 > len(ret.AddsLeft) {
  214. ret.AddsLeft = []*DiffFile{}
  215. }
  216. for _, addLeft := range diff.AddsLeft {
  217. title, parseErr := parseTitleInSnapshot(addLeft.ID, repo, luteEngine)
  218. if "" == title || nil != parseErr {
  219. continue
  220. }
  221. ret.RemovesRight = append(ret.RemovesRight, &DiffFile{
  222. FileID: addLeft.ID,
  223. Title: title,
  224. Path: addLeft.Path,
  225. HSize: humanize.BytesCustomCeil(uint64(addLeft.Size), 2),
  226. Updated: addLeft.Updated,
  227. })
  228. }
  229. if 1 > len(ret.RemovesRight) {
  230. ret.RemovesRight = []*DiffFile{}
  231. }
  232. for _, updateLeft := range diff.UpdatesLeft {
  233. title, parseErr := parseTitleInSnapshot(updateLeft.ID, repo, luteEngine)
  234. if "" == title || nil != parseErr {
  235. continue
  236. }
  237. ret.UpdatesLeft = append(ret.UpdatesLeft, &DiffFile{
  238. FileID: updateLeft.ID,
  239. Title: title,
  240. Path: updateLeft.Path,
  241. HSize: humanize.BytesCustomCeil(uint64(updateLeft.Size), 2),
  242. Updated: updateLeft.Updated,
  243. })
  244. }
  245. if 1 > len(ret.UpdatesLeft) {
  246. ret.UpdatesLeft = []*DiffFile{}
  247. }
  248. for _, updateRight := range diff.UpdatesRight {
  249. title, parseErr := parseTitleInSnapshot(updateRight.ID, repo, luteEngine)
  250. if "" == title || nil != parseErr {
  251. continue
  252. }
  253. ret.UpdatesRight = append(ret.UpdatesRight, &DiffFile{
  254. FileID: updateRight.ID,
  255. Title: title,
  256. Path: updateRight.Path,
  257. HSize: humanize.BytesCustomCeil(uint64(updateRight.Size), 2),
  258. Updated: updateRight.Updated,
  259. })
  260. }
  261. if 1 > len(ret.UpdatesRight) {
  262. ret.UpdatesRight = []*DiffFile{}
  263. }
  264. return
  265. }
  266. func parseTitleInSnapshot(fileID string, repo *dejavu.Repo, luteEngine *lute.Lute) (title string, err error) {
  267. file, err := repo.GetFile(fileID)
  268. if nil != err {
  269. logging.LogErrorf("get file [%s] failed: %s", fileID, err)
  270. return
  271. }
  272. title = path.Base(file.Path)
  273. if strings.HasSuffix(file.Path, ".sy") {
  274. var data []byte
  275. data, err = repo.OpenFile(file)
  276. if nil != err {
  277. logging.LogErrorf("open file [%s] failed: %s", fileID, err)
  278. return
  279. }
  280. var tree *parse.Tree
  281. tree, err = filesys.ParseJSONWithoutFix(data, luteEngine.ParseOptions)
  282. if nil != err {
  283. logging.LogErrorf("parse file [%s] failed: %s", fileID, err)
  284. return
  285. }
  286. title = tree.Root.IALAttr("title")
  287. }
  288. return
  289. }
  290. func parseTreeInSnapshot(data []byte, luteEngine *lute.Lute) (isProtyleDoc bool, tree *parse.Tree, err error) {
  291. isProtyleDoc = 1024*1024*1 <= len(data)
  292. tree, err = filesys.ParseJSONWithoutFix(data, luteEngine.ParseOptions)
  293. if nil != err {
  294. return
  295. }
  296. return
  297. }
  298. type Snapshot struct {
  299. *dejavu.Log
  300. TypesCount []*TypeCount `json:"typesCount"`
  301. }
  302. type TypeCount struct {
  303. Type string `json:"type"`
  304. Count int `json:"count"`
  305. }
  306. func GetRepoSnapshots(page int) (ret []*Snapshot, pageCount, totalCount int, err error) {
  307. ret = []*Snapshot{}
  308. if 1 > len(Conf.Repo.Key) {
  309. err = errors.New(Conf.Language(26))
  310. return
  311. }
  312. repo, err := newRepository()
  313. if nil != err {
  314. return
  315. }
  316. logs, pageCount, totalCount, err := repo.GetIndexLogs(page, 32)
  317. if nil != err {
  318. if dejavu.ErrNotFoundIndex == err {
  319. logs = []*dejavu.Log{}
  320. err = nil
  321. return
  322. }
  323. logging.LogErrorf("get data repo index logs failed: %s", err)
  324. return
  325. }
  326. ret = buildSnapshots(logs)
  327. if 1 > len(ret) {
  328. ret = []*Snapshot{}
  329. }
  330. return
  331. }
  332. func buildSnapshots(logs []*dejavu.Log) (ret []*Snapshot) {
  333. for _, l := range logs {
  334. typesCount := statTypesByPath(l.Files)
  335. l.Files = nil // 置空,否则返回前端数据量太大
  336. ret = append(ret, &Snapshot{
  337. Log: l,
  338. TypesCount: typesCount,
  339. })
  340. }
  341. return
  342. }
  343. func statTypesByPath(files []*entity.File) (ret []*TypeCount) {
  344. for _, f := range files {
  345. ext := path.Ext(f.Path)
  346. if "" == ext {
  347. ext = "NoExt"
  348. }
  349. found := false
  350. for _, tc := range ret {
  351. if tc.Type == ext {
  352. tc.Count++
  353. found = true
  354. break
  355. }
  356. }
  357. if !found {
  358. ret = append(ret, &TypeCount{Type: ext, Count: 1})
  359. }
  360. }
  361. sort.Slice(ret, func(i, j int) bool { return ret[i].Count > ret[j].Count })
  362. if 10 < len(ret) {
  363. otherCount := 0
  364. for _, tc := range ret[10:] {
  365. tc.Count += otherCount
  366. }
  367. other := &TypeCount{
  368. Type: "Other",
  369. Count: otherCount,
  370. }
  371. ret = append(ret[:10], other)
  372. }
  373. return
  374. }
  375. func ImportRepoKey(base64Key string) (err error) {
  376. util.PushMsg(Conf.Language(136), 3000)
  377. base64Key = strings.TrimSpace(base64Key)
  378. base64Key = gulu.Str.RemoveInvisible(base64Key)
  379. if 1 > len(base64Key) {
  380. err = errors.New(Conf.Language(142))
  381. return
  382. }
  383. key, err := base64.StdEncoding.DecodeString(base64Key)
  384. if nil != err {
  385. logging.LogErrorf("import data repo key failed: %s", err)
  386. return errors.New(Conf.Language(157))
  387. }
  388. if 32 != len(key) {
  389. return errors.New(Conf.Language(157))
  390. }
  391. Conf.Repo.Key = key
  392. Conf.Save()
  393. if err = os.RemoveAll(Conf.Repo.GetSaveDir()); nil != err {
  394. return
  395. }
  396. if err = os.MkdirAll(Conf.Repo.GetSaveDir(), 0755); nil != err {
  397. return
  398. }
  399. initDataRepo()
  400. return
  401. }
  402. func ResetRepo() (err error) {
  403. logging.LogInfof("resetting data repo...")
  404. msgId := util.PushMsg(Conf.Language(144), 1000*60)
  405. repo, err := newRepository()
  406. if nil != err {
  407. return
  408. }
  409. if err = repo.Reset(); nil != err {
  410. logging.LogErrorf("reset data repo failed: %s", err)
  411. return
  412. }
  413. logging.LogInfof("reset data repo completed")
  414. Conf.Repo.Key = nil
  415. Conf.Sync.Enabled = false
  416. Conf.Save()
  417. util.PushUpdateMsg(msgId, Conf.Language(145), 3000)
  418. go func() {
  419. time.Sleep(2 * time.Second)
  420. util.ReloadUI()
  421. }()
  422. return
  423. }
  424. func PurgeCloud() (err error) {
  425. msg := Conf.Language(223)
  426. util.PushEndlessProgress(msg)
  427. defer util.PushClearProgress()
  428. repo, err := newRepository()
  429. if nil != err {
  430. return
  431. }
  432. stat, err := repo.PurgeCloud()
  433. if nil != err {
  434. return
  435. }
  436. deletedIndexes := stat.Indexes
  437. deletedObjects := stat.Objects
  438. deletedSize := humanize.BytesCustomCeil(uint64(stat.Size), 2)
  439. msg = fmt.Sprintf(Conf.Language(232), deletedIndexes, deletedObjects, deletedSize)
  440. util.PushMsg(msg, 5000)
  441. return
  442. }
  443. func PurgeRepo() (err error) {
  444. msg := Conf.Language(202)
  445. util.PushEndlessProgress(msg)
  446. defer util.PushClearProgress()
  447. repo, err := newRepository()
  448. if nil != err {
  449. return
  450. }
  451. stat, err := repo.Purge()
  452. if nil != err {
  453. return
  454. }
  455. deletedIndexes := stat.Indexes
  456. deletedObjects := stat.Objects
  457. deletedSize := humanize.BytesCustomCeil(uint64(stat.Size), 2)
  458. msg = fmt.Sprintf(Conf.Language(203), deletedIndexes, deletedObjects, deletedSize)
  459. util.PushMsg(msg, 5000)
  460. return
  461. }
  462. func InitRepoKeyFromPassphrase(passphrase string) (err error) {
  463. passphrase = gulu.Str.RemoveInvisible(passphrase)
  464. passphrase = strings.TrimSpace(passphrase)
  465. if "" == passphrase {
  466. return errors.New(Conf.Language(142))
  467. }
  468. util.PushMsg(Conf.Language(136), 3000)
  469. if err = os.RemoveAll(Conf.Repo.GetSaveDir()); nil != err {
  470. return
  471. }
  472. if err = os.MkdirAll(Conf.Repo.GetSaveDir(), 0755); nil != err {
  473. return
  474. }
  475. var key []byte
  476. base64Data, base64Err := base64.StdEncoding.DecodeString(passphrase)
  477. if nil == base64Err && 32 == len(base64Data) {
  478. // 改进数据仓库 `通过密码生成密钥` https://github.com/siyuan-note/siyuan/issues/6782
  479. logging.LogInfof("passphrase is base64 encoded, use it as key directly")
  480. key = base64Data
  481. } else {
  482. salt := fmt.Sprintf("%x", sha256.Sum256([]byte(passphrase)))[:16]
  483. key, err = encryption.KDF(passphrase, salt)
  484. if nil != err {
  485. logging.LogErrorf("init data repo key failed: %s", err)
  486. return
  487. }
  488. }
  489. Conf.Repo.Key = key
  490. Conf.Save()
  491. initDataRepo()
  492. return
  493. }
  494. func InitRepoKey() (err error) {
  495. util.PushMsg(Conf.Language(136), 3000)
  496. if err = os.RemoveAll(Conf.Repo.GetSaveDir()); nil != err {
  497. return
  498. }
  499. if err = os.MkdirAll(Conf.Repo.GetSaveDir(), 0755); nil != err {
  500. return
  501. }
  502. randomBytes := make([]byte, 16)
  503. _, err = rand.Read(randomBytes)
  504. if nil != err {
  505. return
  506. }
  507. password := string(randomBytes)
  508. randomBytes = make([]byte, 16)
  509. _, err = rand.Read(randomBytes)
  510. if nil != err {
  511. logging.LogErrorf("init data repo key failed: %s", err)
  512. return
  513. }
  514. salt := string(randomBytes)
  515. key, err := encryption.KDF(password, salt)
  516. if nil != err {
  517. logging.LogErrorf("init data repo key failed: %s", err)
  518. return
  519. }
  520. Conf.Repo.Key = key
  521. Conf.Save()
  522. initDataRepo()
  523. return
  524. }
  525. func initDataRepo() {
  526. time.Sleep(1 * time.Second)
  527. util.PushMsg(Conf.Language(138), 3000)
  528. time.Sleep(1 * time.Second)
  529. if initErr := IndexRepo("[Init] Init local data repo"); nil != initErr {
  530. util.PushErrMsg(fmt.Sprintf(Conf.Language(140), initErr), 0)
  531. }
  532. }
  533. func CheckoutRepo(id string) {
  534. task.AppendTask(task.RepoCheckout, checkoutRepo, id)
  535. }
  536. func checkoutRepo(id string) {
  537. var err error
  538. if 1 > len(Conf.Repo.Key) {
  539. util.PushErrMsg(Conf.Language(26), 7000)
  540. return
  541. }
  542. repo, err := newRepository()
  543. if nil != err {
  544. logging.LogErrorf("new repository failed: %s", err)
  545. util.PushErrMsg(Conf.Language(141), 7000)
  546. return
  547. }
  548. util.PushEndlessProgress(Conf.Language(63))
  549. WaitForWritingFiles()
  550. CloseWatchAssets()
  551. defer WatchAssets()
  552. CloseWatchEmojis()
  553. defer WatchEmojis()
  554. // 恢复快照时自动暂停同步,避免刚刚恢复后的数据又被同步覆盖
  555. syncEnabled := Conf.Sync.Enabled
  556. Conf.Sync.Enabled = false
  557. Conf.Save()
  558. _, _, err = repo.Checkout(id, map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress})
  559. if nil != err {
  560. logging.LogErrorf("checkout repository failed: %s", err)
  561. util.PushClearProgress()
  562. util.PushErrMsg(Conf.Language(141), 7000)
  563. return
  564. }
  565. task.AppendTask(task.DatabaseIndexFull, fullReindex)
  566. task.AppendTask(task.DatabaseIndexRef, IndexRefs)
  567. go func() {
  568. sql.WaitForWritingDatabase()
  569. ResetVirtualBlockRefCache()
  570. }()
  571. task.AppendTask(task.ReloadUI, util.ReloadUIResetScroll)
  572. if syncEnabled {
  573. func() {
  574. time.Sleep(5 * time.Second)
  575. util.PushMsg(Conf.Language(134), 0)
  576. }()
  577. }
  578. return
  579. }
  580. func DownloadCloudSnapshot(tag, id string) (err error) {
  581. if 1 > len(Conf.Repo.Key) {
  582. err = errors.New(Conf.Language(26))
  583. return
  584. }
  585. repo, err := newRepository()
  586. if nil != err {
  587. return
  588. }
  589. switch Conf.Sync.Provider {
  590. case conf.ProviderSiYuan:
  591. if !IsSubscriber() {
  592. util.PushErrMsg(Conf.Language(29), 5000)
  593. return
  594. }
  595. case conf.ProviderWebDAV, conf.ProviderS3:
  596. if !IsPaidUser() {
  597. util.PushErrMsg(Conf.Language(214), 5000)
  598. return
  599. }
  600. }
  601. defer util.PushClearProgress()
  602. var downloadFileCount, downloadChunkCount int
  603. var downloadBytes int64
  604. if "" == tag {
  605. downloadFileCount, downloadChunkCount, downloadBytes, err = repo.DownloadIndex(id, map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress})
  606. } else {
  607. downloadFileCount, downloadChunkCount, downloadBytes, err = repo.DownloadTagIndex(tag, id, map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress})
  608. }
  609. if nil != err {
  610. return
  611. }
  612. msg := fmt.Sprintf(Conf.Language(153), downloadFileCount, downloadChunkCount, humanize.BytesCustomCeil(uint64(downloadBytes), 2))
  613. util.PushMsg(msg, 5000)
  614. util.PushStatusBar(msg)
  615. return
  616. }
  617. func UploadCloudSnapshot(tag, id string) (err error) {
  618. if 1 > len(Conf.Repo.Key) {
  619. err = errors.New(Conf.Language(26))
  620. return
  621. }
  622. repo, err := newRepository()
  623. if nil != err {
  624. return
  625. }
  626. switch Conf.Sync.Provider {
  627. case conf.ProviderSiYuan:
  628. if !IsSubscriber() {
  629. util.PushErrMsg(Conf.Language(29), 5000)
  630. return
  631. }
  632. case conf.ProviderWebDAV, conf.ProviderS3:
  633. if !IsPaidUser() {
  634. util.PushErrMsg(Conf.Language(214), 5000)
  635. return
  636. }
  637. }
  638. util.PushEndlessProgress(Conf.Language(116))
  639. defer util.PushClearProgress()
  640. uploadFileCount, uploadChunkCount, uploadBytes, err := repo.UploadTagIndex(tag, id, map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress})
  641. if nil != err {
  642. if errors.Is(err, dejavu.ErrCloudBackupCountExceeded) {
  643. err = fmt.Errorf(Conf.Language(84), Conf.Language(154))
  644. return
  645. }
  646. err = errors.New(fmt.Sprintf(Conf.Language(84), formatRepoErrorMsg(err)))
  647. return
  648. }
  649. msg := fmt.Sprintf(Conf.Language(152), uploadFileCount, uploadChunkCount, humanize.BytesCustomCeil(uint64(uploadBytes), 2))
  650. util.PushMsg(msg, 5000)
  651. util.PushStatusBar(msg)
  652. return
  653. }
  654. func RemoveCloudRepoTag(tag string) (err error) {
  655. if 1 > len(Conf.Repo.Key) {
  656. err = errors.New(Conf.Language(26))
  657. return
  658. }
  659. if "" == tag {
  660. err = errors.New("tag is empty")
  661. return
  662. }
  663. repo, err := newRepository()
  664. if nil != err {
  665. return
  666. }
  667. switch Conf.Sync.Provider {
  668. case conf.ProviderSiYuan:
  669. if !IsSubscriber() {
  670. util.PushErrMsg(Conf.Language(29), 5000)
  671. return
  672. }
  673. case conf.ProviderWebDAV, conf.ProviderS3:
  674. if !IsPaidUser() {
  675. util.PushErrMsg(Conf.Language(214), 5000)
  676. return
  677. }
  678. }
  679. err = repo.RemoveCloudRepoTag(tag)
  680. if nil != err {
  681. return
  682. }
  683. return
  684. }
  685. func GetCloudRepoTagSnapshots() (ret []*dejavu.Log, err error) {
  686. ret = []*dejavu.Log{}
  687. if 1 > len(Conf.Repo.Key) {
  688. err = errors.New(Conf.Language(26))
  689. return
  690. }
  691. repo, err := newRepository()
  692. if nil != err {
  693. return
  694. }
  695. switch Conf.Sync.Provider {
  696. case conf.ProviderSiYuan:
  697. if !IsSubscriber() {
  698. util.PushErrMsg(Conf.Language(29), 5000)
  699. return
  700. }
  701. case conf.ProviderWebDAV, conf.ProviderS3:
  702. if !IsPaidUser() {
  703. util.PushErrMsg(Conf.Language(214), 5000)
  704. return
  705. }
  706. }
  707. logs, err := repo.GetCloudRepoTagLogs(map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar})
  708. if nil != err {
  709. return
  710. }
  711. ret = logs
  712. if 1 > len(ret) {
  713. ret = []*dejavu.Log{}
  714. }
  715. return
  716. }
  717. func GetCloudRepoSnapshots(page int) (ret []*dejavu.Log, pageCount, totalCount int, err error) {
  718. ret = []*dejavu.Log{}
  719. if 1 > len(Conf.Repo.Key) {
  720. err = errors.New(Conf.Language(26))
  721. return
  722. }
  723. repo, err := newRepository()
  724. if nil != err {
  725. return
  726. }
  727. switch Conf.Sync.Provider {
  728. case conf.ProviderSiYuan:
  729. if !IsSubscriber() {
  730. util.PushErrMsg(Conf.Language(29), 5000)
  731. return
  732. }
  733. case conf.ProviderWebDAV, conf.ProviderS3:
  734. if !IsPaidUser() {
  735. util.PushErrMsg(Conf.Language(214), 5000)
  736. return
  737. }
  738. }
  739. if 1 > page {
  740. page = 1
  741. }
  742. logs, pageCount, totalCount, err := repo.GetCloudRepoLogs(page)
  743. if nil != err {
  744. return
  745. }
  746. ret = logs
  747. if 1 > len(ret) {
  748. ret = []*dejavu.Log{}
  749. }
  750. return
  751. }
  752. func GetTagSnapshots() (ret []*Snapshot, err error) {
  753. ret = []*Snapshot{}
  754. if 1 > len(Conf.Repo.Key) {
  755. err = errors.New(Conf.Language(26))
  756. return
  757. }
  758. repo, err := newRepository()
  759. if nil != err {
  760. return
  761. }
  762. logs, err := repo.GetTagLogs()
  763. if nil != err {
  764. return
  765. }
  766. ret = buildSnapshots(logs)
  767. if 1 > len(ret) {
  768. ret = []*Snapshot{}
  769. }
  770. return
  771. }
  772. func RemoveTagSnapshot(tag string) (err error) {
  773. if 1 > len(Conf.Repo.Key) {
  774. err = errors.New(Conf.Language(26))
  775. return
  776. }
  777. repo, err := newRepository()
  778. if nil != err {
  779. return
  780. }
  781. err = repo.RemoveTag(tag)
  782. return
  783. }
  784. func TagSnapshot(id, name string) (err error) {
  785. if 1 > len(Conf.Repo.Key) {
  786. err = errors.New(Conf.Language(26))
  787. return
  788. }
  789. name = strings.TrimSpace(name)
  790. name = gulu.Str.RemoveInvisible(name)
  791. if "" == name {
  792. err = errors.New(Conf.Language(142))
  793. return
  794. }
  795. if !gulu.File.IsValidFilename(name) {
  796. err = errors.New(Conf.Language(151))
  797. return
  798. }
  799. repo, err := newRepository()
  800. if nil != err {
  801. return
  802. }
  803. index, err := repo.GetIndex(id)
  804. if nil != err {
  805. return
  806. }
  807. if err = repo.AddTag(index.ID, name); nil != err {
  808. msg := fmt.Sprintf("Add tag to data snapshot [%s] failed: %s", index.ID, err)
  809. util.PushStatusBar(msg)
  810. return
  811. }
  812. return
  813. }
  814. func IndexRepo(memo string) (err error) {
  815. if 1 > len(Conf.Repo.Key) {
  816. err = errors.New(Conf.Language(26))
  817. return
  818. }
  819. memo = strings.TrimSpace(memo)
  820. memo = gulu.Str.RemoveInvisible(memo)
  821. if "" == memo {
  822. err = errors.New(Conf.Language(142))
  823. return
  824. }
  825. repo, err := newRepository()
  826. if nil != err {
  827. return
  828. }
  829. util.PushEndlessProgress(Conf.Language(143))
  830. start := time.Now()
  831. latest, _ := repo.Latest()
  832. WaitForWritingFiles()
  833. index, err := repo.Index(memo, map[string]interface{}{
  834. eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress,
  835. })
  836. if nil != err {
  837. util.PushStatusBar("Index data repo failed: " + html.EscapeString(err.Error()))
  838. return
  839. }
  840. elapsed := time.Since(start)
  841. if nil == latest || latest.ID != index.ID {
  842. msg := fmt.Sprintf(Conf.Language(147), elapsed.Seconds())
  843. util.PushStatusBar(msg)
  844. util.PushMsg(msg, 5000)
  845. } else {
  846. msg := fmt.Sprintf(Conf.Language(148), elapsed.Seconds())
  847. util.PushStatusBar(msg)
  848. util.PushMsg(msg, 5000)
  849. }
  850. util.PushClearProgress()
  851. return
  852. }
  853. var syncingFiles = sync.Map{}
  854. var syncingStorages = atomic.Bool{}
  855. func waitForSyncingStorages() {
  856. for isSyncingStorages() {
  857. time.Sleep(time.Second)
  858. }
  859. }
  860. func isSyncingStorages() bool {
  861. return syncingStorages.Load() || isBootSyncing.Load()
  862. }
  863. func IsSyncingFile(rootID string) (ret bool) {
  864. _, ret = syncingFiles.Load(rootID)
  865. return
  866. }
  867. func syncRepoDownload() (err error) {
  868. if 1 > len(Conf.Repo.Key) {
  869. planSyncAfter(fixSyncInterval)
  870. msg := Conf.Language(26)
  871. util.PushStatusBar(msg)
  872. util.PushErrMsg(msg, 0)
  873. err = errors.New(msg)
  874. return
  875. }
  876. repo, err := newRepository()
  877. if nil != err {
  878. planSyncAfter(fixSyncInterval)
  879. msg := fmt.Sprintf("sync repo failed: %s", err)
  880. logging.LogErrorf(msg)
  881. util.PushStatusBar(msg)
  882. util.PushErrMsg(msg, 0)
  883. return
  884. }
  885. logging.LogInfof("downloading data repo [device=%s, kernel=%s, provider=%d, mode=%s/%t]", Conf.System.ID, KernelID, Conf.Sync.Provider, "d", true)
  886. start := time.Now()
  887. _, _, err = indexRepoBeforeCloudSync(repo)
  888. if nil != err {
  889. planSyncAfter(fixSyncInterval)
  890. logging.LogErrorf("sync data repo download failed: %s", err)
  891. msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
  892. Conf.Sync.Stat = msg
  893. Conf.Save()
  894. util.PushStatusBar(msg)
  895. util.PushErrMsg(msg, 0)
  896. return
  897. }
  898. syncContext := map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar}
  899. mergeResult, trafficStat, err := repo.SyncDownload(syncContext)
  900. elapsed := time.Since(start)
  901. if nil != err {
  902. planSyncAfter(fixSyncInterval)
  903. logging.LogErrorf("sync data repo download failed: %s", err)
  904. msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
  905. if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
  906. u := Conf.GetUser()
  907. msg = fmt.Sprintf(Conf.Language(43), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
  908. if 2 == u.UserSiYuanSubscriptionPlan {
  909. msg = fmt.Sprintf(Conf.Language(68), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
  910. }
  911. }
  912. Conf.Sync.Stat = msg
  913. Conf.Save()
  914. util.PushStatusBar(msg)
  915. util.PushErrMsg(msg, 0)
  916. return
  917. }
  918. util.PushStatusBar(fmt.Sprintf(Conf.Language(149), elapsed.Seconds()))
  919. Conf.Sync.Synced = util.CurrentTimeMillis()
  920. msg := fmt.Sprintf(Conf.Language(150), trafficStat.UploadFileCount, trafficStat.DownloadFileCount, trafficStat.UploadChunkCount, trafficStat.DownloadChunkCount, humanize.BytesCustomCeil(uint64(trafficStat.UploadBytes), 2), humanize.BytesCustomFloor(uint64(trafficStat.DownloadBytes), 2))
  921. Conf.Sync.Stat = msg
  922. Conf.Save()
  923. autoSyncErrCount = 0
  924. BootSyncSucc = 0
  925. processSyncMergeResult(false, true, mergeResult, trafficStat, "d", elapsed)
  926. return
  927. }
  928. func syncRepoUpload() (err error) {
  929. if 1 > len(Conf.Repo.Key) {
  930. planSyncAfter(fixSyncInterval)
  931. msg := Conf.Language(26)
  932. util.PushStatusBar(msg)
  933. util.PushErrMsg(msg, 0)
  934. err = errors.New(msg)
  935. return
  936. }
  937. repo, err := newRepository()
  938. if nil != err {
  939. planSyncAfter(fixSyncInterval)
  940. msg := fmt.Sprintf("sync repo failed: %s", err)
  941. logging.LogErrorf(msg)
  942. util.PushStatusBar(msg)
  943. util.PushErrMsg(msg, 0)
  944. return
  945. }
  946. logging.LogInfof("uploading data repo [device=%s, kernel=%s, provider=%d, mode=%s/%t]", Conf.System.ID, KernelID, Conf.Sync.Provider, "u", true)
  947. start := time.Now()
  948. _, _, err = indexRepoBeforeCloudSync(repo)
  949. if nil != err {
  950. planSyncAfter(fixSyncInterval)
  951. logging.LogErrorf("sync data repo upload failed: %s", err)
  952. msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
  953. Conf.Sync.Stat = msg
  954. Conf.Save()
  955. util.PushStatusBar(msg)
  956. util.PushErrMsg(msg, 0)
  957. return
  958. }
  959. syncContext := map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar}
  960. trafficStat, err := repo.SyncUpload(syncContext)
  961. elapsed := time.Since(start)
  962. if nil != err {
  963. planSyncAfter(fixSyncInterval)
  964. logging.LogErrorf("sync data repo upload failed: %s", err)
  965. msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
  966. if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
  967. u := Conf.GetUser()
  968. msg = fmt.Sprintf(Conf.Language(43), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
  969. if 2 == u.UserSiYuanSubscriptionPlan {
  970. msg = fmt.Sprintf(Conf.Language(68), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
  971. }
  972. }
  973. Conf.Sync.Stat = msg
  974. Conf.Save()
  975. util.PushStatusBar(msg)
  976. util.PushErrMsg(msg, 0)
  977. return
  978. }
  979. util.PushStatusBar(fmt.Sprintf(Conf.Language(149), elapsed.Seconds()))
  980. Conf.Sync.Synced = util.CurrentTimeMillis()
  981. msg := fmt.Sprintf(Conf.Language(150), trafficStat.UploadFileCount, trafficStat.DownloadFileCount, trafficStat.UploadChunkCount, trafficStat.DownloadChunkCount, humanize.BytesCustomCeil(uint64(trafficStat.UploadBytes), 2), humanize.BytesCustomCeil(uint64(trafficStat.DownloadBytes), 2))
  982. Conf.Sync.Stat = msg
  983. Conf.Save()
  984. autoSyncErrCount = 0
  985. BootSyncSucc = 0
  986. processSyncMergeResult(false, true, &dejavu.MergeResult{}, trafficStat, "u", elapsed)
  987. return
  988. }
  989. var isBootSyncing = atomic.Bool{}
  990. func bootSyncRepo() (err error) {
  991. if 1 > len(Conf.Repo.Key) {
  992. autoSyncErrCount++
  993. planSyncAfter(fixSyncInterval)
  994. msg := Conf.Language(26)
  995. util.PushStatusBar(msg)
  996. util.PushErrMsg(msg, 0)
  997. err = errors.New(msg)
  998. return
  999. }
  1000. repo, err := newRepository()
  1001. if nil != err {
  1002. autoSyncErrCount++
  1003. planSyncAfter(fixSyncInterval)
  1004. msg := fmt.Sprintf("sync repo failed: %s", html.EscapeString(err.Error()))
  1005. logging.LogErrorf(msg)
  1006. util.PushStatusBar(msg)
  1007. util.PushErrMsg(msg, 0)
  1008. return
  1009. }
  1010. isBootSyncing.Store(true)
  1011. start := time.Now()
  1012. _, _, err = indexRepoBeforeCloudSync(repo)
  1013. if nil != err {
  1014. autoSyncErrCount++
  1015. planSyncAfter(fixSyncInterval)
  1016. msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
  1017. Conf.Sync.Stat = msg
  1018. Conf.Save()
  1019. util.PushStatusBar(msg)
  1020. util.PushErrMsg(msg, 0)
  1021. BootSyncSucc = 1
  1022. isBootSyncing.Store(false)
  1023. return
  1024. }
  1025. syncContext := map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar}
  1026. fetchedFiles, err := repo.GetSyncCloudFiles(syncContext)
  1027. if errors.Is(err, dejavu.ErrRepoFatal) {
  1028. autoSyncErrCount++
  1029. planSyncAfter(fixSyncInterval)
  1030. msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
  1031. Conf.Sync.Stat = msg
  1032. Conf.Save()
  1033. util.PushStatusBar(msg)
  1034. util.PushErrMsg(msg, 0)
  1035. BootSyncSucc = 1
  1036. isBootSyncing.Store(false)
  1037. return
  1038. }
  1039. syncingFiles = sync.Map{}
  1040. syncingStorages.Store(false)
  1041. for _, fetchedFile := range fetchedFiles {
  1042. name := path.Base(fetchedFile.Path)
  1043. if strings.HasSuffix(name, ".sy") {
  1044. id := name[:len(name)-3]
  1045. syncingFiles.Store(id, true)
  1046. continue
  1047. }
  1048. if strings.HasPrefix(fetchedFile.Path, "/storage/") {
  1049. syncingStorages.Store(true)
  1050. }
  1051. }
  1052. elapsed := time.Since(start)
  1053. logging.LogInfof("boot get sync cloud files elapsed [%.2fs]", elapsed.Seconds())
  1054. if nil != err {
  1055. autoSyncErrCount++
  1056. planSyncAfter(fixSyncInterval)
  1057. logging.LogErrorf("sync data repo failed: %s", err)
  1058. msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
  1059. if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
  1060. u := Conf.GetUser()
  1061. msg = fmt.Sprintf(Conf.Language(43), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
  1062. if 2 == u.UserSiYuanSubscriptionPlan {
  1063. msg = fmt.Sprintf(Conf.Language(68), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
  1064. }
  1065. }
  1066. Conf.Sync.Stat = msg
  1067. Conf.Save()
  1068. util.PushStatusBar(msg)
  1069. util.PushErrMsg(msg, 0)
  1070. BootSyncSucc = 1
  1071. isBootSyncing.Store(false)
  1072. return
  1073. }
  1074. if 0 < len(fetchedFiles) {
  1075. go func() {
  1076. _, syncErr := syncRepo(false, false)
  1077. isBootSyncing.Store(false)
  1078. if nil != err {
  1079. logging.LogErrorf("boot background sync repo failed: %s", syncErr)
  1080. return
  1081. }
  1082. }()
  1083. } else {
  1084. isBootSyncing.Store(false)
  1085. }
  1086. return
  1087. }
  1088. func syncRepo(exit, byHand bool) (dataChanged bool, err error) {
  1089. if 1 > len(Conf.Repo.Key) {
  1090. autoSyncErrCount++
  1091. planSyncAfter(fixSyncInterval)
  1092. msg := Conf.Language(26)
  1093. util.PushStatusBar(msg)
  1094. util.PushErrMsg(msg, 0)
  1095. err = errors.New(msg)
  1096. return
  1097. }
  1098. repo, err := newRepository()
  1099. if nil != err {
  1100. autoSyncErrCount++
  1101. planSyncAfter(fixSyncInterval)
  1102. msg := fmt.Sprintf("sync repo failed: %s", err)
  1103. logging.LogErrorf(msg)
  1104. util.PushStatusBar(msg)
  1105. util.PushErrMsg(msg, 0)
  1106. return
  1107. }
  1108. logging.LogInfof("syncing data repo [device=%s, kernel=%s, provider=%d, mode=%s/%t]", Conf.System.ID, KernelID, Conf.Sync.Provider, "a", byHand)
  1109. start := time.Now()
  1110. beforeIndex, afterIndex, err := indexRepoBeforeCloudSync(repo)
  1111. if nil != err {
  1112. autoSyncErrCount++
  1113. planSyncAfter(fixSyncInterval)
  1114. logging.LogErrorf("sync data repo failed: %s", err)
  1115. msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
  1116. Conf.Sync.Stat = msg
  1117. Conf.Save()
  1118. util.PushStatusBar(msg)
  1119. if 1 > autoSyncErrCount || byHand {
  1120. util.PushErrMsg(msg, 0)
  1121. }
  1122. if exit {
  1123. ExitSyncSucc = 1
  1124. }
  1125. return
  1126. }
  1127. syncContext := map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar}
  1128. mergeResult, trafficStat, err := repo.Sync(syncContext)
  1129. elapsed := time.Since(start)
  1130. if nil != err {
  1131. autoSyncErrCount++
  1132. planSyncAfter(fixSyncInterval)
  1133. logging.LogErrorf("sync data repo failed: %s", err)
  1134. msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
  1135. if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
  1136. u := Conf.GetUser()
  1137. msg = fmt.Sprintf(Conf.Language(43), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
  1138. if 2 == u.UserSiYuanSubscriptionPlan {
  1139. msg = fmt.Sprintf(Conf.Language(68), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
  1140. }
  1141. }
  1142. Conf.Sync.Stat = msg
  1143. Conf.Save()
  1144. util.PushStatusBar(msg)
  1145. if 1 > autoSyncErrCount || byHand {
  1146. util.PushErrMsg(msg, 0)
  1147. }
  1148. if exit {
  1149. ExitSyncSucc = 1
  1150. }
  1151. return
  1152. }
  1153. dataChanged = nil == beforeIndex || beforeIndex.ID != afterIndex.ID || mergeResult.DataChanged()
  1154. util.PushStatusBar(fmt.Sprintf(Conf.Language(149), elapsed.Seconds()))
  1155. Conf.Sync.Synced = util.CurrentTimeMillis()
  1156. msg := fmt.Sprintf(Conf.Language(150), trafficStat.UploadFileCount, trafficStat.DownloadFileCount, trafficStat.UploadChunkCount, trafficStat.DownloadChunkCount, humanize.BytesCustomCeil(uint64(trafficStat.UploadBytes), 2), humanize.BytesCustomCeil(uint64(trafficStat.DownloadBytes), 2))
  1157. Conf.Sync.Stat = msg
  1158. Conf.Save()
  1159. autoSyncErrCount = 0
  1160. processSyncMergeResult(exit, byHand, mergeResult, trafficStat, "a", elapsed)
  1161. if !exit {
  1162. // 首次数据同步执行完成后再执行索引订正 Index fixing should not be performed before data synchronization https://github.com/siyuan-note/siyuan/issues/10761
  1163. go checkIndex()
  1164. }
  1165. return
  1166. }
  1167. func processSyncMergeResult(exit, byHand bool, mergeResult *dejavu.MergeResult, trafficStat *dejavu.TrafficStat, mode string, elapsed time.Duration) {
  1168. logging.LogInfof("synced data repo [device=%s, kernel=%s, provider=%d, mode=%s/%t, ufc=%d, dfc=%d, ucc=%d, dcc=%d, ub=%s, db=%s] in [%.2fs], merge result [conflicts=%d, upserts=%d, removes=%d]\n\n",
  1169. Conf.System.ID, KernelID, Conf.Sync.Provider, mode, byHand,
  1170. trafficStat.UploadFileCount, trafficStat.DownloadFileCount, trafficStat.UploadChunkCount, trafficStat.DownloadChunkCount, humanize.BytesCustomCeil(uint64(trafficStat.UploadBytes), 2), humanize.BytesCustomCeil(uint64(trafficStat.DownloadBytes), 2),
  1171. elapsed.Seconds(),
  1172. len(mergeResult.Conflicts), len(mergeResult.Upserts), len(mergeResult.Removes))
  1173. //logSyncMergeResult(mergeResult)
  1174. var needReloadFiletree bool
  1175. if 0 < len(mergeResult.Conflicts) {
  1176. luteEngine := util.NewLute()
  1177. if Conf.Sync.GenerateConflictDoc {
  1178. // 云端同步发生冲突时生成副本 https://github.com/siyuan-note/siyuan/issues/5687
  1179. for _, file := range mergeResult.Conflicts {
  1180. if !strings.HasSuffix(file.Path, ".sy") {
  1181. continue
  1182. }
  1183. parts := strings.Split(file.Path[1:], "/")
  1184. if 2 > len(parts) {
  1185. continue
  1186. }
  1187. boxID := parts[0]
  1188. absPath := filepath.Join(util.TempDir, "repo", "sync", "conflicts", mergeResult.Time.Format("2006-01-02-150405"), file.Path)
  1189. tree, loadTreeErr := loadTree(absPath, luteEngine)
  1190. if nil != loadTreeErr {
  1191. logging.LogErrorf("load conflicted file [%s] failed: %s", absPath, loadTreeErr)
  1192. continue
  1193. }
  1194. tree.Box = boxID
  1195. tree.Path = strings.TrimPrefix(file.Path, "/"+boxID)
  1196. resetTree(tree, "Conflicted")
  1197. createTreeTx(tree)
  1198. }
  1199. needReloadFiletree = true
  1200. }
  1201. historyDir := filepath.Join(util.HistoryDir, mergeResult.Time.Format("2006-01-02-150405")+"-sync")
  1202. indexHistoryDir(filepath.Base(historyDir), luteEngine)
  1203. }
  1204. if 1 > len(mergeResult.Upserts) && 1 > len(mergeResult.Removes) && 1 > len(mergeResult.Conflicts) { // 没有数据变更
  1205. syncSameCount.Add(1)
  1206. if 10 < syncSameCount.Load() {
  1207. syncSameCount.Store(5)
  1208. }
  1209. if !byHand {
  1210. delay := time.Minute * time.Duration(int(math.Pow(2, float64(syncSameCount.Load()))))
  1211. if fixSyncInterval.Minutes() > delay.Minutes() {
  1212. delay = time.Minute * 8
  1213. }
  1214. planSyncAfter(delay)
  1215. }
  1216. util.PushClearProgress()
  1217. return
  1218. }
  1219. // 有数据变更,需要重建索引
  1220. var upserts, removes []string
  1221. var upsertTrees int
  1222. // 可能需要重新加载部分功能
  1223. var needReloadFlashcard, needReloadOcrTexts, needReloadPlugin bool
  1224. upsertPluginSet := hashset.New()
  1225. for _, file := range mergeResult.Upserts {
  1226. upserts = append(upserts, file.Path)
  1227. if strings.HasPrefix(file.Path, "/storage/riff/") {
  1228. needReloadFlashcard = true
  1229. }
  1230. if strings.HasPrefix(file.Path, "/assets/ocr-texts.json") {
  1231. needReloadOcrTexts = true
  1232. }
  1233. if strings.HasSuffix(file.Path, "/.siyuan/conf.json") {
  1234. needReloadFiletree = true
  1235. }
  1236. if strings.HasPrefix(file.Path, "/storage/petal/") {
  1237. needReloadPlugin = true
  1238. }
  1239. if strings.HasPrefix(file.Path, "/plugins/") {
  1240. if parts := strings.Split(file.Path, "/"); 2 < len(parts) {
  1241. needReloadPlugin = true
  1242. upsertPluginSet.Add(parts[2])
  1243. }
  1244. }
  1245. if strings.HasSuffix(file.Path, ".sy") {
  1246. upsertTrees++
  1247. }
  1248. }
  1249. removeWidgetDirSet, removePluginSet := hashset.New(), hashset.New()
  1250. for _, file := range mergeResult.Removes {
  1251. removes = append(removes, file.Path)
  1252. if strings.HasPrefix(file.Path, "/storage/riff/") {
  1253. needReloadFlashcard = true
  1254. }
  1255. if strings.HasPrefix(file.Path, "/assets/ocr-texts.json") {
  1256. needReloadOcrTexts = true
  1257. }
  1258. if strings.HasSuffix(file.Path, "/.siyuan/conf.json") {
  1259. needReloadFiletree = true
  1260. }
  1261. if strings.HasPrefix(file.Path, "/storage/petal/") {
  1262. needReloadPlugin = true
  1263. }
  1264. if strings.HasPrefix(file.Path, "/plugins/") {
  1265. if parts := strings.Split(file.Path, "/"); 2 < len(parts) {
  1266. needReloadPlugin = true
  1267. removePluginSet.Add(parts[2])
  1268. }
  1269. }
  1270. if strings.HasPrefix(file.Path, "/widgets/") {
  1271. if parts := strings.Split(file.Path, "/"); 2 < len(parts) {
  1272. removeWidgetDirSet.Add(parts[2])
  1273. }
  1274. }
  1275. }
  1276. if needReloadFlashcard {
  1277. LoadFlashcards()
  1278. }
  1279. if needReloadOcrTexts {
  1280. util.LoadAssetsTexts()
  1281. }
  1282. if needReloadPlugin {
  1283. pushReloadPlugin(upsertPluginSet, removePluginSet)
  1284. }
  1285. for _, widgetDir := range removeWidgetDirSet.Values() {
  1286. widgetDirPath := filepath.Join(util.DataDir, "widgets", widgetDir.(string))
  1287. gulu.File.RemoveEmptyDirs(widgetDirPath)
  1288. }
  1289. syncingFiles = sync.Map{}
  1290. syncingStorages.Store(false)
  1291. if needFullReindex(upsertTrees) { // 改进同步后全量重建索引判断 https://github.com/siyuan-note/siyuan/issues/5764
  1292. FullReindex()
  1293. return
  1294. }
  1295. if needReloadFiletree {
  1296. util.PushReloadFiletree()
  1297. }
  1298. if exit { // 退出时同步不用推送事件
  1299. return
  1300. }
  1301. upsertRootIDs, removeRootIDs := incReindex(upserts, removes)
  1302. go func() {
  1303. util.WaitForUILoaded()
  1304. if 0 < len(upsertRootIDs) || 0 < len(removeRootIDs) {
  1305. util.BroadcastByType("main", "syncMergeResult", 0, "",
  1306. map[string]interface{}{"upsertRootIDs": upsertRootIDs, "removeRootIDs": removeRootIDs})
  1307. }
  1308. time.Sleep(2 * time.Second)
  1309. util.PushStatusBar(fmt.Sprintf(Conf.Language(149), elapsed.Seconds()))
  1310. if 0 < len(mergeResult.Conflicts) {
  1311. syConflict := false
  1312. for _, file := range mergeResult.Conflicts {
  1313. if strings.HasSuffix(file.Path, ".sy") {
  1314. syConflict = true
  1315. break
  1316. }
  1317. }
  1318. if syConflict {
  1319. // 数据同步发生冲突时在界面上进行提醒 https://github.com/siyuan-note/siyuan/issues/7332
  1320. util.PushMsg(Conf.Language(108), 7000)
  1321. }
  1322. }
  1323. }()
  1324. }
  1325. func logSyncMergeResult(mergeResult *dejavu.MergeResult) {
  1326. if 1 > len(mergeResult.Conflicts) && 1 > len(mergeResult.Upserts) && 1 > len(mergeResult.Removes) {
  1327. return
  1328. }
  1329. if 0 < len(mergeResult.Conflicts) {
  1330. logBuilder := bytes.Buffer{}
  1331. for i, f := range mergeResult.Conflicts {
  1332. logBuilder.WriteString(" ")
  1333. logBuilder.WriteString(f.Path)
  1334. if i < len(mergeResult.Conflicts)-1 {
  1335. logBuilder.WriteString("\n")
  1336. }
  1337. }
  1338. logging.LogInfof("sync conflicts:\n%s", logBuilder.String())
  1339. }
  1340. if 0 < len(mergeResult.Upserts) {
  1341. logBuilder := bytes.Buffer{}
  1342. for i, f := range mergeResult.Upserts {
  1343. logBuilder.WriteString(" ")
  1344. logBuilder.WriteString(f.Path)
  1345. if i < len(mergeResult.Upserts)-1 {
  1346. logBuilder.WriteString("\n")
  1347. }
  1348. }
  1349. logging.LogInfof("sync merge upserts:\n%s", logBuilder.String())
  1350. }
  1351. if 0 < len(mergeResult.Removes) {
  1352. logBuilder := bytes.Buffer{}
  1353. for i, f := range mergeResult.Removes {
  1354. logBuilder.WriteString(" ")
  1355. logBuilder.WriteString(f.Path)
  1356. if i < len(mergeResult.Removes)-1 {
  1357. logBuilder.WriteString("\n")
  1358. }
  1359. }
  1360. logging.LogInfof("sync merge removes:\n%s", logBuilder.String())
  1361. }
  1362. }
  1363. func needFullReindex(upsertTrees int) bool {
  1364. return 0.2 < float64(upsertTrees)/float64(treenode.CountTrees())
  1365. }
  1366. var promotedPurgeDataRepo bool
  1367. func indexRepoBeforeCloudSync(repo *dejavu.Repo) (beforeIndex, afterIndex *entity.Index, err error) {
  1368. start := time.Now()
  1369. beforeIndex, _ = repo.Latest()
  1370. afterIndex, err = repo.Index("[Sync] Cloud sync", map[string]interface{}{
  1371. eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar,
  1372. })
  1373. if nil != err {
  1374. logging.LogErrorf("index data repo before cloud sync failed: %s", err)
  1375. return
  1376. }
  1377. elapsed := time.Since(start)
  1378. if nil == beforeIndex || beforeIndex.ID != afterIndex.ID {
  1379. // 对新创建的快照需要更新备注,加入耗时统计
  1380. afterIndex.Memo = fmt.Sprintf("[Sync] Cloud sync, completed in %.2fs", elapsed.Seconds())
  1381. if err = repo.PutIndex(afterIndex); nil != err {
  1382. util.PushStatusBar("Save data snapshot for cloud sync failed")
  1383. logging.LogErrorf("put index into data repo before cloud sync failed: %s", err)
  1384. return
  1385. }
  1386. util.PushStatusBar(fmt.Sprintf(Conf.Language(147), elapsed.Seconds()))
  1387. } else {
  1388. util.PushStatusBar(fmt.Sprintf(Conf.Language(148), elapsed.Seconds()))
  1389. }
  1390. if Conf.Repo.SyncIndexTiming < elapsed.Milliseconds() {
  1391. logging.LogWarnf("index data repo before cloud sync elapsed [%dms]", elapsed.Milliseconds())
  1392. if !promotedPurgeDataRepo {
  1393. go func() {
  1394. util.WaitForUILoaded()
  1395. time.Sleep(3 * time.Second)
  1396. if indexCount, _ := repo.CountIndexes(); 128 > indexCount {
  1397. // 快照数量较少时不推送提示
  1398. return
  1399. }
  1400. util.PushMsg(Conf.language(218), 24000)
  1401. promotedPurgeDataRepo = true
  1402. }()
  1403. }
  1404. }
  1405. return
  1406. }
  1407. func newRepository() (ret *dejavu.Repo, err error) {
  1408. cloudConf, err := buildCloudConf()
  1409. if nil != err {
  1410. return
  1411. }
  1412. var cloudRepo cloud.Cloud
  1413. switch Conf.Sync.Provider {
  1414. case conf.ProviderSiYuan:
  1415. cloudRepo = cloud.NewSiYuan(&cloud.BaseCloud{Conf: cloudConf})
  1416. case conf.ProviderS3:
  1417. s3HTTPClient := &http.Client{Transport: httpclient.NewTransport(cloudConf.S3.SkipTlsVerify)}
  1418. s3HTTPClient.Timeout = time.Duration(cloudConf.S3.Timeout) * time.Second
  1419. cloudRepo = cloud.NewS3(&cloud.BaseCloud{Conf: cloudConf}, s3HTTPClient)
  1420. case conf.ProviderWebDAV:
  1421. webdavClient := gowebdav.NewClient(cloudConf.WebDAV.Endpoint, cloudConf.WebDAV.Username, cloudConf.WebDAV.Password)
  1422. a := cloudConf.WebDAV.Username + ":" + cloudConf.WebDAV.Password
  1423. auth := "Basic " + base64.StdEncoding.EncodeToString([]byte(a))
  1424. webdavClient.SetHeader("Authorization", auth)
  1425. webdavClient.SetHeader("User-Agent", util.UserAgent)
  1426. webdavClient.SetTimeout(time.Duration(cloudConf.WebDAV.Timeout) * time.Second)
  1427. webdavClient.SetTransport(httpclient.NewTransport(cloudConf.WebDAV.SkipTlsVerify))
  1428. cloudRepo = cloud.NewWebDAV(&cloud.BaseCloud{Conf: cloudConf}, webdavClient)
  1429. default:
  1430. err = fmt.Errorf("unknown cloud provider [%d]", Conf.Sync.Provider)
  1431. return
  1432. }
  1433. ignoreLines := getSyncIgnoreLines()
  1434. ignoreLines = append(ignoreLines, "/.siyuan/conf.json") // 忽略旧版同步配置
  1435. ret, err = dejavu.NewRepo(util.DataDir, util.RepoDir, util.HistoryDir, util.TempDir, Conf.System.ID, Conf.System.Name, Conf.System.OS, Conf.Repo.Key, ignoreLines, cloudRepo)
  1436. if nil != err {
  1437. logging.LogErrorf("init data repo failed: %s", err)
  1438. return
  1439. }
  1440. return
  1441. }
  1442. func init() {
  1443. subscribeRepoEvents()
  1444. }
  1445. func subscribeRepoEvents() {
  1446. eventbus.Subscribe(eventbus.EvtIndexBeforeWalkData, func(context map[string]interface{}, path string) {
  1447. msg := fmt.Sprintf(Conf.Language(158), path)
  1448. util.SetBootDetails(msg)
  1449. util.ContextPushMsg(context, msg)
  1450. })
  1451. indexWalkDataCount := 0
  1452. eventbus.Subscribe(eventbus.EvtIndexWalkData, func(context map[string]interface{}, path string) {
  1453. msg := fmt.Sprintf(Conf.Language(158), filepath.Base(path))
  1454. if 0 == indexWalkDataCount%1024 {
  1455. util.SetBootDetails(msg)
  1456. util.ContextPushMsg(context, msg)
  1457. }
  1458. indexWalkDataCount++
  1459. })
  1460. eventbus.Subscribe(eventbus.EvtIndexBeforeGetLatestFiles, func(context map[string]interface{}, total int) {
  1461. msg := fmt.Sprintf(Conf.Language(159), 0, total)
  1462. util.SetBootDetails(msg)
  1463. util.ContextPushMsg(context, msg)
  1464. })
  1465. eventbus.Subscribe(eventbus.EvtIndexGetLatestFile, func(context map[string]interface{}, count int, total int) {
  1466. msg := fmt.Sprintf(Conf.Language(159), count, total)
  1467. if 0 == count%64 {
  1468. util.SetBootDetails(msg)
  1469. util.ContextPushMsg(context, msg)
  1470. }
  1471. })
  1472. eventbus.Subscribe(eventbus.EvtIndexUpsertFiles, func(context map[string]interface{}, total int) {
  1473. msg := fmt.Sprintf(Conf.Language(160), 0, total)
  1474. util.SetBootDetails(msg)
  1475. util.ContextPushMsg(context, msg)
  1476. })
  1477. eventbus.Subscribe(eventbus.EvtIndexUpsertFile, func(context map[string]interface{}, count int, total int) {
  1478. msg := fmt.Sprintf(Conf.Language(160), count, total)
  1479. if 0 == count%32 {
  1480. util.SetBootDetails(msg)
  1481. util.ContextPushMsg(context, msg)
  1482. }
  1483. })
  1484. eventbus.Subscribe(eventbus.EvtCheckoutBeforeWalkData, func(context map[string]interface{}, path string) {
  1485. msg := fmt.Sprintf(Conf.Language(161), path)
  1486. util.SetBootDetails(msg)
  1487. util.ContextPushMsg(context, msg)
  1488. })
  1489. coWalkDataCount := 0
  1490. eventbus.Subscribe(eventbus.EvtCheckoutWalkData, func(context map[string]interface{}, path string) {
  1491. msg := fmt.Sprintf(Conf.Language(161), filepath.Base(path))
  1492. if 0 == coWalkDataCount%512 {
  1493. util.SetBootDetails(msg)
  1494. util.ContextPushMsg(context, msg)
  1495. }
  1496. coWalkDataCount++
  1497. })
  1498. var bootProgressPart int32
  1499. eventbus.Subscribe(eventbus.EvtCheckoutUpsertFiles, func(context map[string]interface{}, total int) {
  1500. msg := fmt.Sprintf(Conf.Language(162), 0, total)
  1501. util.SetBootDetails(msg)
  1502. bootProgressPart = int32(10 / float64(total))
  1503. util.ContextPushMsg(context, msg)
  1504. })
  1505. coUpsertFileCount := 0
  1506. eventbus.Subscribe(eventbus.EvtCheckoutUpsertFile, func(context map[string]interface{}, count, total int) {
  1507. msg := fmt.Sprintf(Conf.Language(162), count, total)
  1508. util.IncBootProgress(bootProgressPart, msg)
  1509. if 0 == coUpsertFileCount%32 {
  1510. util.ContextPushMsg(context, msg)
  1511. }
  1512. coUpsertFileCount++
  1513. })
  1514. eventbus.Subscribe(eventbus.EvtCheckoutRemoveFiles, func(context map[string]interface{}, total int) {
  1515. msg := fmt.Sprintf(Conf.Language(163), 0, total)
  1516. util.SetBootDetails(msg)
  1517. bootProgressPart = int32(10 / float64(total))
  1518. util.ContextPushMsg(context, msg)
  1519. })
  1520. eventbus.Subscribe(eventbus.EvtCheckoutRemoveFile, func(context map[string]interface{}, count, total int) {
  1521. msg := fmt.Sprintf(Conf.Language(163), count, total)
  1522. util.IncBootProgress(bootProgressPart, msg)
  1523. if 0 == count%64 {
  1524. util.ContextPushMsg(context, msg)
  1525. }
  1526. })
  1527. eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadIndex, func(context map[string]interface{}, id string) {
  1528. msg := fmt.Sprintf(Conf.Language(164), id[:7])
  1529. util.IncBootProgress(1, msg)
  1530. util.ContextPushMsg(context, msg)
  1531. })
  1532. eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadFiles, func(context map[string]interface{}, total int) {
  1533. msg := fmt.Sprintf(Conf.Language(165), 0, total)
  1534. util.SetBootDetails(msg)
  1535. bootProgressPart = int32(10 / float64(total))
  1536. util.ContextPushMsg(context, msg)
  1537. })
  1538. eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadFile, func(context map[string]interface{}, count, total int) {
  1539. msg := fmt.Sprintf(Conf.Language(165), count, total)
  1540. util.IncBootProgress(bootProgressPart, msg)
  1541. if 0 == count%8 {
  1542. util.ContextPushMsg(context, msg)
  1543. }
  1544. })
  1545. eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadChunks, func(context map[string]interface{}, total int) {
  1546. msg := fmt.Sprintf(Conf.Language(166), 0, total)
  1547. util.SetBootDetails(msg)
  1548. bootProgressPart = int32(10 / float64(total))
  1549. util.ContextPushMsg(context, msg)
  1550. })
  1551. eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadChunk, func(context map[string]interface{}, count, total int) {
  1552. msg := fmt.Sprintf(Conf.Language(166), count, total)
  1553. util.IncBootProgress(bootProgressPart, msg)
  1554. if 0 == count%8 {
  1555. util.ContextPushMsg(context, msg)
  1556. }
  1557. })
  1558. eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadRef, func(context map[string]interface{}, ref string) {
  1559. msg := fmt.Sprintf(Conf.Language(167), ref)
  1560. util.IncBootProgress(1, msg)
  1561. util.ContextPushMsg(context, msg)
  1562. })
  1563. eventbus.Subscribe(eventbus.EvtCloudBeforeUploadIndex, func(context map[string]interface{}, id string) {
  1564. msg := fmt.Sprintf(Conf.Language(168), id[:7])
  1565. util.IncBootProgress(1, msg)
  1566. util.ContextPushMsg(context, msg)
  1567. })
  1568. eventbus.Subscribe(eventbus.EvtCloudBeforeUploadFiles, func(context map[string]interface{}, total int) {
  1569. msg := fmt.Sprintf(Conf.Language(169), 0, total)
  1570. util.SetBootDetails(msg)
  1571. util.ContextPushMsg(context, msg)
  1572. })
  1573. eventbus.Subscribe(eventbus.EvtCloudBeforeUploadFile, func(context map[string]interface{}, count, total int) {
  1574. msg := fmt.Sprintf(Conf.Language(169), count, total)
  1575. if 0 == count%8 {
  1576. util.SetBootDetails(msg)
  1577. util.ContextPushMsg(context, msg)
  1578. }
  1579. })
  1580. eventbus.Subscribe(eventbus.EvtCloudBeforeUploadChunks, func(context map[string]interface{}, total int) {
  1581. msg := fmt.Sprintf(Conf.Language(170), 0, total)
  1582. util.SetBootDetails(msg)
  1583. util.ContextPushMsg(context, msg)
  1584. })
  1585. eventbus.Subscribe(eventbus.EvtCloudBeforeUploadChunk, func(context map[string]interface{}, count, total int) {
  1586. msg := fmt.Sprintf(Conf.Language(170), count, total)
  1587. if 0 == count%8 {
  1588. util.SetBootDetails(msg)
  1589. util.ContextPushMsg(context, msg)
  1590. }
  1591. })
  1592. eventbus.Subscribe(eventbus.EvtCloudBeforeUploadRef, func(context map[string]interface{}, ref string) {
  1593. msg := fmt.Sprintf(Conf.Language(171), ref)
  1594. util.SetBootDetails(msg)
  1595. util.ContextPushMsg(context, msg)
  1596. })
  1597. eventbus.Subscribe(eventbus.EvtCloudLock, func(context map[string]interface{}) {
  1598. msg := fmt.Sprintf(Conf.Language(186))
  1599. util.SetBootDetails(msg)
  1600. util.ContextPushMsg(context, msg)
  1601. })
  1602. eventbus.Subscribe(eventbus.EvtCloudUnlock, func(context map[string]interface{}) {
  1603. msg := fmt.Sprintf(Conf.Language(187))
  1604. util.SetBootDetails(msg)
  1605. util.ContextPushMsg(context, msg)
  1606. })
  1607. eventbus.Subscribe(eventbus.EvtCloudBeforeUploadIndexes, func(context map[string]interface{}) {
  1608. msg := fmt.Sprintf(Conf.Language(208))
  1609. util.SetBootDetails(msg)
  1610. util.ContextPushMsg(context, msg)
  1611. })
  1612. eventbus.Subscribe(eventbus.EvtCloudBeforeUploadCheckIndex, func(context map[string]interface{}) {
  1613. msg := fmt.Sprintf(Conf.Language(209))
  1614. util.SetBootDetails(msg)
  1615. util.ContextPushMsg(context, msg)
  1616. })
  1617. eventbus.Subscribe(eventbus.EvtCloudBeforeFixObjects, func(context map[string]interface{}, count, total int) {
  1618. msg := fmt.Sprintf(Conf.Language(210), count, total)
  1619. util.SetBootDetails(msg)
  1620. util.ContextPushMsg(context, msg)
  1621. })
  1622. eventbus.Subscribe(eventbus.EvtCloudAfterFixObjects, func(context map[string]interface{}) {
  1623. msg := fmt.Sprintf(Conf.Language(211))
  1624. util.SetBootDetails(msg)
  1625. util.ContextPushMsg(context, msg)
  1626. })
  1627. eventbus.Subscribe(eventbus.EvtCloudCorrupted, func() {
  1628. util.PushErrMsg(Conf.language(220), 30000)
  1629. })
  1630. eventbus.Subscribe(eventbus.EvtCloudPurgeListObjects, func(context map[string]interface{}) {
  1631. util.ContextPushMsg(context, Conf.language(224))
  1632. })
  1633. eventbus.Subscribe(eventbus.EvtCloudPurgeListIndexes, func(context map[string]interface{}) {
  1634. util.ContextPushMsg(context, Conf.language(225))
  1635. })
  1636. eventbus.Subscribe(eventbus.EvtCloudPurgeListRefs, func(context map[string]interface{}) {
  1637. util.ContextPushMsg(context, Conf.language(226))
  1638. })
  1639. eventbus.Subscribe(eventbus.EvtCloudPurgeDownloadIndexes, func(context map[string]interface{}) {
  1640. util.ContextPushMsg(context, fmt.Sprintf(Conf.language(227)))
  1641. })
  1642. eventbus.Subscribe(eventbus.EvtCloudPurgeDownloadFiles, func(context map[string]interface{}) {
  1643. util.ContextPushMsg(context, Conf.language(228))
  1644. })
  1645. eventbus.Subscribe(eventbus.EvtCloudPurgeRemoveIndexes, func(context map[string]interface{}) {
  1646. util.ContextPushMsg(context, Conf.language(229))
  1647. })
  1648. eventbus.Subscribe(eventbus.EvtCloudPurgeRemoveIndexesV2, func(context map[string]interface{}) {
  1649. util.ContextPushMsg(context, Conf.language(230))
  1650. })
  1651. eventbus.Subscribe(eventbus.EvtCloudPurgeRemoveObjects, func(context map[string]interface{}) {
  1652. util.ContextPushMsg(context, Conf.language(231))
  1653. })
  1654. }
  1655. func buildCloudConf() (ret *cloud.Conf, err error) {
  1656. if !cloud.IsValidCloudDirName(Conf.Sync.CloudName) {
  1657. logging.LogWarnf("invalid cloud repo name, rename it to [main]")
  1658. Conf.Sync.CloudName = "main"
  1659. Conf.Save()
  1660. }
  1661. userId, token, availableSize := "0", "", int64(1024*1024*1024*1024*2)
  1662. if nil != Conf.User && conf.ProviderSiYuan == Conf.Sync.Provider {
  1663. u := Conf.GetUser()
  1664. userId = u.UserId
  1665. token = u.UserToken
  1666. availableSize = u.GetCloudRepoAvailableSize()
  1667. }
  1668. ret = &cloud.Conf{
  1669. Dir: Conf.Sync.CloudName,
  1670. UserID: userId,
  1671. Token: token,
  1672. AvailableSize: availableSize,
  1673. Server: util.GetCloudServer(),
  1674. }
  1675. switch Conf.Sync.Provider {
  1676. case conf.ProviderSiYuan:
  1677. ret.Endpoint = util.GetCloudSyncServer()
  1678. case conf.ProviderS3:
  1679. ret.S3 = &cloud.ConfS3{
  1680. Endpoint: Conf.Sync.S3.Endpoint,
  1681. AccessKey: Conf.Sync.S3.AccessKey,
  1682. SecretKey: Conf.Sync.S3.SecretKey,
  1683. Bucket: Conf.Sync.S3.Bucket,
  1684. Region: Conf.Sync.S3.Region,
  1685. PathStyle: Conf.Sync.S3.PathStyle,
  1686. SkipTlsVerify: Conf.Sync.S3.SkipTlsVerify,
  1687. Timeout: Conf.Sync.S3.Timeout,
  1688. }
  1689. case conf.ProviderWebDAV:
  1690. ret.WebDAV = &cloud.ConfWebDAV{
  1691. Endpoint: Conf.Sync.WebDAV.Endpoint,
  1692. Username: Conf.Sync.WebDAV.Username,
  1693. Password: Conf.Sync.WebDAV.Password,
  1694. SkipTlsVerify: Conf.Sync.WebDAV.SkipTlsVerify,
  1695. Timeout: Conf.Sync.WebDAV.Timeout,
  1696. }
  1697. default:
  1698. err = fmt.Errorf("invalid provider [%d]", Conf.Sync.Provider)
  1699. return
  1700. }
  1701. return
  1702. }
  1703. type Backup struct {
  1704. Size int64 `json:"size"`
  1705. HSize string `json:"hSize"`
  1706. Updated string `json:"updated"`
  1707. SaveDir string `json:"saveDir"` // 本地备份数据存放目录路径
  1708. }
  1709. type Sync struct {
  1710. Size int64 `json:"size"`
  1711. HSize string `json:"hSize"`
  1712. Updated string `json:"updated"`
  1713. CloudName string `json:"cloudName"` // 云端同步数据存放目录名
  1714. SaveDir string `json:"saveDir"` // 本地同步数据存放目录路径
  1715. }
  1716. func GetCloudSpace() (s *Sync, b *Backup, hSize, hAssetSize, hTotalSize, hExchangeSize, hTrafficUploadSize, hTrafficDownloadSize, hTrafficAPIGet, hTrafficAPIPut string, err error) {
  1717. stat, err := getCloudSpace()
  1718. if nil != err {
  1719. err = errors.New(Conf.Language(30) + " " + err.Error())
  1720. return
  1721. }
  1722. syncSize := stat.Sync.Size
  1723. syncUpdated := stat.Sync.Updated
  1724. s = &Sync{
  1725. Size: syncSize,
  1726. HSize: "-",
  1727. Updated: syncUpdated,
  1728. }
  1729. backupSize := stat.Backup.Size
  1730. backupUpdated := stat.Backup.Updated
  1731. b = &Backup{
  1732. Size: backupSize,
  1733. HSize: "-",
  1734. Updated: backupUpdated,
  1735. }
  1736. assetSize := stat.AssetSize
  1737. totalSize := syncSize + backupSize + assetSize
  1738. hAssetSize = "-"
  1739. hSize = "-"
  1740. hTotalSize = "-"
  1741. hExchangeSize = "-"
  1742. hTrafficUploadSize = "-"
  1743. hTrafficDownloadSize = "-"
  1744. hTrafficAPIGet = "-"
  1745. hTrafficAPIPut = "-"
  1746. if conf.ProviderSiYuan == Conf.Sync.Provider {
  1747. s.HSize = humanize.BytesCustomCeil(uint64(syncSize), 2)
  1748. b.HSize = humanize.BytesCustomCeil(uint64(backupSize), 2)
  1749. hAssetSize = humanize.BytesCustomCeil(uint64(assetSize), 2)
  1750. hSize = humanize.BytesCustomCeil(uint64(totalSize), 2)
  1751. u := Conf.GetUser()
  1752. hTotalSize = humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2)
  1753. hExchangeSize = humanize.BytesCustomCeil(uint64(u.UserSiYuanPointExchangeRepoSize), 2)
  1754. hTrafficUploadSize = humanize.BytesCustomCeil(uint64(u.UserTrafficUpload), 2)
  1755. hTrafficDownloadSize = humanize.BytesCustomCeil(uint64(u.UserTrafficDownload), 2)
  1756. hTrafficAPIGet = humanize.SIWithDigits(u.UserTrafficAPIGet, 2, "")
  1757. hTrafficAPIPut = humanize.SIWithDigits(u.UserTrafficAPIPut, 2, "")
  1758. }
  1759. return
  1760. }
  1761. func getCloudSpace() (stat *cloud.Stat, err error) {
  1762. repo, err := newRepository()
  1763. if nil != err {
  1764. return
  1765. }
  1766. stat, err = repo.GetCloudRepoStat()
  1767. if nil != err {
  1768. logging.LogErrorf("get cloud repo stat failed: %s", err)
  1769. return
  1770. }
  1771. return
  1772. }
  1773. func pushReloadPlugin(upsertPluginSet, removePluginNameSet *hashset.Set) {
  1774. upsertPlugins, removePlugins := []string{}, []string{}
  1775. for _, n := range upsertPluginSet.Values() {
  1776. upsertPlugins = append(upsertPlugins, n.(string))
  1777. }
  1778. for _, n := range removePluginNameSet.Values() {
  1779. removePlugins = append(removePlugins, n.(string))
  1780. }
  1781. logging.LogInfof("reload plugins [upserts=%v, removes=%v]", upsertPlugins, removePlugins)
  1782. util.BroadcastByType("main", "reloadPlugin", 0, "", map[string]interface{}{
  1783. "upsertPlugins": upsertPlugins,
  1784. "removePlugins": removePlugins,
  1785. })
  1786. }