feat(drivers): alias a new storage with multi path (close #3248)

drivers/alias/driver.go

@@ -0,0 +1,106 @@
+package alias
+
+import (
+	"context"
+	"errors"
+	"strings"
+
+	"github.com/alist-org/alist/v3/internal/driver"
+	"github.com/alist-org/alist/v3/internal/errs"
+	"github.com/alist-org/alist/v3/internal/model"
+	"github.com/alist-org/alist/v3/pkg/utils"
+)
+
+type Alias struct {
+	model.Storage
+	Addition
+	pathMap map[string][]string
+}
+
+func (d *Alias) Config() driver.Config {
+	return config
+}
+
+func (d *Alias) GetAddition() driver.Additional {
+	return &d.Addition
+}
+
+func (d *Alias) Init(ctx context.Context) error {
+	if d.Paths == "" {
+		return errors.New("paths is required")
+	}
+	d.pathMap = make(map[string][]string)
+	for _, path := range strings.Split(d.Paths, "\n") {
+		path = strings.TrimSpace(path)
+		if path == "" {
+			continue
+		}
+		k, v := getPair(path)
+		d.pathMap[k] = append(d.pathMap[k], v)
+	}
+	return nil
+}
+
+func (d *Alias) Drop(ctx context.Context) error {
+	d.pathMap = nil
+	return nil
+}
+
+func (d *Alias) Get(ctx context.Context, path string) (model.Obj, error) {
+	if utils.PathEqual(path, "/") {
+		return &model.Object{
+			Name:     "Root",
+			IsFolder: true,
+			Path:     "/",
+		}, nil
+	}
+	root, sub := getRootAndPath(path)
+	dsts, ok := d.pathMap[root]
+	if !ok {
+		return nil, errs.ObjectNotFound
+	}
+	for _, dst := range dsts {
+		obj, err := d.get(ctx, path, dst, sub)
+		if err == nil {
+			return obj, nil
+		}
+	}
+	return nil, errs.ObjectNotFound
+}
+
+func (d *Alias) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
+	path := dir.GetPath()
+	if utils.PathEqual(path, "/") {
+		return d.listRoot(), nil
+	}
+	root, sub := getRootAndPath(path)
+	dsts, ok := d.pathMap[root]
+	if !ok {
+		return nil, errs.ObjectNotFound
+	}
+	var objs []model.Obj
+	for _, dst := range dsts {
+		tmp, err := d.list(ctx, dst, sub)
+		if err == nil {
+			objs = append(objs, tmp...)
+		}
+	}
+	return objs, nil
+}
+
+func (d *Alias) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
+	root, sub := getRootAndPath(file.GetPath())
+	dsts, ok := d.pathMap[root]
+	if !ok {
+		return nil, errs.ObjectNotFound
+	}
+	for _, dst := range dsts {
+		link, err := d.link(ctx, dst, sub, args)
+		if err == nil {
+			return link, nil
+		}
+	}
+	return nil, errs.ObjectNotFound
+}
+
+var _ driver.Driver = (*Alias)(nil)

drivers/alias/meta.go

@@ -0,0 +1,27 @@
+package alias
+
+import (
+	"github.com/alist-org/alist/v3/internal/driver"
+	"github.com/alist-org/alist/v3/internal/op"
+)
+
+type Addition struct {
+	// Usually one of two
+	// driver.RootPath
+	// define other
+	Paths string `json:"paths" required:"true" type:"text"`
+}
+
+var config = driver.Config{
+	Name:        "Alias",
+	LocalSort:   true,
+	NoCache:     true,
+	NoUpload:    true,
+	DefaultRoot: "/",
+}
+
+func init() {
+	op.RegisterDriver(func() driver.Driver {
+		return &Alias{}
+	})
+}

drivers/alias/types.go

@@ -0,0 +1 @@
+package alias

drivers/alias/util.go

@@ -0,0 +1,81 @@
+package alias
+
+import (
+	"context"
+	stdpath "path"
+	"strings"
+
+	"github.com/alist-org/alist/v3/internal/fs"
+	"github.com/alist-org/alist/v3/internal/model"
+	"github.com/alist-org/alist/v3/pkg/utils"
+)
+
+func (d *Alias) listRoot() []model.Obj {
+	var objs []model.Obj
+	for k, _ := range d.pathMap {
+		obj := model.Object{
+			Name:     k,
+			IsFolder: true,
+			Modified: d.Modified,
+		}
+		objs = append(objs, &obj)
+	}
+	return objs
+}
+
+// do others that not defined in Driver interface
+func getPair(path string) (string, string) {
+	//path = strings.TrimSpace(path)
+	if strings.Contains(path, ":") {
+		pair := strings.SplitN(path, ":", 2)
+		if !strings.Contains(pair[0], "/") {
+			return pair[0], pair[1]
+		}
+	}
+	return stdpath.Base(path), path
+}
+
+func getRootAndPath(path string) (string, string) {
+	path = strings.TrimPrefix(path, "/")
+	parts := strings.SplitN(path, "/", 2)
+	if len(parts) == 1 {
+		return parts[0], ""
+	}
+	return parts[0], parts[1]
+}
+
+func (d *Alias) get(ctx context.Context, path string, dst, sub string) (model.Obj, error) {
+	obj, err := fs.Get(ctx, stdpath.Join(dst, sub))
+	if err != nil {
+		return nil, err
+	}
+	return &model.Object{
+		Path:     path,
+		Name:     obj.GetName(),
+		Size:     obj.GetSize(),
+		Modified: obj.ModTime(),
+		IsFolder: obj.IsDir(),
+	}, nil
+}
+
+func (d *Alias) list(ctx context.Context, dst, sub string) ([]model.Obj, error) {
+	objs, err := fs.List(ctx, stdpath.Join(dst, sub))
+	// the obj must implement the model.SetPath interface
+	// return objs, err
+	if err != nil {
+		return nil, err
+	}
+	return utils.SliceConvert(objs, func(obj model.Obj) (model.Obj, error) {
+		return &model.Object{
+			Name:     obj.GetName(),
+			Size:     obj.GetSize(),
+			Modified: obj.ModTime(),
+			IsFolder: obj.IsDir(),
+		}, nil
+	})
+}
+
+func (d *Alias) link(ctx context.Context, dst, sub string, args model.LinkArgs) (*model.Link, error) {
+	link, _, err := fs.Link(ctx, stdpath.Join(dst, sub), args)
+	return link, err
+}