#!/bin/bash

set -e
#set -x

if [ $(id -u) -ne 0 ]; then
	echo "$0 needs root: please use sudo."
	exit 1
fi

# Source Auxilium (args parser)
. /usr/share/auxilium/auxilium.sh

auxilium_init $(basename $0) -m 'Displays list of devices and their physical location'

auxilium_add redmine -s r -t opt -m 'Format for redmine'
auxilium_add field -s f -t str -l "id|mount_point|majmin|scsi|vendor|serial|model|firmware|size|description|fo_name" -d "serial" -m 'Field to display under Device Name'
auxilium_add csv -t opt -m 'Return a CSV formatted list of devices (id,kname,mount-point,scsi,majmin,status,vendor,model,serial,firmware,size,description)'
auxilium_add locate -t positional -a -m 'List of device to locate (sdx or BoxXDiskX)'


if [ -f "/usr/share/ferreus-orbit/common.sh" ]; then
	. "/usr/share/ferreus-orbit/common.sh"
elif [ -f "share/common.sh" ] ; then
	. "share/common.sh"
elif [ -f "../share/common.sh" ] ; then
	. "../share/common.sh"
else
	echo "Cannot load common.sh: exiting"
	exit 1
fi

auxilium_parse "$@"

fo_lock

fo_get_name(){
	local width="$2"
	local locate="False"
	local fo_name="$1"
	local fullname=$(fo_get_disk_info ${fo_name} mount_point)
	[ -z "$fullname" ] && fullname=$(fo_get_disk_info ${fo_name} kname)
	name=$(basename ${fullname})
	[ "${name}" == "$AUXILIUM_ARG_locate" ] && locate="True"
	mount | grep -q ${fullname} || fullname="${UNMOUNTED}${fullname}"
	fo_center_align "$width" $(fo_contain "$AUXILIUM_ARG_locate" "${name}") "$fullname"
}

fo_get_field(){
	local fo_name="$1"
	local field="$AUXILIUM_ARG_field"
	local width="$2"
	local name=$(fo_get_disk_info ${fo_name} mount_point)
	[ -z "$name" ] && name=$(fo_get_disk_info ${fo_name} kname)
	local shortname=$(basename ${name} 2>/dev/null)
	f=$(fo_get_disk_info ${fo_name} ${field})
	fo_center_align "$width" $(fo_contain "$AUXILIUM_ARG_locate" "${shortname}") "$f"
}


fo_display_array(){
	if [ "$1" == "--header" ]; then
		local header='True'
		shift 1
	elif [ "$1" == "--footer" ]; then
		local footer='True'
		shift 1
	fi
	local fo_name=$1
	local k_name=$2
	[ "${3}" == "null" ] && local size=0 || local size=$3
	local model=$4
	local sn=$5
	local fw=$6
	local descr=$7
	local color_begin=''
	local color_end=''
	local arrow=''
	local LEN_ARROW=4
	local to_locate=0
	local short_fo=$(basename ${fo_name} 2>/dev/null)
	local short_k=$(basename ${k_name} 2>/dev/null)
	if [ ! -z "${fo_name}" ] && [ "$(fo_contain "${AUXILIUM_ARG_locate}" "${short_fo}")" -eq 1 ]; then
		to_locate=1
	elif [ -z "${fo_name}" ] && [ ! -z "${k_name}" ] && [ "$(fo_contain "${AUXILIUM_ARG_locate}" "${short_k}")" -eq 1 ]; then
		to_locate=1
	fi
	if [ "$to_locate" -eq 1 ];then
		if [ -z "$AUXILIUM_ARG_redmine" ]; then
			local color_begin=${RED}
			local color_end=${NORM}
		else
			arrow='--> '
		fi
	fi
	let "lentotal = LEN_ARROW + LEN_MODEL + LEN_SN + LEN_MOUNT + 60"
	let "lenm = LEN_MOUNT + LEN_ARROW"
	if [ -n "${header}" ]; then
		fo_repeat_char '_' $lentotal;echo
		printf -- "|%-${lenm}s|%-12s|%-15s|%-${LEN_MODEL}s|%-${LEN_SN}s|%-10s|%-15s|\n" "$fo_name" "$k_name" "$size" "$model" "$sn" "$fw" "$descr"
		printf -- "|$(fo_repeat_char '-' ${lenm})|------------|---------------|$(fo_repeat_char '-' ${LEN_MODEL})|$(fo_repeat_char '-' ${LEN_SN})|----------|---------------|\n"
	elif [ -n "${footer}" ]; then
		fo_repeat_char '¯' $lentotal;echo
	else
		printf -- "|${color_begin}%-${lenm}s|%-12s|%-15s|%-${LEN_MODEL}s|%-${LEN_SN}s|%-10s|%-15s${color_end}|\n" "${arrow}${fo_name}" "${arrow}${k_name}" "${size}" "$model" "$sn" "$fw" "$descr"
	fi
}

