自动更新Hosts文件

刚开始想网上找一个OpenWRT系统上更新Hosts文件，但是发现要么就是简单粗暴直接下载要么只能下载单个文件再启动服务。
毕竟不是运维出身，于是花了一晚上功夫边学边写的，目前运行良好。

更新日志

• 20250101
  修复了文件获取失败导致无法继续执行的问题。

• 20240716
  修复了log输出时时间戳不准确的问题。

• 20240624
  修改wget方式，合并去重以及格式化。

• 20240601
  增加了下载是否成功并且行数大于0验证。

#!/bin/sh
set -e

# 定义 hosts 文件的 URL
HOSTS_URL1="输入你想去广告的Hosts地址"
HOSTS_URL2="输入你想去广告的Hosts地址"
HOSTS_URL3="输入你想去广告的Hosts地址"

# 定义各类文件路径
UPDTLOG_FILE="/tmp/update_hosts.log"
TMPHOST_FILE=$(mktemp)
ETCHOST_FILE="/etc/hosts"

# 定义错误处理和日志记录
log_message() {
    echo "$(date "+%Y-%m-%d %H:%M:%S") $1" >> "$UPDTLOG_FILE"
}

# 定义下载函数
download_hosts() {
    local url=$1
    local output
    output=$(/usr/bin/wget --timeout=10 --no-hsts -qO- "$url" 2>/dev/null)
    if [ $? -eq 0 ]; then
        echo "$output"
    else
        log_message "Failed to download from $url"
        return 1
    fi
}

# 下载 hosts 文件并合并去重并格式化
{
    download_hosts "$HOSTS_URL1" || true
    download_hosts "$HOSTS_URL2" || true
    download_hosts "$HOSTS_URL3" || true
} | sed -e '/^[[:space:]]*$/d' -e '/^[0-9]/!d' | sort -uk2 | awk '!seen[$0]++' | awk '{printf "%-20s %s\n", $1, $2}' > "$TMPHOST_FILE"

# 获取文件行数
LINE_COUNT=$(wc -l < "$TMPHOST_FILE")


# 检查下载是否成功并且行数大于 0
if [ -s "$TMPHOST_FILE" ] && [ "$LINE_COUNT" -gt 0 ]; then
    # 复制处理后的 hosts 文件到 /etc/hosts，并重启 dnsmasq 服务
    cp "$TMPHOST_FILE" "$ETCHOST_FILE" && \
        if command -v systemctl >/dev/null 2>&1; then
            systemctl restart dnsmasq
        else
            /etc/init.d/dnsmasq restart
        fi && \
        log_message "Update Hosts File Successful[Records:$LINE_COUNT]"
else
    log_message "Update Hosts File Failed[Records:$LINE_COUNT]"
    exit 1
fi

# 删除临时文件
trap 'rm -f "$TMPHOST_FILE"' EXIT