collection_groups.go

package handler

import (
	"net/http"
	"strconv"
	"time"

	"github.com/USACE/instrumentation-api/api/internal/db"
	"github.com/USACE/instrumentation-api/api/internal/dto"
	"github.com/USACE/instrumentation-api/api/internal/httperr"

	"github.com/google/uuid"

	"github.com/labstack/echo/v4"
)

// ListCollectionGroups godoc
//
//	@Summary lists instrument groups
//	@Tags collection-groups
//	@Produce json
//	@Param project_id path string true "project uuid" Format(uuid)
//	@Success 200 {array} db.CollectionGroup
//	@Failure 400 {object} echo.HTTPError
//	@Failure 404 {object} echo.HTTPError
//	@Failure 500 {object} echo.HTTPError
//	@Router /projects/{project_id}/collection_groups [get]
func (h *ApiHandler) ListCollectionGroups(c echo.Context) error {
	pID, err := uuid.Parse(c.Param("project_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	cc, err := h.DBService.CollectionGroupListForProject(c.Request().Context(), pID)
	if err != nil {
		return httperr.InternalServerError(err)
	}
	return c.JSON(http.StatusOK, &cc)
}

// GetCollectionGroupDetails godoc
//
//	@Summary gets all data needed to render collection group form
//	@Tags collection-groups
//	@Produce json
//	@Param project_id path string true "project uuid" Format(uuid)
//	@Param collection_group_id path string true "collection group uuid" Format(uuid)
//	@Success 200 {object} db.VCollectionGroupDetails
//	@Failure 400 {object} echo.HTTPError
//	@Failure 404 {object} echo.HTTPError
//	@Failure 500 {object} echo.HTTPError
//	@Router /projects/{project_id}/collection_groups/{collection_group_id} [get]
func (h *ApiHandler) GetCollectionGroupDetails(c echo.Context) error {
	_, err := uuid.Parse(c.Param("project_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	cgID, err := uuid.Parse(c.Param("collection_group_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	d, err := h.DBService.CollectionGroupDetailsGet(c.Request().Context(), cgID)
	if err != nil {
		return httperr.InternalServerError(err)
	}
	return c.JSON(http.StatusOK, d)
}

// CreateCollectionGroup godoc
//
//	@Summary creates a new collection group
//	@Description lists alert configs for a single project optionally filtered by alert_type_id
//	@Tags collection-groups
//	@Produce json
//	@Param project_id path string true "project uuid" Format(uuid)
//	@Param collection_group body dto.CollectionGroup true "collection group payload"
//	@Param key query string false "api key"
//	@Success 200 {array} db.CollectionGroup
//	@Failure 400 {object} echo.HTTPError
//	@Failure 404 {object} echo.HTTPError
//	@Failure 500 {object} echo.HTTPError
//	@Router /projects/{project_id}/collection_groups [post]
//	@Security Bearer
func (h *ApiHandler) CreateCollectionGroup(c echo.Context) error {
	var cg dto.CollectionGroup
	// Bind Information Provided
	if err := c.Bind(&cg); err != nil {
		return httperr.MalformedBody(err)
	}
	// Project ID from Route Params
	pID, err := uuid.Parse(c.Param("project_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	cg.ProjectID = pID
	p := c.Get("profile").(db.VProfile)
	cg.CreatedBy, cg.CreatedAt = p.ID, time.Now()

	cgNew, err := h.DBService.CollectionGroupCreate(c.Request().Context(), cg)
	if err != nil {
		return httperr.InternalServerError(err)
	}
	return c.JSON(http.StatusCreated, []db.CollectionGroup{cgNew})
}

// UpdateCollectionGroup godoc
//
//	@Summary updates an existing collection group
//	@Tags collection-groups
//	@Produce json
//	@Param project_id path string true "project uuid" Format(uuid)
//	@Param collection_group_id path string true "collection group uuid"
//	@Param collection_group body dto.CollectionGroup true "collection group payload"
//	@Param key query string false "api key"
//	@Success 200 {object} db.CollectionGroup
//	@Failure 400 {object} echo.HTTPError
//	@Failure 404 {object} echo.HTTPError
//	@Failure 500 {object} echo.HTTPError
//	@Router /projects/{project_id}/collection_groups/{collection_group_id} [put]
//	@Security Bearer
func (h *ApiHandler) UpdateCollectionGroup(c echo.Context) error {
	var cg dto.CollectionGroup
	if err := c.Bind(&cg); err != nil {
		return httperr.MalformedBody(err)
	}

	pID, err := uuid.Parse(c.Param("project_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	cg.ProjectID = pID

	cgID, err := uuid.Parse(c.Param("collection_group_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	cg.ID = cgID

	p := c.Get("profile").(db.VProfile)
	t := time.Now()
	cg.UpdatedBy, cg.UpdatedAt = &p.ID, &t
	cgUpdated, err := h.DBService.CollectionGroupUpdate(c.Request().Context(), cg)
	if err != nil {
		return httperr.InternalServerError(err)
	}
	return c.JSON(http.StatusOK, cgUpdated)
}

// DeleteCollectionGroup godoc
//
//	@Summary deletes a collection group using the id of the collection group
//	@Tags collection-groups
//	@Produce json
//	@Param project_id path string true "project uuid" Format(uuid)
//	@Param collection_group_id path string true "collection group uuid" Format(uuid)
//	@Param key query string false "api key"
//	@Success 200 {object} map[string]interface{}
//	@Failure 400 {object} echo.HTTPError
//	@Failure 404 {object} echo.HTTPError
//	@Failure 500 {object} echo.HTTPError
//	@Router /projects/{project_id}/collection_groups/{collection_group_id} [delete]
//	@Security Bearer
func (h *ApiHandler) DeleteCollectionGroup(c echo.Context) error {
	pID, err := uuid.Parse(c.Param("project_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	cgID, err := uuid.Parse(c.Param("collection_group_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	if err := h.DBService.CollectionGroupDelete(c.Request().Context(), db.CollectionGroupDeleteParams{
		ProjectID: pID,
		ID:        cgID,
	}); err != nil {
		return httperr.InternalServerError(err)
	}
	return c.JSON(http.StatusOK, make(map[string]interface{}))
}

// AddTimeseriesToCollectionGroup godoc
//
//	@Summary adds a timeseries to a collection group
//	@Tags collection-groups
//	@Produce json
//	@Param project_id path string true "project uuid" Format(uuid)
//	@Param collection_group_id path string true "collection group uuid" Format(uuid)
//	@Param timeseries_id path string true "timeseries uuid" Format(uuid)
//	@Param key query string false "api key"
//	@Success 201 {object} map[string]interface{}
//	@Failure 400 {object} echo.HTTPError
//	@Failure 404 {object} echo.HTTPError
//	@Failure 500 {object} echo.HTTPError
//	@Router /projects/{project_id}/collection_groups/{collection_group_id}/timeseries/{timeseries_id} [post]
//	@Security Bearer
func (h *ApiHandler) AddTimeseriesToCollectionGroup(c echo.Context) error {
	cgID, err := uuid.Parse(c.Param("collection_group_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	tsID, err := uuid.Parse(c.Param("timeseries_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	var sortOrder int32
	soParam := c.QueryParam("sort_order")
	if soParam != "" {
		so64, err := strconv.ParseInt(soParam, 10, 32)
		if err != nil {
			return httperr.BadRequest(err)
		}
		sortOrder = int32(so64)
	}

	if err := h.DBService.CollectionGroupTimeseriesCreate(c.Request().Context(), db.CollectionGroupTimeseriesCreateParams{
		CollectionGroupID: cgID,
		TimeseriesID:      tsID,
		SortOrder:         sortOrder,
	}); err != nil {
		return httperr.InternalServerError(err)
	}
	return c.JSON(http.StatusCreated, make(map[string]interface{}))
}

// UpdateTimeseriesCollectionGroupSortOrder godoc
//
//	@Summary updates sort order for collection group timesries
//	@Tags collection-groups
//	@Produce json
//	@Param project_id path string true "project uuid" Format(uuid)
//	@Param collection_group_id path string true "collection group uuid" Format(uuid)
//	@Param timeseries_id path string true "timeseries uuid" Format(uuid)
//	@Param key query string false "api key"
//	@Success 200 {object} map[string]interface{}
//	@Failure 400 {object} echo.HTTPError
//	@Failure 404 {object} echo.HTTPError
//	@Failure 500 {object} echo.HTTPError
//	@Router /projects/{project_id}/collection_groups/{collection_group_id}/timeseries/{timeseries_id} [put]
//	@Security Bearer
func (h *ApiHandler) UpdateTimeseriesCollectionGroupSortOrder(c echo.Context) error {
	cgID, err := uuid.Parse(c.Param("collection_group_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	tsID, err := uuid.Parse(c.Param("timeseries_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	var sortOrder int32
	soParam := c.QueryParam("sort_order")
	if soParam != "" {
		so64, err := strconv.ParseInt(soParam, 10, 32)
		if err != nil {
			return httperr.BadRequest(err)
		}
		sortOrder = int32(so64)
	}

	if err := h.DBService.CollectionGroupTimeseriesUpdateSortOrder(c.Request().Context(), db.CollectionGroupTimeseriesUpdateSortOrderParams{
		CollectionGroupID: cgID,
		TimeseriesID:      tsID,
		SortOrder:         sortOrder,
	}); err != nil {
		return httperr.InternalServerError(err)
	}
	return c.JSON(http.StatusOK, make(map[string]interface{}))
}

// RemoveTimeseriesFromCollectionGroup godoc
//
//	@Summary removes a timeseries from a collection group
//	@Tags collection-groups
//	@Produce json
//	@Param project_id path string true "project uuid" Format(uuid)
//	@Param collection_group_id path string true "collection group uuid" Format(uuid)
//	@Param timeseries_id path string true "timeseries uuid" Format(uuid)
//	@Param key query string false "api key"
//	@Success 200 {object} map[string]interface{}
//	@Failure 400 {object} echo.HTTPError
//	@Failure 404 {object} echo.HTTPError
//	@Failure 500 {object} echo.HTTPError
//	@Router /projects/{project_id}/collection_groups/{collection_group_id}/timeseries/{timeseries_id} [delete]
//	@Security Bearer
func (h *ApiHandler) RemoveTimeseriesFromCollectionGroup(c echo.Context) error {
	cgID, err := uuid.Parse(c.Param("collection_group_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	tsID, err := uuid.Parse(c.Param("timeseries_id"))
	if err != nil {
		return httperr.MalformedID(err)
	}
	if err := h.DBService.CollectionGroupTimeseriesDelete(c.Request().Context(), db.CollectionGroupTimeseriesDeleteParams{
		CollectionGroupID: cgID,
		TimeseriesID:      tsID,
	}); err != nil {
		return httperr.InternalServerError(err)
	}
	return c.JSON(http.StatusOK, make(map[string]interface{}))
}