neko/internal/session/session.go

124 lines
2.8 KiB
Go
Raw Normal View History

package session
import (
"github.com/rs/zerolog"
2020-10-30 06:07:04 +13:00
2020-10-29 07:15:48 +13:00
"demodesk/neko/internal/types"
"demodesk/neko/internal/types/event"
"demodesk/neko/internal/types/message"
)
2020-11-02 04:09:48 +13:00
type SessionCtx struct {
2020-11-26 08:02:53 +13:00
id string
logger zerolog.Logger
manager *SessionManagerCtx
profile types.MemberProfile
2020-11-26 08:02:53 +13:00
websocket_peer types.WebSocketPeer
websocket_connected bool
webrtc_peer types.WebRTCPeer
webrtc_connected bool
}
2020-11-02 04:09:48 +13:00
func (session *SessionCtx) ID() string {
return session.id
}
2020-11-02 04:09:48 +13:00
func (session *SessionCtx) Name() string {
return session.profile.Name
}
2020-12-03 04:49:51 +13:00
func (session *SessionCtx) IsAdmin() bool {
return session.profile.IsAdmin
}
2020-11-02 04:09:48 +13:00
func (session *SessionCtx) IsHost() bool {
return session.manager.host != nil && session.manager.host.ID() == session.ID()
}
2020-11-28 07:59:54 +13:00
func (session *SessionCtx) VerifySecret(secret string) bool {
return session.profile.Secret == secret
}
2020-12-03 04:49:51 +13:00
func (session *SessionCtx) IsConnected() bool {
// TODO: Refactor.
2020-12-01 10:28:13 +13:00
return session.websocket_connected// && session.webrtc_connected
}
2020-11-26 06:36:33 +13:00
func (session *SessionCtx) SetWebSocketPeer(websocket_peer types.WebSocketPeer) {
session.websocket_peer = websocket_peer
}
2020-11-26 08:02:53 +13:00
func (session *SessionCtx) SetWebSocketConnected(connected bool) {
if connected {
session.websocket_connected = true
// TODO: Refactor.
//session.manager.emmiter.Emit("websocket_connected", session)
session.manager.emmiter.Emit("connected", session)
2020-11-26 08:02:53 +13:00
} else {
session.websocket_connected = false
2020-11-26 08:26:11 +13:00
// TODO: Refactor.
//session.manager.emmiter.Emit("websocket_disconnected", session)
session.manager.emmiter.Emit("disconnected", session)
session.websocket_peer = nil
2020-11-26 08:02:53 +13:00
}
}
2020-11-26 06:41:40 +13:00
func (session *SessionCtx) SetWebRTCPeer(webrtc_peer types.WebRTCPeer) {
session.webrtc_peer = webrtc_peer
}
2020-11-26 08:02:53 +13:00
func (session *SessionCtx) SetWebRTCConnected(connected bool) {
if connected {
session.webrtc_connected = true
session.manager.emmiter.Emit("webrtc_connected", session)
2020-11-26 08:02:53 +13:00
} else {
session.webrtc_connected = false
session.manager.emmiter.Emit("webrtc_disconnected", session)
2020-11-02 09:03:25 +13:00
2020-11-26 08:26:11 +13:00
session.webrtc_peer = nil
2020-11-02 09:03:25 +13:00
}
2020-11-02 08:53:25 +13:00
}
2020-11-02 04:09:48 +13:00
func (session *SessionCtx) Disconnect(reason string) error {
2020-11-26 10:06:13 +13:00
if err := session.Send(
2020-12-02 07:44:23 +13:00
message.SystemDisconnect{
2020-11-19 08:30:33 +13:00
Event: event.SYSTEM_DISCONNECT,
Message: reason,
2020-11-26 10:06:13 +13:00
}); err != nil {
return err
}
if session.websocket_peer != nil {
if err := session.websocket_peer.Destroy(); err != nil {
return err
}
}
if session.webrtc_peer != nil {
if err := session.webrtc_peer.Destroy(); err != nil {
return err
}
}
return nil
}
2020-11-02 04:09:48 +13:00
func (session *SessionCtx) Send(v interface{}) error {
2020-11-26 06:36:33 +13:00
if session.websocket_peer == nil {
return nil
}
2020-11-01 09:46:29 +13:00
2020-11-26 06:36:33 +13:00
return session.websocket_peer.Send(v)
}
2020-11-02 04:09:48 +13:00
func (session *SessionCtx) SignalAnswer(sdp string) error {
2020-11-26 06:41:40 +13:00
if session.webrtc_peer == nil {
return nil
}
2020-11-01 09:46:29 +13:00
2020-11-26 06:41:40 +13:00
return session.webrtc_peer.SignalAnswer(sdp)
}