setup-env.sh

#!/usr/bin/env bash

if [ -z "${BASH_VERSION}" ]; then
  echo -e "\n\n\n\nYour environment is not correctly set up.\n"
  echo "You need to source $DXRSRC/setup-env.sh into a running bash:"
  echo ". $DXRSRC/setup-env.sh"
  return 0 &>/dev/null
  exit 1
fi

if [ -z "$2" ]; then
  name="${BASH_SOURCE[0]}"
  if [ -z "$name" ]; then
	  name=$0
  fi
  echo "Usage: . $name <dxr.config> <treename>"
  return 0 &>/dev/null
  exit 1
fi

DXRCONFIG="$1"
TREENAME="$2"

function get_var()
{
echo $(PYTHONPATH=$DXRSRC:$PYTHONPATH python - <<HEREDOC
import dxr
cfg=dxr.load_config("$DXRCONFIG")
found=False
for treecfg in cfg.trees:
  if treecfg.tree == "$1":
    found=True
    break

if found == False:
  raise BaseException("Tree[$1] not found in config[$DXRCONFIG]")

try:
  print("%s" % treecfg.$2)
except:
  raise BaseException("Variable[$2] not found in Tree[$1]")
HEREDOC
)
}

SRCDIR=$(get_var "$TREENAME" "sourcedir")
OBJDIR=$(get_var "$TREENAME" "objdir")

if [ -z "$SRCDIR" ]; then
  return 0 &>/dev/null
  exit 1
fi

if [ -z "$OBJDIR" ]; then
  echo -e "\e[1;33mThe object dir equals source dir (not recommended).\e[0m\n"
  export OBJDIR="$SRCDIR"
fi

if [ -z "$DXRSRC" ]; then
  echo "Setting DXRSRC variable"
  scriptsrc=${BASH_SOURCE[0]}
  export DXRSRC=$(dirname $(readlink -f $scriptsrc))
fi

MAKE=${MAKE:-make}

echo "Finding available DXR plugins..."
tools=($(PYTHONPATH=$DXRSRC:$PYTHONPATH python - <<HEREDOC
import dxr
cfg=dxr.load_config("$DXRCONFIG")
for treecfg in cfg.trees:
  if treecfg.tree == "$TREENAME":
    files = [x.__file__ for x in dxr.get_active_plugins(treecfg, "$DXRSRC")]
    break
print ' '.join([x[:x.find('/indexer.py')] for x in files])
HEREDOC
) )
echo -n "Found:"
for plugin in $(seq 0 $((${#tools[@]} - 1))); do
  echo -n " $(basename ${tools[plugin]})"
done
echo ""

echo -n "Cleaning up environment variables from previous runs... "
for plugin in $(seq 0 $((${#tools[@]} - 1))); do
  if [ -e "${tools[plugin]}/unset-env.sh" ]; then
    . "${tools[plugin]}/unset-env.sh"
  fi
done
echo ""

echo -n "Building sqlite tokenizer... "
$MAKE -C ${DXRSRC}/sqlite
if [[ $? != 0 ]]; then
  echo "Bailing!"
  return 1
fi
echo "done!"

for plugin in $(seq 0 $((${#tools[@]} - 1))); do
  echo -n "Prebuilding $(basename ${tools[plugin]})... "
  if [ -e ${tools[plugin]}/Makefile ]; then 
    $MAKE -s -C ${tools[plugin]} prebuild
    if [[ $? != 0 ]]; then
      echo "Bailing!"
      return 1
    fi
  fi
  echo "done!"
done

echo -n "Preparing environment... "
for plugin in $(seq 0 $((${#tools[@]} - 1))); do
  if [ -e "${tools[plugin]}/set-env.sh" ]; then
    . "${tools[plugin]}/set-env.sh"
  fi
done
echo "done!"

echo "DXR setup complete. You can now compile your source code in this shell."

# Check for . $0 instead of ./$0
return 0 &>/dev/null
echo -e "\n\n\n\n\e[1;31mYour environment is not correctly set up.\e[0m\n"
echo "You need to run the following command instead:"
echo ". $DXRSRC/setup-env.sh"