-
Notifications
You must be signed in to change notification settings - Fork 0
/
user_controller.go
152 lines (143 loc) · 3.86 KB
/
user_controller.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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
package api
import (
"errors"
"fmt"
"github.com/gofiber/fiber"
"github.com/zackartz/blog-api/auth"
"github.com/zackartz/blog-api/models"
"github.com/zackartz/blog-api/utils"
"github.com/zackartz/blog-api/utils/formaterror"
"net/http"
"strconv"
)
func (s *Server) GetAllUsers(ctx *fiber.Ctx) {
user := models.User{}
users, err := user.GetAllUsers(s.DB)
if err != nil {
utils.Error(ctx, http.StatusBadRequest, err)
return
}
utils.JSON(ctx, http.StatusOK, users)
}
func (s *Server) CreateUser(ctx *fiber.Ctx) {
user := new(models.User)
if err := ctx.BodyParser(user); err != nil {
utils.Error(ctx, http.StatusUnprocessableEntity, err)
return
}
fmt.Printf("%s", user.Password)
err := user.Validate("")
if err != nil {
utils.Error(ctx, http.StatusUnprocessableEntity, err)
return
}
pwBytes, err := models.Hash(user.Password)
user.Password = string(pwBytes)
if err != nil {
utils.Error(ctx, http.StatusUnprocessableEntity, err)
return
}
userCreated, err := user.CreateUser(s.DB)
fmt.Printf("%s", user.Password)
if err != nil {
formattedError := formaterror.FormatError(err.Error())
utils.Error(ctx, http.StatusInternalServerError, formattedError)
return
}
ctx.Set("Location", fmt.Sprintf("%s%s/%d", ctx.Hostname(), ctx.OriginalURL(), userCreated.ID))
utils.JSON(ctx, http.StatusCreated, userCreated)
}
func (s *Server) GetUser(ctx *fiber.Ctx) {
id := ctx.Params("id")
uid, err := strconv.ParseUint(id, 10, 32)
if err != nil {
utils.JSON(ctx, http.StatusBadRequest, err)
return
}
user := &models.User{}
user, err = user.GetUserByID(s.DB, uint32(uid))
if err != nil {
utils.Error(ctx, http.StatusTeapot, err)
return
}
Return(ctx, user)
}
func (s *Server) GetMe(ctx *fiber.Ctx) {
id, err := auth.ExtractTokenID(ctx)
if err != nil {
utils.JSON(ctx, http.StatusUnauthorized, err)
return
}
user := &models.User{}
user, err = user.GetUserByID(s.DB, id)
if err != nil {
utils.Error(ctx, http.StatusBadRequest, err)
return
}
Return(ctx, user)
}
func (s *Server) UpdateUser(ctx *fiber.Ctx) {
uid, err := strconv.ParseUint(ctx.Params("id"), 10, 32)
if err != nil {
utils.Error(ctx, http.StatusBadRequest, err)
return
}
user := new(models.User)
if err := ctx.BodyParser(user); err != nil {
utils.Error(ctx, http.StatusUnprocessableEntity, err)
return
}
tokenId, err := auth.ExtractTokenID(ctx)
if err != nil {
utils.Error(ctx, http.StatusUnauthorized, errors.New(http.StatusText(http.StatusUnauthorized)))
return
}
if tokenId != uint32(uid) {
utils.Error(ctx, http.StatusUnauthorized, errors.New(http.StatusText(http.StatusUnauthorized)))
return
}
user.Prepare()
err = user.Validate("update")
if err != nil {
utils.Error(ctx, http.StatusUnprocessableEntity, err)
return
}
updatedUser, err := user.UpdateUser(s.DB, uint32(uid))
if err != nil {
formattedError := formaterror.FormatError(err.Error())
utils.Error(ctx, http.StatusInternalServerError, formattedError)
return
}
utils.JSON(ctx, http.StatusOK, updatedUser)
}
func (s *Server) DeleteUser(ctx *fiber.Ctx) {
user := new(models.User)
uid, err := strconv.ParseUint(ctx.Params("id"), 10, 32)
if err != nil {
utils.Error(ctx, http.StatusBadRequest, err)
return
}
tokenId, err := auth.ExtractTokenID(ctx)
if err != nil {
utils.Error(ctx, http.StatusUnauthorized, errors.New(http.StatusText(http.StatusUnauthorized)))
return
}
if tokenId != 0 && tokenId != uint32(uid) {
utils.Error(ctx, http.StatusUnauthorized, errors.New(http.StatusText(http.StatusUnauthorized)))
return
}
_, err = user.DeleteUserByID(s.DB, uint32(uid))
if err != nil {
utils.Error(ctx, http.StatusInternalServerError, err)
return
}
ctx.Set("Entity", fmt.Sprintf("%d", uid))
utils.JSON(ctx, http.StatusNoContent, "")
}
func Return(ctx *fiber.Ctx, user *models.User) {
user.Password = ""
if user.ShowEmail == false {
user.Email = ""
}
utils.JSON(ctx, http.StatusOK, user)
}