156 lines
2.9 KiB
Go
156 lines
2.9 KiB
Go
package plays
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"gitea.konchin.com/ytshih/inp2025/game/models"
|
|
"gitea.konchin.com/ytshih/inp2025/game/tracing"
|
|
"gitea.konchin.com/ytshih/inp2025/game/types"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
var (
|
|
lobbyChoices = []string{"No-op", "Logout"}
|
|
)
|
|
|
|
type Lobby struct {
|
|
*Base
|
|
|
|
choice string
|
|
cursor int
|
|
|
|
updateCh chan struct{}
|
|
users []models.UserStatus
|
|
rooms []models.Room
|
|
}
|
|
|
|
func NewLobby(base *Base) *Lobby {
|
|
m := Lobby{
|
|
Base: base,
|
|
choice: "",
|
|
cursor: 0,
|
|
updateCh: make(chan struct{}, 1),
|
|
}
|
|
|
|
return &m
|
|
}
|
|
|
|
func updateLobbyInfo(m *Lobby) error {
|
|
for {
|
|
select {
|
|
case <-m.updateCh:
|
|
return nil
|
|
default:
|
|
var users []models.UserStatus
|
|
resp, err := m.Base.client.R().
|
|
SetResult(&users).
|
|
ForceContentType("application/json").
|
|
Get("/api/lobby/users")
|
|
if resp.StatusCode() != http.StatusOK {
|
|
tracing.Logger.
|
|
Error("failed to get lobby users",
|
|
zap.Error(err))
|
|
return err
|
|
}
|
|
m.users = users
|
|
|
|
var rooms []models.Room
|
|
_, err = m.Base.client.R().
|
|
SetResult(&rooms).
|
|
ForceContentType("application/json").
|
|
Get("/api/lobby/rooms")
|
|
if err != nil {
|
|
tracing.Logger.
|
|
Error("failed to get lobby rooms",
|
|
zap.Error(err))
|
|
return err
|
|
}
|
|
m.rooms = rooms
|
|
|
|
time.Sleep(refetchTick)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (m *Lobby) Init() tea.Cmd {
|
|
go updateLobbyInfo(m)
|
|
return tea.Sequence(tea.ClearScreen, Tick(refreshTick))
|
|
}
|
|
|
|
func (m *Lobby) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case tea.KeyMsg:
|
|
switch msg.String() {
|
|
case "ctrl+l":
|
|
return m, tea.ClearScreen
|
|
case "ctrl+c", "q":
|
|
return m, tea.Interrupt
|
|
case "enter":
|
|
m.choice = lobbyChoices[m.cursor]
|
|
return m, tea.Quit
|
|
case "tab", "shift+tab", "up", "down":
|
|
s := msg.String()
|
|
|
|
if s == "up" || s == "shift+tab" {
|
|
m.cursor--
|
|
} else {
|
|
m.cursor++
|
|
}
|
|
|
|
if m.cursor >= len(lobbyChoices) {
|
|
m.cursor = 0
|
|
} else if m.cursor < 0 {
|
|
m.cursor = len(lobbyChoices) - 1
|
|
}
|
|
}
|
|
case types.TickMsg:
|
|
return m, Tick(refreshTick)
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func (m *Lobby) View() string {
|
|
var b strings.Builder
|
|
|
|
fmt.Fprintf(&b, "Game lobby\n")
|
|
|
|
fmt.Fprintf(&b, "User Status (%d):\n", len(m.users))
|
|
for _, user := range m.users {
|
|
b.WriteString("- " + user.View() + "\n")
|
|
}
|
|
|
|
fmt.Fprintf(&b, "Room Status (%d):\n", len(m.rooms))
|
|
for _, room := range m.rooms {
|
|
b.WriteString("- " + room.View() + "\n")
|
|
}
|
|
|
|
b.WriteString("==========\n")
|
|
|
|
for i := 0; i < len(lobbyChoices); i++ {
|
|
if m.cursor == i {
|
|
fmt.Fprintf(&b, "(•) %s\n", lobbyChoices[i])
|
|
} else {
|
|
fmt.Fprintf(&b, "( ) %s\n", lobbyChoices[i])
|
|
}
|
|
}
|
|
|
|
return b.String()
|
|
}
|
|
|
|
func (m *Lobby) Next(queue *[]*tea.Program) error {
|
|
m.updateCh <- struct{}{}
|
|
switch m.choice {
|
|
case "No-op":
|
|
*queue = append(*queue,
|
|
tea.NewProgram(NewLobby(m.Base)))
|
|
case "Logout":
|
|
*queue = append(*queue,
|
|
tea.NewProgram(NewLogout(m.Base)))
|
|
}
|
|
return nil
|
|
}
|