#!/bin/bash
# Simona Pisano - 2018-11-18 -
# simona-scripts
# Libertamente utilizzabile sotto GPL v3
#srun-mount-img() {

#to-do add input parameters

[[ $1 == "" ]] && exit 0 #TO-DO attivare invece la richiesta dei parametri?

#color	Foreground	Background
#black	30	40
#red	31	41
#green	32	42
#yellow	33	43
#blue	34	44
#magenta	35	45
#cyan	36	46
#white	37	47
EchoReset="\e[0m"
echo -e -n "$EchoReset"
OK="\E[32;40m[OK]${EchoReset}"
WARN="\E[33;40m[OK]${EchoReset}"
ERR="\E[31;40m[OK]${EchoReset}"
#echo -e "$OK $WARN $ERR"

# ------------------------------------------------------------------------------------------------------------------------
# INPUT PARAMETER --------------------------------------------------------------------------------------------------------
# ------------------------------------------------------------------------------------------------------------------------

allow_formats="raw qcow2 vdi blkdebug blklogwrites blkreplay blkverify bochs cloop copy-on-read dmg file ftp ftps gluster host_cdrom host_device http https iscsi iser luks nbd nfs null-aio null-co nvme parallels qcow qed quorum rbd replication sheepdog ssh throttle vhdx vmdk vpc vvfat"
allow_formats_iso="iso nrg bin img mdf"
allow_formats_crypto="tcrypt plain luks luks1 luks2 loopaes"

if [[ $1 == "-h" || $1 == "--help" ]] ; then
  cat << EOF
image-mount command help

Usage
  image-mount [--help|-h]"
  image-mount [path/]file-name[.format] [format | #partition] [path-where-mount | #partition] [--force-iso-stdmount]
  --force-iso-stdmount   -> force in iso sudo mount 'mount -o loop' instead 'fuseiso'
  -y --ask-confirm       -> ask confirm before proceeed

Warnings
  1) part# in in format 0-9 with one integer value
  2) format must be explicily specified if file extension do not say itself the correct format.
  3) all parameters are case sensitive.

Default pars:
  image-mount image.raw raw 0 \${HOME}/mount/image
Minimal usate:
  image-mount file.ext
Common usage:
  image-mount file.ext 2

Full allowed format list (with 'more partitions inside' supported)
----------------------------------------------------------------------------------------------------
Virtual images allowed ('qemu' for this function package required):
  ${allow_formats}
Iso images (fuse standard mount or optional 'fuseiso' package required):
  ${allow_formats_iso}
Crypto images ('cryptsetup' for this function package is required):
  ${allow_formats_crypto}
Squash readonly file (can ben modified, saved on umount) (package 'squashfs-tools' required):
  sqhashfs
EOF

  exit 0
elseif [[ $1 == "--version" ]]
  echo "image-mount Version 0.0 (simonascripts package)"
  exit 0
fi

# DEFAULTS -------------------------------------------------------------------------------------------------------------
full_path_image="image.raw"
n_part=0
#n_part=2
mount_where="$HOME/mount/image"
format="" #something required (no default allowed)

# PARS -----------------------------------------------------------------------------------------------------------------
#parameter $1: name file used as image
par=$1; [[ $1 != "" && ${par:0:1} != "-" ]] && full_path_image=$1
image="${full_path_image##*/}" # estract substring: take only final part after last '/' char
[[ ! -f "$full_path_image" ]] && { echo -e "$ERR File image '$full_path_image' not found!! Exit."; exit 1; }

ext="${image##*.}" # extract substring: take only final part after last '.' char
[[ $ext != "" ]] && format=$ext

#parameter $2: format or partition number
case $2 in
  [0-9]* ) n_part=$2 ;;
  [a-z]* ) format=$2 ;;
esac

#parameter $3: mount path or partition number
case $3 in
  [0-9]* ) n_part=$3 ;;
  * ) par=$3; [[ $3 != "" && ${par:0:1} != "-" ]] && format=$3 ;;
esac

[[ $format == "" ]] && { echo "Parameter missing. Please insert format as second command parameter."; exit 1; }
if [[ ! $allow_formats =~ $format &&
      ! $allow_formats_iso =~ $format &&
      ! $allow_formats_crypto =~ $format &&
      ! "squashfs" =~ $format ]] ; then
  echo -e "$ERR Format $format unknown or not supported. Sorry. Exit."
  exit 1
fi

#parameter 4: mount path
par=$4; [[ $4 != "" && ${par:0:1} != "-" ]] && mount_where=$4

[[ ! -d $mount_where ]] && { echo "[TO-DO] Please create $mount_where with 'mkdir -p $mount_where'. Exit."; exit 1; }

#display default
#[[ $full_path_image == "image.raw" ]] && echo "Default: 'image.raw' image file"
[[ $allow_formats =~ $format && $n_part == 0 ]] && echo "Default: partition number 0"
[[ $mount_where == "$HOME/mount/image" ]] && echo "Default: '$HOME/mount/image' dir where mount image"

