From fbbac04c4cabee906d7097382e6fa3a0742f64e9 Mon Sep 17 00:00:00 2001
From: Ground-Zerro <ksa65s571dozor@gmail.com>
Date: Mon, 12 Aug 2024 22:40:39 +1100
Subject: [PATCH] bug fix

---
 Win.bat | 71 ++++++++++++++++++++++++++++++++++-----------------------
 1 file changed, 42 insertions(+), 29 deletions(-)

diff --git a/Win.bat b/Win.bat
index 10df376..034724a 100644
--- a/Win.bat
+++ b/Win.bat
@@ -2,7 +2,8 @@
 setlocal enabledelayedexpansion
 chcp 65001 > NUL
 
-REM Проверяем, установлен ли Python 3
+REM Проверка Python 3
+:CheckPython
 python --version 2>NUL | findstr /I "Python 3" >NUL
 if ERRORLEVEL 1 (
     echo Python 3 не установлен.
@@ -12,34 +13,43 @@ if ERRORLEVEL 1 (
         pause
         exit /b 1
     ) else (
-        echo Загрузка дистрибутива...
-        powershell -Command "if ($PSVersionTable.PSVersion.Major -ge 3) {Invoke-WebRequest -Uri 'https://www.python.org/ftp/python/3.12.5/python-3.12.5-amd64.exe' -OutFile 'python_installer.exe'} else {Start-BitsTransfer -Source 'https://www.python.org/ftp/python/3.12.5/python-3.12.5-amd64.exe' -Destination 'python_installer.exe'}"
-        
-        REM Проверяем, был ли успешно скачан файл
-        if not exist "python_installer.exe" (
-            echo Ошибка загрузки установщика Python 3.
-            pause
-            exit /b 1
-        )
-        
-        REM Установка Python 3
-        echo Установка...
-        echo PS - не забудьте ее разрешить в соседнем окне
-        python_installer.exe /quiet InstallAllUsers=1 PrependPath=1
-        del /q /f python_installer.exe
-        
-        REM Нужно обновить системные PATH, но в этом сеансе не получится
-        echo Вроде бы все прошло удачно, но нужно обновить окружение,
-        echo закройте это окно и запустите скрипт еще раз.
-        pause
-        exit
+        call :InstallPython
     )
 ) else (
-    echo Похоже Python 3 имеется в системе.
+    echo Python 3 установлен.
+)
+goto :CheckModules
+
+REM Инсталляция Python 3
+:InstallPython
+echo Загрузка дистрибутива...
+powershell -Command "if ($PSVersionTable.PSVersion.Major -ge 3) {Invoke-WebRequest -Uri 'https://www.python.org/ftp/python/3.12.5/python-3.12.5-amd64.exe' -OutFile 'python_installer.exe'} else {Start-BitsTransfer -Source 'https://www.python.org/ftp/python/3.12.5/python-3.12.5-amd64.exe' -Destination 'python_installer.exe'}"
+
+REM Проверяем успешность загрузки
+if not exist "python_installer.exe" (
+    echo Ошибка загрузки установщика Python 3.
+    pause
+    exit /b 1
 )
 
-REM Проверяем наличие необходимых библиотек
-set "modules=requests dnspython ipaddress configparser httpx"
+REM Установка Python 3
+echo Установка...
+echo PS - не забудьте ее разрешить в соседнем окне
+python_installer.exe /quiet InstallAllUsers=1 PrependPath=1
+del /q /f python_installer.exe
+
+REM Оповещение о перезапуске
+echo.
+echo Установка завершена, но требуется обновить окружение.
+echo - закройте это окно и запустите скрипт снова.
+pause
+exit /b 0
+
+REM Проверка и установка необходимых модулей Python
+:CheckModules
+set "modules=requests dnspython ipaddress configparser httpx colorama"
+echo.
+echo Проверка необходимых библиотек...
 
 for %%m in (%modules%) do (
     pip show %%m >NUL 2>&1
@@ -53,8 +63,10 @@ for %%m in (%modules%) do (
     )
 )
 
+goto :DownloadMain
 
-REM Скачиваем main.py
+REM Загрузка и запуск main.py
+:DownloadMain
 echo Загрузка Domain Mapper...
 powershell -Command "if ($PSVersionTable.PSVersion.Major -ge 3) {Invoke-WebRequest -Uri 'https://raw.githubusercontent.com/Ground-Zerro/DomainMapper/main/main.py' -OutFile 'main.py'} else {Start-BitsTransfer -Source 'https://raw.githubusercontent.com/Ground-Zerro/DomainMapper/main/main.py' -Destination 'main.py'}"
 
@@ -63,8 +75,9 @@ if not exist "main.py" (
     pause
     exit /b 1
 )
+
 cls
-REM Запуск main.py в Python 3
+REM Запуск main.py
 echo Запускаем...
 python main.py
 if ERRORLEVEL 1 (
@@ -76,6 +89,6 @@ if ERRORLEVEL 1 (
 
 move /y domain-ip-resolve.txt %UserProfile%\Desktop\domain-ip-resolve.txt
 echo Программа завершена.
-endlocal
 del /q /f main.py
-exit
+endlocal
+exit /b 0