package httpx

import (
	"bytes"
	"encoding/json"
	"errors"
	"io"
	"net/http"
	"net/url"
)

type HttpX struct {
	resp *http.Response
	err  error
}

func NewHttpX() *HttpX {
	return &HttpX{}
}

type Condition struct {
	Data   interface{}
	Param  url.Values
	Header http.Header
}

func (r *HttpX) Request(method string, _url string, c Condition) *HttpX {
	client := &http.Client{}
	// params
	if c.Param != nil {
		URL, err := url.Parse(_url)
		if err != nil {
			r.err = err
			return r
		}
		URL.RawQuery = c.Param.Encode()
		_url = URL.String()
	}
	// data
	var reader io.Reader = nil
	if c.Data != nil {
		bytesData, err := json.Marshal(c.Data)
		if err != nil {
			r.err = err
			return r
		}
		reader = bytes.NewReader(bytesData)
	}
	req, err := http.NewRequest(method, _url, reader)
	if err != nil {
		r.err = err
		return r
	}
	// header
	req.Header = c.Header
	resp, err := client.Do(req)
	if err != nil {
		r.err = err
		return r
	}
	r.resp = resp
	return r
}

func (r *HttpX) Get(url string, c Condition) *HttpX {
	return r.Request("GET", url, c)
}

func (r *HttpX) Post(url string, c Condition) *HttpX {
	return r.Request("POST", url, c)
}

func (r *HttpX) Then(accept func(data []byte)) *HttpX {
	if r.resp != nil {
		defer r.resp.Body.Close()
	} else {
		r.err = errors.New("remote query failed")
		return r
	}
	bytes, err := io.ReadAll(r.resp.Body)
	if err != nil {
		r.err = err
		return r
	}
	accept(bytes)
	return r
}

func (r *HttpX) Catch(accept func(err error)) *HttpX {
	if r.err != nil {
		accept(r.err)
	}
	return r
}