alist/internal/fs/list.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
}