Feat: nullable roomid

This commit is contained in:
Grail Finder
2025-07-03 12:29:39 +03:00
parent 2a593739ae
commit d8338fe382
6 changed files with 25 additions and 14 deletions

View File

@ -300,11 +300,14 @@ func recoverBots() {
func recoverBot(bm models.Player) error {
// check if room still exists
if _, err := repo.RoomGetByID(context.Background(), bm.RoomID); err != nil {
return fmt.Errorf("no such room: %s; err: %w", bm.RoomID, err)
if bm.RoomID == nil {
return errors.New("bot has no room id")
}
if _, err := repo.RoomGetByID(context.Background(), *bm.RoomID); err != nil {
return fmt.Errorf("no such room: %s; err: %w", *bm.RoomID, err)
}
log.Debug("recovering bot", "bot", bm)
_, err := llmapi.NewBot(string(bm.Role), string(bm.Team), bm.Username, bm.RoomID, cfg, true)
_, err := llmapi.NewBot(string(bm.Role), string(bm.Team), bm.Username, *bm.RoomID, cfg, true)
if err != nil {
return err
}

View File

@ -110,8 +110,11 @@ func HandleFrontLogin(w http.ResponseWriter, r *http.Request) {
// room.PlayerList = append(room.PlayerList, fi.State.Username)
fi.Room = room
fi.List = nil
fi.State.RoomID = room.ID
repo.PlayerSetRoomID(r.Context(), fi.State.Username, room.ID)
fi.State.RoomID = &room.ID
if err := repo.PlayerSetRoomID(r.Context(), fi.State.Username, room.ID); err != nil {
abortWithError(w, err.Error())
return
}
// repo.RoomUpdate()
// save full info instead
// if err := saveFullInfo(r.Context(), fi); err != nil {
@ -191,6 +194,7 @@ func makeCookie(username string, remote string) (*http.Cookie, error) {
return cookie, nil
}
//nolint: unused
func cacheGetSession(key string) (*models.Session, error) {
userSessionB, err := cache.MemCache.Get(key)
if err != nil {

View File

@ -38,7 +38,7 @@ func HandleCreateRoom(w http.ResponseWriter, r *http.Request) {
abortWithError(w, msg)
return
}
fi.State.RoomID = room.ID
fi.State.RoomID = &room.ID
fi.Room = room
if err := repo.RoomCreate(r.Context(), room); err != nil {
log.Error("failed to create a room", "error", err)
@ -223,7 +223,7 @@ func HandleJoinRoom(w http.ResponseWriter, r *http.Request) {
return
}
// room.PlayerList = append(room.PlayerList, fi.State.Username)
fi.State.RoomID = room.ID
fi.State.RoomID = &room.ID
fi.Room = room
fi.List = nil
if err := saveFullInfo(r.Context(), fi); err != nil {

View File

@ -2,8 +2,8 @@ package handlers
import (
"context"
"fmt"
"gralias/models"
"strconv"
"sync"
"time"
)
@ -46,7 +46,7 @@ func StartTurnTimer(roomID string, duration time.Duration) {
if err := repo.RoomUpdate(context.Background(), room); err != nil {
log.Error("failed to save room", "error", err)
}
notify(models.NotifyTurnTimerPrefix+room.ID, fmt.Sprintf("%d", room.Settings.TurnSecondsLeft))
notify(models.NotifyTurnTimerPrefix+room.ID, strconv.FormatUint(uint64(room.Settings.TurnSecondsLeft), 10))
notifyBotIfNeeded(room)
StopTurnTimer(roomID)
return

View File

@ -49,6 +49,7 @@ func convertToSliceOfStrings(value any) ([]string, error) {
}
}
//nolint: unused
func (b *Bot) checkGuesses(tempMap map[string]any, room *models.Room) error {
guesses, err := convertToSliceOfStrings(tempMap["guesses"])
if err != nil {
@ -539,5 +540,5 @@ func (b *Bot) CallLLM(prompt string) ([]byte, error) {
return body, nil
}
// This line should not be reached because each error path returns in the loop.
return nil, fmt.Errorf("unknown error in retry loop")
return nil, errors.New("unknown error in retry loop")
}

View File

@ -38,8 +38,9 @@ func TestPlayersRepo_AddPlayer(t *testing.T) {
repo := &RepoProvider{DB: db}
roomID := "test_room_player_1"
player := &models.Player{
RoomID: "test_room_player_1",
RoomID: &roomID,
Username: "test_player_1",
Team: "blue",
Role: "player",
@ -61,8 +62,9 @@ func TestPlayersRepo_GetPlayer(t *testing.T) {
repo := &RepoProvider{DB: db}
roomID := "test_room_player_2"
player := &models.Player{
RoomID: "test_room_player_2",
RoomID: &roomID,
Username: "test_player_2",
Team: "red",
Role: "player",
@ -84,8 +86,9 @@ func TestPlayersRepo_DeletePlayer(t *testing.T) {
repo := &RepoProvider{DB: db}
roomID := "test_room_player_3"
player := &models.Player{
RoomID: "test_room_player_3",
RoomID: &roomID,
Username: "test_player_3",
Team: "blue",
Role: "player",
@ -95,7 +98,7 @@ func TestPlayersRepo_DeletePlayer(t *testing.T) {
_, err := db.Exec(`INSERT INTO players (room_id, username, team, role, is_bot) VALUES (?, ?, ?, ?, ?)`, player.RoomID, player.Username, player.Team, player.Role, player.IsBot)
assert.NoError(t, err)
err = repo.PlayerDelete(context.Background(), player.RoomID, player.Username)
err = repo.PlayerDelete(context.Background(), *player.RoomID, player.Username)
assert.NoError(t, err)
var count int