traefik/web.go

package main

import (
	"github.com/gorilla/mux"
	"net/http"
	"github.com/unrolled/render"
	"fmt"
	"io/ioutil"
	"encoding/json"
	"github.com/elazarl/go-bindata-assetfs"
)

var renderer = render.New(render.Options{
	Directory: "templates",
	Asset: Asset,
	AssetNames: AssetNames,
})

type WebProvider struct {
	Address string
}

type Page struct {
	Configuration Configuration
}

func (provider *WebProvider) Provide(configurationChan chan <- *Configuration) {
	systemRouter := mux.NewRouter()
	systemRouter.Methods("GET").PathPrefix("/web/").Handler(http.HandlerFunc(GetHtmlConfigHandler))
	systemRouter.Methods("GET").PathPrefix("/api/").Handler(http.HandlerFunc(GetConfigHandler))
	systemRouter.Methods("POST").PathPrefix("/api/").Handler(http.HandlerFunc(
		func(rw http.ResponseWriter, r *http.Request) {
			configuration := new(Configuration)
			b, _ := ioutil.ReadAll(r.Body)
			err := json.Unmarshal(b, configuration)
			if (err == nil) {
				configurationChan <- configuration
				GetConfigHandler(rw, r)
			}else {
				log.Error("Error parsing configuration %+v\n", err)
				http.Error(rw, fmt.Sprintf("%+v", err), http.StatusBadRequest)
			}
		}))
	systemRouter.PathPrefix("/static").Handler(http.StripPrefix("/static/", http.FileServer(&assetfs.AssetFS{Asset: Asset, AssetDir: AssetDir, Prefix: "static"})))

	go http.ListenAndServe(provider.Address, systemRouter)
}

func GetConfigHandler(rw http.ResponseWriter, r *http.Request) {
	renderer.JSON(rw, http.StatusOK, currentConfiguration)
}

func GetHtmlConfigHandler(response http.ResponseWriter, request *http.Request) {
	renderer.HTML(response, http.StatusOK, "configuration", Page{Configuration:*currentConfiguration})
}