package chess

import (
	"mchess_server/types"
	"testing"

	"github.com/stretchr/testify/assert"
)

func Test_CheckMove_validPawnMove(t *testing.T) {
	var board = newBoard()

	board.position[types.Coordinate{Col: 1, Row: 2}] = Pawn{Color: types.White}
	board.position[types.Coordinate{Col: 2, Row: 4}] = Pawn{Color: types.Black}
	board.position[types.Coordinate{Col: 5, Row: 1}] = King{Color: types.White}
	board.position[types.Coordinate{Col: 5, Row: 8}] = King{Color: types.Black}

	move := types.Move{
		StartSquare: types.Coordinate{Col: 1, Row: 2},
		EndSquare:   types.Coordinate{Col: 1, Row: 3},
	}

	good, _ := board.CheckAndPlay(move)
	assert.True(t, good)

	//we take the pawn
	secondMove := types.Move{
		StartSquare: types.Coordinate{Col: 2, Row: 4},
		EndSquare:   types.Coordinate{Col: 1, Row: 3},
	}
	good, _ = board.CheckAndPlay(secondMove)
	assert.True(t, good)
}

func Test_CheckMove_enPassant(t *testing.T) {
	var board = newBoard()

	board.position[types.Coordinate{Col: 6, Row: 4}] = Pawn{Color: types.Black}
	board.position[types.Coordinate{Col: 5, Row: 2}] = Pawn{Color: types.White}
	board.position[types.Coordinate{Col: 5, Row: 1}] = King{Color: types.White}
	board.position[types.Coordinate{Col: 5, Row: 8}] = King{Color: types.Black}

	move := types.Move{
		StartSquare: types.Coordinate{Col: 5, Row: 2},
		EndSquare:   types.Coordinate{Col: 5, Row: 4},
	}

	good, reason := board.CheckAndPlay(move)
	assert.True(t, good)
	assert.Empty(t, reason)
	assert.Equal(t, Pawn{Color: types.White}, board.position[types.Coordinate{Col: 5, Row: 4}])

	newMove := types.Move{
		StartSquare: types.Coordinate{Col: 6, Row: 4},
		EndSquare:   types.Coordinate{Col: 5, Row: 3},
	}
	good, reason = board.CheckAndPlay(newMove)
	assert.True(t, good)
	assert.Empty(t, reason)
	// the black pawn is on its correct square
	assert.Equal(t, Pawn{Color: types.Black}, board.position[types.Coordinate{Col: 5, Row: 3}])
	//the white pawn is gone
	assert.Nil(t, board.position[types.Coordinate{Col: 5, Row: 4}])
}

func Test_CheckMove_invalidPawnMoves(t *testing.T) {
	t.Run("pawn is blocked", func(t *testing.T) {
		var board = newBoard()

		board.position[types.Coordinate{Col: 2, Row: 5}] = Pawn{Color: types.White}
		board.position[types.Coordinate{Col: 1, Row: 5}] = King{Color: types.White}
		board.position[types.Coordinate{Col: 7, Row: 5}] = Queen{Color: types.Black}
		board.position[types.Coordinate{Col: 8, Row: 5}] = King{Color: types.Black}
		board.position[types.Coordinate{Col: 2, Row: 6}] = Pawn{Color: types.Black}

		move := types.Move{
			StartSquare: types.Coordinate{Col: 2, Row: 5},
			EndSquare:   types.Coordinate{Col: 2, Row: 6},
		}
		legalMove, _ := board.CheckAndPlay(move)
		assert.False(t, legalMove)
	})

	t.Run("pawn moves to the side", func(t *testing.T) {
		var board = newBoard()
		boardBeforeMove := board

		board.position[types.Coordinate{Col: 1, Row: 5}] = King{Color: types.White}
		board.position[types.Coordinate{Col: 8, Row: 5}] = King{Color: types.Black}
		board.position[types.Coordinate{Col: 2, Row: 5}] = Pawn{Color: types.White}

		move := types.Move{
			StartSquare: types.Coordinate{Col: 2, Row: 5},
			EndSquare:   types.Coordinate{Col: 3, Row: 5},
		}
		legal, _ := board.CheckAndPlay(move)

		assert.False(t, legal)
		assert.Equal(t, boardBeforeMove, board)

		move = types.Move{
			StartSquare: types.Coordinate{Col: 2, Row: 5},
			EndSquare:   types.Coordinate{Col: 1, Row: 5},
		}
		legal, _ = board.CheckAndPlay(move)

		assert.False(t, legal)
		assert.Equal(t, boardBeforeMove, board)

		move = types.Move{
			StartSquare: types.Coordinate{Col: 2, Row: 5},
			EndSquare:   types.Coordinate{Col: 6, Row: 5},
		}
		legal, _ = board.CheckAndPlay(move)

		assert.False(t, legal)
		assert.Equal(t, boardBeforeMove, board)
	})

	t.Run("king of moving color is in check after move", func(t *testing.T) {
		var board = newBoard()

		board.position[types.Coordinate{Col: 1, Row: 6}] = King{Color: types.White}
		board.position[types.Coordinate{Col: 2, Row: 6}] = Pawn{Color: types.White}
		board.position[types.Coordinate{Col: 7, Row: 6}] = Rook{Color: types.Black}
		board.position[types.Coordinate{Col: 8, Row: 8}] = King{Color: types.Black}
        boardBeforeMove := board

		move := types.Move{
			StartSquare: types.Coordinate{Col: 2, Row: 6},
			EndSquare:   types.Coordinate{Col: 2, Row: 7},
		}
		good, _ := board.CheckAndPlay(move)

		assert.False(t, good)
        assert.Equal(t, boardBeforeMove, board)
	})
}

