#!/bin/sh
set -eu

if [ -z "${INPUT_REMOTE_DOCKER_PORT+x}" ]; then
  INPUT_REMOTE_DOCKER_PORT=22
fi

if [ -z "${INPUT_REMOTE_DOCKER_HOST+x}" ]; then
    echo "Input remote_docker_host is required!"
    exit 1
fi

if [ -z "${INPUT_REMOTE_DOCKER_USER+x}" ]; then
    echo "Input remote_docker_user is required!"
    exit 1
fi

if [ -z "${INPUT_SSH_PRIVATE_KEY+x}" ]; then
    echo "Input ssh_private_key is required!"
    exit 1
fi

if [ -z "${INPUT_ARGS+x}" ]; then
  echo "Input input_args is required!"
  exit 1
fi

CHANGE_DIR=true
if [ -z "${INPUT_DEPLOY_PATH+x}" ]; then
  CHANGE_DIR=false
fi
if $CHANGE_DIR ; then
  echo "Change working directory"
  echo "Current: `pwd`"
  echo "Target: $INPUT_DEPLOY_PATH"
  mkdir -p $INPUT_DEPLOY_PATH
  cp -rfp ./. $INPUT_DEPLOY_PATH
  cd $INPUT_DEPLOY_PATH
fi

if [ -z "${INPUT_DOCKER_REGISTRY_URI+x}" ]; then
  INPUT_DOCKER_REGISTRY_URI=https://registry.hub.docker.com
fi

echo "Registering SSH keys..."

# register the private key with the agent.
mkdir -p ~/.ssh
ls ~/.ssh
printf '%s\n' "$INPUT_SSH_PRIVATE_KEY" > ~/.ssh/id_rsa
chmod 600 ~/.ssh/id_rsa
eval $(ssh-agent)
ssh-add ~/.ssh/id_rsa

SSH_PROXY=true
if [ -z "${INPUT_SSH_PROXY_CMD+x}" ]; then
  SSH_PROXY=false
fi
echo "Add REMOTE_DOCKER_HOST alias to ~/.ssh/config"
touch ~/.ssh/config
echo >> ~/.ssh/config
echo "Host REMOTE_DOCKER_HOST" >> ~/.ssh/config
echo "        HostName $INPUT_REMOTE_DOCKER_HOST" >> ~/.ssh/config
echo "        User $INPUT_REMOTE_DOCKER_USER" >> ~/.ssh/config
echo "        Port $INPUT_REMOTE_DOCKER_PORT" >> ~/.ssh/config
echo "        IdentityFile ~/.ssh/id_rsa" >> ~/.ssh/config
echo "        StrictHostKeyChecking no" >> ~/.ssh/config
echo "        UserKnownHostsFile /dev/null" >> ~/.ssh/config
if $SSH_PROXY ; then
  echo "Add ProxyCommand: $INPUT_SSH_PROXY_CMD"
  echo "        ProxyCommand $INPUT_SSH_PROXY_CMD" >> ~/.ssh/config
fi
echo >> ~/.ssh/config
chmod 600 ~/.ssh/config

set context
echo "Create docker context"
docker context create remote --docker "host=ssh://REMOTE_DOCKER_HOST"
docker context use remote

if ! [ "x${INPUT_DOCKER_REGISTRY_USERNAME}"=="x" ] && ! [ "x${INPUT_DOCKER_REGISTRY_PASSWORD}"=="x" ]; then
  echo "Connecting to $INPUT_DOCKER_REGISTRY_URI... Command: docker login"
  echo "$INPUT_DOCKER_REGISTRY_PASSWORD" | docker login -u "$INPUT_DOCKER_REGISTRY_USERNAME" --password-stdin "$INPUT_DOCKER_REGISTRY_URI"
fi

if ! [ -z "${INPUT_DOCKER_PRUNE+x}" ] && [ $INPUT_DOCKER_PRUNE = 'true' ] ; then
  yes | docker --log-level debug --host "ssh://$INPUT_REMOTE_DOCKER_HOST:$INPUT_REMOTE_DOCKER_PORT" system prune -a 2>&1
fi

DEPLOYMENT_COMMAND="docker --log-level=debug compose"

if ! [ "x${INPUT_STACK_FILE_NAME}"=="x" ]; then
  DEPLOYMENT_COMMAND="$DEPLOYMENT_COMMAND -f ${INPUT_STACK_FILE_NAME//,/ -f }"
fi

if ! [ "x${INPUT_ENV_FILE_NAME}"=="x" ]; then
  DEPLOYMENT_COMMAND="$DEPLOYMENT_COMMAND --env-file ${INPUT_ENV_FILE_NAME//,/ --env-file }"
fi

# DEPLOYMENT_COMMAND_OPTIONS=""

echo "Connecting to $INPUT_REMOTE_DOCKER_HOST... Command: ${DEPLOYMENT_COMMAND} ${INPUT_ARGS}"
${DEPLOYMENT_COMMAND} ${INPUT_ARGS} 2>&1