wakapi/routes/heartbeat.go

package routes

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

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

	_ "github.com/go-sql-driver/mysql"
	"github.com/n1try/wakapi/models"
)

type HeartbeatHandler struct {
	HeartbeatSrvc *services.HeartbeatService
}

func (h *HeartbeatHandler) Post(w http.ResponseWriter, r *http.Request) {
	if r.Method != "POST" {
		w.WriteHeader(415)
		return
	}

	opSys, editor, _ := utils.ParseUserAgent(r.Header.Get("User-Agent"))

	dec := json.NewDecoder(r.Body)
	var heartbeats []*models.Heartbeat
	err := dec.Decode(&heartbeats)
	if err != nil {
		w.WriteHeader(400)
		w.Write([]byte(err.Error()))
		return
	}
	for _, h := range heartbeats {
		h.OperatingSystem = opSys
		h.Editor = editor
	}

	user := r.Context().Value(models.UserKey).(*models.User)
	err = h.HeartbeatSrvc.InsertBatch(heartbeats, user)
	if err != nil {
		w.WriteHeader(500)
		os.Stderr.WriteString(err.Error())
		return
	}

	w.WriteHeader(200)
}