fo_write_disks_infos(){
	local did=$1
	local kname=$2
	local mount_point=$3
	local scsi=$4
	local majmi=$5
	local vendor=$6
	local model=$7
	local serial=$8
	local version=$9
	shift 9
	local size=$1
	local descr=$2
	eval $(udevadm info -q property --export ${kname})
	local majmin="${MAJOR}:${MINOR}"
	if fo_check_raid; then
		local descr="RAIDx Disk"
		local dids=$(fo_get_did)
		[ -e "$kname" ] && local dev=$kname || local dev="/dev/sda"
		for did in $dids; do
			local fo_name=$(fo_get_sorted_name $kname $did)
			unset vendor model serial version
			while IFS=: read k_smt v_smt; do
				k_smt=$(echo $k_smt)
				case $k_smt in
					"Vendor")
						vendor=$(echo ${v_smt})
						;;
					Serial\ [Nn]umber)
						serial=$(echo ${v_smt})
						;;
					"Device Model"|"Product")
						model=$(echo ${v_smt})
						;;
					"Revision"|"Firmware Version")
						version=$(echo ${v_smt})
						;;
					"LU is fully provisioned")
						descr="RAID0 Disk"
						;;
				esac
			done <<< "$(smartctl -i -d megaraid,${did} ${kname} 2>/dev/null)"
			printf "%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s\n" \
						"${did}" \
						"$kname" \
						"$mount_point" \
						"$fo_name" \
						"$scsi" \
						"$majmin" \
						"$vendor" \
						"$model" \
						"$serial" \
						"$version" \
						"$size" \
						"$descr" >> $FO_CACHE_DISKS_INFOS
		done
		unset did kname mount_point scsi majmin size descr
	else
		local fo_name=$(fo_get_sorted_name $kname)
		printf "%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s\n" \
					"${id}" \
					"$kname" \
					"$mount_point" \
					"$fo_name" \
					"$scsi" \
					"$majmin" \
					"$vendor" \
					"$model" \
					"$serial" \
					"$version" \
					"$size" \
					"$descr" >> $FO_CACHE_DISKS_INFOS
	fi
}

fo_load_disks_infos(){
	# File format : did,kname,mountpoint,fo_name,scsi,majmin,
	# vendor,model,serial,version,size,description
	#id,kname,mountpoint,scsi,majmin,status,vendor,model,serial,version,size,description
	fo_use_cache_file "${FO_CACHE_DISKS_INFOS}" && return 0|| true
	fo_use_cache_file "${FO_CACHE_FILE_LSHW}" || lshw -c disk 2>/dev/null > "${FO_CACHE_FILE_LSHW}" || true
	local use_fo=$USE_FO_NAME
	USE_FO_NAME=1
	while IFS=: read k v; do
		k=$(echo ${k})
		[ -z "$skip" ] || continue
		case "$k" in
		'*-cdrom')
				[ ! -z "$id" ] && fo_write_disks_infos "${id}" "$kname" "$mount_point" "$scsi" "$majmin" "$vendor" "$model" "$serial" "$version" "$size" "$descr"
				unset id kname mount_point scsi majmin vendor model serial version size descr
				local skip=1
		;;
		'*-disk')
			if [ -z "$v" ]; then
				[ ! -z "$id" ] && fo_write_disks_infos "${id}" "$kname" "$mount_point" "$scsi" "$majmin" "$vendor" "$model" "$serial" "$version" "$size" "$descr"
				unset id kname mount_point scsi majmin vendor model serial version size descr
				local skip=1
			else
				[ ! -z "$id" ] && fo_write_disks_infos "${id}" "$kname" "$mount_point" "$scsi" "$majmin" "$vendor" "$model" "$serial" "$version" "$size" "$descr"
				unset skip kname id mount_point scsi majmin vendor model serial version size descr
				local id=${v}
			fi
			;;
		'description')
			local descr=$(echo ${v})
		;;
		'logical name')
			case "$v" in
			" /dev/"*)
				local kname=$(echo ${v})
				;;
			" /srv/node/"*)
				local mount_point=$(echo ${v})
				;;
			esac
		;;
		'vendor')
			local vendor=$(echo ${v})
		;;
		'product')
			local model=$(echo ${v})
		;;
		'bus info')
			local scsi=$(echo ${v} | cut -d@ -f2)
		;;
		'version')
			local version=$(echo ${v})
		;;
		'serial')
			local serial=$(echo ${v})
		;;
		'size')
			local size=$(echo ${v})
		;;
		esac
	done < ${FO_CACHE_FILE_LSHW}
	[ -z "$skip" ] && fo_write_disks_infos "$id" "$kname" "$mount_point" "$scsi" "$majmin" "$vendor" "$model" "$serial" "$version" "$size" "$descr"
	USE_FO_NAME=$use_fo
}

