-
Notifications
You must be signed in to change notification settings - Fork 0
/
users.go
95 lines (80 loc) · 2.29 KB
/
users.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
package handling
import (
"encoding/json"
"log"
"net/http"
"gopkg.in/mgo.v2/bson"
"github.com/gorilla/mux"
"github.com/heimdal-rw/chmgt/models"
)
// GetUsers returns a json response with the requested users
func (h *Handler) GetUsers(w http.ResponseWriter, r *http.Request) {
var (
users models.Users
err error
)
vars := mux.Vars(r)
users, err = h.Datasource.GetUsers(vars["id"])
if err != nil {
switch err {
case models.ErrNotFound:
APIWriteFailure(w, "no users found", http.StatusNotFound)
case models.ErrObjID:
APIWriteFailure(w, "invalid object id", http.StatusBadRequest)
default:
APIWriteFailure(w, "", http.StatusInternalServerError)
log.Println(err)
}
return
}
// mask the passwords so that they're not sent in the response
for idx := range users {
users[idx].Password = "********"
}
APIWriteSuccess(w, users)
}
// InsertUser inserts a new user into the database
func (h *Handler) InsertUser(w http.ResponseWriter, r *http.Request) {
var user models.User
if err := json.NewDecoder(r.Body).Decode(&user); err != nil {
APIWriteFailure(w, "error parsing json", http.StatusBadRequest)
return
}
id, err := h.Datasource.InsertUser(user)
if err != nil {
APIWriteFailure(w, "error inserting user", http.StatusInternalServerError)
log.Println(err)
return
}
APIWriteSuccess(w, id)
}
// UpdateUser updates a user's database record
func (h *Handler) UpdateUser(w http.ResponseWriter, r *http.Request) {
var user models.User
vars := mux.Vars(r)
if err := json.NewDecoder(r.Body).Decode(&user); err != nil {
APIWriteFailure(w, "error parsing json", http.StatusBadRequest)
return
}
user.ID = bson.ObjectIdHex(vars["id"])
if err := h.Datasource.UpdateUser(user); err != nil {
APIWriteFailure(w, "error inserting user", http.StatusInternalServerError)
log.Println(err)
return
}
APIWriteSuccess(w, nil)
}
// RemoveUser removes a user from the database
func (h *Handler) RemoveUser(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
if err := h.Datasource.RemoveUser(vars["id"]); err != nil {
if err == models.ErrNotFound {
APIWriteFailure(w, "change not found", http.StatusNotFound)
} else {
APIWriteFailure(w, "error removing user", http.StatusInternalServerError)
log.Println(err)
}
return
}
APIWriteSuccess(w, nil)
}