51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package httpclient
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/photoprism/photoprism/internal/event"
|
|
)
|
|
|
|
var log = event.Log
|
|
|
|
func Client(debug bool) *http.Client {
|
|
if debug {
|
|
return &http.Client{
|
|
Transport: LoggingRoundTripper{http.DefaultTransport},
|
|
}
|
|
}
|
|
return http.DefaultClient
|
|
}
|
|
|
|
// This type implements the http.RoundTripper interface
|
|
type LoggingRoundTripper struct {
|
|
proxy http.RoundTripper
|
|
}
|
|
|
|
func (lrt LoggingRoundTripper) RoundTrip(req *http.Request) (res *http.Response, e error) {
|
|
// Do "before sending requests" actions here.
|
|
log.Debugf("Sending request to %v\n", req.URL.String())
|
|
|
|
// Send the request, get the response (or the error)
|
|
res, e = lrt.proxy.RoundTrip(req)
|
|
|
|
// Handle the result.
|
|
if e != nil {
|
|
log.Errorf("Error: %v", e)
|
|
} else {
|
|
log.Debugf("Received %v response\n", res.Status)
|
|
|
|
// Copy body into buffer for logging
|
|
buf := new(bytes.Buffer)
|
|
_, err := io.Copy(buf, res.Body)
|
|
if err != nil {
|
|
log.Errorf("Error: %v", err)
|
|
}
|
|
log.Debugf("Header: %s\n", res.Header)
|
|
log.Debugf("Reponse Body: %s\n", buf.String())
|
|
res.Body = io.NopCloser(buf)
|
|
}
|
|
return
|
|
}
|