fo_get_disk_info(){
	local field=$2
	local search=",${1},"
	while IFS=, read id kname mount_point fo_name scsi majmin vendor model serial fw size descr; do
		case "$field" in
		id)
			echo ${id}
		;;
		majmin)
			echo ${majmin}
		;;
		scsi)
			echo ${scsi}
		;;
		vendor)
			echo ${vendor}
		;;
		serial)
			echo ${serial}
		;;
		model)
			echo ${model}
		;;
		firmware)
			echo ${fw}
		;;
		description)
			echo ${descr}
		;;
		mount_point)
			echo ${mount_point}
		;;
		fo_name)
			echo ${fo_name}
		;;
		kname)
			echo ${kname}
		;;
		size)
			echo ${size}
		;;
		esac
	done <<< $(grep ${search} ${FO_CACHE_DISKS_INFOS})
}

main(){
	fo_load_disks_infos || true
	if [ -z "$AUXILIUM_ARG_csv" ]; then
		LEN_SN=18
		LEN_MODEL=18
		LEN_MOUNT=15
		while IFS="," read -r id kname mount_point fo_name scsi majmin vendor model serial fw size descr
		do
			[ "${#serial}" -gt "$LEN_SN" ] && LEN_SN=${#serial}
			[ "${#model}" -gt "$LEN_MODEL" ] && LEN_MODEL=${#model}
			[ "${#mount_point}" -gt "$LEN_MOUNT" ] && LEN_MOUNT=${#mount_point}
		done < $FO_CACHE_DISKS_INFOS
		fo_display_array --header "Mount point" "Kernel name" Size Model Serial Firmware Description
		while IFS="," read -r id kname mount_point fo_name scsi majmin vendor model serial fw size descr
		do
			fo_display_array "$mount_point" "$kname" "$size" "$model" "$serial" "$fw" "$descr"
		done <<< $(sort -t ',' -k 3 $FO_CACHE_DISKS_INFOS)
		fo_display_array --footer;echo
		local LEN_FIELD=18
		[ "$AUXILIUM_ARG_field" == "serial" ] && LEN_FIELD=$LEN_SN
		[ "$AUXILIUM_ARG_field" == "model" ] && LEN_FIELD=$LEN_MODEL

		# fo_build_dispo
		local underscore=$(fo_repeat_char '_' ${LEN_FIELD})
		local minus=$(fo_repeat_char '-' ${LEN_FIELD})
		local higher=$(fo_repeat_char '¯' ${LEN_FIELD})
		local FO_DISPOSITION=$(printf "${FO_DISPOSITION}"|sed "s%{UNDERSCORE}%${underscore}%g")
		FO_DISPOSITION=$(printf "${FO_DISPOSITION}"|sed "s%{MINUS}%${minus}%g")
		FO_DISPOSITION=$(printf "${FO_DISPOSITION}"|sed "s%{HIGHER}%${higher}%g")
		while IFS="," read -r id kname mount_point fo_name scsi majmin vendor model serial fw size descr
		do
			# [ -z "$mount_point" ] && local fullname=$kname || local fullname=$mount_point
			# local name=$(basename "$fullname")
			local n=$(fo_get_name "$fo_name" ${LEN_FIELD})
			local f=$(fo_get_field "$fo_name" ${LEN_FIELD})
				FO_DISPOSITION=$(printf "${FO_DISPOSITION}"|sed "s%{${fo_name}}%${n}%" 2>/dev/null)
				FO_DISPOSITION=$(printf "${FO_DISPOSITION}"|sed "s%{SN_${fo_name}}%${f}%" 2>/dev/null)
		done <<< $(sort -t ',' -k 3 $FO_CACHE_DISKS_INFOS)
		# final replace {new_line} by \n and @ by \
		FO_DISPOSITION=$(printf "${FO_DISPOSITION}"|sed "s%{new_line}%\n%g"| tr '@' '\\')
		printf "${FO_DISPOSITION}\n"
	else
		sort -t ',' -k 3 "$FO_CACHE_DISKS_INFOS"
	fi
}

main