#!/bin/bash

APT_OPTIONS="-o Acquire::https::mirrors.martin98.com::Verify-Peer=false -o Acquire::https::mirrors.martin98.com::Verify-Host=false"
lsb_dist=$(cat /etc/*release | grep ^ID= | cut -d= -f2) # ubuntu or debian?
release=$(cat /etc/*release | grep VERSION_CODENAME | cut -d= -f2) #  ubuntu(jammy oracular) debian(bookworm)....
components=$([ "${lsb_dist=}" = "debian" ] && echo "main contrib non-free non-free-firmware" || ([ "${lsb_dist=}" = "ubuntu" ] && echo "main restricted universe multiverse"))

# 删除 DEB822 格式
if [ -f /etc/apt/sources.list.d/${lsb_dist}.sources ]; then
  rm /etc/apt/sources.list.d/${lsb_dist}.sources
fi
# 配置镜像源
cat <<EOF > /etc/apt/sources.list
deb https://mirrors.martin98.com/repository/$lsb_dist $release $components
deb https://mirrors.martin98.com/repository/$lsb_dist $release-updates $components
deb https://mirrors.martin98.com/repository/$lsb_dist $release-backports $components
deb https://mirrors.martin98.com/repository/$lsb_dist $release-security $components
EOF

for RELEASE in "$@"; do
    echo "deb https://mirrors.martin98.com/repository/$lsb_dist $RELEASE $components" >> /etc/apt/sources.list
done


# 关闭交互式弹窗
echo 'debconf debconf/frontend select Noninteractive' | debconf-set-selections

apt update $APT_OPTIONS && apt install $APT_OPTIONS -y ca-certificates curl && update-ca-certificates
apt update && apt upgrade -y

# NTP
# 启用阿里云 ntp
if pidof systemd >/dev/null 2>&1; then
    apt install chrony -y
    cat > /etc/chrony.conf <<EOF
server ntp4.aliyun.com minpoll 4 maxpoll 10 iburst
server ntp5.aliyun.com iburst
driftfile /var/lib/chrony/drift
allow 127.0.0.1
EOF
    timedatectl set-timezone Asia/Shanghai
    systemctl restart chrony && systemctl enable chrony
else
    echo "systemctl 不存在，跳过 chrony 的重启"
fi

# fix vim 粘贴
if ! grep -q "set pastetoggle=" ~/.vimrc; then
    echo "set pastetoggle=" >> ~/.vimrc
fi