diff options
| -rwxr-xr-x | share/hydra/bootless | 79 |
1 files changed, 63 insertions, 16 deletions
diff --git a/share/hydra/bootless b/share/hydra/bootless index 511b317..cc34e1f 100755 --- a/share/hydra/bootless +++ b/share/hydra/bootless @@ -77,20 +77,9 @@ EOF usbdevice=`echo ${device} | sed -e s/[0-9]\$//g` # Issue a warning - cat <<EOF - -****************** -* ATTENTION!!! * -****************** - -If you continue, all data in device "${device}" will be destroyed! - -EOF - - echo -n "Are you sure you want to continue? Type uppercase \"YES\": " - read go + hydra_bootless_warning ${device} - if [ "${go}" != "YES" ]; then + if [ "$?" != "0" ]; then echo "Aborting..." exit 1 fi @@ -313,18 +302,48 @@ function hydra_bootless_device { fi } +function hydra_bootless_warning { + local device=$1 + local go + + echo "******************" + echo "* ATTENTION!!! *" + echo "******************" + echo "" + echo "If you continue, all data in device \"${device}\" will be destroyed!" + echo "" + echo -n "Are you sure you want to continue? Type uppercase \"YES\": " + read go + + if [ "${go}" != "YES" ]; then + false + else + true + fi +} + # Generate a bootless disk image -# TODO: optionally copy to removable media function hydra_bootless_image { local output="$1" + local device="$2" - if [ -z "$output" ]; then + # Fix parameters + if echo ${output} | grep -q "/dev/"; then + output="bootless.img" + device="$1" + elif [ -z "$output" ]; then output="bootless.img" fi # Set folder hydra_bootless_folder + # Set sudo config + local sudo + if [ "`whoami`" != 'root' ]; then + sudo="sudo" + fi + # Copy data tmpdir=`mktemp -d` $sudo rsync -Cav ${BOOTLESS_DIR}/ ${tmpdir}/boot/ @@ -332,9 +351,37 @@ function hydra_bootless_image { # Make rescue disk grub-mkrescue -o ${output} ${tmpdir} + # Optionally copy to removable media + if [ ! -z "$device" ]; then + # Issue a warning + hydra_bootless_warning ${device} + + if [ "$?" != "0" ]; then + echo "Skipping copy of ${output} to ${device}..." + else + # Check if device is mounted + if [ "`mount | grep ${device}`" != "" ]; then + echo "Error: device \"${device}\" is mounted." + echo "Skipping copy of ${output} to ${device}..." + else + $sudo dd if=${output} of=${device} + fi + fi + + fi + # Cleanup rm -rf ${tmpdir} - echo "Image saved at ${output}" + + # Finish + if [ -z "$device" ]; then + echo "Image saved at ${output}" + else + echo "Removing image ${output}..." + rm ${output} + fi + + echo "Done." } # Usage |