package ui

import (
	"fmt"
	"time"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
)

// WaitingScreen shows room members and lets players ready up before starting.
type WaitingScreen struct {
	ready bool
}

func NewWaitingScreen() *WaitingScreen {
	return &WaitingScreen{}
}

func (s *WaitingScreen) pollWaiting() tea.Cmd {
	return tea.Tick(time.Second, func(t time.Time) tea.Msg {
		return tickMsg{}
	})
}

func (s *WaitingScreen) startGame(ctx *Context) (Screen, tea.Cmd) {
	room := ctx.Lobby.GetRoom(ctx.RoomCode)
	if room != nil && room.Session != nil {
		ctx.Session = room.Session
		ctx.Session.StartGame()
		ctx.Lobby.StartRoom(ctx.RoomCode)
		gs := NewGameScreen()
		gs.gameState = ctx.Session.GetState()
		return gs, gs.pollState()
	}
	return s, s.pollWaiting()
}

func (s *WaitingScreen) Update(msg tea.Msg, ctx *Context) (Screen, tea.Cmd) {
	switch msg.(type) {
	case tickMsg:
		// Check if all players are ready → start game
		if ctx.Lobby != nil && ctx.Lobby.AllReady(ctx.RoomCode) {
			return s.startGame(ctx)
		}
		return s, s.pollWaiting()
	}

	if key, ok := msg.(tea.KeyMsg); ok {
		if isEnter(key) && !s.ready {
			s.ready = true
			if ctx.Lobby != nil {
				ctx.Lobby.SetPlayerReady(ctx.RoomCode, ctx.Fingerprint, true)
				// Solo: if only 1 player in room, start immediately
				room := ctx.Lobby.GetRoom(ctx.RoomCode)
				if room != nil && len(room.Players) == 1 {
					return s.startGame(ctx)
				}
			}
		} else if isKey(key, "esc") || key.Type == tea.KeyEsc {
			// Leave room — unready and go back to lobby
			if ctx.Lobby != nil {
				ctx.Lobby.SetPlayerReady(ctx.RoomCode, ctx.Fingerprint, false)
				ctx.Lobby.LeaveRoom(ctx.RoomCode, ctx.Fingerprint)
			}
			ctx.RoomCode = ""
			ls := NewLobbyScreen()
			ls.refreshLobby(ctx)
			return ls, nil
		}
	}
	return s, nil
}

func (s *WaitingScreen) View(ctx *Context) string {
	headerStyle := lipgloss.NewStyle().
		Foreground(lipgloss.Color("205")).
		Bold(true)

	readyStyle := lipgloss.NewStyle().
		Foreground(lipgloss.Color("46"))

	notReadyStyle := lipgloss.NewStyle().
		Foreground(lipgloss.Color("240"))

	header := headerStyle.Render(fmt.Sprintf("── 대기실 [%s] ──", ctx.RoomCode))

	playerList := ""
	if ctx.Lobby != nil {
		room := ctx.Lobby.GetRoom(ctx.RoomCode)
		if room != nil {
			for _, p := range room.Players {
				status := notReadyStyle.Render("...")
				if p.Ready {
					status = readyStyle.Render("준비 완료")
				}
				cls := p.Class
				if cls == "" {
					cls = "?"
				}
				playerList += fmt.Sprintf("  %s (%s) %s\n", p.Name, cls, status)
			}
		}
	}

	menu := "[Enter] 준비"
	if s.ready {
		menu = "다른 플레이어를 기다리는 중..."
	}
	menu += "  [Esc] 나가기"

	return lipgloss.JoinVertical(lipgloss.Left,
		header,
		"",
		playerList,
		"",
		menu,
	)
}