initial commit - from neko open source repository..

internal/session/session.go

@@ -0,0 +1,137 @@
+package session
+
+import (
+	"sync"
+
+	"github.com/rs/zerolog"
+	"n.eko.moe/neko/internal/types"
+	"n.eko.moe/neko/internal/types/event"
+	"n.eko.moe/neko/internal/types/message"
+)
+
+type Session struct {
+	logger    zerolog.Logger
+	id        string
+	name      string
+	admin     bool
+	muted     bool
+	connected bool
+	manager   *SessionManager
+	socket    types.WebSocket
+	peer      types.Peer
+	mu        sync.Mutex
+}
+
+func (session *Session) ID() string {
+	return session.id
+}
+
+func (session *Session) Name() string {
+	return session.name
+}
+
+func (session *Session) Admin() bool {
+	return session.admin
+}
+
+func (session *Session) Muted() bool {
+	return session.muted
+}
+
+func (session *Session) Connected() bool {
+	return session.connected
+}
+
+func (session *Session) Address() string {
+	if session.socket == nil {
+		return ""
+	}
+	return session.socket.Address()
+}
+
+func (session *Session) Member() *types.Member {
+	return &types.Member{
+		ID:    session.id,
+		Name:  session.name,
+		Admin: session.admin,
+		Muted: session.muted,
+	}
+}
+
+func (session *Session) SetMuted(muted bool) {
+	session.muted = muted
+}
+
+func (session *Session) SetName(name string) error {
+	session.name = name
+	return nil
+}
+
+func (session *Session) SetSocket(socket types.WebSocket) error {
+	session.socket = socket
+	return nil
+}
+
+func (session *Session) SetPeer(peer types.Peer) error {
+	session.peer = peer
+	return nil
+}
+
+func (session *Session) SetConnected(connected bool) error {
+	session.connected = connected
+	if connected {
+		session.manager.emmiter.Emit("connected", session.id, session)
+	}
+	return nil
+}
+
+func (session *Session) Kick(reason string) error {
+	if session.socket == nil {
+		return nil
+	}
+	if err := session.socket.Send(&message.Disconnect{
+		Event:   event.SYSTEM_DISCONNECT,
+		Message: reason,
+	}); err != nil {
+		return err
+	}
+
+	return session.destroy()
+}
+
+func (session *Session) Send(v interface{}) error {
+	if session.socket == nil {
+		return nil
+	}
+	return session.socket.Send(v)
+}
+
+func (session *Session) Write(v interface{}) error {
+	if session.socket == nil {
+		return nil
+	}
+	return session.socket.Send(v)
+}
+
+func (session *Session) SignalAnswer(sdp string) error {
+	if session.peer == nil {
+		return nil
+	}
+	return session.peer.SignalAnswer(sdp)
+}
+
+func (session *Session) destroy() error {
+	if session.socket != nil {
+		if err := session.socket.Destroy(); err != nil {
+			return err
+		}
+	}
+
+	if session.peer != nil {
+		if err := session.peer.Destroy(); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}