151 lines
3.4 KiB
Go
151 lines
3.4 KiB
Go
package link
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/jackc/pgx/v4/pgxpool"
|
|
"github.com/labstack/echo/v4"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
)
|
|
|
|
func redirectHandler(ctx echo.Context, serv Service) error {
|
|
linkId := ctx.Param("id")
|
|
linkPassword := ctx.QueryParam("password")
|
|
|
|
redirectUrl, err := serv.AccessLink(linkId, linkPassword)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return ctx.Redirect(http.StatusSeeOther, redirectUrl.String())
|
|
}
|
|
|
|
func creationHandler(ctx echo.Context, serv Service) error {
|
|
var model CreationModel
|
|
if err := json.NewDecoder(ctx.Request().Body).Decode(&model); err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Invalid data format.")
|
|
}
|
|
|
|
entity, err := model.MapModelToEntity()
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Invalid URL.")
|
|
}
|
|
|
|
if err = serv.Create(entity); err != nil {
|
|
return err
|
|
}
|
|
|
|
return ctx.NoContent(http.StatusCreated)
|
|
}
|
|
|
|
func retrievalByIdHandler(ctx echo.Context, serv Service) error {
|
|
linkId := ctx.Param("id")
|
|
|
|
l, err := serv.GetById(linkId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return ctx.JSON(http.StatusOK, MapEntityToModel(l))
|
|
}
|
|
|
|
func allRetrievalHandler(ctx echo.Context, serv Service) error {
|
|
limit := 20
|
|
if v := ctx.QueryParam("limit"); v != "" {
|
|
num, err := strconv.Atoi(v)
|
|
if err != nil || num < 0 {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Invalid limit value.")
|
|
}
|
|
|
|
limit = num
|
|
}
|
|
|
|
offset := 0
|
|
if v := ctx.QueryParam("offset"); v != "" {
|
|
num, err := strconv.Atoi(v)
|
|
if err != nil || num < 0 {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Invalid offset value.")
|
|
}
|
|
|
|
offset = num
|
|
}
|
|
|
|
links, err := serv.GetAll(limit, offset)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
models := make([]ResourceModel, len(links))
|
|
for i, v := range links {
|
|
models[i] = MapEntityToModel(v)
|
|
}
|
|
|
|
return ctx.JSON(http.StatusOK, models)
|
|
}
|
|
|
|
func updateHandler(ctx echo.Context, serv Service) error {
|
|
linkId := ctx.Param("id")
|
|
|
|
var model UpdateModel
|
|
if err := json.NewDecoder(ctx.Request().Body).Decode(&model); err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Invalid data format.")
|
|
}
|
|
|
|
var parsedUrl *url.URL
|
|
parsedUrl, err := url.Parse(model.RedirectURL)
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Invalid URL value.")
|
|
}
|
|
|
|
if err := serv.UpdateById(linkId, struct {
|
|
Name string
|
|
Password string
|
|
RedirectURL *url.URL
|
|
}{Name: model.Name, Password: model.Password, RedirectURL: parsedUrl}); err != nil {
|
|
return err
|
|
}
|
|
|
|
return ctx.NoContent(http.StatusOK)
|
|
}
|
|
|
|
func removalHandler(ctx echo.Context, serv Service) error {
|
|
linkId := ctx.Param("id")
|
|
|
|
if err := serv.DeleteById(linkId); err != nil {
|
|
return err
|
|
}
|
|
|
|
return ctx.NoContent(http.StatusNoContent)
|
|
}
|
|
|
|
func AddHandlers(server *echo.Echo, pool *pgxpool.Pool) {
|
|
serv := NewService(NewRepository(pool))
|
|
|
|
linksGroup := server.Group("/links")
|
|
exactLinkGroup := linksGroup.Group("/:id")
|
|
|
|
linksGroup.POST("", func(ctx echo.Context) error {
|
|
return creationHandler(ctx, serv)
|
|
})
|
|
|
|
linksGroup.GET("", func(ctx echo.Context) error {
|
|
return allRetrievalHandler(ctx, serv)
|
|
})
|
|
|
|
exactLinkGroup.GET("", func(ctx echo.Context) error {
|
|
return retrievalByIdHandler(ctx, serv)
|
|
})
|
|
|
|
exactLinkGroup.PATCH("", func(ctx echo.Context) error {
|
|
return updateHandler(ctx, serv)
|
|
})
|
|
|
|
exactLinkGroup.DELETE("", func(ctx echo.Context) error {
|
|
return removalHandler(ctx, serv)
|
|
})
|
|
|
|
server.GET("/:id", func(ctx echo.Context) error {
|
|
return redirectHandler(ctx, serv)
|
|
})
|
|
}
|