package alidrive

import (
	"bytes"
	"encoding/json"
	"fmt"
	"github.com/Xhofe/alist/conf"
	log "github.com/sirupsen/logrus"
	"io/ioutil"
	"net/http"
	"strings"
	"time"
)

// convert body to json
func BodyToJson(url string, req interface{}, resp RespHandle, drive *conf.Drive) error {
	if body, err := DoPost(url, req, drive.AccessToken); err != nil {
		log.Errorf("doPost出错:%s", err.Error())
		return err
	} else {
		if err = json.Unmarshal(body, &resp); err != nil {
			log.Errorf("解析json[%s]出错:%s", string(body), err.Error())
			return err
		}
	}
	if resp.IsAvailable() {
		return nil
	}
	if resp.GetCode() == conf.AccessTokenInvalid {
		resp.SetCode("")
		if RefreshToken(drive) {
			return BodyToJson(url, req, resp, drive)
		}
	}
	return fmt.Errorf(resp.GetMessage())
}

// do post request
func DoPost(url string, request interface{}, auth string) (body []byte, err error) {
	var (
		resp *http.Response
	)
	requestBody := new(bytes.Buffer)
	err = json.NewEncoder(requestBody).Encode(request)
	if err != nil {
		log.Errorf("创建requestBody出错:%s", err.Error())
	}
	req, err := http.NewRequest("POST", url, requestBody)
	log.Debugf("do_post_req:%+v", req)
	if err != nil {
		log.Errorf("创建request出错:%s", err.Error())
		return
	}
	if auth != "" {
		req.Header.Set("authorization", conf.Bearer+auth)
	}
	req.Header.Add("content-type", "application/json")
	req.Header.Add("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36")
	req.Header.Add("origin", "https://aliyundrive.com")
	req.Header.Add("accept", "*/*")
	req.Header.Add("Accept-Language", "zh-CN,zh;q=0.8,en-US;q=0.5,en;q=0.3")
	req.Header.Add("Connection", "keep-alive")

	for retryCount := 3; retryCount >= 0; retryCount-- {
		if resp, err = conf.Client.Do(req); err != nil && strings.Contains(err.Error(), "timeout") {
			<-time.After(time.Second)
		} else {
			break
		}
	}
	if err != nil {
		log.Errorf("请求阿里云盘api时出错:%s", err.Error())
		return
	}
	if body, err = ioutil.ReadAll(resp.Body); err != nil {
		log.Errorf("读取api返回内容失败")
	}
	log.Debugf("请求返回信息:%s", string(body))
	return
}