github.com/Equinix-Metal/virtlet@v1.5.2-0.20210807010419-342346535dc5/build/cmd.sh

#!/bin/bash
set -o errexit
set -o nounset
set -o pipefail
set -o errtrace

CRIPROXY_DEB_URL="${CRIPROXY_DEB_URL:-https://github.com/Mirantis/criproxy/releases/download/v0.14.0/criproxy-nodeps_0.14.0_amd64.deb}"
VIRTLET_IMAGE_REPO="${VIRTLET_IMAGE_REPO:-ryugu-psie-docker-dev-local.jfrog.io/equinix}"
VIRTLET_SKIP_RSYNC="${VIRTLET_SKIP_RSYNC:-}"
VIRTLET_SKIP_VENDOR="${VIRTLET_SKIP_VENDOR:-false}"
VIRTLET_RSYNC_PORT="${VIRTLET_RSYNC_PORT:-18730}"
VIRTLET_ON_MASTER="${VIRTLET_ON_MASTER:-}"
VIRTLET_MULTI_NODE="${VIRTLET_MULTI_NODE:-}"
# XXX: try to extract the docker socket path from DOCKER_HOST if it's set to unix://...
DOCKER_SOCKET_PATH="${DOCKER_SOCKET_PATH:-/var/run/docker.sock}"
FORCE_UPDATE_IMAGE="${FORCE_UPDATE_IMAGE:-}"
IMAGE_REGEXP_TRANSLATION="${IMAGE_REGEXP_TRANSLATION:-1}"
GH_RELEASE_TEST_USER="ivan4th"
DIND_CRI="${DIND_CRI:-containerd}"
MKDOCS_SERVE_ADDRESS="${MKDOCS_SERVE_ADDRESS:-localhost:8042}"

# Note that project_dir must not end with slash
project_dir="$(cd "$(dirname "${BASH_SOURCE}")/.." && pwd)"
virtlet_image="${VIRTLET_IMAGE_REPO}/virtlet"
remote_project_dir="/go/src/github.com/Equinix-Metal/virtlet"
build_name="virtlet-build"
tmp_container_name="${build_name}-$(openssl rand -hex 16)"
build_image="${VIRTLET_IMAGE_REPO}/${build_name}:latest"
volume_name=virtlet_src
rsync_git=y
exclude=(
    --exclude 'vendor'
    --exclude .git
    --exclude _output
    --exclude '*.png'
)
rsync_pw_file="${project_dir}/_output/rsync.password"
busybox_image=busybox:1.27.2
virtlet_nodes=()
if [[ ${VIRTLET_ON_MASTER} ]]; then
  virtlet_nodes+=(kube-master)
fi
if [[ !${VIRTLET_ON_MASTER} || ${VIRTLET_MULTI_NODE} ]]; then
  virtlet_nodes+=(kube-node-1)
fi
if [[ ${VIRTLET_MULTI_NODE} ]]; then
  virtlet_nodes+=(kube-node-2)
fi
bindata_modtime=1522279343
bindata_out="pkg/tools/bindata.go"
bindata_dir="deploy/data"
bindata_pkg="tools"
ldflags=()
go_package=github.com/Equinix-Metal/virtlet

function image_tags_filter {
    local tag="${1}"
    local prefix=".items[0].spec.template.spec."
    local suffix="|=map(.image=\"equinix/virtlet:${tag}\")"
    echo -n "${prefix}containers${suffix}|${prefix}initContainers${suffix}"
}

function get_version() {
	inside_git=$(git rev-parse --is-inside-work-tree 2>&1 >/dev/null; echo $?)
	if [ $inside_git != 0 ]; then
    echo "Not in GIT repo and VERSION variable is not set."
    version='latest'
	else
    VERSION_HASH=$(git show --summary | head -1 | awk '{print substr($2,0,8)}')
    version="${VERSION_HASH}"
  fi
  echo "$version" >"${project_dir}/_output/version"
  return 0
}

