superagent.go

package pkg

import (
	"bytes"
	"encoding/json"
	"io/ioutil"
	"net/http"

	"github.com/ory-am/hydra/pkg/helper"
	"github.com/pkg/errors"
)

type SuperAgent struct {
	Client *http.Client
	URL    string
	Dry    bool
}

func NewSuperAgent(rawurl string) *SuperAgent {
	return &SuperAgent{
		URL:    rawurl,
		Client: http.DefaultClient,
	}
}

func (s *SuperAgent) DoDry(req *http.Request) error {
	return helper.DoDryRequest(s.Dry, req)
}

func (s *SuperAgent) Delete() error {
	req, err := http.NewRequest("DELETE", s.URL, nil)
	if err != nil {
		return errors.WithStack(err)
	}

	if err := s.DoDry(req); err != nil {
		return err
	}

	resp, err := s.Client.Do(req)
	if err != nil {
		return errors.WithStack(err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusNoContent {
		body, _ := ioutil.ReadAll(resp.Body)
		return errors.Errorf("Expected status code %d, got %d.\n%s\n", http.StatusNoContent, resp.StatusCode, body)
	}

	return nil
}

func (s *SuperAgent) Get(o interface{}) error {
	req, err := http.NewRequest("GET", s.URL, nil)
	if err != nil {
		return errors.WithStack(err)
	} else if o == nil {
		return errors.New("Can not pass nil")
	}

	if err := s.DoDry(req); err != nil {
		return err
	}

	resp, err := s.Client.Do(req)
	if err != nil {
		return errors.WithStack(err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		body, _ := ioutil.ReadAll(resp.Body)
		return errors.Errorf("Expected status code %d, got %d.\n%s\n", http.StatusOK, resp.StatusCode, body)
	} else if err := json.NewDecoder(resp.Body).Decode(o); err != nil {
		return errors.WithStack(err)
	}

	return nil
}

func (s *SuperAgent) Create(o interface{}) error {
	return s.send("POST", o, o)
}

func (s *SuperAgent) POST(in, out interface{}) error {
	return s.send("POST", in, out)
}

func (s *SuperAgent) Update(o interface{}) error {
	return s.send("PUT", o, o)
}

func (s *SuperAgent) send(method string, in interface{}, out interface{}) error {
	if s.Client == nil {
		s.Client = http.DefaultClient
	}

	data, err := json.Marshal(in)
	if err != nil {
		return errors.WithStack(err)
	}

	req, err := http.NewRequest(method, s.URL, bytes.NewReader(data))
	if err != nil {
		return errors.WithStack(err)
	}

	req.Header.Set("Content-Type", "application/json")
	if err := s.DoDry(req); err != nil {
		return err
	}

	resp, err := s.Client.Do(req)
	if err != nil {
		return errors.WithStack(err)
	}

	if resp.StatusCode == http.StatusNoContent {
		return nil
	}

	if resp.StatusCode < 200 || resp.StatusCode >= 300 {
		body, _ := ioutil.ReadAll(resp.Body)
		return errors.Errorf("Expected 2xx status code but got %d.\n%s", resp.StatusCode, body)
	} else if err := json.NewDecoder(resp.Body).Decode(out); err != nil {
		body, _ := ioutil.ReadAll(resp.Body)
		return errors.Errorf("%s: %s", err, body)
	}

	return nil
}