#!/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.                                             #
#--------------------------------------------------------------------------- #

###############################################################################
# This script is used to copy a VM image (SRC) to the image repository as DST
# Several SRC types are supported
###############################################################################

# -------- Set up the environment to source common tools & conf ------------

if [ -z "${ONE_LOCATION}" ]; then
    LIB_LOCATION=/usr/lib/one
else
    LIB_LOCATION=$ONE_LOCATION/lib
fi

. $LIB_LOCATION/sh/scripts_common.sh

DRIVER_PATH=$(dirname $0)
source ${DRIVER_PATH}/../libfs.sh
source ${DRIVER_PATH}/../../etc/datastore/ceph/ceph.conf

# -------- Get cp and datastore arguments from OpenNebula core ------------

DRV_ACTION=`cat -`
ID=$1

TS=`date +"%Y-%m-%d_%H%M%S"`
echo "${TS} [cp] - DRV_ACTION ${DRV_ACTION}" >>  /var/log/one/efs.log
echo "${TS} [cp] - ID ${ID}" >>  /var/log/one/efs.log

export DRV_ACTION

UTILS_PATH="${DRIVER_PATH}/.."

XPATH="$UTILS_PATH/xpath.rb -b $DRV_ACTION"

unset i XPATH_ELEMENTS

while IFS= read -r -d '' element; do
    XPATH_ELEMENTS[i++]="$element"
done < <($XPATH     /DS_DRIVER_ACTION_DATA/DATASTORE/BASE_PATH \
                    /DS_DRIVER_ACTION_DATA/DATASTORE/TEMPLATE/RESTRICTED_DIRS \
                    /DS_DRIVER_ACTION_DATA/DATASTORE/TEMPLATE/SAFE_DIRS \
                    /DS_DRIVER_ACTION_DATA/DATASTORE/TEMPLATE/BRIDGE_LIST \
                    /DS_DRIVER_ACTION_DATA/DATASTORE/TEMPLATE/POOL_NAME \
                    /DS_DRIVER_ACTION_DATA/DATASTORE/TEMPLATE/STAGING_DIR \
                    /DS_DRIVER_ACTION_DATA/IMAGE/PATH \
                    /DS_DRIVER_ACTION_DATA/IMAGE/SIZE \
                    /DS_DRIVER_ACTION_DATA/IMAGE/TEMPLATE/MD5 \
                    /DS_DRIVER_ACTION_DATA/IMAGE/TEMPLATE/SHA1 \
                    /DS_DRIVER_ACTION_DATA/DATASTORE/TEMPLATE/NO_DECOMPRESS \
                    /DS_DRIVER_ACTION_DATA/DATASTORE/TEMPLATE/LIMIT_TRANSFER_BW \
                    /DS_DRIVER_ACTION_DATA/DATASTORE/TEMPLATE/ISCSI_BASE_IQN\
                    /DS_DRIVER_ACTION_DATA/DATASTORE/TEMPLATE/ISCSI_LUN \
                    /DS_DRIVER_ACTION_DATA/IMAGE/NAME )

unset i

BASE_PATH="${XPATH_ELEMENTS[i++]}"
RESTRICTED_DIRS="${XPATH_ELEMENTS[i++]}"
SAFE_DIRS="${XPATH_ELEMENTS[i++]}"
BRIDGE_LIST="${XPATH_ELEMENTS[i++]}"
POOL_NAME="${XPATH_ELEMENTS[i++]:-$POOL_NAME}"
STAGING_DIR="${XPATH_ELEMENTS[i++]:-$STAGING_DIR}"
SRC="${XPATH_ELEMENTS[i++]}"
SIZE="${XPATH_ELEMENTS[i++]}"
MD5="${XPATH_ELEMENTS[i++]}"
SHA1="${XPATH_ELEMENTS[i++]}"
NO_DECOMPRESS="${XPATH_ELEMENTS[i++]}"
LIMIT_TRANSFER_BW="${XPATH_ELEMENTS[i++]}"
ISCSI_BASE_IQN="${XPATH_ELEMENTS[i++]}"
ISCSI_LUN="${XPATH_ELEMENTS[i++]:-1}"
IMAGE_NAME="${XPATH_ELEMENTS[i++]}"
ZPOOL=${POOL_NAME}

DST_HOST=`get_destination_host $ID`

