chess/internal/client/commands/game.go

// FILE: lixenwraith/chess/internal/client/commands/game.go
package commands

import (
	"bufio"
	"fmt"
	"os"
	"strconv"
	"strings"
	"time"

	"chess/internal/client/api"
	"chess/internal/client/display"
)

func (r *Registry) registerGameCommands() {
	r.Register(&Command{
		Name:        "new",
		ShortName:   "n",
		Description: "Create a new game",
		Usage:       "new",
		Handler:     newGameHandler,
	})

	r.Register(&Command{
		Name:        "join",
		ShortName:   "j",
		Description: "Join/set current game ID",
		Usage:       "join <gameId>",
		Handler:     joinGameHandler,
	})

	r.Register(&Command{
		Name:        "move",
		ShortName:   "m",
		Description: "Make a move",
		Usage:       "move <uci-move>",
		Handler:     moveHandler,
	})

	r.Register(&Command{
		Name:        "computer",
		ShortName:   "c",
		Description: "Trigger computer move",
		Usage:       "computer",
		Handler:     computerMoveHandler,
	})

	r.Register(&Command{
		Name:        "undo",
		ShortName:   "u",
		Description: "Undo moves",
		Usage:       "undo [count]",
		Handler:     undoHandler,
	})

	r.Register(&Command{
		Name:        "show",
		ShortName:   "h",
		Description: "Show board and game state",
		Usage:       "show",
		Handler:     showBoardHandler,
	})

	r.Register(&Command{
		Name:        "state",
		ShortName:   "s",
		Description: "Show raw game JSON",
		Usage:       "state",
		Handler:     gameStateHandler,
	})

	r.Register(&Command{
		Name:        "delete",
		ShortName:   "d",
		Description: "Delete a game",
		Usage:       "delete [gameId]",
		Handler:     deleteGameHandler,
	})

	r.Register(&Command{
		Name:        "poll",
		ShortName:   "p",
		Description: "Long-poll for game updates",
		Usage:       "poll",
		Handler:     pollHandler,
	})
}

func newGameHandler(s Session, args []string) error {
	scanner := bufio.NewScanner(os.Stdin)
	c := s.GetClient().(*api.Client)

	fmt.Println("\n" + display.Cyan + "Creating new game..." + display.Reset)

	// White player
	fmt.Print(display.Yellow + "White player type (h/c) [h]: " + display.Reset)
	scanner.Scan()
	whiteType := strings.ToLower(strings.TrimSpace(scanner.Text()))
	if whiteType == "" {
		whiteType = "h"
	}

	white := api.PlayerConfig{Type: 1}
	if whiteType == "c" {
		white.Type = 2

		fmt.Print(display.Yellow + "Computer level (0-20) [10]: " + display.Reset)
		scanner.Scan()
		levelStr := strings.TrimSpace(scanner.Text())
		if levelStr == "" {
			white.Level = 10
		} else {
			level, _ := strconv.Atoi(levelStr)
			white.Level = level
		}

		fmt.Print(display.Yellow + "Search time (100-10000ms) [1000]: " + display.Reset)
		scanner.Scan()
		timeStr := strings.TrimSpace(scanner.Text())
		if timeStr == "" {
			white.SearchTime = 1000
		} else {
			searchTime, _ := strconv.Atoi(timeStr)
			white.SearchTime = searchTime
		}
	}

	// Black player
	fmt.Print(display.Yellow + "Black player type (h/c) [h]: " + display.Reset)
	scanner.Scan()
	blackType := strings.ToLower(strings.TrimSpace(scanner.Text()))
	if blackType == "" {
		blackType = "h"
	}

	black := api.PlayerConfig{Type: 1}
	if blackType == "c" {
		black.Type = 2

		fmt.Print(display.Yellow + "Computer level (0-20) [10]: " + display.Reset)
		scanner.Scan()
		levelStr := strings.TrimSpace(scanner.Text())
		if levelStr == "" {
			black.Level = 10
		} else {
			level, _ := strconv.Atoi(levelStr)
			black.Level = level
		}

		fmt.Print(display.Yellow + "Search time (100-10000ms) [1000]: " + display.Reset)
		scanner.Scan()
		timeStr := strings.TrimSpace(scanner.Text())
		if timeStr == "" {
			black.SearchTime = 1000
		} else {
			searchTime, _ := strconv.Atoi(timeStr)
			black.SearchTime = searchTime
		}
	}

	// Starting position
	fmt.Print(display.Yellow + "Starting position (FEN) [default]: " + display.Reset)
	scanner.Scan()
	fen := strings.TrimSpace(scanner.Text())

	req := &api.CreateGameRequest{
		White: white,
		Black: black,
		FEN:   fen,
	}

	resp, err := c.CreateGame(req)
	if err != nil {
		return err
	}

	s.SetCurrentGame(resp.GameID)
	s.SetLastMoveCount(len(resp.Moves))
	s.SetGameState(resp)

	// Determine player color if authenticated
	if s.GetCurrentUser() != "" {
		if resp.Players.White.ID == s.GetCurrentUser() {
			s.SetPlayerColor("w")
		} else if resp.Players.Black.ID == s.GetCurrentUser() {
			s.SetPlayerColor("b")
		}
	}

	fmt.Printf("%sGame created: %s%s\n", display.Green, resp.GameID, display.Reset)
	fmt.Printf("%sCurrent game set to: %s%s\n", display.Cyan, resp.GameID, display.Reset)

	// If white is computer, trigger first move
	if white.Type == 2 {
		fmt.Printf("\n%sTriggering white computer move...%s\n", display.Magenta, display.Reset)
		time.Sleep(100 * time.Millisecond)
		_, err = c.MakeMove(resp.GameID, "cccc")
		if err != nil {
			fmt.Printf("%sFailed to trigger computer move: %s%s\n", display.Red, err.Error(), display.Reset)
		}
	}

	return nil
}

