#!/bin/bash
# SPDX-License-Identifier: MIT

set -e
set -o posix

SELF_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
TMPDIR=/tmp/cascading-pr-test
export LOOP_DELAY=5
mkdir -p $TMPDIR
source $SELF_DIR/../cascading-pr-lib.sh

function push_self() {
    forgejo-test-helper.sh push_self_action http://user1:admin1234@${options[host_port]} user1 cascading-pr vTest
}

function user_login() {
    local username=$1
    (
	export DOT=$TMPDIR/$username
	forgejo-curl.sh logout
	forgejo-curl.sh --user $username --password "${options[password]}" login ${options[url]}
    )
}

function user_curl() {
    local username=$1
    shift
    DOT=$TMPDIR/$username forgejo-curl.sh "$@"
}

function user_token() {
    local username=$1 name=$2

    curl -sS -f -H Content-Type:application/json --user "$username:${options[password]}" --data '{"name":"'$name'","scopes":["write:repository","write:issue","read:user"]}' ${options[url]}/api/v1/users/$username/tokens  | jq --raw-output .sha1 | tee $TMPDIR/$username/repo-token
}

function user_secret() {
    local username=$1 name=$2 token=$3

    user_curl $username api_json -X PUT --data '{"data":"'$token'"}' ${options[url]}/api/v1/user/actions/secrets/$name
}

function orgs_delete() {
    forgejo-curl.sh api_json ${options[url]}/api/v1/orgs | jq --raw-output '.[] | .name' | while read owner ; do
	forgejo-curl.sh api_json ${options[url]}/api/v1/orgs/$owner/repos | jq --raw-output '.[] | .name' | while read repo ; do
	    forgejo-curl.sh api_json -X DELETE ${options[url]}/api/v1/repos/$owner/$repo
	done
	forgejo-curl.sh api_json -X DELETE ${options[url]}/api/v1/orgs/$owner
    done
}

function user_create() {
    local username="$1" email="$2"
    forgejo-curl.sh api_json -X DELETE ${options[url]}/api/v1/admin/users/$username?purge=true >& /dev/null || true
    forgejo-curl.sh api_json --data '{"username":"'$username'","email":"'$email'","password":"'${options[password]}'","must_change_password":false}' ${options[url]}/api/v1/admin/users
    user_login $username
}

function close_pull_request() {
    local repo=$1

    forgejo-curl.sh api_json ${options[url]}/api/v1/repos/user1/${repo}/pulls | jq --raw-output '.[] | .number' | while read pr ; do
	forgejo-curl.sh api_json -X PATCH --data '{"state":"closed"}' ${options[url]}/api/v1/repos/user1/${repo}/issues/$pr
    done
}

function merge_pull_request() {
    local repo=$1

    forgejo-curl.sh api_json ${options[url]}/api/v1/repos/user1/${repo}/pulls | jq --raw-output '.[] | .number' | while read pr ; do
	forgejo-curl.sh api_json --data '{"Do":"merge"}' ${options[url]}/api/v1/repos/user1/${repo}/pulls/$pr/merge
    done
}

function has_cascade_pull_request() {
    pr_count=$(forgejo-curl.sh api_json ${options[url]}/api/v1/repos/user2/destinationrepo/pulls | jq '[ .[] | select(.state == "open") ] | length')
    test $pr_count -gt 0
}

function create_branch1() {
    local owner=$1 repo=$2 modify=$3

    forgejo-curl.sh api_json -X DELETE ${options[url]}/api/v1/repos/$owner/${repo}/branches/branch1 >& /dev/null || true
    forgejo-curl.sh api_json --data '{"new_branch_name":"branch1"}' ${options[url]}/api/v1/repos/$owner/${repo}/branches
    (
	cd $TMPDIR
	rm -fr ${repo}
	git clone -b branch1 http://user1:admin1234@${options[host_port]}/$owner/${repo}
	cd ${repo}
	echo CONTENT > README
	$modify
	git config user.email root@example.com
	git config user.name username
	git add .
	git commit -m 'update'
	git push origin branch1
	git rev-parse HEAD > ../${owner}-${repo}.sha
    )
}

