/
device.go
127 lines (105 loc) · 3.23 KB
/
device.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
package main
import (
"database/sql"
"encoding/json"
"net/http"
log "github.com/Sirupsen/logrus"
"github.com/gorilla/context"
"github.com/gorilla/mux"
"github.com/irrenhaus/pushmearound_server/httpresponse"
"github.com/irrenhaus/pushmearound_server/models"
)
func DeviceCreateHandler(resp http.ResponseWriter, req *http.Request) {
platform := req.FormValue("platform")
name := req.FormValue("name")
if name == "" || !models.DevicePlatforms[platform] {
httpresponse.BadRequest("Please specify device name and a valid platform").WriteJSON(resp)
return
}
user, ok := context.Get(req, ContextKeyUser).(models.User)
if !ok || user.ID == 0 {
httpresponse.InternalServerError("No user object found").WriteJSON(resp)
return
}
device := models.Device{
UserID: user.ID,
Platform: platform,
Name: name,
Options: models.DeviceOptions{
PushNotifications: true,
},
}
if err := device.Create(DB); err != nil {
log.WithFields(log.Fields{"error": err, "device": device.Name}).Warn("Could not append device")
httpresponse.InternalServerError("Creating the device failed").WriteJSON(resp)
return
}
response := httpresponse.Success("")
response.Data = map[string]string{
"device_id": device.ID,
}
response.WriteJSON(resp)
}
func DeviceOptionsHandler(resp http.ResponseWriter, req *http.Request) {
user, ok := context.Get(req, ContextKeyUser).(models.User)
if !ok || user.ID == 0 {
httpresponse.InternalServerError("No user object found").WriteJSON(resp)
return
}
var options interface{}
decoder := json.NewDecoder(req.Body)
if err := decoder.Decode(&options); err != nil {
httpresponse.BadRequest("Could not parse JSON request").WriteJSON(resp)
return
}
deviceID := req.FormValue("device")
if deviceID == "" {
httpresponse.BadRequest("No device specified").WriteJSON(resp)
return
}
device, err := models.FindDevice(DB, deviceID)
if err != nil {
if err != sql.ErrNoRows {
log.WithFields(log.Fields{"user": user.ID, "device": deviceID, "error": err}).Error("SQL error while loading device")
}
httpresponse.NotFound("No such device").WriteJSON(resp)
return
}
if device.UserID != user.ID {
httpresponse.NotFound("No such device").WriteJSON(resp)
return
}
if err := device.Options.ParseJSONMap(options.(map[string]interface{})); err != nil {
httpresponse.BadRequest(err.Error()).WriteJSON(resp)
return
}
httpresponse.Success("").WriteJSON(resp)
}
func DeviceGetHandler(resp http.ResponseWriter, req *http.Request) {
user, ok := context.Get(req, ContextKeyUser).(models.User)
if !ok || user.ID == 0 {
httpresponse.InternalServerError("No user object found").WriteJSON(resp)
return
}
vars := mux.Vars(req)
deviceID, ok := vars["id"]
if !ok {
httpresponse.BadRequest("Missing device ID parameter").WriteJSON(resp)
return
}
device, err := models.FindDevice(DB, deviceID)
if err != nil {
if err != sql.ErrNoRows {
log.WithFields(log.Fields{"user": user.ID, "device": deviceID, "error": err}).Error("SQL error while finding device")
}
httpresponse.NotFound("No such device").WriteJSON(resp)
return
}
if device.UserID != user.ID {
httpresponse.NotFound("No such device").WriteJSON(resp)
return
}
response := httpresponse.New()
response.Data = device
response.WriteJSON(resp)
}