#!/bin/bash

# SPDX-FileCopyrightText: 2021-2022 Ole Tange, http://ole.tange.dk and Free Software and Foundation, Inc.
#
# SPDX-License-Identifier: GPL-3.0-or-later

unset run_test
unset run_once

# SSH only allowed to localhost/lo

par_z_sshloginfile() {
    echo '### --slf with mIxEd cAsE'
    tmp=$(mktemp)
    (
	echo 2/bash@LO
	echo 3/parallel@Lo:22
	echo 4/ksh@lO:ssh
    ) > "$tmp"
    seq 9 | parallel --slf "$tmp" 'whoami;sleep 1;echo' | sort
    rm -f "$tmp"
}

par__test_different_rsync_versions() {
    echo '### different versions of rsync need fixups'
    echo '### no output is good'
    doit() {
	rm -f 'a`b`c\<d\$e\{#\}g\"h\ i'$2 'a`b`c\<d\$e\{#\}g\"h\ i'$2.out
	touch 'a`b`c\<d\$e\{#\}g\"h\ i'$2
	TMPDIR=/tmp tmp=$(mktemp -d)
	(
	    echo "#!/bin/bash"
	    echo $1' "$@"'
	) > "$tmp"/rsync
	chmod +x "$tmp"/rsync
	PATH="$tmp":"$PATH"
	parallel --trc {}.out -S sh@lo cp {} {}.out ::: 'a`b`c\<d\$e\{#\}g\"h\ i'$2
	rm 'a`b`c\<d\$e\{#\}g\"h\ i'$2 'a`b`c\<d\$e\{#\}g\"h\ i'$2.out
	rm -rf "$tmp"
    }
    export -f doit
    stdout parallel --tagstring {/} -k doit {} {/} ::: /usr/local/bin/rsync-v*
}

par_--nonall_results() {
    echo '### --results --onall'
    tmp="$TMPDIR"/onall
    mkdir -p "$tmp"
    parallel --results "$tmp"/noslash --onall -Scsh@lo,sh@lo ::: id pwd
    parallel --results "$tmp"/slash/ --onall -Scsh@lo,sh@lo ::: id pwd
    parallel --results "$tmp"/rplslash/{}/ --onall -Scsh@lo,sh@lo ::: id pwd
    parallel --results "$tmp"/rplnoslash/{} --onall -Scsh@lo,sh@lo ::: id pwd
    parallel --results "$tmp"/rpl1slash/{1}/ --onall -Scsh@lo,sh@lo ::: id pwd
    parallel --results "$tmp"/rpl1noslash/{1} --onall -Scsh@lo,sh@lo ::: id pwd
    find "$tmp" -print0 | replace_tmpdir
    rm -r "$tmp"    
    echo '### --results --nonall'
    tmp="$TMPDIR"/nonall
    mkdir -p "$tmp"
    parallel --results "$tmp"/noslash --nonall -Scsh@lo,sh@lo pwd
    parallel --results "$tmp"/slash/ --nonall -Scsh@lo,sh@lo pwd
    parallel --results "$tmp"/rplslash/{}/ --nonall -Scsh@lo,sh@lo pwd
    parallel --results "$tmp"/rplnoslash/{} --nonall -Scsh@lo,sh@lo pwd
    parallel --results "$tmp"/rpl1slash/{1}/ --nonall -Scsh@lo,sh@lo pwd
    parallel --results "$tmp"/rpl1noslash/{1} --nonall -Scsh@lo,sh@lo pwd
    find "$tmp" -print0 | replace_tmpdir
    rm -r "$tmp"    
}

