File size: 1,502 Bytes
7107f0b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
package mediatrack
import (
"errors"
"fmt"
"net/http"
"strconv"
"github.com/alist-org/alist/v3/drivers/base"
"github.com/alist-org/alist/v3/pkg/utils"
"github.com/go-resty/resty/v2"
log "github.com/sirupsen/logrus"
)
// do others that not defined in Driver interface
func (d *MediaTrack) request(url string, method string, callback base.ReqCallback, resp interface{}) ([]byte, error) {
req := base.RestyClient.R()
req.SetHeader("Authorization", "Bearer "+d.AccessToken)
if callback != nil {
callback(req)
}
var e BaseResp
req.SetResult(&e)
res, err := req.Execute(method, url)
if err != nil {
return nil, err
}
log.Debugln(res.String())
if e.Status != "SUCCESS" {
return nil, errors.New(e.Message)
}
if resp != nil {
err = utils.Json.Unmarshal(res.Body(), resp)
}
return res.Body(), err
}
func (d *MediaTrack) getFiles(parentId string) ([]File, error) {
files := make([]File, 0)
url := fmt.Sprintf("https://jayce.api.mediatrack.cn/v4/assets/%s/children", parentId)
sort := ""
if d.OrderBy != "" {
if d.OrderDesc {
sort = "-"
}
sort += d.OrderBy
}
page := 1
for {
var resp ChildrenResp
_, err := d.request(url, http.MethodGet, func(req *resty.Request) {
req.SetQueryParams(map[string]string{
"page": strconv.Itoa(page),
"size": "50",
"sort": sort,
})
}, &resp)
if err != nil {
return nil, err
}
if len(resp.Data.Assets) == 0 {
break
}
page++
files = append(files, resp.Data.Assets...)
}
return files, nil
}
|