twitch-manager/api.go

171 lines
3.7 KiB
Go
Raw Normal View History

2020-11-20 21:51:10 +00:00
package main
import (
2020-11-20 23:33:02 +00:00
"crypto/sha256"
"fmt"
2020-11-20 21:51:10 +00:00
"net/http"
2020-11-20 23:33:02 +00:00
"strings"
2020-11-20 21:51:10 +00:00
"sync"
"time"
"github.com/gofrs/uuid"
"github.com/pkg/errors"
log "github.com/sirupsen/logrus"
"github.com/gorilla/mux"
"github.com/gorilla/websocket"
)
2020-11-20 23:01:55 +00:00
const (
msgTypeStore string = "store"
)
2020-11-23 08:48:53 +00:00
var subscriptions = newSubscriptionStore()
type subcriptionStore struct {
socketSubscriptions map[string]func(interface{}) error
socketSubscriptionsLock *sync.RWMutex
}
func newSubscriptionStore() *subcriptionStore {
return &subcriptionStore{
socketSubscriptions: map[string]func(msg interface{}) error{},
socketSubscriptionsLock: new(sync.RWMutex),
}
}
2020-11-20 21:51:10 +00:00
2020-11-23 08:48:53 +00:00
func (s subcriptionStore) SendAllSockets(msgType string, msg interface{}) error {
s.socketSubscriptionsLock.RLock()
defer s.socketSubscriptionsLock.RUnlock()
2020-11-20 21:51:10 +00:00
2020-11-23 08:48:53 +00:00
for _, hdl := range s.socketSubscriptions {
2020-11-20 23:33:02 +00:00
if err := hdl(compileSocketMessage(msgType, msg)); err != nil {
2020-11-20 21:51:10 +00:00
return errors.Wrap(err, "submit message")
}
}
return nil
}
2020-11-23 08:48:53 +00:00
func (s *subcriptionStore) SubscribeSocket(id string, hdl func(interface{}) error) {
s.socketSubscriptionsLock.Lock()
defer s.socketSubscriptionsLock.Unlock()
2020-11-20 21:51:10 +00:00
2020-11-23 08:48:53 +00:00
s.socketSubscriptions[id] = hdl
2020-11-20 21:51:10 +00:00
}
2020-11-23 08:48:53 +00:00
func (s *subcriptionStore) UnsubscribeSocket(id string) {
s.socketSubscriptionsLock.Lock()
defer s.socketSubscriptionsLock.Unlock()
2020-11-20 21:51:10 +00:00
2020-11-23 08:48:53 +00:00
delete(s.socketSubscriptions, id)
2020-11-20 21:51:10 +00:00
}
2020-11-20 23:33:02 +00:00
func compileSocketMessage(msgType string, msg interface{}) map[string]interface{} {
assetVersionsLock.RLock()
defer assetVersionsLock.RUnlock()
versionParts := []string{version}
for _, asset := range assets {
versionParts = append(versionParts, assetVersions[asset])
}
hash := sha256.New()
hash.Write([]byte(strings.Join(versionParts, "/")))
ver := fmt.Sprintf("%x", hash.Sum(nil))
return map[string]interface{}{
"payload": msg,
"type": msgType,
"version": ver,
}
}
2020-11-20 21:51:10 +00:00
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
func registerAPI(r *mux.Router) {
r.HandleFunc("/api/subscribe", handleUpdateSocket)
r.HandleFunc("/api/webhook/{type}", handleWebHookPush)
}
func handleUpdateSocket(w http.ResponseWriter, r *http.Request) {
// Upgrade connection to socket
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.WithError(err).Error("Unable to upgrade socket")
return
}
defer conn.Close()
// Register listener
2020-11-22 07:45:33 +00:00
var (
connLock = new(sync.Mutex)
id = uuid.Must(uuid.NewV4()).String()
)
2020-11-23 08:48:53 +00:00
subscriptions.SubscribeSocket(id, func(msg interface{}) error {
2020-11-22 07:45:33 +00:00
connLock.Lock()
defer connLock.Unlock()
return conn.WriteJSON(msg)
})
2020-11-23 08:48:53 +00:00
defer subscriptions.UnsubscribeSocket(id)
2020-11-20 21:51:10 +00:00
keepAlive := time.NewTicker(5 * time.Second)
defer keepAlive.Stop()
go func() {
for range keepAlive.C {
2020-11-22 07:45:33 +00:00
connLock.Lock()
2020-11-20 21:51:10 +00:00
if err := conn.WriteMessage(websocket.PingMessage, nil); err != nil {
log.WithError(err).Error("Unable to send ping message")
2020-11-22 07:45:33 +00:00
connLock.Unlock()
2020-11-20 21:51:10 +00:00
conn.Close()
2020-11-22 07:45:33 +00:00
return
2020-11-20 21:51:10 +00:00
}
2020-11-22 07:45:33 +00:00
connLock.Unlock()
2020-11-20 21:51:10 +00:00
}
}()
2020-11-22 07:45:33 +00:00
connLock.Lock()
2020-11-20 23:33:02 +00:00
if err := conn.WriteJSON(compileSocketMessage(msgTypeStore, store)); err != nil {
2020-11-20 21:51:10 +00:00
log.WithError(err).Error("Unable to send initial state")
return
}
2020-11-22 07:45:33 +00:00
connLock.Unlock()
2020-11-20 21:51:10 +00:00
// Handle socket
for {
messageType, p, err := conn.ReadMessage()
if err != nil {
log.WithError(err).Error("Unable to read from socket")
return
}
switch messageType {
case websocket.TextMessage:
// This is fine and expected
case websocket.BinaryMessage:
// Wat?
log.Warn("Got binary message from socket, disconnecting...")
return
case websocket.CloseMessage:
// They want to go? Fine, have it that way!
return
default:
log.Debug("Got unhandled message from socket: %d", messageType)
continue
}
// FIXME: Do we need this?
_ = p
}
}