Draft: big refactor

plays/roomWaiting.go

@@ -0,0 +1,90 @@
+package plays
+
+import (
+	"fmt"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+
+	"gitea.konchin.com/ytshih/inp2025/game/tracing"
+	"gitea.konchin.com/ytshih/inp2025/game/types"
+	tea "github.com/charmbracelet/bubbletea"
+	"go.uber.org/zap"
+)
+
+type RoomWaiting struct {
+	*Base
+	roomID string
+	room   *types.Room
+}
+
+func NewRoomWaiting(base *Base, roomID string) *RoomWaiting {
+	return &RoomWaiting{
+		Base:   base,
+		roomID: roomID,
+	}
+}
+
+func (m *RoomWaiting) checkRoomStatus() tea.Msg {
+	resp, err := m.Base.client.R().
+		SetResult(&types.Room{}).
+		Get(fmt.Sprintf("/api/rooms/%s", m.roomID))
+
+	if err != nil || resp.StatusCode() != http.StatusOK {
+		return types.TickMsg{} // Keep polling
+	}
+
+	room := resp.Result().(*types.Room)
+	if room.Status == types.RoomStatusPlaying {
+		m.room = room
+		return tea.Quit() // Stop polling and proceed
+	}
+
+	return types.TickMsg{} // Keep polling
+}
+
+func (m *RoomWaiting) Init() tea.Cmd {
+	return tea.Batch(tea.ClearScreen, func() tea.Msg {
+		return types.TickMsg(time.Now())
+	})
+}
+
+func (m *RoomWaiting) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	switch msg := msg.(type) {
+	case types.TickMsg:
+		time.Sleep(1 * time.Second)
+		return m, m.checkRoomStatus
+	case tea.KeyMsg:
+		switch msg.String() {
+		case "ctrl+c", "q":
+			return m, tea.Quit
+		}
+	}
+	return m, nil
+}
+
+func (m *RoomWaiting) View() string {
+	var b strings.Builder
+	b.WriteString("Waiting for another player to join...\n")
+	b.WriteString(fmt.Sprintf("Room ID: %s\n", m.roomID))
+	return b.String()
+}
+
+func (m *RoomWaiting) Next(queue *[]*tea.Program) error {
+	if m.room != nil && m.room.Status == types.RoomStatusPlaying {
+		hostURL := m.Base.client.HostURL
+		wsHost := strings.Replace(hostURL, "http", "ws", 1)
+		wsURL := fmt.Sprintf("%s/api/rooms/%s/state", wsHost, m.roomID)
+		tracing.Logger.Info("parsing websocket url", zap.String("url", wsURL))
+		u, err := url.Parse(wsURL)
+		if err != nil {
+			return err
+		}
+		*queue = append(*queue, tea.NewProgram(NewGame(m.Base, u, m.roomID)))
+	} else {
+		// If we quit without the room being ready, go back to the lobby.
+		*queue = append(*queue, tea.NewProgram(NewLobby(m.Base)))
+	}
+	return nil
+}