12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036 |
- // SiYuan - Refactor your thinking
- // Copyright (c) 2020-present, b3log.org
- //
- // This program is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Affero General Public License as published by
- // the Free Software Foundation, either version 3 of the License, or
- // (at your option) any later version.
- //
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Affero General Public License for more details.
- //
- // You should have received a copy of the GNU Affero General Public License
- // along with this program. If not, see <https://www.gnu.org/licenses/>.
- package model
- import (
- "bytes"
- "crypto/rand"
- "crypto/sha256"
- "encoding/base64"
- "errors"
- "fmt"
- "math"
- "net/http"
- "os"
- "path"
- "path/filepath"
- "sort"
- "strings"
- "sync"
- "sync/atomic"
- "time"
- "github.com/88250/go-humanize"
- "github.com/88250/gulu"
- "github.com/88250/lute"
- "github.com/88250/lute/ast"
- "github.com/88250/lute/html"
- "github.com/88250/lute/parse"
- "github.com/88250/lute/render"
- "github.com/emirpasic/gods/sets/hashset"
- "github.com/siyuan-note/dejavu"
- "github.com/siyuan-note/dejavu/cloud"
- "github.com/siyuan-note/dejavu/entity"
- "github.com/siyuan-note/encryption"
- "github.com/siyuan-note/eventbus"
- "github.com/siyuan-note/httpclient"
- "github.com/siyuan-note/logging"
- "github.com/siyuan-note/siyuan/kernel/conf"
- "github.com/siyuan-note/siyuan/kernel/filesys"
- "github.com/siyuan-note/siyuan/kernel/sql"
- "github.com/siyuan-note/siyuan/kernel/task"
- "github.com/siyuan-note/siyuan/kernel/treenode"
- "github.com/siyuan-note/siyuan/kernel/util"
- "github.com/studio-b12/gowebdav"
- )
- func GetRepoFile(fileID string) (ret []byte, p string, err error) {
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- file, err := repo.GetFile(fileID)
- if nil != err {
- return
- }
- ret, err = repo.OpenFile(file)
- p = file.Path
- return
- }
- func OpenRepoSnapshotDoc(fileID string) (content string, isProtyleDoc bool, updated int64, err error) {
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- file, err := repo.GetFile(fileID)
- if nil != err {
- return
- }
- data, err := repo.OpenFile(file)
- if nil != err {
- return
- }
- updated = file.Updated
- if strings.HasSuffix(file.Path, ".sy") {
- luteEngine := NewLute()
- var snapshotTree *parse.Tree
- isProtyleDoc, snapshotTree, err = parseTreeInSnapshot(data, luteEngine)
- if nil != err {
- logging.LogErrorf("parse tree from snapshot file [%s] failed", fileID)
- return
- }
- if !isProtyleDoc {
- renderTree := &parse.Tree{Root: &ast.Node{Type: ast.NodeDocument}}
- var unlinks []*ast.Node
- ast.Walk(snapshotTree.Root, func(n *ast.Node, entering bool) ast.WalkStatus {
- if !entering {
- return ast.WalkContinue
- }
- n.RemoveIALAttr("heading-fold")
- n.RemoveIALAttr("fold")
- return ast.WalkContinue
- })
- for _, unlink := range unlinks {
- unlink.Unlink()
- }
- var appends []*ast.Node
- for n := snapshotTree.Root.FirstChild; nil != n; n = n.Next {
- appends = append(appends, n)
- }
- for _, n := range appends {
- renderTree.Root.AppendChild(n)
- }
- snapshotTree = renderTree
- }
- luteEngine.RenderOptions.ProtyleContenteditable = false
- if isProtyleDoc {
- util.PushMsg(Conf.Language(36), 5000)
- formatRenderer := render.NewFormatRenderer(snapshotTree, luteEngine.RenderOptions)
- content = gulu.Str.FromBytes(formatRenderer.Render())
- } else {
- content = luteEngine.Tree2BlockDOM(snapshotTree, luteEngine.RenderOptions)
- }
- } else {
- isProtyleDoc = true
- if strings.HasSuffix(file.Path, ".json") {
- content = gulu.Str.FromBytes(data)
- } else {
- if strings.Contains(file.Path, "assets/") { // 剔除笔记本级或者文档级资源文件路径前缀
- file.Path = file.Path[strings.Index(file.Path, "assets/"):]
- if util.IsDisplayableAsset(file.Path) {
- dir, f := filepath.Split(file.Path)
- tempRepoDiffDir := filepath.Join(util.TempDir, "repo", "diff", dir)
- if mkErr := os.MkdirAll(tempRepoDiffDir, 0755); nil != mkErr {
- logging.LogErrorf("mkdir [%s] failed: %v", tempRepoDiffDir, mkErr)
- } else {
- if wrErr := os.WriteFile(filepath.Join(tempRepoDiffDir, f), data, 0644); nil != wrErr {
- logging.LogErrorf("write file [%s] failed: %v", filepath.Join(tempRepoDiffDir, file.Path), wrErr)
- }
- }
- content = path.Join("repo", "diff", file.Path)
- }
- } else {
- content = file.Path
- }
- }
- }
- return
- }
- type LeftRightDiff struct {
- LeftIndex *DiffIndex `json:"leftIndex"`
- RightIndex *DiffIndex `json:"rightIndex"`
- AddsLeft []*DiffFile `json:"addsLeft"`
- UpdatesLeft []*DiffFile `json:"updatesLeft"`
- UpdatesRight []*DiffFile `json:"updatesRight"`
- RemovesRight []*DiffFile `json:"removesRight"`
- }
- type DiffFile struct {
- FileID string `json:"fileID"`
- Title string `json:"title"`
- Path string `json:"path"`
- HSize string `json:"hSize"`
- Updated int64 `json:"updated"`
- }
- type DiffIndex struct {
- ID string `json:"id"`
- Created int64 `json:"created"`
- }
- func DiffRepoSnapshots(left, right string) (ret *LeftRightDiff, err error) {
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- diff, err := repo.DiffIndex(left, right)
- if nil != err {
- return
- }
- ret = &LeftRightDiff{
- LeftIndex: &DiffIndex{
- ID: diff.LeftIndex.ID,
- Created: diff.LeftIndex.Created,
- },
- RightIndex: &DiffIndex{
- ID: diff.RightIndex.ID,
- Created: diff.RightIndex.Created,
- },
- }
- luteEngine := NewLute()
- for _, removeRight := range diff.RemovesRight {
- title, parseErr := parseTitleInSnapshot(removeRight.ID, repo, luteEngine)
- if "" == title || nil != parseErr {
- continue
- }
- ret.AddsLeft = append(ret.AddsLeft, &DiffFile{
- FileID: removeRight.ID,
- Title: title,
- Path: removeRight.Path,
- HSize: humanize.BytesCustomCeil(uint64(removeRight.Size), 2),
- Updated: removeRight.Updated,
- })
- }
- if 1 > len(ret.AddsLeft) {
- ret.AddsLeft = []*DiffFile{}
- }
- for _, addLeft := range diff.AddsLeft {
- title, parseErr := parseTitleInSnapshot(addLeft.ID, repo, luteEngine)
- if "" == title || nil != parseErr {
- continue
- }
- ret.RemovesRight = append(ret.RemovesRight, &DiffFile{
- FileID: addLeft.ID,
- Title: title,
- Path: addLeft.Path,
- HSize: humanize.BytesCustomCeil(uint64(addLeft.Size), 2),
- Updated: addLeft.Updated,
- })
- }
- if 1 > len(ret.RemovesRight) {
- ret.RemovesRight = []*DiffFile{}
- }
- for _, updateLeft := range diff.UpdatesLeft {
- title, parseErr := parseTitleInSnapshot(updateLeft.ID, repo, luteEngine)
- if "" == title || nil != parseErr {
- continue
- }
- ret.UpdatesLeft = append(ret.UpdatesLeft, &DiffFile{
- FileID: updateLeft.ID,
- Title: title,
- Path: updateLeft.Path,
- HSize: humanize.BytesCustomCeil(uint64(updateLeft.Size), 2),
- Updated: updateLeft.Updated,
- })
- }
- if 1 > len(ret.UpdatesLeft) {
- ret.UpdatesLeft = []*DiffFile{}
- }
- for _, updateRight := range diff.UpdatesRight {
- title, parseErr := parseTitleInSnapshot(updateRight.ID, repo, luteEngine)
- if "" == title || nil != parseErr {
- continue
- }
- ret.UpdatesRight = append(ret.UpdatesRight, &DiffFile{
- FileID: updateRight.ID,
- Title: title,
- Path: updateRight.Path,
- HSize: humanize.BytesCustomCeil(uint64(updateRight.Size), 2),
- Updated: updateRight.Updated,
- })
- }
- if 1 > len(ret.UpdatesRight) {
- ret.UpdatesRight = []*DiffFile{}
- }
- return
- }
- func parseTitleInSnapshot(fileID string, repo *dejavu.Repo, luteEngine *lute.Lute) (title string, err error) {
- file, err := repo.GetFile(fileID)
- if nil != err {
- logging.LogErrorf("get file [%s] failed: %s", fileID, err)
- return
- }
- title = path.Base(file.Path)
- if strings.HasSuffix(file.Path, ".sy") {
- var data []byte
- data, err = repo.OpenFile(file)
- if nil != err {
- logging.LogErrorf("open file [%s] failed: %s", fileID, err)
- return
- }
- var tree *parse.Tree
- tree, err = filesys.ParseJSONWithoutFix(data, luteEngine.ParseOptions)
- if nil != err {
- logging.LogErrorf("parse file [%s] failed: %s", fileID, err)
- return
- }
- title = tree.Root.IALAttr("title")
- }
- return
- }
- func parseTreeInSnapshot(data []byte, luteEngine *lute.Lute) (isProtyleDoc bool, tree *parse.Tree, err error) {
- isProtyleDoc = 1024*1024*1 <= len(data)
- tree, err = filesys.ParseJSONWithoutFix(data, luteEngine.ParseOptions)
- if nil != err {
- return
- }
- return
- }
- type Snapshot struct {
- *dejavu.Log
- TypesCount []*TypeCount `json:"typesCount"`
- }
- type TypeCount struct {
- Type string `json:"type"`
- Count int `json:"count"`
- }
- func GetRepoSnapshots(page int) (ret []*Snapshot, pageCount, totalCount int, err error) {
- ret = []*Snapshot{}
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- logs, pageCount, totalCount, err := repo.GetIndexLogs(page, 32)
- if nil != err {
- if dejavu.ErrNotFoundIndex == err {
- logs = []*dejavu.Log{}
- err = nil
- return
- }
- logging.LogErrorf("get data repo index logs failed: %s", err)
- return
- }
- ret = buildSnapshots(logs)
- if 1 > len(ret) {
- ret = []*Snapshot{}
- }
- return
- }
- func buildSnapshots(logs []*dejavu.Log) (ret []*Snapshot) {
- for _, l := range logs {
- typesCount := statTypesByPath(l.Files)
- l.Files = nil // 置空,否则返回前端数据量太大
- ret = append(ret, &Snapshot{
- Log: l,
- TypesCount: typesCount,
- })
- }
- return
- }
- func statTypesByPath(files []*entity.File) (ret []*TypeCount) {
- for _, f := range files {
- ext := path.Ext(f.Path)
- if "" == ext {
- ext = "NoExt"
- }
- found := false
- for _, tc := range ret {
- if tc.Type == ext {
- tc.Count++
- found = true
- break
- }
- }
- if !found {
- ret = append(ret, &TypeCount{Type: ext, Count: 1})
- }
- }
- sort.Slice(ret, func(i, j int) bool { return ret[i].Count > ret[j].Count })
- if 10 < len(ret) {
- otherCount := 0
- for _, tc := range ret[10:] {
- tc.Count += otherCount
- }
- other := &TypeCount{
- Type: "Other",
- Count: otherCount,
- }
- ret = append(ret[:10], other)
- }
- return
- }
- func ImportRepoKey(base64Key string) (err error) {
- util.PushMsg(Conf.Language(136), 3000)
- base64Key = strings.TrimSpace(base64Key)
- base64Key = gulu.Str.RemoveInvisible(base64Key)
- if 1 > len(base64Key) {
- err = errors.New(Conf.Language(142))
- return
- }
- key, err := base64.StdEncoding.DecodeString(base64Key)
- if nil != err {
- logging.LogErrorf("import data repo key failed: %s", err)
- return errors.New(Conf.Language(157))
- }
- if 32 != len(key) {
- return errors.New(Conf.Language(157))
- }
- Conf.Repo.Key = key
- Conf.Save()
- if err = os.RemoveAll(Conf.Repo.GetSaveDir()); nil != err {
- return
- }
- if err = os.MkdirAll(Conf.Repo.GetSaveDir(), 0755); nil != err {
- return
- }
- initDataRepo()
- return
- }
- func ResetRepo() (err error) {
- logging.LogInfof("resetting data repo...")
- msgId := util.PushMsg(Conf.Language(144), 1000*60)
- repo, err := newRepository()
- if nil != err {
- return
- }
- if err = repo.Reset(); nil != err {
- logging.LogErrorf("reset data repo failed: %s", err)
- return
- }
- logging.LogInfof("reset data repo completed")
- Conf.Repo.Key = nil
- Conf.Sync.Enabled = false
- Conf.Save()
- util.PushUpdateMsg(msgId, Conf.Language(145), 3000)
- go func() {
- time.Sleep(2 * time.Second)
- util.ReloadUI()
- }()
- return
- }
- func PurgeCloud() (err error) {
- msg := Conf.Language(223)
- util.PushEndlessProgress(msg)
- defer util.PushClearProgress()
- repo, err := newRepository()
- if nil != err {
- return
- }
- stat, err := repo.PurgeCloud()
- if nil != err {
- return
- }
- deletedIndexes := stat.Indexes
- deletedObjects := stat.Objects
- deletedSize := humanize.BytesCustomCeil(uint64(stat.Size), 2)
- msg = fmt.Sprintf(Conf.Language(232), deletedIndexes, deletedObjects, deletedSize)
- util.PushMsg(msg, 5000)
- return
- }
- func PurgeRepo() (err error) {
- msg := Conf.Language(202)
- util.PushEndlessProgress(msg)
- defer util.PushClearProgress()
- repo, err := newRepository()
- if nil != err {
- return
- }
- stat, err := repo.Purge()
- if nil != err {
- return
- }
- deletedIndexes := stat.Indexes
- deletedObjects := stat.Objects
- deletedSize := humanize.BytesCustomCeil(uint64(stat.Size), 2)
- msg = fmt.Sprintf(Conf.Language(203), deletedIndexes, deletedObjects, deletedSize)
- util.PushMsg(msg, 5000)
- return
- }
- func InitRepoKeyFromPassphrase(passphrase string) (err error) {
- passphrase = gulu.Str.RemoveInvisible(passphrase)
- passphrase = strings.TrimSpace(passphrase)
- if "" == passphrase {
- return errors.New(Conf.Language(142))
- }
- util.PushMsg(Conf.Language(136), 3000)
- if err = os.RemoveAll(Conf.Repo.GetSaveDir()); nil != err {
- return
- }
- if err = os.MkdirAll(Conf.Repo.GetSaveDir(), 0755); nil != err {
- return
- }
- var key []byte
- base64Data, base64Err := base64.StdEncoding.DecodeString(passphrase)
- if nil == base64Err && 32 == len(base64Data) {
- // 改进数据仓库 `通过密码生成密钥` https://github.com/siyuan-note/siyuan/issues/6782
- logging.LogInfof("passphrase is base64 encoded, use it as key directly")
- key = base64Data
- } else {
- salt := fmt.Sprintf("%x", sha256.Sum256([]byte(passphrase)))[:16]
- key, err = encryption.KDF(passphrase, salt)
- if nil != err {
- logging.LogErrorf("init data repo key failed: %s", err)
- return
- }
- }
- Conf.Repo.Key = key
- Conf.Save()
- initDataRepo()
- return
- }
- func InitRepoKey() (err error) {
- util.PushMsg(Conf.Language(136), 3000)
- if err = os.RemoveAll(Conf.Repo.GetSaveDir()); nil != err {
- return
- }
- if err = os.MkdirAll(Conf.Repo.GetSaveDir(), 0755); nil != err {
- return
- }
- randomBytes := make([]byte, 16)
- _, err = rand.Read(randomBytes)
- if nil != err {
- return
- }
- password := string(randomBytes)
- randomBytes = make([]byte, 16)
- _, err = rand.Read(randomBytes)
- if nil != err {
- logging.LogErrorf("init data repo key failed: %s", err)
- return
- }
- salt := string(randomBytes)
- key, err := encryption.KDF(password, salt)
- if nil != err {
- logging.LogErrorf("init data repo key failed: %s", err)
- return
- }
- Conf.Repo.Key = key
- Conf.Save()
- initDataRepo()
- return
- }
- func initDataRepo() {
- time.Sleep(1 * time.Second)
- util.PushMsg(Conf.Language(138), 3000)
- time.Sleep(1 * time.Second)
- if initErr := IndexRepo("[Init] Init local data repo"); nil != initErr {
- util.PushErrMsg(fmt.Sprintf(Conf.Language(140), initErr), 0)
- }
- }
- func CheckoutRepo(id string) {
- task.AppendTask(task.RepoCheckout, checkoutRepo, id)
- }
- func checkoutRepo(id string) {
- var err error
- if 1 > len(Conf.Repo.Key) {
- util.PushErrMsg(Conf.Language(26), 7000)
- return
- }
- repo, err := newRepository()
- if nil != err {
- logging.LogErrorf("new repository failed: %s", err)
- util.PushErrMsg(Conf.Language(141), 7000)
- return
- }
- util.PushEndlessProgress(Conf.Language(63))
- WaitForWritingFiles()
- CloseWatchAssets()
- defer WatchAssets()
- CloseWatchEmojis()
- defer WatchEmojis()
- // 恢复快照时自动暂停同步,避免刚刚恢复后的数据又被同步覆盖
- syncEnabled := Conf.Sync.Enabled
- Conf.Sync.Enabled = false
- Conf.Save()
- _, _, err = repo.Checkout(id, map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress})
- if nil != err {
- logging.LogErrorf("checkout repository failed: %s", err)
- util.PushClearProgress()
- util.PushErrMsg(Conf.Language(141), 7000)
- return
- }
- task.AppendTask(task.DatabaseIndexFull, fullReindex)
- task.AppendTask(task.DatabaseIndexRef, IndexRefs)
- go func() {
- sql.WaitForWritingDatabase()
- ResetVirtualBlockRefCache()
- }()
- task.AppendTask(task.ReloadUI, util.ReloadUIResetScroll)
- if syncEnabled {
- func() {
- time.Sleep(5 * time.Second)
- util.PushMsg(Conf.Language(134), 0)
- }()
- }
- return
- }
- func DownloadCloudSnapshot(tag, id string) (err error) {
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- switch Conf.Sync.Provider {
- case conf.ProviderSiYuan:
- if !IsSubscriber() {
- util.PushErrMsg(Conf.Language(29), 5000)
- return
- }
- case conf.ProviderWebDAV, conf.ProviderS3:
- if !IsPaidUser() {
- util.PushErrMsg(Conf.Language(214), 5000)
- return
- }
- }
- defer util.PushClearProgress()
- var downloadFileCount, downloadChunkCount int
- var downloadBytes int64
- if "" == tag {
- downloadFileCount, downloadChunkCount, downloadBytes, err = repo.DownloadIndex(id, map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress})
- } else {
- downloadFileCount, downloadChunkCount, downloadBytes, err = repo.DownloadTagIndex(tag, id, map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress})
- }
- if nil != err {
- return
- }
- msg := fmt.Sprintf(Conf.Language(153), downloadFileCount, downloadChunkCount, humanize.BytesCustomCeil(uint64(downloadBytes), 2))
- util.PushMsg(msg, 5000)
- util.PushStatusBar(msg)
- return
- }
- func UploadCloudSnapshot(tag, id string) (err error) {
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- switch Conf.Sync.Provider {
- case conf.ProviderSiYuan:
- if !IsSubscriber() {
- util.PushErrMsg(Conf.Language(29), 5000)
- return
- }
- case conf.ProviderWebDAV, conf.ProviderS3:
- if !IsPaidUser() {
- util.PushErrMsg(Conf.Language(214), 5000)
- return
- }
- }
- util.PushEndlessProgress(Conf.Language(116))
- defer util.PushClearProgress()
- uploadFileCount, uploadChunkCount, uploadBytes, err := repo.UploadTagIndex(tag, id, map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress})
- if nil != err {
- if errors.Is(err, dejavu.ErrCloudBackupCountExceeded) {
- err = fmt.Errorf(Conf.Language(84), Conf.Language(154))
- return
- }
- err = errors.New(fmt.Sprintf(Conf.Language(84), formatRepoErrorMsg(err)))
- return
- }
- msg := fmt.Sprintf(Conf.Language(152), uploadFileCount, uploadChunkCount, humanize.BytesCustomCeil(uint64(uploadBytes), 2))
- util.PushMsg(msg, 5000)
- util.PushStatusBar(msg)
- return
- }
- func RemoveCloudRepoTag(tag string) (err error) {
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- if "" == tag {
- err = errors.New("tag is empty")
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- switch Conf.Sync.Provider {
- case conf.ProviderSiYuan:
- if !IsSubscriber() {
- util.PushErrMsg(Conf.Language(29), 5000)
- return
- }
- case conf.ProviderWebDAV, conf.ProviderS3:
- if !IsPaidUser() {
- util.PushErrMsg(Conf.Language(214), 5000)
- return
- }
- }
- err = repo.RemoveCloudRepoTag(tag)
- if nil != err {
- return
- }
- return
- }
- func GetCloudRepoTagSnapshots() (ret []*dejavu.Log, err error) {
- ret = []*dejavu.Log{}
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- switch Conf.Sync.Provider {
- case conf.ProviderSiYuan:
- if !IsSubscriber() {
- util.PushErrMsg(Conf.Language(29), 5000)
- return
- }
- case conf.ProviderWebDAV, conf.ProviderS3:
- if !IsPaidUser() {
- util.PushErrMsg(Conf.Language(214), 5000)
- return
- }
- }
- logs, err := repo.GetCloudRepoTagLogs(map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar})
- if nil != err {
- return
- }
- ret = logs
- if 1 > len(ret) {
- ret = []*dejavu.Log{}
- }
- return
- }
- func GetCloudRepoSnapshots(page int) (ret []*dejavu.Log, pageCount, totalCount int, err error) {
- ret = []*dejavu.Log{}
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- switch Conf.Sync.Provider {
- case conf.ProviderSiYuan:
- if !IsSubscriber() {
- util.PushErrMsg(Conf.Language(29), 5000)
- return
- }
- case conf.ProviderWebDAV, conf.ProviderS3:
- if !IsPaidUser() {
- util.PushErrMsg(Conf.Language(214), 5000)
- return
- }
- }
- if 1 > page {
- page = 1
- }
- logs, pageCount, totalCount, err := repo.GetCloudRepoLogs(page)
- if nil != err {
- return
- }
- ret = logs
- if 1 > len(ret) {
- ret = []*dejavu.Log{}
- }
- return
- }
- func GetTagSnapshots() (ret []*Snapshot, err error) {
- ret = []*Snapshot{}
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- logs, err := repo.GetTagLogs()
- if nil != err {
- return
- }
- ret = buildSnapshots(logs)
- if 1 > len(ret) {
- ret = []*Snapshot{}
- }
- return
- }
- func RemoveTagSnapshot(tag string) (err error) {
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- err = repo.RemoveTag(tag)
- return
- }
- func TagSnapshot(id, name string) (err error) {
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- name = strings.TrimSpace(name)
- name = gulu.Str.RemoveInvisible(name)
- if "" == name {
- err = errors.New(Conf.Language(142))
- return
- }
- if !gulu.File.IsValidFilename(name) {
- err = errors.New(Conf.Language(151))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- index, err := repo.GetIndex(id)
- if nil != err {
- return
- }
- if err = repo.AddTag(index.ID, name); nil != err {
- msg := fmt.Sprintf("Add tag to data snapshot [%s] failed: %s", index.ID, err)
- util.PushStatusBar(msg)
- return
- }
- return
- }
- func IndexRepo(memo string) (err error) {
- if 1 > len(Conf.Repo.Key) {
- err = errors.New(Conf.Language(26))
- return
- }
- memo = strings.TrimSpace(memo)
- memo = gulu.Str.RemoveInvisible(memo)
- if "" == memo {
- err = errors.New(Conf.Language(142))
- return
- }
- repo, err := newRepository()
- if nil != err {
- return
- }
- util.PushEndlessProgress(Conf.Language(143))
- start := time.Now()
- latest, _ := repo.Latest()
- WaitForWritingFiles()
- index, err := repo.Index(memo, map[string]interface{}{
- eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBarAndProgress,
- })
- if nil != err {
- util.PushStatusBar("Index data repo failed: " + html.EscapeString(err.Error()))
- return
- }
- elapsed := time.Since(start)
- if nil == latest || latest.ID != index.ID {
- msg := fmt.Sprintf(Conf.Language(147), elapsed.Seconds())
- util.PushStatusBar(msg)
- util.PushMsg(msg, 5000)
- } else {
- msg := fmt.Sprintf(Conf.Language(148), elapsed.Seconds())
- util.PushStatusBar(msg)
- util.PushMsg(msg, 5000)
- }
- util.PushClearProgress()
- return
- }
- var syncingFiles = sync.Map{}
- var syncingStorages = atomic.Bool{}
- func waitForSyncingStorages() {
- for isSyncingStorages() {
- time.Sleep(time.Second)
- }
- }
- func isSyncingStorages() bool {
- return syncingStorages.Load() || isBootSyncing.Load()
- }
- func IsSyncingFile(rootID string) (ret bool) {
- _, ret = syncingFiles.Load(rootID)
- return
- }
- func syncRepoDownload() (err error) {
- if 1 > len(Conf.Repo.Key) {
- planSyncAfter(fixSyncInterval)
- msg := Conf.Language(26)
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- err = errors.New(msg)
- return
- }
- repo, err := newRepository()
- if nil != err {
- planSyncAfter(fixSyncInterval)
- msg := fmt.Sprintf("sync repo failed: %s", err)
- logging.LogErrorf(msg)
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- return
- }
- logging.LogInfof("downloading data repo [device=%s, kernel=%s, provider=%d, mode=%s/%t]", Conf.System.ID, KernelID, Conf.Sync.Provider, "d", true)
- start := time.Now()
- _, _, err = indexRepoBeforeCloudSync(repo)
- if nil != err {
- planSyncAfter(fixSyncInterval)
- logging.LogErrorf("sync data repo download failed: %s", err)
- msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
- Conf.Sync.Stat = msg
- Conf.Save()
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- return
- }
- syncContext := map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar}
- mergeResult, trafficStat, err := repo.SyncDownload(syncContext)
- elapsed := time.Since(start)
- if nil != err {
- planSyncAfter(fixSyncInterval)
- logging.LogErrorf("sync data repo download failed: %s", err)
- msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
- if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
- u := Conf.GetUser()
- msg = fmt.Sprintf(Conf.Language(43), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
- if 2 == u.UserSiYuanSubscriptionPlan {
- msg = fmt.Sprintf(Conf.Language(68), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
- }
- }
- Conf.Sync.Stat = msg
- Conf.Save()
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- return
- }
- util.PushStatusBar(fmt.Sprintf(Conf.Language(149), elapsed.Seconds()))
- Conf.Sync.Synced = util.CurrentTimeMillis()
- 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))
- Conf.Sync.Stat = msg
- Conf.Save()
- autoSyncErrCount = 0
- BootSyncSucc = 0
- processSyncMergeResult(false, true, mergeResult, trafficStat, "d", elapsed)
- return
- }
- func syncRepoUpload() (err error) {
- if 1 > len(Conf.Repo.Key) {
- planSyncAfter(fixSyncInterval)
- msg := Conf.Language(26)
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- err = errors.New(msg)
- return
- }
- repo, err := newRepository()
- if nil != err {
- planSyncAfter(fixSyncInterval)
- msg := fmt.Sprintf("sync repo failed: %s", err)
- logging.LogErrorf(msg)
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- return
- }
- logging.LogInfof("uploading data repo [device=%s, kernel=%s, provider=%d, mode=%s/%t]", Conf.System.ID, KernelID, Conf.Sync.Provider, "u", true)
- start := time.Now()
- _, _, err = indexRepoBeforeCloudSync(repo)
- if nil != err {
- planSyncAfter(fixSyncInterval)
- logging.LogErrorf("sync data repo upload failed: %s", err)
- msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
- Conf.Sync.Stat = msg
- Conf.Save()
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- return
- }
- syncContext := map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar}
- trafficStat, err := repo.SyncUpload(syncContext)
- elapsed := time.Since(start)
- if nil != err {
- planSyncAfter(fixSyncInterval)
- logging.LogErrorf("sync data repo upload failed: %s", err)
- msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
- if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
- u := Conf.GetUser()
- msg = fmt.Sprintf(Conf.Language(43), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
- if 2 == u.UserSiYuanSubscriptionPlan {
- msg = fmt.Sprintf(Conf.Language(68), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
- }
- }
- Conf.Sync.Stat = msg
- Conf.Save()
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- return
- }
- util.PushStatusBar(fmt.Sprintf(Conf.Language(149), elapsed.Seconds()))
- Conf.Sync.Synced = util.CurrentTimeMillis()
- 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))
- Conf.Sync.Stat = msg
- Conf.Save()
- autoSyncErrCount = 0
- BootSyncSucc = 0
- processSyncMergeResult(false, true, &dejavu.MergeResult{}, trafficStat, "u", elapsed)
- return
- }
- var isBootSyncing = atomic.Bool{}
- func bootSyncRepo() (err error) {
- if 1 > len(Conf.Repo.Key) {
- autoSyncErrCount++
- planSyncAfter(fixSyncInterval)
- msg := Conf.Language(26)
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- err = errors.New(msg)
- return
- }
- repo, err := newRepository()
- if nil != err {
- autoSyncErrCount++
- planSyncAfter(fixSyncInterval)
- msg := fmt.Sprintf("sync repo failed: %s", html.EscapeString(err.Error()))
- logging.LogErrorf(msg)
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- return
- }
- isBootSyncing.Store(true)
- start := time.Now()
- _, _, err = indexRepoBeforeCloudSync(repo)
- if nil != err {
- autoSyncErrCount++
- planSyncAfter(fixSyncInterval)
- msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
- Conf.Sync.Stat = msg
- Conf.Save()
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- BootSyncSucc = 1
- isBootSyncing.Store(false)
- return
- }
- syncContext := map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar}
- fetchedFiles, err := repo.GetSyncCloudFiles(syncContext)
- if errors.Is(err, dejavu.ErrRepoFatal) {
- autoSyncErrCount++
- planSyncAfter(fixSyncInterval)
- msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
- Conf.Sync.Stat = msg
- Conf.Save()
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- BootSyncSucc = 1
- isBootSyncing.Store(false)
- return
- }
- syncingFiles = sync.Map{}
- syncingStorages.Store(false)
- for _, fetchedFile := range fetchedFiles {
- name := path.Base(fetchedFile.Path)
- if strings.HasSuffix(name, ".sy") {
- id := name[:len(name)-3]
- syncingFiles.Store(id, true)
- continue
- }
- if strings.HasPrefix(fetchedFile.Path, "/storage/") {
- syncingStorages.Store(true)
- }
- }
- elapsed := time.Since(start)
- logging.LogInfof("boot get sync cloud files elapsed [%.2fs]", elapsed.Seconds())
- if nil != err {
- autoSyncErrCount++
- planSyncAfter(fixSyncInterval)
- logging.LogErrorf("sync data repo failed: %s", err)
- msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
- if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
- u := Conf.GetUser()
- msg = fmt.Sprintf(Conf.Language(43), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
- if 2 == u.UserSiYuanSubscriptionPlan {
- msg = fmt.Sprintf(Conf.Language(68), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
- }
- }
- Conf.Sync.Stat = msg
- Conf.Save()
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- BootSyncSucc = 1
- isBootSyncing.Store(false)
- return
- }
- if 0 < len(fetchedFiles) {
- go func() {
- _, syncErr := syncRepo(false, false)
- isBootSyncing.Store(false)
- if nil != err {
- logging.LogErrorf("boot background sync repo failed: %s", syncErr)
- return
- }
- }()
- } else {
- isBootSyncing.Store(false)
- }
- return
- }
- func syncRepo(exit, byHand bool) (dataChanged bool, err error) {
- if 1 > len(Conf.Repo.Key) {
- autoSyncErrCount++
- planSyncAfter(fixSyncInterval)
- msg := Conf.Language(26)
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- err = errors.New(msg)
- return
- }
- repo, err := newRepository()
- if nil != err {
- autoSyncErrCount++
- planSyncAfter(fixSyncInterval)
- msg := fmt.Sprintf("sync repo failed: %s", err)
- logging.LogErrorf(msg)
- util.PushStatusBar(msg)
- util.PushErrMsg(msg, 0)
- return
- }
- logging.LogInfof("syncing data repo [device=%s, kernel=%s, provider=%d, mode=%s/%t]", Conf.System.ID, KernelID, Conf.Sync.Provider, "a", byHand)
- start := time.Now()
- beforeIndex, afterIndex, err := indexRepoBeforeCloudSync(repo)
- if nil != err {
- autoSyncErrCount++
- planSyncAfter(fixSyncInterval)
- logging.LogErrorf("sync data repo failed: %s", err)
- msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
- Conf.Sync.Stat = msg
- Conf.Save()
- util.PushStatusBar(msg)
- if 1 > autoSyncErrCount || byHand {
- util.PushErrMsg(msg, 0)
- }
- if exit {
- ExitSyncSucc = 1
- }
- return
- }
- syncContext := map[string]interface{}{eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar}
- mergeResult, trafficStat, err := repo.Sync(syncContext)
- elapsed := time.Since(start)
- if nil != err {
- autoSyncErrCount++
- planSyncAfter(fixSyncInterval)
- logging.LogErrorf("sync data repo failed: %s", err)
- msg := fmt.Sprintf(Conf.Language(80), formatRepoErrorMsg(err))
- if errors.Is(err, dejavu.ErrCloudStorageSizeExceeded) {
- u := Conf.GetUser()
- msg = fmt.Sprintf(Conf.Language(43), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
- if 2 == u.UserSiYuanSubscriptionPlan {
- msg = fmt.Sprintf(Conf.Language(68), humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2))
- }
- }
- Conf.Sync.Stat = msg
- Conf.Save()
- util.PushStatusBar(msg)
- if 1 > autoSyncErrCount || byHand {
- util.PushErrMsg(msg, 0)
- }
- if exit {
- ExitSyncSucc = 1
- }
- return
- }
- dataChanged = nil == beforeIndex || beforeIndex.ID != afterIndex.ID || mergeResult.DataChanged()
- util.PushStatusBar(fmt.Sprintf(Conf.Language(149), elapsed.Seconds()))
- Conf.Sync.Synced = util.CurrentTimeMillis()
- 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))
- Conf.Sync.Stat = msg
- Conf.Save()
- autoSyncErrCount = 0
- processSyncMergeResult(exit, byHand, mergeResult, trafficStat, "a", elapsed)
- if !exit {
- // 首次数据同步执行完成后再执行索引订正 Index fixing should not be performed before data synchronization https://github.com/siyuan-note/siyuan/issues/10761
- go checkIndex()
- }
- return
- }
- func processSyncMergeResult(exit, byHand bool, mergeResult *dejavu.MergeResult, trafficStat *dejavu.TrafficStat, mode string, elapsed time.Duration) {
- 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",
- Conf.System.ID, KernelID, Conf.Sync.Provider, mode, byHand,
- trafficStat.UploadFileCount, trafficStat.DownloadFileCount, trafficStat.UploadChunkCount, trafficStat.DownloadChunkCount, humanize.BytesCustomCeil(uint64(trafficStat.UploadBytes), 2), humanize.BytesCustomCeil(uint64(trafficStat.DownloadBytes), 2),
- elapsed.Seconds(),
- len(mergeResult.Conflicts), len(mergeResult.Upserts), len(mergeResult.Removes))
- //logSyncMergeResult(mergeResult)
- var needReloadFiletree bool
- if 0 < len(mergeResult.Conflicts) {
- luteEngine := util.NewLute()
- if Conf.Sync.GenerateConflictDoc {
- // 云端同步发生冲突时生成副本 https://github.com/siyuan-note/siyuan/issues/5687
- for _, file := range mergeResult.Conflicts {
- if !strings.HasSuffix(file.Path, ".sy") {
- continue
- }
- parts := strings.Split(file.Path[1:], "/")
- if 2 > len(parts) {
- continue
- }
- boxID := parts[0]
- absPath := filepath.Join(util.TempDir, "repo", "sync", "conflicts", mergeResult.Time.Format("2006-01-02-150405"), file.Path)
- tree, loadTreeErr := loadTree(absPath, luteEngine)
- if nil != loadTreeErr {
- logging.LogErrorf("load conflicted file [%s] failed: %s", absPath, loadTreeErr)
- continue
- }
- tree.Box = boxID
- tree.Path = strings.TrimPrefix(file.Path, "/"+boxID)
- resetTree(tree, "Conflicted")
- createTreeTx(tree)
- }
- needReloadFiletree = true
- }
- historyDir := filepath.Join(util.HistoryDir, mergeResult.Time.Format("2006-01-02-150405")+"-sync")
- indexHistoryDir(filepath.Base(historyDir), luteEngine)
- }
- if 1 > len(mergeResult.Upserts) && 1 > len(mergeResult.Removes) && 1 > len(mergeResult.Conflicts) { // 没有数据变更
- syncSameCount.Add(1)
- if 10 < syncSameCount.Load() {
- syncSameCount.Store(5)
- }
- if !byHand {
- delay := time.Minute * time.Duration(int(math.Pow(2, float64(syncSameCount.Load()))))
- if fixSyncInterval.Minutes() > delay.Minutes() {
- delay = time.Minute * 8
- }
- planSyncAfter(delay)
- }
- util.PushClearProgress()
- return
- }
- // 有数据变更,需要重建索引
- var upserts, removes []string
- var upsertTrees int
- // 可能需要重新加载部分功能
- var needReloadFlashcard, needReloadOcrTexts, needReloadPlugin bool
- upsertPluginSet := hashset.New()
- for _, file := range mergeResult.Upserts {
- upserts = append(upserts, file.Path)
- if strings.HasPrefix(file.Path, "/storage/riff/") {
- needReloadFlashcard = true
- }
- if strings.HasPrefix(file.Path, "/assets/ocr-texts.json") {
- needReloadOcrTexts = true
- }
- if strings.HasSuffix(file.Path, "/.siyuan/conf.json") {
- needReloadFiletree = true
- }
- if strings.HasPrefix(file.Path, "/storage/petal/") {
- needReloadPlugin = true
- }
- if strings.HasPrefix(file.Path, "/plugins/") {
- if parts := strings.Split(file.Path, "/"); 2 < len(parts) {
- needReloadPlugin = true
- upsertPluginSet.Add(parts[2])
- }
- }
- if strings.HasSuffix(file.Path, ".sy") {
- upsertTrees++
- }
- }
- removeWidgetDirSet, removePluginSet := hashset.New(), hashset.New()
- for _, file := range mergeResult.Removes {
- removes = append(removes, file.Path)
- if strings.HasPrefix(file.Path, "/storage/riff/") {
- needReloadFlashcard = true
- }
- if strings.HasPrefix(file.Path, "/assets/ocr-texts.json") {
- needReloadOcrTexts = true
- }
- if strings.HasSuffix(file.Path, "/.siyuan/conf.json") {
- needReloadFiletree = true
- }
- if strings.HasPrefix(file.Path, "/storage/petal/") {
- needReloadPlugin = true
- }
- if strings.HasPrefix(file.Path, "/plugins/") {
- if parts := strings.Split(file.Path, "/"); 2 < len(parts) {
- needReloadPlugin = true
- removePluginSet.Add(parts[2])
- }
- }
- if strings.HasPrefix(file.Path, "/widgets/") {
- if parts := strings.Split(file.Path, "/"); 2 < len(parts) {
- removeWidgetDirSet.Add(parts[2])
- }
- }
- }
- if needReloadFlashcard {
- LoadFlashcards()
- }
- if needReloadOcrTexts {
- util.LoadAssetsTexts()
- }
- if needReloadPlugin {
- pushReloadPlugin(upsertPluginSet, removePluginSet)
- }
- for _, widgetDir := range removeWidgetDirSet.Values() {
- widgetDirPath := filepath.Join(util.DataDir, "widgets", widgetDir.(string))
- gulu.File.RemoveEmptyDirs(widgetDirPath)
- }
- syncingFiles = sync.Map{}
- syncingStorages.Store(false)
- if needFullReindex(upsertTrees) { // 改进同步后全量重建索引判断 https://github.com/siyuan-note/siyuan/issues/5764
- FullReindex()
- return
- }
- if needReloadFiletree {
- util.PushReloadFiletree()
- }
- if exit { // 退出时同步不用推送事件
- return
- }
- upsertRootIDs, removeRootIDs := incReindex(upserts, removes)
- go func() {
- util.WaitForUILoaded()
- if 0 < len(upsertRootIDs) || 0 < len(removeRootIDs) {
- util.BroadcastByType("main", "syncMergeResult", 0, "",
- map[string]interface{}{"upsertRootIDs": upsertRootIDs, "removeRootIDs": removeRootIDs})
- }
- time.Sleep(2 * time.Second)
- util.PushStatusBar(fmt.Sprintf(Conf.Language(149), elapsed.Seconds()))
- if 0 < len(mergeResult.Conflicts) {
- syConflict := false
- for _, file := range mergeResult.Conflicts {
- if strings.HasSuffix(file.Path, ".sy") {
- syConflict = true
- break
- }
- }
- if syConflict {
- // 数据同步发生冲突时在界面上进行提醒 https://github.com/siyuan-note/siyuan/issues/7332
- util.PushMsg(Conf.Language(108), 7000)
- }
- }
- }()
- }
- func logSyncMergeResult(mergeResult *dejavu.MergeResult) {
- if 1 > len(mergeResult.Conflicts) && 1 > len(mergeResult.Upserts) && 1 > len(mergeResult.Removes) {
- return
- }
- if 0 < len(mergeResult.Conflicts) {
- logBuilder := bytes.Buffer{}
- for i, f := range mergeResult.Conflicts {
- logBuilder.WriteString(" ")
- logBuilder.WriteString(f.Path)
- if i < len(mergeResult.Conflicts)-1 {
- logBuilder.WriteString("\n")
- }
- }
- logging.LogInfof("sync conflicts:\n%s", logBuilder.String())
- }
- if 0 < len(mergeResult.Upserts) {
- logBuilder := bytes.Buffer{}
- for i, f := range mergeResult.Upserts {
- logBuilder.WriteString(" ")
- logBuilder.WriteString(f.Path)
- if i < len(mergeResult.Upserts)-1 {
- logBuilder.WriteString("\n")
- }
- }
- logging.LogInfof("sync merge upserts:\n%s", logBuilder.String())
- }
- if 0 < len(mergeResult.Removes) {
- logBuilder := bytes.Buffer{}
- for i, f := range mergeResult.Removes {
- logBuilder.WriteString(" ")
- logBuilder.WriteString(f.Path)
- if i < len(mergeResult.Removes)-1 {
- logBuilder.WriteString("\n")
- }
- }
- logging.LogInfof("sync merge removes:\n%s", logBuilder.String())
- }
- }
- func needFullReindex(upsertTrees int) bool {
- return 0.2 < float64(upsertTrees)/float64(treenode.CountTrees())
- }
- var promotedPurgeDataRepo bool
- func indexRepoBeforeCloudSync(repo *dejavu.Repo) (beforeIndex, afterIndex *entity.Index, err error) {
- start := time.Now()
- beforeIndex, _ = repo.Latest()
- afterIndex, err = repo.Index("[Sync] Cloud sync", map[string]interface{}{
- eventbus.CtxPushMsg: eventbus.CtxPushMsgToStatusBar,
- })
- if nil != err {
- logging.LogErrorf("index data repo before cloud sync failed: %s", err)
- return
- }
- elapsed := time.Since(start)
- if nil == beforeIndex || beforeIndex.ID != afterIndex.ID {
- // 对新创建的快照需要更新备注,加入耗时统计
- afterIndex.Memo = fmt.Sprintf("[Sync] Cloud sync, completed in %.2fs", elapsed.Seconds())
- if err = repo.PutIndex(afterIndex); nil != err {
- util.PushStatusBar("Save data snapshot for cloud sync failed")
- logging.LogErrorf("put index into data repo before cloud sync failed: %s", err)
- return
- }
- util.PushStatusBar(fmt.Sprintf(Conf.Language(147), elapsed.Seconds()))
- } else {
- util.PushStatusBar(fmt.Sprintf(Conf.Language(148), elapsed.Seconds()))
- }
- if Conf.Repo.SyncIndexTiming < elapsed.Milliseconds() {
- logging.LogWarnf("index data repo before cloud sync elapsed [%dms]", elapsed.Milliseconds())
- if !promotedPurgeDataRepo {
- go func() {
- util.WaitForUILoaded()
- time.Sleep(3 * time.Second)
- if indexCount, _ := repo.CountIndexes(); 128 > indexCount {
- // 快照数量较少时不推送提示
- return
- }
- util.PushMsg(Conf.language(218), 24000)
- promotedPurgeDataRepo = true
- }()
- }
- }
- return
- }
- func newRepository() (ret *dejavu.Repo, err error) {
- cloudConf, err := buildCloudConf()
- if nil != err {
- return
- }
- var cloudRepo cloud.Cloud
- switch Conf.Sync.Provider {
- case conf.ProviderSiYuan:
- cloudRepo = cloud.NewSiYuan(&cloud.BaseCloud{Conf: cloudConf})
- case conf.ProviderS3:
- s3HTTPClient := &http.Client{Transport: httpclient.NewTransport(cloudConf.S3.SkipTlsVerify)}
- s3HTTPClient.Timeout = time.Duration(cloudConf.S3.Timeout) * time.Second
- cloudRepo = cloud.NewS3(&cloud.BaseCloud{Conf: cloudConf}, s3HTTPClient)
- case conf.ProviderWebDAV:
- webdavClient := gowebdav.NewClient(cloudConf.WebDAV.Endpoint, cloudConf.WebDAV.Username, cloudConf.WebDAV.Password)
- a := cloudConf.WebDAV.Username + ":" + cloudConf.WebDAV.Password
- auth := "Basic " + base64.StdEncoding.EncodeToString([]byte(a))
- webdavClient.SetHeader("Authorization", auth)
- webdavClient.SetHeader("User-Agent", util.UserAgent)
- webdavClient.SetTimeout(time.Duration(cloudConf.WebDAV.Timeout) * time.Second)
- webdavClient.SetTransport(httpclient.NewTransport(cloudConf.WebDAV.SkipTlsVerify))
- cloudRepo = cloud.NewWebDAV(&cloud.BaseCloud{Conf: cloudConf}, webdavClient)
- default:
- err = fmt.Errorf("unknown cloud provider [%d]", Conf.Sync.Provider)
- return
- }
- ignoreLines := getSyncIgnoreLines()
- ignoreLines = append(ignoreLines, "/.siyuan/conf.json") // 忽略旧版同步配置
- 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)
- if nil != err {
- logging.LogErrorf("init data repo failed: %s", err)
- return
- }
- return
- }
- func init() {
- subscribeRepoEvents()
- }
- func subscribeRepoEvents() {
- eventbus.Subscribe(eventbus.EvtIndexBeforeWalkData, func(context map[string]interface{}, path string) {
- msg := fmt.Sprintf(Conf.Language(158), path)
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- indexWalkDataCount := 0
- eventbus.Subscribe(eventbus.EvtIndexWalkData, func(context map[string]interface{}, path string) {
- msg := fmt.Sprintf(Conf.Language(158), filepath.Base(path))
- if 0 == indexWalkDataCount%1024 {
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- }
- indexWalkDataCount++
- })
- eventbus.Subscribe(eventbus.EvtIndexBeforeGetLatestFiles, func(context map[string]interface{}, total int) {
- msg := fmt.Sprintf(Conf.Language(159), 0, total)
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtIndexGetLatestFile, func(context map[string]interface{}, count int, total int) {
- msg := fmt.Sprintf(Conf.Language(159), count, total)
- if 0 == count%64 {
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- }
- })
- eventbus.Subscribe(eventbus.EvtIndexUpsertFiles, func(context map[string]interface{}, total int) {
- msg := fmt.Sprintf(Conf.Language(160), 0, total)
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtIndexUpsertFile, func(context map[string]interface{}, count int, total int) {
- msg := fmt.Sprintf(Conf.Language(160), count, total)
- if 0 == count%32 {
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- }
- })
- eventbus.Subscribe(eventbus.EvtCheckoutBeforeWalkData, func(context map[string]interface{}, path string) {
- msg := fmt.Sprintf(Conf.Language(161), path)
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- coWalkDataCount := 0
- eventbus.Subscribe(eventbus.EvtCheckoutWalkData, func(context map[string]interface{}, path string) {
- msg := fmt.Sprintf(Conf.Language(161), filepath.Base(path))
- if 0 == coWalkDataCount%512 {
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- }
- coWalkDataCount++
- })
- var bootProgressPart int32
- eventbus.Subscribe(eventbus.EvtCheckoutUpsertFiles, func(context map[string]interface{}, total int) {
- msg := fmt.Sprintf(Conf.Language(162), 0, total)
- util.SetBootDetails(msg)
- bootProgressPart = int32(10 / float64(total))
- util.ContextPushMsg(context, msg)
- })
- coUpsertFileCount := 0
- eventbus.Subscribe(eventbus.EvtCheckoutUpsertFile, func(context map[string]interface{}, count, total int) {
- msg := fmt.Sprintf(Conf.Language(162), count, total)
- util.IncBootProgress(bootProgressPart, msg)
- if 0 == coUpsertFileCount%32 {
- util.ContextPushMsg(context, msg)
- }
- coUpsertFileCount++
- })
- eventbus.Subscribe(eventbus.EvtCheckoutRemoveFiles, func(context map[string]interface{}, total int) {
- msg := fmt.Sprintf(Conf.Language(163), 0, total)
- util.SetBootDetails(msg)
- bootProgressPart = int32(10 / float64(total))
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCheckoutRemoveFile, func(context map[string]interface{}, count, total int) {
- msg := fmt.Sprintf(Conf.Language(163), count, total)
- util.IncBootProgress(bootProgressPart, msg)
- if 0 == count%64 {
- util.ContextPushMsg(context, msg)
- }
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadIndex, func(context map[string]interface{}, id string) {
- msg := fmt.Sprintf(Conf.Language(164), id[:7])
- util.IncBootProgress(1, msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadFiles, func(context map[string]interface{}, total int) {
- msg := fmt.Sprintf(Conf.Language(165), 0, total)
- util.SetBootDetails(msg)
- bootProgressPart = int32(10 / float64(total))
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadFile, func(context map[string]interface{}, count, total int) {
- msg := fmt.Sprintf(Conf.Language(165), count, total)
- util.IncBootProgress(bootProgressPart, msg)
- if 0 == count%8 {
- util.ContextPushMsg(context, msg)
- }
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadChunks, func(context map[string]interface{}, total int) {
- msg := fmt.Sprintf(Conf.Language(166), 0, total)
- util.SetBootDetails(msg)
- bootProgressPart = int32(10 / float64(total))
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadChunk, func(context map[string]interface{}, count, total int) {
- msg := fmt.Sprintf(Conf.Language(166), count, total)
- util.IncBootProgress(bootProgressPart, msg)
- if 0 == count%8 {
- util.ContextPushMsg(context, msg)
- }
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeDownloadRef, func(context map[string]interface{}, ref string) {
- msg := fmt.Sprintf(Conf.Language(167), ref)
- util.IncBootProgress(1, msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeUploadIndex, func(context map[string]interface{}, id string) {
- msg := fmt.Sprintf(Conf.Language(168), id[:7])
- util.IncBootProgress(1, msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeUploadFiles, func(context map[string]interface{}, total int) {
- msg := fmt.Sprintf(Conf.Language(169), 0, total)
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeUploadFile, func(context map[string]interface{}, count, total int) {
- msg := fmt.Sprintf(Conf.Language(169), count, total)
- if 0 == count%8 {
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- }
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeUploadChunks, func(context map[string]interface{}, total int) {
- msg := fmt.Sprintf(Conf.Language(170), 0, total)
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeUploadChunk, func(context map[string]interface{}, count, total int) {
- msg := fmt.Sprintf(Conf.Language(170), count, total)
- if 0 == count%8 {
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- }
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeUploadRef, func(context map[string]interface{}, ref string) {
- msg := fmt.Sprintf(Conf.Language(171), ref)
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudLock, func(context map[string]interface{}) {
- msg := fmt.Sprintf(Conf.Language(186))
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudUnlock, func(context map[string]interface{}) {
- msg := fmt.Sprintf(Conf.Language(187))
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeUploadIndexes, func(context map[string]interface{}) {
- msg := fmt.Sprintf(Conf.Language(208))
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeUploadCheckIndex, func(context map[string]interface{}) {
- msg := fmt.Sprintf(Conf.Language(209))
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudBeforeFixObjects, func(context map[string]interface{}, count, total int) {
- msg := fmt.Sprintf(Conf.Language(210), count, total)
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudAfterFixObjects, func(context map[string]interface{}) {
- msg := fmt.Sprintf(Conf.Language(211))
- util.SetBootDetails(msg)
- util.ContextPushMsg(context, msg)
- })
- eventbus.Subscribe(eventbus.EvtCloudCorrupted, func() {
- util.PushErrMsg(Conf.language(220), 30000)
- })
- eventbus.Subscribe(eventbus.EvtCloudPurgeListObjects, func(context map[string]interface{}) {
- util.ContextPushMsg(context, Conf.language(224))
- })
- eventbus.Subscribe(eventbus.EvtCloudPurgeListIndexes, func(context map[string]interface{}) {
- util.ContextPushMsg(context, Conf.language(225))
- })
- eventbus.Subscribe(eventbus.EvtCloudPurgeListRefs, func(context map[string]interface{}) {
- util.ContextPushMsg(context, Conf.language(226))
- })
- eventbus.Subscribe(eventbus.EvtCloudPurgeDownloadIndexes, func(context map[string]interface{}) {
- util.ContextPushMsg(context, fmt.Sprintf(Conf.language(227)))
- })
- eventbus.Subscribe(eventbus.EvtCloudPurgeDownloadFiles, func(context map[string]interface{}) {
- util.ContextPushMsg(context, Conf.language(228))
- })
- eventbus.Subscribe(eventbus.EvtCloudPurgeRemoveIndexes, func(context map[string]interface{}) {
- util.ContextPushMsg(context, Conf.language(229))
- })
- eventbus.Subscribe(eventbus.EvtCloudPurgeRemoveIndexesV2, func(context map[string]interface{}) {
- util.ContextPushMsg(context, Conf.language(230))
- })
- eventbus.Subscribe(eventbus.EvtCloudPurgeRemoveObjects, func(context map[string]interface{}) {
- util.ContextPushMsg(context, Conf.language(231))
- })
- }
- func buildCloudConf() (ret *cloud.Conf, err error) {
- if !cloud.IsValidCloudDirName(Conf.Sync.CloudName) {
- logging.LogWarnf("invalid cloud repo name, rename it to [main]")
- Conf.Sync.CloudName = "main"
- Conf.Save()
- }
- userId, token, availableSize := "0", "", int64(1024*1024*1024*1024*2)
- if nil != Conf.User && conf.ProviderSiYuan == Conf.Sync.Provider {
- u := Conf.GetUser()
- userId = u.UserId
- token = u.UserToken
- availableSize = u.GetCloudRepoAvailableSize()
- }
- ret = &cloud.Conf{
- Dir: Conf.Sync.CloudName,
- UserID: userId,
- Token: token,
- AvailableSize: availableSize,
- Server: util.GetCloudServer(),
- }
- switch Conf.Sync.Provider {
- case conf.ProviderSiYuan:
- ret.Endpoint = util.GetCloudSyncServer()
- case conf.ProviderS3:
- ret.S3 = &cloud.ConfS3{
- Endpoint: Conf.Sync.S3.Endpoint,
- AccessKey: Conf.Sync.S3.AccessKey,
- SecretKey: Conf.Sync.S3.SecretKey,
- Bucket: Conf.Sync.S3.Bucket,
- Region: Conf.Sync.S3.Region,
- PathStyle: Conf.Sync.S3.PathStyle,
- SkipTlsVerify: Conf.Sync.S3.SkipTlsVerify,
- Timeout: Conf.Sync.S3.Timeout,
- }
- case conf.ProviderWebDAV:
- ret.WebDAV = &cloud.ConfWebDAV{
- Endpoint: Conf.Sync.WebDAV.Endpoint,
- Username: Conf.Sync.WebDAV.Username,
- Password: Conf.Sync.WebDAV.Password,
- SkipTlsVerify: Conf.Sync.WebDAV.SkipTlsVerify,
- Timeout: Conf.Sync.WebDAV.Timeout,
- }
- default:
- err = fmt.Errorf("invalid provider [%d]", Conf.Sync.Provider)
- return
- }
- return
- }
- type Backup struct {
- Size int64 `json:"size"`
- HSize string `json:"hSize"`
- Updated string `json:"updated"`
- SaveDir string `json:"saveDir"` // 本地备份数据存放目录路径
- }
- type Sync struct {
- Size int64 `json:"size"`
- HSize string `json:"hSize"`
- Updated string `json:"updated"`
- CloudName string `json:"cloudName"` // 云端同步数据存放目录名
- SaveDir string `json:"saveDir"` // 本地同步数据存放目录路径
- }
- func GetCloudSpace() (s *Sync, b *Backup, hSize, hAssetSize, hTotalSize, hExchangeSize, hTrafficUploadSize, hTrafficDownloadSize, hTrafficAPIGet, hTrafficAPIPut string, err error) {
- stat, err := getCloudSpace()
- if nil != err {
- err = errors.New(Conf.Language(30) + " " + err.Error())
- return
- }
- syncSize := stat.Sync.Size
- syncUpdated := stat.Sync.Updated
- s = &Sync{
- Size: syncSize,
- HSize: "-",
- Updated: syncUpdated,
- }
- backupSize := stat.Backup.Size
- backupUpdated := stat.Backup.Updated
- b = &Backup{
- Size: backupSize,
- HSize: "-",
- Updated: backupUpdated,
- }
- assetSize := stat.AssetSize
- totalSize := syncSize + backupSize + assetSize
- hAssetSize = "-"
- hSize = "-"
- hTotalSize = "-"
- hExchangeSize = "-"
- hTrafficUploadSize = "-"
- hTrafficDownloadSize = "-"
- hTrafficAPIGet = "-"
- hTrafficAPIPut = "-"
- if conf.ProviderSiYuan == Conf.Sync.Provider {
- s.HSize = humanize.BytesCustomCeil(uint64(syncSize), 2)
- b.HSize = humanize.BytesCustomCeil(uint64(backupSize), 2)
- hAssetSize = humanize.BytesCustomCeil(uint64(assetSize), 2)
- hSize = humanize.BytesCustomCeil(uint64(totalSize), 2)
- u := Conf.GetUser()
- hTotalSize = humanize.BytesCustomCeil(uint64(u.UserSiYuanRepoSize), 2)
- hExchangeSize = humanize.BytesCustomCeil(uint64(u.UserSiYuanPointExchangeRepoSize), 2)
- hTrafficUploadSize = humanize.BytesCustomCeil(uint64(u.UserTrafficUpload), 2)
- hTrafficDownloadSize = humanize.BytesCustomCeil(uint64(u.UserTrafficDownload), 2)
- hTrafficAPIGet = humanize.SIWithDigits(u.UserTrafficAPIGet, 2, "")
- hTrafficAPIPut = humanize.SIWithDigits(u.UserTrafficAPIPut, 2, "")
- }
- return
- }
- func getCloudSpace() (stat *cloud.Stat, err error) {
- repo, err := newRepository()
- if nil != err {
- return
- }
- stat, err = repo.GetCloudRepoStat()
- if nil != err {
- logging.LogErrorf("get cloud repo stat failed: %s", err)
- return
- }
- return
- }
- func pushReloadPlugin(upsertPluginSet, removePluginNameSet *hashset.Set) {
- upsertPlugins, removePlugins := []string{}, []string{}
- for _, n := range upsertPluginSet.Values() {
- upsertPlugins = append(upsertPlugins, n.(string))
- }
- for _, n := range removePluginNameSet.Values() {
- removePlugins = append(removePlugins, n.(string))
- }
- logging.LogInfof("reload plugins [upserts=%v, removes=%v]", upsertPlugins, removePlugins)
- util.BroadcastByType("main", "reloadPlugin", 0, "", map[string]interface{}{
- "upsertPlugins": upsertPlugins,
- "removePlugins": removePlugins,
- })
- }
|