twitch-bot/internal/apimodules/overlays/database.go

package overlays

import (
	"bytes"
	"encoding/json"
	"strings"
	"time"

	"github.com/pkg/errors"
	"gorm.io/gorm"

	"github.com/Luzifer/twitch-bot/v3/internal/helpers"
	"github.com/Luzifer/twitch-bot/v3/pkg/database"
	"github.com/Luzifer/twitch-bot/v3/plugins"
)

type (
	overlaysEvent struct {
		ID        uint64    `gorm:"primaryKey"`
		Channel   string    `gorm:"not null;index:overlays_events_sort_idx"`
		CreatedAt time.Time `gorm:"index:overlays_events_sort_idx"`
		EventType string
		Fields    string
	}
)

func AddChannelEvent(db database.Connector, channel string, evt SocketMessage) error {
	buf := new(bytes.Buffer)
	if err := json.NewEncoder(buf).Encode(evt.Fields); err != nil {
		return errors.Wrap(err, "encoding fields")
	}

	return errors.Wrap(
		helpers.RetryTransaction(db.DB(), func(tx *gorm.DB) error {
			return tx.Create(&overlaysEvent{
				Channel:   channel,
				CreatedAt: evt.Time.UTC(),
				EventType: evt.Type,
				Fields:    strings.TrimSpace(buf.String()),
			}).Error
		}),
		"storing event to database",
	)
}

func GetChannelEvents(db database.Connector, channel string) ([]SocketMessage, error) {
	var evts []overlaysEvent

	if err := helpers.Retry(func() error {
		return db.DB().Where("channel = ?", channel).Order("created_at").Find(&evts).Error
	}); err != nil {
		return nil, errors.Wrap(err, "querying channel events")
	}

	var out []SocketMessage
	for _, e := range evts {
		fields := new(plugins.FieldCollection)
		if err := json.NewDecoder(strings.NewReader(e.Fields)).Decode(fields); err != nil {
			return nil, errors.Wrap(err, "decoding fields")
		}

		out = append(out, SocketMessage{
			IsLive: false,
			Time:   e.CreatedAt,
			Type:   e.EventType,
			Fields: fields,
		})
	}

	return out, nil
}