diff --git a/deploy.sh b/deploy.sh index b15cf53..a6e935e 100644 --- a/deploy.sh +++ b/deploy.sh @@ -14,11 +14,15 @@ SERVICE_NAME="wfl-data.service" TARGET_ENV="/mnt/wfl/.env" show_help() { - echo "Использование: $0 -r <релиз> [-h]" + echo "Использование: $0 -t <домашняя команда> -r <релиз> [-l <лига>]" + echo " -t Домашняя команда" echo " -r Релиз (тег или ветка в git)" + echo " -l Лига (опционально)" echo " -h Показать эту справку" echo "" - echo "Пример: $0 -r Barabanov_TEST" + echo "Пример: $0 -t cska -r main" + echo "Пример: $0 -t zenit -r Barabanov_TEST -l vtb" + echo "Пример: $0 -t avtodor -r main" echo "" exit 0 } @@ -190,10 +194,11 @@ except ImportError as e: log_info "Установленные пакеты:" pip list --format=columns } - # Функция создания systemd сервиса create_systemd_service() { log_info "Создание systemd сервиса" + local team="$1" + local league="$2" # Останавливаем и отключаем старый сервис если он есть if systemctl is-active --quiet "$SERVICE_NAME"; then @@ -206,6 +211,30 @@ create_systemd_service() { systemctl disable "$SERVICE_NAME" fi + + log_info "Создание systemd сервиса для команды: $team" + if [ -n "$league" ]; then + log_info "Лига: $league" + fi + + # Останавливаем и отключаем старый сервис если он есть + if systemctl is-active --quiet wfl-data.service; then + log_info "Остановка wfl-data.service..." + systemctl stop wfl-data.service + fi + + if systemctl is-enabled --quiet wfl-data.service; then + log_info "Отключение wfl-data.service..." + systemctl disable wfl-data.service + fi + + # Формируем команду для data сервиса + local data_command="python3 $TARGET_DIR/get_data.py --team \"$team\"" + if [ -n "$league" ]; then + data_command="$data_command --league \"$league\"" + fi + + log_info "Создание файла сервиса: $SERVICE_NAME" # Формируем команду для data сервиса local data_service_file="/etc/systemd/system/$SERVICE_NAME" @@ -213,7 +242,7 @@ create_systemd_service() { cat > "$data_service_file" << EOF [Unit] Description=WFL Data Service -Documentation=https://git.tvstart.ru/ychernenko/WHL +Documentation=https://git.tvstart.ru/ychernenko/WFL After=network.target Wants=network.target @@ -223,8 +252,7 @@ User=root WorkingDirectory=$TARGET_DIR Environment=PATH=$TARGET_DIR/.venv/bin EnvironmentFile=$TARGET_ENV -# ExecStart=$TARGET_DIR/.venv/bin/uvicorn app:app --host 0.0.0.0 --port 8000 -ExecStart=/root/WFL/.venv/bin/uvicorn app:app --host 0.0.0.0 --port 8000 +ExecStart=$data_command # Лимиты ресурсов MemoryMax=1G @@ -261,33 +289,58 @@ EOF fi } -# Функция проверки файлов -check_required_files() { - local target_dir="$TARGET_DIR" - cd "$target_dir" || exit 1 +# Функция определения IP и команды (исправленная) +detect_team() { + local team_arg="$1" - local missing_files=() - local required_files=("app.py") + # Определение IP адреса + if [[ "$(uname -s)" == "Linux" ]]; then + ip_address=$(ip route get 1 2>/dev/null | awk '{print $7; exit}' || + ip a 2>/dev/null | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' | + grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1' | head -n1) + else + log_error "Скрипт работает только на Linux" + exit 1 + fi + + log_info "Определен IP адрес: $ip_address" + + # База данных хостов в виде пар IP-команда + declare -A hosts=( + ["10.10.35.21"]="cska" + ["10.10.35.22"]="Lokomotiv Kuban" + ["10.10.35.23"]="uralmash" + ["10.10.35.24"]="betcity parma" + ["10.10.35.25"]="avtodor" + ["10.10.35.26"]="zenit" + ["10.10.35.27"]="samara" + ["10.10.35.28"]="mba-mai" + ["10.10.35.29"]="Pari Nizhny Novgorod" + ["10.10.35.30"]="unics" + ["10.10.35.31"]="enisey" + ) + + # Определение команды по IP + detected_team="${hosts[$ip_address]}" - for file in "${required_files[@]}"; do - if [ ! -f "$file" ]; then - missing_files+=("$file") - fi - done - - if [ ${#missing_files[@]} -ne 0 ]; then - log_error "Отсутствуют необходимые файлы: ${missing_files[*]}" - log_error "Содержимое директории $target_dir:" - ls -la + # Определение финальной команды (приоритет у аргумента, затем у автоопределения) + if [[ -n "$team_arg" ]]; then + final_team="$team_arg" + log_info "Используется команда из аргумента: $final_team" + elif [[ -n "$detected_team" ]]; then + final_team="$detected_team" + log_info "Используется автоопределенная команда: $final_team" + else + log_error "Не удалось определить команду. Укажите явно через -t" + echo "Доступные команды:" >&2 + for ip in "${!hosts[@]}"; do + echo " $ip: ${hosts[$ip]}" >&2 + done exit 1 fi - log_info "Все необходимые файлы присутствуют" - - # Дополнительная проверка на исполняемость Python файла - if ! python3 -c "import ast; ast.parse(open('app.py').read())" 2>/dev/null; then - log_warn "Файл get_data.py содержит синтаксические ошибки Python" - fi + # Возвращаем только чистую строку с названием команды (без цветовых кодов) + echo "$final_team" } # Функция настройки firewall @@ -296,20 +349,15 @@ setup_firewall() { # Проверяем наличие ufw if command -v ufw &> /dev/null && systemctl is-active --quiet ufw; then - if ufw allow 8000/tcp; then - log_info "Порт 8000 открыт в ufw" - else - log_warn "Не удалось открыть порт 8000 в ufw" - fi + ufw allow 8000/tcp + log_info "Порт 8000 открыт в ufw" fi # Для firewalld (CentOS/RHEL) if command -v firewall-cmd &> /dev/null && systemctl is-active --quiet firewalld; then - if firewall-cmd --permanent --add-port=8000/tcp && firewall-cmd --reload; then - log_info "Порт 8000 открыт в firewalld" - else - log_warn "Не удалось открыть порт 8000 в firewalld" - fi + firewall-cmd --permanent --add-port=8000/tcp + firewall-cmd --reload + log_info "Порт 8000 открыт в firewalld" fi } @@ -318,9 +366,9 @@ check_port() { local port=8000 local occupied=false - if command -v netstat &> /dev/null && netstat -tuln 2>/dev/null | grep -q ":$port "; then + if command -v netstat &> /dev/null && netstat -tuln 2>/dev/null | grep ":$port " > /dev/null; then occupied=true - elif command -v ss &> /dev/null && ss -tuln 2>/dev/null | grep -q ":$port "; then + elif command -v ss &> /dev/null && ss -tuln 2>/dev/null | grep ":$port " > /dev/null; then occupied=true fi @@ -387,12 +435,7 @@ check_system() { fi # Проверка свободного места - local available_space - available_space=$(df "$TARGET_DIR" --output=avail 2>/dev/null | tail -1) - if [ -z "$available_space" ]; then - available_space=$(df / --output=avail | tail -1) - fi - + local available_space=$(df /root --output=avail | tail -1) if [ "$available_space" -lt 1048576 ]; then # Меньше 1GB log_warn "Мало свободного места: $((available_space / 1024)) MB" else @@ -400,14 +443,8 @@ check_system() { fi # Проверка памяти - if [ -f /proc/meminfo ]; then - local total_mem - total_mem=$(grep MemTotal /proc/meminfo | awk '{print $2}') - log_info "Оперативная память: $((total_mem / 1024)) MB" - fi - - # Проверка архитектуры - log_info "Архитектура: $(uname -m)" + local total_mem=$(grep MemTotal /proc/meminfo | awk '{print $2}') + log_info "Оперативная память: $((total_mem / 1024)) MB" } # Функция очистки при прерывании @@ -423,23 +460,33 @@ cleanup() { # Основная функция main() { + local team="" # переменная для команды local release="main" # значение по умолчанию + local league="" # переменная для лиги # Обработка аргументов командной строки - while getopts "r:h" opt; do + while getopts "t:r:l:h" opt; do case $opt in + t) team="$OPTARG" ;; r) release="$OPTARG" ;; + l) league="$OPTARG" ;; h) show_help ;; *) log_error "Неверный аргумент"; exit 1 ;; esac done - + + # Проверка обязательных аргументов + if [ -z "$team" ] && [ -z "$release" ]; then + log_error "Необходимо указать команду (-t) и релиз (-r)" + show_help + fi + log_info "Начало установки WFL Data Service..." - if [ -n "$release" ]; then - log_info "Релиз: $release" + log_info "Команда: $team, Релиз: $release" + if [ -n "$league" ]; then + log_info "Лига: $league" else - log_info "Релиз: не указан (будет использовано значение по умолчанию: main)" - release="main" + log_info "Лига: не указана (будет использовано значение по умолчанию)" fi # Проверка прав root @@ -448,9 +495,6 @@ main() { exit 1 fi - # Установка обработчика прерывания - trap 'log_error "Прервано пользователем"; cleanup; exit 1' INT TERM - # Проверка системы check_system @@ -463,12 +507,19 @@ main() { # Загрузка кода download_code "$release" - # Проверка файлов - check_required_files - # Настройка виртуального окружения setup_venv + # Определение команды + final_team=$(detect_team "$team") + log_info "Финальная команда: '$final_team'" + + # Проверка что команда не пустая и не содержит спецсимволов + if [[ -z "$final_team" || "$final_team" =~ [^a-zA-Z0-9[:space:]-] ]]; then + log_error "Некорректное название команды: '$final_team'" + exit 1 + fi + # Настройка firewall setup_firewall @@ -476,11 +527,11 @@ main() { check_port # Создание systemd сервиса - create_systemd_service + create_systemd_service "$final_team" "$league" log_info "Настройка завершена!" - # Запуск сервиса + # Запуск сервиса if manage_services "start"; then log_info "Сервис успешно запущен" else @@ -492,7 +543,6 @@ main() { check_service_status # Получаем IP для вывода - local ip_address ip_address=$(ip route get 1 2>/dev/null | awk '{print $7; exit}' || ip a 2>/dev/null | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' | grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1' | head -n1) @@ -501,6 +551,10 @@ main() { log_info "Установка завершена успешно!" log_info "Релиз: $release" log_info "Рабочая директория: $TARGET_DIR" + log_info "Команда: $final_team" + if [ -n "$league" ]; then + log_info "Лига: $league" + fi log_info "" log_info "Для просмотра логов:" log_info " journalctl -u $SERVICE_NAME -f"