1 files changed, 95 insertions, 0 deletions
diff --git a/pkg/tools/enable_ipdb.sh b/pkg/tools/enable_ipdb.sh
new file mode 100755
index 00000000..00a9235b
--- /dev/null
+++ b/pkg/tools/enable_ipdb.sh
@@ -0,0 +1,95 @@
+#!/bin/sh
+
+# enable_ipdb.sh
+# This script installs modules needed for using IPython debug shell in a
+# Bitmask bundle directory. It uses a python virtual environment in which it
+# installs needed modules and then links them into the appropriate directory
+# inside the bundle directory.
+
+# Copyright (C) 2013 LEAP
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+
+MODULES="ast.py runpy.py"
+SITE_MODULES="ipdb IPython simplegeneric.py decorator.py pexpect"
+
+if [ $# != 1 ]; then
+  echo "Usage: $0 bundle_path"
+  exit 1
+fi
+
+BUNDLE_PATH=`echo $1 | sed -e "s/\/\$//"`
+BUNDLE_LIB=${BUNDLE_PATH}/lib
+BUNDLE_VENV=${BUNDLE_PATH}/.venv
+
+function check_bundle_dirs() {
+  if [ ! -d ${BUNDLE_PATH} ]; then
+    echo "Argument ${BUNDLE_PATH} is not a directory."
+    exit 2
+  fi
+
+  if [ ! -d ${BUNDLE_LIB} ]; then
+    echo "Expected library directory ${BUNDLE_LIB} is not a directory."
+    exit 2
+  fi
+
+  if [ ! -w ${BUNDLE_LIB} ]; then
+    echo "Directory ${BUNDLE_LIB} is not writable."
+    exit 2
+  fi
+}
+
+function confirm_installation() {
+  echo -n "Are you sure you want to enable IPython debugger in ${BUNDLE_PATH} (y/N)? "
+  read confirm
+  if [[ "${confirm}" != "y" && "${confirm}" != "Y" ]]; then
+    echo "Bailing out..."
+    exit 0
+  fi
+}
+
+function setup_virtualenv() {
+  if [ ! -d ${BUNDLE_VENV} ]; then
+    virtualenv ${BUNDLE_VENV}
+  fi
+  source ${BUNDLE_VENV}/bin/activate
+  pip install ipdb
+}
+
+function link_modules() {
+  for package in ${MODULES}; do
+    package_path=${BUNDLE_LIB}/${package}
+    if [[ ! -f ${package_path} && ! -d ${package_path} ]]; then
+      ln -sf /usr/lib/python2.7/${package} ${BUNDLE_LIB}
+    fi
+  done
+  for package in ${SITE_MODULES}; do
+    package_path=${BUNDLE_LIB}/${package}
+    if [[ ! -f ${package_path} && ! -d ${package_path} ]]; then
+      ln -sf ${BUNDLE_VENV}/lib/python2.7/site-packages/${package} ${BUNDLE_LIB}
+    fi
+  done
+}
+
+function main() {
+  check_bundle_dirs
+  confirm_installation
+  setup_virtualenv
+  link_modules
+  echo "All done."
+  exit 0
+}
+
+main