#!/bin/sh

set -e
PROJECT_ROOT="spec/integration"
PROJECT_NAME=rundock_spec
PLATFORM_NAME=centos6
PLATFORM_DIR="${PROJECT_ROOT}/platforms/${PLATFORM_NAME}"
DOCKER_IMAGE_NAME="${PROJECT_NAME}/${PLATFORM_NAME}"
DOCKER_BRIDGE_ADDRESS=`ip addr show | grep "global docker0" | awk '{print $2}' | cut -d'/' -f1`
DOCKER_BRIDGE_SSH_PORT=22222
DOCKER_CACHE_DIR="${HOME}/docker"
DOCKER_CACHE_IMAGE_PATH="${DOCKER_CACHE_DIR}/${PLATFORM_NAME}.tar"
DOCKER_SSH_USER=tester
DOCKER_SSH_KEY_PRIVATE="${HOME}/.ssh/id_rsa_${PROJECT_NAME}_${PLATFORM_NAME}_tmp"
DOCKER_SSH_KEY_PUBLIC_LOCAL="${HOME}/.ssh/id_rsa_${PROJECT_NAME}_${PLATFORM_NAME}_tmp.pub"
DOCKER_SSH_KEY_PUBLIC_REMOTE="${PLATFORM_DIR}/authorized_keys"
DOCKER_SSH_CONFIG="${HOME}/.ssh/config_${PROJECT_NAME}_${PLATFORM_NAME}"
RUNDOCK_SCENARIO_DIR="${PROJECT_ROOT}/scenarios"
RUNDOCK_GROUP_DIR="${PROJECT_ROOT}/groups"
RUNDOCK_TARGET_DIR="${PROJECT_ROOT}/targets"
RUNDOCK_CACHE_DIR="${HOME}/.rundock/${PLATFORM_NAME}"
RUNDOCK_DEFAULT_SSH_YML="${RUNDOCK_CACHE_DIR}/integration_default_ssh.yml"
RUNDOCK_SCENARIO_CACHE_DIR="${RUNDOCK_CACHE_DIR}/scenarios"
RUNDOCK_GROUP_CACHE_DIR="${RUNDOCK_CACHE_DIR}/groups"
RUNDOCK_TARGET_CACHE_DIR="${RUNDOCK_CACHE_DIR}/targets"

