Initial commit

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