#!/bin/sh
#
# Copyright (c) 2018-2020 Jiang Xin
#

test_description='Test proc-receive hook on special git-push refspec'

. ./test-lib.sh

bare=bare.git

create_commits_in () {
	repo="$1" &&
	if ! parent=$(git -C "$repo" rev-parse HEAD^{} 2>/dev/null)
	then
		parent=
	fi &&
	T=$(git -C "$repo" write-tree) &&
	shift &&
	while test $# -gt 0
	do
		name=$1 &&
		test_tick &&
		if test -z "$parent"
		then
			oid=$(echo $name | git -C "$repo" commit-tree $T)
		else
			oid=$(echo $name | git -C "$repo" commit-tree -p $parent $T)
		fi &&
		eval $name=$oid &&
		parent=$oid &&
		shift ||
		return 1
	done &&
	git -C "$repo" update-ref refs/heads/master $oid
}

test_expect_success "setup repository" '
	git init --bare $bare &&

	# Enable push options for bare.git.
	git -C $bare config receive.advertisePushOptions true &&

	git clone --no-local $bare work &&
	create_commits_in work A B
'

test_expect_success "setup hooks" '
	## pre-receive hook
	cat >$bare/hooks/pre-receive <<-EOF &&
	#!/bin/sh

	printf >&2 "execute: pre-receive hook\n"

	while read old new ref
	do
		printf >&2 ">> old: \$old, new: \$new, ref: \$ref.\n"
	done
	EOF

	## proc-receive hook
	cat >$bare/hooks/proc-receive <<-EOF &&
	#!/bin/sh

	printf >&2 "execute: proc-receive hook\n"

	while read old new ref
	do
		printf >&2 ">> old: \$old, new: \$new, ref: \$ref.\n"
	done
	EOF

	## post-receive hook
	cat >$bare/hooks/post-receive <<-EOF &&
	#!/bin/sh

	printf >&2 "execute: post-receive hook\n"

	while read old new ref
	do
		printf >&2 ">> old: \$old, new: \$new, ref: \$ref.\n"
	done
	EOF
	chmod a+x \
		$bare/hooks/pre-receive \
		$bare/hooks/post-receive \
		$bare/hooks/proc-receive \
'