if [ "${1}x" = "--cleanx" ];then
  set +e
  rm -f ${RUNDOCK_DEFAULT_SSH_YML}
  rm -f ${RUNDOCK_SCENARIO_CACHE_DIR}/*.yml
  rm -f ${RUNDOCK_GROUP_CACHE_DIR}/*.yml
  rm -f ${RUNDOCK_TARGET_CACHE_DIR}/*.yml
  if sudo docker ps | grep "${DOCKER_IMAGE_NAME}" > /dev/null; then
    rm -f ${DOCKER_CACHE_IMAGE_PATH}
    rm -f ${DOCKER_SSH_KEY_PRIVATE}
    rm -f ${DOCKER_SSH_KEY_PUBLIC_LOCAL}
    rm -f ${DOCKER_SSH_CONFIG}
    rm -f ${DOCKER_SSH_KEY_PUBLIC_REMOTE}
    sudo docker ps -q | xargs sudo docker rm -f
  fi

  exit 0
fi

mkdir -p "${RUNDOCK_SCENARIO_CACHE_DIR}"
mkdir -p "${RUNDOCK_GROUP_CACHE_DIR}"
mkdir -p "${RUNDOCK_TARGET_CACHE_DIR}"

if [ ! -f ${RUNDOCK_DEFAULT_SSH_YML} ]; then
(
cat << EOP
:port: ${DOCKER_BRIDGE_SSH_PORT}
:paranoid: false
:user: "${DOCKER_SSH_USER}"
:keys: ["${DOCKER_SSH_KEY_PRIVATE}"]
EOP
) > ${RUNDOCK_DEFAULT_SSH_YML}
fi

cp ${RUNDOCK_SCENARIO_DIR}/* ${RUNDOCK_SCENARIO_CACHE_DIR}
cp ${RUNDOCK_GROUP_DIR}/* ${RUNDOCK_GROUP_CACHE_DIR}
cp ${RUNDOCK_TARGET_DIR}/* ${RUNDOCK_TARGET_CACHE_DIR}

find ${RUNDOCK_SCENARIO_CACHE_DIR} -type f -name "*_scenario.yml" | \
  xargs sed -i -e "s#<replaced_by_platforms_key>#${DOCKER_SSH_KEY_PRIVATE}#g"
find ${RUNDOCK_GROUP_CACHE_DIR} -type f -name "*_group.yml" | \
  xargs sed -i -e "s#<replaced_by_platforms_key>#${DOCKER_SSH_KEY_PRIVATE}#g"
find ${RUNDOCK_TARGET_CACHE_DIR} -type f -name "*.yml" | \
  xargs sed -i -e "s#<replaced_by_platforms_key>#${DOCKER_SSH_KEY_PRIVATE}#g"
find ${RUNDOCK_SCENARIO_CACHE_DIR} -type f -name "*_scenario.yml" | \
  xargs sed -i -e "s#<replaced_by_platforms_host>#${DOCKER_BRIDGE_ADDRESS}#g"
find ${RUNDOCK_GROUP_CACHE_DIR} -type f -name "*_group.yml" | \
  xargs sed -i -e "s#<replaced_by_platforms_host>#${DOCKER_BRIDGE_ADDRESS}#g"
find ${RUNDOCK_TARGET_CACHE_DIR} -type f -name "*.yml" | \
  xargs sed -i -e "s#<replaced_by_platforms_host>#${DOCKER_BRIDGE_ADDRESS}#g"
find ${RUNDOCK_SCENARIO_CACHE_DIR} -type f -name "*_scenario.yml" | \
  xargs sed -i -e "s#<replaced_by_platforms_port>#${DOCKER_BRIDGE_SSH_PORT}#g"
find ${RUNDOCK_GROUP_CACHE_DIR} -type f -name "*_group.yml" | \
  xargs sed -i -e "s#<replaced_by_platforms_port>#${DOCKER_BRIDGE_SSH_PORT}#g"
find ${RUNDOCK_TARGET_CACHE_DIR} -type f -name "*.yml" | \
  xargs sed -i -e "s#<replaced_by_platforms_port>#${DOCKER_BRIDGE_SSH_PORT}#g"

sudo docker ps | grep "${DOCKER_IMAGE_NAME}" && { echo "docker image is already standing."; exit 0; }

yes | ssh-keygen -N "" -t rsa -f ${DOCKER_SSH_KEY_PRIVATE}
cp ${DOCKER_SSH_KEY_PUBLIC_LOCAL} ${DOCKER_SSH_KEY_PUBLIC_REMOTE}

if sudo docker ps | grep "${DOCKER_IMAGE_NAME}" > /dev/null 2>&1; then
  sudo docker ps -q | xargs sudo docker rm -f > /dev/null
fi

if file ${DOCKER_CACHE_IMAGE_PATH} | grep empty; then
  sudo docker load --input ${DOCKER_CACHE_IMAGE_PATH}
fi

sudo docker build -t "${DOCKER_IMAGE_NAME}" ${PLATFORM_DIR}
rm -f ${DOCKER_SSH_KEY_PUBLIC_REMOTE}
mkdir -p ${DOCKER_CACHE_DIR}
sudo docker save "${DOCKER_IMAGE_NAME}" > ${DOCKER_CACHE_IMAGE_PATH}
sudo docker run -d --privileged -p ${DOCKER_BRIDGE_ADDRESS}:${DOCKER_BRIDGE_SSH_PORT}:22 "${DOCKER_IMAGE_NAME}"

echo "Host ${PLATFORM_NAME}"                          >  $DOCKER_SSH_CONFIG
echo "        HostName ${DOCKER_BRIDGE_ADDRESS}"      >> $DOCKER_SSH_CONFIG
echo "        User     ${DOCKER_SSH_USER}"            >> $DOCKER_SSH_CONFIG
echo "        Port     ${DOCKER_BRIDGE_SSH_PORT}"     >> $DOCKER_SSH_CONFIG
echo "        IdentityFile ${DOCKER_SSH_KEY_PRIVATE}" >> $DOCKER_SSH_CONFIG
echo "        StrictHostKeyChecking no"               >> $DOCKER_SSH_CONFIG