254 lines
8.5 KiB
Bash
254 lines
8.5 KiB
Bash
#!/bin/sh
|
||
# 定义颜色输出函数
|
||
red() { echo -e "\033[31m\033[01m$1\033[0m"; }
|
||
green() { echo -e "\033[32m\033[01m$1\033[0m"; }
|
||
greeninfo() { echo -e "\033[32m\033[01m[INFO] $1\033[0m"; }
|
||
blueinfo() { echo -e "\033[32m\033[01m$1\033[0m"; }
|
||
yellow() { echo -e "\033[33m\033[01m$1\033[0m"; }
|
||
blue() { echo -e "\033[34m\033[01m$1\033[0m"; }
|
||
light_magenta() { echo -e "\033[95m\033[01m$1\033[0m"; }
|
||
light_yellow() { echo -e "\033[93m\033[01m$1\033[0m"; }
|
||
purple() { echo -e "\033[38;5;141m$1\033[0m"; }
|
||
cyan() { echo -e "\033[38;2;0;255;255m$1\033[0m"; }
|
||
|
||
install_depends_apps() {
|
||
cyan "正在安装必备工具...."
|
||
opkg update >/dev/null 2>&1
|
||
for pkg in lsblk fdisk; do
|
||
if opkg list-installed | grep -qw "$pkg"; then
|
||
cyan "$pkg 已安装。"
|
||
else
|
||
if opkg install "$pkg" >/dev/null 2>&1; then
|
||
green "$pkg 安装成功。"
|
||
else
|
||
red "$pkg 安装失败。"
|
||
exit 1
|
||
fi
|
||
fi
|
||
done
|
||
}
|
||
|
||
# 卸载USB设备
|
||
unmount_usb_device() {
|
||
for mount in $(mount | grep "$1" | awk '{print $3}'); do
|
||
yellow "正在尝试卸载U盘挂载点：$mount"
|
||
umount $mount || {
|
||
red "警告：无法卸载挂载点 $mount。可能有文件正在被访问或权限不足。"
|
||
exit 1
|
||
}
|
||
blueinfo "U盘挂载点 $mount 卸载成功。"
|
||
done
|
||
}
|
||
|
||
create_and_format_partitions() {
|
||
local device=$1
|
||
# 使用fdisk -l获取设备的总容量（以字节为单位）并转换为GB
|
||
local total_bytes=$(fdisk -l $device | grep "Disk $device:" | awk '{print $5}')
|
||
local total_gb=$(echo "$total_bytes" | awk '{print int($1/(1024*1024*1024))}')
|
||
|
||
if [ -n "$CUSTOM_OPKG_SIZE" ]; then
|
||
part1_gb=$CUSTOM_OPKG_SIZE
|
||
yellow "U盘总容量约为 $total_gb GB,您设置的自定义软件包大小为 ${part1_gb}GB。"
|
||
else
|
||
# 计算10%的大小，以GB为单位
|
||
part1_gb=$((total_gb / 10))
|
||
yellow "U盘总容量约为 $total_gb GB,第一分区大小设置为U盘容量的10% 即 ${part1_gb}GB。"
|
||
fi
|
||
green "计划将第一分区分配给软件包 其大小为 ${part1_gb}GB"
|
||
cyan "没错～你没有看错,让我们任性的告别 容 量 焦 虑！"
|
||
# 创建分区并分配空间
|
||
{
|
||
echo g # 创建一个新的空DOS分区表
|
||
echo n # 添加一个新分区
|
||
echo p # 主分区
|
||
echo 1 # 分区号1
|
||
echo # 第一个可用扇区（默认）
|
||
echo +${part1_gb}G # 为第一个分区分配计算出的GB数
|
||
echo n # 添加第二个新分区
|
||
echo p # 主分区
|
||
echo 2 # 分区号2
|
||
echo # 第一个可用扇区（默认，自动）
|
||
echo # 最后一个扇区（默认，使用剩余空间）
|
||
echo w # 写入并退出
|
||
} | fdisk $device >/dev/null 2>&1
|
||
|
||
# 给系统一点时间来识别新分区
|
||
sleep 5
|
||
|
||
# 格式化第一个分区为EXT4
|
||
local new_partition1="${device}1"
|
||
cyan "正在将 $new_partition1 格式化为EXT4文件系统..."
|
||
mkfs.ext4 -F $new_partition1 >/dev/null 2>&1
|
||
cyan "$new_partition1 已成功格式化为EXT4文件系统。"
|
||
green "第2分区 ${device}2 暂不格式化,未来您可分配给docker使用"
|
||
}
|
||
|
||
# 换区到U盘
|
||
change_overlay_usb() {
|
||
install_depends_apps
|
||
blueinfo "现在开始查找USB设备分区 请稍后......"
|
||
local USB_PARTITION=$(lsblk -dn -o NAME,RM,TYPE | awk '$2=="1" && $3=="disk" {print "/dev/"$1; exit}')
|
||
if [ -z "$USB_PARTITION" ]; then
|
||
red "未找到USB磁盘。"
|
||
exit 1
|
||
fi
|
||
yellow "找到USB磁盘 $USB_PARTITION"
|
||
# 清零磁盘开始部分以清除分区表和文件系统签名
|
||
dd if=/dev/zero of=$USB_PARTITION bs=1M count=10
|
||
sync
|
||
# 卸载所有与该磁盘相关的挂载点
|
||
unmount_usb_device "$USB_PARTITION"
|
||
red "正在将U盘${USB_PARTITION}分为2个区 ..."
|
||
create_and_format_partitions "$USB_PARTITION"
|
||
|
||
# U盘分区的挂载点
|
||
MOUNT_POINT="/mnt/usb_overlay"
|
||
# 临时目录用于复制数据
|
||
TMP_DIR="/tmp/overlay_backup"
|
||
# 创建挂载点目录
|
||
mkdir -p $MOUNT_POINT
|
||
# 挂载U盘分区
|
||
cyan "重新挂载第一分区 ${USB_PARTITION}1 到 $MOUNT_POINT"
|
||
mount ${USB_PARTITION}1 $MOUNT_POINT >/dev/null 2>&1
|
||
# 创建临时目录用于备份overlay数据
|
||
mkdir -p $TMP_DIR
|
||
# 复制当前overlay到临时目录
|
||
cp -a /overlay/. $TMP_DIR
|
||
# 将临时目录的数据复制到U盘
|
||
blueinfo "正在拷贝 当前系统文件到U盘"
|
||
cp -a $TMP_DIR/. $MOUNT_POINT
|
||
# 更新fstab配置，以便在启动时自动挂载U盘为overlay
|
||
blueinfo "正在更新启动时的配置文件"
|
||
uci set fstab.overlay=mount
|
||
uci set fstab.overlay.uuid="$(blkid -o value -s UUID ${USB_PARTITION}1)"
|
||
uci set fstab.overlay.target="/overlay"
|
||
uci commit fstab
|
||
# 清理临时目录
|
||
rm -rf $TMP_DIR
|
||
cyan "overlay更换分区完成 重启验证是否成功."
|
||
red "是否立即重启？(y/n)"
|
||
read -r answer
|
||
if [ "$answer" = "y" ] || [ -z "$answer" ]; then
|
||
red "正在重启..."
|
||
reboot
|
||
else
|
||
yellow "您选择了不重启"
|
||
fi
|
||
}
|
||
|
||
check_overlay_size() {
|
||
# 使用df命令获取/overlay分区的总大小（以1K块为单位）
|
||
OVERLAY_SIZE=$(df /overlay | awk '/\/overlay/{print $2}')
|
||
# 将1GB转换为1K块单位，即1GB = 1*1024*1024 1K块
|
||
ONE_GB_IN_1K_BLOCKS=$((1024 * 1024))
|
||
# 比较/overlay分区的大小是否大于1GB
|
||
if [ "$OVERLAY_SIZE" -gt "$ONE_GB_IN_1K_BLOCKS" ]; then
|
||
yellow "检测到您已经换区到U盘啦,可以继续"
|
||
else
|
||
echo "您还没有换区到U盘,请先执行选项1."
|
||
exit 1
|
||
fi
|
||
}
|
||
|
||
# 安装 Docker 和 dockerd
|
||
install_docker() {
|
||
check_overlay_size
|
||
green "正在更新 OPKG 软件包..."
|
||
opkg update >/dev/null 2>&1
|
||
cyan "正在安装 Docker 及相关服务...请耐心等待一会...大约需要1-2分钟\n"
|
||
opkg install luci-app-dockerman >/dev/null 2>&1
|
||
opkg install luci-i18n-dockerman-zh-cn >/dev/null 2>&1
|
||
opkg install dockerd --force-depends >/dev/null 2>&1
|
||
cyan "Docker 运行环境部署完成 重启后生效\n"
|
||
red "正在重启..."
|
||
reboot
|
||
}
|
||
|
||
# 重新绑定
|
||
rebind_usb_overlay() {
|
||
cyan "正在重新绑定U盘设备...."
|
||
if opkg list-installed | grep -qw "lsblk"; then
|
||
echo
|
||
else
|
||
opkg update >/dev/null 2>&1
|
||
if opkg install "lsblk" >/dev/null 2>&1; then
|
||
echo
|
||
else
|
||
red "$pkg 安装失败。"
|
||
exit 1
|
||
fi
|
||
fi
|
||
local USB_DEVICE=$(lsblk -dn -o NAME,RM,TYPE | awk '$2=="1" && $3=="disk" {print "/dev/"$1; exit}')
|
||
if [ -z "$USB_DEVICE" ]; then
|
||
red "未找到USB磁盘。"
|
||
exit 1
|
||
fi
|
||
uci set fstab.overlay=mount
|
||
uci set fstab.overlay.uuid="$(blkid -o value -s UUID ${USB_DEVICE}1)"
|
||
uci set fstab.overlay.target="/overlay"
|
||
uci commit fstab
|
||
green "重新绑定成功！ 重启后生效"
|
||
red "正在重启..."
|
||
reboot
|
||
}
|
||
|
||
#自定义软件包的大小
|
||
#默认为U盘容量的10%
|
||
custom_package_size() {
|
||
while :; do
|
||
echo "请输入想分配的软件包的大小(数字,单位:GB):"
|
||
read size
|
||
# 检查输入是否为数字
|
||
if [[ $size =~ ^[0-9]+$ ]]; then
|
||
CUSTOM_OPKG_SIZE=$size
|
||
yellow "已设置软件包大小为:$CUSTOM_OPKG_SIZE GB"
|
||
green "接下来,您可以执行第一项啦"
|
||
break # 跳出循环
|
||
else
|
||
red "错误: 请输入一个有效的数字。"
|
||
fi
|
||
done
|
||
}
|
||
|
||
while true; do
|
||
clear
|
||
echo "***********************************************************************"
|
||
green " MT-3000 软件包更换分区助手 "
|
||
echo "**********************************************************************"
|
||
echo
|
||
cyan " 1. MT-3000 一键更换overlay分区到U盘"
|
||
cyan " 2. MT-3000 安装Docker"
|
||
cyan " 3. 自定义设置软件包大小(GB)"
|
||
light_yellow " 4. 重新绑定U盘"
|
||
echo
|
||
echo " Q. 退出本程序"
|
||
echo
|
||
read -p "请选择一个选项: " choice
|
||
echo
|
||
|
||
case $choice in
|
||
|
||
1)
|
||
change_overlay_usb
|
||
;;
|
||
2)
|
||
install_docker
|
||
;;
|
||
3)
|
||
custom_package_size
|
||
;;
|
||
4)
|
||
rebind_usb_overlay
|
||
;;
|
||
|
||
q | Q)
|
||
echo "退出"
|
||
exit 0
|
||
;;
|
||
*)
|
||
echo "无效选项，请重新选择。"
|
||
;;
|
||
esac
|
||
read -p "按 Enter 键继续..."
|
||
done
|