wakapi/routes/api/diagnostics.go

package api

import (
	"encoding/json"
	"github.com/gorilla/mux"
	conf "github.com/muety/wakapi/config"
	"github.com/muety/wakapi/middlewares"
	"github.com/muety/wakapi/services"
	"github.com/muety/wakapi/utils"
	"net/http"

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

type DiagnosticsApiHandler struct {
	config          *conf.Config
	userSrvc        services.IUserService
	diagnosticsSrvc services.IDiagnosticsService
}

func NewDiagnosticsApiHandler(userService services.IUserService, diagnosticsService services.IDiagnosticsService) *DiagnosticsApiHandler {
	return &DiagnosticsApiHandler{
		config:          conf.Get(),
		userSrvc:        userService,
		diagnosticsSrvc: diagnosticsService,
	}
}

func (h *DiagnosticsApiHandler) RegisterRoutes(router *mux.Router) {
	r := router.PathPrefix("/plugins/errors").Subrouter()
	r.Use(
		middlewares.NewAuthenticateMiddleware(h.userSrvc).Handler,
	)
	r.Path("").Methods(http.MethodPost).HandlerFunc(h.Post)
}

// @Summary Push a new diagnostics object
// @ID post-diagnostics
// @Tags diagnostics
// @Accept json
// @Param diagnostics body models.Diagnostics true "A single diagnostics object sent by WakaTime CLI"
// @Security ApiKeyAuth
// @Success 201
// @Router /api/plugins/errors [post]
func (h *DiagnosticsApiHandler) Post(w http.ResponseWriter, r *http.Request) {
	var diagnostics models.Diagnostics

	user := middlewares.GetPrincipal(r)
	if user == nil {
		w.WriteHeader(http.StatusUnauthorized)
		w.Write([]byte(conf.ErrUnauthorized))
		return
	}

	if err := json.NewDecoder(r.Body).Decode(&diagnostics); err != nil {
		w.WriteHeader(http.StatusBadRequest)
		w.Write([]byte(conf.ErrBadRequest))
		conf.Log().Request(r).Error("failed to parse diagnostics for user %s - %v", err)
		return
	}
	diagnostics.UserID = user.ID

	if _, err := h.diagnosticsSrvc.Create(&diagnostics); err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		w.Write([]byte(conf.ErrInternalServerError))
		conf.Log().Request(r).Error("failed to insert diagnostics for user %s - %v", err)
		return
	}

	utils.RespondJSON(w, r, http.StatusCreated, struct{}{})
}