# from build/common.sh in k8s
function rsync_probe {
    # Wait unil rsync is up and running.
    local tries=20
    while (( ${tries} > 0 )) ; do
        if rsync "rsync://k8s@${1}:${2}/" \
                 --password-file="${project_dir}/_output/rsyncd.password" \
           &> /dev/null ; then
            return 0
        fi
        tries=$(( ${tries} - 1))
        sleep 0.1
    done

    return 1
}

function image_exists {
    local name="${1}"
    # can't use 'docker images -q' due to https://github.com/docker/docker/issues/28895
    docker history -q "${name}" >& /dev/null || return 1
}

function ensure_build_image {
    mkdir -p "${project_dir}/_output"
    get_version
    build_tag="$(cat "${project_dir}/_output/version")"
    build_image="${VIRTLET_IMAGE_REPO}/${build_name}:${build_tag}"
    docker login ryugu-psie-docker-dev-local.jfrog.io
    virtlet_base_image="${VIRTLET_IMAGE_REPO}/virtlet-base:${build_tag}"
    build_base_image="${VIRTLET_IMAGE_REPO}/virtlet-build-base:${build_tag}"
    echo >&2 "Trying to pull the base image ${virtlet_base_image}..."
    if ! docker pull "${virtlet_base_image}"; then
        docker build -t "${virtlet_base_image}" \
               -f "${project_dir}/images/Dockerfile.virtlet-base" \
               "${project_dir}/images"
        docker push "${virtlet_base_image}"
    fi
    echo >&2 "Trying to pull the build base image ${build_base_image}..."
    if ! docker pull "${build_base_image}"; then
        docker build -t "${build_base_image}" \
               --label virtlet_image=build-base \
               --build-arg BASE_TAG=${build_tag} \
               -f "${project_dir}/images/Dockerfile.build-base" "${project_dir}/images"
        docker push "${build_base_image}"
    fi
    tar -C "${project_dir}/images" -c image_skel/ qemu-build.conf Dockerfile.build |
        docker build -t "${build_image}" --build-arg BUILD_TAG=${build_tag} \
        -f Dockerfile.build -
    docker tag "${build_image}" "${VIRTLET_IMAGE_REPO}/${build_name}:latest"
    docker push "${build_image}"
    docker push "${VIRTLET_IMAGE_REPO}/${build_name}:latest"
}

function get_rsync_addr {
    local container_ip
    container_ip=$(docker inspect --format '{{ .NetworkSettings.IPAddress }}' virtlet-build)

    # Sometimes we can reach rsync through localhost and a NAT'd port.  Other
    # times (when we are running in another docker container on the Jenkins
    # machines) we have to talk directly to the container IP.  There is no one
    # strategy that works in all cases so we test to figure out which situation we
    # are in.
    if rsync_probe 127.0.0.1 "${VIRTLET_RSYNC_PORT}"; then
        echo "127.0.0.1:${VIRTLET_RSYNC_PORT}" >"${project_dir}/_output/rsync_addr"
        return 0
    elif rsync_probe "${container_ip}" ${VIRTLET_RSYNC_PORT}; then
        echo "${container_ip}:${VIRTLET_RSYNC_PORT}" >"${project_dir}/_output/rsync_addr"
        return 0
    else
        echo "Could not probe the rsync port" >&2
    fi
}

