package util

import (
	"crypto/tls"
	"errors"
	"github.com/valyala/fasthttp"
	"gitlab.wodcloud.com/smart-operation/so-operation-api/src/common/conf"
	"go.uber.org/zap"
	"io"
	"net"
	"net/http"
	"strings"
	"sync"
	"time"
)

var lock sync.RWMutex

const (
	MediaTypeJSON          = "application/json"
	MediaTypeXML           = "application/xml"
	MediaTypeForm          = "application/x-www-form-urlencoded"
	MediaTypeMultipartForm = "multipart/form-data"
	MediaTypeOctetStream   = "application/octet-stream"
	MediaTypeHTML          = "text/html"
	MediaTypePlain         = "text/plain"
	MediaTypeJavascript    = "application/javascript"
	MediaTypeCSS           = "text/css"
	MediaTypeGIF           = "image/gif"
	MediaTypeJPEG          = "image/jpeg"
	MediaTypePNG           = "image/png"
	MediaTypeBMP           = "image/bmp"
	MediaTypeIcon          = "image/x-icon"
	MediaTypePDF           = "application/pdf"
	MediaTypeZip           = "application/zip"
	MediaTypeTAR           = "application/x-tar"
	MediaTypeGZ            = "application/x-gzip"
	MediaTypeProtobuf      = "application/protobuf"
	MediaTypeGRPC          = "application/grpc"
)

// Request 包装fasthttp的请求工具
/*
请求示例
1.GET请求
Request("https://httpbin.org/get", http.MethodGet, nil, map[string]string{"Cookie": "token=e69f5609-378f-4461-89d0-7508f2a06022"})

2.POST请求
Request("https://httpbin.org/post",
		http.MethodPost,
		body,
		map[string]string{"Content-Type": MediaTypeJSON})

3.PUT请求
Request("https://httpbin.org/put",
		http.MethodPut,
		body,
		map[string]string{
			"Content-Type": MediaTypeJSON,
			"Cookie":       "aweToken=3ab9f63f-b0b3-4935-80ec-405d76ac111d",
		})
*/
func Request(url string, method string, body []byte, headers map[string]string) (*fasthttp.Response, error) {
	req := fasthttp.AcquireRequest()
	defer fasthttp.ReleaseRequest(req)

	req.SetRequestURI(url)
	req.Header.SetMethod(method)

	// 设置默认请求头部
	req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/100.0.0.0 Safari/537.36")

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

	// 设置请求体
	if body != nil {
		req.SetBody(body)
	}

	// 创建一个忽略证书问题的客户端
	client := &fasthttp.Client{
		Dial: func(addr string) (net.Conn, error) {
			return net.DialTimeout("tcp", addr, 30*time.Second)
		},
		TLSConfig: &tls.Config{
			InsecureSkipVerify: true,
		},
	}

	resp := fasthttp.AcquireResponse()
	defer fasthttp.ReleaseResponse(resp)

	// 发送请求
	if err := client.Do(req, resp); err != nil {
		return nil, err
	}

	result := new(fasthttp.Response)
	resp.CopyTo(result)

	// 返回响应体和错误信息
	return result, nil
}

// HttpSend , http请求  GET/DELETE/POST/PUT
func HttpSend(sendType string, url string, body string, header map[string]string) (result *http.Response, err error) {
	conf.Logger.Debug("消息请求url", zap.String("url", url))
	conf.Logger.Debug("请求body", zap.Any("body", string(body)))
	client := &http.Client{}
	var reqest *http.Request
	if sendType == "GET" || sendType == "DELETE" {
		reqest, _ = http.NewRequest(sendType, url, nil)
	} else if sendType == "POST" || sendType == "PUT" {
		reqest, _ = http.NewRequest(sendType, url, strings.NewReader(body))
	}
	if header == nil {
		header = make(map[string]string)
	}
	header["Content-Type"] = "application/json"
	for k, v := range header {
		reqest.Header.Set(k, v)
	}
	response, err := client.Do(reqest)
	if err != nil {
		conf.Logger.Error("消息发送失败", zap.Error(err))
	}
	return response, err
}

func ProxySendRes(sendType string, url string, body string, header map[string]string) (result []byte, err error) {
	client := &http.Client{}
	var reqest *http.Request
	if sendType == "GET" || sendType == "DELETE" {
		reqest, _ = http.NewRequest(sendType, url, nil)
	} else if sendType == "POST" || sendType == "PUT" {
		reqest, _ = http.NewRequest(sendType, url, strings.NewReader(body))
	}
	http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
	lock.Lock()
	if header == nil {
		header = make(map[string]string)
	}
	reqest.Header.Set("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8")
	reqest.Header.Set("Accept-Language", "ja,zh-CN;q=0.8,zh;q=0.6")
	reqest.Header.Set("Connection", "keep-alive")
	for k, v := range header {
		reqest.Header.Set(k, v)
	}
	lock.Unlock()
	response, errc := client.Do(reqest)
	if errc != nil {
		return nil, errc
	} else {
		defer response.Body.Close()
		body, erra := io.ReadAll(response.Body)
		if response.StatusCode >= 400 {
			err = errors.New(string(body))
			return
		}
		return body, erra
	}
}