X-Git-Url: https://git.treefish.org/~alex/backmeupscotty.git/blobdiff_plain/1aa0158b6b4d4aaca7f413d76aec814a206a212b..d0707b3df434a0028ac2e588a8bb7a6a8714d764:/backmeupscotty

diff --git a/backmeupscotty b/backmeupscotty
index ed41ae2..257c192 100644
--- a/backmeupscotty
+++ b/backmeupscotty
@@ -18,22 +18,6 @@ function scottyerror {
     echo $(upperme): $@ >&2 
 }
 
-if [ $(pidof -x $(basename $0) | wc -w) -gt 2 ]; then 
-    scottyerror Another instance of $(basename $0) is already running!
-    exit 1
-fi
-
-while getopts "qn:" opt; do
-    case $opt in
-	q)
-	    exec > /dev/null
-	    ;;
-	n)
-	    BACKUP_RUNEVERYNTHDAY=$OPTARG
-	    ;;
-    esac
-done
-
 function grepbackups {
     ssh $REMOTE_HOST "ls $REMOTE_DIR" | grep -E '[0-9]+-[0-9]+'
 }
@@ -116,21 +100,30 @@ function scottysync {
     done
 }
 
+function deleteLock {
+    if ! rmdir /tmp/$(basename $0).lock; then
+	scottyerror "Could not delete lockfile /tmp/$(basename $0).lock!"
+    fi
+}
+
 function prepare {
     scottyinfo "Preparing for sync"
 }
 
-function cleanup_abrt {
-    scottyinfo "No cleanup_abrt function was defined."
+function cleanup_abort {
+    scottyinfo "No cleanup_abort function was defined."
 }
 
-function _cleanup_abrt {
+function _cleanup_abort {
     scottyerror "Caught exit signal! Cleaning up..."
-    cleanup_abrt
+
+    cleanup_abort
+
     if [ $(jobs -p) ]; then
 	scottyerror TERMinating remaining child processes...
 	kill $(jobs -p)
     fi
+    deleteLock
     exit
 }
 
@@ -138,29 +131,56 @@ function prepare {
     scottyinfo "No prepare function was defined."
 }
 
-function _prepare {
-    prepare
-}
-
 function cleanup_normal {
     scottyinfo "No cleanup_normal function was defined."
 }
 
-function _cleanup_normal {
-    cleanup_normal
+function printhelp {
+    cat <<EOF
+Usage: $(basename $0) [OPTION]...
+
+Recognized options:
+  -q   Only output errors
+  -n   Run only on nth day
+  -h   Print out this help
+EOF
 }
 
 function backmeupscotty {
+    while getopts "qn:h" opt; do
+	case $opt in
+	    q)
+		exec > /dev/null
+		;;
+	    n)
+		BACKUP_RUNEVERYNTHDAY=$OPTARG
+		;;
+	    h)
+		printhelp
+                exit 0
+                ;;
+	esac
+    done
+    
     if ! isIncompleteOrNthDay; then
 	scottyinfo "This is not the nth day and no incomplete backup exists."
 	exit 0
     fi
 
-    trap _cleanup_abrt EXIT
+    trap _cleanup_abort EXIT
 
-    _prepare
+    prepare
     scottysync
-    _cleanup_normal
+    cleanup_normal
+
+    trap deleteLock EXIT
 
-    trap - EXIT
+    exit 0
 }
+
+if ! mkdir /tmp/$(basename $0).lock; then
+    scottyerror "Another instance of $(basename $0) is still running!"
+    exit 1
+else
+    trap deleteLock EXIT
+fi