Split Web into API/Dashboard, ping, metric and Rest Provider

provider/rest/rest.go

@@ -0,0 +1,65 @@
+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)
+	}
+}