xref: /linux/tools/perf/perf-completion.sh (revision d91517839e5d95adc0cf4b28caa7af62a71de526)
1# perf bash and zsh completion
2
3# Taken from git.git's completion script.
4__my_reassemble_comp_words_by_ref()
5{
6	local exclude i j first
7	# Which word separators to exclude?
8	exclude="${1//[^$COMP_WORDBREAKS]}"
9	cword_=$COMP_CWORD
10	if [ -z "$exclude" ]; then
11		words_=("${COMP_WORDS[@]}")
12		return
13	fi
14	# List of word completion separators has shrunk;
15	# re-assemble words to complete.
16	for ((i=0, j=0; i < ${#COMP_WORDS[@]}; i++, j++)); do
17		# Append each nonempty word consisting of just
18		# word separator characters to the current word.
19		first=t
20		while
21			[ $i -gt 0 ] &&
22			[ -n "${COMP_WORDS[$i]}" ] &&
23			# word consists of excluded word separators
24			[ "${COMP_WORDS[$i]//[^$exclude]}" = "${COMP_WORDS[$i]}" ]
25		do
26			# Attach to the previous token,
27			# unless the previous token is the command name.
28			if [ $j -ge 2 ] && [ -n "$first" ]; then
29				((j--))
30			fi
31			first=
32			words_[$j]=${words_[j]}${COMP_WORDS[i]}
33			if [ $i = $COMP_CWORD ]; then
34				cword_=$j
35			fi
36			if (($i < ${#COMP_WORDS[@]} - 1)); then
37				((i++))
38			else
39				# Done.
40				return
41			fi
42		done
43		words_[$j]=${words_[j]}${COMP_WORDS[i]}
44		if [ $i = $COMP_CWORD ]; then
45			cword_=$j
46		fi
47	done
48}
49
50type _get_comp_words_by_ref &>/dev/null ||
51_get_comp_words_by_ref()
52{
53	local exclude cur_ words_ cword_
54	if [ "$1" = "-n" ]; then
55		exclude=$2
56		shift 2
57	fi
58	__my_reassemble_comp_words_by_ref "$exclude"
59	cur_=${words_[cword_]}
60	while [ $# -gt 0 ]; do
61		case "$1" in
62		cur)
63			cur=$cur_
64			;;
65		prev)
66			prev=${words_[$cword_-1]}
67			;;
68		words)
69			words=("${words_[@]}")
70			;;
71		cword)
72			cword=$cword_
73			;;
74		esac
75		shift
76	done
77}
78
79type __ltrim_colon_completions &>/dev/null ||
80__ltrim_colon_completions()
81{
82	if [[ "$1" == *:* && "$COMP_WORDBREAKS" == *:* ]]; then
83		# Remove colon-word prefix from COMPREPLY items
84		local colon_word=${1%"${1##*:}"}
85		local i=${#COMPREPLY[*]}
86		while [[ $((--i)) -ge 0 ]]; do
87			COMPREPLY[$i]=${COMPREPLY[$i]#"$colon_word"}
88		done
89	fi
90}
91
92__perfcomp ()
93{
94	COMPREPLY=( $( compgen -W "$1" -- "$2" ) )
95}
96
97__perfcomp_colon ()
98{
99	__perfcomp "$1" "$2"
100	__ltrim_colon_completions $cur
101}
102
103__perf_main ()
104{
105	local cmd
106
107	cmd=${words[0]}
108	COMPREPLY=()
109
110	# List perf subcommands or long options
111	if [ $cword -eq 1 ]; then
112		if [[ $cur == --* ]]; then
113			__perfcomp '--help --version \
114			--exec-path --html-path --paginate --no-pager \
115			--perf-dir --work-tree --debugfs-dir' -- "$cur"
116		else
117			cmds=$($cmd --list-cmds)
118			__perfcomp "$cmds" "$cur"
119		fi
120	# List possible events for -e option
121	elif [[ $prev == "-e" && "${words[1]}" == @(record|stat|top) ]]; then
122		evts=$($cmd list --raw-dump)
123		__perfcomp_colon "$evts" "$cur"
124	# List subcommands for 'perf kvm'
125	elif [[ $prev == "kvm" ]]; then
126		subcmds="top record report diff buildid-list stat"
127		__perfcomp_colon "$subcmds" "$cur"
128	# List long option names
129	elif [[ $cur == --* ]];  then
130		subcmd=${words[1]}
131		opts=$($cmd $subcmd --list-opts)
132		__perfcomp "$opts" "$cur"
133	fi
134}
135
136if [[ -n ${ZSH_VERSION-} ]]; then
137	autoload -U +X compinit && compinit
138
139	__perfcomp ()
140	{
141		emulate -L zsh
142
143		local c IFS=$' \t\n'
144		local -a array
145
146		for c in ${=1}; do
147			case $c in
148			--*=*|*.) ;;
149			*) c="$c " ;;
150			esac
151			array[${#array[@]}+1]="$c"
152		done
153
154		compset -P '*[=:]'
155		compadd -Q -S '' -a -- array && _ret=0
156	}
157
158	__perfcomp_colon ()
159	{
160		emulate -L zsh
161
162		local cur_="${2-$cur}"
163		local c IFS=$' \t\n'
164		local -a array
165
166		if [[ "$cur_" == *:* ]]; then
167			local colon_word=${cur_%"${cur_##*:}"}
168		fi
169
170		for c in ${=1}; do
171			case $c in
172			--*=*|*.) ;;
173			*) c="$c " ;;
174			esac
175			array[$#array+1]=${c#"$colon_word"}
176		done
177
178		compset -P '*[=:]'
179		compadd -Q -S '' -a -- array && _ret=0
180	}
181
182	_perf ()
183	{
184		local _ret=1 cur cword prev
185		cur=${words[CURRENT]}
186		prev=${words[CURRENT-1]}
187		let cword=CURRENT-1
188		emulate ksh -c __perf_main
189		let _ret && _default && _ret=0
190		return _ret
191	}
192
193	compdef _perf perf
194	return
195fi
196
197type perf &>/dev/null &&
198_perf()
199{
200	local cur words cword prev
201	_get_comp_words_by_ref -n =: cur words cword prev
202	__perf_main
203} &&
204
205complete -o bashdefault -o default -o nospace -F _perf perf 2>/dev/null \
206	|| complete -o default -o nospace -F _perf perf
207