1 files changed, 8 insertions, 1 deletions
diff --git a/lock.sh b/lock.sh
index 6a22392..5a7ef13 100755
--- a/lock.sh
+++ b/lock.sh
@@ -197,8 +197,15 @@ case "$1" in
 			mkdir -p "$folder"
 		fi
 
+		# get random file in dir if passed argument is a dir
+		user_input=$2
+		if [ -d $user_input ]; then
+			user_input=($user_input/*)
+			user_input=${user_input[RANDOM % ${#user_input[@]}]}
+		fi
+		
 		# get user image
-		cp "$2" "$user_image"
+		cp "$user_input" "$user_image"
 		if [ ! -f $user_image ]; then
 			echo "Please specify the path to the image you would like to use"
 			exit 1