79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package websocket
|
|
|
|
import (
|
|
"n.eko.moe/neko/internal/types"
|
|
"n.eko.moe/neko/internal/types/event"
|
|
"n.eko.moe/neko/internal/types/message"
|
|
)
|
|
|
|
func (h *MessageHandler) SessionCreated(id string, session types.Session) error {
|
|
if err := session.Send(message.Identity{
|
|
Event: event.IDENTITY_PROVIDE,
|
|
ID: id,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (h *MessageHandler) SessionConnected(id string, session types.Session) error {
|
|
// send list of members to session
|
|
if err := session.Send(message.MembersList{
|
|
Event: event.MEMBER_LIST,
|
|
Memebers: h.sessions.Members(),
|
|
}); err != nil {
|
|
h.logger.Warn().Str("id", id).Err(err).Msgf("sending event %s has failed", event.MEMBER_LIST)
|
|
return err
|
|
}
|
|
|
|
// tell session there is a host
|
|
host, ok := h.sessions.GetHost()
|
|
if ok {
|
|
if err := session.Send(message.Control{
|
|
Event: event.CONTROL_LOCKED,
|
|
ID: host.ID(),
|
|
}); err != nil {
|
|
h.logger.Warn().Str("id", id).Err(err).Msgf("sending event %s has failed", event.CONTROL_LOCKED)
|
|
return err
|
|
}
|
|
}
|
|
|
|
// let everyone know there is a new session
|
|
if err := h.sessions.Brodcast(
|
|
message.Member{
|
|
Event: event.MEMBER_CONNECTED,
|
|
Member: session.Member(),
|
|
}, nil); err != nil {
|
|
h.logger.Warn().Err(err).Msgf("brodcasting event %s has failed", event.CONTROL_RELEASE)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (h *MessageHandler) SessionDestroyed(id string) error {
|
|
// clear host if exists
|
|
if h.sessions.IsHost(id) {
|
|
h.sessions.ClearHost()
|
|
if err := h.sessions.Brodcast(message.Control{
|
|
Event: event.CONTROL_RELEASE,
|
|
ID: id,
|
|
}, nil); err != nil {
|
|
h.logger.Warn().Err(err).Msgf("brodcasting event %s has failed", event.CONTROL_RELEASE)
|
|
}
|
|
}
|
|
|
|
// let everyone know session disconnected
|
|
if err := h.sessions.Brodcast(
|
|
message.MemberDisconnected{
|
|
Event: event.MEMBER_DISCONNECTED,
|
|
ID: id,
|
|
}, nil); err != nil {
|
|
h.logger.Warn().Err(err).Msgf("brodcasting event %s has failed", event.MEMBER_DISCONNECTED)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|