Этот коммит содержится в:
stevefoxru
2025-04-22 14:35:51 +03:00
коммит произвёл GitHub
родитель 82845e7978
Коммит 35e8795e92
+142 -270
Просмотреть файл
@@ -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"
shift
local description="$1"; shift
local cmd="$@"
if [ "$ENABLE_LOGS" = true ]; then
echo -e "${BLUE}${description}...${NC}"
eval "$cmd"
local status=$?
if [ $status -eq 0 ]; then
echo -e "${GREEN}${description}... Done!${NC}\n"
else
echo -e "${RED}${description}... Failed!${NC}"
echo -e "${RED}Ошибка при выполнении команды: $cmd${NC}\n"
exit 1
fi
local stat=$?
if [ $stat -eq 0 ]; then echo -e "${GREEN}${description}... Done!${NC}\n"; else echo -e "${RED}${description}... Failed!${NC}\n"; exit 1; fi
else
local stdout_temp=$(mktemp)
local stderr_temp=$(mktemp)
eval "$cmd" >"$stdout_temp" 2>"$stderr_temp" &
local pid=$!
local spinner='|/-\'
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
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
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
rm -f "$out" "$err"
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}"
return 0
echo -e "\n${GREEN}Python 3.11 уже установлен${NC}"; return 0
fi
echo -e "\n${RED}Python 3.11 не установлен${NC}"
read -p "Установить Python 3.11? (y/n): " install_python
if [[ "$install_python" =~ ^[Yy]$ ]]; then
if [[ "$UBUNTU_VERSION" == "24.04" ]]; then
local max_attempts=30
local attempt=1
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
echo -e "\n${YELLOW}Устанавливаю Python 3.11...${NC}"
# Ожидание dpkg lock на Ubuntu 24.04
if [[ "$UBUNTU_VERSION" == "24.04" ]]; then
local max=30 cnt=1
while fuser /var/lib/dpkg/lock-frontend >/dev/null 2>&1; do
echo -e "${YELLOW}Ожидание dpkg lock ($cnt/$max)${NC}"; ((cnt++))
[ $cnt -gt $max ] && { echo -e "${RED}Таймаут ожидания dpkg lock${NC}"; exit 1; }
sleep 10
done
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"
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
run_with_spinner "Установка needrestart" "sudo apt-get install -y needrestart -qq"
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 -a /etc/needrestart/needrestart.conf >/dev/null
}
# Клонирование репозитория
clone_repository() {
if [[ -d "amnezia-bot" ]]; then
echo -e "\n${YELLOW}Репозиторий существует${NC}"
cd amnezia-bot || { echo -e "\n${RED}Ошибка перехода в директорию${NC}"; exit 1; }
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; }
if [[ -d "amnezia-bot" ]]; then cd amnezia-bot; return; fi
run_with_spinner "Клонирование репозитория" "git clone https://github.com/stevefoxru/amnezia-bot.git -q"
cd amnezia-bot
}
# Настройка виртуального окружения
setup_venv() {
if [[ -d "myenv" ]]; then
echo -e "\n${YELLOW}Виртуальное окружение существует${NC}"
return 0
fi
run_with_spinner "Настройка виртуального окружения" "python3.11 -m venv myenv && source myenv/bin/activate && pip install --upgrade pip && pip install -r $(pwd)/requirements.txt && deactivate"
if [[ -d "myenv" ]]; then return; fi
run_with_spinner "Создание virtualenv" "python3.11 -m venv myenv"
source myenv/bin/activate
run_with_spinner "Установка Python-зависимостей" "pip install --upgrade pip && pip install -r requirements.txt"
deactivate
}
# Права на скрипты
set_permissions() {
find . -type f -name "*.sh" -exec chmod +x {} \; 2>chmod_error.log
local status=$?
[[ $status -ne 0 ]] && { cat chmod_error.log; rm -f chmod_error.log; exit 1; }
rm -f chmod_error.log
find . -type f -name "*.sh" -exec chmod +x {} \; || exit 1
}
# Инициализация бота для генерации 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 "$BOT_PID"
wait "$BOT_PID" 2>/dev/null
kill "$PID" && wait "$PID" 2>/dev/null
cd ..
}
# Создание systemd-сервиса
create_service() {
# Проверяем, что директории существуют
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
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 "Обновление systemd" "sudo systemctl daemon-reload -qq"
run_with_spinner "Запуск службы" "sudo systemctl start $SERVICE_NAME -qq"
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}"
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 "Запуск службы" "sudo systemctl start $SERVICE_NAME -q"
run_with_spinner "Включение автозапуска" "sudo systemctl enable $SERVICE_NAME -q"
}
# Проверка и применение обновлений
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