|
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 { |
|
|
|
code, err := d.getCode(ctx1) |
|
if err != nil { |
|
return err |
|
} |
|
|
|
err = d.getRefreshTokenByTV(ctx1, code, false) |
|
if err != nil { |
|
return err |
|
} |
|
} |
|
} |
|
|
|
if d.QuarkUCTVCommon.AccessToken == "" { |
|
err := d.getRefreshTokenByTV(ctx1, d.Addition.RefreshToken, true) |
|
if err != nil { |
|
return err |
|
} |
|
} |
|
|
|
|
|
_, 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) |
|
|