func joinGameHandler(s Session, args []string) error {
	if len(args) < 1 {
		return fmt.Errorf("usage: join <gameId>")
	}

	gameID := args[0]
	c := s.GetClient().(*api.Client)

	// Verify game exists
	resp, err := c.GetGame(gameID)
	if err != nil {
		return err
	}

	s.SetCurrentGame(gameID)
	s.SetLastMoveCount(len(resp.Moves))
	s.SetGameState(resp)

	// Determine player color if authenticated
	if s.GetCurrentUser() != "" {
		if resp.Players.White.ID == s.GetCurrentUser() {
			s.SetPlayerColor("w")
		} else if resp.Players.Black.ID == s.GetCurrentUser() {
			s.SetPlayerColor("b")
		} else {
			s.SetPlayerColor("")
		}
	}

	fmt.Printf("%sJoined game: %s%s\n", display.Green, gameID, display.Reset)
	fmt.Printf("Turn: %s | State: %s | Moves: %d\n", resp.Turn, resp.State, len(resp.Moves))

	return nil
}

func moveHandler(s Session, args []string) error {
	if len(args) < 1 {
		return fmt.Errorf("usage: move <uci-move>")
	}

	gameID := s.GetCurrentGame()
	if gameID == "" {
		return fmt.Errorf("no current game, use 'new' or 'join <gameId>'")
	}

	move := args[0]
	c := s.GetClient().(*api.Client)

	resp, err := c.MakeMove(gameID, move)
	if err != nil {
		return err
	}

	s.SetLastMoveCount(len(resp.Moves))
	s.SetGameState(resp)
	fmt.Printf("%sMove accepted%s\n", display.Green, display.Reset)

	// Check if computer should move
	currentTurn := resp.Turn
	var computerPlayer *api.PlayerInfo
	if currentTurn == "w" && resp.Players.White.Type == 2 {
		computerPlayer = &resp.Players.White
	} else if currentTurn == "b" && resp.Players.Black.Type == 2 {
		computerPlayer = &resp.Players.Black
	}

	if computerPlayer != nil && resp.State == "ongoing" {
		fmt.Printf("\n%sComputer's turn, triggering move...%s\n", display.Magenta, display.Reset)
		time.Sleep(100 * time.Millisecond)
		resp2, err := c.MakeMove(gameID, "cccc")
		if err != nil {
			fmt.Printf("%sFailed to trigger computer move: %s%s\n", display.Red, err.Error(), display.Reset)
		} else if resp2.State == "pending" {
			fmt.Printf("%sComputer is thinking...%s\n", display.Magenta, display.Reset)
			// Wait for completion
			for i := 0; i < 50; i++ {
				time.Sleep(200 * time.Millisecond)
				resp3, err := c.GetGame(gameID)
				if err == nil && resp3.State != "pending" {
					s.SetLastMoveCount(len(resp3.Moves))
					if resp3.LastMove != nil {
						fmt.Printf("%sComputer played: %s%s", display.Magenta, resp3.LastMove.Move, display.Reset)
						if resp3.LastMove.Depth > 0 {
							fmt.Printf(" (depth %d, score %d)", resp3.LastMove.Depth, resp3.LastMove.Score)
						}
						fmt.Println()
					}
					break
				}
			}
		}
	}

	return nil
}

func computerMoveHandler(s Session, args []string) error {
	gameID := s.GetCurrentGame()
	if gameID == "" {
		return fmt.Errorf("no current game, use 'new' or 'join <gameId>'")
	}

	c := s.GetClient().(*api.Client)

	resp, err := c.MakeMove(gameID, "cccc")
	if err != nil {
		return err
	}

	if resp.State == "pending" {
		fmt.Printf("%sComputer is thinking...%s\n", display.Magenta, display.Reset)

		// Poll for completion
		for i := 0; i < 50; i++ {
			time.Sleep(200 * time.Millisecond)
			resp2, err := c.GetGame(gameID)
			if err == nil && resp2.State != "pending" {
				s.SetLastMoveCount(len(resp2.Moves))
				s.SetGameState(resp2)
				if resp2.LastMove != nil {
					fmt.Printf("%sComputer played: %s%s", display.Magenta, resp2.LastMove.Move, display.Reset)
					if resp2.LastMove.Depth > 0 {
						fmt.Printf(" (depth %d, score %d)", resp2.LastMove.Depth, resp2.LastMove.Score)
					}
					fmt.Println()
				}
				return nil
			}
		}
		return fmt.Errorf("timeout waiting for computer move")
	}

	s.SetLastMoveCount(len(resp.Moves))
	fmt.Printf("%sMove triggered%s\n", display.Green, display.Reset)
	return nil
}

