#!/bin/bash
# https://github.com/Hyy2001X/AutoBuild-Actions
# AutoBuild Module by Hyy2001
# AutoBuild_Tools for Openwrt

Version=V1.2.1-BETA

AutoBuild_Tools() {
	while :
	do
		clear
		echo -e "AutoBuild 固件工具箱 ${Version}\n"
		echo "1.内部空间扩展"
		echo "2.Samba 共享"
		echo "3.挂载硬盘"
		echo "4.查看挂载点"
		echo "5.安装软件包"
		echo -e "\nq.退出\n"
		read -p "请从上方选择一个操作:" Choose
		case $Choose in
		q)
			rm -rf /tmp/AutoExpand /tmp/AutoSamba
			clear
			exit
		;;
		1)
			
			AutoExpand_UI
		;;
		2)
			Samba_UI
		;;
		3)
			block mount
		;;
		4)
			clear && mount
			Enter
		;;
		5)
			Install_UI	
		;;
		esac
	done
}

AutoExpand_UI() {
	uci set fstab.@global[0].auto_mount='0'
	uci set fstab.@global[0].auto_swap='0'
	uci commit fstab
	clear
	echo -e "Newifi-D2 一键扩展内部空间\n"
	USB_Check_Core
	if [[ ! -z "${Check_Disk}" ]];then
		for ((i=1;i<=${Disk_Number};i++));
		do
			Disk_info=$(sed -n ${i}p ${Disk_Processed_List})
			List_Disk ${Disk_info}
		done
		echo -e "\nq.返回"
		echo "r.重新载入列表"
	else
		echo "未检测到外接硬盘!" && sleep 2
		return
	fi
	echo ""
	read -p "请输入要操作的硬盘编号[1-${Disk_Number}]:" Choose
	echo ""
	case ${Choose} in
	q)
		return
	;;
	r)
		AutoExpand_UI
	;;
	*)
		if [ ${Choose} -gt 0 ] > /dev/null 2>&1 && [ ${Choose} -le ${Disk_Number} ] > /dev/null 2>&1;then
			
			which mkfs.ext4 > /dev/null 2>&1
			if [ "$?" -eq 0 ];then
				AutoExpand_Core
			else
				echo "请先安装 [e2fsprogs] !" && sleep 3
			fi			
		else
			echo "选择错误,请输入正确的选项!"
			sleep 2 && AutoExpand_UI
			exit
		fi
	;;
	esac
}

