Move mountfile copying code

Signed-off-by: Aaron Griffin <aaronmgriffin@gmail.com>

1 files changed, 6 insertions, 3 deletions

diff --git a/archiso/mkarchiso b/archiso/mkarchiso
index 2fcc361..857d588 100755
--- a/archiso/mkarchiso
+++ b/archiso/mkarchiso
@@ -143,9 +143,6 @@ command_install () {
 
     # pacman DBs are big, delete all sync dbs
     rm -rf "${work_dir}/var/lib/pacman/sync"
-
-    #TODO test for existance of images in work_dir
-    cp "${MOUNTFILE}" "${work_dir}/mounts"
 }
 
 # command_squash path image
@@ -189,6 +186,12 @@ command_image () {
         exit 1
     fi
 
+    if [ ! -e "${MOUNTFILE}" ]; then
+        echo "error: mount file '${MOUNTFILE}' does not exist, aborting."
+        exit 1
+    fi
+    cp "${MOUNTFILE}" "${work_dir}/mounts"
+
     kernelver=$(_kversion)
     basedir=${work_dir}
     [ "${work_dir:0:1}" != "/" ] && basedir="$(pwd)/${work_dir}"