#!/usr/bin/env bash
set -e

echo "================================="
echo "   ArvanCloud Ubuntu Mirror Setup"
echo "================================="

source /etc/os-release

if [[ "$ID" != "ubuntu" ]]; then
    echo "This script only supports Ubuntu."
    exit 1
fi

CODENAME="$VERSION_CODENAME"

echo "[1/3] Backing up sources.list..."
cp /etc/apt/sources.list /etc/apt/sources.list.bak

echo "[2/3] Writing ArvanCloud mirror..."

cat > /etc/apt/sources.list <<EOF
deb http://mirror.arvancloud.ir/ubuntu ${CODENAME} universe
EOF

echo "[3/3] Updating package index..."
apt update

echo "Ubuntu mirror configured."