echo "${TS} [cp] - BASE_PATH ${BASE_PATH}" >>  /var/log/one/efs.log
echo "${TS} [cp] - RESTRICTED_DIRS ${RESTRICTED_DIRS}" >>  /var/log/one/efs.log
echo "${TS} [cp] - SAFE_DIRS ${SAFE_DIRS}" >>  /var/log/one/efs.log
echo "${TS} [cp] - STAGING_DIR ${STAGING_DIR}" >>  /var/log/one/efs.log
echo "${TS} [cp] - BRIDGE_LIST ${BRIDGE_LIST}" >>  /var/log/one/efs.log
echo "${TS} [cp] - POOL_NAME ${POOL_NAME}" >>  /var/log/one/efs.log
echo "${TS} [cp] - SRC ${SRC}" >>  /var/log/one/efs.log
echo "${TS} [cp] - SIZE ${SIZE}" >>  /var/log/one/efs.log
echo "${TS} [cp] - DST_HOST ${DST_HOST}" >>  /var/log/one/efs.log

#error_message "Copy stub"
#exit -1

if [ -z "$DST_HOST" ]; then
    error_message "Datastore template missing 'BRIDGE_LIST' attribute."
    exit -1
fi

set_up_datastore "$BASE_PATH" "$RESTRICTED_DIRS" "$SAFE_DIRS"
#set_up_datastore "/nfs/${ZPOOL}/vms/efs/" "$RESTRICTED_DIRS" "$SAFE_DIRS"

IMAGE_HASH=`generate_image_hash`
TMP_DST="$STAGING_DIR/$IMAGE_HASH"


ZFS_SOURCE="${POOL_NAME}/one/${ID}"
IMAGE_PATH="/dev/zvol/${ZFS_SOURCE}"

DOWNLOADER_ARGS=`set_downloader_args "$MD5" "$SHA1" "$NO_DECOMPRESS" "$LIMIT_TRANSFER_BW" "$SRC" -`

COPY_COMMAND="$UTILS_PATH/downloader.sh $DOWNLOADER_ARGS"

case $SRC in
http://*|https://*)
    log "Downloading $SRC to the image repository"

    DUMP="$COPY_COMMAND"
    ;;

*)
    if [ `check_restricted $SRC` -eq 1 ]; then
        log_error "Not allowed to copy images from $RESTRICTED_DIRS"
        error_message "Not allowed to copy image file $SRC"
        exit -1
    fi

    log "Copying local image $SRC to the image repository"

    DUMP="$COPY_COMMAND"
    ;;
esac

multiline_exec_and_log "set -e -o pipefail; $DUMP | $SSH $DST_HOST $DD of=$TMP_DST bs=${DD_BLOCK_SIZE:-64k}" \
             "Error dumping $SRC to $DST_HOST:$TMP_DST"


TGTCONF=$(cat <<EOF
<target "${ISCSI_BASE_IQN}:${ID}">
    backing-store ${IMAGE_PATH}
</target>
EOF
)


REGISTER_CMD=$(cat <<EOF
    set -e -o pipefail
    umask 0006
    sudo zfs create -p ${POOL_NAME}/one
    sudo zfs set canmount=noauto ${POOL_NAME}/one
    IMGSIZE=\$($QEMU_IMG info $TMP_DST | grep "^virtual size" | grep -o -E "[0-9]+ bytes" | awk '{ print \$1 }')
    sudo zfs create -sV \$IMGSIZE ${ZFS_SOURCE}
    sudo zfs set volmode=dev ${ZFS_SOURCE}

    FORMAT=\$($QEMU_IMG info $TMP_DST | grep "^file format:" | awk '{print \$3}' || :)
    sleep 3
#    if [ "\$FORMAT" != "raw" ]; then
        sudo $QEMU_IMG dd -O raw if=$TMP_DST of=${IMAGE_PATH} bs=1m
#    else
#        sudo $QEMU_IMG dd -O raw if=$TMP_DST of=${IMAGE_PATH} bs=1m
#    fi
    sudo zfs set one:current=${TS} ${ZFS_SOURCE}
    sudo zfs snapshot ${ZFS_SOURCE}@${TS}
UNAME=\$(uname)

case "\${UNAME}" in
    "FreeBSD")
       sudo ctladd ${ISCSI_BASE_IQN}:${ID} ${ISCSI_LUN} ${IMAGE_PATH} "${IMAGE_NAME}"
       sudo service ctld reload
       ;;
    "Linux")
       sudo /usr/bin/echo '$TGTCONF' > /etc/tgt/conf.d/img-${ID}.conf
       sudo /usr/sbin/tgt-admin --update ALL -c /etc/tgt/targets.conf
       ;;
    *)
       exit 1
       ;;
esac
    # remove original
    $RM -f $TMP_DST
EOF
)

echo "${TS} [cp] - REGISTER_CMD ${REGISTER_CMD}" >>  /var/log/one/efs.log

ssh_exec_and_log    "$DST_HOST" "$REGISTER_CMD" \
                    "Error registering $ZFS_SOURCE (${ISCSI_BASE_IQN}:${ID}) in $DST_HOST"

echo "${ISCSI_BASE_IQN}:${ID} raw"
