Melora Hugues
d5aa640df0
Because polyculeconnect is a OIDC proxy, we need to know which auth backend to use. This is provided using a query param or a form, so we need to get it from our own middleware. This commit adds the following elements: - basic DB storage for the backends - support for DB migrations and a first test migration (not definitive) - middleware to get the backend from the request and put it in the context - test that the backend exists in the auth flow
40 lines
932 B
Go
40 lines
932 B
Go
package middlewares
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
)
|
|
|
|
const (
|
|
backendNameQueryParam = "connector_id"
|
|
backendCtxKeyName = "backendName"
|
|
)
|
|
|
|
type BackendFromRequestMiddleware struct {
|
|
h http.Handler
|
|
}
|
|
|
|
func (m *BackendFromRequestMiddleware) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
if err := r.ParseForm(); err != nil {
|
|
// TODO: handle this better
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
backendName := r.Form.Get(backendNameQueryParam)
|
|
// TODO this should be explicitly handled
|
|
if backendName == "" {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
w.Write([]byte("no backend id provided"))
|
|
return
|
|
}
|
|
|
|
// TODO we should test that this backend actually exists here
|
|
|
|
ctx := context.WithValue(r.Context(), backendCtxKeyName, backendName)
|
|
m.h.ServeHTTP(w, r.WithContext(ctx))
|
|
}
|
|
|
|
func WithBackendFromRequestMiddleware(input http.Handler) http.Handler {
|
|
return &BackendFromRequestMiddleware{h: input}
|
|
}
|