From 19fa4d23910beeb7ea6d5249dd4be536b7442ec9 Mon Sep 17 00:00:00 2001
From: hzy <hzyitc@outlook.com>
Date: Sun, 9 Oct 2022 13:47:41 +0800
Subject: [PATCH] Split `/lib/functions/cli/utils-cli.sh` from `/compile.sh`

---
 compile.sh                     | 161 +------------------------------
 lib/functions/cli/utils-cli.sh | 168 +++++++++++++++++++++++++++++++++
 2 files changed, 172 insertions(+), 157 deletions(-)
 create mode 100644 lib/functions/cli/utils-cli.sh

diff --git a/compile.sh b/compile.sh
index 7a3f847df..1559f9dbb 100755
--- a/compile.sh
+++ b/compile.sh
@@ -48,86 +48,9 @@ else
 
 fi
 
-#  Add the variables needed at the beginning of the path
-check_args() {
-
-	for p in "$@"; do
-
-		case "${p%=*}" in
-			LIB_TAG)
-				# Take a variable if the branch exists locally
-				if [ "${p#*=}" == "$(git branch |
-					gawk -v b="${p#*=}" '{if ( $NF == b ) {print $NF}}')" ]; then
-					echo -e "[\e[0;35m warn \x1B[0m] Setting $p"
-					eval "$p"
-				else
-					echo -e "[\e[0;35m warn \x1B[0m] Skip $p setting as LIB_TAG=\"\""
-					eval LIB_TAG=""
-				fi
-				;;
-		esac
-
-	done
-
-}
-
 check_args "$@"
 
-update_src() {
-
-	cd "${SRC}" || exit
-	if [[ ! -f "${SRC}"/.ignore_changes ]]; then
-		echo -e "[\e[0;32m o.k. \x1B[0m] This script will try to update"
-
-		CHANGED_FILES=$(git diff --name-only)
-		if [[ -n "${CHANGED_FILES}" ]]; then
-			echo -e "[\e[0;35m warn \x1B[0m] Can't update since you made changes to: \e[0;32m\n${CHANGED_FILES}\x1B[0m"
-			while true; do
-				echo -e "Press \e[0;33m<Ctrl-C>\x1B[0m or \e[0;33mexit\x1B[0m to abort compilation" \
-					", \e[0;33m<Enter>\x1B[0m to ignore and continue, \e[0;33mdiff\x1B[0m to display changes"
-				read -r
-				if [[ "${REPLY}" == "diff" ]]; then
-					git diff
-				elif [[ "${REPLY}" == "exit" ]]; then
-					exit 1
-				elif [[ "${REPLY}" == "" ]]; then
-					break
-				else
-					echo "Unknown command!"
-				fi
-			done
-		elif [[ $(git branch | grep "*" | awk '{print $2}') != "${LIB_TAG}" && -n "${LIB_TAG}" ]]; then
-			git checkout "${LIB_TAG:-master}"
-			git pull
-		fi
-	fi
-
-}
-
-TMPFILE=$(mktemp)
-chmod 644 "${TMPFILE}"
-{
-
-	echo SRC="$SRC"
-	echo LIB_TAG="$LIB_TAG"
-	declare -f update_src
-	echo "update_src"
-
-} > "$TMPFILE"
-
-#do not update/checkout git with root privileges to messup files onwership.
-#due to in docker/VM, we can't su to a normal user, so do not update/checkout git.
-if [[ $(systemd-detect-virt) == 'none' ]]; then
-
-	if [[ "${EUID}" == "0" ]]; then
-		su "$(stat --format=%U "${SRC}"/.git)" -c "bash ${TMPFILE}"
-	else
-		bash "${TMPFILE}"
-	fi
-
-fi
-
-rm "${TMPFILE}"
+do_update_src
 
 if [[ "${EUID}" == "0" ]] || [[ "${1}" == "vagrant" ]]; then
 	:
@@ -154,12 +77,7 @@ else
 
 fi
 
-# Check for Vagrant
-if [[ "${1}" == vagrant && -z "$(command -v vagrant)" ]]; then
-	display_alert "Vagrant not installed." "Installing"
-	sudo apt-get update
-	sudo apt-get install -y vagrant virtualbox
-fi
+handle_vagrant "$@"
 
 # Purge Armbian Docker images
 if [[ "${1}" == dockerpurge && -f /etc/debian_version ]]; then
