feat: add support for quark tv driver and uc tv driver (#6959)

AlistGo · Aug 10, 2024 · 62ed169 · 62ed169
1 parent 979d0cf
commit 62ed169
Show file tree

Hide file tree

Showing 5 changed files with 555 additions and 0 deletions.
diff --git a/drivers/all.go b/drivers/all.go
@@ -41,6 +41,7 @@ import (
 	_ "github.com/alist-org/alist/v3/drivers/pikpak"
 	_ "github.com/alist-org/alist/v3/drivers/pikpak_share"
 	_ "github.com/alist-org/alist/v3/drivers/quark_uc"
+	_ "github.com/alist-org/alist/v3/drivers/quark_uc_tv"
 	_ "github.com/alist-org/alist/v3/drivers/quqi"
 	_ "github.com/alist-org/alist/v3/drivers/s3"
 	_ "github.com/alist-org/alist/v3/drivers/seafile"

diff --git a/drivers/quark_uc_tv/driver.go b/drivers/quark_uc_tv/driver.go
@@ -0,0 +1,174 @@
+package quark_uc_tv
+
+import (
+	"context"
+	"fmt"
+	"github.com/alist-org/alist/v3/internal/op"
+	"github.com/alist-org/alist/v3/pkg/utils"
+	"github.com/go-resty/resty/v2"
+	"strconv"
+	"time"
+
+	"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 QuarkUCTV struct {
+	*QuarkUCTVCommon
+	model.Storage
+	Addition
+	config driver.Config
+	conf   Conf
+}
+
+func (d *QuarkUCTV) Config() driver.Config {
+	return d.config
+}
+
+func (d *QuarkUCTV) GetAddition() driver.Additional {
+	return &d.Addition
+}
+
+func (d *QuarkUCTV) Init(ctx context.Context) error {
+
+	if d.Addition.DeviceID == "" {
+		d.Addition.DeviceID = utils.GetMD5EncodeStr(time.Now().String())
+	}
+	op.MustSaveDriverStorage(d)
+
+	if d.QuarkUCTVCommon == nil {
+		d.QuarkUCTVCommon = &QuarkUCTVCommon{
+			AccessToken: "",
+		}
+	}
+	ctx1, cancelFunc := context.WithTimeout(ctx, 5*time.Second)
+	defer cancelFunc()
+	if d.Addition.RefreshToken == "" {
+		if d.Addition.QueryToken == "" {
+			qrData, err := d.getLoginCode(ctx1)
+			if err != nil {
+				return err
+			}
+			// 展示二维码
+			qrTemplate := `<body>
+        <img src="data:image/jpeg;base64,%s"/>
+    </body>`
+			qrPage := fmt.Sprintf(qrTemplate, qrData)
+			return fmt.Errorf("need verify: \n%s", qrPage)
+		} else {
+			// 通过query token获取code -> refresh token
+			code, err := d.getCode(ctx1)
+			if err != nil {
+				return err
+			}
+			// 通过code获取refresh token
+			err = d.getRefreshTokenByTV(ctx1, code, false)
+			if err != nil {
+				return err
+			}
+		}
+	}
+	// 通过refresh token获取access token
+	if d.QuarkUCTVCommon.AccessToken == "" {
+		err := d.getRefreshTokenByTV(ctx1, d.Addition.RefreshToken, true)
+		if err != nil {
+			return err
+		}
+	}
+
+	// 验证 access token 是否有效
+	_, err := d.isLogin(ctx1)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (d *QuarkUCTV) Drop(ctx context.Context) error {
+	return nil
+}
+
+func (d *QuarkUCTV) List(ctx context.Context, dir model.Obj, args model.ListArgs) ([]model.Obj, error) {
+	files := make([]model.Obj, 0)
+	pageIndex := int64(0)
+	pageSize := int64(100)
+	for {
+		var filesData FilesData
+		_, err := d.request(ctx, "/file", "GET", func(req *resty.Request) {
+			req.SetQueryParams(map[string]string{
+				"method":     "list",
+				"parent_fid": dir.GetID(),
+				"order_by":   "3",
+				"desc":       "1",
+				"category":   "",
+				"source":     "",
+				"ex_source":  "",
+				"list_all":   "0",
+				"page_size":  strconv.FormatInt(pageSize, 10),
+				"page_index": strconv.FormatInt(pageIndex, 10),
+			})
+		}, &filesData)
+		if err != nil {
+			return nil, err
+		}
+		for i := range filesData.Data.Files {
+			files = append(files, &filesData.Data.Files[i])
+		}
+		if pageIndex*pageSize >= filesData.Data.TotalCount {
+			break
+		} else {
+			pageIndex++
+		}
+	}
+	return files, nil
+}
+
+func (d *QuarkUCTV) Link(ctx context.Context, file model.Obj, args model.LinkArgs) (*model.Link, error) {
+	files := &model.Link{}
+	var fileLink FileLink
+	_, err := d.request(ctx, "/file", "GET", func(req *resty.Request) {
+		req.SetQueryParams(map[string]string{
+			"method":     "download",
+			"group_by":   "source",
+			"fid":        file.GetID(),
+			"resolution": "low,normal,high,super,2k,4k",
+			"support":    "dolby_vision",
+		})
+	}, &fileLink)
+	if err != nil {
+		return nil, err
+	}
+	files.URL = fileLink.Data.DownloadURL
+	return files, nil
+}
+
+func (d *QuarkUCTV) MakeDir(ctx context.Context, parentDir model.Obj, dirName string) (model.Obj, error) {
+	return nil, errs.NotImplement
+}
+
+func (d *QuarkUCTV) Move(ctx context.Context, srcObj, dstDir model.Obj) (model.Obj, error) {
+	return nil, errs.NotImplement
+}
+
+func (d *QuarkUCTV) Rename(ctx context.Context, srcObj model.Obj, newName string) (model.Obj, error) {
+	return nil, errs.NotImplement
+}
+
+func (d *QuarkUCTV) Copy(ctx context.Context, srcObj, dstDir model.Obj) (model.Obj, error) {
+	return nil, errs.NotImplement
+}
+
+func (d *QuarkUCTV) Remove(ctx context.Context, obj model.Obj) error {
+	return errs.NotImplement
+}
+
+func (d *QuarkUCTV) Put(ctx context.Context, dstDir model.Obj, stream model.FileStreamer, up driver.UpdateProgress) (model.Obj, error) {
+	return nil, errs.NotImplement
+}
+
+type QuarkUCTVCommon struct {
+	AccessToken string
+}
+
+var _ driver.Driver = (*QuarkUCTV)(nil)
diff --git a/drivers/quark_uc_tv/meta.go b/drivers/quark_uc_tv/meta.go
@@ -0,0 +1,67 @@
+package quark_uc_tv
+
+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.RootID
+	// define other
+	RefreshToken string `json:"refresh_token" required:"false" default:""`
+	// 必要且影响登录,由签名决定
+	DeviceID string `json:"device_id"  required:"false" default:""`
+	// 登陆所用的数据 无需手动填写
+	QueryToken string `json:"query_token" required:"false" default:"" help:"don't edit'"`
+}
+
+type Conf struct {
+	api      string
+	clientID string
+	signKey  string
+	appVer   string
+	channel  string
+	codeApi  string
+}
+
+func init() {
+	op.RegisterDriver(func() driver.Driver {
+		return &QuarkUCTV{
+			config: driver.Config{
+				Name:              "QuarkTV",
+				OnlyLocal:         false,
+				DefaultRoot:       "0",
+				NoOverwriteUpload: true,
+				NoUpload:          true,
+			},
+			conf: Conf{
+				api:      "https://open-api-drive.quark.cn",
+				clientID: "d3194e61504e493eb6222857bccfed94",
+				signKey:  "kw2dvtd7p4t3pjl2d9ed9yc8yej8kw2d",
+				appVer:   "1.5.6",
+				channel:  "CP",
+				codeApi:  "http://api.extscreen.com/quarkdrive",
+			},
+		}
+	})
+	op.RegisterDriver(func() driver.Driver {
+		return &QuarkUCTV{
+			config: driver.Config{
+				Name:              "UCTV",
+				OnlyLocal:         false,
+				DefaultRoot:       "0",
+				NoOverwriteUpload: true,
+				NoUpload:          true,
+			},
+			conf: Conf{
+				api:      "https://open-api-drive.uc.cn",
+				clientID: "5acf882d27b74502b7040b0c65519aa7",
+				signKey:  "l3srvtd7p42l0d0x1u8d7yc8ye9kki4d",
+				appVer:   "1.6.5",
+				channel:  "UCTVOFFICIALWEB",
+				codeApi:  "http://api.extscreen.com/ucdrive",
+			},
+		}
+	})
+}
diff --git a/drivers/quark_uc_tv/types.go b/drivers/quark_uc_tv/types.go
@@ -0,0 +1,102 @@
+package quark_uc_tv
+
+import (
+	"github.com/alist-org/alist/v3/internal/model"
+	"github.com/alist-org/alist/v3/pkg/utils"
+	"time"
+)
+
+type Resp struct {
+	CommonRsp
+	Errno     int    `json:"errno"`
+	ErrorInfo string `json:"error_info"`
+}
+
+type CommonRsp struct {
+	Status int    `json:"status"`
+	ReqID  string `json:"req_id"`
+}
+
+type RefreshTokenAuthResp struct {
+	Code    int    `json:"code"`
+	Message string `json:"message"`
+	Data    struct {
+		Status       int    `json:"status"`
+		Errno        int    `json:"errno"`
+		ErrorInfo    string `json:"error_info"`
+		ReqID        string `json:"req_id"`
+		AccessToken  string `json:"access_token"`
+		RefreshToken string `json:"refresh_token"`
+		ExpiresIn    int    `json:"expires_in"`
+		Scope        string `json:"scope"`
+	} `json:"data"`
+}
+type Files struct {
+	Fid          string `json:"fid"`
+	ParentFid    string `json:"parent_fid"`
+	Category     int    `json:"category"`
+	Filename     string `json:"filename"`
+	Size         int64  `json:"size"`
+	FileType     string `json:"file_type"`
+	SubItems     int    `json:"sub_items,omitempty"`
+	Isdir        int    `json:"isdir"`
+	Duration     int    `json:"duration"`
+	CreatedAt    int64  `json:"created_at"`
+	UpdatedAt    int64  `json:"updated_at"`
+	IsBackup     int    `json:"is_backup"`
+	ThumbnailURL string `json:"thumbnail_url,omitempty"`
+}
+
+func (f *Files) GetSize() int64 {
+	return f.Size
+}
+
+func (f *Files) GetName() string {
+	return f.Filename
+}
+
+func (f *Files) ModTime() time.Time {
+	//return time.Unix(f.UpdatedAt, 0)
+	return time.Unix(0, f.UpdatedAt*int64(time.Millisecond))
+}
+
+func (f *Files) CreateTime() time.Time {
+	//return time.Unix(f.CreatedAt, 0)
+	return time.Unix(0, f.CreatedAt*int64(time.Millisecond))
+}
+
+func (f *Files) IsDir() bool {
+	return f.Isdir == 1
+}
+
+func (f *Files) GetHash() utils.HashInfo {
+	return utils.HashInfo{}
+}
+
+func (f *Files) GetID() string {
+	return f.Fid
+}
+
+func (f *Files) GetPath() string {
+	return ""
+}
+
+var _ model.Obj = (*Files)(nil)
+
+type FilesData struct {
+	CommonRsp
+	Data struct {
+		TotalCount int64   `json:"total_count"`
+		Files      []Files `json:"files"`
+	} `json:"data"`
+}
+
+type FileLink struct {
+	CommonRsp
+	Data struct {
+		Fid         string `json:"fid"`
+		FileName    string `json:"file_name"`
+		Size        int64  `json:"size"`
+		DownloadURL string `json:"download_url"`
+	} `json:"data"`
+}