Update install.sh

install.sh

@@ -14,6 +14,7 @@ SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
 SCRIPT_NAME="$(basename "$0")"
 SCRIPT_PATH="$SCRIPT_DIR/$SCRIPT_NAME"
 
+# Параметры скрипта
 while [[ "$#" -gt 0 ]]; do
     case $1 in
         --quiet) ENABLE_LOGS=false ;;
@@ -27,6 +28,7 @@ while [[ "$#" -gt 0 ]]; do
     shift
 done
 
+# Получение версии Ubuntu
 get_ubuntu_version() {
     if command -v lsb_release &>/dev/null; then
         UBUNTU_VERSION=$(lsb_release -rs)
@@ -34,316 +36,110 @@ get_ubuntu_version() {
         DISTRIB_ID=$(lsb_release -is)
         [[ "$DISTRIB_ID" != "Ubuntu" ]] && { echo -e "${RED}Скрипт поддерживает только Ubuntu. Обнаружена система: $DISTRIB_ID${NC}"; exit 1; }
     else
-        echo -e "${RED}lsb_release не установлен. Необходимо установить lsb-release${NC}"
+        echo -e "${RED}lsb_release не установлен. Установите пакет lsb-release.${NC}"
         exit 1
     fi
 }
 
+# Функция запуска команд с индикатором
 run_with_spinner() {
-    local description="$1"
+    local description="$1"; shift
-    shift
     local cmd="$@"
-
     if [ "$ENABLE_LOGS" = true ]; then
         echo -e "${BLUE}${description}...${NC}"
         eval "$cmd"
-        local status=$?
+        local stat=$?
-        if [ $status -eq 0 ]; then
+        if [ $stat -eq 0 ]; then echo -e "${GREEN}${description}... Done!${NC}\n"; else echo -e "${RED}${description}... Failed!${NC}\n"; exit 1; fi
-            echo -e "${GREEN}${description}... Done!${NC}\n"
-        else
-            echo -e "${RED}${description}... Failed!${NC}"
-            echo -e "${RED}Ошибка при выполнении команды: $cmd${NC}\n"
-            exit 1
-        fi
     else
-        local stdout_temp=$(mktemp)
+        local out=$(mktemp) err=$(mktemp)
-        local stderr_temp=$(mktemp)
+        eval "$cmd" >"$out" 2>"$err" & pid=$!
-
+        local spinner='|/-\\' i=0
-        eval "$cmd" >"$stdout_temp" 2>"$stderr_temp" &
+        while kill -0 "$pid" 2>/dev/null; do printf "\r${BLUE}${description}...${NC} ${spinner:i++%${#spinner}:1}"; sleep 0.1; done
-        local pid=$!
+        wait "$pid"; stat=$?
-
+        if [ $stat -eq 0 ]; then printf "\r${BLUE}${description}...${NC} ${GREEN}Done!${NC}\n\n"; else printf "\r${BLUE}${description}...${NC} ${RED}Failed!${NC}\n\n"; echo -e "${RED}Ошибка: $cmd${NC}"; cat "$err"; rm -f "$out" "$err"; exit 1; fi
-        local spinner='|/-\'
+        rm -f "$out" "$err"
-        local i=0
-
-        echo -ne "${BLUE}${description}...${NC} ${spinner:i++%${#spinner}:1}"
-
-        while kill -0 "$pid" 2>/dev/null; do
-            printf "\r${BLUE}${description}...${NC} ${spinner:i++%${#spinner}:1}"
-            sleep 0.1
-        done
-
-        wait "$pid"
-        local status=$?
-        
-        if [ $status -eq 0 ]; then
-            printf "\r${BLUE}${description}...${NC} ${spinner:i%${#spinner}:1} ${GREEN}Done!${NC}\n\n"
-        else
-            printf "\r${BLUE}${description}...${NC} ${spinner:i%${#spinner}:1} ${RED}Failed!${NC}\n\n"
-            echo -e "${RED}Ошибка при выполнении команды: $cmd${NC}"
-            echo -e "${RED}Вывод ошибки:${NC}"
-            cat "$stderr_temp"
-        fi
-
-        rm -f "$stdout_temp" "$stderr_temp"
-
-        if [ $status -ne 0 ]; then
-            exit 1
-        fi
     fi
 }
 
-check_updates() {
+# Обновление и очистка системы
-    # Переход в директорию репозитория
-    if [[ ! -d "amnezia-bot" ]]; then
-        echo -e "${RED}Директория amnezia-bot не найдена. Проверьте установку.${NC}\n"
-        return 1
-    fi
-    cd amnezia-bot || { echo -e "${RED}Не удалось перейти в директорию amnezia-bot${NC}\n"; return 1; }
-
-    # Проверка, является ли текущая директория Git-репозиторием
-    if [[ ! -d ".git" ]]; then
-        echo -e "${RED}Текущая директория не является Git-репозиторием${NC}\n"
-        return 1
-    fi
-
-    local stdout_temp=$(mktemp)
-    local stderr_temp=$(mktemp)
-    
-    if ! git pull >"$stdout_temp" 2>"$stderr_temp"; then
-        echo -e "${RED}Ошибка при проверке обновлений:${NC}"
-        cat "$stderr_temp"
-        rm -f "$stdout_temp" "$stderr_temp"
-        echo
-        return 1
-    fi
-
-    local changes=$(cat "$stdout_temp")
-    rm -f "$stdout_temp" "$stderr_temp"
-
-    if [[ "$changes" == "Already up to date." ]]; then
-        echo -e "${GREEN}Обновления не требуются${NC}\n"
-    elif [[ -n "$changes" ]]; then
-        echo -e "${YELLOW}Обнаружены обновления:${NC}"
-        echo "$changes"
-        echo
-        echo -e "${GREEN}1${NC}. Установить обновления"
-        echo -e "${YELLOW}2${NC}. Отменить"
-        
-        echo -ne "\n${BLUE}Выберите действие:${NC} "
-        read action
-        
-        case $action in
-            1)
-                echo -e "\n${GREEN}Установка обновлений...${NC}"
-                git pull
-                run_with_spinner "Перезапуск службы" "sudo systemctl restart $SERVICE_NAME -qq"
-                echo -e "${GREEN}Обновления установлены и применены${NC}\n"
-                ;;
-            2)
-                echo -e "${YELLOW}Установка обновлений отменена${NC}\n"
-                return 0
-                ;;
-            *)
-                echo -e "${RED}Некорректный ввод${NC}\n"
-                return 1
-                ;;
-        esac
-    else
-        echo -e "${RED}Неожиданный ответ при проверке обновлений${NC}\n"
-    fi
-
-    # Возвращаемся в исходную директорию (опционально)
-    cd .. || { echo -e "${RED}Не удалось вернуться в исходную директорию${NC}\n"; return 1; }
-}
-
-remove_amneziawg() {
-    echo -e "\n${YELLOW}Внимание: Это действие удалит AmneziaWG и связанные данные. Продолжить? (y/n)${NC}"
-    read confirm
-    if [[ "$confirm" =~ ^[Yy]$ ]]; then
-        # Остановка службы, если она активна
-        systemctl is-active --quiet "$SERVICE_NAME" && run_with_spinner "Остановка службы" "sudo systemctl stop $SERVICE_NAME -qq"
-        
-        # Удаление Docker контейнеров и образов, связанных с AmneziaWG (предполагается использование Docker)
-        run_with_spinner "Удаление Docker контейнеров AmneziaWG" "docker ps -a -q -f name=amnezia-wg | xargs -r docker rm -f"
-        run_with_spinner "Удаление Docker образов AmneziaWG" "docker images -q amneziawg/amnezia-wg | uniq | xargs -r docker rmi -f"
-        
-        # Удаление конфигурационных файлов (предполагается, что они в директории awg/files)
-        run_with_spinner "Удаление конфигураций AmneziaWG" "rm -rf $(pwd)/awg/files"
-        
-        echo -e "${GREEN}AmneziaWG успешно удален${NC}"
-    else
-        echo -e "${YELLOW}Удаление отменено${NC}"
-    fi
-}
-
-service_control_menu() {
-    while true; do
-        echo -e "\n${BLUE}=== Управление службой $SERVICE_NAME ===${NC}"
-        sudo systemctl status "$SERVICE_NAME" | grep -E "Active:|Loaded:"
-        echo -e "\n${GREEN}1${NC}. Остановить службу"
-        echo -e "${GREEN}2${NC}. Перезапустить службу"
-        echo -e "${GREEN}3${NC}. Переустановить службу"
-        echo -e "${RED}4${NC}. Удалить службу"
-        echo -e "${GREEN}5${NC}. Удалить AmneziaWG"
-        echo -e "${YELLOW}5${NC}. Назад"
-        
-        echo -ne "\n${BLUE}Выберите действие:${NC} "
-        read action
-        case $action in
-            1) run_with_spinner "Остановка службы" "sudo systemctl stop $SERVICE_NAME -qq" ;;
-            2) run_with_spinner "Перезапуск службы" "sudo systemctl restart $SERVICE_NAME -qq" ;;
-            3) create_service ;;
-            4) run_with_spinner "Удаление службы" "sudo systemctl stop $SERVICE_NAME -qq && sudo systemctl disable $SERVICE_NAME -qq && sudo rm /etc/systemd/system/$SERVICE_NAME.service && sudo systemctl daemon-reload -qq" ;;
-            5) remove_amneziawg ;;
-            6) return 0 ;;
-            *) echo -e "${RED}Некорректный ввод${NC}" ;;
-        esac
-    done
-}
-
-reinstall_bot() {
-    echo -e "\n${YELLOW}Внимание: Переустановка бота удалит текущие настройки и данные. Продолжить? (y/n)${NC}"
-    read confirm
-    if [[ "$confirm" =~ ^[Yy]$ ]]; then
-        # Остановка и удаление службы, если она существует
-        systemctl is-active --quiet "$SERVICE_NAME" && run_with_spinner "Остановка службы" "sudo systemctl stop $SERVICE_NAME -qq"
-        systemctl list-units --type=service --all | grep -q "$SERVICE_NAME.service" && run_with_spinner "Удаление службы" "sudo systemctl disable $SERVICE_NAME -qq && sudo rm /etc/systemd/system/$SERVICE_NAME.service && sudo systemctl daemon-reload -qq"
-        
-        # Удаление старых файлов бота
-        run_with_spinner "Удаление старых файлов" "rm -rf amnezia-bot"
-        
-        # Запуск полной установки заново
-        install_bot
-        
-        echo -e "${GREEN}Бот успешно переустановлен${NC}"
-    else
-        echo -e "${YELLOW}Переустановка отменена${NC}"
-    fi
-}
-
-installed_menu() {
-    while true; do
-        echo -e "\n${BLUE}=== AWG Docker Telegram Bot ===${NC}"
-        echo -e "${GREEN}1${NC}. Проверить обновления"
-        echo -e "${GREEN}2${NC}. Управление службой"
-        echo -e "${GREEN}3${NC}. Переустановить бота"
-        echo -e "${YELLOW}4${NC}. Выход"
-        
-        echo -ne "\n${BLUE}Выберите действие:${NC} "
-        read action
-        case $action in
-            1) check_updates ;;
-            2) service_control_menu ;;
-            3) reinstall_bot ;;
-            4) exit 0 ;;
-            *) echo -e "${RED}Некорректный ввод${NC}" ;;
-        esac
-    done
-}
-
 update_and_clean_system() {
+    # Ждём освобождения APT
+    while fuser /var/lib/dpkg/lock-frontend >/dev/null 2>&1; do
+        echo -e "${YELLOW}APT занят. Ждём...${NC}"; sleep 2
+    done
     run_with_spinner "Обновление системы" "sudo apt-get update -qq && sudo apt-get upgrade -y -qq"
     run_with_spinner "Очистка системы" "sudo apt-get autoclean -qq && sudo apt-get autoremove --purge -y -qq"
 }
 