[[ -d "$mount_where/$image" ]] && { echo -e "$ERR $image found. umount/rm -r \"$mount_where/$image\". Exit."; exit 1; }

forceIsoStdMount=false #default par
flagAskConfirm=false
while [ $# -ne 0 ] ; do
  case $1 in
    --force-iso-stdmount) forceIsoStdMount=true ;;
    -y|--ask-confirm) flagAskConfirm=true ;;
    *) ;;
  esac
  shift
done

echo "Now mounting image <$image> format <$format> on path <$mount_where>, partition number <$n_part>."
if [[ "$flagAskConfirm" == true ]] ; then
  echo -n "Continue? (Y/N): "; read ask
  [[ ! $ask =~ ^(yes|y|YES|y|s|S|si|SI|sì)$ ]] && { echo "Exit forced by user."; exit 0; }
fi


# ------------------------------------------------------------------------------------------------------------------------
# WORK -------------------------------------------------------------------------------------------------------------------
# ------------------------------------------------------------------------------------------------------------------------

mkdir -p "$mount_where/$image"
[[ $? -ne 0 ]] && { echo -e "$ERR Fail creation dir $mount_where/$image! Exit."; exit 1; }

cd "$mount_where/$image"
if [[ $? -ne 0 ]] ; then
  echo -e "$ERR Fail 'cd \"$mount_where/$image\"' cmd! Exit."
  rmdir "$mount_where/$image" #undo
  [ $? -ne 0 ] && echo -e "$WARN Fail undo removing dir 'rmdir $mount_where/$image' cmd!"
  exit 1
fi
echo "[OK] prepare success."

# vm images -----------------------------------------------------------------------------------------------------------
if [[ $allow_formats =~ $format ]] ; then

  [[ ! -r /bin/qemu-nbd ]] && { echo -e "$ERR Missing qemu-nbd cmd (qemu pkg?)! Exit"; exit 1; }

  sudo modprobe nbd max_part=63
  [[ $? -ne 0 ]] && { echo -e "$ERR Fail 'modprobe nbd max_part=63' cmd! Exit."; exit 1; }
  echo "[OK] nbd module prepared... success."

  sudo qemu-nbd -c /dev/nbd0 --format=$format "$full_path_image"
  [[ $? -ne 0 ]] && { echo -e "$ERR Fail 'qemu-nbd -c /dev/nbd0 --format=$format \"$full_path_image\"' cmd! Exit."; exit 1; }
  echo -e "$OK qemu nbd started... success."

  if [[ n_part -eq 0 ]] ; then
    #ls /dev/nbd0p?
    #echo -n "* "; for dev_part in /dev/nbd0p* ; do echo $dev_part; done
    #echo -n "? "; for dev_part in /dev/nbd0p? ; do echo $dev_part; done
    #echo -n "[0-9] "; for dev_part in /dev/nbd0p[0-9] ; do echo $dev_part; done
    for dev_part in /dev/nbd0p[0-9] ; do
      part=${dev_part:10:1} # estrae substring: dopo n-esimo carattere (0 based) per N caratteri
      mkdir -p "$mount_where/$image/$part"
      if [ $? -ne 0 ]; then
        echo -e "$WARN Fail creation dir $mount_where/$image/$part and mount part $part..."
      els
        sudo mount $dev_part "$mount_where/$image/$part"
        if [ $? -ne 0 ]; then
          sudo qemu-nbd -d /dev/nbd0 >/dev/null #undo
          [[ $? -ne 0 ]] && echo -e "$WARN Fail undo 'qemu-nbd -d /dev/nbd0' cmd!" || echo -e "$OK qemu nbd stopped... success."
          #rmdir -r "$mount_where/$image" #undo
          #[ $? -ne 0 ] && echo "Can't execute command 'rmdir $mount_where/$image'!"
          echo -e "$ERR Fail'mount $dev_part $mount_where/\"$image/$part\"' cmd! Exit."
          exit 1
        fi
      fi
      echo -e "$OK Mount $dev_part ... [part# $part] on $mount_where/$image/$part success."
    done
  else
    sudo mount /dev/nbd0p$n_part "$mount_where/$image"
    if [ $? -ne 0 ]; then
      sudo qemu-nbd -d /dev/nbd0 >/dev/null #undo
      [[ $? -ne 0 ]] && echo -e "$WARN Fail undo 'qemu-nbd -d /dev/nbd0' cmd!" || echo -e "$OK qemu nbd stopped... success."
      rmdir "$mount_where/$image" #undo
      [ $? -ne 0 ] && echo -e "$WARN Fail undo removing dir 'rmdir $mount_where/$image' cmd!"
      echo -e "$ERR Part num $n_part really exist? Fail mount /dev/nbd0p$n_part on $mount_where/\"$image\"'! Exit."
      exit 1
    fi
    echo -e "$OK Mount /dev/nbd0p$n_part ... [partition $n_part] on $mount_where/$image success."
  fi