function delete_pull_requests() {
    local owner=$1 repo=$2

    forgejo-curl.sh api_json ${options[url]}/api/v1/repos/$owner/${repo}/pulls | jq --raw-output '.[] | .number' | while read pr ; do
	forgejo-curl.sh api_json -X DELETE ${options[url]}/api/v1/repos/$owner/${repo}/issues/$pr
    done
}


function create_pull_request() {
    local baseowner=$1 headowner=$2 repo=$3

    local head
    if test $baseowner == $headowner; then
	head=branch1
    else
	head=$headowner:branch1
    fi
    cat > $TMPDIR/data <<EOF
{"title":"PR","base":"main","head":"$head"}
EOF
    forgejo-curl.sh api_json --data @$TMPDIR/data ${options[url]}/api/v1/repos/$baseowner/${repo}/pulls
}

function create_pull_request_case1() {
    local baseowner=$1 headowner=$2 repo=$3

    delete_pull_requests $baseowner $repo
    create_branch1 $headowner $repo
    create_pull_request $baseowner $headowner $repo
}

function finalize_options() {
    if test -f forgejo-ip; then
	: ${options[host_port]:=$(cat forgejo-ip):3000}
    fi
    options[url]=http://${options[host_port]}
    if test -f forgejo-token; then
	: ${options[token]:=$(cat forgejo-token)}
    fi
    options[password]=admin1234
}

function unit_retry_fail() {
    local file=$1
    local value=$(cat $file)
    expr $value - 1 > $file
    echo RESULT
    if test $value -gt 0 ; then
	return 1
    else
	return 0
    fi
}

function unit_retry() {
    local two=$TMPDIR/unit_retry_two

    rm -f $TMPDIR/retry.*

    #
    # Succeeds after two tries
    #
    echo 2 > $TMPDIR/unit_retry_two
    RETRY_DELAYS='1 1 1 1' retry unit_retry_fail $two > $TMPDIR/retry.result 2> $TMPDIR/retry.log
    test "$(cat $TMPDIR/retry.result)" = "RESULT"
    test "$(grep -c 'waiting 1 unit_retry_fail' $TMPDIR/retry.log)" = 2

    #
    # Succeeds immediately
    #
    RETRY_DELAYS='1' retry unit_retry_fail $two > $TMPDIR/retry.result 2> $TMPDIR/retry.log
    test "$(cat $TMPDIR/retry.result)" = "RESULT"
    test "$(grep -c 'waiting 1 unit_retry_fail' $TMPDIR/retry.log)" = 0

    #
    # Verify the output is only the output of the last run and is not polluted by
    # unrelated output
    #
    echo 2 > $TMPDIR/unit_retry_two
    test "$(RETRY_DELAYS='1 1 1' retry unit_retry_fail $two)" = "RESULT"

    #
    # Fails after one try
    #
    echo 2 > $TMPDIR/unit_retry_two
    if RETRY_DELAYS='1' retry unit_retry_fail $two |& tee $TMPDIR/retry.log ; then
	return 1
    fi
    grep --quiet 'retry failed' $TMPDIR/retry.log
}

function fixture() {
    local origin=$1
    local destination=$2

    orgs_delete

    user_create user3 user3@example.com
    user_curl user3 api_json --data '{"username":"destination-fork"}' ${options[url]}/api/v1/orgs

    user_create user2 user2@example.com
    forgejo-test-helper.sh push tests/${destination} http://user2:admin1234@${options[host_port]} user2 ${destination}

    user_create user1 user1@example.com
    forgejo-test-helper.sh push tests/${origin} http://user1:admin1234@${options[host_port]} user1 ${origin} cascading-pr
    user_curl user1 api_json --data '{"username":"origin-fork"}' ${options[url]}/api/v1/orgs
    user_curl user1 api_json --data '{"organization":"origin-fork"}' ${options[url]}/api/v1/repos/user1/${origin}/forks
    user_secret user1 ORIGIN_TOKEN $(user_token user1 ORIGIN_TOKEN)

    push_self
}

