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(dashboardMetrics(metrics)))
	renderTemplate("dashboard.html", pongo2.Context{
		"dashid":  params["dashid"],
		"metrics": metrics,
		"apikey":  dash.APIKey,
		"baseurl": cfg.BaseURL,
	}, res)
}

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)
}