/
user_handler.go
85 lines (75 loc) · 1.85 KB
/
user_handler.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package api
import (
"errors"
"github.com/joshdstockdale/go-booking/db"
"github.com/joshdstockdale/go-booking/types"
"go.mongodb.org/mongo-driver/mongo"
"github.com/gofiber/fiber/v2"
)
type UserHandler struct {
userStore db.UserStore
}
func NewUserHandler(userStore db.UserStore) *UserHandler {
return &UserHandler{
userStore: userStore,
}
}
func (h *UserHandler) HandlePutUser(c *fiber.Ctx) error {
var (
//values bson.M
params types.UpdateUserParams
userID = c.Params("id")
)
if err := c.BodyParser(¶ms); err != nil {
return ErrBadRequest()
}
if err := h.userStore.UpdateUser(c.Context(), userID, params); err != nil {
return err
}
return c.JSON(map[string]string{"updated": userID})
}
func (h *UserHandler) HandleDeleteUser(c *fiber.Ctx) error {
userID := c.Params("id")
if err := h.userStore.DeleteUser(c.Context(), userID); err != nil {
return err
}
return c.JSON(map[string]string{"deleted": userID})
}
func (h *UserHandler) HandlePostUser(c *fiber.Ctx) error {
var params types.InsertUserParams
if err := c.BodyParser(¶ms); err != nil {
return ErrBadRequest()
}
if errors := params.Validate(); len(errors) > 0 {
return c.JSON(errors)
}
user, err := types.NewUserFromParams(params)
if err != nil {
return err
}
insertedUser, err := h.userStore.InsertUser(c.Context(), user)
if err != nil {
return err
}
return c.JSON(insertedUser)
}
func (h *UserHandler) HandleGetUser(c *fiber.Ctx) error {
var (
id = c.Params("id")
)
user, err := h.userStore.GetUserByID(c.Context(), id)
if err != nil {
if errors.Is(err, mongo.ErrNoDocuments) {
return c.JSON(map[string]string{"error": "Not Found"})
}
return err
}
return c.JSON(user)
}
func (h *UserHandler) HandleGetUsers(c *fiber.Ctx) error {
users, err := h.userStore.GetUsers(c.Context())
if err != nil {
return ErrNotFound("User")
}
return c.JSON(users)
}