ye-cd

# -*- shell-script -*-

# If ye-cd has been source'd already, YE_SCRIPT will be in the
# environment, so we should not re-source it, unless YE_FORCE_RELOAD
# is set.
if [ -n "$YE_SCRIPT" ] && [ ! -n "$YE_FORCE_RELOAD" ]; then
    cat <<EOF >&2
ye-cd has already been source'd. If you really want to re-evaluate it,
set the YE_FORCE_RELOAD environment variable and source it again.
EOF
    return
fi

YE_SCRIPT=`which ye`
if [ $? -ne 0 ]; then
    echo "ERROR: could not find ye (is it in \$PATH?)." >&2
    return
fi

export YE_SCRIPT

YE_PLUMBING_DIR=$HOME/.ye/plumbing

yecd_error() {
    echo "ERROR: " $* >&2
}

yecd_get_bb_var() {
    if [ -e "$YE_PLUMBING_DIR/x" ]; then
        awk -F= "/^$1/"' {print $2}' $YE_PLUMBING_DIR/x
    else
        yecd_error "yecd_get_bb_var: could not determine $1."
    fi
}

ye() {

    if [ "$1" != "cd" ]; then
        $YE_SCRIPT "$@"
        return $?
    fi

    shift

    if [ -z "$BUILDDIR" ]; then
        yecd_error "BUILDDIR is not set"
        return 1
    fi

    local dir_shortcut=$1
    local topdir

    case $dir_shortcut in
        "top"|"topdir")
            topdir=`$YE_SCRIPT plumbing topdir`
            [ $? -eq 0 ] && cd $topdir
            ;;
        "wd"|"work"|"workdir")
            local recipe=$2
            if [ -z "$recipe" ]; then
                cd $BUILDDIR/tmp/work/
            else
                $YE_SCRIPT plumbing wd "$recipe"
                if [ -e "$YE_PLUMBING_DIR/wd" ]; then
                    cd `cat $YE_PLUMBING_DIR/wd`
                fi
            fi
            ;;
        "bd"|"build"|"builddir")
            cd $BUILDDIR
            ;;
        "bh"|"buildhistory")
            cd $BUILDDIR/buildhistory
            ;;
        "sd"|"sysroot"|"sysrootdir")
            $YE_SCRIPT plumbing x MACHINE
            local machine=`yecd_get_bb_var MACHINE | tr - _`
            [ -n "$machine" ] && cd $BUILDDIR/tmp/sysroots-components/$machine
            ;;
        "src"|"sources")
            local recipe=$2
            if [ -z "$recipe" ]; then
                topdir=`$YE_SCRIPT plumbing topdir`
                [ $? -eq 0 ] && cd $topdir/sources
            else
                cd $BUILDDIR
                $YE_SCRIPT plumbing find $recipe
                cd `dirname $(cat $YE_PLUMBING_DIR/find)`
            fi
            ;;
        "img"|"images")
            $YE_SCRIPT plumbing x MACHINE
            local machine=`yecd_get_bb_var MACHINE`
            [ -n "$machine" ] && cd $BUILDDIR/tmp/deploy/images/$machine
            ;;
        "pkg"|"packages")
            $YE_SCRIPT plumbing x IMAGE_PKGTYPE MACHINE
            local pkg_type=`yecd_get_bb_var IMAGE_PKGTYPE`
            local machine=`yecd_get_bb_var MACHINE | tr - _`
            [ -n "$pkg_type" ] && [ -n "$machine" ] && \
                cd $BUILDDIR/tmp/deploy/$pkg_type/$machine
            ;;
        "manifest"|"manifests")
            topdir=`$YE_SCRIPT plumbing topdir`
            [ $? -eq 0 ] && cd $topdir/.repo/manifests
            ;;
        *)
            if [ -z "$dir_shortcut" ]; then
                cd $BUILDDIR
            else
                $YE_SCRIPT -h
                return $?
            fi
            ;;
    esac
}