caitong93 · December 5, 2018 02:55 · Dec 5, 2018
diff --git a/client.go b/client.go
@@ -0,0 +1,40 @@
+var defaultTransport = &http.Transport{
+	Proxy: http.ProxyFromEnvironment,
+	Dial: (&net.Dialer{
+		Timeout:   30 * time.Second,
+		KeepAlive: 30 * time.Second,
+	}).Dial,
+	TLSHandshakeTimeout: 10 * time.Second,
+	TLSClientConfig: &tls.Config{
+		InsecureSkipVerify: true,
+	},
+}
+
+type AuthTransport struct {
+	*http.Transport
+	Token    string
+	Username string
+	Password string
+}
+
+func NewAuthTransport(username, password, token string) *AuthTransport {
+	return &AuthTransport{
+		Transport: defaultTransport,
+		Token:     token,
+		Username:  username,
+		Password:  password,
+	}
+}
+
+func (t *AuthTransport) RoundTrip(req *http.Request) (*http.Response, error) {
+	if len(t.Token) != 0 {
+		token := oauth2.Token{
+			TokenType:   "Bearer",
+			AccessToken: t.Token,
+		}
+		token.SetAuthHeader(req)
+	} else {
+		req.SetBasicAuth(t.Username, t.Password)
+	}
+	return t.Transport.RoundTrip(req)
+}