mondash/web_handlers.go

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"sort"
	"time"

	"github.com/flosch/pongo2"
	"github.com/gorilla/mux"
)

func handleRedirectWelcome(res http.ResponseWriter, req *http.Request) {
	http.Redirect(res, req, "/welcome", 302)
}

func handleCreateRandomDashboard(res http.ResponseWriter, req *http.Request) {
	var urlProposal string
	for {
		urlProposal = generateAPIKey()[0:20]
		if exists, err := store.Exists(urlProposal); err == nil && !exists {
			break
		}
	}
	http.Redirect(res, req, fmt.Sprintf("/%s", urlProposal), http.StatusTemporaryRedirect)
}

func handleDisplayDashboard(res http.ResponseWriter, req *http.Request) {
	params := mux.Vars(req)
	dash, err := loadDashboard(params["dashid"], store)
	if err != nil {
		dash = &dashboard{APIKey: generateAPIKey(), Metrics: dashboardMetrics{}}
	}

	// Filter out expired metrics
	metrics := dashboardMetrics{}
	for _, m := range dash.Metrics {
		if m.Meta.LastUpdate.After(time.Now().Add(time.Duration(m.Expires*-1) * time.Second)) {
			metrics = append(metrics, m)
		}
	}

	sort.Sort(sort.Reverse(metrics))
	renderTemplate("dashboard.html", pongo2.Context{
		"dashid":  params["dashid"],
		"metrics": metrics,
		"apikey":  dash.APIKey,
		"baseurl": cfg.BaseURL,
	}, res)
}

func handleDisplayDashboardJSON(res http.ResponseWriter, req *http.Request) {
	params := mux.Vars(req)
	dash, err := loadDashboard(params["dashid"], store)
	if err != nil {
		dash = &dashboard{APIKey: generateAPIKey(), Metrics: dashboardMetrics{}}
	}

	response := struct {
		APIKey  string `json:"api_key,omitempty"`
		Metrics []struct {
			ID          string    `json:"id"`
			Title       string    `json:"title"`
			Description string    `json:"description"`
			Status      string    `json:"status"`
			Value       float64   `json:"value"`
			LastUpdate  time.Time `json:"last_update"`
		} `json:"metrics"`
	}{}

	// Filter out expired metrics
	for _, m := range dash.Metrics {
		if m.Meta.LastUpdate.After(time.Now().Add(time.Duration(m.Expires*-1) * time.Second)) {
			response.Metrics = append(response.Metrics, struct {
				ID          string    `json:"id"`
				Title       string    `json:"title"`
				Description string    `json:"description"`
				Status      string    `json:"status"`
				Value       float64   `json:"value"`
				LastUpdate  time.Time `json:"last_update"`
			}{
				ID:          m.MetricID,
				Title:       m.Title,
				Description: m.Description,
				Status:      m.PreferredStatus(),
				Value:       m.Value,
				LastUpdate:  m.Meta.LastUpdate,
			})
		}
	}

	if len(response.Metrics) == 0 {
		response.APIKey = dash.APIKey
	}

	res.Header().Set("Content-Type", "application/json")
	json.NewEncoder(res).Encode(response)
}

func handleDeleteDashboard(res http.ResponseWriter, req *http.Request) {
	params := mux.Vars(req)
	dash, err := loadDashboard(params["dashid"], store)
	if err != nil {
		http.Error(res, "This dashboard does not exist.", http.StatusInternalServerError)
		return
	}

	if dash.APIKey != req.Header.Get("Authorization") {
		http.Error(res, "APIKey did not match.", http.StatusUnauthorized)
		return
	}

	store.Delete(params["dashid"])
	http.Error(res, "OK", http.StatusOK)
}

func handlePutMetric(res http.ResponseWriter, req *http.Request) {
	params := mux.Vars(req)
	body, err := ioutil.ReadAll(req.Body)
	if err != nil {
		http.Error(res, "Internal Server Error", http.StatusInternalServerError)
		return
	}

	metricUpdate := newDashboardMetric()
	err = json.Unmarshal(body, metricUpdate)
	if err != nil {
		http.Error(res, "Unable to unmarshal json", http.StatusInternalServerError)
		return
	}

	dash, err := loadDashboard(params["dashid"], store)
	if err != nil {
		if len(req.Header.Get("Authorization")) < 10 {
			http.Error(res, "APIKey is too insecure", http.StatusUnauthorized)
			return
		}
		dash = &dashboard{
			APIKey:      req.Header.Get("Authorization"),
			Metrics:     dashboardMetrics{},
			DashboardID: params["dashid"],
			storage:     store,
		}
	}

	if dash.APIKey != req.Header.Get("Authorization") {
		http.Error(res, "APIKey did not match.", http.StatusUnauthorized)
		return
	}

	valid, reason := metricUpdate.IsValid()
	if !valid {
		http.Error(res, fmt.Sprintf("Invalid data: %s", reason), http.StatusInternalServerError)
		return
	}

	updated := false
	for _, m := range dash.Metrics {
		if m.MetricID == params["metricid"] {
			m.Update(metricUpdate)
			updated = true
			break
		}
	}

	if !updated {
		tmp := newDashboardMetric()
		tmp.MetricID = params["metricid"]
		tmp.Update(metricUpdate)
		dash.Metrics = append(dash.Metrics, tmp)
	}

	dash.Save()

	http.Error(res, "OK", http.StatusOK)
}

func handleDeleteMetric(res http.ResponseWriter, req *http.Request) {
	params := mux.Vars(req)
	dash, err := loadDashboard(params["dashid"], store)
	if err != nil {
		dash = &dashboard{APIKey: req.Header.Get("Authorization"), Metrics: dashboardMetrics{}, DashboardID: params["dashid"]}
	}

	if dash.APIKey != req.Header.Get("Authorization") {
		http.Error(res, "APIKey did not match.", http.StatusUnauthorized)
		return
	}

	tmp := dashboardMetrics{}
	for _, m := range dash.Metrics {
		if m.MetricID != params["metricid"] {
			tmp = append(tmp, m)
		}
	}
	dash.Metrics = tmp
	dash.Save()

	http.Error(res, "OK", http.StatusOK)
}