Draft: feat login

plays/lobby.go

@@ -0,0 +1,133 @@
+package plays
+
+import (
+	"fmt"
+	"net/http"
+	"strings"
+
+	"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{"Logout"}
+)
+
+type Lobby struct {
+	*Base
+
+	choice string
+	cursor int
+
+	users []models.UserStatus
+	rooms []models.Room
+}
+
+func NewLobby(base *Base) *Lobby {
+	m := Lobby{
+		Base:   base,
+		choice: "",
+		cursor: 0,
+	}
+
+	return &m
+}
+
+func (m *Lobby) Init() tea.Cmd {
+	return tea.Sequence(tea.ClearScreen, Tick())
+}
+
+func updateLobbyInfo(m *Lobby) error {
+	resp, err := m.Base.client.R().
+		SetResult(&m.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
+	}
+	_, err = m.Base.client.R().
+		SetResult(&m.rooms).
+		ForceContentType("application/json").
+		Get("/api/lobby/rooms")
+	if err != nil {
+		tracing.Logger.
+			Error("failed to get lobby rooms",
+				zap.Error(err))
+		return err
+	}
+	return nil
+}
+
+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:
+		go updateLobbyInfo(m)
+		return m, Tick()
+	}
+	return m, nil
+}
+
+func (m *Lobby) View() string {
+	var b strings.Builder
+
+	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 {
+	switch m.choice {
+	case "Logout":
+		*queue = append(*queue,
+			tea.NewProgram(NewLogout(m.Base)))
+	}
+	return nil
+}