func undoHandler(s Session, args []string) error {
	gameID := s.GetCurrentGame()
	if gameID == "" {
		return fmt.Errorf("no current game, use 'new' or 'join <gameId>'")
	}

	count := 1
	if len(args) > 0 {
		var err error
		count, err = strconv.Atoi(args[0])
		if err != nil {
			return fmt.Errorf("invalid count: %s", args[0])
		}
	}

	c := s.GetClient().(*api.Client)
	resp, err := c.UndoMoves(gameID, count)
	if err != nil {
		return err
	}

	s.SetLastMoveCount(len(resp.Moves))
	fmt.Printf("%sUndid %d move(s)%s\n", display.Green, count, display.Reset)
	return nil
}

func showBoardHandler(s Session, args []string) error {
	gameID := s.GetCurrentGame()
	if gameID == "" {
		return fmt.Errorf("no current game, use 'new' or 'join <gameId>'")
	}

	c := s.GetClient().(*api.Client)

	// Get full game state
	game, err := c.GetGame(gameID)
	if err != nil {
		return err
	}

	// Get ASCII board
	board, err := c.GetBoard(gameID)
	if err != nil {
		return err
	}

	s.SetLastMoveCount(len(game.Moves))
	s.SetGameState(game)

	// Display board with colors
	fmt.Println()
	display.RenderBoard(board.Board)

	// Display game info
	fmt.Printf("\nFEN: %s\n", game.FEN)
	fmt.Printf("Turn: %s | State: %s | Moves: %d\n",
		display.ColorForTurn(game.Turn), game.State, len(game.Moves))

	// Display move history
	if len(game.Moves) > 0 {
		fmt.Printf("\nHistory: ")
		for i, move := range game.Moves {
			if i > 0 {
				fmt.Print(" ")
			}
			if i%2 == 0 {
				fmt.Printf("%d.%s", (i/2)+1, move)
			} else {
				fmt.Printf(" %s", move)
			}
		}
		fmt.Println()
	}

	// Display last move info
	if game.LastMove != nil {
		color := "White"
		if game.LastMove.PlayerColor == "b" {
			color = "Black"
		}
		fmt.Printf("Last move: %s by %s", game.LastMove.Move, color)
		if game.LastMove.Depth > 0 {
			fmt.Printf(" (depth %d, score %d)", game.LastMove.Depth, game.LastMove.Score)
		}
		fmt.Println()
	}

	return nil
}

func gameStateHandler(s Session, args []string) error {
	gameID := s.GetCurrentGame()
	if gameID == "" {
		return fmt.Errorf("no current game, use 'new' or 'join <gameId>'")
	}

	c := s.GetClient().(*api.Client)
	resp, err := c.GetGame(gameID)
	if err != nil {
		return err
	}

	s.SetLastMoveCount(len(resp.Moves))

	// Pretty print JSON
	fmt.Printf("%sGame State:%s\n", display.Cyan, display.Reset)
	display.PrettyPrintJSON(resp)

	return nil
}

func deleteGameHandler(s Session, args []string) error {
	gameID := s.GetCurrentGame()
	if len(args) > 0 {
		gameID = args[0]
	}

	if gameID == "" {
		return fmt.Errorf("specify game ID or set current game")
	}

	c := s.GetClient().(*api.Client)
	err := c.DeleteGame(gameID)
	if err != nil {
		return err
	}

	if gameID == s.GetCurrentGame() {
		s.SetCurrentGame("")
		s.SetLastMoveCount(0)
	}

	fmt.Printf("%sGame deleted: %s%s\n", display.Green, gameID, display.Reset)
	return nil
}

func pollHandler(s Session, args []string) error {
	gameID := s.GetCurrentGame()
	if gameID == "" {
		return fmt.Errorf("no current game, use 'new' or 'join <gameId>'")
	}

	c := s.GetClient().(*api.Client)
	moveCount := s.GetLastMoveCount()

	fmt.Printf("%sLong-polling for updates (move count: %d)...%s\n",
		display.Cyan, moveCount, display.Reset)
	fmt.Printf("%sThis may take up to 25 seconds%s\n", display.Cyan, display.Reset)

	resp, err := c.GetGameWithPoll(gameID, moveCount)
	if err != nil {
		return err
	}

	s.SetLastMoveCount(len(resp.Moves))
	s.SetGameState(resp)

	if len(resp.Moves) > moveCount {
		fmt.Printf("%sGame updated! New moves detected%s\n", display.Green, display.Reset)
		if resp.LastMove != nil {
			fmt.Printf("Last move: %s\n", resp.LastMove.Move)
		}
	} else {
		fmt.Printf("%sNo updates (timeout)%s\n", display.Yellow, display.Reset)
	}

	return nil
}