update_onedrive_hosts.sh

#!/bin/bash
# Function: 更新 Onedrive Host - 可以选择多个源，也许有助于改善中国大陆的 Onedrive 网络封锁。
# Function: Update Onedrive Host - Multiple sources can be selected, which may help improve the Onedrive network blockade in mainland China.
# Author: 10935336
# Creation date：2024-01-06
# Modified date：2024-02-29

#### Variable ####
# 默认更新源
UPDATE_SOURCE=1

# 更新源 1  https://learningman.top/archives/245
SOURCE1_URL=https://onedrive-hosts.learningman.top/

# 更新源 2  
#SOURCE2_URL=

# 取消标志
CANCEL_FLAG=0


# 解析命令行参数
while [[ "$#" -gt 0 ]]; do
    case $1 in
        -c|--cancel) CANCEL_FLAG=1;;
        -u|--update-source) UPDATE_SOURCE="$2"; shift;;
        *) echo "Unknown parameter passed: $1"; exit 1;;
    esac
    shift
done


# 取消操作
if [ $CANCEL_FLAG -eq 1 ]; then
    echo "取消更新并删除现有 Onenote Hosts"
    sed -i '/####### Onenote Hosts Start #######/,/####### Onenote Hosts End #######/d' /etc/hosts
    exit 0
fi


# 根据选择的更新源设置URL
case $UPDATE_SOURCE in
    1) HOSTS_URL=$SOURCE1_URL;;
    2) HOSTS_URL=$SOURCE2_URL;;
    *) echo "无效的更新源: $UPDATE_SOURCE"; exit 1;;
esac


#### Execution ####
# 获取 hosts 内容
wget -q -O onedrive-hosts $HOSTS_URL

# 检查是否获取成功
if [ $? -ne 0 ]; then
    echo "获取 onedrive-hosts 失败"
    exit 1
else
    echo "获取 onedrive-hosts 成功，开始更新"
fi

# 检查hosts文件是否存在
if [ ! -f /etc/hosts ]; then
    echo "未找到 hosts 文件"
    exit 1
fi

# 获取现有hosts内容暂存
cat /etc/hosts > hosts-temp

# 记录文件内容
content=$(cat onedrive-hosts) 

# 如果为空,还原内容 
[ -s onedrive-hosts ] || echo "$content" > onedrive-hosts


# 删除现有hosts中已有的 Onenote Hosts
sed -i '/####### Onenote Hosts Start #######/,/####### Onenote Hosts End #######/d' hosts-temp

# 将新的hosts内容添加到现有hosts暂存中
cat onedrive-hosts >> hosts-temp

# 将更新后的hosts内容写入/etc/hosts
cat hosts-temp > /etc/hosts

# 删除临时文件
rm -f onedrive-hosts hosts-temp

echo "更新完成"