/
wshandlers.go
112 lines (96 loc) · 2.76 KB
/
wshandlers.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
package main
import (
"net/http"
"github.com/gorilla/websocket"
"github.com/iryonetwork/network-poc/requests"
)
type wsStruct struct {
*handlers
}
var upgrader = websocket.Upgrader{
CheckOrigin: checkOrigin,
}
func checkOrigin(r *http.Request) bool {
// TODO: check origin??
return true
}
func (h *handlers) wsHandler(w http.ResponseWriter, r *http.Request) {
ws := wsStruct{h}
r.ParseForm()
// Upgrade connection
c, err := upgrader.Upgrade(w, r, nil)
if err != nil {
h.log.Debugf("Error upgrading request: %v", err)
return
}
defer c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "Closing connection"))
// Authentication
token := r.Form["token"][0]
h.log.Debugf("Token: %s", token)
if token == "" {
h.log.Debugf("Token field empty")
c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "No token recieved"))
return
}
user, exists := h.token.ValidateGetInfo(token)
if !exists {
h.log.Debugf("Invalid token")
c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "Unatuhorized"))
return
}
h.log.Debugf("Token ok")
c.WriteMessage(websocket.BinaryMessage, []byte("Authorized"))
// Add user to hub
h.hub.Register(c, user)
defer h.hub.Unregister(c, user)
for {
_, message, err := c.ReadMessage()
if err != nil {
if websocket.IsUnexpectedCloseError(err, websocket.CloseNormalClosure) {
h.log.Debugf("Error reading message: %v", err)
} else {
h.log.Debugf("User %s disconnected", user)
}
break
}
err = ws.HandleRequest(message, user, h.db)
if err != nil {
h.log.Debugf("Error HandlingRequest: %v", err)
}
}
}
// notify all users that are online and connected to `owner` that new file has been uploaded
func (s *storage) notifyConnectedUpload(owner, uploader, fileID string) {
// generate message
notification := s.notifyUploadRequest(owner, fileID)
// list users to notify
connected, err := s.eos.ListConnected(owner)
if err != nil {
s.log.Printf("Error getting list of connections, %v", err)
}
connected = append(connected, owner)
// notify all connected users and skip the creator of this request
for _, v := range connected {
if v == uploader {
continue
}
// if user is connected send notification
if s.hub.Connected(v) {
c, err := s.hub.GetConn(v)
if err != nil {
s.log.Printf("Error getting ws.conn; %v", err)
}
c.WriteMessage(websocket.BinaryMessage, notification)
}
}
}
func (s *storage) notifyUploadRequest(owner, fileID string) []byte {
req := requests.NewReq("NewUpload")
req.Append("user", owner)
req.Append("fileID", fileID)
out, err := req.Encode()
if err != nil {
s.log.Printf("Error encoding request NewUpload; %v", err)
}
return out
}