+# Обязательная установка Python 3.11
 check_python() {
     if command -v python3.11 &>/dev/null; then
-        echo -e "\n${GREEN}Python 3.11 установлен${NC}"
+        echo -e "\n${GREEN}Python 3.11 уже установлен${NC}"; return 0
-        return 0
     fi
-    
+    echo -e "\n${YELLOW}Устанавливаю Python 3.11...${NC}"
-    echo -e "\n${RED}Python 3.11 не установлен${NC}"
+    # Ожидание dpkg lock на Ubuntu 24.04
-    read -p "Установить Python 3.11? (y/n): " install_python
+    if [[ "$UBUNTU_VERSION" == "24.04" ]]; then
-    
+        local max=30 cnt=1
-    if [[ "$install_python" =~ ^[Yy]$ ]]; then
+        while fuser /var/lib/dpkg/lock-frontend >/dev/null 2>&1; do
-       if [[ "$UBUNTU_VERSION" == "24.04" ]]; then
+            echo -e "${YELLOW}Ожидание dpkg lock ($cnt/$max)${NC}"; ((cnt++))
-            local max_attempts=30
+            [ $cnt -gt $max ] && { echo -e "${RED}Таймаут ожидания dpkg lock${NC}"; exit 1; }
-            local attempt=1
+            sleep 10
-            
+        done
-            while fuser /var/lib/dpkg/lock-frontend >/dev/null 2>&1; do
-                echo -e "${YELLOW}Ожидание освобождения dpkg lock (попытка $attempt из $max_attempts)${NC}"
-                attempt=$((attempt + 1))
-                if [ $attempt -gt $max_attempts ]; then
-                    echo -e "${RED}Превышено время ожидания освобождения dpkg lock${NC}"
-                    exit 1
-                fi
-                sleep 10
-            done
-        fi
-        
-        run_with_spinner "Установка Python 3.11" "sudo apt-get install software-properties-common -y && sudo add-apt-repository ppa:deadsnakes/ppa -y && sudo apt-get update -qq && sudo apt-get install python3.11 python3.11-venv python3.11-dev -y -qq"
-            
-        if ! command -v python3.11 &>/dev/null; then
-            echo -e "\n${RED}Не удалось установить Python 3.11${NC}"
-            exit 1
-        fi
-    else
-        echo -e "\n${RED}Установка Python 3.11 обязательна${NC}"
-        exit 1
     fi
+    run_with_spinner "Установка Python 3.11" "sudo apt-get install -y software-properties-common && sudo add-apt-repository -y ppa:deadsnakes/ppa && sudo apt-get update -qq && sudo apt-get install -y python3.11 python3.11-venv python3.11-dev -qq"
+    command -v python3.11 &>/dev/null || { echo -e "\n${RED}Не удалось установить Python 3.11${NC}"; exit 1; }
+    echo -e "\n${GREEN}Python 3.11 успешно установлен${NC}"
 }
 
