Hosts文件更新

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

更新日志

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

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

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

#!/bin/sh

set -e

# 定义hosts文件的URL
HOSTS_URL1="https://hosts.gitcdn.top/hosts.txt"
HOSTS_URL2="https://hosts.gitcdn.top/hosts.txt"
HOSTS_URL3="https://raw.hellogithub.com/hosts"

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

# 下载hosts文件并合并去重并格式化
{
/usr/bin/wget --no-hsts -qO- "$HOSTS_URL1"
/usr/bin/wget --no-hsts -qO- "$HOSTS_URL2"
/usr/bin/wget --no-hsts -qO- "$HOSTS_URL3"
} | 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")

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

# 检查下载是否成功并且行数大于0
if [ -s "$TMPHOST_FILE" ] && [ "$LINE_COUNT" -gt 0 ]; then
    # 复制处理后的hosts文件到/etc/hosts，并重启dnsmasq服务
    cp "$TMPHOST_FILE" "$ETCHOST_FILE" && \
        /etc/init.d/dnsmasq restart && \
        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