Feat(booking): Added updateBookingStatus
All checks were successful
ktm-booking-bot/ktm-booking-bot/pipeline/head This commit looks good

This commit is contained in:
2022-09-29 00:26:12 +08:00
parent f3f4b93ebd
commit 6aa67554ee
6 changed files with 186 additions and 0 deletions

View File

@@ -83,3 +83,34 @@ func (env *Env) deleteBooking(
return &newBooking, nil
}
func (env *Env) updateBookingStatus(
user *user.User,
bookingIDStr string,
status string,
) (*Booking, error) {
var newBooking Booking
bookingID, err := uuid.Parse(bookingIDStr)
if err != nil {
log.Println(err)
return nil, errors.New("invalid uuid")
}
err = env.DB.Where(&Booking{ID: bookingID}).Where("user_id = ?", user.ID).First(&newBooking).Error
if err != nil {
log.Println(err)
return nil, errors.New("failed retrieve booking")
}
newBooking.Status = status
err = env.DB.Save(&newBooking).Error
if err != nil {
log.Println(err)
return nil, errors.New("failed to update booking status")
}
return &newBooking, nil
}

View File

@@ -123,3 +123,37 @@ func (env *Env) deleteBookingRoute(w http.ResponseWriter, r *http.Request) {
render.Render(w, r, env.NewBookingResponse(booking))
}
// Change booking status
// @Summary Changing booking status
// @Description Description
// @Tags ktmtrainbot Booking
// @Accept json
// @Produce json
// @Param bookingID path string true "Booking ID"
// @Param status path string true "Status"
// @Success 200 {object} BookingResponse
// @Failure 400 {object} common.ErrResponse
// @Router /api/v1/ktmtrainbot/booking/{bookingID}/{status} [put]
func (env *Env) updateBookingStatusRoute(w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
currUser, ok := ctx.Value(user.UserContextKey).(*user.User)
if !ok {
err := errors.New("user not logged in")
render.Render(w, r, common.ErrInternalError(err))
return
}
_ = currUser
bookingID := chi.URLParam(r, "bookingID")
status := chi.URLParam(r, "status")
booking, err := env.updateBookingStatus(currUser, bookingID, status)
if err != nil {
render.Render(w, r, common.ErrInvalidRequest(err))
return
}
render.Render(w, r, env.NewBookingResponse(booking))
}

View File

@@ -26,6 +26,7 @@ func KTMTrainBotRoutes(db *gorm.DB) chi.Router {
checkLoggedInUserGroup.Get("/booking", env.getBookingRoute)
checkLoggedInUserGroup.Post("/booking", env.createBookingRoute)
checkLoggedInUserGroup.Delete("/booking/{bookingID}", env.deleteBookingRoute)
checkLoggedInUserGroup.Put("/booking/{bookingID}/{status}", env.updateBookingStatusRoute)
return r
}