feat(drivers): add BaiduYun share link support (#3801)

新增百度网盘分享链接挂载

drivers/all.go

@@ -13,6 +13,7 @@ import (
 	_ "github.com/alist-org/alist/v3/drivers/aliyundrive_share"
 	_ "github.com/alist-org/alist/v3/drivers/baidu_netdisk"
 	_ "github.com/alist-org/alist/v3/drivers/baidu_photo"
+	_ "github.com/alist-org/alist/v3/drivers/baidu_share"
 	_ "github.com/alist-org/alist/v3/drivers/cloudreve"
 	_ "github.com/alist-org/alist/v3/drivers/ftp"
 	_ "github.com/alist-org/alist/v3/drivers/google_drive"

drivers/baidu_share/driver.go

@@ -0,0 +1,153 @@
+package baidu_share
+
+import (
+	"context"
+	"fmt"
+	"net/http"
+	"net/url"
+	"path"
+	"strconv"
+	"time"
+
+	"github.com/Xhofe/go-cache"
+	"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"
+)
+
+type BaiduShare struct {
+	model.Storage
+	Addition
+	config     map[string]string
+	dlinkCache cache.ICache[string]
+}
+
+func (d *BaiduShare) Config() driver.Config {
+	return config
+}
+
+func (d *BaiduShare) GetAddition() driver.Additional {
+	return &d.Addition
+}
+
+func (d *BaiduShare) Init(ctx context.Context) error {
+	// TODO login / refresh token
+	//op.MustSaveDriverStorage(d)
+	d.config = map[string]string{}
+	d.dlinkCache = cache.NewMemCache(cache.WithClearInterval[string](time.Duration(d.CacheExpiration) * time.Minute))
+	return nil
+}
+
+func (d *BaiduShare) Drop(ctx context.Context) error {
+	d.dlinkCache.Clear()
+	return nil
+}
+
+func (d *BaiduShare) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
+	// TODO return the files list
+	body := url.Values{
+		"shorturl": {d.Surl},
+		"dir":      {dir.GetPath()},
+		"root":     {"0"},
+		"pwd":      {d.Pwd},
+		"num":      {"1000"},
+		"order":    {"time"},
+	}
+	if body.Get("dir") == "" || body.Get("dir") == d.config["root"] {
+		body.Set("root", "1")
+	}
+	res := []model.Obj{}
+	var err error
+	var page int64 = 1
+	more := true
+	for more {
+		body.Set("page", strconv.FormatInt(page, 10))
+		req := base.RestyClient.R().
+			SetCookies([]*http.Cookie{{Name: "BDUSS", Value: d.BDUSS}}).
+			SetBody(body.Encode())
+		resp, e := req.Post("https://pan.baidu.com/share/wxlist?channel=weixin&version=2.2.2&clienttype=25&web=1")
+		err = e
+		jsonresp := jsonResp{}
+		if err == nil {
+			err = base.RestyClient.JSONUnmarshal(resp.Body(), &jsonresp)
+		}
+		if err == nil && jsonresp.Errno == 0 {
+			more = jsonresp.Data.More
+			page += 1
+			for _, v := range jsonresp.Data.List {
+				size, _ := v.Size.Int64()
+				mtime, _ := v.Time.Int64()
+				res = append(res, &model.Object{
+					ID:       v.ID.String(),
+					Path:     v.Path,
+					Name:     v.Name,
+					Size:     size,
+					Modified: time.Unix(mtime, 0),
+					IsFolder: v.Dir.String() == "1",
+				})
+				d.dlinkCache.Set(v.Path, v.Dlink, cache.WithEx[string](time.Duration(d.CacheExpiration/2)*time.Minute))
+			}
+			if len(res) > 0 && body.Get("root") == "1" {
+				d.config["root"] = path.Dir(res[0].GetPath())
+			}
+		} else {
+			if err == nil {
+				err = fmt.Errorf("errno:%d", jsonresp.Errno)
+			}
+			break
+		}
+	}
+	return res, err
+}
+
+func (d *BaiduShare) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
+	// TODO return link of file
+	var err error
+	req := base.RestyClient.R().SetCookies([]*http.Cookie{{Name: "BDUSS", Value: d.BDUSS}})
+	url, found := d.dlinkCache.Get(file.GetPath())
+	if !found {
+		_, err = d.List(ctx, &model.Object{Path: path.Dir(file.GetPath())}, model.ListArgs{})
+		url, found = d.dlinkCache.Get(file.GetPath())
+		if err == nil && !found {
+			err = errs.NotSupport
+		}
+	}
+	return &model.Link{URL: url, Header: req.Header}, err
+}
+
+func (d *BaiduShare) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) error {
+	// TODO create folder
+	return errs.NotSupport
+}
+
+func (d *BaiduShare) Move(ctx context.Context, srcObj, dstDir model.Obj) error {
+	// TODO move obj
+	return errs.NotSupport
+}
+
+func (d *BaiduShare) Rename(ctx context.Context, srcObj model.Obj, newName string) error {
+	// TODO rename obj
+	return errs.NotSupport
+}
+
+func (d *BaiduShare) Copy(ctx context.Context, srcObj, dstDir model.Obj) error {
+	// TODO copy obj
+	return errs.NotSupport
+}
+
+func (d *BaiduShare) Remove(ctx context.Context, obj model.Obj) error {
+	// TODO remove obj
+	return errs.NotSupport
+}
+
+func (d *BaiduShare) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) error {
+	// TODO upload file
+	return errs.NotSupport
+}
+
+//func (d *Template) Other(ctx context.Context, args model.OtherArgs) (interface{}, error) {
+//	return nil, errs.NotSupport
+//}
+
+var _ driver.Driver = (*BaiduShare)(nil)

drivers/baidu_share/meta.go

@@ -0,0 +1,34 @@
+package baidu_share
+
+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
+	// driver.RootID
+	// define other
+	// Field string `json:"field" type:"select" required:"true" options:"a,b,c" default:"a"`
+	Surl  string `json:"surl"`
+	Pwd   string `json:"pwd"`
+	BDUSS string `json:"BDUSS"`
+}
+
+var config = driver.Config{
+	Name:        "BaiduShare",
+	LocalSort:   true,
+	OnlyLocal:   false,
+	OnlyProxy:   false,
+	NoCache:     false,
+	NoUpload:    true,
+	NeedMs:      false,
+	DefaultRoot: "",
+}
+
+func init() {
+	op.RegisterDriver(func() driver.Driver {
+		return &BaiduShare{}
+	})
+}

drivers/baidu_share/types.go

@@ -0,0 +1,19 @@
+package baidu_share
+
+import "encoding/json"
+
+type jsonResp struct {
+	Errno int64 `json:"errno"`
+	Data  struct {
+		More bool `json:"has_more"`
+		List []struct {
+			ID    json.Number `json:"fs_id"`
+			Dir   json.Number `json:"isdir"`
+			Path  string      `json:"path"`
+			Name  string      `json:"server_filename"`
+			Time  json.Number `json:"server_mtime"`
+			Size  json.Number `json:"size"`
+			Dlink string      `json:"dlink"`
+		} `json:"list"`
+	} `json:"data"`
+}

drivers/baidu_share/util.go

@@ -0,0 +1,3 @@
+package baidu_share
+
+// do others that not defined in Driver interface