From fdcedd957d5bcb5a675a5f4a519dd6f691821d0e Mon Sep 17 00:00:00 2001
From: stevefoxru <53931856+stevefoxru@users.noreply.github.com>
Date: Wed, 23 Apr 2025 14:30:55 +0300
Subject: [PATCH] Update install.sh

---
 install.sh | 33 ++++++++++++++++++++++++---------
 1 file changed, 24 insertions(+), 9 deletions(-)

diff --git a/install.sh b/install.sh
index 5f314bc..2ce64e8 100644
--- a/install.sh
+++ b/install.sh
@@ -231,7 +231,12 @@ setup_python_env() {
 
 # Проверка и создание конфигурации
 check_config() {
-    cd awg || error_exit "Каталог awg не найден"
+    # Проверка и создание каталога awg
+    if [[ ! -d "awg" ]]; then
+        echo -e "${YELLOW}Каталог awg не найден. Создаём...${NC}"
+        mkdir -p awg || error_exit "Не удалось создать каталог awg"
+    fi
+    cd awg || error_exit "Не удалось перейти в каталог awg"
     echo -e "${BLUE}Проверка конфигурации...${NC}"
     if [[ ! -f "files/setting.ini" ]]; then
         echo -e "${YELLOW}Файл setting.ini не найден. Запрашиваем конфигурацию...${NC}"
@@ -268,15 +273,25 @@ set_permissions() {
 
 # Инициализация бота для генерации config
 initialize_bot() {
-    cd awg || error_exit "Каталог awg не найден"
+    # Проверка и создание каталога awg
+    if [[ ! -d "awg" ]]; then
+        echo -e "${YELLOW}Каталог awg не найден. Создаём...${NC}"
+        mkdir -p awg || error_exit "Не удалось создать каталог awg"
+    fi
+    cd awg || error_exit "Не удалось перейти в каталог awg"
     if [[ ! -f "files/setting.ini" ]]; then
-        ../myenv/bin/python3.11 bot_manager.py < /dev/tty &
-        local PID=$!
-        while [[ ! -f "files/setting.ini" ]]; do
-            sleep 2
-            kill -0 "$PID" 2>/dev/null || error_exit "Ошибка инициализации бота"
-        done
-        kill "$PID" && wait "$PID" 2>/dev/null
+        if [[ -f "bot_manager.py" ]]; then
+            ../myenv/bin/python3.11 bot_manager.py < /dev/tty &
+            local PID=$!
+            while [[ ! -f "files/setting.ini" ]]; do
+                sleep 2
+                kill -0 "$PID" 2>/dev/null || error_exit "Ошибка инициализации бота"
+            done
+            kill "$PID" && wait "$PID" 2>/dev/null
+        else
+            echo -e "${RED}Файл bot_manager.py не найден в каталоге awg. Проверьте репозиторий: $REPO_URL${NC}"
+            error_exit "Отсутствует необходимый файл bot_manager.py"
+        fi
     fi
     cd ..
 }