+# Установка зависимостей VPN-бота
 install_dependencies() {
-    run_with_spinner "Установка зависимостей" "sudo apt-get install jq net-tools iptables resolvconf git -y -qq"
+    run_with_spinner "Установка зависимостей" "sudo apt-get install -y jq net-tools iptables resolvconf git -qq"
 }
 
+# Установка и конфиг needrestart
 install_and_configure_needrestart() {
-    run_with_spinner "Установка needrestart" "sudo apt-get install needrestart -y -qq"
+    run_with_spinner "Установка needrestart" "sudo apt-get install -y needrestart -qq"
-    sudo sed -i 's/^#\?\(nrconf{restart} = "\).*$/\1a";/' /etc/needrestart/needrestart.conf
+    sudo sed -i 's/^#\?\(nrconf{restart} = \"\).*$/\1a\";/' /etc/needrestart/needrestart.conf
-    grep -q 'nrconf{restart} = "a";' /etc/needrestart/needrestart.conf || echo 'nrconf{restart} = "a";' | sudo tee /etc/needrestart/needrestart.conf >/dev/null 2>&1
+    grep -q 'nrconf{restart} = "a";' /etc/needrestart/needrestart.conf || echo 'nrconf{restart} = "a";' | sudo tee -a /etc/needrestart/needrestart.conf >/dev/null
 }
 