function ensure_build_container {
    if ! docker ps --filter=label=virtlet_build | grep -q virtlet-build; then
        ensure_build_image
        cd "${project_dir}"
        # Need to mount docker socket into the container because of
        # CRI proxy deployment tests & building the image
        # We also pass --tmpfs /tmp because log tailing doesn't work
        # on overlayfs. This breaks 'go test' though unless we also
        # remount /tmp with exec option (it creates and runs executable files
        # under /tmp)
        declare -a docker_cert_args=()
        if [[ ${DOCKER_CERT_PATH:-} ]]; then
           docker_cert_args=(-e DOCKER_CERT_PATH=/docker-cert)
        fi
        docker run -d --privileged --net=host \
               -l virtlet_build \
               -v "virtlet_src:${remote_project_dir}" \
               -v "virtlet_pkg:/go/pkg" \
               -v /sys/fs/cgroup:/sys/fs/cgroup \
               -v /lib/modules:/lib/modules:ro \
               -v /boot:/boot:ro \
               -v "${DOCKER_SOCKET_PATH}:/var/run/docker.sock" \
               -e DOCKER_MACHINE_NAME="${DOCKER_MACHINE_NAME:-}" \
               -e DOCKER_TLS_VERIFY="${DOCKER_TLS_VERIFY:-}" \
               -e TRAVIS="${TRAVIS:-}" \
               -e TRAVIS_PULL_REQUEST="${TRAVIS_PULL_REQUEST:-}" \
               -e TRAVIS_BRANCH="${TRAVIS_BRANCH:-}" \
               -e CIRCLECI="${CIRCLECI:-}" \
               -e CIRCLE_PULL_REQUEST="${CIRCLE_PULL_REQUEST:-}" \
               -e CIRCLE_BRANCH="${CIRCLE_PULL_REQUEST:-}" \
               -e VIRTLET_ON_MASTER="${VIRTLET_ON_MASTER:-}" \
               -e VIRTLET_MULTI_NODE="${VIRTLET_MULTI_NODE:-}" \
               -e GITHUB_TOKEN="${GITHUB_TOKEN:-}" \
               -e MKDOCS_SERVE_ADDRESS="${MKDOCS_SERVE_ADDRESS:-}" \
               -e VIRTLET_SKIP_VENDOR="${VIRTLET_SKIP_VENDOR:-false}" \
               ${docker_cert_args[@]+"${docker_cert_args[@]}"} \
               --name virtlet-build \
               --tmpfs /tmp \
               "${build_image}" \
               /bin/bash -c "mount /tmp -o remount,exec && sleep Infinity" >/dev/null
        if [[ ! ${VIRTLET_SKIP_RSYNC} ]]; then
            # from build/common.sh in k8s
            mkdir -p "${project_dir}/_output"
            dd if=/dev/urandom bs=512 count=1 2>/dev/null | LC_ALL=C tr -dc 'A-Za-z0-9' | dd bs=32 count=1 2>/dev/null >"${rsync_pw_file}"
            chmod 600 "${rsync_pw_file}"

            docker cp "${rsync_pw_file}" virtlet-build:/rsyncd.password
            docker exec -d -i virtlet-build /rsyncd.sh "${VIRTLET_RSYNC_PORT}"
            get_rsync_addr
        fi
        if [[ ${DOCKER_CERT_PATH:-} ]]; then
            tar -C "${DOCKER_CERT_PATH}" -c . | docker exec -i virtlet-build /bin/bash -c 'mkdir /docker-cert && tar -C /docker-cert -x'
        fi
    fi
    if [[ ! ${VIRTLET_SKIP_RSYNC} ]]; then
        get_rsync_addr
        RSYNC_ADDR="$(cat "${project_dir}/_output/rsync_addr")"
    fi
}

function vsh {
    ensure_build_container
    cd "${project_dir}"
    docker exec -it virtlet-build env TERM=xterm bash
}

function sync_source {
    ensure_build_container
    cd "${project_dir}"
    if [[ ! ${VIRTLET_SKIP_RSYNC} ]]; then
        local -a filters=(
            --filter '- /_output/'
        )
        if [[ ${VIRTLET_SKIP_VENDOR:-false} != "true" ]]; then
            filters+=(--filter '- /vendor/')
        fi
        if [[ ! ${rsync_git} ]]; then
            filters+=(--filter '- /.git/')
        fi
        rsync "${filters[@]}" \
              --password-file "${project_dir}/_output/rsync.password" \
              -a --delete --compress-level=9 \
              "${project_dir}/" "rsync://virtlet@${RSYNC_ADDR}/virtlet/"
    fi
}

