package saving_account import ( "github.com/labstack/echo/v4" "gitnet.fr/deblan/budget/backend/controller/crud" "gitnet.fr/deblan/budget/backend/message" "gitnet.fr/deblan/budget/database/model" "gorm.io/gorm" ) type Controller struct { crud *crud.Controller } func (ctrl *Controller) Config() crud.Configuration { return crud.Configuration{ Table: "saving_accounts", Model: model.SavingAccount{}, Models: []model.SavingAccount{}, ValidOrders: []string{"id", "label", "blocked_amount", "released_amount"}, DefaultLimit: 20, CreateModel: func() interface{} { return new(model.SavingAccount) }, } } func New(e *echo.Echo) *Controller { c := Controller{ crud: crud.New(), } listRoute := "/api/saving_account" itemRoute := "/api/saving_account/:id" e.GET(listRoute, c.List) e.POST(listRoute, c.Create) e.GET(itemRoute, c.Show) e.POST(itemRoute, c.Update) e.DELETE(itemRoute, c.Delete) return &c } func (ctrl *Controller) List(c echo.Context) error { if nil == model.LoadApiUser(c) { return c.JSON(403, crud.Error{Code: 403, Message: message.LoginRequired}) } return ctrl.crud.With(ctrl.Config()).List(c) } func (ctrl *Controller) Show(c echo.Context) error { if nil == model.LoadApiUser(c) { return c.JSON(403, crud.Error{Code: 403, Message: message.LoginRequired}) } return ctrl.crud.With(ctrl.Config()).Show(c) } func (ctrl *Controller) Delete(c echo.Context) error { if nil == model.LoadApiUser(c) { return c.JSON(403, crud.Error{Code: 403, Message: message.LoginRequired}) } return ctrl.crud.With(ctrl.Config()).Delete(c) } func (ctrl *Controller) Create(c echo.Context) error { if nil == model.LoadApiUser(c) { return c.JSON(403, crud.Error{Code: 403, Message: message.LoginRequired}) } type body struct { Label string `json:"label" form:"label" validate:"required"` BlockedAmount float64 `json:"blocked_amount" form:"blocked_amount"` ReleasedAmount float64 `json:"released_amount" form:"released_amount"` } return ctrl.crud.With(ctrl.Config()).Create(c, new(body), func(db *gorm.DB, v interface{}) (interface{}, error) { value := v.(*body) item := model.NewSavingAccount(value.Label, value.BlockedAmount, value.ReleasedAmount) db.Model(ctrl.crud.Config.Model).Create(&item) return item, nil }) } func (ctrl *Controller) Update(c echo.Context) error { if nil == model.LoadApiUser(c) { return c.JSON(403, crud.Error{Code: 403, Message: message.LoginRequired}) } type body struct { Label string `json:"label" form:"label" validate:"required"` BlockedAmount float64 `json:"blocked_amount" form:"blocked_amount"` ReleasedAmount float64 `json:"released_amount" form:"released_amount"` } return ctrl.crud.With(ctrl.Config()).Update(c, new(body), func(db *gorm.DB, a, b interface{}) (interface{}, error) { item := a.(*model.SavingAccount) update := b.(*body) item.Label = update.Label item.ReleasedAmount = update.ReleasedAmount item.BlockedAmount = update.BlockedAmount db.Model(ctrl.crud.Config.Model).Where("id = ?", item.ID).Save(&item) return item, nil }) }