package rest import ( "encoding/json" "fmt" "io/ioutil" "net/http" "github.com/containous/mux" "github.com/containous/traefik/log" "github.com/containous/traefik/safe" "github.com/containous/traefik/types" "github.com/unrolled/render" ) // Provider is a provider.Provider implementation that provides a Rest API type Provider struct { configurationChan chan<- types.ConfigMessage EntryPoint string `description:"EntryPoint" export:"true"` CurrentConfigurations *safe.Safe } var templatesRenderer = render.New(render.Options{Directory: "nowhere"}) // AddRoutes add rest provider routes on a router func (p *Provider) AddRoutes(systemRouter *mux.Router) { systemRouter.Methods("PUT").Path("/api/providers/{provider}").HandlerFunc(func(response http.ResponseWriter, request *http.Request) { vars := mux.Vars(request) // TODO: Deprecated configuration - Need to be removed in the future if vars["provider"] != "web" && vars["provider"] != "rest" { response.WriteHeader(http.StatusBadRequest) fmt.Fprint(response, "Only 'rest' provider can be updated through the REST API") return } else if vars["provider"] == "web" { log.Warn("The provider web is deprecated. Please use /rest instead") } configuration := new(types.Configuration) body, _ := ioutil.ReadAll(request.Body) err := json.Unmarshal(body, configuration) if err == nil { // TODO: Deprecated configuration - Change to `rest` in the future p.configurationChan <- types.ConfigMessage{ProviderName: "web", Configuration: configuration} p.getConfigHandler(response, request) } else { log.Errorf("Error parsing configuration %+v", err) http.Error(response, fmt.Sprintf("%+v", err), http.StatusBadRequest) } }) } // Provide allows the provider to provide configurations to traefik // using the given configuration channel. func (p *Provider) Provide(configurationChan chan<- types.ConfigMessage, pool *safe.Pool, _ types.Constraints) error { p.configurationChan = configurationChan return nil } func (p *Provider) getConfigHandler(response http.ResponseWriter, request *http.Request) { currentConfigurations := p.CurrentConfigurations.Get().(types.Configurations) err := templatesRenderer.JSON(response, http.StatusOK, currentConfigurations) if err != nil { log.Error(err) } }