function vcmd {
    sync_source >&2
    local t=""
    if [[ ${USE_TERM:-} ]]; then
        t="t"
    fi
    docker exec -i"${t}" virtlet-build bash -c "$*"
}

function vcmd_simple {
    local cmd="${1}"
    docker exec -i virtlet-build bash -c "${cmd}"
}

function stop {
    docker ps -a -q --filter=label=virtlet_build | while read container_id; do
        echo >&2 "Removing container:" "${container_id}"
        docker rm -fv "${container_id}"
    done
}

function copy_output {
    ensure_build_container
    cd "${project_dir}"
    vcmd_simple "tar -C '${remote_project_dir}' -cz \$(find . -path '*/_output/*' -type f)" | tar -xvz
}

function copy_back {
    ensure_build_container
    cd "${project_dir}"
    tar -cz $(find . -path '*/_output/*' -type f | grep -v rsync) | vcmd_simple "tar -C '${remote_project_dir}' -xvz"
}

function copy_dind_internal {
    local virtlet_node="${1}"
    if ! docker volume ls -q | grep -q "^kubeadm-dind-${virtlet_node}$"; then
        echo "No active or snapshotted kubeadm-dind-cluster" >&2
        exit 1
    fi
    tar -C _output -c . |
        docker run -i --rm \
               -v "kubeadm-dind-${virtlet_node}:/dind" \
               --name ${tmp_container_name} \
               "${busybox_image}" \
               /bin/sh -c 'tar -C /dind -xv && chmod ug+s /dind/vmwrapper'
}

function kvm_ok {
    build_tag="$(cat "${project_dir}/_output/version")"
    # The check is done inside the virtlet node container because it
    # has proper /lib/modules from the docker host. Also, it'll have
    # to use the virtlet image later anyway.
    # Use kube-master node as all of the DIND nodes in the cluster are similar
    if ! docker exec kube-master docker run --privileged --rm -v /lib/modules:/lib/modules "${virtlet_image}:${build_tag}" kvm-ok; then
        return 1
    fi
}

function prepare_node {
    local node="${1}"
    if docker exec "${node}" dpkg-query -W criproxy-nodeps >&/dev/null; then
        return 0
    fi
    ensure_build_container
    echo >&2 "Installing CRI proxy package in the node container (${node})..."
    if [[ ${DIND_CRI:-} = containerd ]]; then
        docker exec "${node}" /bin/bash -c 'echo criproxy-nodeps criproxy/primary_cri select containerd | debconf-set-selections'
    fi
    docker exec "${node}" /bin/bash -c "curl -sSL '${CRIPROXY_DEB_URL}' >/criproxy.deb && dpkg -i /criproxy.deb && rm /criproxy.deb"

    docker exec "${node}" mount --make-shared /dind
    docker exec "${node}" mount --make-shared /dev
    docker exec "${node}" mount --make-shared /boot
    docker exec "${node}" mount --make-shared /sys/fs/cgroup

    if [[ ${VIRTLET_ON_MASTER} ]]; then
        if [[ $(kubectl get node kube-master -o jsonpath='{.spec.taints[?(@.key=="node-role.kubernetes.io/master")]}') ]]; then
            kubectl taint nodes kube-master node-role.kubernetes.io/master-
        fi
    fi
    if [[ ${FORCE_UPDATE_IMAGE} ]] || ! docker exec "${node}" docker history -q "${virtlet_image}:latest" >&/dev/null; then
        echo >&2 "Propagating Virtlet image to the node container..."
        if [[ ${DIND_CRI} = containerd ]]; then
          vcmd "docker save '${virtlet_image}:latest' | docker exec -i '${node}' ctr -n k8s.io images import -"
        else
          vcmd "docker save '${virtlet_image}:latest' | docker exec -i '${node}' docker load"
        fi
    fi
}

function prepare_all_nodes {
    for node in $(kubectl get nodes -o jsonpath='{.items[?(@.metadata.name!="kube-master")].metadata.name}'); do
        prepare_node "${node}"
    done
}

