alist/drivers/chaoxing/util.go

package template

import (
	"encoding/json"
	"fmt"
	"github.com/Xhofe/alist/conf"
	"github.com/Xhofe/alist/drivers/base"
	"github.com/Xhofe/alist/model"
	"github.com/Xhofe/alist/utils"
	"github.com/bitly/go-simplejson"
	"github.com/go-resty/resty/v2"
	"regexp"
	"strconv"
	"strings"
	"time"
)

// write util func here, such as cal sign

var chaoxingClient = resty.New()

var form_login_fmt = "fid=-1&uname=%s&password=%s&t=true&forbidotherlogin=0&validate=&doubleFactorLogin=0"

var api_list_root = "https://pan-yz.chaoxing.com/opt/listres?page=1&size=%d&enc=%s"
var api_list_file = "https://pan-yz.chaoxing.com/opt/listres?puid=%s&shareid=%s&parentId=%s&page=1&size=%d&enc=%s"
var api_list_shared_root = "https://pan-yz.chaoxing.com/opt/listres?puid=0&shareid=-1&parentId=0&page=1&size=%d&enc=%s"

var reg_enc_fmt = regexp.MustCompile("enc[ ]*=\"(.*)\"")

func (driver ChaoxingDrive) Login(account *model.Account) error {
	url := "https://passport2.chaoxing.com/fanyalogin"
	var resp base.Json
	var err Resp

	req_body := fmt.Sprintf(form_login_fmt, account.Username, account.Password)

	loginReq, e := chaoxingClient.R().SetBody(req_body).
		SetResult(&resp).SetError(&err).
		SetHeader("Content-Length", strconv.FormatInt(int64(len(req_body)), 10)).
		SetHeader("Content-Type", "application/x-www-form-urlencoded; charset=UTF-8").
		SetHeader("Host", "passport2.chaoxing.com").
		SetHeader("User-Agent", " Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/95.0.4638.69 Safari/537.36").
		SetHeader("X-Requested-With", "XMLHttpRequest").
		Post(url)

	if e != nil {
		return e
	}

	account.AccessToken = ""
	for _, cookie := range loginReq.Cookies() {
		//route=9d169c0aea4b7c89fa0d073417b5645f;
		account.AccessToken += fmt.Sprintf("%s=%s; ", cookie.Name, cookie.Value)
	}

	return nil
}

func (driver ChaoxingDrive) GetEnc(account *model.Account) error {
	url := "https://pan-yz.chaoxing.com/"

	encReq, e := chaoxingClient.R().SetHeader("Cookie", account.AccessToken).Get(url)
	if e != nil {
		return e
	}

	//直接读取响应
	sresp := string(encReq.Body())
	submatch := reg_enc_fmt.FindAllStringSubmatch(sresp, 1)
	//第一次获取失败，可能是未登录
	if len(submatch) == 0 {
		e = driver.Login(account)
		if e != nil {
			return e
		}
		encReq, e = chaoxingClient.R().SetHeader("Cookie", account.AccessToken).Get(url)
		if e != nil {
			return e
		}
		sresp = string(encReq.Body())
		submatch = reg_enc_fmt.FindAllStringSubmatch(sresp, 1)
		if len(submatch) == 0 {
			account.Status = "failed"
			return fmt.Errorf("登录失败，服务器返回信息：%s", sresp)
		}
	}
	enc := submatch[0][1]
	account.AccessSecret = enc
	account.Status = "work"
	return nil
}

func (driver ChaoxingDrive) ListFile(folder_id string, account *model.Account) ([]model.File, error) {
	var url string
	//按规则解析 id 号
	folder_id_info := strings.Split(folder_id, "_")
	if len(folder_id_info) == 3 {
		folder_id = folder_id_info[0]
		folder_puid := folder_id_info[1]
		folder_shareid := folder_id_info[2]
		if folder_id == "0" {
			//访问“共享给我的文件夹”
			url = fmt.Sprintf(api_list_shared_root, account.Limit, account.AccessSecret)
		} else {
			//访问其他目录
			url = fmt.Sprintf(api_list_file, folder_puid, folder_shareid, folder_id, account.Limit, account.AccessSecret)
		}
	} else {
		//id无法解析为三段，应当是访问根目录（此时为 ""）
		url = fmt.Sprintf(api_list_root, account.Limit, account.AccessSecret)
	}

	listFileReq, e := chaoxingClient.R().SetHeader("Cookie", account.AccessToken).Post(url)
	resp, e := simplejson.NewJson(listFileReq.Body())
	if e != nil || resp == nil {
		return nil, e
	}

	files := make([]model.File, 0)
	array, _ := resp.Get("list").Array()

	for _, file := range array {
		var f = model.File{}
		file_ := file.(map[string]interface{})
		//f.Id = file_["id"].(string)
		f.Id = fmt.Sprintf("%s_%s_%s", file_["id"].(string), file_["puid"].(json.Number).String(), file_["shareid"].(json.Number).String())
		f.Name = file_["name"].(string)
		f_server_type, _ := file_["type"].(json.Number).Int64()
		if f_server_type != TYPE_CX_SHARED_ROOT {
			f.Size, _ = file_["filesize"].(json.Number).Int64()
		}
		// 为文件分配类型
		switch f_server_type {
		case TYPE_CX_FILE:
			{
				f.Type = utils.GetFileType(file_["suffix"].(string))
			}
		case TYPE_CX_FOLDER:
			f.Type = conf.FOLDER
		case TYPE_CX_SHARED_ROOT:
			f.Type = conf.FOLDER
		}
		modifyDate, e := time.Parse("2006-01-02 15:04:05", file_["modifyDate"].(string))
		if e == nil {
			f.UpdatedAt = &modifyDate
		}
		f.Thumbnail = file_["thumbnail"].(string)
		files = append(files, f)
	}
	return files, nil
}