#!/bin/bash -e
set -u
# Copyright 2016 The Rook Authors. All rights reserved.
#
# 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.

BUILD_HOST=$(hostname)
BUILD_REPO=github.com/rook/rook
BUILD_ROOT=$( cd "$( dirname "${BASH_SOURCE[0]}" )/.." && pwd -P)
SHA256CMD=${SHA256CMD:-shasum -a 256}
BUILD_REGISTRY=build-$(echo "${BUILD_HOST}"-"${BUILD_ROOT}" | ${SHA256CMD} | cut -c1-8)

DOCKERCMD=${DOCKERCMD:-docker}

export scriptdir
scriptdir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
export OUTPUT_DIR=${BUILD_ROOT}/_output
export WORK_DIR=${BUILD_ROOT}/.work
export CACHE_DIR=${BUILD_ROOT}/.cache

CROSS_IMAGE=${BUILD_REGISTRY}/cross-amd64
CROSS_IMAGE_VOLUME=cross-volume
CROSS_RSYNC_PORT=10873

function ver() {
    local full_ver maj min bug build
    full_ver="$1" # functions should name input params for easier understanding
    maj="$(echo "${full_ver}" | cut -f1 -d'.')" # when splitting a param, name the components for easier understanding
    min="$(echo "${full_ver}" | cut -f2 -d'.')"
    bug="$(echo "${full_ver}" | cut -f3 -d'.')"
    build="$(echo "${full_ver}" | cut -f4 -d'.')"
  printf "%d%03d%03d%03d" "${maj}" "${min}" "${bug}" "${build}"
}

function check_git() {
    # git version 2.6.6+ through 2.8.3 had a bug with submodules. this makes it hard
    # to share a cloned directory between host and container
    # see https://github.com/git/git/blob/master/Documentation/RelNotes/2.8.3.txt#L33
    local gitversion
    gitversion=$(git --version | cut -d" " -f3)

    if (( $(ver "${gitversion}") > $(ver 2.6.6) && $(ver "${gitversion}") < $(ver 2.8.3) )); then
        echo WARN: you are running git version "${gitversion}" which has a bug related to relative
        echo WARN: submodule paths. Please consider upgrading to 2.8.3 or later
    fi
}

function start_rsync_container() {
    ${DOCKERCMD} run \
        -d \
        -e OWNER=root \
        -e GROUP=root \
        -e MKDIRS="/volume/go/src/${BUILD_REPO}" \
        -p ${CROSS_RSYNC_PORT}:873 \
        -v ${CROSS_IMAGE_VOLUME}:/volume \
        --entrypoint "/tini" \
        "${CROSS_IMAGE}" \
        -- /build/rsyncd.sh
}

function wait_for_rsync() {
    # wait for rsync to come up
    local tries=100
    while (( tries > 0 )); do
        if rsync "rsync://localhost:${CROSS_RSYNC_PORT}/"  &> /dev/null ; then
            return 0
        fi
        (( tries-- ))
        sleep 0.1
    done
    echo ERROR: rsyncd did not come up >&2
    exit 1
}

function stop_rsync_container() {
    local id=$1

    ${DOCKERCMD} stop "${id}" &> /dev/null || true
    ${DOCKERCMD} rm "${id}" &> /dev/null || true
}

function run_rsync() {
    local src=$1
    shift

    local dst=$1
    shift

    # run the container as an rsyncd daemon so that we can copy the
    # source tree to the container volume.
    local id
    id=$(start_rsync_container)

    # wait for rsync to come up
    wait_for_rsync || { stop_rsync_container "${id}"; return 1; }

    # NOTE: add --progress to show files being syncd
    rsync \
        --archive \
        --delete \
        --prune-empty-dirs \
        "$@" \
        "$src" "$dst" || { stop_rsync_container "${id}"; return 1; }

    stop_rsync_container "${id}"
}

function rsync_host_to_container() {
    run_rsync "${scriptdir}"/.. rsync://localhost:${CROSS_RSYNC_PORT}/volume/go/src/${BUILD_REPO} "$@"
}

function rsync_container_to_host() {
    run_rsync rsync://localhost:${CROSS_RSYNC_PORT}/volume/go/src/${BUILD_REPO}/ "${scriptdir}"/.. "$@"
}