Set up virtualenv for PyPy

Author: BanzaiMan

bin/compile

@@ -21,13 +21,25 @@ if [[ $ALIAS ]] ; then
   sudo ln -s $INSTALL_DEST/$VERSION $INSTALL_DEST/$ALIAS
 fi
 
+if [[ $VERSION = pypy-* ]]; then
+	PYTHON_BIN=pypy
+elif [[ $VERSION = pypy3* ]]; then
+	PYTHON_BIN=pypy3
+else
+	PYTHON_BIN=python
+fi
+
+if [[ $VERSION != pypy* ]]; then
+	VIRTENV_VERSION=python$VERSION
+fi
+
 virtualenv --distribute --python=$INSTALL_DEST/$VERSION/bin/python \
-  /home/travis/virtualenv/python$VERSION
+  /home/travis/virtualenv/$VIRTENV_VERSION
 
 if [[ $ALIAS ]] ; then
-  ln -s $HOME/virtualenv/python$VERSION $HOME/virtualenv/python$ALIAS
+  ln -s $HOME/virtualenv/$VIRTENV_VERSION $HOME/virtualenv/python$ALIAS
 fi
 
 if [[ $PACKAGES ]] ; then
-  $HOME/virtualenv/python$VERSION/bin/pip install --upgrade $PACKAGES
+  $HOME/virtualenv/$VIRTENV_VERSION/bin/pip install --upgrade $PACKAGES
 fi