Rewrite logic, use env variables, and add comments to the following code to make it more readable and maintainable.
This commit is contained in:
parent
9a449517ed
commit
31d07d4484
4 changed files with 259 additions and 35 deletions
|
|
@ -1,40 +1,39 @@
|
|||
#!/bin/bash
|
||||
set -e
|
||||
set -euo pipefail
|
||||
|
||||
# Пути к конфигурационным файлам
|
||||
TARGET_DIR="/opt/appdata/pihole/etc/dnsmasq.d"
|
||||
IPSET_CONF="91-ipset-bbrkn.conf"
|
||||
RESOLVE_CONF="92-resolve-bbrkn.conf"
|
||||
# ==============================
|
||||
# Конфигурация через переменные
|
||||
# ==============================
|
||||
TARGET_DIR="${TARGET_DIR:-/opt/appdata/pihole/etc/dnsmasq.d}"
|
||||
IPSET_CONF="${IPSET_CONF:-91-ipset-bbrkn.conf}"
|
||||
RESOLVE_CONF="${RESOLVE_CONF:-92-resolve-bbrkn.conf}"
|
||||
|
||||
DOCKER_CONTAINER="${DOCKER_CONTAINER:-pihole}"
|
||||
|
||||
echo "Deploying configuration files to $TARGET_DIR"
|
||||
|
||||
# Создаем резервные копии существующих файлов
|
||||
# Бэкап существующих файлов
|
||||
if [ -f "$TARGET_DIR/$IPSET_CONF" ]; then
|
||||
cp "$TARGET_DIR/$IPSET_CONF" "$TARGET_DIR/$IPSET_CONF.backup.$(date +%Y%m%d-%H%M%S)"
|
||||
fi
|
||||
|
||||
if [ -f "$TARGET_DIR/$RESOLVE_CONF" ]; then
|
||||
cp "$TARGET_DIR/$RESOLVE_CONF" "$TARGET_DIR/$RESOLVE_CONF.backup.$(date +%Y%m%d-%H%M%S)"
|
||||
fi
|
||||
|
||||
# Копируем новые файлы
|
||||
# Копирование новых файлов
|
||||
cp "/tmp/$IPSET_CONF" "$TARGET_DIR/$IPSET_CONF"
|
||||
cp "/tmp/$RESOLVE_CONF" "$TARGET_DIR/$RESOLVE_CONF"
|
||||
|
||||
# Перезапуск контейнера
|
||||
echo "Restarting $DOCKER_CONTAINER container..."
|
||||
docker restart "$DOCKER_CONTAINER"
|
||||
|
||||
# Перезапускаем dnsmasq через Pi-hole контейнер
|
||||
echo "Restarting Pi-hole container..."
|
||||
docker restart pihole
|
||||
|
||||
# Проверяем успешность перезапуска
|
||||
sleep 5
|
||||
if ! docker ps | grep -q pihole; then
|
||||
echo "Error: Pi-hole container failed to start"
|
||||
if ! docker ps | grep -q "$DOCKER_CONTAINER"; then
|
||||
echo "Error: $DOCKER_CONTAINER container failed to start"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo "Deployment completed successfully"
|
||||
|
||||
# Уведомляем о количестве обработанных доменов
|
||||
DOMAIN_COUNT=$(grep -c '^ipset=' "$TARGET_DIR/$IPSET_CONF" || echo "0")
|
||||
echo "Applied configuration for $DOMAIN_COUNT domains"
|
||||
|
|
|
|||
|
|
@ -1,24 +1,31 @@
|
|||
#!/bin/bash
|
||||
set -euo pipefail
|
||||
|
||||
INPUT_FILE="domains.txt"
|
||||
IPSET_CONF="/tmp/91-ipset-bbrkn.conf"
|
||||
RESOLVE_CONF="/tmp/92-resolve-bbrkn.conf"
|
||||
API_URL="http://10.100.1.2:3000/domains?domain="
|
||||
# ==============================
|
||||
# Конфигурация через переменные
|
||||
# ==============================
|
||||
INPUT_FILE="${INPUT_FILE:-domains.txt}"
|
||||
|
||||
IPSET_CONF="${IPSET_CONF:-/tmp/91-ipset-bbrkn.conf}"
|
||||
RESOLVE_CONF="${RESOLVE_CONF:-/tmp/92-resolve-bbrkn.conf}"
|
||||
|
||||
API_URL="${API_URL:-http://10.100.1.2:3000/domains?domain=}"
|
||||
DNS_SERVER="${DNS_SERVER:-8.8.8.8}"
|
||||
|
||||
DRY_RUN=false
|
||||
if [[ "${1:-}" == "--dry-run" ]]; then
|
||||
DRY_RUN=true
|
||||
fi
|
||||
|
||||
# Очистка файлов только если не dry-run
|
||||
if ! $DRY_RUN; then
|
||||
: > "$IPSET_CONF"
|
||||
: > "$RESOLVE_CONF"
|
||||
fi
|
||||
|
||||
declare -A DOM_ROLE # роли для исходных доменов: site/service
|
||||
declare -A EXPANDED # все уникальные домены (ключи = домены)
|
||||
declare -A SOURCES # источник: base или related
|
||||
declare -A DOM_ROLE
|
||||
declare -A EXPANDED
|
||||
declare -A SOURCES
|
||||
|
||||
normalize_domain() {
|
||||
local raw="$1"
|
||||
|
|
@ -34,7 +41,7 @@ normalize_domain() {
|
|||
fi
|
||||
}
|
||||
|
||||
# Обработка доменов из входного списка
|
||||
# Обработка доменов
|
||||
while IFS= read -r line || [ -n "$line" ]; do
|
||||
dom="$(normalize_domain "$line" || true)"
|
||||
[ -z "$dom" ] && continue
|
||||
|
|
@ -60,25 +67,23 @@ while IFS= read -r line || [ -n "$line" ]; do
|
|||
fi
|
||||
done < "$INPUT_FILE"
|
||||
|
||||
# Сортировка доменов
|
||||
# Сортировка
|
||||
mapfile -t ALL_DOMAINS < <(printf "%s\n" "${!EXPANDED[@]}" | sort -u)
|
||||
|
||||
# Генерация конфигов
|
||||
if ! $DRY_RUN; then
|
||||
for d in "${ALL_DOMAINS[@]}"; do
|
||||
echo "ipset=/$d/bbrkn" >> "$IPSET_CONF"
|
||||
echo "server=/$d/8.8.8.8" >> "$RESOLVE_CONF"
|
||||
echo "server=/$d/$DNS_SERVER" >> "$RESOLVE_CONF"
|
||||
done
|
||||
fi
|
||||
|
||||
# Подсчёты
|
||||
count_in=$(wc -l < "$INPUT_FILE")
|
||||
count_total=${#ALL_DOMAINS[@]}
|
||||
count_base=$(printf "%s\n" "${!SOURCES[@]}" | grep -c '.*' || true)
|
||||
count_related=$(printf "%s\n" "${!SOURCES[@]}" | grep -c '.*' || true)
|
||||
count_related=$(( count_total - count_in ))
|
||||
|
||||
# Отладочный вывод
|
||||
# Отладочный отчёт
|
||||
echo
|
||||
echo "===== DEBUG REPORT ====="
|
||||
echo "Original domains file: $count_in entries"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue