mirror of https://github.com/Xhofe/alist
104 lines
2.2 KiB
Go
104 lines
2.2 KiB
Go
package common
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/Xhofe/alist/drivers/base"
|
|
"github.com/Xhofe/alist/model"
|
|
"github.com/Xhofe/alist/utils"
|
|
"github.com/gin-gonic/gin"
|
|
log "github.com/sirupsen/logrus"
|
|
"strings"
|
|
)
|
|
|
|
type Resp struct {
|
|
Code int `json:"code"`
|
|
Message string `json:"message"`
|
|
Data interface{} `json:"data"`
|
|
}
|
|
|
|
type PathReq struct {
|
|
Path string `json:"path"`
|
|
Password string `json:"password"`
|
|
PageNum int `json:"page_num"`
|
|
PageSize int `json:"page_size"`
|
|
}
|
|
|
|
func ParsePath(rawPath string) (*model.Account, string, base.Driver, error) {
|
|
var path, name string
|
|
switch model.AccountsCount() {
|
|
case 0:
|
|
return nil, "", nil, fmt.Errorf("no accounts,please add one first")
|
|
case 1:
|
|
path = rawPath
|
|
break
|
|
default:
|
|
if path == "/" {
|
|
return nil, "", nil, errors.New("can't operate root of multiple accounts")
|
|
}
|
|
paths := strings.Split(rawPath, "/")
|
|
path = "/" + strings.Join(paths[2:], "/")
|
|
name = paths[1]
|
|
}
|
|
account, ok := model.GetBalancedAccount(name)
|
|
if !ok {
|
|
return nil, "", nil, fmt.Errorf("no [%s] account", name)
|
|
}
|
|
driver, ok := base.GetDriver(account.Type)
|
|
if !ok {
|
|
return nil, "", nil, fmt.Errorf("no [%s] driver", account.Type)
|
|
}
|
|
return &account, path, driver, nil
|
|
}
|
|
|
|
func ErrorResp(c *gin.Context, err error, code int) {
|
|
log.Error(err.Error())
|
|
c.JSON(200, Resp{
|
|
Code: code,
|
|
Message: err.Error(),
|
|
Data: nil,
|
|
})
|
|
c.Abort()
|
|
}
|
|
|
|
func ErrorStrResp(c *gin.Context, str string, code int) {
|
|
log.Error(str)
|
|
c.JSON(200, Resp{
|
|
Code: code,
|
|
Message: str,
|
|
Data: nil,
|
|
})
|
|
c.Abort()
|
|
}
|
|
|
|
func SuccessResp(c *gin.Context, data ...interface{}) {
|
|
if len(data) == 0 {
|
|
c.JSON(200, Resp{
|
|
Code: 200,
|
|
Message: "success",
|
|
Data: nil,
|
|
})
|
|
return
|
|
}
|
|
c.JSON(200, Resp{
|
|
Code: 200,
|
|
Message: "success",
|
|
Data: data[0],
|
|
})
|
|
}
|
|
|
|
func Hide(meta *model.Meta, files []model.File) []model.File {
|
|
//meta, _ := model.GetMetaByPath(path)
|
|
if meta != nil && meta.Hide != "" {
|
|
tmpFiles := make([]model.File, 0)
|
|
hideFiles := strings.Split(meta.Hide, ",")
|
|
for _, item := range files {
|
|
if !utils.IsContain(hideFiles, item.Name) {
|
|
tmpFiles = append(tmpFiles, item)
|
|
}
|
|
}
|
|
files = tmpFiles
|
|
}
|
|
return files
|
|
}
|