Pocketmine-Cache/request/request.go

package request

import (
	"errors"
	"fmt"
	"io"
	"net/http"
	"os"
)

var (
	ErrNoUrl        = errors.New("no url informed")
	ErrPageNotExist = errors.New("page not exists")
)

var DefaultHeader = http.Header{
	"Accept": {"text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7"},
	// "Accept-Encoding":           {"gzip, deflate"},
	"Accept-Language":           {"en-US;q=0.9,en;q=0.8"},
	"Sec-Ch-Ua":                 {`"Google Chrome";v="123", "Not:A-Brand";v="8", "Chromium";v="123\"`},
	"Sec-Ch-Ua-Mobile":          {"?0"},
	"Sec-Ch-Ua-Platform":        {`"Windows"`},
	"Sec-Fetch-Dest":            {"document"},
	"Sec-Fetch-Mode":            {"navigate"},
	"Sec-Fetch-Site":            {"none"},
	"Sec-Fetch-User":            {"?1"},
	"Upgrade-Insecure-Requests": {"1"},
	"User-Agent":                {"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36"},
}

type RequestOptions struct {
	Url       string      // Request url
	Method    string      // Request Method
	Body      io.Reader   // Body Reader
	Headers   http.Header // Extra HTTP Headers
	HttpError bool        // Return if status code is equal or then 300
}

// Make custom request and return request, response and error if exist
func Request(opt RequestOptions) (http.Response, error) {
	if len(opt.Url) == 0 {
		return http.Response{}, ErrNoUrl
	} else if len(opt.Method) == 0 {
		opt.Method = "GET"
	}

	// Create request
	req, err := http.NewRequest(opt.Method, opt.Url, opt.Body)
	if err != nil {
		return http.Response{}, err
	}

	// Project headers
	for key, value := range DefaultHeader {
		req.Header[key] = value
	}

	// Set headers
	for key, value := range opt.Headers {
		req.Header[key] = value
	}

	// Create response from request
	client := &http.Client{}
	res, err := client.Do(req)

	if opt.HttpError && res.StatusCode >= 300 {
		if res.StatusCode == 404 {
			err = ErrPageNotExist
		} else {
			err = fmt.Errorf("response non < 299, code %d, url: %q", res.StatusCode, opt.Url)
		}
	}

	// User tratement
	return *res, err
}

func SaveFile(filePath string, opt RequestOptions) (http.Response, error) {
	res, err := Request(opt)
	if err != nil {
		return res, err
	}

	defer res.Body.Close()
	file, err := os.Create(filePath)
	if err != nil {
		return res, err
	}

	defer file.Close()
	_, err = io.Copy(file, res.Body)
	return res, err
}