✨ upyun uss support

drivers/uss/uss.go

@@ -0,0 +1,82 @@
+package uss
+
+import (
+	"errors"
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/drivers/base"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/utils"
+	"github.com/upyun/go-sdk/v3/upyun"
+	"path"
+	"strings"
+)
+
+var clientsMap map[string]*upyun.UpYun
+
+func (driver USS) NewUpYun(account *model.Account) (*upyun.UpYun, error) {
+	return upyun.NewUpYun(&upyun.UpYunConfig{
+		Bucket:   account.Bucket,
+		Operator: account.AccessKey,
+		Password: account.AccessToken,
+	}), nil
+}
+
+func (driver USS) GetClient(account *model.Account) (*upyun.UpYun, error) {
+	client, ok := clientsMap[account.Name]
+	if ok {
+		return client, nil
+	}
+	return nil, errors.New("can't get client")
+}
+
+func (driver USS) List(prefix string, account *model.Account) ([]model.File, error) {
+	prefix = driver.GetKey(prefix, account, true)
+	client, err := driver.GetClient(account)
+	if err != nil {
+		return nil, err
+	}
+	objsChan := make(chan *upyun.FileInfo, 10)
+	defer close(objsChan)
+	go func() {
+		err = client.List(&upyun.GetObjectsConfig{
+			Path:           prefix,
+			ObjectsChan:    objsChan,
+			MaxListObjects: 0,
+			MaxListLevel:   1,
+		})
+	}()
+	if err != nil {
+		return nil, err
+	}
+	res := make([]model.File, 0)
+	for obj := range objsChan {
+		t := obj.Time
+		f := model.File{
+			Name:      obj.Name,
+			Size:      obj.Size,
+			UpdatedAt: &t,
+			Driver:    driver.Config().Name,
+		}
+		if obj.IsDir {
+			f.Type = conf.FOLDER
+		} else {
+			f.Type = utils.GetFileType(path.Ext(obj.Name))
+		}
+		res = append(res, f)
+	}
+	return res, err
+}
+
+func (driver USS) GetKey(path string, account *model.Account, dir bool) string {
+	path = utils.Join(account.RootFolder, path)
+	path = strings.TrimPrefix(path, "/")
+	if dir {
+		path += "/"
+	}
+	return path
+}
+
+func init() {
+	clientsMap = make(map[string]*upyun.UpYun)
+	base.RegisterDriver(&USS{})
+}