mirror of https://github.com/muety/wakapi.git
60 lines
1.3 KiB
Go
60 lines
1.3 KiB
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)
|
|
}
|