59 lines
1.3 KiB
Plaintext
59 lines
1.3 KiB
Plaintext
package lobby
|
|
|
|
import (
|
|
"context"
|
|
"gitea/price-is-wrong/pkg/lib/db"
|
|
lobby "gitea/price-is-wrong/pkg/state/lobby/internal"
|
|
"slices"
|
|
)
|
|
|
|
func (l *Lobby) Join(ctx context.Context, id int) error {
|
|
return l.upsertEvent(ctx, &lobby.PlayerJoin{ID: id})
|
|
}
|
|
|
|
func (l *Lobby) Leave(ctx context.Context, id int) error {
|
|
return l.upsertEvent(ctx, &lobby.PlayerLeave{ID: id})
|
|
}
|
|
|
|
func (l *Lobby) Close(ctx context.Context) error {
|
|
return l.upsertEvent(ctx, &lobby.Close{})
|
|
}
|
|
|
|
func (l *Lobby) upsertEvent(ctx context.Context, e lobby.Event) error {
|
|
if err := upsertEvent(ctx, l.id, e); err != nil {
|
|
return err
|
|
}
|
|
|
|
l2, err := openID(ctx, l.id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*l = *l2
|
|
return nil
|
|
}
|
|
|
|
func upsertEvent(ctx context.Context, lobbyID int, e lobby.Event) error {
|
|
b, err := lobby.MarshalEvent(e)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = db.From(ctx).ExecContext(ctx, `
|
|
INSERT INTO lobby_events (lobby_id, payload) VALUES (?, ?)
|
|
`, lobbyID, b)
|
|
return err
|
|
}
|
|
|
|
func (l *Lobby) apply(e lobby.Event) error {
|
|
switch e := e.(type) {
|
|
case *lobby.PlayerJoin:
|
|
if !slices.Contains(l.Players, e.ID) {
|
|
l.Players = append(l.Players, e.ID)
|
|
}
|
|
case *lobby.PlayerLeave:
|
|
l.Players = slices.DeleteFunc(l.Players, func(id int) bool { return id == e.ID })
|
|
}
|
|
return nil
|
|
}
|