k-smart 1.1

#!/bin/bash

Main() {
	export PATH=/bin:/usr/bin:/opt/local/bin:/usr/local/bin
	PathToSmartCtl="$(GetSmartCtl)"
	PathToKeys="${0%/*}"/keys

	CreateShellAlias

	sleep 1
	IsShiftPressed=$(${PathToKeys} shift)
	IsOptionPressed=$(${PathToKeys} option)

	case ${IsShiftPressed} in
		1)
			# shift is pressed, we are in test mode!
			# if option is pressed as well we're
			# performing a long test otherwise short
			if [ "X${IsOptionPressed}" = "X1" ]; then
				TestMode=long
			else
				TestMode=short
			fi
			GetListOfVolumes | while read ; do
				# check whether device is SMART capable
				"${PathToSmartCtl}" -a ${REPLY} >/dev/null 2>&1
				if [ $? -eq 0 ]; then
					echo -e "${AbstandSchraddel}Previous test results for ${REPLY}:\n"
					"${PathToSmartCtl}" -l selftest ${REPLY}
					echo -e "\nNow performing a ${TestMode} test on ${REPLY} in the background. You can close this window and try again in several minutes. Please see below the smartctl output.\n"
					"${PathToSmartCtl}" -t ${TestMode} ${REPLY}
					AbstandSchraddel="\n\n\n"
				fi
			done
			;;
		*)
			# we simply display SMART status for all devices
			GetListOfVolumes | while read ; do
				echo -e "${AbstandSchraddel}Examining ${REPLY}\n"
				"${PathToSmartCtl}" -a ${REPLY}
				AbstandSchraddel="\n\n\n"
			done
			;;
	esac
} # Main

CreateShellAlias() {
	if [ -f "${HOME}/.profile" ]; then
		MyTmpFile="$(mktemp /tmp/${0##*/}.XXXXXX || (echo -e "Not able to create temporary file below /tmp/. Skip creating shell alias." | logger ; return))"
		trap "rm -f \"${MyTmpFile}\"; exit 0" 0 1 2 3 15
		grep -v "^alias smartctl" <"${HOME}/.profile" >"${MyTmpFile}"
		echo -e "alias smartctl=\"${PathToSmartCtl}\"" >>"${MyTmpFile}"
		cat "${MyTmpFile}" >"${HOME}/.profile"
	else
		echo -e "alias smartctl=\"${PathToSmartCtl}\"" >>"${HOME}/.profile"
	fi
} # CreateShellAlias

GetListOfVolumes() {
	/usr/sbin/diskutil list | grep "^/dev/" | while read ; do
		/usr/sbin/diskutil corestorage info ${REPLY} >/dev/null 2>&1 || echo ${REPLY}
	done
} # GetListOfVolumes

GetSmartCtl() {
	mdfind "kMDItemCFBundleIdentifier == 'de.kaiser-edv.k-smart'" | grep "\.app$" | while read ; do
		SmartCtlVersion=$("${REPLY}"/Contents/Resources/smartctl | awk -F" " '/^smartctl / {print $2}')
		echo -e "${SmartCtlVersion}\t${REPLY}/Contents/Resources/smartctl"
	done | sort -r | head -n1 | cut -f2
} # GetSmartCtl

Main "$@"