#!/bin/bash

# -------------------------------------------------------------------------- #
# Copyright 2002-2019, OpenNebula Project, OpenNebula Systems                #
#                                                                            #
# Licensed under the Apache License, Version 2.0 (the "License"); you may    #
# not use this file except in compliance with the License. You may obtain    #
# a copy of the License at                                                   #
#                                                                            #
# http://www.apache.org/licenses/LICENSE-2.0                                 #
#                                                                            #
# Unless required by applicable law or agreed to in writing, software        #
# distributed under the License is distributed on an "AS IS" BASIS,          #
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   #
# See the License for the specific language governing permissions and        #
# limitations under the License.                                             #
#--------------------------------------------------------------------------- #

# resize image size vmid

SRC=$1
SIZE=$2
VM_ID=$3

#--------------------------------------------------------------------------------

if [ -z "${ONE_LOCATION}" ]; then
    TMCOMMON=/var/lib/one/remotes/tm/tm_common.sh
    LIB_LOCATION=/usr/lib/one
else
    TMCOMMON=$ONE_LOCATION/var/remotes/tm/tm_common.sh
    LIB_LOCATION=$ONE_LOCATION/lib
fi

DRIVER_PATH=$(dirname $0)

source $TMCOMMON

#source ${DRIVER_PATH}/efs_utils.sh
source ${DRIVER_PATH}/../../datastore/libfs.sh
source ${DRIVER_PATH}/../../efs_utils

efs_log_var "SRC"
efs_log_var "SIZE"
efs_log_var "VM_ID"

#-------------------------------------------------------------------------------
# Set dst path and dir
#-------------------------------------------------------------------------------

SRC_HOST=`arg_host $SRC`
SRC_PATH=`arg_path $SRC`

#-------------------------------------------------------------------------------
# Get Image information
#-------------------------------------------------------------------------------

DS_ID=$(echo $SRC | sed s#//*#/#g | awk -F/ '{print $(NF-2)}')
DISK_ID=$(echo "$SRC_PATH" | $AWK -F. '{print $NF}')

XPATH="${DRIVER_PATH}/../../datastore/xpath.rb --stdin"

unset i j XPATH_ELEMENTS

while IFS= read -r -d '' element; do
    XPATH_ELEMENTS[i++]="$element"
done < <(onevm show -x $VM_ID| $XPATH \
                    /VM/TEMPLATE/CONTEXT/AXC_VMTYPE \
                    /VM/TEMPLATE/DISK[DISK_ID=$DISK_ID]/IMAGE_ID \
                    /VM/TEMPLATE/DISK[DISK_ID=$DISK_ID]/SOURCE \
                    /VM/TEMPLATE/DISK[DISK_ID=$DISK_ID]/PERSISTENT)

AXC_VMTYPE="${XPATH_ELEMENTS[j++]:-efs}"
IMGID="${XPATH_ELEMENTS[j++]}"
ISCSI_SRC="${XPATH_ELEMENTS[j++]}"
PERSISTENT="${XPATH_ELEMENTS[j++]}"

case "$AXC_VMTYPE" in
  aa|acr|anc|cf|efs|rb|sp|ve)
    ;;
  *)
    AXC_VMTYPE="efs"
    ;;
esac


efs_log_var "ISCSI_SRC"
efs_log_var "PERSISTENT"

if [ "x${PERSISTENT}" = 'xYES' ]; then
  unset i j XPATH_ELEMENTS

  while IFS= read -r -d '' element; do
        XPATH_ELEMENTS[i++]="$element"
  done < <(oneimage show -x $IMGID | $XPATH \
                    /IMAGE/DATASTORE_ID)
  DS_ID="${XPATH_ELEMENTS[j++]}"
fi

#-------------------------------------------------------------------------------
# Get System Datastore information
#-------------------------------------------------------------------------------

unset i j XPATH_ELEMENTS

while IFS= read -r -d '' element; do
        XPATH_ELEMENTS[i++]="$element"
done < <(onedatastore show -x $DS_ID | $XPATH \
                    /DATASTORE/TEMPLATE/BRIDGE_LIST\
                    /DATASTORE/TEMPLATE/POOL_NAME\
                    /DATASTORE/TEMPLATE/ISCSI_LUN)

BRIDGE_LIST="${XPATH_ELEMENTS[j++]}"
POOL_NAME="${XPATH_ELEMENTS[j++]}"
ISCSI_LUN="${XPATH_ELEMENTS[j++]}"

DST_HOST=`get_destination_host $DS_ID`
#DST_HOST=`pool_to_host ${POOL_NAME}`

efs_log_var "DST_HOST"
efs_log_var "POOL_NAME"
efs_log_var "ISCSI_LUN"


#if [ -n "${ISCSI_SRC}" ]; then
#    if [ "${PERSISTENT}" = 'YES' ]; then
#        ISCSI_DST=$ISCSI_SRC
#    else
#        ISCSI_DST="${ISCSI_SRC}-${VM_ID}-${DISK_ID}"
#    fi
#else
#    # volatile ?
#    RBD_DST="${ISCSI_SRC}:sys-${VM_ID}-${DISK_ID}"
#fi

IMGSRC=$(echo $ISCSI_SRC | awk -F':' '{ print $2 }')
IMGFILE="${POOL_NAME}/one/${AXC_VMTYPE}/system/${VM_ID}/${DISK_ID}"
if [ "${PERSISTENT}" = 'YES' ]; then
    IMGFILE="${POOL_NAME}/one/${IMGID}"
fi
#-------------------------------------------------------------------------------
# Resize disk
#-------------------------------------------------------------------------------

# for running VMs, leave disk resize on kvm/resize_disk
#if [ `lcm_state $VM_ID` -eq 62 ]; then
#    log "Not resizing disk ${SRC_PATH} for running virt. machine directly"
#    exit 0
#fi

RESIZE_CMD=$(cat <<EOF
    sudo zfs set volsize=${SIZE}M ${IMGFILE}
    sleep 15
#truncate -s ${SIZE}M ${IMGFILE}
    if [ \$(uname) == "Linux" ]
    then
        sudo /usr/bin/systemctl restart tgt.service
    elif [ \$(uname) == "FreeBSD" ]
    then
        sudo /usr/sbin/service ctld reload
    else
        exit 1
    fi
    sleep 15
EOF
)

ssh_exec_and_log_stdin $DST_HOST "$RESIZE_CMD" \
                 "Error resizing disk $IMGFILE"

exit 0
