Commit d86a774a authored by Eric Chiang's avatar Eric Chiang Committed by GitHub

Merge pull request #670 from ericchiang/example-app-debug

cmd/example-app: add a --debug flag
parents ce703a7f 35d6423a
......@@ -11,6 +11,7 @@ import (
"log"
"net"
"net/http"
"net/http/httputil"
"net/url"
"os"
"strings"
......@@ -62,6 +63,31 @@ func httpClientForRootCAs(rootCAs string) (*http.Client, error) {
}, nil
}
type debugTransport struct {
t http.RoundTripper
}
func (d debugTransport) RoundTrip(req *http.Request) (*http.Response, error) {
reqDump, err := httputil.DumpRequest(req, true)
if err != nil {
return nil, err
}
log.Printf("%s", reqDump)
resp, err := d.t.RoundTrip(req)
if err != nil {
return nil, err
}
respDump, err := httputil.DumpResponse(resp, true)
if err != nil {
resp.Body.Close()
return nil, err
}
log.Printf("%s", respDump)
return resp, nil
}
func cmd() *cobra.Command {
var (
a app
......@@ -70,6 +96,7 @@ func cmd() *cobra.Command {
tlsCert string
tlsKey string
rootCAs string
debug bool
)
c := cobra.Command{
Use: "example-app",
......@@ -101,6 +128,17 @@ func cmd() *cobra.Command {
a.ctx = context.WithValue(a.ctx, oauth2.HTTPClient, client)
}
if debug {
client, ok := a.ctx.Value(oauth2.HTTPClient).(*http.Client)
if ok {
client.Transport = debugTransport{client.Transport}
} else {
a.ctx = context.WithValue(a.ctx, oauth2.HTTPClient, &http.Client{
Transport: debugTransport{http.DefaultTransport},
})
}
}
// TODO(ericchiang): Retry with backoff
provider, err := oidc.NewProvider(a.ctx, issuerURL)
if err != nil {
......@@ -161,6 +199,7 @@ func cmd() *cobra.Command {
c.Flags().StringVar(&tlsCert, "tls-cert", "", "X509 cert file to present when serving HTTPS.")
c.Flags().StringVar(&tlsKey, "tls-key", "", "Private key for the HTTPS cert.")
c.Flags().StringVar(&rootCAs, "issuer-root-ca", "", "Root certificate authorities for the issuer. Defaults to host certs.")
c.Flags().BoolVar(&debug, "debug", false, "Print all request and responses from the OpenID Connect issuer.")
return &c
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment