blob: f7480a4977a3bd65cf6c6761aee69fec21acd93e [file] [log] [blame]
# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
# Programmable completion for some Chromium OS build scripts.
_list_repo_commands() {
local repo=${COMP_WORDS[0]}
"${repo}" help --all | grep -E '^ ' | sed 's/ \([^ ]\+\) .\+/\1/'
}
_list_repo_branches() {
local repo=${COMP_WORDS[0]}
"${repo}" branches 2>&1 | grep \| | sed 's/[ *][Pp ] *\([^ ]\+\) .*/\1/'
}
_list_repo_projects() {
local repo=${COMP_WORDS[0]}
local manifest=$(mktemp)
"${repo}" manifest -o "${manifest}" >& /dev/null
grep 'project name=' "${manifest}" | sed 's/.\+name="\([^"]\+\)".\+/\1/'
rm -f "${manifest}" >& /dev/null
}
# Complete the repo <command> argument.
_complete_repo_command() {
[ ${COMP_CWORD} -eq 1 ] || return 1
local command=${COMP_WORDS[1]}
COMPREPLY=($(compgen -W "$(_list_repo_commands)" -- "${command}"))
return 0
}
_complete_repo_arg() {
[ ${COMP_CWORD} -gt 1 ] || return 1
local command=${COMP_WORDS[1]}
local current=${COMP_WORDS[COMP_CWORD]}
if [[ ${command} == "abandon" ]]; then
if [[ ${COMP_CWORD} -eq 2 ]]; then
COMPREPLY=($(compgen -W "$(_list_repo_branches)" -- "${current}"))
else
COMPREPLY=($(compgen -W "$(_list_repo_projects)" -- "${current}"))
fi
return 0
fi
if [[ ${command} == "help" ]]; then
[ ${COMP_CWORD} -eq 2 ] && \
COMPREPLY=($(compgen -W "$(_list_repo_commands)" -- "${current}"))
return 0
fi
if [[ ${command} == "start" ]]; then
[ ${COMP_CWORD} -gt 2 ] && \
COMPREPLY=($(compgen -W "$(_list_repo_projects)" -- "${current}"))
return 0
fi
return 1
}
# Complete the repo arguments.
_complete_repo() {
COMPREPLY=()
_complete_repo_command && return 0
_complete_repo_arg && return 0
return 0
}
complete -F _complete_repo repo
source /usr/share/git/git-prompt.sh 2>/dev/null
# Add a way to get the "m" branch from repo easily; used by __git_branch_ps1()
#
# Repo seems to maintain a phony 'm/' remote and it always seems to be the name
# of the manifest branch. This will retrieve it.
__git_m_branch() {
local git_dir=$(git rev-parse --git-dir 2> /dev/null)
if [ -n "${git_dir}" ]; then
echo $(cd ${git_dir}/refs/remotes/m 2> /dev/null && ls)
fi
}
# A "subclass" of __git_ps1 that adds the manifest branch name into the prompt.
# ...if you're on manifest branch "0.11.257.B" and local branch "lo" and
# pass " (%s)", we'll output " (0.11.257.B/lo)". Note that we'll never show
# the manifest branch 'master', since it's so common.
__git_branch_ps1() {
local format_str="${1:- (%s)}"
local m_branch=$(__git_m_branch)
if [ "${m_branch}" != "master" -a -n "${m_branch}" ]; then
format_str=$(printf "${format_str}" "${m_branch}/%s")
fi
# for subshells, prefix the prompt with the shell nesting level
local lshlvl=""
[ ! -z "${SHLVL##*[!0-9]*}" ] && [ ${SHLVL} -gt 1 ] && lshlvl="${SHLVL} "
__git_ps1 "${lshlvl}${format_str}"
}
# Prompt functions should not error when in subshells
export -f __gitdir
export -f __git_ps1
export -f __git_m_branch
export -f __git_branch_ps1