Something went wrong. Try again.
A privacy-first, self-hosted, fully open source personal knowledge management software, written in typescript and golang. (PERSONAL FORK)
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750// 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 api
import ( "encoding/hex" "io" "net/http" "os" "path/filepath" "strings" "time"
"github.com/siyuan-note/logging"
"github.com/88250/gulu" "github.com/gin-gonic/gin" "github.com/siyuan-note/siyuan/kernel/conf" "github.com/siyuan-note/siyuan/kernel/model" "github.com/siyuan-note/siyuan/kernel/util")
func importSyncProviderWebDAV(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(200, ret)
form, err := c.MultipartForm() if err != nil { logging.LogErrorf("read upload file failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
files := form.File["file"] if 1 != len(files) { ret.Code = -1 ret.Msg = "invalid upload file" return }
f := files[0] fh, err := f.Open() if err != nil { logging.LogErrorf("read upload file failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
data, err := io.ReadAll(fh) fh.Close() if err != nil { logging.LogErrorf("read upload file failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
importDir := filepath.Join(util.TempDir, "import") if err = os.MkdirAll(importDir, 0755); err != nil { logging.LogErrorf("import WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
tmp := filepath.Join(importDir, f.Filename) if err = os.WriteFile(tmp, data, 0644); err != nil { logging.LogErrorf("import WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
tmpDir := filepath.Join(importDir, "webdav") os.RemoveAll(tmpDir) if strings.HasSuffix(strings.ToLower(tmp), ".zip") { if err = gulu.Zip.Unzip(tmp, tmpDir); err != nil { logging.LogErrorf("import WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return } } else if strings.HasSuffix(strings.ToLower(tmp), ".json") { if err = gulu.File.CopyFile(tmp, filepath.Join(tmpDir, f.Filename)); err != nil { logging.LogErrorf("import WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() } } else { logging.LogErrorf("invalid WebDAV provider package") ret.Code = -1 ret.Msg = "invalid WebDAV provider package" return }
entries, err := os.ReadDir(tmpDir) if err != nil { logging.LogErrorf("import WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
if 1 != len(entries) { logging.LogErrorf("invalid WebDAV provider package") ret.Code = -1 ret.Msg = "invalid WebDAV provider package" return }
tmp = filepath.Join(tmpDir, entries[0].Name()) data, err = os.ReadFile(tmp) if err != nil { logging.LogErrorf("import WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
data = util.AESDecrypt(string(data)) data, _ = hex.DecodeString(string(data)) webdav := &conf.WebDAV{} if err = gulu.JSON.UnmarshalJSON(data, webdav); err != nil { logging.LogErrorf("import WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
err = model.SetSyncProviderWebDAV(webdav) if err != nil { logging.LogErrorf("import WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
ret.Data = map[string]interface{}{ "webdav": model.Conf.Sync.WebDAV, }}
func exportSyncProviderWebDAV(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
name := "siyuan-webdav-" + time.Now().Format("20060102150405") + ".json" tmpDir := filepath.Join(util.TempDir, "export") if err := os.MkdirAll(tmpDir, 0755); err != nil { logging.LogErrorf("export WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
webdav := model.Conf.Sync.WebDAV if nil == webdav { webdav = &conf.WebDAV{} }
data, err := gulu.JSON.MarshalJSON(model.Conf.Sync.WebDAV) if err != nil { logging.LogErrorf("export WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
dataStr := util.AESEncrypt(string(data)) tmp := filepath.Join(tmpDir, name) if err = os.WriteFile(tmp, []byte(dataStr), 0644); err != nil { logging.LogErrorf("export WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
zipFile, err := gulu.Zip.Create(tmp + ".zip") if err != nil { logging.LogErrorf("export WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
if err = zipFile.AddEntry(name, tmp); err != nil { logging.LogErrorf("export WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
if err = zipFile.Close(); err != nil { logging.LogErrorf("export WebDAV provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
zipPath := "/export/" + name + ".zip" ret.Data = map[string]interface{}{ "name": name, "zip": zipPath, }}
func importSyncProviderS3(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(200, ret)
form, err := c.MultipartForm() if err != nil { logging.LogErrorf("read upload file failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
files := form.File["file"] if 1 != len(files) { ret.Code = -1 ret.Msg = "invalid upload file" return }
f := files[0] fh, err := f.Open() if err != nil { logging.LogErrorf("read upload file failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
data, err := io.ReadAll(fh) fh.Close() if err != nil { logging.LogErrorf("read upload file failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
importDir := filepath.Join(util.TempDir, "import") if err = os.MkdirAll(importDir, 0755); err != nil { logging.LogErrorf("import S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
tmp := filepath.Join(importDir, f.Filename) if err = os.WriteFile(tmp, data, 0644); err != nil { logging.LogErrorf("import S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
tmpDir := filepath.Join(importDir, "s3") os.RemoveAll(tmpDir) if strings.HasSuffix(strings.ToLower(tmp), ".zip") { if err = gulu.Zip.Unzip(tmp, tmpDir); err != nil { logging.LogErrorf("import S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return } } else if strings.HasSuffix(strings.ToLower(tmp), ".json") { if err = gulu.File.CopyFile(tmp, filepath.Join(tmpDir, f.Filename)); err != nil { logging.LogErrorf("import S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() } } else { logging.LogErrorf("invalid S3 provider package") ret.Code = -1 ret.Msg = "invalid S3 provider package" return }
entries, err := os.ReadDir(tmpDir) if err != nil { logging.LogErrorf("import S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
if 1 != len(entries) { logging.LogErrorf("invalid S3 provider package") ret.Code = -1 ret.Msg = "invalid S3 provider package" return }
tmp = filepath.Join(tmpDir, entries[0].Name()) data, err = os.ReadFile(tmp) if err != nil { logging.LogErrorf("import S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
data = util.AESDecrypt(string(data)) data, _ = hex.DecodeString(string(data)) s3 := &conf.S3{} if err = gulu.JSON.UnmarshalJSON(data, s3); err != nil { logging.LogErrorf("import S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
err = model.SetSyncProviderS3(s3) if err != nil { logging.LogErrorf("import S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
ret.Data = map[string]interface{}{ "s3": model.Conf.Sync.S3, }}
func exportSyncProviderS3(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
name := "siyuan-s3-" + time.Now().Format("20060102150405") + ".json" tmpDir := filepath.Join(util.TempDir, "export") if err := os.MkdirAll(tmpDir, 0755); err != nil { logging.LogErrorf("export S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
s3 := model.Conf.Sync.S3 if nil == s3 { s3 = &conf.S3{} }
data, err := gulu.JSON.MarshalJSON(model.Conf.Sync.S3) if err != nil { logging.LogErrorf("export S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
dataStr := util.AESEncrypt(string(data)) tmp := filepath.Join(tmpDir, name) if err = os.WriteFile(tmp, []byte(dataStr), 0644); err != nil { logging.LogErrorf("export S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
zipFile, err := gulu.Zip.Create(tmp + ".zip") if err != nil { logging.LogErrorf("export S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
if err = zipFile.AddEntry(name, tmp); err != nil { logging.LogErrorf("export S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
if err = zipFile.Close(); err != nil { logging.LogErrorf("export S3 provider failed: %s", err) ret.Code = -1 ret.Msg = err.Error() return }
zipPath := "/export/" + name + ".zip" ret.Data = map[string]interface{}{ "name": name, "zip": zipPath, }}
func getSyncInfo(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
stat := model.Conf.Sync.Stat if !model.Conf.Sync.Enabled { stat = model.Conf.Language(53) }
ret.Data = map[string]interface{}{ "synced": model.Conf.Sync.Synced, "stat": stat, "kernels": model.GetOnlineKernels(), "kernel": model.KernelID, }}
func getBootSync(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
if !model.IsAdminRoleContext(c) { return }
if model.Conf.Sync.Enabled && 1 == model.BootSyncSucc { ret.Code = 1 ret.Msg = model.Conf.Language(17) return }}
func performSync(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
// Android 端前后台切换时自动触发同步 https://github.com/siyuan-note/siyuan/issues/7122 var mobileSwitch bool if mobileSwitchArg := arg["mobileSwitch"]; nil != mobileSwitchArg { mobileSwitch = mobileSwitchArg.(bool) } if mobileSwitch { if nil == model.Conf.GetUser() || !model.Conf.Sync.Enabled { return } }
if 3 != model.Conf.Sync.Mode { model.SyncData(true) return }
// 云端同步模式支持 `完全手动同步` 模式 https://github.com/siyuan-note/siyuan/issues/7295 uploadArg := arg["upload"] if nil == uploadArg { // 必须传入同步方向,未传的话不执行同步 return }
upload := uploadArg.(bool) if upload { model.SyncDataUpload() } else { model.SyncDataDownload() }}
func performBootSync(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret) model.BootSyncData() ret.Code = model.BootSyncSucc}
func listCloudSyncDir(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
syncDirs, hSize, err := model.ListCloudSyncDir() if err != nil { ret.Code = 1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }
ret.Data = map[string]interface{}{ "syncDirs": syncDirs, "hSize": hSize, "checkedSyncDir": model.Conf.Sync.CloudName, }}
func removeCloudSyncDir(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
name := arg["name"].(string) err := model.RemoveCloudSyncDir(name) if err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }
ret.Data = model.Conf.Sync.CloudName}
func createCloudSyncDir(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
name := arg["name"].(string) err := model.CreateCloudSyncDir(name) if err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }}
func setSyncGenerateConflictDoc(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
enabled := arg["enabled"].(bool) model.SetSyncGenerateConflictDoc(enabled)}
func setSyncEnable(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
enabled := arg["enabled"].(bool) model.SetSyncEnable(enabled)}
func setSyncInterval(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret) arg, ok := util.JsonArg(c, ret) if !ok { return } interval := int(arg["interval"].(float64)) model.SetSyncInterval(interval)}
func setSyncPerception(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
enabled := arg["enabled"].(bool) model.SetSyncPerception(enabled)}
func setSyncMode(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
mode := int(arg["mode"].(float64)) model.SetSyncMode(mode)}
func setSyncProvider(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
provider := int(arg["provider"].(float64)) err := model.SetSyncProvider(provider) if err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }}
func setSyncProviderS3(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
s3Arg := arg["s3"].(interface{}) data, err := gulu.JSON.MarshalJSON(s3Arg) if err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }
s3 := &conf.S3{} if err = gulu.JSON.UnmarshalJSON(data, s3); err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }
err = model.SetSyncProviderS3(s3) if err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }}
func setSyncProviderWebDAV(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
webdavArg := arg["webdav"].(interface{}) data, err := gulu.JSON.MarshalJSON(webdavArg) if err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }
webdav := &conf.WebDAV{} if err = gulu.JSON.UnmarshalJSON(data, webdav); err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }
err = model.SetSyncProviderWebDAV(webdav) if err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }}
func setSyncProviderLocal(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
localArg := arg["local"].(interface{}) data, err := gulu.JSON.MarshalJSON(localArg) if err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }
local := &conf.Local{} if err = gulu.JSON.UnmarshalJSON(data, local); err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }
err = model.SetSyncProviderLocal(local) if err != nil { ret.Code = -1 ret.Msg = err.Error() ret.Data = map[string]interface{}{"closeTimeout": 5000} return }
ret.Data = map[string]interface{}{ "local": local, }}
func setCloudSyncDir(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
name := arg["name"].(string) model.SetCloudSyncDir(name)}