feat(drivers): add BaiduYun share link support

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