@@ -178,80 +96,9 @@ if [[ "${1}" == docker-shell ]]; then
 	set -- "docker" "$@"
 fi
 
-# Install Docker if not there but wanted. We cover only Debian based distro install. On other distros, manual Docker install is needed
-if [[ "${1}" == docker && -f /etc/debian_version && -z "$(command -v docker)" ]]; then
+handle_docker "$@"
 
-	DOCKER_BINARY="docker-ce"
-
-	# add exception for Ubuntu Focal until Docker provides dedicated binary
-	codename=$(cat /etc/os-release | grep VERSION_CODENAME | cut -d"=" -f2)
-	codeid=$(cat /etc/os-release | grep ^NAME | cut -d"=" -f2 | awk '{print tolower($0)}' | tr -d '"' | awk '{print $1}')
-	[[ "${codename}" == "debbie" ]] && codename="buster" && codeid="debian"
-	[[ "${codename}" == "ulyana" || "${codename}" == "jammy" ]] && codename="focal" && codeid="ubuntu"
-
-	# different binaries for some. TBD. Need to check for all others
-	[[ "${codename}" =~ focal|hirsute ]] && DOCKER_BINARY="docker containerd docker.io"
-
-	display_alert "Docker not installed." "Installing" "Info"
-	sudo bash -c "echo \"deb [arch=$(dpkg --print-architecture)] https://download.docker.com/linux/${codeid} ${codename} stable\" > /etc/apt/sources.list.d/docker.list"
-
-	sudo bash -c "curl -fsSL \"https://download.docker.com/linux/${codeid}/gpg\" | apt-key add -qq - > /dev/null 2>&1 "
-	export DEBIAN_FRONTEND=noninteractive
-	sudo apt-get update
-	sudo apt-get install -y -qq --no-install-recommends ${DOCKER_BINARY}
-	display_alert "Add yourself to docker group to avoid root privileges" "" "wrn"
-	"${SRC}/compile.sh" "$@"
-	exit $?
-
-fi
-
-# Create userpatches directory if not exists
-mkdir -p "${SRC}"/userpatches
-
-# Create example configs if none found in userpatches
-if ! ls "${SRC}"/userpatches/{config-default.conf,config-docker.conf,config-vagrant.conf} 1> /dev/null 2>&1; then
-
-	# Migrate old configs
-	if ls "${SRC}"/*.conf 1> /dev/null 2>&1; then
-		display_alert "Migrate config files to userpatches directory" "all *.conf" "info"
-		cp "${SRC}"/*.conf "${SRC}"/userpatches || exit 1
-		rm "${SRC}"/*.conf
-		[[ ! -L "${SRC}"/userpatches/config-example.conf ]] && ln -fs config-example.conf "${SRC}"/userpatches/config-default.conf || exit 1
-	fi
-
-	display_alert "Create example config file using template" "config-default.conf" "info"
-
-	# Create example config
-	if [[ ! -f "${SRC}"/userpatches/config-example.conf ]]; then
-		cp "${SRC}"/config/templates/config-example.conf "${SRC}"/userpatches/config-example.conf || exit 1
-	fi
-
-	# Link default config to example config
-	if [[ ! -f "${SRC}"/userpatches/config-default.conf ]]; then
-		ln -fs config-example.conf "${SRC}"/userpatches/config-default.conf || exit 1
-	fi
-
-	# Create Docker config
-	if [[ ! -f "${SRC}"/userpatches/config-docker.conf ]]; then
-		cp "${SRC}"/config/templates/config-docker.conf "${SRC}"/userpatches/config-docker.conf || exit 1
-	fi
-
-	# Create Docker file
-	if [[ ! -f "${SRC}"/userpatches/Dockerfile ]]; then
-		cp "${SRC}"/config/templates/Dockerfile "${SRC}"/userpatches/Dockerfile || exit 1
-	fi
-
-	# Create Vagrant config
-	if [[ ! -f "${SRC}"/userpatches/config-vagrant.conf ]]; then
-		cp "${SRC}"/config/templates/config-vagrant.conf "${SRC}"/userpatches/config-vagrant.conf || exit 1
-	fi
-
-	# Create Vagrant file
-	if [[ ! -f "${SRC}"/userpatches/Vagrantfile ]]; then
-		cp "${SRC}"/config/templates/Vagrantfile "${SRC}"/userpatches/Vagrantfile || exit 1
-	fi
-
-fi
+prepare_userpatches
 
 if [[ -z "${CONFIG}" && -n "$1" && -f "${SRC}/userpatches/config-$1.conf" ]]; then
 	CONFIG="userpatches/config-$1.conf"
diff --git a/lib/functions/cli/utils-cli.sh b/lib/functions/cli/utils-cli.sh
new file mode 100644
index 000000000..27ca25608
--- /dev/null
+++ b/lib/functions/cli/utils-cli.sh
@@ -0,0 +1,168 @@
+#  Add the variables needed at the beginning of the path
+check_args() {
+
+	for p in "$@"; do
+
+		case "${p%=*}" in
+			LIB_TAG)
+				# Take a variable if the branch exists locally
+				if [ "${p#*=}" == "$(git branch |
+					gawk -v b="${p#*=}" '{if ( $NF == b ) {print $NF}}')" ]; then
+					echo -e "[\e[0;35m warn \x1B[0m] Setting $p"
+					eval "$p"
+				else
+					echo -e "[\e[0;35m warn \x1B[0m] Skip $p setting as LIB_TAG=\"\""
+					eval LIB_TAG=""
+				fi
+				;;
+		esac
+
+	done
+
+}
+
+update_src() {
+
+	cd "${SRC}" || exit
+	if [[ ! -f "${SRC}"/.ignore_changes ]]; then
+		echo -e "[\e[0;32m o.k. \x1B[0m] This script will try to update"
+
+		CHANGED_FILES=$(git diff --name-only)
+		if [[ -n "${CHANGED_FILES}" ]]; then
+			echo -e "[\e[0;35m warn \x1B[0m] Can't update since you made changes to: \e[0;32m\n${CHANGED_FILES}\x1B[0m"
+			while true; do
+				echo -e "Press \e[0;33m<Ctrl-C>\x1B[0m or \e[0;33mexit\x1B[0m to abort compilation" \
+					", \e[0;33m<Enter>\x1B[0m to ignore and continue, \e[0;33mdiff\x1B[0m to display changes"
+				read -r
+				if [[ "${REPLY}" == "diff" ]]; then
+					git diff
+				elif [[ "${REPLY}" == "exit" ]]; then
+					exit 1
+				elif [[ "${REPLY}" == "" ]]; then
+					break
+				else
+					echo "Unknown command!"
+				fi
+			done
+		elif [[ $(git branch | grep "*" | awk '{print $2}') != "${LIB_TAG}" && -n "${LIB_TAG}" ]]; then
+			git checkout "${LIB_TAG:-master}"
+			git pull
+		fi
+	fi
+
+}
+
+function do_update_src() {
+	TMPFILE=$(mktemp)
+	chmod 644 "${TMPFILE}"
+	{
+
+		echo SRC="$SRC"
+		echo LIB_TAG="$LIB_TAG"
+		declare -f update_src
+		echo "update_src"
+
+	} > "$TMPFILE"
+
+	#do not update/checkout git with root privileges to messup files onwership.
+	#due to in docker/VM, we can't su to a normal user, so do not update/checkout git.
+	if [[ $(systemd-detect-virt) == 'none' ]]; then
+
+		if [[ "${EUID}" == "0" ]]; then
+			su "$(stat --format=%U "${SRC}"/.git)" -c "bash ${TMPFILE}"
+		else
+			bash "${TMPFILE}"
+		fi
+
+	fi
+
+	rm "${TMPFILE}"
+}
+
+function handle_vagrant() {
+	# Check for Vagrant
+	if [[ "${1}" == vagrant && -z "$(command -v vagrant)" ]]; then
+		display_alert "Vagrant not installed." "Installing"
+		sudo apt-get update
+		sudo apt-get install -y vagrant virtualbox
+	fi
+}
+
+function handle_docker() {
+	# Install Docker if not there but wanted. We cover only Debian based distro install. On other distros, manual Docker install is needed
+	if [[ "${1}" == docker && -f /etc/debian_version && -z "$(command -v docker)" ]]; then
+
+		DOCKER_BINARY="docker-ce"
+
+		# add exception for Ubuntu Focal until Docker provides dedicated binary
+		codename=$(cat /etc/os-release | grep VERSION_CODENAME | cut -d"=" -f2)
+		codeid=$(cat /etc/os-release | grep ^NAME | cut -d"=" -f2 | awk '{print tolower($0)}' | tr -d '"' | awk '{print $1}')
+		[[ "${codename}" == "debbie" ]] && codename="buster" && codeid="debian"
+		[[ "${codename}" == "ulyana" || "${codename}" == "jammy" ]] && codename="focal" && codeid="ubuntu"
+
+		# different binaries for some. TBD. Need to check for all others
+		[[ "${codename}" =~ focal|hirsute ]] && DOCKER_BINARY="docker containerd docker.io"
+
+		display_alert "Docker not installed." "Installing" "Info"
+		sudo bash -c "echo \"deb [arch=$(dpkg --print-architecture)] https://download.docker.com/linux/${codeid} ${codename} stable\" > /etc/apt/sources.list.d/docker.list"
+
+		sudo bash -c "curl -fsSL \"https://download.docker.com/linux/${codeid}/gpg\" | apt-key add -qq - > /dev/null 2>&1 "
+		export DEBIAN_FRONTEND=noninteractive
+		sudo apt-get update
+		sudo apt-get install -y -qq --no-install-recommends ${DOCKER_BINARY}
+		display_alert "Add yourself to docker group to avoid root privileges" "" "wrn"
+		"${SRC}/compile.sh" "$@"
+		exit $?
+
+	fi
+}
+
+function prepare_userpatches() {
+	# Create userpatches directory if not exists
+	mkdir -p "${SRC}"/userpatches
+
+	# Create example configs if none found in userpatches
+	if ! ls "${SRC}"/userpatches/{config-default.conf,config-docker.conf,config-vagrant.conf} 1> /dev/null 2>&1; then
+
+		# Migrate old configs
+		if ls "${SRC}"/*.conf 1> /dev/null 2>&1; then
+			display_alert "Migrate config files to userpatches directory" "all *.conf" "info"
+			cp "${SRC}"/*.conf "${SRC}"/userpatches || exit 1
+			rm "${SRC}"/*.conf
+			[[ ! -L "${SRC}"/userpatches/config-example.conf ]] && ln -fs config-example.conf "${SRC}"/userpatches/config-default.conf || exit 1
+		fi
+
+		display_alert "Create example config file using template" "config-default.conf" "info"
+
+		# Create example config
+		if [[ ! -f "${SRC}"/userpatches/config-example.conf ]]; then
+			cp "${SRC}"/config/templates/config-example.conf "${SRC}"/userpatches/config-example.conf || exit 1
+		fi
+
+		# Link default config to example config
+		if [[ ! -f "${SRC}"/userpatches/config-default.conf ]]; then
+			ln -fs config-example.conf "${SRC}"/userpatches/config-default.conf || exit 1
+		fi
+
+		# Create Docker config
+		if [[ ! -f "${SRC}"/userpatches/config-docker.conf ]]; then
+			cp "${SRC}"/config/templates/config-docker.conf "${SRC}"/userpatches/config-docker.conf || exit 1
+		fi
+
+		# Create Docker file
+		if [[ ! -f "${SRC}"/userpatches/Dockerfile ]]; then
+			cp "${SRC}"/config/templates/Dockerfile "${SRC}"/userpatches/Dockerfile || exit 1
+		fi
+
+		# Create Vagrant config
+		if [[ ! -f "${SRC}"/userpatches/config-vagrant.conf ]]; then
+			cp "${SRC}"/config/templates/config-vagrant.conf "${SRC}"/userpatches/config-vagrant.conf || exit 1
+		fi
+
+		# Create Vagrant file
+		if [[ ! -f "${SRC}"/userpatches/Vagrantfile ]]; then
+			cp "${SRC}"/config/templates/Vagrantfile "${SRC}"/userpatches/Vagrantfile || exit 1
+		fi
+
+	fi
+}