resthandlers.go

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"net/http"
	"time"
)

type errorMessage struct {
	Message string `json:"message"`
}

func jsonError(w http.ResponseWriter, err interface{}, code int) {
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Header().Set("X-Content-Type-Options", "nosniff")
	w.WriteHeader(code)
	j, _ := json.Marshal(err)
	w.Write(j)
	// json.NewEncoder(w).Encode(err)
}

func middlewareContentType(nextHandler func(http.ResponseWriter, *http.Request)) func(http.ResponseWriter, *http.Request) {
	return func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		apikey := r.Header["X-Api-Key"]
		if len(apikey) > 0 && apikey[0] != config.APIKey {
			jsonError(w, errorMessage{Message: "unauthorised"}, http.StatusUnauthorized)
			return
		}
		nextHandler(w, r)
	}
}

func catchAllHandler(w http.ResponseWriter, r *http.Request) {
	buf := new(bytes.Buffer)
	buf.ReadFrom(r.Body)
	body := buf.String()

	rsp := catchAllResponse{
		Method:        r.Method,
		RequestURI:    r.RequestURI,
		URL:           r.URL,
		ContentLength: r.ContentLength,
		Host:          r.Host,
		Proto:         r.Proto,
		RemoteAddr:    r.RemoteAddr,
		Body:          body,
	}
	j, _ := json.Marshal(rsp)
	w.Write(j)
}

// GET /api/health
func healthHandler(w http.ResponseWriter, r *http.Request) {
	if r.Method != "GET" {
		jsonError(
			w,
			unsupportedResponse{SupportedMethods: []string{"GET"}},
			http.StatusMethodNotAllowed,
		)
		return
	}

	status := "ok"
	var info string
	d, err := db.Ping()
	if err != nil {
		status = "degraded"
		info = err.Error()
	}

	rsp := healthResponse{
		Status:     status,
		Version:    buildNumber,
		Info:       info,
		DBResponse: d.String(),
		LiveTime:   liveTime,
		DebugLevel: config.DebugLevel,
	}
	j, _ := json.Marshal(rsp)
	w.Write(j)
}

// GET /api/events?guildID={Guild_ID}
func listEventsHandler(w http.ResponseWriter, r *http.Request) {
	if r.Method != "GET" {
		jsonError(
			w,
			unsupportedResponse{SupportedMethods: []string{"GET"}},
			http.StatusMethodNotAllowed,
		)
		return
	}

	guilID := r.URL.Query().Get("guildId")
	if guilID == "" {
		jsonError(w, errorMessage{Message: "guildId not provided"}, http.StatusBadRequest)
		return
	}

	events, err := db.GetEvents(guilID, "all", time.Time{})
	if err != nil {
		jsonError(w, errorMessage{Message: err.Error()}, http.StatusInternalServerError)
		return
	}

	rsp, err := json.Marshal(events)
	if err != nil {
		jsonError(w, errorMessage{Message: err.Error()}, http.StatusInternalServerError)
		return
	}

	fmt.Fprint(w, string(rsp))
}