From 474fabf287cca3c65175bf05a75ae850326d9c36 Mon Sep 17 00:00:00 2001
From: Luke Shumaker <LukeShu@sbcglobal.net>
Date: Thu, 25 Apr 2013 19:13:54 -0400
Subject: lib/common.sh: Make setup_workdir()/cleanup() safe for programs to
 not use

---
 lib/common.sh | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

(limited to 'lib')

diff --git a/lib/common.sh b/lib/common.sh
index 932799e..d6fbe7c 100644
--- a/lib/common.sh
+++ b/lib/common.sh
@@ -57,13 +57,19 @@ stat_done() {
 	printf "${BOLD}done${ALL_OFF}\n" >&2
 }
 
+_setup_workdir=false
 setup_workdir() {
 	[[ -z $WORKDIR ]] && WORKDIR=$(mktemp -d --tmpdir "${0##*/}.XXXXXXXXXX")
+	_setup_workdir=true
+	trap 'trap_abort' INT QUIT TERM HUP
+	trap 'trap_exit' EXIT
 }
 
 cleanup() {
-	[[ -n $WORKDIR ]] && rm -rf "$WORKDIR"
-	[[ $1 ]] && exit $1
+	if [[ -n $WORKDIR ]] && $_setup_workdir; then
+		rm -rf "$WORKDIR"
+	fi
+	[[ -n $1 ]] && exit $1
 }
 
 abort() {
@@ -86,9 +92,6 @@ die() {
 	cleanup 1
 }
 
-trap 'trap_abort' INT QUIT TERM HUP
-trap 'trap_exit' EXIT
-
 ##
 #  usage : in_array( $needle, $haystack )
 # return : 0 - found
-- 
cgit v1.2.3-70-g09d2