traefik/provider/web/web.go

331 lines
12 KiB
Go
Raw Normal View History

package web
2015-09-08 11:33:10 +00:00
import (
2015-09-08 22:22:34 +00:00
"encoding/json"
"expvar"
2015-09-12 13:10:03 +00:00
"fmt"
"io/ioutil"
"net/http"
"runtime"
"github.com/containous/mux"
"github.com/containous/traefik/autogen"
"github.com/containous/traefik/log"
"github.com/containous/traefik/middlewares"
mauth "github.com/containous/traefik/middlewares/auth"
"github.com/containous/traefik/safe"
"github.com/containous/traefik/types"
2016-10-02 18:07:25 +00:00
"github.com/containous/traefik/version"
2015-09-11 16:47:54 +00:00
"github.com/elazarl/go-bindata-assetfs"
2017-01-12 13:34:54 +00:00
"github.com/prometheus/client_golang/prometheus/promhttp"
thoas_stats "github.com/thoas/stats"
"github.com/unrolled/render"
"github.com/urfave/negroni"
2015-09-08 11:33:10 +00:00
)
// Provider is a provider.Provider implementation that provides the UI
type Provider struct {
2017-10-02 08:32:02 +00:00
Address string `description:"Web administration port" export:"true"`
CertFile string `description:"SSL certificate" export:"true"`
KeyFile string `description:"SSL certificate" export:"true"`
ReadOnly bool `description:"Enable read only API" export:"true"`
Statistics *types.Statistics `description:"Enable more detailed statistics" export:"true"`
Metrics *types.Metrics `description:"Enable a metrics exporter" export:"true"`
Path string `description:"Root path for dashboard and API"`
2017-10-02 08:32:02 +00:00
Auth *types.Auth `export:"true"`
Debug bool `export:"true"`
CurrentConfigurations *safe.Safe
Stats *thoas_stats.Stats
StatsRecorder *middlewares.StatsRecorder
2015-09-08 11:33:10 +00:00
}
var (
templatesRenderer = render.New(render.Options{
Directory: "nowhere",
})
)
func init() {
expvar.Publish("Goroutines", expvar.Func(goroutines))
}
func goroutines() interface{} {
return runtime.NumGoroutine()
}
// Provide allows the provider to provide configurations to traefik
// using the given configuration channel.
func (provider *Provider) Provide(configurationChan chan<- types.ConfigMessage, pool *safe.Pool, _ types.Constraints) error {
2015-09-08 11:33:10 +00:00
systemRouter := mux.NewRouter()
2017-03-03 22:09:44 +00:00
if provider.Path != "/" {
systemRouter.Methods("GET").Path("/").HandlerFunc(func(response http.ResponseWriter, request *http.Request) {
http.Redirect(response, request, provider.Path, 302)
})
}
2017-01-12 13:34:54 +00:00
// Prometheus route
if provider.Metrics != nil && provider.Metrics.Prometheus != nil {
2017-03-03 22:09:44 +00:00
systemRouter.Methods("GET").Path(provider.Path + "metrics").Handler(promhttp.Handler())
2017-01-12 13:34:54 +00:00
}
2015-10-05 14:01:34 +00:00
// health route
2017-03-03 22:09:44 +00:00
systemRouter.Methods("GET").Path(provider.Path + "health").HandlerFunc(provider.getHealthHandler)
2015-10-05 14:01:34 +00:00
2016-08-19 16:02:26 +00:00
// ping route
systemRouter.Methods("GET", "HEAD").Path(provider.Path + "ping").HandlerFunc(provider.getPingHandler)
2015-10-05 14:01:34 +00:00
// API routes
2017-03-03 22:09:44 +00:00
systemRouter.Methods("GET").Path(provider.Path + "api").HandlerFunc(provider.getConfigHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/version").HandlerFunc(provider.getVersionHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/providers").HandlerFunc(provider.getConfigHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/providers/{provider}").HandlerFunc(provider.getProviderHandler)
systemRouter.Methods("PUT").Path(provider.Path + "api/providers/{provider}").HandlerFunc(func(response http.ResponseWriter, request *http.Request) {
2016-01-03 19:42:09 +00:00
if provider.ReadOnly {
response.WriteHeader(http.StatusForbidden)
fmt.Fprint(response, "REST API is in read-only mode")
2016-01-03 19:42:09 +00:00
return
}
2015-10-05 14:01:34 +00:00
vars := mux.Vars(request)
if vars["provider"] != "web" {
response.WriteHeader(http.StatusBadRequest)
fmt.Fprint(response, "Only 'web' provider can be updated through the REST API")
2015-10-05 14:01:34 +00:00
return
}
configuration := new(types.Configuration)
2015-10-05 14:01:34 +00:00
body, _ := ioutil.ReadAll(request.Body)
err := json.Unmarshal(body, configuration)
if err == nil {
configurationChan <- types.ConfigMessage{ProviderName: "web", Configuration: configuration}
provider.getConfigHandler(response, request)
2015-10-05 14:01:34 +00:00
} else {
log.Errorf("Error parsing configuration %+v", err)
http.Error(response, fmt.Sprintf("%+v", err), http.StatusBadRequest)
}
})
2017-03-03 22:09:44 +00:00
systemRouter.Methods("GET").Path(provider.Path + "api/providers/{provider}/backends").HandlerFunc(provider.getBackendsHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/providers/{provider}/backends/{backend}").HandlerFunc(provider.getBackendHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/providers/{provider}/backends/{backend}/servers").HandlerFunc(provider.getServersHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/providers/{provider}/backends/{backend}/servers/{server}").HandlerFunc(provider.getServerHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/providers/{provider}/frontends").HandlerFunc(provider.getFrontendsHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/providers/{provider}/frontends/{frontend}").HandlerFunc(provider.getFrontendHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/providers/{provider}/frontends/{frontend}/routes").HandlerFunc(provider.getRoutesHandler)
systemRouter.Methods("GET").Path(provider.Path + "api/providers/{provider}/frontends/{frontend}/routes/{route}").HandlerFunc(provider.getRouteHandler)
2015-10-05 14:01:34 +00:00
// Expose dashboard
2017-03-03 22:09:44 +00:00
systemRouter.Methods("GET").Path(provider.Path).HandlerFunc(func(response http.ResponseWriter, request *http.Request) {
http.Redirect(response, request, provider.Path+"dashboard/", 302)
2015-10-05 14:01:34 +00:00
})
systemRouter.Methods("GET").PathPrefix(provider.Path + "dashboard/").
Handler(http.StripPrefix(provider.Path+"dashboard/", http.FileServer(&assetfs.AssetFS{Asset: autogen.Asset, AssetInfo: autogen.AssetInfo, AssetDir: autogen.AssetDir, Prefix: "static"})))
2015-09-08 11:33:10 +00:00
// expvars
if provider.Debug {
systemRouter.Methods("GET").Path(provider.Path + "debug/vars").HandlerFunc(expVarHandler)
}
safe.Go(func() {
2016-09-19 15:00:12 +00:00
var err error
2017-08-21 21:18:02 +00:00
var negroniInstance = negroni.New()
if provider.Auth != nil {
authMiddleware, err := mauth.NewAuthenticator(provider.Auth)
if err != nil {
log.Fatal("Error creating Auth: ", err)
}
2017-08-21 21:18:02 +00:00
authMiddlewareWrapper := negroni.HandlerFunc(func(w http.ResponseWriter, r *http.Request, next http.HandlerFunc) {
if r.URL.Path == "/ping" {
next.ServeHTTP(w, r)
} else {
authMiddleware.ServeHTTP(w, r, next)
}
})
negroniInstance.Use(authMiddlewareWrapper)
2016-09-19 15:00:12 +00:00
}
2017-08-21 21:18:02 +00:00
negroniInstance.UseHandler(systemRouter)
2016-09-19 15:00:12 +00:00
if len(provider.CertFile) > 0 && len(provider.KeyFile) > 0 {
2017-08-21 21:18:02 +00:00
err = http.ListenAndServeTLS(provider.Address, provider.CertFile, provider.KeyFile, negroniInstance)
2015-09-22 19:00:29 +00:00
} else {
2017-08-21 21:18:02 +00:00
err = http.ListenAndServe(provider.Address, negroniInstance)
2016-09-19 15:00:12 +00:00
}
2016-09-19 15:00:12 +00:00
if err != nil {
log.Fatal("Error creating server: ", err)
2015-09-22 19:00:29 +00:00
}
})
return nil
2015-09-08 11:33:10 +00:00
}
// healthResponse combines data returned by thoas/stats with statistics (if
// they are enabled).
type healthResponse struct {
*thoas_stats.Data
*middlewares.Stats
}
func (provider *Provider) getHealthHandler(response http.ResponseWriter, request *http.Request) {
health := &healthResponse{Data: provider.Stats.Data()}
if provider.StatsRecorder != nil {
health.Stats = provider.StatsRecorder.Data()
}
templatesRenderer.JSON(response, http.StatusOK, health)
2015-09-08 11:33:10 +00:00
}
func (provider *Provider) getPingHandler(response http.ResponseWriter, request *http.Request) {
fmt.Fprint(response, "OK")
2016-08-19 16:02:26 +00:00
}
func (provider *Provider) getConfigHandler(response http.ResponseWriter, request *http.Request) {
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
templatesRenderer.JSON(response, http.StatusOK, currentConfigurations)
2015-09-08 11:33:10 +00:00
}
func (provider *Provider) getVersionHandler(response http.ResponseWriter, request *http.Request) {
2016-10-02 18:07:25 +00:00
v := struct {
Version string
Codename string
}{
Version: version.Version,
Codename: version.Codename,
}
templatesRenderer.JSON(response, http.StatusOK, v)
}
func (provider *Provider) getProviderHandler(response http.ResponseWriter, request *http.Request) {
2015-10-05 14:01:34 +00:00
vars := mux.Vars(request)
providerID := vars["provider"]
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
if provider, ok := currentConfigurations[providerID]; ok {
2015-10-05 14:01:34 +00:00
templatesRenderer.JSON(response, http.StatusOK, provider)
} else {
http.NotFound(response, request)
}
2015-10-03 14:50:53 +00:00
}
func (provider *Provider) getBackendsHandler(response http.ResponseWriter, request *http.Request) {
2015-10-05 14:01:34 +00:00
vars := mux.Vars(request)
providerID := vars["provider"]
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
if provider, ok := currentConfigurations[providerID]; ok {
2015-10-05 14:01:34 +00:00
templatesRenderer.JSON(response, http.StatusOK, provider.Backends)
2015-09-23 05:48:32 +00:00
} else {
2015-10-05 14:01:34 +00:00
http.NotFound(response, request)
2015-09-23 05:48:32 +00:00
}
2015-09-15 16:35:32 +00:00
}
func (provider *Provider) getBackendHandler(response http.ResponseWriter, request *http.Request) {
2015-10-05 14:01:34 +00:00
vars := mux.Vars(request)
providerID := vars["provider"]
backendID := vars["backend"]
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
if provider, ok := currentConfigurations[providerID]; ok {
2015-10-05 14:01:34 +00:00
if backend, ok := provider.Backends[backendID]; ok {
templatesRenderer.JSON(response, http.StatusOK, backend)
2015-09-23 05:48:32 +00:00
return
}
2015-09-16 17:08:01 +00:00
}
2015-10-05 14:01:34 +00:00
http.NotFound(response, request)
2015-09-15 16:35:32 +00:00
}
func (provider *Provider) getServersHandler(response http.ResponseWriter, request *http.Request) {
vars := mux.Vars(request)
providerID := vars["provider"]
backendID := vars["backend"]
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
if provider, ok := currentConfigurations[providerID]; ok {
if backend, ok := provider.Backends[backendID]; ok {
templatesRenderer.JSON(response, http.StatusOK, backend.Servers)
return
}
}
http.NotFound(response, request)
}
func (provider *Provider) getServerHandler(response http.ResponseWriter, request *http.Request) {
vars := mux.Vars(request)
providerID := vars["provider"]
backendID := vars["backend"]
serverID := vars["server"]
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
if provider, ok := currentConfigurations[providerID]; ok {
if backend, ok := provider.Backends[backendID]; ok {
if server, ok := backend.Servers[serverID]; ok {
templatesRenderer.JSON(response, http.StatusOK, server)
return
}
}
}
http.NotFound(response, request)
}
func (provider *Provider) getFrontendsHandler(response http.ResponseWriter, request *http.Request) {
2015-10-05 14:01:34 +00:00
vars := mux.Vars(request)
providerID := vars["provider"]
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
if provider, ok := currentConfigurations[providerID]; ok {
2015-10-05 14:01:34 +00:00
templatesRenderer.JSON(response, http.StatusOK, provider.Frontends)
2015-09-23 05:48:32 +00:00
} else {
2015-10-05 14:01:34 +00:00
http.NotFound(response, request)
2015-09-23 05:48:32 +00:00
}
2015-09-15 16:35:32 +00:00
}
func (provider *Provider) getFrontendHandler(response http.ResponseWriter, request *http.Request) {
2015-10-05 14:01:34 +00:00
vars := mux.Vars(request)
providerID := vars["provider"]
frontendID := vars["frontend"]
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
if provider, ok := currentConfigurations[providerID]; ok {
2015-10-05 14:01:34 +00:00
if frontend, ok := provider.Frontends[frontendID]; ok {
templatesRenderer.JSON(response, http.StatusOK, frontend)
2015-09-23 05:48:32 +00:00
return
}
2015-09-16 17:08:01 +00:00
}
2015-10-05 14:01:34 +00:00
http.NotFound(response, request)
2015-09-15 16:35:32 +00:00
}
2015-09-16 17:08:01 +00:00
func (provider *Provider) getRoutesHandler(response http.ResponseWriter, request *http.Request) {
2015-10-05 14:01:34 +00:00
vars := mux.Vars(request)
providerID := vars["provider"]
frontendID := vars["frontend"]
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
if provider, ok := currentConfigurations[providerID]; ok {
if frontend, ok := provider.Frontends[frontendID]; ok {
templatesRenderer.JSON(response, http.StatusOK, frontend.Routes)
2015-09-23 05:48:32 +00:00
return
}
2015-09-16 17:08:01 +00:00
}
2015-10-05 14:01:34 +00:00
http.NotFound(response, request)
2015-09-16 17:08:01 +00:00
}
func (provider *Provider) getRouteHandler(response http.ResponseWriter, request *http.Request) {
2015-10-05 14:01:34 +00:00
vars := mux.Vars(request)
providerID := vars["provider"]
frontendID := vars["frontend"]
routeID := vars["route"]
currentConfigurations := provider.CurrentConfigurations.Get().(types.Configurations)
if provider, ok := currentConfigurations[providerID]; ok {
if frontend, ok := provider.Frontends[frontendID]; ok {
if route, ok := frontend.Routes[routeID]; ok {
templatesRenderer.JSON(response, http.StatusOK, route)
2015-09-23 05:48:32 +00:00
return
}
2015-09-16 17:08:01 +00:00
}
}
2015-10-05 14:01:34 +00:00
http.NotFound(response, request)
2015-09-22 08:33:37 +00:00
}
func expVarHandler(w http.ResponseWriter, _ *http.Request) {
w.Header().Set("Content-Type", "application/json; charset=utf-8")
fmt.Fprint(w, "{\n")
first := true
expvar.Do(func(kv expvar.KeyValue) {
if !first {
fmt.Fprint(w, ",\n")
}
first = false
fmt.Fprintf(w, "%q: %s", kv.Key, kv.Value)
})
fmt.Fprint(w, "\n}\n")
}