#!/bin/bash
#
# Copyright (C) - 2017 Julien Desfossez <jdesfossez@efficios.com>
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License, version 2 only, as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc., 51
# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

SH_TAP=1

if [ "x${BT_TESTS_SRCDIR:-}" != "x" ]; then
	UTILSSH="$BT_TESTS_SRCDIR/utils/utils.sh"
else
	UTILSSH="$(dirname "$0")/../utils/utils.sh"
fi

# shellcheck source=../utils/utils.sh
source "$UTILSSH"

clean_tmp() {
	rm -rf "${out_path}" "${text_output1}" "${text_output2_intermediary}" "${text_output2}"
}

SUCCESS_TRACES=("${BT_CTF_TRACES_PATH}/succeed/"*)

# -2 because there is an empty trace that we skip
NUM_TESTS=$((${#SUCCESS_TRACES[@]} * 3 - 2))

plan_tests $NUM_TESTS

for path in "${SUCCESS_TRACES[@]}"; do
	out_path="$(mktemp -d)"
	text_output1="$(mktemp)"
	text_output2_intermediary="$(mktemp)"
	text_output2="$(mktemp)"
	trace="$(basename "${path}")"
	sort_cmd="cat" # by default do not sort the trace

	bt_cli "${text_output1}" "/dev/null" --no-delta "${path}"
	ret=$?
	cnt="$(wc -l < "${text_output1}")"
	if test "$ret" == 0 && test "${cnt// /}" == 0; then
		pass "Empty trace ${trace}, nothing to copy"
		clean_tmp
		continue
	fi

	# If the trace has a timestamp (starts with [), check if there are
	# duplicate timestamps in the output.
	# If there are, we have to sort the text output to make sure it is
	# always the same.
	head -1 "${text_output1}" | "${BT_TESTS_GREP_BIN}" "^\[" >/dev/null
	if test $? = 0; then
		uniq_ts_cnt="$("${BT_TESTS_AWK_BIN}" '{ print $1 }' < "${text_output1}" | sort | uniq | wc -l)"
		# Extract only the timestamp columns and compare the number of
		# unique lines with the total number of lines to see if there
		# are duplicate timestamps.
		if test "${cnt// /}" != "${uniq_ts_cnt// /}"; then
			diag "Trace with non unique timestamps, sorting the output"
			sort_cmd="sort"
			tmp="$(mktemp)"
			sort "${text_output1}" > "$tmp"
			rm "${text_output1}"
			text_output1="$tmp"
		fi
	fi

	bt_cli "/dev/null" "/dev/null" "${path}" --component sink.ctf.fs "--params=path=\"${out_path}\""
	ok $? "Copy trace ${trace} with ctf-fs sink"

	bt_cli "/dev/null" "/dev/null" "${out_path}"
	ok $? "Read the new trace in ${out_path}"

	bt_cli "${text_output2_intermediary}" "/dev/null" --no-delta "${out_path}"
	$sort_cmd "${text_output2_intermediary}" > "${text_output2}"
	cnt=$(diff "${text_output1}" "${text_output2}" | wc -l)
	test "${cnt// /}" == 0
	ok $? "Exact same content between the two traces"

	clean_tmp
done