365 строки
14 KiB
Bash
365 строки
14 KiB
Bash
#!/bin/bash
|
|
|
|
SERVICE_NAME="awg_bot"
|
|
|
|
GREEN=$'\033[0;32m'
|
|
YELLOW=$'\033[1;33m'
|
|
RED=$'\033[0;31m'
|
|
BLUE=$'\033[0;34m'
|
|
NC=$'\033[0m'
|
|
|
|
ENABLE_LOGS=true
|
|
|
|
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 ;;
|
|
--verbose) ENABLE_LOGS=true ;;
|
|
*)
|
|
echo -e "${RED}Неизвестный параметр: $1${NC}"
|
|
echo "Использование: $0 [--quiet|--verbose]"
|
|
exit 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
get_ubuntu_version() {
|
|
if command -v lsb_release &>/dev/null; then
|
|
UBUNTU_VERSION=$(lsb_release -rs)
|
|
UBUNTU_CODENAME=$(lsb_release -cs)
|
|
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}"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
run_with_spinner() {
|
|
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
|
|
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
|
|
fi
|
|
}
|
|
|
|
check_updates() {
|
|
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
|
|
}
|
|
|
|
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=amneziawg | xargs -r docker rm -f"
|
|
run_with_spinner "Удаление Docker образов AmneziaWG" "docker images -q amneziawg/amneziawg | 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 python_bot_amnezia"
|
|
|
|
# Запуск полной установки заново
|
|
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}3${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() {
|
|
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"
|
|
}
|
|
|
|
check_python() {
|
|
if command -v python3.11 &>/dev/null; then
|
|
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
|
|
fi
|
|
}
|
|
|
|
install_dependencies() {
|
|
run_with_spinner "Установка зависимостей" "sudo apt-get install jq net-tools iptables resolvconf git -y -qq"
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
clone_repository() {
|
|
if [[ -d "awg-docker-bot" ]]; then
|
|
echo -e "\n${YELLOW}Репозиторий существует${NC}"
|
|
cd awg-docker-bot || { echo -e "\n${RED}Ошибка перехода в директорию${NC}"; exit 1; }
|
|
return 0
|
|
fi
|
|
|
|
run_with_spinner "Клонирование репозитория" "git clone https://github.com/JB-SelfCompany/awg-docker-bot.git >/dev/null 2>&1"
|
|
cd awg-docker-bot || { echo -e "\n${RED}Ошибка перехода в директорию${NC}"; exit 1; }
|
|
}
|
|
|
|
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"
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
initialize_bot() {
|
|
cd awg || { echo -e "\n${RED}Ошибка перехода в директорию${NC}"; exit 1; }
|
|
|
|
../myenv/bin/python3.11 bot_manager.py < /dev/tty &
|
|
local BOT_PID=$!
|
|
|
|
while [ ! -f "files/setting.ini" ]; do
|
|
sleep 2
|
|
kill -0 "$BOT_PID" 2>/dev/null || { echo -e "\n${RED}Бот завершил работу до инициализации${NC}"; exit 1; }
|
|
done
|
|
|
|
kill "$BOT_PID"
|
|
wait "$BOT_PID" 2>/dev/null
|
|
cd ..
|
|
}
|
|
|
|
create_service() {
|
|
cat > /tmp/service_file << EOF
|
|
[Unit]
|
|
Description=AmneziaVPN Docker Telegram Bot
|
|
After=network.target
|
|
|
|
[Service]
|
|
User=$USER
|
|
WorkingDirectory=$(pwd)/awg
|
|
ExecStart=$(pwd)/myenv/bin/python3.11 bot_manager.py
|
|
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}"
|
|
}
|
|
|
|
install_bot() {
|
|
get_ubuntu_version
|
|
update_and_clean_system
|
|
check_python
|
|
install_dependencies
|
|
install_and_configure_needrestart
|
|
clone_repository
|
|
setup_venv
|
|
set_permissions
|
|
initialize_bot
|
|
create_service
|
|
}
|
|
|
|
main() {
|
|
systemctl list-units --type=service --all | grep -q "$SERVICE_NAME.service" && installed_menu || { install_bot; ( sleep 1; rm -- "$SCRIPT_PATH" ) & exit 0; }
|
|
}
|
|
|
|
main
|
© 2026 StackWorks