init

2023-07-16 20:03:11 +00:00 · 2023-07-16 20:03:11 +00:00 · 545473f1df
commit 545473f1df
9 changed files with 1017 additions and 0 deletions
--- a/bin/main.sh
+++ b/bin/main.sh
@ -0,0 +1,131 @@
+#! /bin/bash 
+#set -e 
+APP_PATH=$( cd $(dirname $0) && pwd)
+SRC_PATH=$( cd $(dirname $0)/../src && pwd)
+cd $APP_PATH
+clear
+echo -e "## DEVICES\n"
+DEVICES=$(lsblk /dev/sd? --nodeps --output NAME,MODEL,VENDOR,SIZE,TYPE,STATE)
+echo "$DEVICES"
+MOUNT_ROOT=$(mount |grep " / " |awk '{print $1}' |sed -r 's=/dev/(...)[0-9]$=\1=')
+declare -a NOT_ROOT
+while read device others; do 
+	[[ "$device" != $MOUNT_ROOT ]] && [[ "$device" != "NAME" ]] && NOT_ROOT+=( $device )
+done <<< "$DEVICES" 
+
+NOT_ROOT_STR=$( echo ${NOT_ROOT[@]} )
+echo -e "\n## ROOT MOUNTED DEVICE\n$MOUNT_ROOT"
+echo -e "\n## OTHER DEVICES\n${NOT_ROOT_STR}\n"
+
+
+echo -e "\n## SMARTCTL TESTS" 
+read -e -p "Do you want to run tests? [y/N] "  -n 1
+REPLY=${REPLY:-N}
+if [[ "N" != "${REPLY^^}" ]] ; then 
+	echo -e "\n## SMARTCTL DISKS SELECTION" 
+	echo -e "Do you want to USE ALL non root devices (empty reply)? \nOr else please type specific DEVICES NAMES to include (ex: '${NOT_ROOT_STR}')?"
+	read -e -p "Type your answer: " 
+	CMD="$SRC_PATH/smartjson.py long "
+	if [[ -z "$REPLY" ]] ; then 
+		CMD+=" -e $MOUNT_ROOT"
+	else
+
+		DEVICES=${REPLY}
+		for i in $DEVICES; do 
+			CMD+=" -d $i"
+		done
+	fi
+
+	echo -e "\n## READY"
+	read -e -i Y -n 1 -p "About to run command '$CMD'. OK [Y/n]? " 
+
+	# $CMD
+	START=$(date "+%s")
+	set +e
+	while true; do 
+		clear
+		echo -e "\n## CHECKING STATUS"
+
+		date
+		echo -e "\nRunning since: $(( $( date +%s ) - $START )) seconds.\n"
+		$SRC_PATH/smartjson.py status && break
+		echo -e "\n## MANUAL HALT"
+		read -e -i Y -p "Stop ?" -t 15
+		[[ ${REPLY^^} == "Y" ]] && break
+	done 
+
+fi
+
+echo -e "\n## SMARTCTL DISKS STATUS"
+$SRC_PATH/smartjson.py list
+
+
+echo -e "\n## DISKS BACKGROUND ERASE\nCaution: this might cause data loss."
+read -e -p "Do you want to run a background erasure? [y/N] "  -n 1
+REPLY=${REPLY:-N}
+if [[ "N" != "${REPLY^^}" ]] ; then 
+	ERASE_DEVICES=""
+	while [[ -z "$ERASE_DEVICES" ]] ; do 
+		echo "Please provide device names. Ex: 'sda sdc'"
+		read -e -p "Devices to erase: " -i "${NOT_ROOT_STR}" ERASE_DEVICES
+		declare -A DD_CMD
+		ERROR="false"
+		for f in $ERASE_DEVICES; do 
+			if [[ ! -b /dev/$f ]] || [[ "$MOUNT_ROOT" == $f ]] ; then  
+							echo "ERROR. $f is not a valid device."
+							ERROR="true"
+							break
+			fi	
+			DD_CMD["$f"]="dd if=/dev/zero of=/dev/$f bs=512K" 
+		done
+		if [[ "$ERROR" == "true" ]]; then
+			ERASE_DEVICES=""
+			continue
+		fi
+	done
+	echo -e "\n## CONFIRMATION\nYou are about to run the following commands."
+	for i in ${!DD_CMD[@]}; do 
+		echo "Disk $i: ${DD_CMD[$i]}"
+	done
+	read -e -p "Please type 'Yes' to validate: " VALIDATE
+	if [[ "YES" != "${VALIDATE^^}" ]]; then 
+		echo "EXIT"
+		exit
+	else
+		echo -e "\n## RUNNING ERASURE\n"
+		declare -A DD_PID
+		for i in ${!DD_CMD[@]}; do
+			${DD_CMD[$i]}&
+			DD_PID[$i]=$!
+			echo "$i PID ${DD_PID[$i]}..."
+			sleep 1
+		done
+		sleep 3
+	fi
+fi
+
+[[ -n ${DD_PID[@]} ]] && while true; do 
+	clear
+	echo -e "\n## CHECKING STATUS"
+	FINISHED_COUNT=O
+
+	if [[ ${#DD_PID[@]} -ne 0 ]] ; then 
+		for i in ${!DD_PID[@]}; do 
+			PID=${DD_PID[$i]}
+			ps -f -p $PID &>/dev/null
+			R=$?
+			if [[ 0 -ne $R ]] ; then
+				MSG="Finished" 
+				let $(( FINISHED_COUNT++ ))
+			else
+				MSG="Process running"
+			fi 
+			echo "DISK $i PID $i : $MSG"
+		done
+	fi
+	[[ ${#DD_PID[@]} -eq $FINISHED_COUNT ]] && break 
+	echo -e "\n## MANUAL HALT"
+	read -e -i Y -p "Stop ?" -t 15
+	[[ ${REPLY^^} == "Y" ]] && break
+done 
+