# Copyright (C) 2012 - 2018 Cloudera, Inc.
# All Rights Reserved.

#!/usr/bin/env bash

# This helper script is designed to be sourced, at which time its functions are made available
# to the user. Most functions defined have functionality defined by environmental variables, which
# can be set during invocation. For example,
#
# CLUSTERDOCK_PULL=false clusterdock_run ./bin/...

## @description  Run a clusterdock Python script within a Docker container
## @audience     public
## @stability    stable
## @replaceable  no
## @param        Python script to run relative to the ./dev-support/clusterdock/clusterdock folder
clusterdock_run() {
  # Supported environmental variables:
  # - CLUSTERDOCK_DOCKER_REGISTRY_INSECURE: whether the Docker registry is insecure (either true or
  #                                         false)
  # - CLUSTERDOCK_DOCKER_REGISTRY_USERNAME: username needed to login to any secure Docker registry
  # - CLUSTERDOCK_DOCKER_REGISTRY_PASSWORD: password needed to login to any secure Docker registry
  # - CLUSTERDOCK_IMAGE: the Docker image from which to start the clusterdock framework container
  # - CLUSTERDOCK_PULL: whether to pull the clusterdock image and CLUSTERDOCK_TOPOLOGY_IMAGE, if
  #                     specified (either true or false; defaults to true)
  # - CLUSTERDOCK_TARGET_DIR: a folder on the host to mount into /root/target in the clusterdock
  #                           container
  # - CLUSTERDOCK_TOPOLOGY_IMAGE: a Docker image to mount into clusterdock's topology folder and
  #                               make available to users

  if [ -z "${CLUSTERDOCK_IMAGE}" ]; then
    local CONSTANTS_CONFIG_URL='https://raw.githubusercontent.com/cloudera/clusterdock/master/clusterdock/constants.cfg'

    # awk -F argument allows for any number of spaces around equal sign.
    local DOCKER_REGISTRY_URL=$(curl -s "${CONSTANTS_CONFIG_URL}" \
        | awk -F " *= *" '/^docker_registry_url/ {print $2}')
    local CLOUDERA_NAMESPACE=$(curl -s "${CONSTANTS_CONFIG_URL}" \
        | awk -F " *= *" '/^cloudera_namespace/ {print $2}')

    CLUSTERDOCK_IMAGE="${DOCKER_REGISTRY_URL}/${CLOUDERA_NAMESPACE}/clusterdock:latest"
  fi

  if [ "${CLUSTERDOCK_PULL}" != "false" ]; then
    sudo docker pull "${CLUSTERDOCK_IMAGE}" &> /dev/null
  fi

  if [ -n "${CLUSTERDOCK_TARGET_DIR}" ]; then
    local TARGET_DIR_MOUNT="-v ${CLUSTERDOCK_TARGET_DIR}:/root/target"
  fi

  if [ -n "${CLUSTERDOCK_DOCKER_REGISTRY_INSECURE}" ]; then
    local REGISTRY_INSECURE="-e DOCKER_REGISTRY_INSECURE=${CLUSTERDOCK_DOCKER_REGISTRY_INSECURE}"
  fi

  if [ -n "${CLUSTERDOCK_DOCKER_REGISTRY_USERNAME}" ]; then
    local REGISTRY_USERNAME="-e DOCKER_REGISTRY_USERNAME=${CLUSTERDOCK_DOCKER_REGISTRY_USERNAME}"
  fi

  if [ -n "${CLUSTERDOCK_DOCKER_REGISTRY_PASSWORD}" ]; then
    local REGISTRY_PASSWORD="-e DOCKER_REGISTRY_PASSWORD=${CLUSTERDOCK_DOCKER_REGISTRY_PASSWORD}"
  fi

  if [ -n "${CLUSTERDOCK_TOPOLOGY_IMAGE}" ]; then
    if [ "${CLUSTERDOCK_PULL}" != "false" ]; then
      sudo docker pull "${CLUSTERDOCK_TOPOLOGY_IMAGE}" &> /dev/null
    fi

    local TOPOLOGY_CONTAINER_ID=$(sudo docker create "${CLUSTERDOCK_TOPOLOGY_IMAGE}")
    local TOPOLOGY_VOLUME="--volumes-from=${TOPOLOGY_CONTAINER_ID}"
  fi

  # The /etc/hosts bind-mount allows clusterdock to update /etc/hosts on the host machine for
  # better access to internal container addresses.
  sudo docker run --net=host -t \
      --privileged \
      ${TARGET_DIR_MOUNT} \
      ${TOPOLOGY_VOLUME} \
      ${REGISTRY_INSECURE} \
      ${REGISTRY_USERNAME} \
      ${REGISTRY_PASSWORD} \
      -v /tmp/clusterdock \
      -v /etc/hosts:/etc/hosts \
      -v /etc/localtime:/etc/localtime \
      -v /var/run/docker.sock:/var/run/docker.sock \
      "${CLUSTERDOCK_IMAGE}" $@

  if [ -n "${TOPOLOGY_CONTAINER_ID}" ]; then
    sudo docker rm -v "${TOPOLOGY_CONTAINER_ID}" &> /dev/null
  fi
}

## @description  SSH to a clusterdock-created container cluster node
## @audience     public
## @stability    stable
## @replaceable  no
## @param        Fully-qualified domain name of container to which to connect
## @param        An optional command to run
clusterdock_ssh() {
  local NODE=${1}
  # Shift away arguments by 1. If anything is left, it's a command to run over SSH.
  shift 1

  if [ -z "${CLUSTERDOCK_IMAGE}" ]; then
    local CONSTANTS_CONFIG_URL='https://raw.githubusercontent.com/cloudera/clusterdock/master/clusterdock/constants.cfg'

    # awk -F argument allows for any number of spaces around equal sign.
    local DOCKER_REGISTRY_URL=$(curl -s "${CONSTANTS_CONFIG_URL}" \
        | awk -F " *= *" '/^docker_registry_url/ {print $2}')
    local CLOUDERA_NAMESPACE=$(curl -s "${CONSTANTS_CONFIG_URL}" \
        | awk -F " *= *" '/^cloudera_namespace/ {print $2}')

    CLUSTERDOCK_IMAGE="${DOCKER_REGISTRY_URL}/${CLOUDERA_NAMESPACE}/clusterdock:latest"
  fi

  if [ "${CLUSTERDOCK_PULL}" != "false" ]; then
    sudo docker pull "${CLUSTERDOCK_IMAGE}" &> /dev/null
  fi

  if [ -n "${CLUSTERDOCK_TOPOLOGY_IMAGE}" ]; then
    if [ "${CLUSTERDOCK_PULL}" != "false" ]; then
      sudo docker pull "${CLUSTERDOCK_TOPOLOGY_IMAGE}" &> /dev/null
    fi

    local TOPOLOGY_CONTAINER_ID=$(sudo docker create "${CLUSTERDOCK_TOPOLOGY_IMAGE}")
    local TOPOLOGY_VOLUME="--volumes-from=${TOPOLOGY_CONTAINER_ID}"
  fi

  local ID
  for ID in $(docker ps -qa); do
    if [ "$(docker inspect --format '{{.Config.Hostname}}' ${ID})" = "${NODE}" ]; then
      break
    fi
  done

  # In order to get a proper login shell with expected files sourced (e.g. /etc/environment), we
  # get into the container using docker exec running ssh (instead of Bash).
  sudo docker exec -it ${ID} ssh localhost "$*"
}