open-webui/start_server_with_nginx_on_windows.bat

@echo off
chcp 65001 >nul
setlocal EnableDelayedExpansion

rem Set basic variables
set "SCRIPT_DIR=%~dp0"
set "LOG_DIR=%SCRIPT_DIR%logs"
set "BACKEND_PID_FILE=%SCRIPT_DIR%backend.pid"
set "NGINX_PID_FILE=%SCRIPT_DIR%nginx.pid"

if not exist "%LOG_DIR%" mkdir "%LOG_DIR%"
set "LOG_FILE=%LOG_DIR%\server_%date:~0,4%%date:~5,2%%date:~8,2%.log"

rem Clean up old processes and files
taskkill /F /IM nginx.exe >nul 2>&1
taskkill /F /IM python.exe /FI "WINDOWTITLE eq OpenWebUI*" >nul 2>&1
timeout /t 2 >nul

if exist "%BACKEND_PID_FILE%" del "%BACKEND_PID_FILE%"
if exist "%NGINX_PID_FILE%" del "%NGINX_PID_FILE%"

rem Create necessary directories for Nginx
cd /d "%SCRIPT_DIR%nginx-1.26.3"
if not exist "logs" mkdir logs
if not exist "temp" mkdir temp
if not exist "temp\client_body_temp" mkdir "temp\client_body_temp"
if not exist "temp\proxy_temp" mkdir "temp\proxy_temp"
if not exist "temp\fastcgi_temp" mkdir "temp\fastcgi_temp"
if not exist "temp\uwsgi_temp" mkdir "temp\uwsgi_temp"
if not exist "temp\scgi_temp" mkdir "temp\scgi_temp"

goto :main

:write_log
echo [%date% %time%] %~1
echo [%date% %time%] %~1 >> "%LOG_FILE%"
exit /b

:start_backend
cd /d "%SCRIPT_DIR%backend"
call :write_log "Starting backend service..."
start "OpenWebUI Backend" /B cmd /c start_windows.bat
if errorlevel 1 (
    call :write_log "Failed to start backend"
    exit /b 1
)

timeout /t 5 >nul

for /f "tokens=2" %%a in ('wmic process where "commandline like '%%python%%' and name='python.exe'" get processid ^| findstr /r "[0-9]"') do (
    echo %%a > "%BACKEND_PID_FILE%"
    call :write_log "Backend started with PID: %%a"
    exit /b 0
)

call :write_log "Failed to get backend PID"
exit /b 1

:start_nginx
cd /d "%SCRIPT_DIR%nginx-1.26.3"
call :write_log "Starting Nginx..."
start "" /B nginx.exe
if errorlevel 1 (
    call :write_log "Failed to start Nginx"
    exit /b 1
)

timeout /t 2 >nul

for /f "tokens=2" %%a in ('tasklist /fi "imagename eq nginx.exe" /nh ^| findstr "nginx"') do (
    echo %%a > "%NGINX_PID_FILE%"
    call :write_log "Nginx started with PID: %%a"
    exit /b 0
)

call :write_log "Failed to get Nginx PID"
exit /b 1

:start
call :write_log "Starting services..."

call conda activate open-webui
if errorlevel 1 (
    call :write_log "Failed to activate conda environment"
    goto :cleanup
)

call :start_backend
if errorlevel 1 goto :cleanup

call :start_nginx
if errorlevel 1 goto :cleanup

cd /d "%SCRIPT_DIR%"
call :write_log "All services started successfully"
exit /b 0

:cleanup
call :write_log "Start failed, cleaning up..."
call :stop
exit /b 1

:stop
call :write_log "Stopping services..."

if exist "%BACKEND_PID_FILE%" (
    for /f %%a in (%BACKEND_PID_FILE%) do (
        taskkill /F /T /PID %%a >nul 2>&1
        if not errorlevel 1 call :write_log "Stopped backend process (PID: %%a)"
    )
    del "%BACKEND_PID_FILE%"
)

for /f "tokens=2" %%a in ('wmic process where "commandline like '%%python%%' and name='python.exe'" get processid ^| findstr /r "[0-9]"') do (
    taskkill /F /T /PID %%a >nul 2>&1
    call :write_log "Stopped additional Python process (PID: %%a)"
)

if exist "%NGINX_PID_FILE%" (
    for /f %%a in (%NGINX_PID_FILE%) do (
        taskkill /F /PID %%a >nul 2>&1
        if not errorlevel 1 call :write_log "Stopped Nginx (PID: %%a)"
    )
    del "%NGINX_PID_FILE%"
)

taskkill /F /IM nginx.exe >nul 2>&1
if not errorlevel 1 call :write_log "Stopped all remaining Nginx processes"

call :write_log "All services stopped"
exit /b 0

:status
set "running=0"

if exist "%BACKEND_PID_FILE%" (
    for /f %%a in (%BACKEND_PID_FILE%) do (
        tasklist /FI "PID eq %%a" 2>nul | find "%%a" >nul
        if not errorlevel 1 (
            call :write_log "Backend is running (PID: %%a)"
            set /a running+=1
        )
    )
)

if exist "%NGINX_PID_FILE%" (
    for /f %%a in (%NGINX_PID_FILE%) do (
        tasklist /FI "PID eq %%a" 2>nul | find "%%a" >nul
        if not errorlevel 1 (
            call :write_log "Nginx is running (PID: %%a)"
            set /a running+=1
        )
    )
)

if %running% equ 2 (
    call :write_log "All services are running"
    exit /b 0
) else (
    call :write_log "Some services are not running"
    exit /b 1
)

:main
if "%1"=="" (
    echo Usage: %~nx0 {start^|stop^|status}
    exit /b 1
)

if "%1"=="start" (
    call :stop
    timeout /t 2 >nul
    call :start
) else if "%1"=="stop" (
    call :stop
) else if "%1"=="status" (
    call :status
) else (
    echo Invalid command. Use: %~nx0 {start^|stop^|status}
    exit /b 1
)
exit /b %errorlevel%