repository.go 50 KB

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