functions

#!/bin/bash

# yad hackery

if [ -e "./bin/yad" ]
then
	PATH="${PATH}:./bin"
	ICONS="./icons"
elif [ -e "../bin/yad" ]
then
	PATH="${PATH}:../bin"
	ICONS="../icons"
fi

# buttons

CANCEL_BTN=1
CLOSE_BTN=252
OK_BTN=0
LOG_BTN=2

# menu options

DEFAULT_TITLE="Genesi i.MX53 Hardware Tests"
DEFAULT_WIDTH=350
DEFAULT_HEIGHT=250
SCREEN_RIGHT=600
SCREEN_TOP=0

MAIN_MENU="--title=${DEFAULT_TITLE} --list --print-column=2 --column=icon:IMG --column=Test --column=Status --column=@fore@"
FEEDBACK_MENU="--title=${DEFAULT_TITLE} --width=${DEFAULT_WIDTH} --list --column=:RD --column=Status --column=Action:HD"

# general functions

yad_default_menu()
{
	yad --title="${DEFAULT_TITLE}" \
--width=${DEFAULT_WIDTH} --height=${DEFAULT_HEIGHT} "$@"
}

yad_default_feedback_menu()
{
	yad_default_menu --list \
--column=:RD --column=Status --column=Action:HD \
"$@"
}

yad_stat_menu()
{
	local stat=${1}

	yad --title="${DEFAULT_TITLE}" --width=${DEFAULT_WIDTH} \
--text=${stat}
}

txtmenu_from_file()
{
	local file=${1}
	local title=${2}

	if [ -z "${title}" ]
	then
		title="${DEFAULT_TITLE}"
	fi

	 yad --title="${title}" --text="$(cat ${file})"
}

ask_continue()
{
	local btn=$? # TODO UGLY!

	if close_menu ${btn}
	then
		if [ -z "${stat}" ] # this is bad and I should feel bad
		then
			echo "Incomplete"
		else
			echo "${stat}"
		fi

		exit 1
	fi
}

if_failure()
{
	local stat=${1}

	if [ "${stat}" = "Failure" ]
	then
		yad_default_menu \
--text="The previous test ended in failure. Would you like to continue?"
		return $?
	else
		return 0
	fi
}

## progress windows
progress_view_dummy()
{
	local ttime=${1}
	local opts="$@"

	for i in $(seq 1 $ttime)
	do
		sleep 1
		echo ""
	done | progress_view ${ttime} "" "" ${opts}
}

progress_view()
{
	local ttime=${1}
	local title=${2}
	local text=${3}
	
	shift 3

	local opts="$@"

	local start_time=${SECONDS}

    if [ -z "${title}" ]
    then
        title="${DEFAULT_TITLE}"
    fi

    if [ -z "${text}" ]
    then
        text="Test in progress. Please wait."
    fi

	while read line
	do
		if [ ! -z "${line}" ] # coming from dummy
		then
			echo "#${line}" | tee -a ./log # dirty but gets the job done
		fi

		echo "100 * (${SECONDS} - ${start_time}) / ${ttime}" | bc -l
	done | yad --title="${title}" --on-top \
--geometry=${DEFAULT_WIDTH}x${DEFAULT_HEIGHT}+${SCREEN_RIGHT}+${SCREEN_TOP} \
--progress --text="${text}" ${opts}
}

# when time unknown but on step complete can read from stdin
progress_view_steps()
{
	local steps=${1}
	local title=${2}
	local text=${3}
	
	shift 3

	local opts="$@"

	local start_time=${SECONDS}

    if [ -z "${title}" ]
    then
        title="${DEFAULT_TITLE}"
    fi

    if [ -z "${text}" ]
    then
        text="Test in progress. Please wait."
    fi

	for i in $(seq 1 ${steps})
	do
		read line
		#echo "#${line}" | cut -c 1-100 | tee -a ./log # still dirty, still working
		echo "100 * (${i} / ${steps})" | bc -l
	done | yad --title="${title}" --width=${DEFAULT_WIDTH} \
--progress --text="${text}" ${opts}
}

continue_after_failure()
{
	local stat=${1}
	local res=0

	if [ ${stat} = "Failure" ]
	then
	    res=$(yad_default_menu --text="The previoius test ended in failure. Would you like to continue?")
	fi

	return ${res}
}

# TODO not linked with the structure of FEEDBACK_MENU at all!
get_feedback_status()
{
	local result="${1}"

	echo $($(echo "${result}" | cut -d "|" -f3) )
}