+# Клонирование репозитория
 clone_repository() {
-    if [[ -d "amnezia-bot" ]]; then
+    if [[ -d "amnezia-bot" ]]; then cd amnezia-bot; return; fi
-        echo -e "\n${YELLOW}Репозиторий существует${NC}"
+    run_with_spinner "Клонирование репозитория" "git clone https://github.com/stevefoxru/amnezia-bot.git -q"
-        cd amnezia-bot || { echo -e "\n${RED}Ошибка перехода в директорию${NC}"; exit 1; }
+    cd amnezia-bot
-        return 0
-    fi
-    
-    run_with_spinner "Клонирование репозитория" "git clone https://github.com/stevefoxru/amnezia-bot.git >/dev/null 2>&1"
-    cd amnezia-bot || { echo -e "\n${RED}Ошибка перехода в директорию${NC}"; exit 1; }
 }
 
+# Настройка виртуального окружения
 setup_venv() {
-    if [[ -d "myenv" ]]; then
+    if [[ -d "myenv" ]]; then return; fi
-        echo -e "\n${YELLOW}Виртуальное окружение существует${NC}"
+    run_with_spinner "Создание virtualenv" "python3.11 -m venv myenv"
-        return 0
+    source myenv/bin/activate
-    fi
+    run_with_spinner "Установка Python-зависимостей" "pip install --upgrade pip && pip install -r requirements.txt"
-    
+    deactivate
-    run_with_spinner "Настройка виртуального окружения" "python3.11 -m venv myenv && source myenv/bin/activate && pip install --upgrade pip && pip install -r $(pwd)/requirements.txt && deactivate"
 }
 
