134 lines
3.3 KiB
Go
134 lines
3.3 KiB
Go
package worker
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"sync"
|
|
"time"
|
|
|
|
"code.nonshy.com/nonshy/website/pkg/config"
|
|
"code.nonshy.com/nonshy/website/pkg/log"
|
|
)
|
|
|
|
// ChatStatistics is the json result of the BareRTC /api/statistics endpoint.
|
|
type ChatStatistics struct {
|
|
UserCount int
|
|
Usernames []string
|
|
Cameras struct {
|
|
Blue int
|
|
Red int
|
|
}
|
|
}
|
|
|
|
// GetChatStatistics returns the latest (cached) chat statistics.
|
|
func GetChatStatistics() *ChatStatistics {
|
|
chatStatisticsMu.RLock()
|
|
defer chatStatisticsMu.RUnlock()
|
|
return cachedChatStatistics
|
|
}
|
|
|
|
// SetChatStatistics updates the cached chat statistics, holding a write lock briefly.
|
|
func SetChatStatistics(stats *ChatStatistics) {
|
|
chatStatisticsMu.Lock()
|
|
defer chatStatisticsMu.Unlock()
|
|
cachedChatStatistics = stats
|
|
}
|
|
|
|
// IsOnline returns whether the username is currently logged-in to chat.
|
|
func (cs *ChatStatistics) IsOnline(username string) bool {
|
|
for _, user := range cs.Usernames {
|
|
if user == username {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// SetOnlineNow patches the current ChatStatistics to mark a user as online immediately, e.g.
|
|
// because the main site has just sent them to the chat with a JWT token.
|
|
func (cs *ChatStatistics) SetOnlineNow(username string) {
|
|
if !cs.IsOnline(username) {
|
|
chatStatisticsMu.Lock()
|
|
defer chatStatisticsMu.Unlock()
|
|
cs.Usernames = append(cs.Usernames, username)
|
|
}
|
|
}
|
|
|
|
type UserOnChatMap map[string]bool
|
|
|
|
// MapUsersOnline returns a hashmap of usernames to online status.
|
|
func (cs *ChatStatistics) MapUsersOnline(usernames []string) UserOnChatMap {
|
|
var result = UserOnChatMap{}
|
|
for _, user := range cs.Usernames {
|
|
result[user] = true
|
|
}
|
|
return result
|
|
}
|
|
|
|
// Get a result from the UserOnChatMap.
|
|
func (m UserOnChatMap) Get(username string) bool {
|
|
return m[username]
|
|
}
|
|
|
|
var (
|
|
cachedChatStatistics = &ChatStatistics{}
|
|
chatStatisticsMu sync.RWMutex
|
|
)
|
|
|
|
// WatchBareRTC is a worker goroutine that caches the current online chatters in the chat room.
|
|
func WatchBareRTC() {
|
|
if config.Current.BareRTC.JWTSecret == "" || config.Current.BareRTC.URL == "" {
|
|
log.Error("Worker (WatchBareRTC): chat room is not configured, will not watch chat room status")
|
|
return
|
|
}
|
|
|
|
// Check it immediately.
|
|
DoCheckBareRTC()
|
|
|
|
// And on an interval forever.
|
|
for {
|
|
time.Sleep(config.ChatStatusRefreshInterval)
|
|
DoCheckBareRTC()
|
|
}
|
|
}
|
|
|
|
// DoCheckBareRTC invokes the attempt to refresh data from the chat server about who's online.
|
|
func DoCheckBareRTC() {
|
|
log.Info("Refresh BareRTC")
|
|
req, err := http.NewRequest(http.MethodGet, config.Current.BareRTC.URL+"/api/statistics", nil)
|
|
|
|
if err != nil {
|
|
log.Error("WatchBareRTC: couldn't make request: %s", err)
|
|
SetChatStatistics(nil)
|
|
return
|
|
}
|
|
|
|
client := http.Client{
|
|
Timeout: 10 * time.Second,
|
|
}
|
|
|
|
res, err := client.Do(req)
|
|
if err != nil {
|
|
log.Error("WatchBareRTC: request error: %s", err)
|
|
SetChatStatistics(nil)
|
|
return
|
|
} else if res.StatusCode != http.StatusOK {
|
|
log.Error("WatchBareRTC: didn't get expected 200 OK from statistics endpoint, instead got: %s", res.Status)
|
|
SetChatStatistics(nil)
|
|
return
|
|
}
|
|
|
|
if res.StatusCode == http.StatusOK {
|
|
var cs ChatStatistics
|
|
body, _ := io.ReadAll(res.Body)
|
|
res.Body.Close()
|
|
if err = json.Unmarshal(body, &cs); err != nil {
|
|
log.Error("WatchBareRTC: json decode error: %s", err)
|
|
return
|
|
}
|
|
|
|
SetChatStatistics(&cs)
|
|
}
|
|
}
|