book_cntrl.go

package controller

import (
	"fmt"
	"net/http"

	"github.com/labstack/echo/v4"
	"github.com/typical-go/typical-rest-server/internal/app/service"
	"github.com/typical-go/typical-rest-server/internal/app/entity"
	"github.com/typical-go/typical-rest-server/pkg/cachekit"
	"github.com/typical-go/typical-rest-server/pkg/echokit"
	"go.uber.org/dig"
)

type (
	// BookCntrl is controller to book entity
	BookCntrl struct {
		dig.In
		Svc   service.BookSvc
		Cache *cachekit.Store
	}
)

var _ echokit.Router = (*BookCntrl)(nil)

// SetRoute to define API Route
func (c *BookCntrl) SetRoute(e echokit.Server) {
	e.GET("/books", c.Find, c.Cache.Middleware)
	e.GET("/books/:id", c.FindOne, c.Cache.Middleware)
	e.HEAD("/books/:id", c.FindOne, c.Cache.Middleware)
	e.POST("/books", c.Create)
	e.PUT("/books/:id", c.Update)
	e.PATCH("/books/:id", c.Patch)
	e.DELETE("/books/:id", c.Delete)
}

// Create book
func (c *BookCntrl) Create(ec echo.Context) (err error) {
	var book entity.Book
	if err = ec.Bind(&book); err != nil {
		return err
	}
	ctx := ec.Request().Context()
	newBook, err := c.Svc.Create(ctx, &book)
	if err != nil {
		return echokit.HTTPError(err)
	}
	ec.Response().Header().Set(echo.HeaderLocation, fmt.Sprintf("/books/%d", newBook.ID))
	return ec.JSON(http.StatusCreated, newBook)
}

// Find books
func (c *BookCntrl) Find(ec echo.Context) (err error) {
	var req service.FindBookReq
	if err = ec.Bind(&req); err != nil {
		return err
	}
	ctx := ec.Request().Context()
	resp, err := c.Svc.Find(ctx, &req)
	if err != nil {
		return echokit.HTTPError(err)
	}
	ec.Response().Header().Add(echokit.HeaderTotalCount, resp.TotalCount)
	return ec.JSON(http.StatusOK, resp.Books)
}

// FindOne book
func (c *BookCntrl) FindOne(ec echo.Context) error {
	book, err := c.Svc.FindOne(
		ec.Request().Context(),
		ec.Param("id"),
	)
	if err != nil {
		return echokit.HTTPError(err)
	}
	return ec.JSON(http.StatusOK, book)
}

// Delete book
func (c *BookCntrl) Delete(ec echo.Context) (err error) {
	ctx := ec.Request().Context()
	id := ec.Param("id")
	if err = c.Svc.Delete(ctx, id); err != nil {
		return echokit.HTTPError(err)
	}
	return ec.NoContent(http.StatusNoContent)
}

// Update book
func (c *BookCntrl) Update(ec echo.Context) (err error) {
	var book entity.Book
	if err = ec.Bind(&book); err != nil {
		return err
	}
	ctx := ec.Request().Context()
	paramID := ec.Param("id")
	updatedBook, err := c.Svc.Update(ctx, paramID, &book)
	if err != nil {
		return echokit.HTTPError(err)
	}
	return ec.JSON(http.StatusOK, updatedBook)
}

// Patch book
func (c *BookCntrl) Patch(ec echo.Context) (err error) {
	var book entity.Book
	if err = ec.Bind(&book); err != nil {
		return err
	}
	ctx := ec.Request().Context()
	paramID := ec.Param("id")
	patchedBook, err := c.Svc.Patch(ctx, paramID, &book)
	if err != nil {
		return echokit.HTTPError(err)
	}
	return ec.JSON(http.StatusOK, patchedBook)
}