+# Права на скрипты
 set_permissions() {
-    find . -type f -name "*.sh" -exec chmod +x {} \; 2>chmod_error.log
+    find . -type f -name "*.sh" -exec chmod +x {} \; || exit 1
-    local status=$?
-    [[ $status -ne 0 ]] && { cat chmod_error.log; rm -f chmod_error.log; exit 1; }
-    rm -f chmod_error.log
 }
 
+# Инициализация бота для генерации config
 initialize_bot() {
-    cd awg || { echo -e "\n${RED}Ошибка перехода в директорию${NC}"; exit 1; }
+    cd awg || exit 1
-    
     ../myenv/bin/python3.11 bot_manager.py < /dev/tty &
-    local BOT_PID=$!
+    local PID=$!
-    
     while [ ! -f "files/setting.ini" ]; do
         sleep 2
-        kill -0 "$BOT_PID" 2>/dev/null || { echo -e "\n${RED}Бот завершил работу до инициализации${NC}"; exit 1; }
+        kill -0 "$PID" 2>/dev/null || { echo -e "${RED}Ошибка инициализации бота${NC}"; exit 1; }
     done
-    
+    kill "$PID" && wait "$PID" 2>/dev/null
-    kill "$BOT_PID"
-    wait "$BOT_PID" 2>/dev/null
     cd ..
 }
 
+# Создание systemd-сервиса
 create_service() {
-    # Проверяем, что директории существуют
+    cat > /tmp/service_file <<EOF
-    if [[ ! -d "/root/amnezia-bot/awg" ]] || [[ ! -f "/root/amnezia-bot/myenv/bin/python3.11" ]]; then
-        echo -e "${RED}Ошибка: необходимые директории или файлы отсутствуют${NC}"
-        exit 1
-    fi
-
-    cat > /tmp/service_file << EOF
 [Unit]
 Description=AmneziaVPN Docker Telegram Bot
 After=network.target
@@ -357,15 +153,83 @@ Restart=always
 [Install]
 WantedBy=multi-user.target
 EOF
-
+    run_with_spinner "Установка службы" "sudo mv /tmp/service_file /etc/systemd/system/$SERVICE_NAME.service"
-    run_with_spinner "Создание службы" "sudo mv /tmp/service_file /etc/systemd/system/$SERVICE_NAME.service"
+    run_with_spinner "Обновление systemd" "sudo systemctl daemon-reload -q"
-    run_with_spinner "Обновление systemd" "sudo systemctl daemon-reload -qq"
+    run_with_spinner "Запуск службы" "sudo systemctl start $SERVICE_NAME -q"
-    run_with_spinner "Запуск службы" "sudo systemctl start $SERVICE_NAME -qq"
+    run_with_spinner "Включение автозапуска" "sudo systemctl enable $SERVICE_NAME -q"
-    run_with_spinner "Включение автозапуска" "sudo systemctl enable $SERVICE_NAME -qq"
-    
-    systemctl is-active --quiet "$SERVICE_NAME" && echo -e "\n${GREEN}Служба запущена${NC}" || echo -e "\n${RED}Ошибка запуска службы${NC}"
 }
 