par_env_underscore() {
    echo '### --env _'
    echo ignored_var >> ~/.parallel/ignored_vars
    unset $(compgen -A function | grep par_)
    ignored_var="should not be copied"
    export ignored_var
    fUbAr="OK FUBAR" parallel -S parallel@lo --env _ echo '$fUbAr $ignored_var' ::: test
    echo 'In csh this may fail with ignored_var: Undefined variable.'
    fUbAr="OK FUBAR" parallel -S csh@lo --env _ echo '$fUbAr $ignored_var' ::: test 2>&1

    echo '### --env _ with explicit mentioning of normally ignored var $ignored_var'
    ignored_var="should be copied"
    fUbAr="OK FUBAR" parallel -S parallel@lo --env ignored_var,_ echo '$fUbAr $ignored_var' ::: test
    fUbAr="OK FUBAR" parallel -S csh@lo --env ignored_var,_ echo '$fUbAr $ignored_var' ::: test 2>&1
}    

par_warn_when_exporting_func() {
    echo 'bug #40137: SHELL not bash: Warning when exporting funcs'
    myrun() {
	. <(printf 'myfunc() {\necho Function run: $1\n}')
	export -f myfunc
	echo "Run function in $1"
	PARALLEL_SHELL=$1 parallel --env myfunc -S lo myfunc ::: OK
    }
    export -f myrun
    parallel -k --tag myrun ::: /bin/{sh,bash} /usr/bin/{ash,csh,dash,ksh,tcsh,zsh}
}

par_exporting_in_zsh() {
    echo '### zsh'
    
    echo 'env in zsh'
    echo 'Normal variable export'
    export B=\'"  Var with quote"
    PARALLEL_SHELL=/usr/bin/zsh parallel --env B echo '$B' ::: OK

    echo 'Function export as variable'
    export myfuncvar="() { echo myfuncvar as var \$*; }"
    PARALLEL_SHELL=/usr/bin/zsh parallel --env myfuncvar myfuncvar ::: OK

    echo 'Function export as function'
    myfunc() { echo myfunc ran $*; }
    export -f myfunc
    PARALLEL_SHELL=/usr/bin/zsh parallel --env myfunc myfunc ::: OK

    ssh zsh@lo 'fun="() { echo function from zsh to zsh \$*; }"; 
              export fun; 
              parallel --env fun fun ::: OK'

    ssh zsh@lo 'fun="() { echo function from zsh to bash \$*; }"; 
              export fun; 
              parallel -S parallel@lo --env fun fun ::: OK'
}

par_bigvar_csh() {
    echo '### csh'
    echo "3 big vars run remotely - length(base64) > 1000"
    stdout ssh csh@lo 'setenv A `seq 200|xargs`; 
                     setenv B `seq 200 -1 1|xargs`; 
                     setenv C `seq 300 -2 1|xargs`; 
                     parallel -Scsh@lo --env A,B,C -k echo \$\{\}\|wc ::: A B C'
    echo '### csh2'
    echo "3 big vars run locally"
    stdout ssh csh@lo 'setenv A `seq 200|xargs`; 
                     setenv B `seq 200 -1 1|xargs`; 
                     setenv C `seq 300 -2 1|xargs`; 
                     parallel --env A,B,C -k echo \$\{\}\|wc ::: A B C'
}

par_bigvar_rc() {
    echo '### rc'
    echo "3 big vars run remotely - length(base64) > 1000"
    stdout ssh rc@lo 'A=`{seq 200}; 
                    B=`{seq 200 -1 1}; 
                    C=`{seq 300 -2 1}; 
                    parallel -Src@lo --env A,B,C -k echo '"'"'${}|wc'"'"' ::: A B C'

    echo '### rc2'
    echo "3 big vars run locally"
    stdout ssh rc@lo 'A=`{seq 200}; 
                    B=`{seq 200 -1 1}; 
                    C=`{seq 300 -2 1}; 
                    parallel --env A,B,C -k echo '"'"'${}|wc'"'"' ::: A B C'
}

