✨ support shandianpan #234

drivers/shandian/shandian.go

@@ -0,0 +1,147 @@
+package shandian
+
+import (
+	"errors"
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/drivers/base"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/utils"
+	log "github.com/sirupsen/logrus"
+	"strconv"
+	"time"
+)
+
+type Resp struct {
+	Code int    `json:"code"`
+	Msg  string `json:"msg"`
+}
+
+type LoginResp struct {
+	Resp
+	Data struct {
+		Token string `json:"token"`
+	} `json:"data"`
+}
+
+func (driver Shandian) Login(account *model.Account) error {
+	var resp LoginResp
+	_, err := base.RestyClient.R().SetResult(&resp).SetHeader("Accept", "application/json").SetBody(base.Json{
+		"mobile":   account.Username,
+		"password": account.Password,
+		"smscode":  "",
+	}).Post("https://shandianpan.com/api/login")
+	if err != nil {
+		return err
+	}
+	if resp.Code != 0 {
+		return errors.New(resp.Msg)
+	}
+	account.AccessToken = resp.Data.Token
+	_ = model.SaveAccount(account)
+	return nil
+}
+
+type File struct {
+	Id         int64  `json:"id"`
+	Type       int    `json:"type"`
+	Name       string `json:"name"`
+	UpdateTime int64  `json:"update_time"`
+	Size       int64  `json:"size"`
+	Ext        string `json:"ext"`
+}
+
+func (driver Shandian) FormatFile(file *File) *model.File {
+	t := time.Unix(file.UpdateTime, 0)
+	f := &model.File{
+		Id:        strconv.FormatInt(file.Id, 10),
+		Name:      file.Name,
+		Size:      file.Size,
+		Driver:    driver.Config().Name,
+		UpdatedAt: &t,
+	}
+	if file.Type == 1 {
+		f.Type = conf.FOLDER
+	} else {
+		f.Type = utils.GetFileType(file.Ext)
+		if file.Ext != "" {
+			f.Name += "." + file.Ext
+		}
+	}
+	return f
+}
+
+func (driver Shandian) Post(url string, data map[string]interface{}, resp interface{}, account *model.Account) ([]byte, error) {
+	req := base.RestyClient.R()
+	req.SetHeader("Accept", "application/json")
+	data["token"] = account.AccessToken
+	req.SetBody(data)
+	var e Resp
+	if resp != nil {
+		req.SetResult(resp)
+	} else {
+		req.SetResult(&e)
+	}
+	req.SetError(&e)
+	res, err := req.Post(url)
+	if err != nil {
+		return nil, err
+	}
+	log.Debug(res.String())
+	if e.Code != 0 {
+		if e.Code == 10 {
+			err = driver.Login(account)
+			if err != nil {
+				return nil, err
+			}
+			return driver.Post(url, data, resp, account)
+		}
+		return nil, errors.New(e.Msg)
+	}
+	return res.Body(), nil
+}
+
+type FilesResp struct {
+	Resp
+	Data []File `json:"data"`
+}
+
+func (driver Shandian) GetFiles(id string, account *model.Account) ([]File, error) {
+	// TODO page not wok
+	//res := make([]File, 0)
+	page := 1
+	//for {
+	data := map[string]interface{}{
+		"id":        id,
+		"page":      page,
+		"page_size": 100,
+	}
+	var resp FilesResp
+	_, err := driver.Post("https://shandianpan.com/api/pan", data, &resp, account)
+	if err != nil {
+		return nil, err
+	}
+	//res = append(res, resp.Data...)
+	//	if len(resp.Data) == 0 {
+	//		break
+	//	}
+	//}
+	//return res, nil
+	return resp.Data, nil
+}
+
+type UploadResp struct {
+	Resp
+	Data struct {
+		Accessid  string `json:"accessid"`
+		Policy    string `json:"policy"`
+		Expire    int    `json:"expire"`
+		Callback  string `json:"callback"`
+		Key       string `json:"key"`
+		Host      string `json:"host"`
+		Signature string `json:"signature"`
+	} `json:"data"`
+}
+
+func init() {
+	base.RegisterDriver(&Shandian{})
+}