+# Проверка и применение обновлений
+check_updates() {
+    cd amnezia-bot || return 1
+    local out err
+    out=$(mktemp); err=$(mktemp)
+    git pull >"$out" 2>"$err" || { cat "$err"; rm -f "$out" "$err"; cd ..; return 1; }
+    local changes=$(<"$out"); rm -f "$out" "$err"
+    if [[ "$changes" == "Already up to date." ]]; then
+        echo -e "${GREEN}Обновлений нет${NC}"; cd ..; return 0
+    fi
+    echo -e "${YELLOW}Найдено обновление:${NC}\n$changes"
+    echo -ne "${BLUE}1) Установить 2) Отменить: ${NC}"; read choice
+    case $choice in 1) git pull; run_with_spinner "Перезапуск службы" "sudo systemctl restart $SERVICE_NAME -q" ;; *) echo "Отменено" ;; esac
+    cd ..
+}
+
+# Удаление AmneziaWG
+remove_amneziawg() {
+    echo -ne "${YELLOW}Удалить AmneziaWG? (y/n): ${NC}"; read ans
+    [[ "$ans" =~ ^[Yy]$ ]] || return
+    systemctl is-active --quiet "$SERVICE_NAME" && run_with_spinner "Остановка" "sudo systemctl stop $SERVICE_NAME -q"
+    run_with_spinner "Удаление контейнеров" "docker ps -aq -f name=amnezia-wg | xargs -r docker rm -f"
+    run_with_spinner "Удаление образов" "docker images -q amneziawg/amnezia-wg | uniq | xargs -r docker rmi -f"
+    run_with_spinner "Удаление конфигов" "rm -rf $(pwd)/awg/files"
+}
+
+# Меню управления службой
+service_control_menu() {
+    while true; do
+        echo -e "\n${BLUE}Управление службой${NC}"
+        sudo systemctl status "$SERVICE_NAME" | grep -E "Active:|Loaded:"
+        echo -e "1) Остановить 2) Перезапустить 3) Переустановить 4) Удалить службу 5) Удалить AmneziaWG 6) Назад"
+        echo -ne "${BLUE}Выберите: ${NC}"; read act
+        case $act in
+            1) run_with_spinner "Остановка" "sudo systemctl stop $SERVICE_NAME -q" ;;  
+            2) run_with_spinner "Перезапуск" "sudo systemctl restart $SERVICE_NAME -q" ;;  
+            3) reinstall_bot ;;  
+            4) run_with_spinner "Удаление службы" "sudo systemctl disable $SERVICE_NAME -q && sudo rm /etc/systemd/system/$SERVICE_NAME.service && sudo systemctl daemon-reload -q" ;;  
+            5) remove_amneziawg ;;  
+            6) break ;;  
+            *) echo "Неверно" ;;  
+        esac
+    done
+}
+
+# Переустановка бота
+reinstall_bot() {
+    echo -ne "${YELLOW}Переустановить бота? (y/n): ${NC}"; read ans
+    [[ "$ans" =~ ^[Yy]$ ]] || return
+    systemctl is-active --quiet "$SERVICE_NAME" && run_with_spinner "Остановка" "sudo systemctl stop $SERVICE_NAME -q"
+    systemctl list-units --type=service --all | grep -q "$SERVICE_NAME.service" && run_with_spinner "Удаление службы" "sudo systemctl disable $SERVICE_NAME -q && sudo rm /etc/systemd/system/$SERVICE_NAME.service && sudo systemctl daemon-reload -q"
+    run_with_spinner "Удаление файлов" "rm -rf amnezia-bot"
+    install_bot
+}
+
+# Главное меню если установлено
+installed_menu() {
+    while true; do
+        echo -e "\n${GREEN}1) Проверить обновления 2) Управление службой 3) Переустановить 4) Выход${NC}"
+        echo -ne "${BLUE}Выберите: ${NC}"; read opt
+        case $opt in
+            1) check_updates ;;  
+            2) service_control_menu ;;  
+            3) reinstall_bot ;;  
+            4) exit 0 ;;  
+            *) echo "Неверно" ;;  
+        esac
+    done
+}
+
+# Полная установка бота
 install_bot() {
     get_ubuntu_version
     update_and_clean_system
@@ -377,10 +241,18 @@ install_bot() {
     set_permissions
     initialize_bot
     create_service
+    echo -e "${GREEN}Установка завершена!${NC}"   
 }
 
+# Точка входа
 main() {
-    systemctl list-units --type=service --all | grep -q "$SERVICE_NAME.service" && installed_menu || { install_bot; ( sleep 1; rm -- "$SCRIPT_PATH" ) & exit 0; }
+    get_ubuntu_version
+    if systemctl list-units --type=service --all | grep -q "$SERVICE_NAME.service"; then
+        installed_menu
+    else
+        install_bot
+        rm -- "$SCRIPT_PATH"
+    fi
 }
 
 main