#!/bin/sh
FAILED=0

if [ "$(echo '\e')" != '\e' ] ; then
	safe_echo() {
		echo -E "$@"
	}
else
	safe_echo() {
		echo "$@"
	}
fi

run_test() {
	SH="$1"
	SESNAME="powerline-shell-test-$$"
	screen -L -c tests/test_shells/screenrc -d -m -S "$SESNAME" \
		env LANG=en_US.UTF-8 BINDFILE="$BINDFILE" "$@"
	screen -S "$SESNAME" -X readreg a tests/test_shells/input.$SH
	sleep 0.3s
	screen -S "$SESNAME" -p 0 -X width 300 1
	screen -S "$SESNAME" -p 0 -X logfile tests/shell/screen.log
	screen -S "$SESNAME" -p 0 -X paste a
	while screen -S "$SESNAME" -X blankerprg "" > /dev/null ; do
		sleep 1s
	done
	./tests/test_shells/postproc.py tests/shell/screen.log
	if ! diff -u tests/test_shells/${SH}.ok tests/shell/screen.log | cat -v ; then
		return 1
	fi
	return 0
}

mkdir tests/shell
git init tests/shell/3rd
git --git-dir=tests/shell/3rd/.git checkout -b BRANCH

if ! run_test bash --norc --noprofile -i ; then
	echo "Failed bash"
	FAILED=1
fi
cp tests/shell/screen.log tests/bash.log
rm tests/shell/screen.log

if ! run_test zsh -f -i ; then
	echo "Failed zsh"
	FAILED=1
fi
cp tests/shell/screen.log tests/zsh.log
rm tests/shell/screen.log

rm -r tests/shell
exit $FAILED