#!/usr/bin/env bash

# Cloud Foundry Python Conda Buildpack

# Copyright (c) 2014-2015 the original author or authors.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

#

echo "-----> Starting compile step..."

BUILD_DIR=$1

CACHE=$2

BASH=$(which bash)

WGET=$(which wget)

CONDA_HOME="$1/.conda"

CONDA_BIN="$CONDA_HOME/bin"

CONDA_ENVS="$CONDA_HOME/envs"

CONDA_ENVS_CACHE="$CACHE/envs"

RUNTIME="$BUILD_DIR/runtime.txt"

# Get the runtime version and download appropriate Miniconda

MINICONDA_VERSION="4.2.12"

if [ -e $RUNTIME ]; then

PYTHON_VERSION=$(cut -d- -f2 $RUNTIME)

if [ ${PYTHON_VERSION:0:1} -eq 3 ]; then

PYTHON_MAJOR_VERSION=3

elif [ ${PYTHON_VERSION:0:1} -eq 2 ]; then

PYTHON_MAJOR_VERSION=2

else

PYTHON_MAJOR_VERSION=""

fi

MINICONDA_FILE="Miniconda${PYTHON_MAJOR_VERSION}-${MINICONDA_VERSION}-Linux-x86_64.sh"

else

MINICONDA_FILE="Miniconda2-${MINICONDA_VERSION}-Linux-x86_64.sh"

fi

MINICONDA_URI="http://repo.continuum.io/miniconda/$MINICONDA_FILE"

MINICONDA_CACHE="$CACHE/$MINICONDA_FILE"

PROFILE_PATH="$BUILD_DIR/.profile.d/conda.sh"

echo "-----> Preparing Python Environment..."

if [ ! -e $MINICONDA_CACHE ] ; then

echo "-----> Downloading Miniconda..."

if [ ! -d $CACHE ]; then mkdir $CACHE; fi

$WGET -q -O $MINICONDA_CACHE $MINICONDA_URI

chmod +x $MINICONDA_CACHE

fi

if [ -e $CONDA_HOME ]; then rm -rf $CONDA_HOME; fi

# Install miniconda

$MINICONDA_CACHE -b -p $CONDA_HOME #&> /dev/null

if [ -e $CONDA_ENVS_CACHE ] ; then

echo "-----> Using dependency cache at $CONDA_ENVS_CACHE"

cp -R $CONDA_ENVS_CACHE/* $CONDA_ENVS

else

mkdir -p $CONDA_ENVS_CACHE

fi

CONDA_ENV="dep_env"

echo "-----> Installing Dependencies..."

echo "-----> Installing conda environment from environment.yml..."

$CONDA_BIN/conda env update --quiet -n $CONDA_ENV -f "$BUILD_DIR/environment.yml"

$CONDA_BIN/conda clean -pt

echo "-----> Caching dependencies to $CONDA_ENVS_CACHE"

cp -R $CONDA_ENVS/* $CONDA_ENVS_CACHE

echo "-----> Fixing paths..."

grep -rlI $BUILD_DIR $BUILD_DIR | xargs sed -i.bak "s|$BUILD_DIR|/home/vcap/app|g"

# Add Conda path to profile

mkdir -p $(dirname $PROFILE_PATH)

echo "export PATH=$HOME/app/.conda/bin:\$PATH" >> $PROFILE_PATH

echo "source activate $CONDA_ENV" >> $PROFILE_PATH

if test -f $BUILD_DIR/runtime.txt && grep 'python=' $BUILD_DIR/environment.yml; then

echo "WARNING: you have specified the version of Python runtime both in 'runtime.txt' and 'environment.yml'. You should remove one of the two versions"

fi

echo "-----> Finished compile step"