USB_Check_Core() {
	block mount
	rm -rf ${AutoExpend_Tmp}/*
	echo "$(block info)" > ${Block_Info}
	Check_Disk="$(cat  ${Block_Info} | awk  -F ':' '/sd/{print $1}')"
	if [ ! -z "${Check_Disk}" ];then
		echo "${Check_Disk}" > ${Disk_List}
		Disk_Number=$(sed -n '$=' ${Disk_List})
		for Disk_Name in $(cat ${Disk_List})
		do
			Mounted_on="$(df -h | grep "${Disk_Name}" | awk '{print $6}')"
			Disk_Available="$(df -m | grep "${Disk_Name}" | awk '{print $4}')"
			Disk_Format="$(cat  ${Block_Info} | grep "${Disk_Name}" | egrep -o 'TYPE="[a-z].+' | awk -F '["]' '/TYPE/{print $2}')"
			touch ${Disk_Processed_List}
			if [ ! -z "$Mounted_on" ];then
				echo "${Disk_Name} ${Mounted_on} ${Disk_Format} ${Disk_Available}MB" >> ${Disk_Processed_List}
			else
				echo "${Disk_Name} ${Disk_Format}" >> ${Disk_Processed_List}
			fi
		done
	fi
}

AutoExpand_Core() {
	Choosed_Disk="$(sed -n ${Choose}p ${Disk_Processed_List} | awk '{print $1}')"
	echo "警告: 本次操作将把硬盘: '${Choosed_Disk}' 格式化为 'ext4' 格式,请提前做好数据备份工作!"
	read -p "是否继续本次操作?[Y/n]:" Choose
	if [ ${Choose} == Y ] || [ ${Choose} == y ];then
		sleep 3 && echo ""
	else
		sleep 3
		echo "用户已取消操作."
		break
	fi
	if [[ "$(mount)" =~ "${Choosed_Disk}" ]] > /dev/null 2>&1 ;then
		Choosed_Disk_Mounted="$(mount | grep "${Choosed_Disk}" | awk '{print $3}')"
		echo "取消挂载: '${Choosed_Disk}' on '${Choosed_Disk_Mounted}' ..."
		umount -l ${Choosed_Disk_Mounted} > /dev/null 2>&1
		if [ "$(mount)" =~ "${Choosed_Disk_Mounted}" ] > /dev/null 2>&1 ;then
			echo "取消挂载: '${Choosed_Disk_Mounted}' 失败 !"
			exit
		fi
	fi
	echo "正在格式化硬盘: '${Choosed_Disk}' 为 'ext4' 格式 ..."
	mkfs.ext4 -F ${Choosed_Disk} > /dev/null 2>&1
	echo "格式化完成! 挂载硬盘: '${Choosed_Disk}' 到 ' /tmp/extroot' ..."
	mkdir -p /tmp/introot && mkdir -p /tmp/extroot
	mount --bind / /tmp/introot
	mount ${Choosed_Disk} /tmp/extroot
	echo "正在备份系统文件到 硬盘: '${Choosed_Disk}',请耐心等待 ..."
	tar -C /tmp/introot -cf - . | tar -C /tmp/extroot -xf -
	echo "取消挂载: '/tmp/introot' '/tmp/extroot' ..."
	umount /tmp/introot && umount /tmp/extroot
	[ ! -d /mnt/bak ] && mkdir -p /mnt/bak
	mount ${Choosed_Disk} /mnt/bak
	echo "同步系统文件改动 ..."
	sync
	echo "写入 '分区表' 到 '/etc/config/fstab' ..."
	block detect > /etc/config/fstab
	sed -i "s?/mnt/bak?/?g" /etc/config/fstab
	for ((i=0;i<=${Disk_Number};i++));
	do
		uci set fstab.@mount[${i}].enabled='1' > /dev/null 2>&1
	done
	uci commit fstab
	umount -l /mnt/bak
	echo -e "操作结束,外接硬盘: '${Choosed_Disk}' 已挂载到 '/'.\n"
	read -p "挂载完成后需要重启生效,是否立即重启路由器?[Y/n]:" Choose
	if [ ${Choose} == Y ] || [ ${Choose} == y ];then
		sleep 3 && echo -e "\n正在重启路由器,请耐心等待 ..."
		sync
		reboot
	else
		echo "用户已取消重启操作."
		sleep 3
		break
	fi
}

List_Disk() {
	if [[ ${2} == "/" ]];then
		_Type="[不可用]"
	else
		_Type="[可用]"
	fi
	if [[ ! -z ${3} ]];then
		echo "${i}.${_Type}硬盘: '${1}' 挂载点: '${2}' 格式: '${3}' 可用空间: ${4}"
	else
		echo "${i}.硬盘: '${1}' 格式: '${2}' 未挂载"
	fi
}

Samba_UI() {
	while :
	do
		clear
		echo -e "Samba 工具箱\n"
		echo "1.删除所有 Samba 挂载点"
		echo "2.自动共享所有已连接的硬盘"
		echo -e "\nq.返回\n"
		read -p "请从上方选择一个操作:" Choose
		case $Choose in
		1)
			Remove_Samba_Settings
		;;
		2)
			Mount_Samba_Devices
		;;
		q)
			break
		;;
		esac
	done
}

Remove_Samba_Settings() {
	while :
	do
		Samba_config="$(grep "sambashare" /etc/config/samba | wc -l)"
		[ "${Samba_config}" -eq 0 ] && break
		uci delete samba.@sambashare[0]
		uci commit
	done
	echo -e "\n已删除所有共享挂载点!"
	sleep 2
}

Mount_Samba_Devices() {
	echo "$(cat /proc/mounts  | awk  -F ':' '/sd/{print $1}')" > ${Disk_List}
	Disk_Number=$(sed -n '$=' ${Disk_List})
	echo ""
	for ((i=1;i<=${Disk_Number};i++));
	do
		Disk_Name=$(sed -n ${i}p ${Disk_List} | awk '{print $1}')
		Disk_Mounted_on=$(sed -n ${i}p ${Disk_List} | awk '{print $2}')
		Samba_Name=${Disk_Mounted_on#*/mnt/}
		uci show 2>&1 | grep "sambashare" > ${UCI_Show_List}
		if [[ ! "$(cat ${UCI_Show_List})" =~ "${Disk_Name}" ]] > /dev/null 2>&1 ;then
			echo "共享硬盘: '${Disk_Name}' on '${Disk_Mounted_on}' 到 '${Samba_Name}' ..."
			echo -e "\nconfig sambashare" >> ${Samba_Config_File}
			echo -e "\toption auto '1'" >> ${Samba_Config_File}
			echo -e "\toption name '${Samba_Name}'" >> ${Samba_Config_File}
			echo -e "\toption device '${Disk_Name}'" >> ${Samba_Config_File}
			echo -e "\toption path '${Disk_Mounted_on}'" >> ${Samba_Config_File}
			echo -e "\toption read_only 'no'" >> ${Samba_Config_File}
			echo -e "\toption guest_ok 'yes'" >> ${Samba_Config_File}
			echo -e "\toption create_mask '0666'" >> ${Samba_Config_File}
			echo -e "\toption dir_mask '0777'" >> ${Samba_Config_File}
		else
			echo "硬盘: '${Disk_Name}' 已设置共享."
		fi
	done
	/etc/init.d/samba restart
	sleep 2
}

