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/operations"
	"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) ([]model.Obj, error) {
	meta := ctx.Value("meta").(*model.Meta)
	user := ctx.Value("user").(*model.User)
	storage, actualPath, err := operations.GetStorageAndActualPath(path)
	virtualFiles := operations.GetStorageVirtualFilesByPath(path)
	if err != nil {
		if len(virtualFiles) != 0 {
			return virtualFiles, nil
		}
		return nil, errors.WithMessage(err, "failed get storage")
	}
	objs, err := operations.List(ctx, storage, actualPath, model.ListArgs{
		ReqPath: path,
	})
	if err != nil {
		log.Errorf("%+v", err)
		if len(virtualFiles) != 0 {
			return virtualFiles, nil
		}
		return nil, errors.WithMessage(err, "failed get objs")
	}
	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.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.Compile(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
}