mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
9a072adff3
This package doesn't really seem to do anything of real interest. Removing it and replacing with a few helper functions. Most of this was maintaining a fork of ctxhttp to support a mock that was unnecessary. We could probably do with a further refactor of the client interface. There is a lot of confusion of between transport, http layer and application layer that makes for some awkward code. This change improves the situation to the point where no breaking changes are introduced. Signed-off-by: Stephen J Day <stephen.day@docker.com>
51 lines
1.5 KiB
Go
51 lines
1.5 KiB
Go
package client
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"errors"
|
|
"net/http"
|
|
)
|
|
|
|
var errTLSConfigUnavailable = errors.New("TLSConfig unavailable")
|
|
|
|
// transportFunc allows us to inject a mock transport for testing. We define it
|
|
// here so we can detect the tlsconfig and return nil for only this type.
|
|
type transportFunc func(*http.Request) (*http.Response, error)
|
|
|
|
func (tf transportFunc) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
return tf(req)
|
|
}
|
|
|
|
// resolveTLSConfig attempts to resolve the tls configuration from the
|
|
// RoundTripper.
|
|
func resolveTLSConfig(transport http.RoundTripper) (*tls.Config, error) {
|
|
switch tr := transport.(type) {
|
|
case *http.Transport:
|
|
return tr.TLSClientConfig, nil
|
|
case transportFunc:
|
|
return nil, nil // detect this type for testing.
|
|
default:
|
|
return nil, errTLSConfigUnavailable
|
|
}
|
|
}
|
|
|
|
// resolveScheme detects a tls config on the transport and returns the
|
|
// appropriate http scheme.
|
|
//
|
|
// TODO(stevvooe): This isn't really the right way to write clients in Go.
|
|
// `NewClient` should probably only take an `*http.Client` and work from there.
|
|
// Unfortunately, the model of having a host-ish/url-thingy as the connection
|
|
// string has us confusing protocol and transport layers. We continue doing
|
|
// this to avoid breaking existing clients but this should be addressed.
|
|
func resolveScheme(transport http.RoundTripper) (string, error) {
|
|
c, err := resolveTLSConfig(transport)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if c != nil {
|
|
return "https", nil
|
|
}
|
|
|
|
return "http", nil
|
|
}
|