iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

session id to session.

Browse Source
This commit is contained in:
Miroslav Šedivý 2020-11-14 16:03:12 +01:00
parent fdf5839547
commit e5eaf5c60c
3 changed files with 12 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
internal/websocket/handler/handler.go
View File

@ -42,7 +42,7 @@ type MessageHandlerCtx struct {
locked bool
}
func (h *MessageHandlerCtx) Connected(id string, socket types.WebSocket) (bool, string) {
func (h *MessageHandlerCtx) Connected(session types.Session, socket types.WebSocket) (bool, string) {
address := socket.Address()
if address != "" {
ok, banned := h.banned[address]
@ -54,12 +54,9 @@ func (h *MessageHandlerCtx) Connected(id string, socket types.WebSocket) (bool,
h.logger.Debug().Msg("no remote address")
}
if h.locked {
session, ok := h.sessions.Get(id)
if !ok || !session.Admin() {
h.logger.Debug().Msg("server locked")
return false, "locked"
}
if h.locked && !session.Admin(){
h.logger.Debug().Msg("server locked")
return false, "locked"
}
return true, ""
@ -74,17 +71,12 @@ func (h *MessageHandlerCtx) Disconnected(id string) error {
return h.sessions.Destroy(id)
}
func (h *MessageHandlerCtx) Message(id string, raw []byte) error {
func (h *MessageHandlerCtx) Message(session types.Session, raw []byte) error {
header := message.Message{}
if err := json.Unmarshal(raw, &header); err != nil {
return err
}
session, ok := h.sessions.Get(id)
if !ok {
return errors.Errorf("unknown session id %s", id)
}
switch header.Event {
// Signal Events
case event.SIGNAL_ANSWER:

6
internal/websocket/manager.go
View File

@ -149,13 +149,13 @@ func (ws *WebSocketManagerCtx) Upgrade(w http.ResponseWriter, r *http.Request) e
// }
socket := &WebSocketCtx{
id: session.ID(),
session: session,
ws: ws,
address: ip,
connection: connection,
}
ok, reason := ws.handler.Connected(session.ID(), socket)
ok, reason := ws.handler.Connected(session, socket)
if !ok {
// TODO: Refactor
if err = connection.WriteJSON(message.Disconnect{
@ -226,7 +226,7 @@ func (ws *WebSocketManagerCtx) handle(connection *websocket.Conn, session types.
Str("raw", string(raw)).
Msg("received message from client")
if err := ws.handler.Message(session.ID(), raw); err != nil {
if err := ws.handler.Message(session, raw); err != nil {
ws.logger.Error().Err(err).Msg("message handler has failed")
}
case <-cancel:

6
internal/websocket/websocket.go
View File

@ -6,10 +6,12 @@ import (
"sync"
"github.com/gorilla/websocket"
"demodesk/neko/internal/types"
)
type WebSocketCtx struct {
id string
session types.Session
address string
ws *WebSocketManagerCtx
connection *websocket.Conn
@ -40,7 +42,7 @@ func (socket *WebSocketCtx) Send(v interface{}) error {
}
socket.ws.logger.Debug().
Str("session", socket.id).
Str("session", socket.session.ID()).
Str("address", socket.connection.RemoteAddr().String()).
Str("raw", string(raw)).
Msg("sending message to client")