Install_UI() {
while :
	do
		clear
		echo -e "安装软件包\n"
		echo "1.更新软件包列表"
		Install_UI_Mod 2 block-mount
		Install_UI_Mod 3 e2fsprogs
		echo "x.自定义软件包名"
		echo -e "\nq.返回\n"
		read -p "请从上方选择一个操作:" Choose
		echo ""
		case $Choose in
		q)
			break
		;;
		x)
			echo "常用的附加参数:"
			echo "--force-depends		在安装、删除软件包时无视失败的依赖"
			echo "--force-downgrade	允许降级安装软件包"
			echo -e "--force-reinstall	重新安装软件包\n"
			read -p "请输入你想安装的软件包名:" PKG_NAME
			Install_opkg_mod $PKG_NAME
		;;
		1)
			opkg update
		;;
		2)
			Install_opkg_mod block-mount	
		;;
		3)
			Install_opkg_mod e2fsprogs
		;;
		esac
	done
}

Install_UI_Mod() {
	if [[ "$(opkg list | awk '{print $1}')" =~ "${2}" ]] > /dev/null 2>&1 ;then
		echo "${1}.安装 [${2}] [已安装]"
	else
		echo "${1}.未安装 [${2}] [已安装]"
	fi
}

Install_opkg_mod() {
	opkg install ${*}
	if [[ "$(opkg list | awk '{print $1}')" =~ "${1}" ]] > /dev/null 2>&1 ;then
		echo -e "\n${1} 安装成功!"
	else
		echo -e "\n${1} 安装失败!"
	fi
	sleep 2
}

Enter() {
	echo "" && read -p "按下[回车]键以继续..." Key
}

AutoExpend_Tmp="/tmp/AutoExpand"
Disk_List="${AutoExpend_Tmp}/Disk_List"
Block_Info="${AutoExpend_Tmp}/Block_Info"
Disk_Processed_List="${AutoExpend_Tmp}/Disk_Processed_List"
[ ! -d ${AutoExpend_Tmp} ] && mkdir -p ${AutoExpend_Tmp}
Samba_Config_File="/etc/config/samba"
Samba_Tmp="/tmp/AutoSamba"
Disk_List="${Samba_Tmp}/Disk_List"
UCI_Show_List="${Samba_Tmp}/UCI_List"
[ ! -d ${Samba_Tmp} ] && mkdir -p ${Samba_Tmp}
which block > /dev/null 2>&1
if [ "$?" -eq 0 ];then
	AutoBuild_Tools
else
	echo -e "\nAutoBuild_Tools 不适用于此固件,请先安装 [block-mount] !"
fi