neko-custom/server/internal/session/manager.go

package session

import (
	"fmt"

	"github.com/gorilla/websocket"
	"github.com/kataras/go-events"
	"github.com/pion/webrtc/v2"

	"n.eko.moe/neko/internal/utils"
)

func New() *SessionManager {
	return &SessionManager{
		host:    "",
		members: make(map[string]*Session),
		emmiter: events.New(),
	}
}

type SessionManager struct {
	host    string
	members map[string]*Session
	emmiter events.EventEmmiter
}

func (m *SessionManager) New(id string, admin bool, socket *websocket.Conn) *Session {
	session := &Session{
		ID:     id,
		Admin:  admin,
		socket: socket,
	}

	m.members[id] = session
	m.emmiter.Emit("created", id, session)

	return session
}

func (m *SessionManager) IsHost(id string) bool {
	return m.host == id
}

func (m *SessionManager) HasHost() bool {
	return m.host != ""
}

func (m *SessionManager) SetHost(id string) error {
	_, ok := m.members[id]
	if ok {
		m.host = id
		m.emmiter.Emit("host", id)
		return nil
	}
	return fmt.Errorf("invalid session id %s", id)
}

func (m *SessionManager) GetHost() (*Session, bool) {
	host, ok := m.members[m.host]
	return host, ok
}

func (m *SessionManager) ClearHost() {
	id := m.host
	m.host = ""
	m.emmiter.Emit("host_cleared", id)
}

func (m *SessionManager) Has(id string) bool {
	_, ok := m.members[id]
	return ok
}

func (m *SessionManager) Get(id string) (*Session, bool) {
	session, ok := m.members[id]
	return session, ok
}

func (m *SessionManager) Set(id string, session *Session) {
	m.members[id] = session
}

func (m *SessionManager) Destroy(id string) error {
	session, ok := m.members[id]
	if ok {
		err := session.destroy()
		delete(m.members, id)
		m.emmiter.Emit("destroyed", id)
		return err
	}
	return nil
}

func (m *SessionManager) SetSocket(id string, socket *websocket.Conn) (bool, error) {
	session, ok := m.members[id]
	if ok {
		session.socket = socket
		return true, nil
	}

	return false, fmt.Errorf("invalid session id %s", id)
}

func (m *SessionManager) SetPeer(id string, peer *webrtc.PeerConnection) (bool, error) {
	session, ok := m.members[id]
	if ok {
		session.peer = peer
		return true, nil
	}

	return false, fmt.Errorf("invalid session id %s", id)
}

func (m *SessionManager) SetName(id string, name string) (bool, error) {
	session, ok := m.members[id]
	if ok {
		session.Name = name
		return true, nil
	}

	return false, fmt.Errorf("invalid session id %s", id)
}

func (m *SessionManager) Clear() error {
	return nil
}

func (m *SessionManager) Brodcast(v interface{}, exclude interface{}) error {
	if exclude != nil {
		for id, sess := range m.members {
			if in, _ := utils.ArrayIn(id, exclude); in {
				continue
			}

			if err := sess.Send(v); err != nil {
				return err
			}
		}
	} else {
		for _, sess := range m.members {
			if err := sess.Send(v); err != nil {
				return err
			}
		}
	}

	return nil
}

func (m *SessionManager) OnHost(listener func(id string)) {
	m.emmiter.On("host", func(payload ...interface{}) {
		listener(payload[0].(string))
	})
}

func (m *SessionManager) OnHostCleared(listener func(id string)) {
	m.emmiter.On("host_cleared", func(payload ...interface{}) {
		listener(payload[0].(string))
	})
}

func (m *SessionManager) OnCreated(listener func(id string, session *Session)) {
	m.emmiter.On("created", func(payload ...interface{}) {
		listener(payload[0].(string), payload[1].(*Session))
	})
}

func (m *SessionManager) OnDestroy(listener func(id string)) {
	m.emmiter.On("destroyed", func(payload ...interface{}) {
		listener(payload[0].(string))
	})
}