siyuan/kernel/api/repo.go
Jeffrey Chen 698ee3d357
♻️ Improve validation of some JSON parameters (#17412)
* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters

* 🎨 Improve validation of some JSON parameters
2026-04-05 17:03:13 +08:00

556 lines
12 KiB
Go

// 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 (
"fmt"
"mime"
"net/http"
"path/filepath"
"github.com/88250/gulu"
"github.com/gabriel-vasile/mimetype"
"github.com/gin-gonic/gin"
"github.com/siyuan-note/siyuan/kernel/model"
"github.com/siyuan-note/siyuan/kernel/util"
)
func setRepoIndexRetentionDays(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var days float64
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("days", &days, true, false)) {
return
}
daysInt := int(days)
if 1 > daysInt {
daysInt = 180
}
model.Conf.Repo.IndexRetentionDays = daysInt
model.Conf.Save()
}
func setRetentionIndexesDaily(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var indexes float64
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("indexes", &indexes, true, false)) {
return
}
indexesInt := int(indexes)
if 1 > indexesInt {
indexesInt = 180
}
model.Conf.Repo.RetentionIndexesDaily = indexesInt
model.Conf.Save()
}
func getRepoFile(c *gin.Context) {
// Add internal kernel API `/api/repo/getRepoFile` https://github.com/siyuan-note/siyuan/issues/10101
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var id string
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("id", &id, true, true)) {
return
}
data, p, err := model.GetRepoFile(id)
if err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
contentType := mime.TypeByExtension(filepath.Ext(p))
if "" == contentType {
if m := mimetype.Detect(data); nil != m {
contentType = m.String()
}
}
if "" == contentType {
contentType = "application/octet-stream"
}
c.Data(http.StatusOK, contentType, data)
}
func rollbackRepoSnapshotFile(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var id string
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("id", &id, true, true)) {
return
}
err := model.RollbackRepoSnapshotFile(id)
if nil != err {
ret.Code = -1
ret.Msg = err.Error()
return
}
}
func openRepoSnapshotFile(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var id string
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("id", &id, true, true)) {
return
}
title, content, displayInText, updated, err := model.OpenRepoSnapshotFile(id)
if err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
ret.Data = map[string]any{
"title": title,
"content": content,
"displayInText": displayInText,
"updated": updated,
}
}
func diffRepoSnapshots(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var left, right string
if !util.ParseJsonArgs(arg, ret,
util.BindJsonArg("left", &left, true, true),
util.BindJsonArg("right", &right, true, true),
) {
return
}
diff, err := model.DiffRepoSnapshots(left, right)
if err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
ret.Data = map[string]any{
"addsLeft": diff.AddsLeft,
"updatesLeft": diff.UpdatesLeft,
"updatesRight": diff.UpdatesRight,
"removesRight": diff.RemovesRight,
"left": diff.LeftIndex,
"right": diff.RightIndex,
}
}
func getCloudSpace(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
sync, backup, hSize, hAssetSize, hTotalSize, exchangeSize, hTrafficUploadSize, hTrafficDownloadSize, htrafficAPIGet, hTrafficAPIPut, err := model.GetCloudSpace()
if err != nil {
ret.Code = 1
ret.Msg = err.Error()
util.PushErrMsg(err.Error(), 3000)
return
}
ret.Data = map[string]any{
"sync": sync,
"backup": backup,
"hAssetSize": hAssetSize,
"hSize": hSize,
"hTotalSize": hTotalSize,
"hExchangeSize": exchangeSize,
"hTrafficUploadSize": hTrafficUploadSize,
"hTrafficDownloadSize": hTrafficDownloadSize,
"hTrafficAPIGet": htrafficAPIGet,
"hTrafficAPIPut": hTrafficAPIPut,
}
}
func checkoutRepo(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var id string
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("id", &id, true, true)) {
return
}
model.CheckoutRepo(id)
}
func downloadCloudSnapshot(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var id, tag string
if !util.ParseJsonArgs(arg, ret,
util.BindJsonArg("id", &id, true, true),
util.BindJsonArg("tag", &tag, true, false),
) {
return
}
if err := model.DownloadCloudSnapshot(tag, id); err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
}
func uploadCloudSnapshot(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var id, tag string
if !util.ParseJsonArgs(arg, ret,
util.BindJsonArg("id", &id, true, true),
util.BindJsonArg("tag", &tag, true, false),
) {
return
}
if err := model.UploadCloudSnapshot(tag, id); err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
}
func getRepoSnapshots(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var page float64
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("page", &page, true, false)) {
return
}
snapshots, pageCount, totalCount, err := model.GetRepoSnapshots(int(page))
if err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
ret.Data = map[string]any{
"snapshots": snapshots,
"pageCount": pageCount,
"totalCount": totalCount,
}
}
func getCloudRepoSnapshots(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var page float64
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("page", &page, true, false)) {
return
}
snapshots, pageCount, totalCount, err := model.GetCloudRepoSnapshots(int(page))
if err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
ret.Data = map[string]any{
"snapshots": snapshots,
"pageCount": pageCount,
"totalCount": totalCount,
}
}
func getCloudRepoTagSnapshots(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
snapshots, err := model.GetCloudRepoTagSnapshots()
if err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
ret.Data = map[string]any{
"snapshots": snapshots,
}
}
func removeCloudRepoTagSnapshot(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var tag string
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("tag", &tag, true, true)) {
return
}
err := model.RemoveCloudRepoTag(tag)
if err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
}
func getRepoTagSnapshots(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
snapshots, err := model.GetTagSnapshots()
if err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
ret.Data = map[string]any{
"snapshots": snapshots,
}
}
func removeRepoTagSnapshot(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var tag string
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("tag", &tag, true, true)) {
return
}
err := model.RemoveTagSnapshot(tag)
if err != nil {
ret.Code = -1
ret.Msg = err.Error()
return
}
}
func createSnapshot(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var memo string
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("memo", &memo, true, false)) {
return
}
if err := model.IndexRepo(memo); err != nil {
ret.Code = -1
ret.Msg = fmt.Sprintf(model.Conf.Language(140), err)
ret.Data = map[string]any{"closeTimeout": 5000}
return
}
}
func tagSnapshot(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var id, name string
if !util.ParseJsonArgs(arg, ret,
util.BindJsonArg("id", &id, true, true),
util.BindJsonArg("name", &name, true, false),
) {
return
}
if err := model.TagSnapshot(id, name); err != nil {
ret.Code = -1
ret.Msg = fmt.Sprintf(model.Conf.Language(140), err)
ret.Data = map[string]any{"closeTimeout": 5000}
return
}
}
func importRepoKey(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var base64Key string
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("key", &base64Key, true, false)) {
return
}
retKey, err := model.ImportRepoKey(base64Key)
if err != nil {
ret.Code = -1
ret.Msg = fmt.Sprintf(model.Conf.Language(137), err)
ret.Data = map[string]any{"closeTimeout": 5000}
return
}
ret.Data = map[string]any{
"key": retKey,
}
}
func initRepoKeyFromPassphrase(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret)
if !ok {
return
}
var pass string
if !util.ParseJsonArgs(arg, ret, util.BindJsonArg("pass", &pass, true, false)) {
return
}
if err := model.InitRepoKeyFromPassphrase(pass); err != nil {
ret.Code = -1
ret.Msg = fmt.Sprintf(model.Conf.Language(137), err)
ret.Data = map[string]any{"closeTimeout": 5000}
return
}
ret.Data = map[string]any{
"key": model.Conf.Repo.Key,
}
}
func initRepoKey(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
if err := model.InitRepoKey(); err != nil {
ret.Code = -1
ret.Msg = fmt.Sprintf(model.Conf.Language(137), err)
ret.Data = map[string]any{"closeTimeout": 5000}
return
}
ret.Data = map[string]any{
"key": model.Conf.Repo.Key,
}
}
func resetRepo(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
if err := model.ResetRepo(); err != nil {
ret.Code = -1
ret.Msg = fmt.Sprintf(model.Conf.Language(146), err.Error())
ret.Data = map[string]any{"closeTimeout": 5000}
return
}
}
func purgeRepo(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
if err := model.PurgeRepo(); err != nil {
ret.Code = -1
ret.Msg = fmt.Sprintf(model.Conf.Language(201), err.Error())
ret.Data = map[string]any{"closeTimeout": 5000}
return
}
}
func purgeCloudRepo(c *gin.Context) {
ret := gulu.Ret.NewResult()
defer c.JSON(http.StatusOK, ret)
if err := model.PurgeCloud(); err != nil {
ret.Code = -1
ret.Msg = fmt.Sprintf(model.Conf.Language(201), err.Error())
ret.Data = map[string]any{"closeTimeout": 5000}
return
}
}