Expected extensions

src/freedombone-image

@@ -163,6 +163,10 @@ rm $CURR_DIR/${PROJECT_NAME}*.sig
 rm $CURR_DIR/${PROJECT_NAME}*.vdi
 rm $CURR_DIR/${PROJECT_NAME}*.qcow2
 
+if [ -f ${CURR_DIR}/${PROJECT_NAME}_login_credentials.txt ]; then
+    rm ${CURR_DIR}/${PROJECT_NAME}_login_credentials.txt
+fi
+
 if [ -d $BUILD_DIR ]; then
     rm -rf $BUILD_DIR
 fi
@@ -192,8 +196,16 @@ make $IMAGE_TYPE \
     SSH_PUBKEY="$SSH_PUBKEY" \
     GENERIC_IMAGE="$GENERIC_IMAGE"
 
+EXPECTED_EXTENSION='img'
+if [[ $IMAGE_TYPE == "qemu"* ]]; then
+    EXPECTED_EXTENSION='qcow2'
+fi
+if [[ $IMAGE_TYPE == "virtualbox"* ]]; then
+    EXPECTED_EXTENSION='vdi'
+fi
+
 shopt -s nullglob
-imgfiles=(build/${PROJECT_NAME}*.img)
+imgfiles=(build/${PROJECT_NAME}*.${EXPECTED_EXTENSION})
 if [ ${#imgfiles[@]} -eq 0 ]; then
     echo 'Image was not created'
     rm -rf $BUILD_DIR