mirror of https://github.com/coder/coder.git
41 lines
1.0 KiB
Go
41 lines
1.0 KiB
Go
package httpmw
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
|
|
"github.com/coder/coder/v2/coderd/externalauth"
|
|
"github.com/coder/coder/v2/coderd/httpapi"
|
|
)
|
|
|
|
type externalAuthParamContextKey struct{}
|
|
|
|
func ExternalAuthParam(r *http.Request) *externalauth.Config {
|
|
config, ok := r.Context().Value(externalAuthParamContextKey{}).(*externalauth.Config)
|
|
if !ok {
|
|
panic("developer error: external auth param middleware not provided")
|
|
}
|
|
return config
|
|
}
|
|
|
|
func ExtractExternalAuthParam(configs []*externalauth.Config) func(next http.Handler) http.Handler {
|
|
configByID := make(map[string]*externalauth.Config)
|
|
for _, c := range configs {
|
|
configByID[c.ID] = c
|
|
}
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
config, ok := configByID[chi.URLParam(r, "externalauth")]
|
|
if !ok {
|
|
httpapi.ResourceNotFound(w)
|
|
return
|
|
}
|
|
|
|
r = r.WithContext(context.WithValue(r.Context(), externalAuthParamContextKey{}, config))
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|