Files
astra-events/handler/event.go
andrusyakka 6752536325 WIP service
2023-07-14 18:39:50 +03:00

113 lines
2.4 KiB
Go

package handler
import (
"astra-events/model"
"astra-events/pkg/utils"
"astra-events/services"
"github.com/hashicorp/go-set"
"github.com/labstack/echo/v4"
"net/http"
"strconv"
)
func (h *Handler) CreateEvent(c echo.Context) (err error) {
newEvent := new(model.Event)
event := new(model.CreateEvent)
authorId, err := strconv.Atoi(c.Request().Header.Get("UserId"))
if err != nil {
return c.NoContent(http.StatusUnauthorized)
}
if err = c.Bind(event); err != nil {
return c.NoContent(http.StatusBadRequest)
}
usersMap, err := services.GetUsers([]uint{event.EmployeeId, uint(authorId)}, c.Request().Header)
if err != nil {
return c.HTML(http.StatusBadGateway, err.Error())
}
employee := usersMap[event.EmployeeId]
author := usersMap[uint(authorId)]
newEvent.Start = event.Start
newEvent.End = event.End
newEvent.AuthorId = uint(authorId)
newEvent.EmployeeId = event.EmployeeId
newEvent.MemberId = event.MemberId
newEvent.MedicalCardId = event.MedicalCardId
newEvent.Status = event.Status
newEvent.Services = event.Services
tx := h.DB.Create(&newEvent)
if tx.RowsAffected == 0 {
return c.NoContent(http.StatusBadGateway)
}
newEvent.Author = author
newEvent.Employee = employee
return c.JSON(http.StatusCreated, newEvent)
}
func (h *Handler) GetEvents(c echo.Context) (err error) {
var events []model.Event
query := c.Request().URL.Query()
_, _, filterMap := utils.GetFilterParams(query)
tx := h.DB
if len(filterMap) > 0 {
tx = tx.Where("start >= ?", filterMap["start"]).Where("\"end\" <= ?", filterMap["end"])
}
tx.Find(&events)
ids := set.New[uint](len(events) * 2)
for _, event := range events {
ids.Insert(event.EmployeeId)
ids.Insert(event.AuthorId)
}
usersMap, err := services.GetUsers(ids.Slice(), c.Request().Header)
if err != nil {
return c.HTML(http.StatusBadGateway, err.Error())
}
for i, event := range events {
events[i].Author = usersMap[event.AuthorId]
events[i].Employee = usersMap[event.EmployeeId]
}
return c.JSON(http.StatusOK, events)
}
func (h *Handler) GetEvent(c echo.Context) error {
var event model.Event
id := c.Param("id")
tx := h.DB.First(&event, id)
if tx.RowsAffected == 0 {
return c.NoContent(http.StatusNotFound)
}
return c.JSON(http.StatusOK, event)
}
func (h *Handler) DeleteEvent(c echo.Context) error {
id := c.Param("id")
tx := h.DB.Delete(&model.Event{}, id)
if tx.RowsAffected == 0 {
return c.NoContent(http.StatusNotFound)
}
return c.NoContent(http.StatusOK)
}