Rewrite logic, use env variables, and add comments to the following code to make it more readable and maintainable.

This commit is contained in:
Kirill Kodanev 2025-09-11 19:49:28 +03:00
parent 9a449517ed
commit 31d07d4484
4 changed files with 259 additions and 35 deletions

View file

@ -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"