package session import ( "github.com/rs/zerolog" "demodesk/neko/internal/types" "demodesk/neko/internal/types/event" "demodesk/neko/internal/types/message" ) type SessionCtx struct { id string token string logger zerolog.Logger manager *SessionManagerCtx profile types.MemberProfile state types.SessionState websocketPeer types.WebSocketPeer webrtcPeer types.WebRTCPeer } func (session *SessionCtx) ID() string { return session.id } func (session *SessionCtx) Profile() types.MemberProfile { return session.profile } func (session *SessionCtx) profileChanged() { if !session.profile.CanHost && session.IsHost() { session.manager.ClearHost() } if !session.profile.CanWatch && session.state.IsWatching { if err := session.webrtcPeer.Destroy(); err != nil { session.logger.Warn().Err(err).Msgf("webrtc destroy has failed") } } if (!session.profile.CanConnect || !session.profile.CanLogin) && session.state.IsConnected { if err := session.Disconnect("profile changed"); err != nil { session.logger.Warn().Err(err).Msgf("websocket destroy has failed") } } } func (session *SessionCtx) State() types.SessionState { return session.state } func (session *SessionCtx) IsHost() bool { return session.manager.host != nil && session.manager.host == session } // --- // webscoket // --- func (session *SessionCtx) SetWebSocketPeer(websocketPeer types.WebSocketPeer) { if session.websocketPeer != nil { if err := session.websocketPeer.Destroy(); err != nil { session.logger.Warn().Err(err).Msgf("websocket destroy has failed") } } session.websocketPeer = websocketPeer } func (session *SessionCtx) SetWebSocketConnected(connected bool) { session.state.IsConnected = connected if connected { session.manager.emmiter.Emit("connected", session) return } session.manager.emmiter.Emit("disconnected", session) session.websocketPeer = nil if session.webrtcPeer != nil { if err := session.webrtcPeer.Destroy(); err != nil { session.logger.Warn().Err(err).Msgf("webrtc destroy has failed") } } } func (session *SessionCtx) Send(v interface{}) error { if session.websocketPeer == nil { return nil } return session.websocketPeer.Send(v) } func (session *SessionCtx) Disconnect(reason string) error { if err := session.Send( message.SystemDisconnect{ Event: event.SYSTEM_DISCONNECT, Message: reason, }); err != nil { return err } if session.websocketPeer != nil { if err := session.websocketPeer.Destroy(); err != nil { return err } } if session.webrtcPeer != nil { if err := session.webrtcPeer.Destroy(); err != nil { return err } } return nil } // --- // webrtc // --- func (session *SessionCtx) SetWebRTCPeer(webrtcPeer types.WebRTCPeer) { if session.webrtcPeer != nil { if err := session.webrtcPeer.Destroy(); err != nil { session.logger.Warn().Err(err).Msgf("webrtc destroy has failed") } } session.webrtcPeer = webrtcPeer } func (session *SessionCtx) SetWebRTCConnected(webrtcPeer types.WebRTCPeer, connected bool) { if webrtcPeer != session.webrtcPeer { return } session.state.IsWatching = connected session.manager.emmiter.Emit("state_changed", session) if !connected { session.webrtcPeer = nil } } func (session *SessionCtx) GetWebRTCPeer() types.WebRTCPeer { return session.webrtcPeer }