resty_client.go

package sdk

import (
	"bytes"
	"github.com/go-resty/resty"
	"io"
	"net"
	"net/http"
	"os"
	"time"
)

//var (
//	//token = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhcGlLZXkiOiJaT0dKNVI0QTZPU1JIWUlOTUlMQSIsImV4cCI6MTY4NDgzNDUxNSwiaWRlbnRpdHkiOnsiSWQiOjYsIk5pY2tOYW1lIjoidXNlcjEiLCJQcm9maWxlSW1hZ2UiOiIiLCJXYWxsZXRBZGRyIjoidXNlcjEiLCJ0ZXJtaW5hbCI6InBjL3NkayJ9LCJvcmlnX2lhdCI6MTY4MjI0MjUyNH0.w0lQ-RGufy5j3yMfbnL94hOvLeeCSccZbSIwd6kXvbI"
//	token     = "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhcGlLZXkiOiJaT0dKNVI0QTZPU1JIWUlOTUlMQSIsImV4cCI6MTY4NDgzNDUxNSwiaWRlbnRpdHkiOnsiSWQiOjYsIk5pY2tOYW1lIjoidXNlcjEiLCJQcm9maWxlSW1hZ2UiOiIiLCJXYWxsZXRBZGRyIjoidXNlcjEiLCJ0ZXJtaW5hbCI6InBjL3NkayJ9LCJvcmlnX2lhdCI6MTY4MjI0MjUyNH0.w0lQ-RGufy5j3yMfbnL94hOvLeeCSccZbSIwd6kXvbI"
//	userAgent = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36"
//	headers   = map[string]string{
//		"Accept":        "application/json",
//		"Authorization": token,
//		"User-Agent":    userAgent,
//	}
//)

type RestyClient struct {
	Config *Configuration
	//logger *golog.Logger
}

func (r *RestyClient) getHeaders() *map[string]string {
	apiToken := r.Config.ChainStorageApiToken
	httpRequestUserAgent := r.Config.HttpRequestUserAgent

	return &map[string]string{
		"Accept":        "application/json",
		"Authorization": apiToken,
		"User-Agent":    httpRequestUserAgent,
	}
}

func (r *RestyClient) RestyGet(url string) (httpStatus int, body []byte, err error) {
	resp, err := resty.
		SetTimeout(time.Duration(60) * time.Second).
		SetHeaders(*r.getHeaders()).
		R().Get(url)
	if err != nil {
		if perr, ok := err.(net.Error); ok && perr.Timeout() {
			return http.StatusGatewayTimeout, nil, err
		} else {
			return http.StatusServiceUnavailable, nil, err
		}
	}

	defer resp.RawResponse.Body.Close()
	body = resp.Body()
	return resp.StatusCode(), body, nil
}

func (r *RestyClient) RestyPut(url string, data interface{}) (httpStatus int, body []byte, err error) {
	resp, err := resty.
		SetTimeout(time.Duration(60) * time.Second).
		SetHeaders(*r.getHeaders()).
		R().SetBody(data).Put(url)
	if err != nil {
		if perr, ok := err.(net.Error); ok && perr.Timeout() {
			return http.StatusGatewayTimeout, nil, err
		} else {
			return http.StatusServiceUnavailable, nil, err

		}
	}
	defer resp.RawResponse.Body.Close()
	body = resp.Body()
	return resp.StatusCode(), body, nil
}

func (r *RestyClient) RestyPost(url string, data interface{}) (httpStatus int, body []byte, err error) {
	resp, err := resty.
		SetTimeout(time.Duration(60) * time.Second).
		SetHeaders(*r.getHeaders()).
		R().SetBody(data).Post(url)
	if err != nil {
		if perr, ok := err.(net.Error); ok && perr.Timeout() {
			return http.StatusGatewayTimeout, nil, err
		} else {
			return http.StatusServiceUnavailable, nil, err

		}
	}
	defer resp.RawResponse.Body.Close()
	body = resp.Body()
	return resp.StatusCode(), body, nil
}

func (r *RestyClient) RestyPostForm(filename string, filePath string, formData map[string]string, url string) (httpStatus int, body []byte, err error) {
	fileb, err := os.ReadFile(filePath)
	if err != nil {
		return http.StatusInternalServerError, nil, err
	}

	resp, err := resty.
		SetTimeout(time.Duration(60)*time.Second).
		SetHeaders(*r.getHeaders()).
		R().SetFileReader("file", filename, bytes.NewReader(fileb)).
		SetFormData(formData).
		Post(url)

	if err != nil {
		if perr, ok := err.(net.Error); ok && perr.Timeout() {
			return http.StatusGatewayTimeout, nil, err
		} else {
			return http.StatusServiceUnavailable, nil, err

		}
	}
	defer resp.RawResponse.Body.Close()
	body = resp.Body()

	return resp.StatusCode(), body, err
}

func (r *RestyClient) RestyDelete(url string, data interface{}) (httpStatus int, body []byte, err error) {
	resp, err := resty.
		SetTimeout(time.Duration(60) * time.Second).
		SetHeaders(*r.getHeaders()).
		R().
		SetBody(data).
		Delete(url)
	if err != nil {
		if perr, ok := err.(net.Error); ok && perr.Timeout() {
			return http.StatusGatewayTimeout, nil, err
		} else {
			return http.StatusServiceUnavailable, nil, err

		}
	}

	defer resp.RawResponse.Body.Close()

	body = resp.Body()

	return resp.StatusCode(), body, nil
}

type ExtReader func(reader *io.Reader) io.Reader

func (r *RestyClient) RestyPostFormExt(filename string, filePath string, formData map[string]string, url string, extReader io.Reader) (httpStatus int, body []byte, err error) {

	var reader io.Reader
	if extReader != nil {
		//reader = bytes.NewReader(fileb)
		reader = extReader
	} else {
		fileb, err := os.ReadFile(filePath)
		if err != nil {
			return http.StatusInternalServerError, nil, err
		}
		reader = bytes.NewReader(fileb)
	}

	resp, err := resty.
		SetTimeout(time.Duration(60)*time.Second).
		SetHeaders(*r.getHeaders()).
		R().SetFileReader("file", filename, reader).
		SetFormData(formData).
		Post(url)

	if err != nil {
		if perr, ok := err.(net.Error); ok && perr.Timeout() {
			return http.StatusGatewayTimeout, nil, err
		} else {
			return http.StatusServiceUnavailable, nil, err

		}
	}
	defer resp.RawResponse.Body.Close()
	body = resp.Body()

	return resp.StatusCode(), body, err
}