feat: casino framework and basic slot machine

1 files changed, 17 insertions, 5 deletions

diff --git a/internal/net/server.go b/internal/net/server.go
index 9da90ea..da1c01b 100644
--- a/internal/net/server.go
+++ b/internal/net/server.go
@@ -100,10 +100,11 @@ type Server struct {
 }
 
 type Hub struct {
-	mu       sync.Mutex
-	sessions map[*Session]bool
-	rooms    map[int]map[*Session]bool
-	onRemove func(*Session)
+	mu           sync.Mutex
+	sessions     map[*Session]bool
+	rooms        map[int]map[*Session]bool
+	onRemove     func(*Session)
+	onDisconnect func(*Session)
 }
 
 func NewHub() *Hub {
@@ -119,6 +120,12 @@ func (h *Hub) OnRemove(cb func(*Session)) {
 	h.onRemove = cb
 }
 
+func (h *Hub) OnDisconnect(cb func(*Session)) {
+	h.mu.Lock()
+	defer h.mu.Unlock()
+	h.onDisconnect = cb
+}
+
 func (h *Hub) Add(s *Session) {
 	h.mu.Lock()
 	defer h.mu.Unlock()
@@ -128,13 +135,18 @@ func (h *Hub) Add(s *Session) {
 
 func (h *Hub) Remove(s *Session) {
 	h.mu.Lock()
-	defer h.mu.Unlock()
 	if s.Player != nil && s.State == StateGame && !s.Disconnecting {
 		s.Disconnecting = true
 		s.DisconnectTicks = 10
+		onDisconnect := h.onDisconnect
+		h.mu.Unlock()
+		if onDisconnect != nil {
+			onDisconnect(s)
+		}
 		return
 	}
 	h.hardRemoveLocked(s)
+	h.mu.Unlock()
 }
 
 func (h *Hub) HardRemove(s *Session) {