#!/usr/bin/env bash

# we use -doesnt_install- easy_install for python packages. Due to instability
# with the easy_install servers mostly these come prepacked. A few have narly
# dependency issues so they need to be called from the easy_install server
# (which has a wider range of dependencies pre-packed)
# 
# options:
#
# -c remove any files or directories from $egg_source before doing anything
# -g get packages (will happen after cleaning egg source if -c is set); do not
#    overwrite any existing packages
# -i install packages (will happen after getting packages if -g is set)
#
# if no options are set, this does nothing(!)
egg_source=${MAUS_THIRD_PARTY}/third_party/source/easy_install
# these are packages in MAUS third party tarball
download_package_list="\
 anyjson python-dateutil>=1.5,<2.0 kombu==2.1.8 amqplib>=1.0 six>=1.4.0 \
 logilab-common logilab-astng suds validictory nose==1.1 nose-exclude  \
 coverage ipython doxypy pylint==0.25.1 bitarray celery==2.5.5 \
 pymongo==2.3 readline matplotlib==1.1.0 scons==2.2.0\
 pil django==1.5.1 magickwand
"
# scons v2.2.0 is no longer on pypi
# temporary hack to specify the download url for scons==2.2.0
# pending upgrade to 2.3.0
# DR March 13, 2014
scons_version="2.2.0"
scons_url="http://sourceforge.net/projects/scons/files/scons/${scons_version}/scons-${scons_version}.tar.gz"

# these are the packages to install - note the version dependencies
package_list="\
 anyjson python-dateutil amqplib six kombu \
 logilab-common logilab-astng  suds validictory nose nose-exclude \
 coverage ipython doxypy pylint bitarray celery \
 pymongo scons readline matplotlib \
 pil django magickwand
"
# note numpy was installed previously, not in this script. We test it's import
# here for convenience
module_test_list="numpy suds validictory nose coverage \
 pylint bitarray matplotlib celery pymongo \
 Image django magickwand" #Image is pil bottom level
binary_test_list="scons"
cleanup="0"
get_packages="0"
install_packages="0"

while getopts ":cgi" opt; do
  case $opt in
    c)
      cleanup="1"
      ;;
    g)
      get_packages="1"
      ;;
    i)
      install_packages="1"
      ;;
    \?)
      echo "FATAL: Invalid option: -$OPTARG" >&2
      exit 1
      ;;
  esac
done

if [ -z ${MAUS_THIRD_PARTY} ]; then
    echo "FATAL: MAUS_THIRD_PARTY was not set - aborting"
    exit 1
fi

if [ "$cleanup" == "1" ]; then
    echo "INFO: Cleaning $egg_source directory" 
    cd $egg_source
    rm -r *
fi

if [ "$get_packages" == "1" ]; then
    echo "INFO: Attempting to download eggs and pack for distribution"
    cd $egg_source
    for package in $download_package_list
    do
        # tmp hack for scons 2.2.0 which has disappeared from pypi
        if [[ $package =~ .*scons*. ]]
        then
            easy_install -zmaxeb . -f$scons_url $package
        else
            echo "easy_install -zmaxeb . $package"
            easy_install -zmaxeb . $package &
            # kill the command after <timeout> seconds
            ((timeout = 120))
            while ((timeout > 0)); do
                sleep 1
                kill -0 $! >& /dev/null
                if [ "$?" == "0" ]; then # pid is running (can be killed)
                    ((timeout -= 1))
                else # pid is finished (can't be killed)
                    ((timeout = 0))
                fi
            done
            kill -0 $! >& /dev/null
            if [ "$?" == "0" ]; then # pid is running (can be killed)
                echo "easy_install of $package has timed out - killing"
                kill -9 $!
            fi
        fi
    done
    downloaded_list=`ls --hide=*.tar.gz`
    echo "INFO: Downloaded the following packages"
    echo "$downloaded_list"
    echo "INFO: Packing them now - they can be found tarred in $egg_source"
    echo "INFO: I will skip any packages which are already packed"
    for item in $downloaded_list
    do
        if [ ! -e $item.tar.gz ]; then # ignore existing tarballs
            echo "INFO:       $item"
            tar -czf $item.tar.gz $item
        else
            echo "INFO:       $item skipped"
        fi
        echo ">>>>>removing $item"
        rm -rf $item # cleanup anything not tarred
    done
fi

if [ "$install_packages" == "1" ]; then
    # first try a local install
    for package in $package_list
    do
        echo "INFO: Installing $package"
        easy_install -H None -f $egg_source $package
    done
    # now check that packages were installed
    for module in $module_test_list
    do
        echo "INFO: Checking import of package $module"
        python -c "import $module" || { echo "FATAL: Failed to install python module $module"; exit 1; }
        echo "INFO: ok"
    done
    for bin in $binary_test_list
    do
        echo "INFO: Checking python script $bin"
        which $bin || { echo "FATAL: Failed to install python script $bin"; exit 1; }
        echo "INFO: ok"
    done

    echo "INFO: easy_install libraries were installed in your third party"
    echo "INFO: directory $MAUS_THIRD_PARTY"
fi