diff options
Diffstat (limited to 'conda.sh')
| -rw-r--r-- | conda.sh | 231 |
1 files changed, 231 insertions, 0 deletions
diff --git a/conda.sh b/conda.sh new file mode 100644 index 0000000..049354b --- /dev/null +++ b/conda.sh @@ -0,0 +1,231 @@ +#!/bin/bash + +export SSB_CHANNEL_PUBLIC=http://ssb.stsci.edu/astroconda +export SSB_CHANNEL_DEV=http://ssb.stsci.edu/conda-dev + + +function ssb_emesg +{ + echo "ERROR: $@" +} + + +function ssb_entry_checkpoint +{ + if ! hash conda 2>/dev/null; then + ssb_emesg "'conda' was not found in your PATH" + return 1 + fi + + return 0 +} + +ssb_entry_checkpoint +retval=$? +if [[ $retval != 0 ]]; then + return $retval +fi +unset -f ssb_entry_checkpoint + + +function ssb_usage +{ + echo "Usage: ssb {-i|-u} {-p|-d} [-an] [[package] ...] + +General Arguments: + --help -h This usage message + --name [env] -n Update existing environment by name + --public -p Use astroconda + --dev -d Use conda-dev + --all -a Update all packages (supersedes [package] requests) + --yes -y Do not prompt (use with care) + +Mode Arguments: + --install -i Perform an installation + --update -u Perform an update + +Positional Arguments: +package Name of package(s) to update + +" + return 0 +} + + +function ssb +{ + local cmd="conda " + local opts="" + local is_public=0 + local is_dev=0 + local is_all=0 + local is_named=0 + local is_create=0 + local is_update=0 + local is_install=0 + + if [[ $# < 1 ]]; then + ssb_emesg "Not enough arguments" + ssb_usage + return 1 + fi + + while [[ $# > 0 ]] + do + key=$1 + + case $key in + --help|-h) + ssb_usage + return 0 + ;; + + --create|-c) + is_create=1 + ;; + + --install|-i) + is_install=1 + ;; + + --update|-u) + is_update=1 + ;; + + --name|-n) + opts+="--name $2 " + is_named=1 + shift + ;; + + --dev|-d) + opts+="--override-channels -c $SSB_CHANNEL_DEV -c defaults " + is_public=1 + ;; + + --public|-p) + opts+="--override-channels -c $SSB_CHANNEL_PUBLIC -c defaults " + is_dev=1 + ;; + + --all|-a) + opts+="--all " + is_all=1 + ;; + + --yes|-y) + opts+="--yes " + ;; + *) + opts+="$key " + ;; + esac + + shift + done + + + if [[ $is_public != 0 ]] && [[ $is_dev != 0 ]]; then + ssb_emesg "--public and --dev are mutually exclusive options" + ssb_usage + return 1 + fi + + if [[ $is_public == 0 ]] && [[ $is_dev == 0 ]]; then + ssb_emesg "--public or --dev must be specified" + ssb_usage + return 1 + fi + + if [[ $is_install == 0 ]] && [[ $is_update == 0 ]] && [[ $is_create == 0 ]]; then + ssb_emesg "--install, --update, or --create is required" + ssb_usage + return 1 + fi + + if [[ $is_all != 0 ]]; then + if [[ $is_public == 0 ]] && [[ $is_dev == 0 ]]; then + ssb_emesg "--all cannot be used without --public or --dev" + ssb_usage + return 1 + fi + + if [[ $is_install != 0 ]]; then + ssb_emesg "--all cannot be used with --install" + ssb_usage + return 1 + fi + + if [[ $is_create != 0 ]]; then + ssb_emesg "--all cannot be used with --create" + ssb_usage + return 1 + fi + fi + + if [[ $is_install != 0 ]]; then + if [[ $is_update != 0 ]]; then + ssb_emesg "--install and --update are mutually exclusive options" + ssb_usage + return 1 + fi + + if [[ $is_create != 0 ]]; then + ssb_emesg "--install and --create are mutually exclusive options" + ssb_usage + return 1 + fi + fi + + if [[ $is_update != 0 ]]; then + if [[ $is_install != 0 ]]; then + ssb_emesg "--update and --install are mutually exclusive options" + ssb_usage + return 1 + fi + + if [[ $is_create != 0 ]]; then + ssb_emesg "--update and --create are mutually exclusive options" + ssb_usage + return 1 + fi + fi + + if [[ $is_create != 0 ]]; then + if [[ $is_install != 0 ]]; then + ssb_emesg "--create and --install are mutually exclusive options" + ssb_usage + return 1 + fi + + if [[ $is_update != 0 ]]; then + ssb_emesg "--create and --update are mutually exclusive options" + ssb_usage + return 1 + fi + fi + + if [[ $is_install != 0 ]]; then + opts="install $opts" + fi + + if [[ $is_update != 0 ]]; then + opts="update $opts" + fi + + if [[ $is_create != 0 ]]; then + opts="create $opts" + fi + + if [[ -z $CONDA_PREFIX ]] && [[ $is_named == 0 ]]; then + ssb_emesg "Refusing to corrupt base installation!" + ssb_emesg "You are not in an active conda environment!" + ssb_emesg "Use --name to update a specific environment!" + ssb_usage + return 1 + fi + + $cmd $opts +} + +export -f ssb |