#!/bin/bash

# -------------------------------------------------------------------------- #
# Copyright 2002-2017, 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.                                             #
#--------------------------------------------------------------------------- #
# mvds host:remote_system_ds/disk.i fe:SOURCE
#   - fe is the front-end hostname
#   - SOURCE is the path of the disk image in the form DS_BASE_PATH/disk
#   - host is the target host to deploy the VM
#   - remote_system_ds is the path for the system datastore in the host

SRC=$1
DST=$2
VM_ID=$3
DS_ID=$4

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

source $TMCOMMON

#-------------------------------------------------------------------------------
# Set dst path and dir
#-------------------------------------------------------------------------------
SRC_PATH=`arg_path $SRC`
SRC_HOST=`arg_host $SRC`

DST_PATH=`arg_path $DST`
DST_HOST=`arg_host $DST`

DISK_ID=$(echo $SRC_PATH|$AWK -F. '{print $NF}')
DS_SYS_ID=$(echo $SRC_PATH|$AWK -F/ '{print $(NF-2)}')

LV_NAME="lv-one-vm-$VM_ID-$DISK_ID"
VG_NAME="vg-one-$DS_SYS_ID"

DUMP_CMD=$(cat <<EOF
    set -e -o pipefail
    hostn=\$(hostname -s)
    DEV="/dev/${VG_NAME}-\${hostn}/${LV_NAME}"
    $SUDO $LVCHANGE -ay $DST
    SIZE=\$($SUDO lvdisplay --units m -o lv_size --nosuffix --noheadings -C ${DST} | sed -e 's/^[[:space:]]*//' | awk '{print (\$0-int(\$0)>0)?int(\$0)+1:int(\$0)}' 2>/dev/null)
    TMP_LV_NAME=\$(basename $DST)
    set \$($SUDO lvdisplay -o lv_name,vg_name --nosuffix --noheadings -C $DST)
    TMP_LV_NAME="\${1}_tmp"
    DST_TMP=/dev/\${2}/\${TMP_LV_NAME}
    $SUDO $LVCREATE --wipesignatures n -L\${SIZE}M -n \$TMP_LV_NAME \$2
    $SUDO $QEMU_IMG convert -O raw \${DEV} \$DST_TMP
    $SUDO ${DD} if=\$DST_TMP of=$DST bs=1M
    $SUDO $LVREMOVE -f \$DST_TMP
EOF
)

DEL_CMD=$(cat <<EOF
    set -e -o pipefail
    hostn=\$(hostname -s)
    DEV="${VG_NAME}-\${hostn}/${LV_NAME}"
    if [ -e /dev/\$DEV ]; then
        $SUDO $LVREMOVE -f \$DEV
    fi
EOF
)

#-------------------------------------------------------------------------------
# Move the image back to the datastore
#-------------------------------------------------------------------------------
log "Dumping $SRC to $DST"

ssh_exec_and_log "$SRC_HOST" "$DUMP_CMD" \
    "Error dumping $SRC to $DST"

ssh_exec_and_log "$SRC_HOST" "$DEL_CMD" \
    "Error dell $SRC to $DST"

exit 0
