diff --git a/internal/handlers/collections.go b/internal/handlers/collections.go index f2b4c6b..9572024 100644 --- a/internal/handlers/collections.go +++ b/internal/handlers/collections.go @@ -487,3 +487,13 @@ func (h *CollectionHandler) GetCollectionData(c echo.Context, collectionID uuid. func (h *CollectionHandler) GetCollectionBooksData(c echo.Context, collectionID uuid.UUID) ([]database.GetCollectionItemsRow, error) { return h.collectionService.GetCollectionBooks(c.Request().Context(), collectionID) } + +func (h *CollectionHandler) GetDeviceMappingsData(c echo.Context, deviceID uuid.UUID) ([]database.GetDeviceShelfMappingsRow, error) { + return h.collectionService.GetDeviceShelfMappings(c.Request().Context(), deviceID) +} + +func (h *CollectionHandler) GetUserCollectionsList(c echo.Context) ([]database.Collections, error) { + user := c.Get("user").(database.Users) + userUUID := uuid.UUID(user.ID.Bytes) + return h.collectionService.GetUserCollections(c.Request().Context(), userUUID) +}