Move to 'source' and own return value variable

steps/10-wipe.step

@@ -5,16 +5,16 @@ echo -n "~> Wipe block device? (Y/n) "
 read answer
 
 if [ "$answer" == "n" ] || [ "$answer" == "N" ]; then
-	exit 0
-fi
-
-echo -n "~> You sure? (y/N) "
-read answer
-
-if [ "$answer" == "y" ] || [ "$answer" == "Y" ]; then
-	echo "~> OK. Be patient now, that could take some time..."
-	dd if=/dev/zero of=$blockdevice status=progress
-	exit 0
+	amf_return=0
 else
-	exit 1
+	echo -n "~> You sure? (y/N) "
+	read answer
+
+	if [ "$answer" == "y" ] || [ "$answer" == "Y" ]; then
+		echo "~> OK. Be patient now, that could take some time..."
+		dd if=/dev/zero of=$blockdevice status=progress
+		amf_return=0
+	else
+		amf_return=1
+	fi
 fi

steps/20-partition.step

@@ -8,8 +8,8 @@ echo -e "g\nn\n1\n2048\n+300M\nn\n2\n\n\np\nw" | fdisk $blockdevice
 
 if [ "$?" -eq 0 ]; then
 	echo "~> Seems to have worked. Yay!"
-	exit 0
+	amf_return=0
 else
 	echo "~> Failed. :("
-	exit 1
+	amf_return=1
 fi

steps/60-user.step

@@ -7,12 +7,12 @@ echo -n "~> Do you want to create an user account? (Y/n) "
 read answer
 
 if [ "$answer" == "n" ] || [ "$answer" == "N" ]; then
-	exit 0
+	amf_return=0
 else
 	echo -n "   Name of your new user? "
 	read name
 	export username=$name
 	arch-chroot /mnt useradd -m $name
 	arch-chroot /mnt passwd $name
-	exit 0
+	amf_return=0
 fi