wakapi/routes/heartbeat.go

package routes

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

	"github.com/muety/wakapi/services"
	"github.com/muety/wakapi/utils"

	"github.com/muety/wakapi/models"
)

type HeartbeatHandler struct {
	config        *models.Config
	heartbeatSrvc *services.HeartbeatService
}

func NewHeartbeatHandler(heartbeatService *services.HeartbeatService) *HeartbeatHandler {
	return &HeartbeatHandler{
		config:        models.GetConfig(),
		heartbeatSrvc: heartbeatService,
	}
}

func (h *HeartbeatHandler) ApiPost(w http.ResponseWriter, r *http.Request) {
	var heartbeats []*models.Heartbeat
	user := r.Context().Value(models.UserKey).(*models.User)
	opSys, editor, _ := utils.ParseUserAgent(r.Header.Get("User-Agent"))

	dec := json.NewDecoder(r.Body)
	if err := dec.Decode(&heartbeats); err != nil {
		w.WriteHeader(http.StatusBadRequest)
		w.Write([]byte(err.Error()))
		return
	}

	for _, hb := range heartbeats {
		hb.OperatingSystem = opSys
		hb.Editor = editor
		hb.User = user
		hb.UserID = user.ID
		hb.Augment(h.config.CustomLanguages)

		if !hb.Valid() {
			w.WriteHeader(http.StatusBadRequest)
			w.Write([]byte("Invalid heartbeat object."))
			return
		}
	}

	if err := h.heartbeatSrvc.InsertBatch(heartbeats); err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		os.Stderr.WriteString(err.Error())
		return
	}

	w.WriteHeader(http.StatusOK)
}