#!/bin/bash

# Copyright (C) 2014 Jakob Borg and other contributors. All rights reserved.
# Use of this source code is governed by an MIT-style license that can be
# found in the LICENSE file.

id1=I6KAH76-66SLLLB-5PFXSOA-UFJCDZC-YAOMLEK-CP2GB32-BV5RQST-3PSROAU
id2=JMFJCXB-GZDE4BN-OCJE3VF-65GYZNU-AIVJRET-3J6HMRQ-AUQIGJO-FKNHMQU

go build json.go
go build md5r.go
go build genfiles.go

start() {
	echo "Starting..."
	STTRACE=model,scanner STPROFILER=":9091" syncthing -home "f1" > 1.out 2>&1 &
	STTRACE=model,scanner STPROFILER=":9092" syncthing -home "f2" > 2.out 2>&1 &
	sleep 1
}

stop() {
	echo "Stopping..."
	for i in 1 2 ; do
		curl -HX-API-Key:abc123 -X POST "http://localhost:808$i/rest/shutdown"
	done
	sleep 1
}

setup() {
	echo "Setting up..."
	rm -rf s? s??-?
	rm -rf f?/*.idx.gz f?/index
	mkdir -p s1
	pushd s1 >/dev/null
	../genfiles
	../md5r > ../md5-1
	popd >/dev/null
}

testConvergence() {
	torestart="$1"
	prevcomp=0

	while true ; do
		sleep 5
		comp=$(curl -HX-API-Key:abc123 -s "http://localhost:8081/rest/connections" | ./json "$id2/Completion")
		comp=${comp:-0}
		echo $comp / 100

		if [[ $comp == 100 ]] ; then
			echo Done
			break
		fi

		# Restart if the destination has made some progress
		if [[ $comp -gt $prevcomp ]] ; then
			prevcomp=$comp
			curl -HX-API-Key:abc123 -X POST "http://localhost:$torestart/rest/restart"
		fi
	done

	echo "Verifying..."

	pushd s2 >/dev/null
	../md5r | grep -v .stversions > ../md5-2
	popd >/dev/null

	if ! cmp md5-1 md5-2 ; then
		echo Repos differ
		stop
		exit 1
	fi
}

echo Testing reconnects during pull where the source node restarts
setup
start
testConvergence 8081
stop

echo Testing reconnects during pull where the destination node restarts
setup
start
testConvergence 8082
stop