package grequest

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

// RequestPost 函数用于发送POST请求
func Post(options RequestPostOptions) (Response, error) {
	var res Response

	if options.URL == "" {
		return res, errors.New("必须提供url")
	}

	// 设置默认的headers
	defaultHeaders := map[string]string{"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.149 Safari/537.36"}

	// 使用默认值或提供的值
	headers := defaultHeaders
	if options.Headers != nil {
		headers = options.Headers
	}

	client := &http.Client{}
	req, err := http.NewRequest("POST", options.URL, nil)
	if err != nil {
		return res, err
	}

	// 设置请求头
	for key, value := range headers {
		req.Header.Set(key, value)
	}

	// 如果提供了数据，则将其添加到请求体中
	if options.Data != nil {
		dataBytes, err := json.Marshal(options.Data)
		if err != nil {
			return res, err
		}
		req.Body = io.NopCloser(bytes.NewBuffer(dataBytes))
		req.Header.Set("Content-Type", "application/json")
	}

	resp, err := client.Do(req)
	if err != nil {
		res.Msg = err.Error()
		return res, nil
	}
	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		res.Msg = err.Error()
		return res, nil
	}

	res.Code = resp.StatusCode
	if err := json.Unmarshal(body, &res.Content); err != nil {
		res.Content = string(body)
	}

	return res, nil
}