helm/dm/transport.go

package dm

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

type debugTransport struct {
	// Writer is the logging destination
	Writer io.Writer

	http.RoundTripper
}

func NewDebugTransport(rt http.RoundTripper) http.RoundTripper {
	return debugTransport{
		RoundTripper: rt,
		Writer:       os.Stderr,
	}
}

func (tr debugTransport) CancelRequest(req *http.Request) {
	type canceler interface {
		CancelRequest(*http.Request)
	}
	if cr, ok := tr.transport().(canceler); ok {
		cr.CancelRequest(req)
	}
}

func (tr debugTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	tr.logRequest(req)
	resp, err := tr.transport().RoundTrip(req)
	if err != nil {
		return nil, err
	}
	tr.logResponse(resp)
	return resp, err
}

func (tr debugTransport) transport() http.RoundTripper {
	if tr.RoundTripper != nil {
		return tr.RoundTripper
	}
	return http.DefaultTransport
}

func (tr debugTransport) logRequest(req *http.Request) {
	dump, err := httputil.DumpRequestOut(req, true)
	if err != nil {
		fmt.Fprintf(tr.Writer, "%s: %s\n", "could not dump request", err)
	}
	fmt.Fprint(tr.Writer, string(dump))
}

func (tr debugTransport) logResponse(resp *http.Response) {
	dump, err := httputil.DumpResponse(resp, true)
	if err != nil {
		fmt.Fprintf(tr.Writer, "%s: %s\n", "could not dump response", err)
	}
	fmt.Fprint(tr.Writer, string(dump))
}