package httptransport

import (
	"errors"
	"net/http"
	"strings"

	"gitee.com/aly1a/boat-go/module/logger"
)

type HtpCli struct {
	HtpClient *http.Client
}

var httpClient *http.Client

// 创建FastHttpClient
func New() *HtpCli {
	if httpClient != nil {
		return &HtpCli{
			HtpClient: httpClient,
		}
	}

	httpClient = CreateHttpClient()

	return &HtpCli{
		HtpClient: httpClient,
	}
}

func Client() *http.Client {
	return httpClient
}

// 一个post请求
func (f HtpCli) Post(data string, url string, headers map[string]string) (*http.Response, error) {
	if f.HtpClient == nil {
		return nil, errors.New("未初始化【http.Transport】")
	}

	// 创建请求
	body := strings.NewReader(data)
	req, err := http.NewRequest("POST", url, body)
	if err != nil {
		return nil, err
	}
	// 设置请求头
	req.Header.Set("Content-Type", "application/json")
	for k, v := range headers {
		req.Header.Set(k, v)
	}
	//
	return f.HtpClient.Do(req)
}

// 一个get请求
func (f HtpCli) Get(url string, headers map[string]string) (*http.Response, error) {
	if f.HtpClient == nil {
		return nil, errors.New("未初始化【http.Transport】")
	}

	// 创建请求
	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
		return nil, err
	}

	for k, v := range headers {
		req.Header.Set(k, v)
	}
	//
	return f.HtpClient.Do(req)
}

// 关闭FastHttpClient
func Close() {
	if httpClient != nil {
		httpClient.CloseIdleConnections()
		logger.LogInfoln("关闭HttpClient成功")
	} else {
		logger.LogInfoln("未开启HttpClient")
	}
}