par_--tmux_different_shells() {
    echo '### Test tmux works on different shells'
    short_TMPDIR() {
	# TMPDIR must be short for -M                                                         
	export TMPDIR=/tmp/ssh/'                                                              
`touch /tmp/tripwire`                                                                     
'
	TMPDIR=/tmp
	mkdir -p "$TMPDIR"
    }
    short_TMPDIR
    (
	stdout parallel -Scsh@lo,tcsh@lo,parallel@lo,zsh@lo --tmux echo ::: 1 2 3 4; echo $?
	stdout parallel -Scsh@lo,tcsh@lo,parallel@lo,zsh@lo --tmux false ::: 1 2 3 4; echo $?

	export PARTMUX='parallel -Scsh@lo,tcsh@lo,parallel@lo,zsh@lo --tmux '; 
	stdout ssh zsh@lo      "$PARTMUX" 'true  ::: 1 2 3 4; echo $status' 
	stdout ssh zsh@lo      "$PARTMUX" 'false ::: 1 2 3 4; echo $status' 
	stdout ssh parallel@lo "$PARTMUX" 'true  ::: 1 2 3 4; echo $?'      
	stdout ssh parallel@lo "$PARTMUX" 'false ::: 1 2 3 4; echo $?'      
	stdout ssh tcsh@lo     "$PARTMUX" 'true  ::: 1 2 3 4; echo $status' 
	stdout ssh tcsh@lo     "$PARTMUX" 'false ::: 1 2 3 4; echo $status' 
	echo "# command is currently too long for csh. Maybe it can be fixed?"; 
	stdout ssh csh@lo      "$PARTMUX" 'true  ::: 1 2 3 4; echo $status'
	stdout ssh csh@lo      "$PARTMUX" 'false ::: 1 2 3 4; echo $status'
    ) | replace_tmpdir | perl -pe 's/tms...../tmsXXXXX/g'
}

par_--tmux_length() {
    echo '### works'
    short_TMPDIR() {
	# TMPDIR must be short for -M                                                         
	export TMPDIR=/tmp/ssh/'                                                              
`touch /tmp/tripwire`                                                                     
'
	TMPDIR=/tmp
	mkdir -p "$TMPDIR"
    }
    short_TMPDIR
    (
	stdout parallel -Sparallel@lo --tmux echo ::: \\\\\\\"\\\\\\\"\\\;\@
        stdout parallel -Sparallel@lo --tmux echo ::: xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx

	echo '### These blocked due to length'
	stdout parallel -Slo      --tmux echo ::: \\\\\\\"\\\\\\\"\\\;\@
	stdout parallel -Scsh@lo  --tmux echo ::: \\\\\\\"\\\\\\\"\\\;\@
	stdout parallel -Stcsh@lo --tmux echo ::: \\\\\\\"\\\\\\\"\\\;\@
	stdout parallel -Szsh@lo  --tmux echo ::: \\\\\\\"\\\\\\\"\\\;\@
	stdout parallel -Scsh@lo  --tmux echo ::: 111111111111111111111111111111111111111111111111111111111
     ) | replace_tmpdir |
	perl -pe 's:tms.....:tmsXXXXX:'
}

par_transfer_return_multiple_inputs() {
    echo '### bug #43746: --transfer and --return of multiple inputs {1} and {2}'
    echo '### and:'
    echo '### bug #44371: --trc with csh complains'
    cd /tmp; echo file1 output line 1 > file1; echo file2 output line 3 > file2
    parallel -Scsh@lo --transferfile {1} --transferfile {2} --trc {1}.a --trc {2}.b \
	     '(cat {1}; echo A {1} output line 2) > {1}.a; (cat {2};echo B {2} output line 4) > {2}.b' ::: file1 ::: file2
    cat file1.a file2.b
    rm /tmp/file1 /tmp/file2 /tmp/file1.a /tmp/file2.b
}

par_z_csh_nice() {
    echo '### bug #44143: csh and nice'
    parallel --nice 1 -S csh@lo setenv B {}\; echo '$B' ::: OK
}

par_z_multiple_hosts_repeat_arg() {
    echo '### bug #45575: -m and multiple hosts repeats first args'
    seq 1 3 | parallel -X -S 2/lo,2/: -k echo 
}

export -f $(compgen -A function | grep par_)
compgen -A function | grep par_ | LC_ALL=C sort |
    parallel --timeout 3000% -j6 --tag -k --joblog /tmp/jl-`basename $0` '{} 2>&1' |
    perl -pe 's:/usr/bin:/bin:g;'