mirror of
https://github.com/argoproj/argo-cd
synced 2026-04-21 17:07:16 +00:00
72 lines
1.9 KiB
Go
72 lines
1.9 KiB
Go
package oidc
|
|
|
|
import (
|
|
"errors"
|
|
"html/template"
|
|
"log"
|
|
"net/http"
|
|
"reflect"
|
|
)
|
|
|
|
type tokenTmplData struct {
|
|
IDToken string
|
|
RefreshToken string
|
|
RedirectURL string
|
|
Claims string
|
|
}
|
|
|
|
var tokenTmpl = template.Must(template.New("token.html").Parse(`<html>
|
|
<head>
|
|
<style>
|
|
/* make pre wrap */
|
|
pre {
|
|
white-space: pre-wrap; /* css-3 */
|
|
white-space: -moz-pre-wrap; /* Mozilla, since 1999 */
|
|
white-space: -pre-wrap; /* Opera 4-6 */
|
|
white-space: -o-pre-wrap; /* Opera 7 */
|
|
word-wrap: break-word; /* Internet Explorer 5.5+ */
|
|
}
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<p> Token: <pre><code>{{ .IDToken }}</code></pre></p>
|
|
<p> Claims: <pre><code>{{ .Claims }}</code></pre></p>
|
|
{{ if .RefreshToken }}
|
|
<p> Refresh Token: <pre><code>{{ .RefreshToken }}</code></pre></p>
|
|
<form action="{{ .RedirectURL }}" method="post">
|
|
<input type="hidden" name="refresh_token" value="{{ .RefreshToken }}">
|
|
<input type="submit" value="Redeem refresh token">
|
|
</form>
|
|
{{ end }}
|
|
</body>
|
|
</html>
|
|
`))
|
|
|
|
func renderToken(w http.ResponseWriter, redirectURL, idToken, refreshToken string, claims []byte) {
|
|
renderTemplate(w, tokenTmpl, tokenTmplData{
|
|
IDToken: idToken,
|
|
RefreshToken: refreshToken,
|
|
RedirectURL: redirectURL,
|
|
Claims: string(claims),
|
|
})
|
|
}
|
|
|
|
func renderTemplate(w http.ResponseWriter, tmpl *template.Template, data any) {
|
|
err := tmpl.Execute(w, data)
|
|
if err == nil {
|
|
return
|
|
}
|
|
|
|
var templateErr *template.Error
|
|
switch {
|
|
case errors.As(err, &templateErr):
|
|
// An ExecError guarantees that Execute has not written to the underlying reader.
|
|
log.Printf("Error rendering template %s: %s", tmpl.Name(), reflect.TypeOf(err))
|
|
|
|
// TODO(ericchiang): replace with better internal server error.
|
|
http.Error(w, "Internal server error", http.StatusInternalServerError)
|
|
default:
|
|
// An error with the underlying write, such as the connection being
|
|
// dropped. Ignore for now.
|
|
}
|
|
}
|