package fetcher

import (
	"bufio"
	"bytes"
	"errors"
	"io"
	"io/ioutil"
	"log"
	"net/http"
	"time"

	"golang.org/x/net/html/charset"
	"golang.org/x/text/encoding"
	"golang.org/x/text/encoding/unicode"
)

// Get get 请求
func Get(url string) ([]byte, error) {
	client := &http.Client{}
	req, _ := http.NewRequest("GET", url, nil)

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

	res, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return nil, err
	}

	return res, nil
}

// PostJson application/json post 请求
func PostJson(url, parameter string, timeout int) ([]byte, error) {
	client := &http.Client{Timeout: time.Second * time.Duration(timeout)}
	byteParameter := bytes.NewBuffer([]byte(parameter))
	request, _ := http.NewRequest("POST", url, byteParameter)
	request.Header.Set("Content-type", "application/json")
	response, _ := client.Do(request)
	if response.StatusCode != 200 {
		return nil, errors.New("网络请求失败")
	}
	all, err := io.ReadAll(response.Body)
	if err != nil {
		return nil, errors.New("读取网络内容失败")
	}
	return all, nil
}

// PostString application/x-www-form-urlencoded 请求
func PostString(url, parameter string, timeout int) ([]byte, error) {
	client := &http.Client{Timeout: time.Second * time.Duration(timeout)}
	byteParameter := bytes.NewBuffer([]byte(parameter))
	request, _ := http.NewRequest("POST", url, byteParameter)
	request.Header.Set("Content-type", "application/x-www-form-urlencoded")
	response, _ := client.Do(request)
	if response.StatusCode != 200 {
		return nil, errors.New("网络请求失败")
	}
	all, err := ioutil.ReadAll(response.Body)
	if err != nil {
		return nil, errors.New("读取网络内容失败")
	}
	return all, nil
}

func determinEncoding(r *bufio.Reader) encoding.Encoding {
	bytes, err := r.Peek(1024)
	if err != nil && err != io.EOF {
		log.Printf("Fetcher error: %v", err)
		return unicode.UTF8
	}

	e, _, _ := charset.DetermineEncoding(bytes, "")
	return e
}