adapter.go

package gotcha

import (
	"net/http"
	"sync"
)

type Adapter interface {
	// DoRequest is a custom function that will be used by gotcha to make the actual request.
	DoRequest(options *Options) (*Response, error)
}

// RequestAdapter is a default implementation of Adapter.
// Gotcha will use this adapter when no other is specified.
type RequestAdapter struct {
	// RoundTripper is a http.RoundTripper that will be used to do the request.
	//
	// Defaults to http.DefaultTransport.
	RoundTripper http.RoundTripper

	// Request is a function that builds the http.Request to send.
	//
	// Defaults to a function that derives the Request from the specified Options.
	Request func(*Options) *http.Request

	mu sync.Mutex
}

// Initializes adapter defaults.
func (ra *RequestAdapter) init(options *Options) {
	ra.mu.Lock()

	if ra.Request == nil {
		ra.Request = func(o *Options) *http.Request {
			return &http.Request{
				Method: o.Method,
				URL:    o.FullUrl,
				Header: o.Headers,
				Body:   o.Body,
			}
		}
	}

	if ra.RoundTripper == nil {
		var defaultTransport = http.DefaultTransport.(*http.Transport)
		if p := options.Proxy; p != nil {
			defaultTransport.Proxy = http.ProxyURL(p)
		}
		ra.RoundTripper = defaultTransport
	}

	ra.mu.Unlock()
}

func (ra *RequestAdapter) DoRequest(options *Options) (*Response, error) {
	ra.init(options)

	req := ra.Request(options)

	if options.CookieJar != nil {
		for _, cookie := range options.CookieJar.Cookies(options.FullUrl) {
			req.AddCookie(cookie)
		}
	}

	res, err := ra.RoundTripper.RoundTrip(req)
	if err != nil {
		return nil, err
	}

	if options.CookieJar != nil {
		if rc := res.Cookies(); len(rc) > 0 {
			options.CookieJar.SetCookies(options.FullUrl, rc)
		}
	}

	return &Response{res, options.UnmarshalJson}, nil
}

// mockAdapter is only used for testing Adapter.
type mockAdapter struct {
	OnCalledDoRequest func(*Options) *Response
}

func (ma *mockAdapter) DoRequest(options *Options) (*Response, error) {
	return ma.OnCalledDoRequest(options), nil
}