function apply_runtime_label {
    local node="${1}"
    kubectl label node --overwrite "${node}" extraRuntime=virtlet
}

function start_dind {
    local -a virtlet_config=(--from-literal=image_regexp_translation="${IMAGE_REGEXP_TRANSLATION}")
    if ! kvm_ok || [[ ${VIRTLET_DISABLE_KVM:-} ]]; then
        virtlet_config+=(--from-literal=disable_kvm=y)
    fi
    kubectl create configmap -n kube-system virtlet-config "${virtlet_config[@]}"
    kubectl create configmap -n kube-system virtlet-image-translations --from-file "${project_dir}/deploy/images.yaml"
    start_virtlet
}

function start_virtlet {
    local -a opts=(--dev)
    if kubectl version | tail -n1 | grep -q 'v1\.7\.'; then
        # apply mount propagation hacks for 1.7
        opts+=(--compat)
    fi
    docker exec virtlet-build "${remote_project_dir}/_output/virtletctl" gen "${opts[@]}" |
        kubectl apply -f -
}

function virtlet_subdir {
    local dir="${1:-$(pwd)}"
    local prefix="${project_dir}/"
    if [[ ${#dir} -lt ${#prefix} || ${dir:0:${#prefix}} != ${prefix} ]]; then
        echo >&2 "must be in a project subdir"
        exit 1
    fi
    echo -n "${dir:${#prefix}}"
}

function clean {
    stop
    docker volume rm virtlet_src || true
    docker volume rm virtlet_pkg || true
    docker rmi "${build_image}" || true
    # find command may produce zero results
    # -exec rm -rf '{}' ';' produces errors when trying to
    # enter deleted directories
    find . -name _output -type d | while read dir; do
        rm -rf "${dir}"
    done
}

function gotest {
    # FIXME: exit 1 in $(virtlet_subdir) doesn't cause the script to exit
    virtlet_subdir >/dev/null
    subdir="$(virtlet_subdir)"
    if ! vcmd "cd '${subdir}' && go test $*"; then
        vcmd_simple "find . -name 'Test*.out.*' | xargs tar -c -T -" | tar -C "${project_dir}" -x
        exit 1
    fi
}

function gobuild {
    # FIXME: exit 1 in $(virtlet_subdir) doesn't cause the script to exit
    virtlet_subdir >/dev/null
    # -gcflags -e removes the limit on error message count, which helps
    # with using it for syntax checking
    vcmd "cd '$(virtlet_subdir)' && go build -gcflags -e $*"
}

function build_image_internal {
    build_internal
    build_tag="$(cat "${project_dir}/_output/version")"
    tar -c _output -C "${project_dir}/images" image_skel/ Dockerfile.virtlet |
        docker build -t "${virtlet_image}:${build_tag}" --build-arg VIRLET_BASE_TAG=build_tag \
          -f Dockerfile.virtlet -
    docker push "${virtlet_image}:${build_tag}"
    docker tag "${virtlet_image}:${build_tag}" "${virtlet_image}:latest"
    docker push "${virtlet_image}:latest"
}

function install_vendor_internal {
    if [ ! -d vendor ]; then
        glide install --strip-vendor
    fi
}

function run_tests_internal {
    install_vendor_internal
    go test -v ./pkg/... ./tests/network/...
}

function run_integration_internal {
    install_vendor_internal
    ( cd tests/integration && ./go.test )
}

function get_ldflags {
    # XXX: use kube::version::ldflag (-ldflags -X package.Var=...)
    # see also versioning.mk in helm
    # https://stackoverflow.com/questions/11354518/golang-application-auto-build-versioning
    # see pkg/version/version.go in k8s
    # for GoVersion / Compiler / Platform
    local vfile="${project_dir}/pkg/version/version.go"
    local git_version="$(git describe --tags --abbrev=14 'HEAD^{commit}' | sed "s/-g\([0-9a-f]\{14\}\)$/+\1/")"
    local git_commit="$(git rev-parse "HEAD^{commit}")"
    local git_tree_state=$([[ $(git status --porcelain) ]] && echo "dirty" || echo "clean")
    if [[ ${git_tree_state} == dirty ]]; then
        git_version+="-dirty"
    fi
    local build_date="$(date -u +'%Y-%m-%dT%H:%M:%SZ')"
    local git_major=""
    local git_minor=""
    local version_pkg="${go_package}/pkg/version"
    local ldflags=(-X "${version_pkg}.gitVersion=${git_version}"
                   -X "${version_pkg}.gitCommit=${git_commit}"
                   -X "${version_pkg}.gitTreeState=${git_tree_state}"
                   -X "${version_pkg}.buildDate=${build_date}")
    if [[ ${git_version} =~ ^v([0-9]+)\.([0-9]+)(\.[0-9]+)?([-].*)?([+].*)?$ ]]; then
        git_major=${BASH_REMATCH[1]}
        git_minor=${BASH_REMATCH[2]}
        ldflags+=(-X "${version_pkg}.gitMajor=${git_major}"
                  -X "${version_pkg}.gitMinor=${git_minor}")
    fi
    if [[ ${SET_VIRTLET_IMAGE_TAG:-} ]]; then
        ldflags+=(-X "${version_pkg}.imageTag=${SET_VIRTLET_IMAGE_TAG}")
    fi
    echo "${ldflags[*]}"
}

function build_internal {
    # we don't just always generate the bindata right there because we
    # want to keep the source buildable outside this build container.
    go-bindata -mode 0644 -o /tmp/bindata.go -modtime "${bindata_modtime}" -pkg "${bindata_pkg}" "${bindata_dir}"
    if ! cmp /tmp/bindata.go "${bindata_out}"; then
        echo >&2 "${bindata_dir} changed, please re-run ${0} update-bindata"
        exit 1
    fi
    install_vendor_internal
    ldflags="$(get_ldflags)"
    mkdir -p "${project_dir}/_output"
    go build -i -o "${project_dir}/_output/virtlet" -ldflags "${ldflags}" ./cmd/virtlet
    go build -i -o "${project_dir}/_output/virtletctl" -ldflags "${ldflags}" ./cmd/virtletctl
    GOOS=darwin go build -i -o "${project_dir}/_output/virtletctl.darwin" -ldflags "${ldflags}" ./cmd/virtletctl
    go build -i -o "${project_dir}/_output/vmwrapper" ./cmd/vmwrapper
    go build -i -o "${project_dir}/_output/flexvolume_driver" ./cmd/flexvolume_driver
    go test -i -c -o "${project_dir}/_output/virtlet-e2e-tests" ./tests/e2e
    go build -i -o "${project_dir}/_output/virtlet-longevity-tests" -ldflags "${ldflags}" ./cmd/longevity
    get_version
}

function release_description {
    local -a tag="${1}"
    shift
    git tag -l --format='%(contents:body)' "${tag}"
    echo
    echo "SHA256 sums for the files:"
    echo '```'
    (cd _output && sha256sum "$@")
    echo '```'
}

function release_internal {
    local tag="${1}"
    local gh_user="Mirantis"
    if [[ ${tag} =~ test ]]; then
        gh_user="${GH_RELEASE_TEST_USER}"
    fi
    local -a opts=(--user "${gh_user}" --repo virtlet --tag "${tag}")
    local -a files=(virtletctl virtletctl.darwin)
    local description="$(release_description "${tag}" "${files[@]}")"
    local pre_release=
    if [[ ${tag} =~ -(test|pre).*$ ]]; then
        pre_release="--pre-release"
    fi
    if github-release --quiet delete "${opts[@]}"; then
        echo >&2 "Replacing the old Virtlet release"
    fi
    github-release release "${opts[@]}" \
                   --name "$(git tag -l --format='%(contents:subject)' "${tag}")" \
                   --description "${description}" \
                   ${pre_release}
    for filename in "${files[@]}"; do
        echo >&2 "Uploading: ${filename}"
        github-release upload "${opts[@]}" \
                       --name "${filename}" \
                       --replace \
                       --file "_output/${filename}"
    done
}

function e2e {
    ensure_build_container
    local cluster_url
    cluster_url="$(kubectl config view -o jsonpath='{.clusters[?(@.name=="dind")].cluster.server}')"
    docker exec virtlet-build _output/virtlet-e2e-tests -include-unsafe-tests=true -cluster-url "${cluster_url}" "$@"
}

function update_bindata_internal {
    # set fixed modtime to avoid unwanted differences during the checks
    # that are done by build/cmd.sh build
    go-bindata -mode 0644 -modtime "${bindata_modtime}" -o "${bindata_out}" -pkg "${bindata_pkg}" "${bindata_dir}"
}

function update_generated_docs_internal {
  if [[ ! -f _output/virtletctl ]]; then
    echo >&2 "Please run build/cmd.sh build first"
  fi
  virtletctl gendoc docs/docs/reference
  tempfile="$(tempfile)"
  _output/virtletctl gendoc --config >"${tempfile}"
  sed -i "/<!-- begin -->/,/<!-- end -->/{
//!d
/begin/r ${tempfile}
}" docs/docs/reference/config.md
  rm -f "${tempfile}"
}

function update_generated_internal {
  install_vendor_internal
  vendor/k8s.io/code-generator/generate-groups.sh all \
    github.com/Equinix-Metal/virtlet/pkg/client github.com/Equinix-Metal/virtlet/pkg/api \
    virtlet.k8s:v1 \
    --go-header-file "build/custom-boilerplate.go.txt"
  # fix import url case issues
  find pkg/client \
       -name '*.go' \
       -exec sed -i 's@github\.com/Equinix/virtlet@github\.com/Equinix/virtlet@g' \
       '{}' \;
}

function serve_docs_internal {
    (cd docs && mkdocs serve -a "${MKDOCS_SERVE_ADDRESS}")
}

function build_docs_internal {
    site_dir="$(mktemp -d)"
    trap 'rm -rf "${site_dir}"' EXIT
    # Use strict mode (-s) for mkdocs so that any broken links
    # etc. are caught
    (cd docs && mkdocs build -s -d "${site_dir}" >&2)
    tar -C "${site_dir}" -c .
}

function build_docs {
    cd "${project_dir}"
    rm -rf _docs
    git clone -b docs . _docs
    local docs_hash="$(git ls-tree HEAD -- docs | awk '{print $3}')"
    if [[ ! -e _docs/source_hash || ${docs_hash} != $(cat _docs/source_hash) ]]; then
        echo >&2 "docs/ directory changed since the last doc build, rebuilding docs"
    elif [[ $(git status --porcelain) ]]; then
        echo >&2 "Source directory dirty, rebuilding docs"
    else
        echo >&2 "Docs unchanged, no need to rebuild"
        return 0
    fi
    # clean up _docs except for .git and CNAME
    find _docs -name .git -prune -o -type f \! -name CNAME -exec rm -f '{}' \;
    vcmd "build/cmd.sh build-docs-internal" | tar -C _docs -xv
    echo "${docs_hash}" > _docs/source_hash
    (
        cd _docs
        git add .
        git commit -m "Update generated docs [ci skip]"
        # this pushes the changes into the local repo (not github!)
        git push origin docs
    )
}

function usage {
    echo >&2 "Usage:"
    echo >&2 "  $0 build"
    echo >&2 "  $0 test"
    echo >&2 "  $0 copy"
    echo >&2 "  $0 copy-dind"
    echo >&2 "  $0 start-dind"
    echo >&2 "  $0 vsh"
    echo >&2 "  $0 stop"
    echo >&2 "  $0 clean"
    echo >&2 "  $0 update-bindata"
    echo >&2 "  $0 update-generated-docs"
    echo >&2 "  $0 gotest [TEST_ARGS...]"
    echo >&2 "  $0 gobuild [BUILD_ARGS...]"
    echo >&2 "  $0 run CMD..."
    echo >&2 "  $0 release TAG"
    echo >&2 "  $0 serve-docs"
    echo >&2 "  $0 build-docs"
    echo >&2 "  $0 sync"
    exit 1
}

cmd="${1:-}"
if [[ ! $cmd ]]; then
    usage
fi
shift

case "${cmd}" in
    gotest)
        gotest "$@"
        ;;
    gobuild)
        rsync_git=
        gobuild "$@"
        ;;
    prepare-vendor)
        vcmd "build/cmd.sh install-vendor-internal"
        ;;
    build)
        vcmd "SET_VIRTLET_IMAGE_TAG='${SET_VIRTLET_IMAGE_TAG:-}' build/cmd.sh build-image-internal"
        ;;
    build-image-internal)
        # this is executed inside the container
        build_image_internal "$@"
        ;;
    test)
        vcmd 'build/cmd.sh run-tests-internal'
        ;;
    integration)
        vcmd 'build/cmd.sh run-integration-internal'
        ;;
    install-vendor-internal)
        install_vendor_internal
        ;;
    run-tests-internal)
        run_tests_internal
        ;;
    run-integration-internal)
        run_integration_internal
        ;;
    update-bindata)
        vcmd "build/cmd.sh update-bindata-internal"
        docker cp "virtlet-build:${remote_project_dir}/pkg/tools/bindata.go" pkg/tools/bindata.go
        ;;
    update-bindata-internal)
        update_bindata_internal
        ;;
    update-generated)
        vcmd "build/cmd.sh update-generated-internal"
        docker exec virtlet-build \
               /bin/bash -c \
               "tar -C '${remote_project_dir}' -c $(find pkg/ -name 'zz_generated.*') pkg/client" |
            tar -C "${project_dir}" -xv
        ;;
    update-generated-internal)
        update_generated_internal
        ;;
    update-generated-docs)
        vcmd "build/cmd.sh update-generated-docs-internal"
        docker exec virtlet-build tar -C "${remote_project_dir}" -c docs/docs/reference/config.md docs/docs/reference/virtletctl.md | tar -C "${project_dir}" -xv
        ;;
    update-generated-docs-internal)
        update_generated_docs_internal
        ;;
    run)
        vcmd "$*"
        ;;
    vsh)
        vsh
        ;;
    stop)
        stop
        ;;
    clean)
        clean
        ;;
    copy)
        copy_output
        ;;
    copy-back)
        copy_back
        ;;
    copy-dind)
        VIRTLET_SKIP_RSYNC=y vcmd "build/cmd.sh copy-dind-internal"
        ;;
    e2e)
        e2e "$@"
        ;;
    copy-dind-internal)
        for virtlet_node in "${virtlet_nodes[@]}"; do
            copy_dind_internal "${virtlet_node}"
        done
        ;;
    prepare-all-nodes)
        prepare_all_nodes
        ;;
    start-dind)
        for virtlet_node in "${virtlet_nodes[@]}"; do
            prepare_node "${virtlet_node}"
            apply_runtime_label "${virtlet_node}"
        done
        start_dind
        ;;
    start-build-container)
        ensure_build_container
        ;;
    release)
        if [[ ! ${1:-} ]]; then
            echo >&2 "must specify the tag"
            exit 1
        fi
        ( vcmd "build/cmd.sh release-internal '${1}'" )
        ;;
    release-internal)
        release_internal "$@"
        ;;
    serve-docs-internal)
        serve_docs_internal
        ;;
    serve-docs)
        ( USE_TERM=1 vcmd "build/cmd.sh serve-docs-internal" )
        ;;
    build-docs-internal)
        build_docs_internal
        ;;
    build-docs)
        build_docs
        ;;
    sync)
        sync_source
        ;;
    *)
        usage
        ;;
esac

# TODO: make it possible to run e2e from within the build container, too
# (although we don't need to use that for CircleCI)
# TODO: fix indentation in this file (use 2 spaces)