Update install.sh

install.sh

@@ -5,9 +5,6 @@ SERVICE_NAME="awg_bot"
 REPO_URL="https://github.com/stevefoxru/amnezia-bot.git"
 REPO_API="https://api.github.com/repos/stevefoxru/amnezia-bot"
 LOCAL_VERSION_FILE="/root/amnezia-bot/.version"
-BOT_DIR="/root/amnezia-bot"
-AWG_DIR="$BOT_DIR/awg"
-FILES_DIR="$AWG_DIR/files"
 
 # Цвета для вывода
 GREEN=$'\033[0;32m'
@@ -33,45 +30,35 @@ run_with_spinner() {
     local cmd="$@"
     if [ "$ENABLE_LOGS" = true ]; then
         echo -e "${BLUE}${description}...${NC}"
-        eval "$cmd" || error_exit "Не удалось выполнить: $cmd"
-        echo -e "${GREEN}${description}... Done!${NC}\n"
+        eval "$cmd"
+        local stat=$?
+        if [ $stat -eq 0 ]; then echo -e "${GREEN}${description}... Done!${NC}\n"; else echo -e "${RED}${description}... Failed!${NC}\n"; error_exit "$cmd"; fi
     else
         local out=$(mktemp) err=$(mktemp)
         eval "$cmd" >"$out" 2>"$err" & pid=$!
         local spinner='|/-\\' i=0
-        while kill -0 "$pid" 2>/dev/null; do
-            printf "\r${BLUE}${description}...${NC} ${spinner:i++%${#spinner}:1}"
-            sleep 0.1
-        done
+        while kill -0 "$pid" 2>/dev/null; do printf "\r${BLUE}${description}...${NC} ${spinner:i++%${#spinner}:1}"; sleep 0.1; done
         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"
-            error_exit "$cmd"
-        fi
+        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"; error_exit "$cmd"; fi
         rm -f "$out" "$err"
     fi
 }
 
 # Функция проверки обновлений на GitHub
 check_github_updates() {
-    local auto_mode="$1"
-    cd "$BOT_DIR" || error_exit "Каталог $BOT_DIR не найден"
-
-    # Проверка наличия git
-    command -v git &>/dev/null || error_exit "git не установлен"
+    local current_sha local_sha latest_sha auto_mode="$1"
+    cd /root/amnezia-bot || { echo -e "${RED}Каталог amnezia-bot не найден${NC}"; return 1; }
     
     # Получение текущего SHA коммита
     local_sha=$(git rev-parse HEAD 2>/dev/null || echo "unknown")
     
     # Получение последнего коммита через GitHub API
-    command -v curl &>/dev/null || error_exit "curl не установлен"
-    latest_sha=$(curl -s "$REPO_API/commits/main" | jq -r '.sha' 2>/dev/null)
-    [[ -z "$latest_sha" ]] && { echo -e "${RED}Не удалось получить данные с GitHub${NC}"; cd ..; return 1; }
+    if command -v curl &>/dev/null; then
+        latest_sha=$(curl -s "$REPO_API/commits/main" | jq -r '.sha' 2>/dev/null)
+        [[ -z "$latest_sha" ]] && { echo -e "${RED}Не удалось получить данные с GitHub${NC}"; cd ..; return 1; }
+    else
+        echo -e "${RED}curl не установлен${NC}"; cd ..; return 1
+    fi
     
     # Сравнение версий
     if [[ "$local_sha" == "$latest_sha" ]]; then
@@ -81,6 +68,7 @@ check_github_updates() {
     
     echo -e "${YELLOW}Доступно обновление (текущий SHA: $local_sha, последний SHA: $latest_sha)${NC}"
     if [[ "$auto_mode" == "--auto" ]]; then
+        # Проверка на наличие локальных изменений
         if git status --porcelain | grep -q .; then
             echo -e "${YELLOW}Обнаружены локальные изменения. Сбрасываем их...${NC}"
             run_with_spinner "Сброс локальных изменений" "git reset --hard && git clean -fd"
@@ -88,9 +76,7 @@ check_github_updates() {
         run_with_spinner "Обновление репозитория" "git pull"
         echo "$latest_sha" > "$LOCAL_VERSION_FILE"
         check_script_update
-        setup_python_env
-        check_config
-        create_service
+        run_with_spinner "Обновление Python-зависимостей" "source myenv/bin/activate && pip install aiogram==2.25.1 aiohttp==3.8.6 apscheduler==3.10.4 humanize==4.9.0 pytz==2023.3.post1 && deactivate"
         run_with_spinner "Перезапуск службы" "systemctl restart $SERVICE_NAME"
     else
         echo -ne "${BLUE}1) Установить 2) Отменить: ${NC}"; read choice
@@ -102,9 +88,7 @@ check_github_updates() {
             run_with_spinner "Обновление репозитория" "git pull"
             echo "$latest_sha" > "$LOCAL_VERSION_FILE"
             check_script_update
-            setup_python_env
-            check_config
-            create_service
+            run_with_spinner "Обновление Python-зависимостей" "source myenv/bin/activate && pip install aiogram==2.25.1 aiohttp==3.8.6 apscheduler==3.10.4 humanize==4.9.0 pytz==2023.3.post1 && deactivate"
             run_with_spinner "Перезапуск службы" "systemctl restart $SERVICE_NAME"
         else
             echo -e "${YELLOW}Обновление отменено${NC}"
@@ -113,32 +97,11 @@ check_github_updates() {
     cd ..
 }
 
-# Проверка и применение обновлений
-check_updates() {
-    [[ ! -d "$BOT_DIR/.git" ]] && error_exit "Репозиторий не найден. Пожалуйста, установите бот сначала."
-    check_github_updates --auto
-}
-
-# Параметры скрипта
-while [[ "$#" -gt 0 ]]; do
-    case $1 in
-        --quiet) ENABLE_LOGS=false ;;
-        --verbose) ENABLE_LOGS=true ;;
-        --check-update) check_updates; exit 0 ;;
-        *)
-            echo -e "${RED}Неизвестный параметр: $1${NC}"
-            echo "Использование: $0 [--quiet|--verbose|--check-update]"
-            exit 1
-            ;;
-    esac
-    shift
-done
-
 # Проверка обновления самого скрипта
 check_script_update() {
     local temp_script=$(mktemp)
-    if [[ -f "$BOT_DIR/install.sh" ]]; then
-        cp "$BOT_DIR/install.sh" "$temp_script"
+    if [[ -f "/root/amnezia-bot/install.sh" ]]; then
+        cp "/root/amnezia-bot/install.sh" "$temp_script"
         if ! cmp -s "$SCRIPT_PATH" "$temp_script"; then
             echo -e "${YELLOW}Обнаружено обновление скрипта install.sh${NC}"
             run_with_spinner "Обновление скрипта" "mv $temp_script $SCRIPT_PATH && chmod +x $SCRIPT_PATH"
@@ -149,301 +112,23 @@ check_script_update() {
         fi
     else
         echo -e "${YELLOW}Скрипт install.sh не найден в репозитории, копируем локальную версию${NC}"
-        cp "$SCRIPT_PATH" "$BOT_DIR/install.sh"
-        chmod +x "$BOT_DIR/install.sh"
+        cp "$SCRIPT_PATH" "/root/amnezia-bot/install.sh"
+        chmod +x "/root/amnezia-bot/install.sh"
     fi
 }
 
-# Получение версии Ubuntu
-get_ubuntu_version() {
-    command -v lsb_release &>/dev/null || error_exit "lsb_release не установлен. Установите пакет lsb-release."
-    UBUNTU_VERSION=$(lsb_release -rs)
-    UBUNTU_CODENAME=$(lsb_release -cs)
-    DISTRIB_ID=$(lsb_release -is)
-    [[ "$DISTRIB_ID" != "Ubuntu" ]] && error_exit "Скрипт поддерживает только Ubuntu. Обнаружена система: $DISTRIB_ID"
-}
-
-# Обновление и очистка системы
-update_and_clean_system() {
-    local max_attempts=30 attempt=1
-    while fuser /var/lib/dpkg/lock-frontend >/dev/null 2>&1; do
-        echo -e "${YELLOW}APT занят. Ждём ($attempt/$max_attempts)...${NC}"
-        ((attempt++))
-        [[ $attempt -gt $max_attempts ]] && error_exit "Таймаут ожидания dpkg lock"
-        sleep 2
-    done
-    run_with_spinner "Обновление системы" "apt-get update -qq && apt-get upgrade -y -qq"
-    run_with_spinner "Очистка системы" "apt-get autoclean -qq && apt-get autoremove --purge -y -qq"
-}
-
-# Проверка и установка Python 3.11
-check_python() {
-    if command -v python3.11 &>/dev/null; then
-        echo -e "${GREEN}Python 3.11 уже установлен${NC}"
-        return 0
+# Проверка и применение обновлений
+check_updates() {
+    if [[ ! -d "/root/amnezia-bot/.git" ]]; then
+        echo -e "${RED}Репозиторий не найден. Пожалуйста, установите бот сначала.${NC}"
+        exit 1
     fi
-    echo -e "${YELLOW}Устанавливаю Python 3.11...${NC}"
-    run_with_spinner "Установка Python 3.11" "apt-get install -y software-properties-common && add-apt-repository -y ppa:deadsnakes/ppa && apt-get update -qq && apt-get install -y python3.11 python3.11-venv python3.11-dev -qq"
-    command -v python3.11 &>/dev/null || error_exit "Не удалось установить Python 3.11"
-    echo -e "${GREEN}Python 3.11 успешно установлен${NC}"
-}
-
-# Установка зависимостей
-install_dependencies() {
-    local packages=("jq" "net-tools" "iptables" "resolvconf" "git" "curl" "sqlite3" "docker.io")
-    local to_install=()
-    for pkg in "${packages[@]}"; do
-        dpkg -s "$pkg" &>/dev/null || to_install+=("$pkg")
-    done
-    [[ ${#to_install[@]} -gt 0 ]] && run_with_spinner "Установка зависимостей" "apt-get install -y ${to_install[*]} -qq"
-    run_with_spinner "Включение Docker" "systemctl enable docker && systemctl start docker"
-}
-
-# Установка и конфигурация needrestart
-install_and_configure_needrestart() {
-    dpkg -s needrestart &>/dev/null || run_with_spinner "Установка needrestart" "apt-get install -y needrestart -qq"
-    sed -i 's/^#\?\(nrconf{restart} = \"\).*$/\1a\";/' /etc/needrestart/needrestart.conf
-    grep -q 'nrconf{restart} = "a";' /etc/needrestart/needrestart.conf || echo 'nrconf{restart} = "a";' >> /etc/needrestart/needrestart.conf
-}
-
-# Клонирование репозитория
-clone_repository() {
-    cd /root || error_exit "Не удалось перейти в /root"
-    if [[ -d "$BOT_DIR/.git" ]]; then
-        echo -e "${YELLOW}Репозиторий уже присутствует, пропускаем клонирование...${NC}"
-        return
-    fi
-    run_with_spinner "Клонирование репозитория" "git clone $REPO_URL -q $BOT_DIR"
-    cd "$BOT_DIR" || error_exit "Не удалось перейти в каталог $BOT_DIR"
-}
-
-# Настройка виртуального окружения
-setup_python_env() {
-    echo -e "${BLUE}Настройка Python окружения...${NC}"
-    cd "$BOT_DIR" || error_exit "Каталог $BOT_DIR не найден"
-    [[ -d "myenv" ]] && rm -rf myenv
-    run_with_spinner "Создание virtualenv" "python3.11 -m venv myenv"
-    source myenv/bin/activate || error_exit "Не удалось активировать виртуальное окружение"
-    run_with_spinner "Обновление pip" "pip install --upgrade pip -q"
-    if [[ -f "requirements.txt" ]]; then
-        run_with_spinner "Установка зависимостей из requirements.txt" "pip install -r requirements.txt -q"
-    else
-        echo -e "${YELLOW}Файл requirements.txt не найден. Устанавливаем минимальные зависимости...${NC}"
-        run_with_spinner "Установка минимальных зависимостей" "pip install aiogram==2.25.1 aiohttp==3.8.6 apscheduler==3.10.4 humanize==4.9.0 pytz==2023.3.post1 yoomoney==0.1.0 -q"
-    fi
-    deactivate
-    echo -e "${GREEN}Настройка Python окружения... Done!${NC}"
-}
-
-# Проверка наличия bot_manager.py и настройка структуры
-check_bot_files() {
-    cd "$AWG_DIR" || error_exit "Каталог $AWG_DIR не найден"
-    echo -e "${BLUE}Проверка наличия bot_manager.py...${NC}"
-    if [[ ! -f "bot_manager.py" ]]; then
-        error_exit "Файл bot_manager.py не найден в $AWG_DIR. Проверьте репозиторий: $REPO_URL"
-    fi
-    # Проверка наличия других необходимых файлов
-    for file in db.py awg-decode.py newclient.sh removeclient.sh; do
-        [[ -f "$file" ]] || echo -e "${YELLOW}Файл $file не найден в $AWG_DIR${NC}"
-    done
-    # Создание каталога files, если отсутствует
-    mkdir -p files || error_exit "Не удалось создать каталог files"
-    # Проверка и перемещение файлов из awg/awg, если они там
-    if [[ -d "awg" ]]; then
-        echo -e "${YELLOW}Обнаружен подкаталог awg. Перемещаем файлы в $AWG_DIR...${NC}"
-        mv awg/* . 2>/dev/null || echo -e "${YELLOW}Не удалось переместить некоторые файлы из awg${NC}"
-        rmdir awg 2>/dev/null || echo -e "${YELLOW}Не удалось удалить пустой подкаталог awg${NC}"
-    fi
-    echo -e "${GREEN}Проверка файлов завершена${NC}"
-}
-
-# Проверка и создание конфигурации
-check_config() {
-    cd "$AWG_DIR" || error_exit "Не удалось перейти в каталог $AWG_DIR"
-    echo -e "${BLUE}Проверка конфигурации...${NC}"
-    mkdir -p files || error_exit "Не удалось создать директорию files"
-    if [[ ! -f "$FILES_DIR/setting.ini" ]]; then
-        echo -e "${YELLOW}Файл setting.ini не найден. Запрашиваем конфигурацию...${NC}"
-        read -p "Введите токен Telegram бота: " bot_token
-        [[ -z "$bot_token" ]] && error_exit "Токен бота не может быть пустым"
-        read -p "Введите Telegram ID администраторов через запятую (например, 12345,67890): " admin_ids
-        [[ -z "$admin_ids" ]] && error_exit "ID администраторов не могут быть пустыми"
-        read -p "Введите endpoint VPN (например, vpn.example.com:51820): " endpoint
-        [[ -z "$endpoint" ]] && error_exit "Endpoint не может быть пустым"
-        # Валидация формата admin_ids
-        if ! echo "$admin_ids" | grep -qE '^[0-9]+(,[0-9]+)*$'; then
-            error_exit "Некорректный формат admin_ids. Используйте числа, разделённые запятыми (например, 12345,67890)"
-        fi
-        # Валидация формата endpoint
-        if ! echo "$endpoint" | grep -qE '^[^:]+:[0-9]+$'; then
-            error_exit "Некорректный формат endpoint. Используйте формат host:port (например, vpn.example.com:51820)"
-        fi
-        cat << EOF > "$FILES_DIR/setting.ini"
-[Settings]
-bot_token = $bot_token
-admin_ids = $admin_ids
-wg_config_file = $FILES_DIR/wg0.conf
-docker_container = amnezia-awg
-endpoint = $endpoint
-EOF
-        chmod 600 "$FILES_DIR/setting.ini"
-        echo -e "${GREEN}Конфигурация сохранена в $FILES_DIR/setting.ini${NC}"
-    fi
-    if docker ps -q -f name=amnezia-awg >/dev/null; then
-        echo -e "${GREEN}Найден Docker-контейнер: amnezia-awg${NC}"
-    else
-        echo -e "${YELLOW}ВНИМАНИЕ: Docker-контейнер amnezia-awg не найден. Попытка запуска...${NC}"
-        docker pull amneziawg/amnezia-wg &>/dev/null || echo -e "${YELLOW}Не удалось загрузить образ amneziawg/amnezia-wg${NC}"
-        docker run -d --name amnezia-awg \
-            --cap-add=NET_ADMIN \
-            --cap-add=SYS_MODULE \
-            -v "$FILES_DIR:/etc/wireguard" \
-            -p 51820:51820/udp \
-            --restart=always \
-            amneziawg/amnezia-wg &>/dev/null || echo -e "${YELLOW}Не удалось запустить контейнер amnezia-awg${NC}"
-    fi
-    if [[ ! -f "$FILES_DIR/wg0.conf" ]]; then
-        echo -e "${YELLOW}ВНИМАНИЕ: Файл wg0.conf не найден. Создайте $FILES_DIR/wg0.conf для WireGuard.${NC}"
-    fi
-    cd ..
-    echo -e "${GREEN}Проверка конфигурации... Done!${NC}"
-}
-
-# Права на скрипты
-set_permissions() {
-    cd "$AWG_DIR" || error_exit "Каталог $AWG_DIR не найден"
-    find . -type f -name "*.sh" -exec chmod +x {} \; || error_exit "Не удалось установить права на скрипты"
-    chmod 644 bot_manager.py db.py awg-decode.py 2>/dev/null || echo -e "${YELLOW}Не удалось установить права на Python-файлы${NC}"
-    chmod 600 files/* 2>/dev/null || echo -e "${YELLOW}Не удалось установить права на файлы в files${NC}"
-}
-
-# Инициализация бота для генерации config
-initialize_bot() {
-    cd "$AWG_DIR" || error_exit "Не удалось перейти в каталог $AWG_DIR"
-    if [[ ! -f "$FILES_DIR/setting.ini" ]]; then
-        if [[ -f "bot_manager.py" ]]; then
-            timeout 30s ../myenv/bin/python3.11 bot_manager.py < /dev/tty &
-            local PID=$!
-            local max_wait=30 wait_count=0
-            while [[ ! -f "$FILES_DIR/setting.ini" && $wait_count -lt $max_wait ]]; do
-                sleep 1
-                ((wait_count++))
-                kill -0 "$PID" 2>/dev/null || error_exit "Ошибка инициализации бота: bot_manager.py завершился с ошибкой"
-            done
-            [[ -f "$FILES_DIR/setting.ini" ]] || error_exit "Файл setting.ini не был создан после запуска bot_manager.py"
-            kill "$PID" && wait "$PID" 2>/dev/null
-        else
-            error_exit "Файл bot_manager.py не найден в $AWG_DIR"
-        fi
-    fi
-    chmod 600 "$FILES_DIR/setting.ini"
-    cd ..
-}
-
-# Создание systemd-сервиса
-create_service() {
-    cat << EOF > /tmp/service_file
-[Unit]
-Description=AmneziaVPN Docker Telegram Bot
-After=network.target
-
-[Service]
-User=root
-WorkingDirectory=$AWG_DIR
-ExecStart=$BOT_DIR/myenv/bin/python3.11 bot_manager.py
-Restart=always
-RestartSec=10
-
-[Install]
-WantedBy=multi-user.target
-EOF
-    run_with_spinner "Установка службы" "mv /tmp/service_file /etc/systemd/system/$SERVICE_NAME.service && chmod 644 /etc/systemd/system/$SERVICE_NAME.service"
-    run_with_spinner "Обновление systemd" "systemctl daemon-reload"
-    run_with_spinner "Запуск службы" "systemctl start $SERVICE_NAME"
-    run_with_spinner "Включение автозапуска" "systemctl enable $SERVICE_NAME"
-}
-
-# Удаление AmneziaWG
-remove_amneziawg() {
-    echo -ne "${YELLOW}Удалить AmneziaWG? (y/n): ${NC}"; read ans
-    [[ "$ans" =~ ^[Yy]$ ]] || return
-    systemctl is-active --quiet "$SERVICE_NAME" && run_with_spinner "Остановка" "systemctl stop $SERVICE_NAME"
-    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 $FILES_DIR"
-}
-
-# Меню управления службой
-service_control_menu() {
-    while true; do
-        echo -e "\n${BLUE}Управление службой${NC}"
-        systemctl status "$SERVICE_NAME" | grep -E "Active:|Loaded:"
-        echo -e "1) Остановить 2) Перезапустить 3) Переустановить 4) Удалить службу 5) Удалить AmneziaWG 6) Проверить обновления 7) Назад"
-        echo -ne "${BLUE}Выберите: ${NC}"; read act
-        case $act in
-            1) run_with_spinner "Остановка" "systemctl stop $SERVICE_NAME" ;;
-            2) run_with_spinner "Перезапуск" "systemctl restart $SERVICE_NAME" ;;
-            3) reinstall_bot ;;
-            4) run_with_spinner "Удаление службы" "systemctl disable $SERVICE_NAME && rm /etc/systemd/system/$SERVICE_NAME.service && systemctl daemon-reload" ;;
-            5) remove_amneziawg ;;
-            6) check_updates ;;
-            7) break ;;
-            *) echo -e "${RED}Неверный выбор${NC}" ;;
-        esac
-    done
-}
-
-# Переустановка бота
-reinstall_bot() {
-    echo -ne "${YELLOW}Переустановить бота? (y/n): ${NC}"; read ans
-    [[ "$ans" =~ ^[Yy]$ ]] || return
-    systemctl is-active --quiet "$SERVICE_NAME" && run_with_spinner "Остановка" "systemctl stop $SERVICE_NAME"
-    systemctl list-units --type=service --all | grep -q "$SERVICE_NAME.service" && run_with_spinner "Удаление службы" "systemctl disable $SERVICE_NAME && rm /etc/systemd/system/$SERVICE_NAME.service && systemctl daemon-reload"
-    run_with_spinner "Удаление файлов" "rm -rf $BOT_DIR"
-    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 -e "${RED}Неверный выбор${NC}" ;;
-        esac
-    done
-}
-
-# Полная установка бота
-install_bot() {
-    get_ubuntu_version
-    update_and_clean_system
-    check_python
-    install_dependencies
-    install_and_configure_needrestart
-    clone_repository
-    check_bot_files
-    setup_python_env
-    set_permissions
-    check_config
-    initialize_bot
-    create_service
-    cp "$SCRIPT_PATH" "$BOT_DIR/install.sh" && chmod +x "$BOT_DIR/install.sh"
-    echo -e "${GREEN}Установка завершена!${NC}"
+    check_github_updates "$1"
 }
 
 # Точка входа
-main() {
-    get_ubuntu_version
-    if systemctl list-units --type=service --all | grep -q "$SERVICE_NAME.service"; then
-        installed_menu
-    else
-        install_bot
-    fi
-}
-
-main
+if [[ "$1" == "--check-update" ]]; then
+    check_updates --auto
+else
+    check_updates
+fi