logged_feedback_menu()
{
	local stage=${1}
	local feedback_menu="${2}"

	log "After running stage ${stage}"
	log_default_feedback "${feedback_menu}"
}

result_good()
{
	echo "Success"
}

result_problem()
{
	log_default_describe_problem "$@"

	echo "Failure"
}

result_none()
{
	log "Nothing happened!"
	echo "Failure"
}

current_status()
{
	oldStat=${1}
	newStat=${2}

	if [ -z "${oldStat}" ]
	then
		echo ${newStat}
	elif [[ "$oldStat" = $newStat  ]]
	then
		echo $oldStat
	else
		echo "Mixed"
	fi
}

close_menu()
{
	btn=${1}

	if [ ${btn} -eq ${OK_BTN} ] || [ ${btn} -eq ${LOG_BTN} ]
	then
		return 1
	else
		return 0
	fi
}

mk_menu_body()
{
	local file=${1}
	local process=${2}

	if [ -z "${process}" ]
	then
		process="echo"
	fi

	local accum_body=""
	local tmp=""

	while read line
	do
		tmp=$(${process} ${line})
		accum_body="${accum_body} ${tmp}"
	done < ${file}

	echo ${accum_body}
}

parse_item()
{
	local item_num="${1}"
	local menu_result="${2}"

	echo "${menu_result}" | cut -d"|" -f"${item_num}"
}

# logging
init_test_log()
{
	# bad form - implicitly takes logging directory as
	# current directory

	local logfile="$(pwd)/log"

	echo "----------------------------------------" >> "${logfile}"
	date >> "${logfile}"
	echo "Logging for test:"		>> "${logfile}"
	echo $(basename "$(pwd)")		>> "${logfile}"
}

log()
{
	local msg="${1}"
	local logfile="$(pwd)/log"

	echo -e $(date "+[%H:%M:%S] ${msg}") >> "${logfile}"
}

log_default_feedback()
{
	local options_filename="${1}"
	local res=""

	while :
	do
		res=$(yad_default_feedback_menu --text="Please select the most appropriate" --rest="${options_filename}")

		if [ -z "${res}" ]
		then
			yad_default_menu --text="Please select at least one option!"
		else
			break
		fi
	done

	log "User feedback:\n${res}"
	echo "${res}"
}

log_default_describe_problem()
{
	local form_fields=""
	local res=""

	while [ ! $# -eq 0 ]
	do
		form_fields="${form_fields} --field=${1}:CHK FALSE"
		shift 1
	done

	# TODO harder to make sure the user selects one
	res=$(yad_default_menu \
--text="Please select the most appropriate" --form \
${form_fields} \
--field="Leave any additional comments below":TXT)a

	log "\nUser asked to describe problem:"
	log "${form_fields}"
	log "${res}"
}

# misc menu
get_icon_and_color()
{
    local sstatus=${1}
    local color=""
    local icon=""

    case ${sstatus} in
        Incomplete)
            color="black"
            icon="none"
            ;;
        Failure)
            color="red"
            icon="${ICONS}/failure.png"
            ;;
        Success)
            color="green"
            icon="${ICONS}/success.png"
            ;;
        Mixed)
            color="orange"
            icon="${ICONS}/mixed.png"
            ;;
        *)
            color="purple"
            icon="none"
            ;;
    esac

    echo "${icon} ${color}"
}

get_button()
{
	local btn=$?

	echo ${btn}
	return ${btn}
}

update_test_status()
{
    local testName="${1}"
    local sstatus=${2}
    local fileName=${3}

    if [ -z "${sstatus}" ]
    then
        exit 0
    fi

    # this is bad and I feel bad
    while read icon_line
    do # if there's one then there's 3 more
        read test_line
        read stat_line
        read color_line

        if [ "${test_line}" = "${testName}" ] && \
[ ! "${sstatus}" = "Incomplete" ]
        then
            local new_icon_color="$(get_icon_and_color \
${sstatus})"

            echo "${new_icon_color}" | cut -d ' ' -f1
            echo "${test_line}"
            echo "${sstatus}"
            echo "${new_icon_color}" | cut -d ' ' -f2
        else
            echo "${icon_line}"
            echo "${test_line}"
            echo "${stat_line}"
            echo "${color_line}"
        fi

    done < ${fileName} > tmp_menu_file

    mv tmp_menu_file ${fileName}
}