test_expect_success "(1) standard git-push command" '
	(
		cd work &&
		git update-ref HEAD $A &&
		git push origin HEAD HEAD:maint 2>&1
	) >out &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $A, ref: refs/heads/master.
	remote: >> old: $ZERO_OID, new: $A, ref: refs/heads/maint.
	remote: execute: post-receive hook
	remote: >> old: $ZERO_OID, new: $A, ref: refs/heads/master.
	remote: >> old: $ZERO_OID, new: $A, ref: refs/heads/maint.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(1) push one special ref (create one)" '
	(
		cd work &&
		git update-ref HEAD $B &&
		git push origin HEAD:refs/for/master/my/topic
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: execute: post-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$B refs/for/master/my/topic
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(1) cleanup" '
	(
		cd $bare &&
		git update-ref -d refs/for/master/my/topic
	)
'

test_expect_success "(2) add config to turn on proc-receive" '
	git -C $bare config --add receive.procReceiveRefs refs/for/
'

test_expect_success "(2) standard git-push command" '
	(
		cd work &&
		git update-ref HEAD $A &&
		git push origin HEAD:a/b/c 2>&1
	) >out &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $A, ref: refs/heads/a/b/c.
	remote: execute: post-receive hook
	remote: >> old: $ZERO_OID, new: $A, ref: refs/heads/a/b/c.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(2) push one special ref" '
	(
		cd work &&
		git update-ref HEAD $B &&
		git push origin HEAD:refs/for/master/my/topic
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: execute: proc-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: execute: post-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(2) push both a normal and a special refs" '
	(
		cd work &&
		git update-ref HEAD $B &&
		git push origin \
			HEAD:refs/for/maint/my/topic \
			HEAD:refs/heads/master
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: execute: proc-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: execute: post-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$B refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(2) cleanup" '
	(
		cd $bare &&
		git update-ref refs/heads/master $A $B &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(2) push two special references (one is not registered)" '
	(
		cd work &&
		git push origin \
			HEAD:refs/for/maint/my/topic \
			HEAD:refs/drafts/maint/my/topic
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/drafts/maint/my/topic.
	remote: execute: proc-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: execute: post-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/drafts/maint/my/topic.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$B refs/drafts/maint/my/topic
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(2) cleanup" '
	(
		cd $bare &&
		git update-ref -d refs/drafts/maint/my/topic &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(2) register new ref prefix" '
	git -C $bare config --add receive.procReceiveRefs refs/drafts/
'

test_expect_success "(2) push to two special references (all registered)" '
	(
		cd work &&
		git push origin \
			HEAD:refs/for/master/my/topic \
			HEAD:refs/drafts/maint/my/topic
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/drafts/maint/my/topic.
	remote: execute: proc-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/drafts/maint/my/topic.
	remote: execute: post-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/drafts/maint/my/topic.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(3) remove proc-receive hook" '
	mv $bare/hooks/proc-receive $bare/hooks/proc-receive.ok
'

test_expect_success "(3) standard git-push command" '
	(
		cd work &&
		git update-ref HEAD $B &&
		git push origin HEAD:master 2>&1
	) >out &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	remote: execute: post-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$B refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(3) cleanup" '
	(
		cd $bare &&
		git update-ref refs/heads/master $A $B &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(3) push one special ref (failed)" '
	(
		cd work &&
		git update-ref HEAD $B &&
		test_must_fail git push origin HEAD:refs/for/a/b/c/my/topic
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/a/b/c/my/topic.
	remote: error: cannot to find hook '"'"'proc-receive'"'"'
	EOF
	test_cmp expect actual
'

test_expect_success "(3) push both a normal and a special refs (one failed)" '
	(
		cd work &&
		git update-ref HEAD $B &&
		test_must_fail git push origin \
			HEAD:refs/for/maint/my/topic \
			HEAD:refs/heads/master
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: error: cannot to find hook '"'"'proc-receive'"'"'
	remote: execute: post-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$B refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(3) cleanup" '
	(
		cd $bare &&
		git update-ref refs/heads/master $A $B &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(3) atomic push both a normal and a special refs (failed)" '
	(
		cd work &&
		git update-ref HEAD $B &&
		test_must_fail git push --atomic origin \
			HEAD:refs/for/maint/my/topic \
			HEAD:refs/heads/master
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: error: cannot to find hook '"'"'proc-receive'"'"'
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(4) new proc-receive hook (return error)" '
	cat >$bare/hooks/proc-receive <<-EOF &&
	#!/bin/sh

	printf >&2 "execute: proc-receive hook\n"

	while read old new ref
	do
		printf >&2 ">> old: \$old, new: \$new, ref: \$ref.\n"
	done

	printf >&2 "fail to run proc-receive\n"
	exit 1
	EOF
	chmod a+x $bare/hooks/proc-receive
'

test_expect_success "(4) standard git-push command" '
	(
		cd work &&
		git update-ref HEAD $B &&
		git push origin HEAD:master 2>&1
	) >out &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	remote: execute: post-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$B refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(4) cleanup" '
	(
		cd $bare &&
		git update-ref refs/heads/master $A $B &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(4) push both a normal and a special refs (one failed)" '
	(
		cd work &&
		git update-ref HEAD $B &&
		test_must_fail git push origin \
			HEAD:refs/for/maint/my/topic \
			HEAD:refs/heads/master
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: execute: proc-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: fail to run proc-receive
	remote: execute: post-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$B refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(4) cleanup" '
	(
		cd $bare &&
		git update-ref refs/heads/master $A $B &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(4) atomic push a normal and a special refs (failed)" '
	(
		cd work &&
		git update-ref HEAD $B &&
		test_must_fail git push --atomic origin \
			HEAD:refs/for/maint/my/topic \
			HEAD:refs/heads/master
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: execute: proc-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: fail to run proc-receive
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(5) restore proc-receive hook" '
	mv $bare/hooks/proc-receive $bare/hooks/proc-receive.fail &&
	mv $bare/hooks/proc-receive.ok $bare/hooks/proc-receive
'

test_expect_success "(5) push two special references" '
	(
		cd work &&
		git update-ref HEAD $B &&
		git push origin \
			HEAD:refs/for/maint/my/topic \
			HEAD:refs/for/a/b/c/my/topic
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/a/b/c/my/topic.
	remote: execute: proc-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/a/b/c/my/topic.
	remote: execute: post-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/a/b/c/my/topic.
	EOF
	test_cmp expect actual &&
	(
		cd $bare &&
		git show-ref
	) >actual &&
	cat >expect <<-EOF &&
	$A refs/heads/a/b/c
	$A refs/heads/maint
	$A refs/heads/master
	EOF
	test_cmp expect actual
'

test_expect_success "(6) new pre-receive hook hook (return error)" '
	mv $bare/hooks/pre-receive $bare/hooks/pre-receive.ok &&
	cat >$bare/hooks/pre-receive <<-EOF &&
	#!/bin/sh

	printf >&2 "execute: pre-receive hook\n"

	while read old new ref
	do
		printf >&2 ">> old: \$old, new: \$new, ref: \$ref.\n"
	done

	printf >&2 ">> ERROR: declined in pre-receive hook\n"
	exit 1
	EOF
	chmod a+x $bare/hooks/pre-receive
'

test_expect_success "(6) push two special references (declined)" '
	(
		cd work &&
		test_must_fail git push origin \
			HEAD:refs/for/master/my/topic \
			HEAD:refs/for/maint/my/topic
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/maint/my/topic.
	remote: >> ERROR: declined in pre-receive hook
	EOF
	test_cmp expect actual
'

test_expect_success "(6) push both a normal and a special refs (declined)" '
	(
		cd work &&
		test_must_fail git push origin \
			HEAD:refs/for/master/my/topic \
			HEAD:refs/heads/master
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $A, new: $B, ref: refs/heads/master.
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: >> ERROR: declined in pre-receive hook
	EOF
	test_cmp expect actual
'

test_expect_success "(7) restore pre-receive hook" '
	mv $bare/hooks/pre-receive $bare/hooks/pre-receive.fail &&
	mv $bare/hooks/pre-receive.ok $bare/hooks/pre-receive
'

test_expect_success "(7) new proc-receive and post-receive hooks (pass environment variables)" '
	## proc-receive hook
	mv $bare/hooks/proc-receive $bare/hooks/proc-receive.ok &&
	cat >$bare/hooks/proc-receive <<-EOF &&
	#!/bin/sh

	printf >&2 "execute: proc-receive hook\n"

	while read old new ref
	do
		printf >&2 ">> old: \$old, new: \$new, ref: \$ref.\n"
	done

	printf "GIT_VAR1=var1\n"
	printf "GIT_VAR2=var2\n"
	printf "AGIT_VAR1=foo\n"
	printf "AGIT_VAR2=bar\n"
	EOF
	chmod a+x $bare/hooks/proc-receive &&

	## post-receive hook
	mv $bare/hooks/post-receive $bare/hooks/post-receive.ok &&
	cat >$bare/hooks/post-receive <<-EOF &&
	#!/bin/sh

	printf >&2 "execute: post-receive hook\n"

	while read old new ref
	do
		printf >&2 ">> old: \$old, new: \$new, ref: \$ref.\n"
	done

	for k in GIT_VAR1 GIT_VAR2 AGIT_VAR1 AGIT_VAR2
	do
		if test -n "\$(eval echo \\"\\\$\$k\")"
		then
			printf >&2 ">> has env: \$k=\$(eval echo \\"\\\$\$k\").\n"
		fi
	done
	EOF
	chmod a+x $bare/hooks/post-receive
'

test_expect_success "(7) push one special ref (show environments)" '
	(
		cd work &&
		git push origin \
			HEAD:refs/for/master/my/topic
	) >out 2>&1 &&
	grep "^remote:" out | sed -e "s/  *\$//g" >actual &&
	cat >expect <<-EOF &&
	remote: execute: pre-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: execute: proc-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: execute: post-receive hook
	remote: >> old: $ZERO_OID, new: $B, ref: refs/for/master/my/topic.
	remote: >> has env: GIT_VAR1=var1.
	remote: >> has env: GIT_VAR2=var2.
	remote: >> has env: AGIT_VAR1=foo.
	remote: >> has env: AGIT_VAR2=bar.
	EOF
	test_cmp expect actual
'

test_done