#!/usr/bin/env bash



  #
  # Copyright (C) 2010 Cloud.com, Inc.  All rights reserved.
  # 
  # This software is licensed under the GNU General Public License v3 or later.
  # 
  # It is free software: you can redistribute it and/or modify
  # it under the terms of the GNU General Public License as published by
  # the Free Software Foundation, either version 3 of the License, or any later version.
  # This program is distributed in the hope that it will be useful,
  # but WITHOUT ANY WARRANTY; without even the implied warranty of
  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  # GNU General Public License for more details.
  # 
  # You should have received a copy of the GNU General Public License
  # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  #
 

# $Id: managevolume.sh 9132 2010-06-04 20:17:43Z manuel $ $HeadURL: svn://svn.lab.vmops.com/repos/vmdev/java/scripts/storage/qcow2/managevolume.sh $
# modifyvolume.sh -- add or delete a disk volume 
#

usage() {
  echo "Usage: modifydisk.sh -f <folder of volume> -p <path of volume> -s <size of volume> [-d <delete>] \n"
}

check_params() {
  # The folder, path, and size must be passed in
  if [ "$fflag$pflag$sflag" != "111" ]
  then
    return 1
  else
    return 0
  fi
}

cleanup_and_exit_if_error() {
  local return_code=$1
  local msg=$2
  local path=$3
  
  if [ $return_code -gt 0 ]
  then
    delete_disk $path
    exit_if_error $return_code "$msg"
  fi
}

exit_if_error() {
  local return_code=$1
  local msg=$2
  
  if [ $return_code -gt 0 ]
  then
    printf "${msg}\n"
    exit 1
  fi
}

make_folder() {
  local folder=$1
  
  if [ ! -d ${folder} ]
  then
    mkdir -p ${folder}
  fi
}

check_disk() {
  local folder=$1
  local path=$2
  
  make_folder $folder
  
  if [ -f $path ]
  then
    return 1
  else
    return 0
  fi
}

delete_disk() {
  local path=$1
  
  if [ -f $path ]
  then
    rm $path
  fi
  
  return 0
}

create_disk() {
  local path=$1
  local size=$2
  
  size=$(convert_size_to_gb $size)
  
  if [ $? -gt 0 ]
  then
    return 1
  fi

  qemu-img create -f qcow2 $path $size
  
  return $?
}

convert_size_to_gb() {
  local size=$1
  
  suffix=${size:(-1)}
  case $suffix in
    G)   
         ;;
    [0-9])   size=${size}G
         ;;
    *)   printf "Error in disk size: expect G as a suffix or no suffix\n"
         return 1
         ;;
  esac
  
  echo $size
  return 0
}

# set -x

fflag=0
pflag=0
sflag=0
dflag=0

while getopts 'f:p:s:d:' OPTION
do
  case $OPTION in
  f)	fflag=1
  		folder="$OPTARG"
  		;;
  p)	pflag=1
		path="$OPTARG"
		;;
  s)	sflag=1
  		size="$OPTARG"
  		;;
  d)	dflag=1
  		;;
  ?)	usage
		exit 2
		;;
  esac
done

# Check all parameters
check_params
exit_if_error $? "$(usage)"

if [ "$dflag" == "0" ]
then
  # Add the volume
  
  # Create the folder if necessary, and make sure there is no existing disk there
  check_disk $folder $path
  cleanup_and_exit_if_error $? "Failed to create disk at $path; path already exists." $path
  
  # Create the disk
  create_disk $path $size
  cleanup_and_exit_if_error $? "Failed to create disk at $path of size $datadisksize." $path
  
else
  # Delete the volume
  
  delete_disk $path
  
fi

exit 0