#!/bin/bash
  
  # The script downloads and installs jieba
  
  set -e
  
  # Make sure we are in the tools/ directory.
  if [ `basename $PWD` == extras ]; then
    cd ..
  fi
  
  ! [ `basename $PWD` == tools ] && \
     echo "You must call this script from the tools/ directory" && exit 1;
  
  echo "Installing jieba"
  
  if [ -d ./jieba ] ; then
    echo  >&2 "$0: Warning: old installation of jieba found. You should manually"
    echo  >&2 "  delete the directory tools/jieba and "
    echo  >&2 "  edit the file tools/env.sh and remove manually all references to it"
    exit 1
  fi
  
  if [ ! -d ./jieba ]; then
    git clone https://github.com/fxsjy/jieba.git || exit 1;
  fi
  
  (
  cd jieba
  pyver=`python --version 2>&1 | sed -e 's:.*\([2-3]\.[0-9]\+\).*:\1:g'`
  export PYTHONPATH=$PYTHONPATH:$PWD/lib/python${pyver}/site-packages/
  # we have to create those dir, as the install target does not create it
  mkdir -p $PWD/lib/python${pyver}/site-packages/
  python setup.py install --prefix `pwd`
  cd ..
  )
  
  lib_dir=./lib/
  site_packages_dir=$(cd ./jieba; find $lib_dir -name "site-packages" -type d | head -n1)
  (
    echo "export JIEBA=\"$PWD/jieba\""
    echo "export PYTHONPATH=\"\${PYTHONPATH:-}:\$JIEBA/${site_packages_dir}\""
  ) >> env.sh
  
  echo >&2 "Installation of jieba finished successfully"
  echo >&2 "Please source tools/env.sh in your path.sh to enable it"