function no_change_no_cascade_pr() {
    fixture originrepo-do-nothing destinationrepo
    user_secret user1 DESTINATION_TOKEN $(user_token user2 DESTINATION_TOKEN)

    create_pull_request_case1 user1 user1 originrepo-do-nothing
    wait_success ${options[url]}/api/v1/repos/user1/originrepo-do-nothing $(cat $TMPDIR/user1-originrepo-do-nothing.sha)
    ! has_cascade_pull_request
}

function create_and_close() {
    fixture originrepo destinationrepo
    user_secret user1 DESTINATION_TOKEN $(user_token user2 DESTINATION_TOKEN)

    create_pull_request_case1 user1 user1 originrepo
    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/user1-originrepo.sha)
    has_cascade_pull_request
    close_pull_request originrepo
    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/user1-originrepo.sha)
}

function taint_update() {
    if ! test -f upgraded ; then
	echo upgraded file not found
	return 1
    fi
    echo 'TAINTED' > upgraded
}

function create_from_origin_fork_and_close() {
    fixture originrepo destinationrepo
    user_secret user1 DESTINATION_TOKEN $(user_token user2 DESTINATION_TOKEN)

    delete_pull_requests user1 originrepo
    create_branch1 origin-fork originrepo taint_update
    create_pull_request user1 origin-fork originrepo

    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/origin-fork-originrepo.sha)
    has_cascade_pull_request
    close_pull_request originrepo
    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/origin-fork-originrepo.sha)
}

function create_and_merge() {
    fixture originrepo destinationrepo
    user_secret user1 DESTINATION_TOKEN $(user_token user2 DESTINATION_TOKEN)

    create_pull_request_case1 user1 user1 originrepo

    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/user1-originrepo.sha)
    has_cascade_pull_request
    merge_pull_request originrepo
    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/user1-originrepo.sha)
    has_cascade_pull_request
}

function create_in_destination_fork_and_close() {
    fixture origin-fork-destination destinationrepo
    user_secret user1 DESTINATION_TOKEN $(user_token user3 DESTINATION_TOKEN)

    create_pull_request_case1 user1 user1 origin-fork-destination

    wait_success ${options[url]}/api/v1/repos/user1/origin-fork-destination $(cat $TMPDIR/user1-origin-fork-destination.sha)
    has_cascade_pull_request
    close_pull_request origin-fork-destination
    wait_success ${options[url]}/api/v1/repos/user1/origin-fork-destination $(cat $TMPDIR/user1-origin-fork-destination.sha)
}

function create_and_merge_close() {
    fixture originrepo-close-merge destinationrepo
    user_secret user1 DESTINATION_TOKEN $(user_token user2 DESTINATION_TOKEN)

    create_pull_request_case1 user1 user1 originrepo-close-merge

    wait_success ${options[url]}/api/v1/repos/user1/originrepo-close-merge $(cat $TMPDIR/user1-originrepo-close-merge.sha)
    has_cascade_pull_request
    merge_pull_request originrepo-close-merge
    wait_success ${options[url]}/api/v1/repos/user1/originrepo-close-merge $(cat $TMPDIR/user1-originrepo-close-merge.sha)
    ! has_cascade_pull_request
}

function integration() {
    for t in create_in_destination_fork_and_close no_change_no_cascade_pr create_from_origin_fork_and_close create_and_close create_and_merge create_and_merge_close; do
	log_info "running $t"
	$t
    done
}

function unit() {
    unit_retry
}

function run() {
    unit
    integration
}

function main() {
    local command=run

    while true; do
	case "$1" in
	    --verbose)
		shift
		verbose
		;;
	    --debug)
		shift
		debug
		;;
	    --host_port)
		shift
		options[host_port]=$1
		shift
		;;
	    --url)
		shift
		options[url]=$1
		shift
		;;
	    --token)
		shift
		options[token]=$1
		shift
		;;
	    *)
		finalize_options
		"${1:-run}"
		return 0
		;;
	esac
    done
}

dependencies

${MAIN:-main} "${@}"