#!/usr/bin/env bash

# ---------------------------------------------------------------------------------

# 控制台颜色

BLACK="\033[1;30m"

RED="\033[1;31m"

GREEN="\033[1;32m"

YELLOW="\033[1;33m"

BLUE="\033[1;34m"

PURPLE="\033[1;35m"

CYAN="\033[1;36m"

RESET="$(tput sgr0)"

# ---------------------------------------------------------------------------------

printf "${BLUE}\n"

cat << EOF

###################################################################################

# 本脚本用于替换 yum repo，使用国内 yum 仓库，加速下载

# 要求：仅适用于 Linux CentOS 发行版本，并且环境必须已支持 yum 、lsb_release 命令

# @author: Zhang Peng

###################################################################################

EOF

printf "${RESET}\n"

printf "\n${GREEN}>>>>>>>>> 替换 yum repo 源开始${RESET}\n"

# 备份

cp /etc/yum.repos.d/CentOS-Base.repo /etc/yum.repos.d/CentOS-Base.repo.bak

# 执行 lsb_release 命令，获取系统发行版本

# version=`lsb_release -r | awk '{print substr($2,1,1)}'` # 很多机器没有 lsb_release 命令

version=`cat /etc/redhat-release | awk '{print substr($4,1,1)}'`

# 根据发型版本选择相应 yum 镜像

if [[ ${version} == 5 ]]; then

# Cento5 已废弃，只能使用 http://vault.CentOS.org/ 替换，但由于是国外镜像，速度较慢

wget -N https://gitee.com/turnon/linux-tutorial/raw/master/codes/linux/sys/yum/Centos-5.repo -O /etc/yum.repos.d/CentOS-Base.repo

# 根据实际发型版本情况替换

detailVersion=`lsb_release -r | awk '{print substr($2,1,3)}'`

sed -i 's/$releasever/'"${detailVersion}"'/g' /etc/yum.repos.d/CentOS-Base.repo

# 不替换下面的开关，可能会出现错误：Could not open/read repomd.xml

sed -i 's/enabled=1/enabled=0/g' /etc/yum.repos.d/CentOS-Media.repo

elif [[ ${version} == 6 ]]; then

wget -N https://gitee.com/turnon/linux-tutorial/raw/master/codes/linux/sys/yum/Centos-6.repo -O /etc/yum.repos.d/CentOS-Base.repo

elif [[ ${version} == 7 ]]; then

wget -N https://gitee.com/turnon/linux-tutorial/raw/master/codes/linux/sys/yum/Centos-7.repo -O /etc/yum.repos.d/CentOS-Base.repo

else

printf "\n${RED}版本不支持，替换 yum repo 失败${RESET}\n"

fi

# 更新缓存

yum clean all

yum makecache

printf "\n${GREEN}<<<<<<<< 替换 yum repo 源结束${RESET}\n"