Skip to content

Instantly share code, notes, and snippets.

@tvlooy
Last active February 4, 2024 04:20
Show Gist options
  • Star 61 You must be signed in to star a gist
  • Fork 8 You must be signed in to fork a gist
  • Save tvlooy/cbfbdb111a4ebad8b93e to your computer and use it in GitHub Desktop.
Save tvlooy/cbfbdb111a4ebad8b93e to your computer and use it in GitHub Desktop.
Bash test: get the directory of a script
#!/bin/bash
function test {
MESSAGE=$1
RECEIVED=$2
EXPECTED=$3
if [ "$RECEIVED" = "$EXPECTED" ]; then
echo -e "\033[32m✔︎ Tested $MESSAGE"
else
echo -e "\033[31m✘ Tested $MESSAGE"
echo -e " Received: $RECEIVED"
echo -e " Expected: $EXPECTED"
fi
echo -en "\033[0m"
}
function testSuite {
test 'absolute call' `bash /tmp/1234/test.sh` /tmp/1234
test 'via symlinked dir' `bash /tmp/current/test.sh` /tmp/1234
test 'via symlinked file' `bash /tmp/test.sh` /tmp/1234
test 'via multiple symlinked dirs' `bash /tmp/current/loop/test.sh` /tmp/1234
pushd /tmp >/dev/null
test 'relative call' `bash 1234/test.sh` /tmp/1234
popd >/dev/null
test 'with space in dir' `bash /tmp/12\ 34/test.sh` /tmp/1234
test 'with space in file' `bash /tmp/1234/te\ st.sh` /tmp/1234
echo
}
function setup {
DIR=/tmp/1234
FILE=test.sh
if [ -e $DIR ]; then rm -rf $DIR; fi; mkdir $DIR
if [ -f $DIR/$FILE ]; then rm -rf $DIR/$FILE; fi; touch $DIR/$FILE
if [ -f /tmp/$FILE ]; then rm /tmp/$FILE; fi; ln -s $DIR/$FILE /tmp
if [ -f /tmp/current ]; then rm /tmp/current; fi; ln -s $DIR /tmp/current
if [ -f /tmp/current/loop ]; then rm /tmp/current/loop; fi; ln -s $DIR /tmp/current/loop
DIR2="/tmp/12 34"
FILE2="te st.sh"
if [ -e "$DIR2" ]; then rm -rf "$DIR2"; fi; mkdir "$DIR2"
if [ -f "$DIR/$FILE2" ]; then rm -rf "$DIR/$FILE2"; fi; ln -s $DIR/$FILE "$DIR/$FILE2"
if [ -f "$DIR2/$FILE" ]; then rm -rf "$DIR2/$FILE"; fi; ln -s $DIR/$FILE "$DIR2/$FILE"
if [ -f "$DIR2/$FILE2" ]; then rm -rf "$DIR2/$FILE2"; fi; ln -s $DIR/$FILE "$DIR2/$FILE2"
}
function test1 {
echo 'Test 1: via dirname'
cat <<- EOF >/tmp/1234/test.sh
echo \`dirname \$0\`
EOF
testSuite
}
function test2 {
echo 'Test 2: via pwd'
cat <<- EOF >/tmp/1234/test.sh
CACHE_DIR=\$( cd "\$( dirname "\${BASH_SOURCE[0]}" )" && pwd )
echo \$CACHE_DIR
EOF
testSuite
}
function test3 {
echo 'Test 3: overcomplicated stackoverflow solution'
cat <<- EOF >/tmp/1234/test.sh
SOURCE="\${BASH_SOURCE[0]}"
while [ -h "\$SOURCE" ]; do
DIR="\$( cd -P "\$( dirname "\$SOURCE" )" && pwd )"
SOURCE="\$(readlink "\$SOURCE")"
[[ \$SOURCE != /* ]] && SOURCE="\$DIR/\$SOURCE"
done
DIR="\$( cd -P "\$( dirname "\$SOURCE" )" && pwd )"
echo \$DIR
EOF
testSuite
}
function test4 {
echo 'Test 4: via readlink'
cat <<- EOF >/tmp/1234/test.sh
echo \`dirname \$(readlink -f \$0)\`
EOF
testSuite
}
function test5 {
echo 'Test 5: via readlink with space'
cat <<- EOF >/tmp/1234/test.sh
echo \`dirname \$(readlink -f "\$0")\`
EOF
testSuite
}
echo
setup
if [ "$1" != "" ]; then
$1
else
test1
test2
test3
test4
test5
fi
@hasufell
Copy link

hasufell commented Oct 15, 2018

No portable POSIX version here. This should do:

function test6 {
    echo 'Test 6: posix compliant'
    cat <<- EOF >/tmp/1234/test.sh
	SOURCE="\$0"

	script_dir() {
		mysource=\${SOURCE}

		while [ -h "\${mysource}" ]; do
			DIR="\$( cd -P "\$( dirname "\${mysource}" )" > /dev/null && pwd )"
			mysource="\$(readlink "\${mysource}")"
			[ "\${mysource%\${mysource#?}}"x != '/x' ] && mysource="\${DIR}/\${mysource}"
		done
		DIR="\$( cd -P "\$( dirname "\${mysource}" )" > /dev/null && pwd )"
		echo "\${DIR}"

		unset mysource DIR
	}
	echo \$(script_dir)
	EOF
    testSuite
}

@gkop
Copy link

gkop commented Apr 30, 2019

greadlink -f unfortunately doesn't work effectively when sourceing the script on Mac :(

@ptc-mrucci
Copy link

New fork https://gist.github.com/ptc-mrucci/61772387878ed53a6c717d51a21d9371 includes:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment