Configurable chat settings

Author: sleroq

.env.development

@@ -7,3 +7,7 @@ SSL_KEY=
 
 # /etc/letsencrypt/live/<your-domain-name>/fullchain.pem
 SSL_CERT=
+
+CHAT_MAX_HISTORY=10000
+CHAT_DEFAULT_TTL=72h
+CHAT_CLEANUP_INTERVAL=1h

.env.production

@@ -7,3 +7,7 @@ SSL_KEY=
 
 # /etc/letsencrypt/live/<your-domain-name>/fullchain.pem
 SSL_CERT=
+
+CHAT_MAX_HISTORY=10000
+CHAT_DEFAULT_TTL=72h
+CHAT_CLEANUP_INTERVAL=1h

internal/chat/chat.go

@@ -2,16 +2,18 @@ package chat
 
 import (
 	"fmt"
+	"os"
+	"strconv"
 	"sync"
 	"time"
 
 	"github.com/google/uuid"
 )
 
 const (
-	MaxHistory      = 10000
-	DefaultTTL      = 72 * time.Hour
-	CleanupInterval = 1 * time.Hour
+	DefaultMaxHistory      = 10000
+	DefaultTTL             = 72 * time.Hour
+	DefaultCleanupInterval = 1 * time.Hour
 
 	EventTypeMessage   = "message"
 	EventTypeConnected = "connected"
@@ -53,12 +55,40 @@ type Manager struct {
 	mu       sync.RWMutex
 	rooms    map[string]*Room
 	sessions map[string]*Session
+
+	maxHistory      int
+	defaultTTL      time.Duration
+	cleanupInterval time.Duration
 }
 
 func NewManager() *Manager {
+	maxHistory := DefaultMaxHistory
+	if val := os.Getenv("CHAT_MAX_HISTORY"); val != "" {
+		if i, err := strconv.Atoi(val); err == nil {
+			maxHistory = i
+		}
+	}
+
+	defaultTTL := DefaultTTL
+	if val := os.Getenv("CHAT_DEFAULT_TTL"); val != "" {
+		if d, err := time.ParseDuration(val); err == nil {
+			defaultTTL = d
+		}
+	}
+
+	cleanupInterval := DefaultCleanupInterval
+	if val := os.Getenv("CHAT_CLEANUP_INTERVAL"); val != "" {
+		if d, err := time.ParseDuration(val); err == nil {
+			cleanupInterval = d
+		}
+	}
+
 	m := &Manager{
-		rooms:    make(map[string]*Room),
-		sessions: make(map[string]*Session),
+		rooms:           make(map[string]*Room),
+		sessions:        make(map[string]*Session),
+		maxHistory:      maxHistory,
+		defaultTTL:      defaultTTL,
+		cleanupInterval: cleanupInterval,
 	}
 	go m.cleanupLoop()
 	return m
@@ -80,7 +110,7 @@ func (m *Manager) Connect(streamKey string) string {
 		m.rooms[streamKey] = &Room{
 			streamKey:    streamKey,
 			subscribers:  make(map[string]*subscriber),
-			history:      make([]Event, 0, MaxHistory),
+			history:      make([]Event, 0, m.maxHistory),
 			nextEventID:  1,
 			lastActivity: now,
 		}
@@ -198,7 +228,7 @@ func (m *Manager) Send(sessionID string, text string, displayName string) error
 	}
 	room.nextEventID++
 
-	if len(room.history) >= MaxHistory {
+	if len(room.history) >= m.maxHistory {
 		room.history = append(room.history[1:], event)
 	} else {
 		room.history = append(room.history, event)
@@ -216,7 +246,7 @@ func (m *Manager) Send(sessionID string, text string, displayName string) error
 }
 
 func (m *Manager) cleanupLoop() {
-	ticker := time.NewTicker(CleanupInterval)
+	ticker := time.NewTicker(m.cleanupInterval)
 	for range ticker.C {
 		m.cleanup()
 	}
@@ -228,14 +258,14 @@ func (m *Manager) cleanup() {
 
 	now := time.Now()
 	for id, s := range m.sessions {
-		if now.Sub(s.LastActivity) > DefaultTTL {
+		if now.Sub(s.LastActivity) > m.defaultTTL {
 			delete(m.sessions, id)
 		}
 	}
 
 	for key, r := range m.rooms {
 		r.mu.Lock()
-		if len(r.subscribers) == 0 && now.Sub(r.lastActivity) > DefaultTTL {
+		if len(r.subscribers) == 0 && now.Sub(r.lastActivity) > m.defaultTTL {
 			delete(m.rooms, key)
 		}
 		r.mu.Unlock()