#!/bin/bash find_charm_dirs() { # Hopefully, $JUJU_CHARM_DIR is set so which venv to use in unambiguous. if [[ -n "$JUJU_CHARM_DIR" || -n "$CHARM_DIR" ]]; then if [[ -z "$JUJU_CHARM_DIR" ]]; then # accept $CHARM_DIR to be more forgiving export JUJU_CHARM_DIR="$CHARM_DIR" fi if [[ -z "$CHARM_DIR" ]]; then # set CHARM_DIR as well to help with backwards compatibility export CHARM_DIR="$JUJU_CHARM_DIR" fi return fi # Try to guess the value for JUJU_CHARM_DIR by looking for a non-subordinate # (because there's got to be at least one principle) charm directory; # if there are several, pick the first by alpha order. agents_dir="/var/lib/juju/agents" if [[ -d "$agents_dir" ]]; then non_subordinates="$(grep -L 'subordinate:.*true' "$agents_dir"/*/charm/metadata.yaml | wc -l)" if [[ "$non_subordinates" -gt 1 ]]; then >&2 echo 'Ambiguous possibilities for JUJU_CHARM_DIR; please run within a Juju hook context' exit 1 elif [[ "$non_subordinates" -eq 1 ]]; then for unit_dir in $(/bin/ls -d "$agents_dir/unit-"*); do if grep -q 'subordinate:.*true' "$unit_dir/charm/metadata.yaml"; then continue fi export JUJU_CHARM_DIR="$unit_dir/charm" export CHARM_DIR="$JUJU_CHARM_DIR" return done fi fi >&2 echo 'Unable to determine JUJU_CHARM_DIR; please run within a Juju hook context' exit 1 } try_activate_venv() { if [[ -d "$JUJU_CHARM_DIR/../.venv" ]]; then . "$JUJU_CHARM_DIR/../.venv/bin/activate" fi } find_wrapped() { PATH="${PATH/\/usr\/local\/sbin:}" which "$(basename "$0")" } find_charm_dirs try_activate_venv export PYTHONPATH="$JUJU_CHARM_DIR/lib:$PYTHONPATH" if [[ "$(basename "$0")" == "charm-env" ]]; then # being used as a shebang exec "$@" elif [[ "$0" == "$BASH_SOURCE" ]]; then # being invoked as a symlink wrapping something to find in the venv exec "$(find_wrapped)" "$@" elif [[ "$(basename "$BASH_SOURCE")" == "charm-env" ]]; then # being sourced directly; do nothing /bin/true else # being sourced for wrapped bash helpers . "$(find_wrapped)" fi