@echo off
title Battleships Game
setlocal EnableDelayedExpansion

:: Board dimensions
set width=5
set height=5

:: Set empty player's and computer's boards
for /L %%i in (1,1,%width%) do (
    for /L %%j in (1,1,%height%) do (
        set player_%%i%%j=.
        set computer_%%i%%j=.
    )
)

:: Let the player place their ship
:place_ship
cls
echo Place your ship on the board.
echo Enter the coordinates in the format X Y (e.g., 2 3)
set /P "shipX=Enter X coordinate (1-%width%): "
set /P "shipY=Enter Y coordinate (1-%height%): "
set player_%shipX%%shipY%=S
cls

:: Set computer's ship at a random position
set /A compShipX=%random% %% %width% + 1
set /A compShipY=%random% %% %height% + 1
set computer_%compShipX%%compShipY%=S

:: Function to display the player's board with ship and computer's shots
:display_board
echo Your Board:
for /L %%i in (1,1,%width%) do (
    for /L %%j in (1,1,%height%) do (
        if "!player_%%i%%j!"=="S" (
            set display=S
        ) else if "!player_%%i%%j!"=="X" (
            set display=X
        ) else if "!player_%%i%%j!"=="O" (
            set display=O
        ) else (
            set display=.
        )
        set /P cell=!display!<nul
    )
    echo.
)
echo.

echo Computer's Board (Shots you've taken):
for /L %%i in (1,1,%width%) do (
    for /L %%j in (1,1,%height%) do (
        if "!computer_%%i%%j!"=="X" (
            set display=X
        ) else if "!computer_%%i%%j!"=="O" (
            set display=O
        ) else if "!computer_%%i%%j!"=="S" (
            set display=.
        ) else (
            set display=.
        )
        set /P cell=!display!<nul
    )
    echo.
)
echo.

:: Start the game
echo Welcome to the Battleships Game!
echo Try to sink the computer's ship while avoiding its shots.
echo.

:game_loop
:: Player's turn to guess
set /P "input=Enter your coordinates (X Y) to fire: "
for /f "tokens=1,2" %%a in ("%input%") do (
    set guessX=%%a
    set guessY=%%b
)

:: Check if player hit the computer's ship
if "!computer_%guessX%%guessY%!"=="S" (
    set computer_%guessX%%guessY%=X
    cls
    echo Hit! You sank the computer's ship!
    goto end
) else (
    set computer_%guessX%%guessY%=O
    cls
    echo Miss! The computer takes a shot.
)

:: Computer's turn to take a shot
set /A compGuessX=%random% %% %width% + 1
set /A compGuessY=%random% %% %height% + 1
if "!player_%compGuessX%%compGuessY%!"=="S" (
    set player_%compGuessX%%compGuessY%=X
    cls
    echo The computer hit your ship at %compGuessX% %compGuessY%!
    goto reveal_computer_ship
) else (
    set player_%compGuessX%%compGuessY%=O
    cls
    echo The computer missed its shot at %compGuessX% %compGuessY%.
)

:: Show updated boards
goto display_board
goto game_loop

:reveal_computer_ship
cls
echo.
echo You lost! Here's where the computer's ship was:
for /L %%i in (1,1,%width%) do (
    for /L %%j in (1,1,%height%) do (
        if "!computer_%%i%%j!"=="S" (
            set display=S
        ) else if "!computer_%%i%%j!"=="X" (
            set display=X
        ) else if "!computer_%%i%%j!"=="O" (
            set display=O
        ) else (
            set display=.
        )
        set /P cell=!display!<nul
    )
    echo.
)
echo.
goto end

:end
echo.
echo Thanks for playing!
pause