git-bd/git-bd.bashrc at master · nnutter/git-bd · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
function __git_bd_list_branches {
    local base_dir="$(git config --get bd.base-dir 2> /dev/null || true)"
    git config --get-regexp bd\..*\.work-tree \
        | sed -e 's/bd\..*\.work-tree //' \
        | sed -e "s|$base_dir/*||"

}

function _git_bd {
    local cur="${COMP_WORDS[COMP_CWORD]}"

    local prev=""
    if [ $COMP_CWORD -gt 0 ]; then
        prev="${COMP_WORDS[COMP_CWORD - 1]}"
    fi

    local BRANCHES=$(__git_bd_list_branches)
    if [ -z "$BRANCHES" ]; then
        return
    fi

    if [[ "$prev" =~ -[dDmMi] ]]; then
        COMPREPLY=($(compgen -W "$BRANCHES" -- $cur))
    else
        if [[ ${#COMP_WORDS[@]} -eq 3 && "$cur" =~ -[dDmMi]? ]]; then
            OPTIONS="-d -D -m -M --init"
            COMPREPLY=($(compgen -W "$OPTIONS" -- $cur))
        fi
    fi

}
complete -F _git_bd git-bd

function __complete_bd {
    local cur="${COMP_WORDS[COMP_CWORD]}"
    local BRANCHES=$(__git_bd_list_branches)
    if [ -z "$BRANCHES" ]; then
        return
    else
        if [ ${#COMP_WORDS[@]} -eq 2 ]; then
            COMPREPLY=($(compgen -W "$BRANCHES" -- $cur))
        fi
    fi
}
complete -F __complete_bd bd

function bd {
    local base_dir="$(git config --get bd.base-dir 2> /dev/null || true)"
    local git_dir="$(git config --get bd.git-dir 2> /dev/null || true)"
    local branch=$(git rev-parse --abbrev-ref HEAD 2> /dev/null)

    if [ "$1" == "-h" ] || [ "$1" == "-?" ] || [ "$1" == "--help" ]; then
        echo 'Usage:'
        echo '   bd                 changes to primary working directory'
        echo '   bd <branchdir>     changes to specified branchdir'
        echo ''
        echo 'Must be used within an existing branchdir or the primary working directory.  <branchdir> can be tab completed.'
        return
    fi

    if [ -z "$base_dir" ]; then
        echo 'ERROR: Have you `git bd --init` yet?'
        return
    fi

    local DIR
    if [ -n "$1" ]; then
        DIR="$base_dir/$1"
    else
        DIR="$git_dir/.."
    fi

    if [ -n "$branch" ] && [ -d "$DIR/${PWD/$base_dir\/$branch\//}" ]; then
        DIR="$DIR/${PWD/$base_dir\/$branch\//}"
    fi

    if [ -d "$DIR" ]; then
        cd "$DIR"
    else
        echo "ERROR: branchdir does not exist: $DIR"
    fi
}