# iso images ----------------------------------------------------------------------------------------------------------
elif [[ $allow_formats_iso =~ $format ]] ; then
  iso_mount_type=""
  if [[ -r /bin/fuseiso && "$forceIsoStdMount" != true ]] ; then

    fuseiso "$full_path_image" "$mount_where/$image"
    if [ $? -ne 0 ]; then
      rmdir "$mount_where/$image" #undo
      [ $? -ne 0 ] && echo -e "$WARN Fail undo removing dir 'rmdir $mount_where/$image' cmd!"
      echo -e "$ERR Fail'fuseiso "$full_path_image" /mnt' cmd! Exit."
      exit 1
    fi
    echo -e "$OK Mount (fuseiso) $mount_where/$image success."
    iso_mount_type="fuseiso"
  else

    sudo mount -o loop "$full_path_image" "$mount_where/$image"
    if [ $? -ne 0 ]; then
      rmdir "$mount_where/$image" #undo
      [ $? -ne 0 ] && echo -e "$WARN Fail undo removing dir 'rmdir $mount_where/$image' cmd!"
      echo -e "$ERR Fail 'mount -o loop \"$full_path_image\" \"$mount_where/$image\"' cmd! Exit."
      exit 1
    fi

    echo -e "$OK Mount (sudo mount loop) $mount_where/$image success."
    iso_mount_type="mount-loop"

  fi

# crypto images -------------------------------------------------------------------------------------------------------
elif [[ $allow_formats_crypto =~ $format ]] ; then

  [[ ! -r /bin/cryptsetup ]] && { echo -e "$ERR Missing cryptsetup cmd (cryptsetup pkg?)! Exit"; exit 1; }

  sudo cryptsetup open --type $format "$full_path_image" crypt-disk
  if [ $? -ne 0 ]; then
    rmdir "$mount_where/$image" #undo
    [ $? -ne 0 ] && echo -e "$WARN Fail removing dir 'rmdir $mount_where/$image' cmd!"
    echo -e "$ERR Fail 'cryptsetup open --type $format \"$full_path_image\" crypt-disk' cmd! Exit."
    exit 1
  fi
  sudo mount -o uid=1000 /dev/mapper/crypt-disk "$mount_where/$image"
  if [ $? -ne 0 ]; then
    sudo cryptsetup close crypt-disk #undo
    [ $? -ne 0 ] && echo -e "$WARN Fail 'cryptsetup close crypt-disk' cmd!"
    rmdir "$mount_where/$image" #undo
    [ $? -ne 0 ] && echo -e "$WARN Fail undo removing dir 'rmdir $mount_where/$image' cmd!"
    echo -e "$ERR Fail 'mount -o uid=1000 /dev/mapper/crypt-disk \"$mount_where/$image\"' cmd! Exit."
    exit 1
  fi

  echo -e "$OK Mount 'pwd-crypt..."

# sqhash images -------------------------------------------------------------------------------------------------------
elif [[ "squashfs" =~ $format ]] ; then
  [[ ! -r /bin/mksquash ]] && { echo -e "$ERR Missing mksqhash cmd (squash-tool pkg?)! Exit"; exit 1; }

  cp "$full_path_image" "$mount_where/$image/."
  if [[ $? -ne 0 ]] ; then
    echo -e "$ERR Fail copy squashfs image! Exit."
    rmdir "$mount_where/$image" #undo
    [ $? -ne 0 ] && echo -e "$WARN Fail undo removing dir 'rmdir $mount_where/$image' cmd!"
    exit 1
  fi

  #sudo mount root.sqsh mnt -t squashfs -o loop #ok, but read only, because squash file cannot be modified
  unsquashfs "$full_path_image"
  if [[ $? -ne 0 ]] ; then
    echo -e "$ERR Fail unsquash cmd! Exit."
    rmdir "$mount_where/$image" #undo
    [ $? -ne 0 ] && echo -e "$WARN Fail undo removing dir 'rmdir $mount_where/$image' cmd!"
    exit 1
  else
    echo -e "$OK unsquash success."
  fi

  rm "$mount_where/$image/$image" #erase copy of sqhash filesystem image file (after decompressed)
  [ $? -ne 0 ] && echo -e "$WARN Fail removing image file dupe '$mount_where/$image/$image'!"

fi

start_evid="\e[1;39;41m"
stop_evid="\e[0m"
exit_string="All done. [MEMO] Dont' forget to use ${start_evid}srun-umount-img \"${image}\""
[[ $format != $ext ]] && exit_string="${exit_string} ${format}"
[[ $iso_mount_type == "mount-loop" && "$forceIsoStdMount" == true ]] && exit_string="${exit_string} --force-iso-stdmount"
exit_string="${exit_string}${stop_evid} when finished."
echo -e "$exit_string"
exit 0