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
50# Define preload_get_comp_words_by_ref="false", if the function
51# __perf_get_comp_words_by_ref() is required instead.
52preload_get_comp_words_by_ref="true"
53
54if [ $preload_get_comp_words_by_ref = "true" ]; then
55	type _get_comp_words_by_ref &>/dev/null ||
56	preload_get_comp_words_by_ref="false"
57fi
58[ $preload_get_comp_words_by_ref = "true" ] ||
59__perf_get_comp_words_by_ref()
60{
61	local exclude cur_ words_ cword_
62	if [ "$1" = "-n" ]; then
63		exclude=$2
64		shift 2
65	fi
66	__my_reassemble_comp_words_by_ref "$exclude"
67	cur_=${words_[cword_]}
68	while [ $# -gt 0 ]; do
69		case "$1" in
70		cur)
71			cur=$cur_
72			;;
73		prev)
74			prev=${words_[$cword_-1]}
75			;;
76		words)
77			words=("${words_[@]}")
78			;;
79		cword)
80			cword=$cword_
81			;;
82		esac
83		shift
84	done
85}
86
87# Define preload__ltrim_colon_completions="false", if the function
88# __perf__ltrim_colon_completions() is required instead.
89preload__ltrim_colon_completions="true"
90
91if [ $preload__ltrim_colon_completions = "true" ]; then
92	type __ltrim_colon_completions &>/dev/null ||
93	preload__ltrim_colon_completions="false"
94fi
95[ $preload__ltrim_colon_completions = "true" ] ||
96__perf__ltrim_colon_completions()
97{
98	if [[ "$1" == *:* && "$COMP_WORDBREAKS" == *:* ]]; then
99		# Remove colon-word prefix from COMPREPLY items
100		local colon_word=${1%"${1##*:}"}
101		local i=${#COMPREPLY[*]}
102		while [[ $((--i)) -ge 0 ]]; do
103			COMPREPLY[$i]=${COMPREPLY[$i]#"$colon_word"}
104		done
105	fi
106}
107
108__perfcomp ()
109{
110	COMPREPLY=( $( compgen -W "$1" -- "$2" ) )
111}
112
113__perfcomp_colon ()
114{
115	__perfcomp "$1" "$2"
116	if [ $preload__ltrim_colon_completions = "true" ]; then
117		__ltrim_colon_completions $cur
118	else
119		__perf__ltrim_colon_completions $cur
120	fi
121}
122
123__perf_prev_skip_opts ()
124{
125	local i cmd_ cmds_
126
127	let i=cword-1
128	cmds_=$($cmd $1 --list-cmds)
129	prev_skip_opts=()
130	while [ $i -ge 0 ]; do
131		if [[ ${words[i]} == $1 ]]; then
132			return
133		fi
134		for cmd_ in $cmds_; do
135			if [[ ${words[i]} == $cmd_ ]]; then
136				prev_skip_opts=${words[i]}
137				return
138			fi
139		done
140		((i--))
141	done
142}
143
144__perf_main ()
145{
146	local cmd
147
148	cmd=${words[0]}
149	COMPREPLY=()
150
151	# Skip options backward and find the last perf command
152	__perf_prev_skip_opts
153	# List perf subcommands or long options
154	if [ -z $prev_skip_opts ]; then
155		if [[ $cur == --* ]]; then
156			cmds=$($cmd --list-opts)
157		else
158			cmds=$($cmd --list-cmds)
159		fi
160		__perfcomp "$cmds" "$cur"
161	# List possible events for -e option
162	elif [[ $prev == @("-e"|"--event") &&
163		$prev_skip_opts == @(record|stat|top) ]]; then
164		evts=$($cmd list --raw-dump)
165		__perfcomp_colon "$evts" "$cur"
166	else
167		# List subcommands for perf commands
168		if [[ $prev_skip_opts == @(kvm|kmem|mem|lock|sched|
169			|data|help|script|test|timechart|trace) ]]; then
170			subcmds=$($cmd $prev_skip_opts --list-cmds)
171			__perfcomp_colon "$subcmds" "$cur"
172		fi
173		# List long option names
174		if [[ $cur == --* ]];  then
175			subcmd=$prev_skip_opts
176			__perf_prev_skip_opts $subcmd
177			subcmd=$subcmd" "$prev_skip_opts
178			opts=$($cmd $subcmd --list-opts)
179			__perfcomp "$opts" "$cur"
180		fi
181	fi
182}
183
184if [[ -n ${ZSH_VERSION-} ]]; then
185	autoload -U +X compinit && compinit
186
187	__perfcomp ()
188	{
189		emulate -L zsh
190
191		local c IFS=$' \t\n'
192		local -a array
193
194		for c in ${=1}; do
195			case $c in
196			--*=*|*.) ;;
197			*) c="$c " ;;
198			esac
199			array[${#array[@]}+1]="$c"
200		done
201
202		compset -P '*[=:]'
203		compadd -Q -S '' -a -- array && _ret=0
204	}
205
206	__perfcomp_colon ()
207	{
208		emulate -L zsh
209
210		local cur_="${2-$cur}"
211		local c IFS=$' \t\n'
212		local -a array
213
214		if [[ "$cur_" == *:* ]]; then
215			local colon_word=${cur_%"${cur_##*:}"}
216		fi
217
218		for c in ${=1}; do
219			case $c in
220			--*=*|*.) ;;
221			*) c="$c " ;;
222			esac
223			array[$#array+1]=${c#"$colon_word"}
224		done
225
226		compset -P '*[=:]'
227		compadd -Q -S '' -a -- array && _ret=0
228	}
229
230	_perf ()
231	{
232		local _ret=1 cur cword prev
233		cur=${words[CURRENT]}
234		prev=${words[CURRENT-1]}
235		let cword=CURRENT-1
236		emulate ksh -c __perf_main
237		let _ret && _default && _ret=0
238		return _ret
239	}
240
241	compdef _perf perf
242	return
243fi
244
245type perf &>/dev/null &&
246_perf()
247{
248	local cur words cword prev
249	if [ $preload_get_comp_words_by_ref = "true" ]; then
250		_get_comp_words_by_ref -n =: cur words cword prev
251	else
252		__perf_get_comp_words_by_ref -n =: cur words cword prev
253	fi
254	__perf_main
255} &&
256
257complete -o bashdefault -o default -o nospace -F _perf perf 2>/dev/null \
258	|| complete -o default -o nospace -F _perf perf
259