@echo off
setlocal

REM Always use 'nathan' as NAS username
set NASUSER=nathan

REM Menu for operation selection
echo.
echo Select operation:
echo 1. Generate thumbnails only
echo 2. SSH cleanup only  
echo 3. Both (generate thumbnails + SSH cleanup)
echo.
set /p CHOICE=Enter your choice (1-3): 

if "%CHOICE%"=="1" goto THUMBS_ONLY
if "%CHOICE%"=="2" goto SSH_ONLY
if "%CHOICE%"=="3" goto BOTH
echo Invalid choice. Exiting.
exit /b

:THUMBS_ONLY
REM Prompt for Windows path
set /p WINPATH=Enter the full Windows path to your NAS directory (e.g., R:\YouTube\Streams\MixerTwitch): 
if "%WINPATH%"=="" (
    echo No path provided. Exiting.
    exit /b
)

REM Run psthumbgen.py
python psthumbgen.py --directory "%WINPATH%"
goto END

:SSH_ONLY
REM Prompt for Windows path
set /p WINPATH=Enter the full Windows path to your NAS directory (e.g., R:\YouTube\Streams\MixerTwitch): 
if "%WINPATH%"=="" (
    echo No path provided. Exiting.
    exit /b
)

REM Convert Windows path to NAS path
set "RELPATH=%WINPATH:~3%"
set "RELPATH=%RELPATH:\=/%"
set "NASPATH=/volume1/Hydra/%RELPATH%"

echo DEBUG: WINPATH = %WINPATH%
echo DEBUG: RELPATH = %RELPATH%
echo DEBUG: NASPATH = %NASPATH%
echo.

REM SSH cleanup commands (run separately)
echo Running SSH cleanup commands...
echo.
echo Please run these commands manually on your NAS:
echo.
echo ssh nathan@hydra
echo find %NASPATH% -type d -name '@eaDir' -exec rm -rf '{}' \;
echo find %NASPATH% -type d -name 'eaDir_tmp' -exec bash -c 'mv "$0" "${0%%/*}/@eaDir"' {} \;
echo.
echo Press any key when done...
goto END

:BOTH
REM Prompt for Windows path
set /p WINPATH=Enter the full Windows path to your NAS directory (e.g., R:\YouTube\Streams\MixerTwitch): 
if "%WINPATH%"=="" (
    echo No path provided. Exiting.
    exit /b
)

REM Run psthumbgen.py
python psthumbgen.py --directory "%WINPATH%"

REM Convert Windows path to NAS path
set "RELPATH=%WINPATH:~3%"
set "RELPATH=%RELPATH:\=/%"
set "NASPATH=/volume1/Hydra/%RELPATH%"

echo DEBUG: WINPATH = %WINPATH%
echo DEBUG: RELPATH = %RELPATH%
echo DEBUG: NASPATH = %NASPATH%
echo.

REM SSH cleanup commands (run separately)
echo Running SSH cleanup commands...
echo.
echo Please run these commands manually on your NAS:
echo.
echo ssh nathan@hydra
echo find %NASPATH% -type d -name '@eaDir' -exec rm -rf '{}' \;
echo find %NASPATH% -type d -name 'eaDir_tmp' -exec bash -c 'mv "$0" "${0%%/*}/@eaDir"' {} \;
echo.
echo Press any key when done...

:END
pause