86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
nethttp "net/http"
|
|
"strings"
|
|
|
|
idpctx "github.com/influxdata/platform/context"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
// PlatformHandler is a collection of all the service handlers.
|
|
type PlatformHandler struct {
|
|
BucketHandler *BucketHandler
|
|
UserHandler *UserHandler
|
|
OrgHandler *OrgHandler
|
|
AuthorizationHandler *AuthorizationHandler
|
|
DashboardHandler *DashboardHandler
|
|
}
|
|
|
|
func setCORSResponseHeaders(w nethttp.ResponseWriter, r *nethttp.Request) {
|
|
if origin := r.Header.Get("Origin"); origin != "" {
|
|
w.Header().Set("Access-Control-Allow-Origin", origin)
|
|
w.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
|
|
w.Header().Set("Access-Control-Allow-Headers", "Accept, Content-Type, Content-Length, Accept-Encoding, Authorization")
|
|
}
|
|
}
|
|
|
|
// ServeHTTP delegates a request to the appropriate subhandler.
|
|
func (h *PlatformHandler) ServeHTTP(w nethttp.ResponseWriter, r *nethttp.Request) {
|
|
|
|
setCORSResponseHeaders(w, r)
|
|
if r.Method == "OPTIONS" {
|
|
return
|
|
}
|
|
|
|
ctx := r.Context()
|
|
var err error
|
|
if ctx, err = extractAuthorization(ctx, r); err != nil {
|
|
nethttp.Error(w, err.Error(), nethttp.StatusBadRequest)
|
|
return
|
|
}
|
|
r = r.WithContext(ctx)
|
|
|
|
if strings.HasPrefix(r.URL.Path, "/v1/buckets") {
|
|
h.BucketHandler.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
if strings.HasPrefix(r.URL.Path, "/v1/users") {
|
|
h.UserHandler.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
if strings.HasPrefix(r.URL.Path, "/v1/orgs") {
|
|
h.OrgHandler.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
if strings.HasPrefix(r.URL.Path, "/v1/authorizations") {
|
|
h.AuthorizationHandler.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
if strings.HasPrefix(r.URL.Path, "/v1/dashboards") {
|
|
h.DashboardHandler.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
nethttp.NotFound(w, r)
|
|
}
|
|
|
|
// PrometheusCollectors satisfies the prom.PrometheusCollector interface.
|
|
func (h *PlatformHandler) PrometheusCollectors() []prometheus.Collector {
|
|
// TODO: collect and return relevant metrics.
|
|
return nil
|
|
}
|
|
|
|
func extractAuthorization(ctx context.Context, r *nethttp.Request) (context.Context, error) {
|
|
t, err := ParseAuthHeaderToken(r)
|
|
if err != nil {
|
|
return ctx, err
|
|
}
|
|
return idpctx.SetToken(ctx, t), nil
|
|
}
|