roxma / envrc-rs Goto Github PK

#!/bin/bash
PROMPT_COMMAND=envrc_prompt_command

function envrc_find() {
    local d="$PWD"

    while :
    do
        if [ -f "$d/.envrc" ]
        then
            echo "$d/.envrc"
        fi

        if [ "$d" == "/" -o "$d" == "." ]
        then
            return
        fi

        d="$(exec dirname "$d")"
    done
}

function envrc_prompt_command() {
    local rc_found="$(envrc_find)"

    if [ "$rc_found" == "$ENVRC_LOAD" ]
    then
        if [ -n "$ENVRC_LOAD" -a -z "$ENVRC_LOADED" -a "$ENVRC_PPID" == "$PPID" ]
        then
            ENVRC_LOADED=1
            . "$ENVRC_LOAD"
        elif [ -n "$ENVRC_LOAD" -a "$ENVRC_PPID" != "$PPID" ]
        then
            unset ENVRC_LOAD
            unset ENVRC_LOADED
            unset ENVRC_PPID
            unset ENVRC_TMP
            envrc_prompt_command
        fi
        return
    fi

    if [ "$ENVRC_LOAD" == "" ]
    then
        local tmp="$(mktemp --suffix=_envrc)"
        if [ -z "$tmp" ]
        then
            return
        fi
        echo "exit 0" > "$tmp"

        echo "envrc: spawn for [$rc_found]"
        ENVRC_TMP="$tmp" ENVRC_LOAD="$rc_found" ENVRC_PPID=$$ $BASH
        eval "$(cat "$tmp"; rm "$tmp")"
        envrc_prompt_command
    else
        echo "cd '$PWD'
        export OLDPWD='$OLDPWD'" > $ENVRC_TMP
        echo "envrc: exit [$ENVRC_LOAD]"
        exit 0
    fi
}