feat: add yandex disk driver

drivers/yandex_disk/driver.go

@@ -0,0 +1,146 @@
+package yandex_disk
+
+import (
+	"context"
+	"net/http"
+	"path"
+	"strconv"
+
+	"github.com/alist-org/alist/v3/drivers/base"
+	"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"
+	"github.com/go-resty/resty/v2"
+)
+
+type YandexDisk struct {
+	model.Storage
+	Addition
+	AccessToken string
+}
+
+func (d *YandexDisk) Config() driver.Config {
+	return config
+}
+
+func (d *YandexDisk) GetAddition() driver.Additional {
+	return d.Addition
+}
+
+func (d *YandexDisk) Init(ctx context.Context, storage model.Storage) error {
+	d.Storage = storage
+	err := utils.Json.UnmarshalFromString(d.Storage.Addition, &d.Addition)
+	if err != nil {
+		return err
+	}
+	return d.refreshToken()
+}
+
+func (d *YandexDisk) Drop(ctx context.Context) error {
+	return nil
+}
+
+func (d *YandexDisk) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
+	files, err := d.getFiles(dir.GetPath())
+	if err != nil {
+		return nil, err
+	}
+	return utils.SliceConvert(files, func(src File) (model.Obj, error) {
+		return fileToObj(src), nil
+	})
+}
+
+//func (d *YandexDisk) Get(ctx context.Context, path string) (model.Obj, error) {
+//	// this is optional
+//	return nil, errs.NotImplement
+//}
+
+func (d *YandexDisk) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
+	var resp DownResp
+	_, err := d.request("/download", http.MethodGet, func(req *resty.Request) {
+		req.SetQueryParam("path", file.GetPath())
+	}, &resp)
+	if err != nil {
+		return nil, err
+	}
+	link := model.Link{
+		URL: resp.Href,
+	}
+	return &link, nil
+}
+
+func (d *YandexDisk) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
+	_, err := d.request("", http.MethodPut, func(req *resty.Request) {
+		req.SetQueryParam("path", path.Join(parentDir.GetPath(), dirName))
+	}, nil)
+	return err
+}
+
+func (d *YandexDisk) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
+	_, err := d.request("/move", http.MethodPost, func(req *resty.Request) {
+		req.SetQueryParams(map[string]string{
+			"from":      srcObj.GetPath(),
+			"path":      path.Join(dstDir.GetPath(), srcObj.GetName()),
+			"overwrite": "true",
+		})
+	}, nil)
+	return err
+}
+
+func (d *YandexDisk) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
+	_, err := d.request("/move", http.MethodPost, func(req *resty.Request) {
+		req.SetQueryParams(map[string]string{
+			"from":      srcObj.GetPath(),
+			"path":      path.Join(path.Dir(srcObj.GetPath()), newName),
+			"overwrite": "true",
+		})
+	}, nil)
+	return err
+}
+
+func (d *YandexDisk) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
+	_, err := d.request("/copy", http.MethodPost, func(req *resty.Request) {
+		req.SetQueryParams(map[string]string{
+			"from":      srcObj.GetPath(),
+			"path":      path.Join(dstDir.GetPath(), srcObj.GetName()),
+			"overwrite": "true",
+		})
+	}, nil)
+	return err
+}
+
+func (d *YandexDisk) Remove(ctx context.Context, obj model.Obj) error {
+	_, err := d.request("", http.MethodDelete, func(req *resty.Request) {
+		req.SetQueryParam("path", obj.GetPath())
+	}, nil)
+	return err
+}
+
+func (d *YandexDisk) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
+	var resp UploadResp
+	_, err := d.request("/upload", http.MethodGet, func(req *resty.Request) {
+		req.SetQueryParams(map[string]string{
+			"path":      path.Join(dstDir.GetPath(), stream.GetName()),
+			"overwrite": "true",
+		})
+	}, &resp)
+	if err != nil {
+		return err
+	}
+	req, err := http.NewRequest(resp.Method, resp.Href, stream)
+	if err != nil {
+		return err
+	}
+	req.Header.Set("Content-Length", strconv.FormatInt(stream.GetSize(), 10))
+	req.Header.Set("Content-Type", "application/octet-stream")
+	res, err := base.HttpClient.Do(req)
+	res.Body.Close()
+	return err
+}
+
+func (d *YandexDisk) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
+	return nil, errs.NotSupport
+}
+
+var _ driver.Driver = (*YandexDisk)(nil)