#!/bin/bash

if [ "$1" == "-v" ]; then
	VERBOSE='true'
fi

SOCKET='sock'

if pgrep go-pluginserver -l; then
	PREVIOUS_SERVER="yes"
else
	echo "starting server..."
	[ -S "$SOCKET" ] && rm "$SOCKET"
	./go-pluginserver -socket "$SOCKET" &
	pgrep go-pluginserver -l
fi

msg() {
	query="$1"
	[ -v VERBOSE ] && rq <<< "$query"
	METHOD="$(rq <<< "$1" -- 'at([2])')"
	response="$(rq <<< "$query" -M | nc -U "$SOCKET" -N | rq -m)"
	[ -v VERBOSE ] && rq <<< "$response"

	ERROR="$(rq <<< "$response" -- 'at [2] ')"
	RESULT="$(rq <<< "$response" -- 'at [3] ')"
}

assert_noerr() {
	if [ "$ERROR" != "null" ]; then
		echo "query: $query"
		echo "response: $response"
		echo "$METHOD : $ERROR" > /dev/stderr
		exit 1
	fi
	echo "$METHOD: ok"
}

assert_fld_match() {
	fld="$1"
	pattern="$2"

	fld_v="$(query_result 'at "'$fld'"')"
	if [[ "$fld_v" =~ "$pattern" ]]; then
		echo "==> $fld_v : ok"
	else
		echo "==> $fld_v : no match '$pattern'"
		exit 1
	fi
}

query_result() {
# 	echo "$RESULT"
	rq <<< "$RESULT" -- "$1"
}

msg '[0, 19, "plugin.SetPluginDir", ["/home/javier/devel/kong_dev/kong"]]'
assert_noerr

msg '[0, 19, "plugin.GetPluginInfo", ["go-log"]]'
assert_noerr

msg '[0, 19, "plugin.StartInstance", [{"Name":"go-log", "Config":"{\"reopen\":false, \"path\":\"/some/where/else/\"}"}]]'
assert_noerr
instanceID=$(query_result 'at "Id"')
echo "instanceID: $instanceID"

msg '[0, 19, "plugin.HandleEvent", [{"InstanceId": '$instanceID', "EventName": "access", "Params": [45, 23]}]]'
assert_noerr
eventId=$(query_result 'at "EventId"')

msg '[0, 19, "plugin.Step", [{"EventId": '$eventId', "Data": "access", "Params": [45, 23]}]]'
assert_noerr
callBack=$(query_result 'at "Data"')
assert_fld_match 'Data' 'kong.request.get_header:[\"host\"]'
# echo "callBack: $callBack"		# get_header('host')

msg '[0, 19, "plugin.Step", [{"EventId": '$eventId', "Data": "example.com"}]]'
assert_noerr
# callBack=$(query_result 'at "Data"')
assert_fld_match 'Data' 'kong.response.set_header:[\"x-hello-go\",\"Go says hello to example.com (/some/where/else/)\"]'
# echo "callBack: $callBack"		# set_header('x-hello-go', ....)

msg '[0, 19, "plugin.Step", [{"EventId": '$eventId', "Data": "ok"}]]'
assert_noerr
callBack=$(query_result 'at "Data"')
assert_fld_match 'Data' '"ret"'
# [ "$(query_result 'at "Data"')" == '"ret"' ] || exit 1


msg '[0, 19, "plugin.InstanceStatus", ['$instanceID']]'
assert_noerr

msg "[0, 19, \"plugin.CloseInstance\", [$instanceID]]"
assert_noerr


if [ ! -v PREVIOUS_SERVER ]; then
	pkill go-pluginserver
	rm "$SOCKET"
fi