echo "Preparing inventory file"

if [ "$ENV_NAME" = "integration" ]; then
    echo -e "[nomad_servers]\nnomad.p81zero.int.corp.$DOMAIN_NAME ansible_user=ubuntu" > inventory
elif [ "$ENV_NAME" = "staging" ]; then
    echo -e "[nomad_servers]\nc0nomad.p81zero.corp.$DOMAIN_NAME ansible_user=ubuntu" > inventory
elif [ "$ENV_NAME" = "production" ]; then
    echo -e "[nomad_servers]\nc1nomad.p81zero.corp.$DOMAIN_NAME ansible_user=ubuntu" > inventory
else
    echo -e "[nomad_servers]\nnomad.p81zero.$ENV_NAME.corp.$DOMAIN_NAME ansible_user=ubuntu" > inventory
fi
cat inventory

echo "Adding ssh keys"
ssh-keyscan -t ecdsa,ed25519 -H github.com >> ~/.ssh/known_hosts 2>&1

echo "Installing Ansible roles"
ansible-galaxy install --ignore-errors -r ../requirements/services.yml -f -p roles/services

echo "Executing Ansible playbook"

if [ "$ENV_NAME" == "integration" ] || [ "$ENV_NAME" == "staging" ] || [ "$ENV_NAME" == "production" ]; then
    if [ "$EXECUTION_MODE" = "check" ]; then
        ansible-playbook -C -D upgrade_cluster.yml
    else
        ansible-playbook upgrade_cluster.yml
    fi
else 
    if [ "$EXECUTION_MODE" = "check" ]; then
        ansible-playbook -C -D upgrade_single.yml
    else
        ansible-playbook upgrade_single.yml
    fi
fi