package api import ( "encoding/json" "github.com/emvi/logbuch" "github.com/gorilla/mux" conf "github.com/muety/wakapi/config" customMiddleware "github.com/muety/wakapi/middlewares/custom" "github.com/muety/wakapi/services" "github.com/muety/wakapi/utils" "net/http" "github.com/muety/wakapi/models" ) type HeartbeatApiHandler struct { config *conf.Config heartbeatSrvc services.IHeartbeatService languageMappingSrvc services.ILanguageMappingService } func NewHeartbeatApiHandler(heartbeatService services.IHeartbeatService, languageMappingService services.ILanguageMappingService) *HeartbeatApiHandler { return &HeartbeatApiHandler{ config: conf.Get(), heartbeatSrvc: heartbeatService, languageMappingSrvc: languageMappingService, } } type heartbeatResponseVm struct { Responses [][]interface{} `json:"responses"` } func (h *HeartbeatApiHandler) RegisterRoutes(router *mux.Router) { r := router.PathPrefix("/heartbeat").Subrouter() r.Use( customMiddleware.NewWakatimeRelayMiddleware().Handler, ) r.Methods(http.MethodPost).HandlerFunc(h.Post) } func (h *HeartbeatApiHandler) Post(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")) machineName := r.Header.Get("X-Machine-Name") 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.Machine = machineName hb.User = user hb.UserID = user.ID if !hb.Valid() { w.WriteHeader(http.StatusBadRequest) w.Write([]byte("Invalid heartbeat object.")) return } hb.Hashed() } if err := h.heartbeatSrvc.InsertBatch(heartbeats); err != nil { w.WriteHeader(http.StatusInternalServerError) logbuch.Error(err.Error()) return } utils.RespondJSON(w, http.StatusCreated, constructSuccessResponse(len(heartbeats))) } // construct weird response format (see https://github.com/wakatime/wakatime/blob/2e636d389bf5da4e998e05d5285a96ce2c181e3d/wakatime/api.py#L288) // to make the cli consider all heartbeats to having been successfully saved // response looks like: { "responses": [ [ { "data": {...} }, 201 ], ... ] } func constructSuccessResponse(n int) *heartbeatResponseVm { responses := make([][]interface{}, n) for i := 0; i < n; i++ { r := make([]interface{}, 2) r[0] = nil r[1] = http.StatusCreated responses[i] = r } return &heartbeatResponseVm{ Responses: responses, } }