http.go

package godingtalk

import (
	"bytes"
	"encoding/json"
	"errors"
	"io/ioutil"
	"net/http"
	"net/url"
)

var (
	ErrEmptyParams = errors.New("params is require")
)

type RequestData map[string]interface{}

func (rd RequestData) Set(key string, value interface{}) {
	switch value := value.(type) {
	case []interface{}:
		rd[key] = value
		return
	}
	rd[key] = value
}

func httpRequest(cli *DingtalkClient, path string, params url.Values, reqData interface{}, respData Unmarshalable, flag bool) error {
	if params == nil {
		return ErrEmptyParams
	}
	if params.Get("appkey") == "" || params.Get("appsecret") == "" {
		params.Set("appkey", cli.AppKey)
		params.Set("appsecret", cli.AppSecret)
	}
	if flag && checkExpireTime(cli.AccessToken.ExpiresTime) {
		if err := cli.RefreshToken(); err != nil {
			return err
		}
		params.Set("access_token", cli.AccessToken.Token)
	}
	uri := &url.URL{
		Scheme:   "https",
		Host:     cli.BaseURL,
		Path:     path,
		RawQuery: params.Encode(),
	}
	data, err := request(cli.Client, uri.String(), reqData)
	if err != nil {
		return err
	}
	if err := json.Unmarshal(data, respData); err != nil {
		return err
	}
	return respData.checkErr()
}

// httpRequestWithStd 使用原生http实现
// 以reqData为判断, nil == Get, !nil == Post
func (d *DingtalkClient) httpRequestWithStd(path string, params url.Values, reqData interface{}, respData Unmarshalable) error {
	return httpRequest(d, path, params, reqData, respData, true)
}

func (d *DingtalkClient) getNewAccessToken(path string, params url.Values, reqData interface{}, respData Unmarshalable) error {
	return httpRequest(d, path, params, reqData, respData, false)
}

func request(client *http.Client, uri string, reqData interface{}) ([]byte, error) {
	var req *http.Request
	var err error
	if reqData != nil {
		// POST
		rd, err := json.Marshal(reqData)
		if err != nil {
			return nil, err
		}
		req, err = http.NewRequest(http.MethodPost, uri, bytes.NewReader(rd))
		if err != nil {
			return nil, err
		}
		req.Header.Set("Content-Type", "application/json")
	} else {
		// GET
		req, err = http.NewRequest(http.MethodGet, uri, nil)
		if err != nil {
			return nil, err
		}
	}
	response, err := client.Do(req)
	if err != nil {
		return nil, err
	}
	defer response.Body.Close()
	data, err := ioutil.ReadAll(response.Body)
	if err != nil {
		return nil, err
	}
	return data, nil
}