func Test_CheckMove_validPromotion(t *testing.T) {
	var board Board = newBoard()

	board.position[types.Coordinate{Col: 1, Row: 7}] = Pawn{Color: types.White}
	board.position[types.Coordinate{Col: 1, Row: 1}] = King{Color: types.White}

	board.position[types.Coordinate{Col: 7, Row: 7}] = King{Color: types.Black}

	shortName := types.QueenShortName
	move := types.Move{
		StartSquare:      types.Coordinate{Col: 1, Row: 7},
		EndSquare:        types.Coordinate{Col: 1, Row: 8},
		PromotionToPiece: &shortName,
	}
	good, reason := board.CheckAndPlay(move)

	assert.Empty(t, reason)
	assert.True(t, good)
	assert.Equal(t, Queen{Color: types.White}, board.getPieceAt(types.Coordinate{Row: 8, Col: 1}))
}

func Test_CheckMove_HistoryWorks(t *testing.T) {
	var board = newBoard()

	board.position[types.Coordinate{Col: 3, Row: 7}] = Pawn{Color: types.Black}
	board.position[types.Coordinate{Col: 1, Row: 2}] = Pawn{Color: types.White}
	board.position[types.Coordinate{Col: 1, Row: 5}] = King{Color: types.White}
	board.position[types.Coordinate{Col: 8, Row: 5}] = King{Color: types.Black}

	firstMove := types.Move{
		StartSquare: types.Coordinate{Col: 1, Row: 2},
		EndSquare:   types.Coordinate{Col: 1, Row: 3},
	}
	secondMove := types.Move{
		StartSquare: types.Coordinate{Col: 3, Row: 7},
		EndSquare:   types.Coordinate{Col: 3, Row: 5},
	}
	thirdMove := types.Move{
		StartSquare: types.Coordinate{Col: 1, Row: 3},
		EndSquare:   types.Coordinate{Col: 1, Row: 4},
	}

	good, _ := board.CheckAndPlay(firstMove)
	assert.True(t, good)

	good, _ = board.CheckAndPlay(secondMove)
	assert.True(t, good)

	good, _ = board.CheckAndPlay(thirdMove)
	assert.True(t, good)

	expectedHistory := []types.Move{
		{
			StartSquare: types.Coordinate{Col: 1, Row: 2},
			EndSquare:   types.Coordinate{Col: 1, Row: 3},
			PieceMoved:  'p',
			ColorMoved:  "white",
		},
		{
			StartSquare: types.Coordinate{Col: 3, Row: 7},
			EndSquare:   types.Coordinate{Col: 3, Row: 5},
			PieceMoved:  'p',
			ColorMoved:  "black",
		},
		{
			StartSquare: types.Coordinate{Col: 1, Row: 3},
			EndSquare:   types.Coordinate{Col: 1, Row: 4},
			PieceMoved:  'p',
			ColorMoved:  "white",
		},
	}
	assert.Equal(t, expectedHistory, board.history)
}