mirror of https://github.com/Xhofe/alist
104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package fs
|
|
|
|
import (
|
|
"context"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/alist-org/alist/v3/internal/model"
|
|
"github.com/alist-org/alist/v3/internal/op"
|
|
"github.com/alist-org/alist/v3/pkg/utils"
|
|
"github.com/pkg/errors"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// List files
|
|
func list(ctx context.Context, path string, refresh ...bool) ([]model.Obj, error) {
|
|
meta := ctx.Value("meta").(*model.Meta)
|
|
user := ctx.Value("user").(*model.User)
|
|
var objs []model.Obj
|
|
storage, actualPath, err := op.GetStorageAndActualPath(path)
|
|
virtualFiles := op.GetStorageVirtualFilesByPath(path)
|
|
if err != nil {
|
|
if len(virtualFiles) == 0 {
|
|
return nil, errors.WithMessage(err, "failed get storage")
|
|
}
|
|
} else {
|
|
_objs, err := op.List(ctx, storage, actualPath, model.ListArgs{
|
|
ReqPath: path,
|
|
}, refresh...)
|
|
if err != nil {
|
|
log.Errorf("%+v", err)
|
|
if len(virtualFiles) == 0 {
|
|
return nil, errors.WithMessage(err, "failed get objs")
|
|
}
|
|
}
|
|
objs = make([]model.Obj, len(_objs))
|
|
copy(objs, _objs)
|
|
}
|
|
if objs == nil {
|
|
objs = virtualFiles
|
|
} else {
|
|
for _, storageFile := range virtualFiles {
|
|
if !containsByName(objs, storageFile) {
|
|
objs = append(objs, storageFile)
|
|
}
|
|
}
|
|
}
|
|
if whetherHide(user, meta, path) {
|
|
objs = hide(objs, meta)
|
|
}
|
|
// sort objs
|
|
if storage != nil {
|
|
if storage.Config().LocalSort {
|
|
model.SortFiles(objs, storage.GetStorage().OrderBy, storage.GetStorage().OrderDirection)
|
|
}
|
|
model.ExtractFolder(objs, storage.GetStorage().ExtractFolder)
|
|
}
|
|
return objs, nil
|
|
}
|
|
|
|
func whetherHide(user *model.User, meta *model.Meta, path string) bool {
|
|
// if is admin, don't hide
|
|
if user.CanSeeHides() {
|
|
return false
|
|
}
|
|
// if meta is nil, don't hide
|
|
if meta == nil {
|
|
return false
|
|
}
|
|
// if meta.Hide is empty, don't hide
|
|
if meta.Hide == "" {
|
|
return false
|
|
}
|
|
// if meta doesn't apply to sub_folder, don't hide
|
|
if !utils.PathEqual(meta.Path, path) && !meta.HSub {
|
|
return false
|
|
}
|
|
// if is guest, hide
|
|
return true
|
|
}
|
|
|
|
func hide(objs []model.Obj, meta *model.Meta) []model.Obj {
|
|
var res []model.Obj
|
|
deleted := make([]bool, len(objs))
|
|
rs := strings.Split(meta.Hide, "\n")
|
|
for _, r := range rs {
|
|
re := regexp.MustCompile(r)
|
|
for i, obj := range objs {
|
|
if deleted[i] {
|
|
continue
|
|
}
|
|
if re.MatchString(obj.GetName()) {
|
|
deleted[i] = true
|
|
}
|
|
}
|
|
}
|
|
for i, obj := range objs {
|
|
if !deleted[i] {
|
|
res = append(res, obj)
|
|
}
|
|
}
|
|
return res
|
|
}
|