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