mirror of
https://github.com/IceWhaleTech/CasaOS.git
synced 2025-06-16 05:55:33 +00:00
80 lines
3.0 KiB
Go
80 lines
3.0 KiB
Go
package v1
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/IceWhaleTech/CasaOS-Common/utils/logger"
|
|
"github.com/IceWhaleTech/CasaOS/drivers/dropbox"
|
|
"github.com/IceWhaleTech/CasaOS/drivers/google_drive"
|
|
"github.com/IceWhaleTech/CasaOS/drivers/onedrive"
|
|
"github.com/IceWhaleTech/CasaOS/model"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/utils/common_err"
|
|
"github.com/IceWhaleTech/CasaOS/pkg/utils/httper"
|
|
"github.com/IceWhaleTech/CasaOS/service"
|
|
"github.com/labstack/echo/v4"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func ListStorages(ctx echo.Context) error {
|
|
r, err := service.MyService.Storage().GetStorages()
|
|
if err != nil {
|
|
return ctx.JSON(common_err.SUCCESS, model.Result{Success: common_err.SERVICE_ERROR, Message: common_err.GetMsg(common_err.SERVICE_ERROR), Data: err.Error()})
|
|
}
|
|
|
|
for i := 0; i < len(r.MountPoints); i++ {
|
|
dataMap, err := service.MyService.Storage().GetConfigByName(r.MountPoints[i].Fs)
|
|
if err != nil {
|
|
logger.Error("GetConfigByName", zap.Any("err", err))
|
|
continue
|
|
}
|
|
if dataMap["type"] == "drive" {
|
|
r.MountPoints[i].Icon = google_drive.ICONURL
|
|
}
|
|
if dataMap["type"] == "dropbox" {
|
|
r.MountPoints[i].Icon = dropbox.ICONURL
|
|
}
|
|
if dataMap["type"] == "onedrive" {
|
|
r.MountPoints[i].Icon = onedrive.ICONURL
|
|
}
|
|
r.MountPoints[i].Name = dataMap["username"]
|
|
}
|
|
list := []httper.MountPoint{}
|
|
|
|
for _, v := range r.MountPoints {
|
|
list = append(list, httper.MountPoint(v))
|
|
}
|
|
|
|
return ctx.JSON(common_err.SUCCESS, model.Result{Success: common_err.SUCCESS, Message: common_err.GetMsg(common_err.SUCCESS), Data: list})
|
|
}
|
|
|
|
func UmountStorage(ctx echo.Context) error {
|
|
json := make(map[string]string)
|
|
ctx.Bind(&json)
|
|
mountPoint := json["mount_point"]
|
|
if mountPoint == "" {
|
|
return ctx.JSON(common_err.CLIENT_ERROR, model.Result{Success: common_err.CLIENT_ERROR, Message: common_err.GetMsg(common_err.CLIENT_ERROR), Data: "mount_point is empty"})
|
|
}
|
|
err := service.MyService.Storage().UnmountStorage(mountPoint)
|
|
if err != nil {
|
|
return ctx.JSON(common_err.SERVICE_ERROR, model.Result{Success: common_err.SERVICE_ERROR, Message: common_err.GetMsg(common_err.SERVICE_ERROR), Data: err.Error()})
|
|
}
|
|
service.MyService.Storage().DeleteConfigByName(strings.ReplaceAll(mountPoint, "/mnt/", ""))
|
|
return ctx.JSON(common_err.SUCCESS, model.Result{Success: common_err.SUCCESS, Message: common_err.GetMsg(common_err.SUCCESS), Data: "success"})
|
|
}
|
|
|
|
func GetStorage(ctx echo.Context) error {
|
|
// idStr := ctx.QueryParam("id")
|
|
// id, err := strconv.Atoi(idStr)
|
|
// if err != nil {
|
|
// return ctx.JSON(common_err.SUCCESS, model.Result{Success: common_err.CLIENT_ERROR, Message: common_err.GetMsg(common_err.CLIENT_ERROR), Data: err.Error()})
|
|
// return
|
|
// }
|
|
// storage, err := service.MyService.Storage().GetStorageById(uint(id))
|
|
// if err != nil {
|
|
// return ctx.JSON(common_err.SUCCESS, model.Result{Success: common_err.SERVICE_ERROR, Message: common_err.GetMsg(common_err.SERVICE_ERROR), Data: err.Error()})
|
|
// return
|
|
// }
|
|
// return ctx.JSON(common_err.SUCCESS, model.Result{Success: common_err.SUCCESS, Message: common_err.GetMsg(common_err.SUCCESS), Data: storage})
|
|
return nil
|
|
}
|