promcertcheck/http.go

package main

//go:generate go-bindata display.html

import (
	"encoding/json"
	"net/http"

	"github.com/flosch/pongo2"
	log "github.com/sirupsen/logrus"
)

func htmlHandler(res http.ResponseWriter, r *http.Request) {
	tplsrc := MustAsset("display.html")

	template, err := pongo2.FromString(string(tplsrc))
	if err != nil {
		log.Fatal(err)
	}

	if err := template.ExecuteWriter(pongo2.Context{
		"results":                probeMonitors,
		"certificateOK":          certificateOK,
		"certificateExpiresSoon": certificateExpiresSoon,
		"version":                version,
	}, res); err != nil {
		log.WithError(err).Error("Unable to render display template")
	}
}

func httpStatusHandler(res http.ResponseWriter, r *http.Request) {
	httpStatus := http.StatusOK
	for _, mon := range probeMonitors {
		if mon.Status != certificateOK {
			httpStatus = http.StatusInternalServerError
		}
	}

	res.WriteHeader(httpStatus)
}

func jsonHandler(res http.ResponseWriter, r *http.Request) {
	res.Header().Set("Content-Type", "application/json")
	json.NewEncoder(res).Encode(probeMonitors)
}