@echo off
setlocal EnableExtensions EnableDelayedExpansion

set "WUDL=C:\Windows\SoftwareDistribution\Download"
set "WINTEMP=C:\Windows\Temp"
set "TMPCOPY=%TEMP%\diskclean_elevated.bat"

if /i "%~1"=="ADMIN" goto ADMIN_ENTRY

title Windows 디스크 정리

echo ==============================================
echo   Windows 디스크 정리
echo ==============================================
echo   지우는 것
echo     - 임시 폴더 : %TEMP%
echo     - 임시 폴더 : %WINTEMP%   (관리자 권한)
echo     - Windows Update 잔여 파일           (관리자 권한)
echo.
echo   건드리지 않는 것
echo     - 문서 / 바탕화면 / 다운로드 / 휴지통
echo     - 레지스트리, 서비스 설정, 시작 프로그램
echo ==============================================

set "BEFORE="
call :FREEMB
if defined FREEMB set "BEFORE=!FREEMB!"
if defined BEFORE (
    call :MB2GB !BEFORE!
    echo   현재 C드라이브 여유 공간 : !GBTXT!
    echo ==============================================
)

choice /c YN /n /m "정리를 시작할까요? (Y/N): "
if errorlevel 2 goto ENDX
echo.

echo [1/3] 사용자 임시 폴더 정리...
call :CLEANDIR "%TEMP%"

echo.
echo [2/3] 관리자 영역 정리 - Windows 임시 폴더 / Update 잔여
set "ADMINDONE="
fltmc >nul 2>&1
if errorlevel 1 goto ELEVATE
call :MACHINE_CLEAN
set "ADMINDONE=1"
goto RESULT

:ELEVATE
echo       관리자 권한이 필요합니다. UAC 창에서 "예" 를 선택하세요.
set "RUNFILE=%~f0"
copy /y "%~f0" "%TMPCOPY%" >nul 2>&1
if exist "%TMPCOPY%" set "RUNFILE=%TMPCOPY%"
powershell -NoProfile -Command "Start-Process -FilePath '%RUNFILE%' -ArgumentList 'ADMIN' -Verb RunAs -Wait"
if not errorlevel 1 set "ADMINDONE=1"
if exist "%TMPCOPY%" del /f /q "%TMPCOPY%" >nul 2>&1
if not defined ADMINDONE (
    echo       관리자 권한을 얻지 못했습니다. 이 부분은 건너뜁니다.
    echo       사용자 임시 폴더 정리는 이미 끝났습니다.
)

:RESULT
echo.
echo [3/3] 결과
set "AFTER="
call :FREEMB
if defined FREEMB set "AFTER=!FREEMB!"
if defined BEFORE if defined AFTER (
    set /a GAINED=AFTER-BEFORE
    if !GAINED! LSS 0 set /a GAINED=0
    call :MB2GB !AFTER!
    echo       확보한 공간   : !GAINED! MB
    echo       현재 여유 공간 : !GBTXT!
)
if not defined ADMINDONE echo       ^(관리자 영역은 정리되지 않았습니다^)

call :BIGLEFTOVERS

echo.
pause
exit /b

:ENDX
echo.
echo 취소했습니다.
exit /b


:ADMIN_ENTRY
title Windows 디스크 정리 - 관리자 단계
fltmc >nul 2>&1
if errorlevel 1 goto NOADMIN
call :MACHINE_CLEAN
echo.
echo 관리자 단계 완료. 이 창을 닫으면 원래 창에서 결과가 나옵니다.
echo.
pause
exit /b

:NOADMIN
echo.
echo [실패] 관리자 권한을 얻지 못했습니다.
echo.
pause
exit /b 1


:MACHINE_CLEAN
echo       (a) %WINTEMP% 정리...
call :CLEANDIR "%WINTEMP%"

echo       (b) Windows Update 서비스 정지...
net stop wuauserv >nul 2>&1
net stop bits >nul 2>&1

echo       (c) Windows Update 잔여 파일 삭제...
if exist "%WUDL%" (
    rd /s /q "%WUDL%" 2>nul
    if exist "%WUDL%" (
        call :CLEANDIR "%WUDL%"
    ) else (
        md "%WUDL%" 2>nul
        echo           삭제 완료.
    )
) else (
    echo           잔여 파일이 없습니다.
)

echo       (d) Windows Update 서비스 재시작...
net start bits >nul 2>&1
net start wuauserv >nul 2>&1
goto :eof


:CLEANDIR
if not exist %1 (
    echo           대상 없음 : %~1
    goto :eof
)
for /f "delims=" %%d in ('dir /b /ad %1 2^>nul') do rd /s /q "%~1\%%d" 2>nul
del /f /q /a "%~1\*" >nul 2>&1
set "LEFT="
for /f "delims=" %%d in ('dir /b /a %1 2^>nul') do set "LEFT=1"
if defined LEFT (
    echo           정리했습니다. 사용 중인 파일 일부는 남습니다 - 정상입니다.
) else (
    echo           정리 완료 : %~1
)
goto :eof


:BIGLEFTOVERS
set "BIG="
if exist "C:\Windows.old" set "BIG=1"
if exist "C:\$WinREAgent" set "BIG=1"
if not defined BIG goto :eof
echo.
echo ==============================================
echo  [참고] 이 배치가 건드리지 않는 대용량 항목이 있습니다.
if exist "C:\Windows.old" echo     - C:\Windows.old      : 이전 Windows 설치본. 보통 20~30 GB
if exist "C:\$WinREAgent" echo     - C:\$WinREAgent      : 업데이트 잔여
echo.
echo  삭제하려면 Windows 기본 기능을 쓰세요. 수동 삭제는 권장하지 않습니다.
echo     설정 - 시스템 - 저장소 - 임시 파일
echo ==============================================
goto :eof


:FREEMB
set "FREEMB="
for /f "delims=" %%a in ('powershell -NoProfile -Command "[int]((Get-PSDrive C).Free/1MB)" 2^>nul') do set "FREEMB=%%a"
goto :eof


:MB2GB
set /a _g=%1/1024
set /a _d=(%1*10/1024) %% 10
set "GBTXT=!_g!.!_d! GB"
goto :eof