blob: bdd40350870ae1b190f6b84eed1fc1dfd7f01a3f [file] [log] [blame]
Ramkumar Ramachandraf38ab8af2013-11-17 21:43:26 +05301# perf bash and zsh completion
Frederic Weisbecker98a41792012-08-09 16:31:51 +02002
Ramkumar Ramachandrac3fb6712013-07-04 18:11:30 +05303# 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
Ramkumar Ramachandra4685a6c2013-07-04 18:11:29 +053079type __ltrim_colon_completions &>/dev/null ||
Namhyung Kimae0c1f92012-10-04 14:23:54 +090080__ltrim_colon_completions()
81{
82 if [[ "$1" == *:* && "$COMP_WORDBREAKS" == *:* ]]; then
83 # Remove colon-word prefix from COMPREPLY items
Ramkumar Ramachandra30079d12013-07-04 18:11:26 +053084 local colon_word=${1%"${1##*:}"}
Namhyung Kimae0c1f92012-10-04 14:23:54 +090085 local i=${#COMPREPLY[*]}
86 while [[ $((--i)) -ge 0 ]]; do
87 COMPREPLY[$i]=${COMPREPLY[$i]#"$colon_word"}
88 done
89 fi
90}
91
Ramkumar Ramachandra12f9dd52013-11-17 21:43:24 +053092__perfcomp ()
93{
94 COMPREPLY=( $( compgen -W "$1" -- "$2" ) )
95}
96
Ramkumar Ramachandra37e72c32013-11-17 21:43:25 +053097__perfcomp_colon ()
98{
99 __perfcomp "$1" "$2"
100 __ltrim_colon_completions $cur
101}
102
Yunlong Song67afff42015-03-18 21:35:47 +0800103__perf_prev_skip_opts ()
104{
105 local i cmd_ cmds_
106
107 let i=cword-1
Yunlong Songeee200a2015-03-18 21:35:48 +0800108 cmds_=$($cmd $1 --list-cmds)
Yunlong Song67afff42015-03-18 21:35:47 +0800109 prev_skip_opts=()
110 while [ $i -ge 0 ]; do
Yunlong Songeee200a2015-03-18 21:35:48 +0800111 if [[ ${words[i]} == $1 ]]; then
112 return
113 fi
Yunlong Song67afff42015-03-18 21:35:47 +0800114 for cmd_ in $cmds_; do
115 if [[ ${words[i]} == $cmd_ ]]; then
116 prev_skip_opts=${words[i]}
117 return
118 fi
119 done
120 ((i--))
121 done
122}
Yunlong Song6fdd9cb2015-03-18 21:35:57 +0800123
Ramkumar Ramachandra2cf025e2013-11-17 21:43:23 +0530124__perf_main ()
Frederic Weisbecker98a41792012-08-09 16:31:51 +0200125{
Ramkumar Ramachandra2cf025e2013-11-17 21:43:23 +0530126 local cmd
Frederic Weisbecker98a41792012-08-09 16:31:51 +0200127
Ramkumar Ramachandra6e0dc372013-07-04 18:11:31 +0530128 cmd=${words[0]}
Ramkumar Ramachandra2cf025e2013-11-17 21:43:23 +0530129 COMPREPLY=()
Frederic Weisbecker98a41792012-08-09 16:31:51 +0200130
Yunlong Song67afff42015-03-18 21:35:47 +0800131 # Skip options backward and find the last perf command
132 __perf_prev_skip_opts
Namhyung Kim35c2fde2012-10-03 00:21:33 +0900133 # List perf subcommands or long options
Yunlong Songe003ce52015-03-18 21:35:51 +0800134 if [ -z $prev_skip_opts ]; then
Namhyung Kim35c2fde2012-10-03 00:21:33 +0900135 if [[ $cur == --* ]]; then
Yunlong Song73353992015-02-27 18:21:31 +0800136 cmds=$($cmd --list-opts)
Namhyung Kim35c2fde2012-10-03 00:21:33 +0900137 else
138 cmds=$($cmd --list-cmds)
Namhyung Kim35c2fde2012-10-03 00:21:33 +0900139 fi
Yunlong Song73353992015-02-27 18:21:31 +0800140 __perfcomp "$cmds" "$cur"
Yunlong Song6fdd9cb2015-03-18 21:35:57 +0800141 # List possible events for -e option
142 elif [[ $prev == @("-e"|"--event") &&
143 $prev_skip_opts == @(record|stat|top) ]]; then
Namhyung Kim4d8061f2012-10-03 00:21:34 +0900144 evts=$($cmd list --raw-dump)
Ramkumar Ramachandra37e72c32013-11-17 21:43:25 +0530145 __perfcomp_colon "$evts" "$cur"
Yunlong Song02fde322015-03-18 21:35:46 +0800146 else
147 # List subcommands for perf commands
Yunlong Song6fdd9cb2015-03-18 21:35:57 +0800148 if [[ $prev_skip_opts == @(kvm|kmem|mem|lock|sched|
149 |data|help|script|test|timechart|trace) ]]; then
Yunlong Song67afff42015-03-18 21:35:47 +0800150 subcmds=$($cmd $prev_skip_opts --list-cmds)
Yunlong Song02fde322015-03-18 21:35:46 +0800151 __perfcomp_colon "$subcmds" "$cur"
152 fi
153 # List long option names
154 if [[ $cur == --* ]]; then
Yunlong Songeee200a2015-03-18 21:35:48 +0800155 subcmd=$prev_skip_opts
156 __perf_prev_skip_opts $subcmd
157 subcmd=$subcmd" "$prev_skip_opts
Yunlong Song02fde322015-03-18 21:35:46 +0800158 opts=$($cmd $subcmd --list-opts)
159 __perfcomp "$opts" "$cur"
160 fi
Frederic Weisbecker98a41792012-08-09 16:31:51 +0200161 fi
Ramkumar Ramachandra2cf025e2013-11-17 21:43:23 +0530162}
163
Ramkumar Ramachandraf38ab8af2013-11-17 21:43:26 +0530164if [[ -n ${ZSH_VERSION-} ]]; then
165 autoload -U +X compinit && compinit
166
167 __perfcomp ()
168 {
169 emulate -L zsh
170
171 local c IFS=$' \t\n'
172 local -a array
173
174 for c in ${=1}; do
175 case $c in
176 --*=*|*.) ;;
177 *) c="$c " ;;
178 esac
179 array[${#array[@]}+1]="$c"
180 done
181
182 compset -P '*[=:]'
183 compadd -Q -S '' -a -- array && _ret=0
184 }
185
186 __perfcomp_colon ()
187 {
188 emulate -L zsh
189
190 local cur_="${2-$cur}"
191 local c IFS=$' \t\n'
192 local -a array
193
194 if [[ "$cur_" == *:* ]]; then
195 local colon_word=${cur_%"${cur_##*:}"}
196 fi
197
198 for c in ${=1}; do
199 case $c in
200 --*=*|*.) ;;
201 *) c="$c " ;;
202 esac
203 array[$#array+1]=${c#"$colon_word"}
204 done
205
206 compset -P '*[=:]'
207 compadd -Q -S '' -a -- array && _ret=0
208 }
209
210 _perf ()
211 {
212 local _ret=1 cur cword prev
213 cur=${words[CURRENT]}
214 prev=${words[CURRENT-1]}
215 let cword=CURRENT-1
216 emulate ksh -c __perf_main
217 let _ret && _default && _ret=0
218 return _ret
219 }
220
221 compdef _perf perf
222 return
223fi
224
Ramkumar Ramachandra2cf025e2013-11-17 21:43:23 +0530225type perf &>/dev/null &&
226_perf()
227{
228 local cur words cword prev
229 _get_comp_words_by_ref -n =: cur words cword prev
230 __perf_main
Frederic Weisbecker98a41792012-08-09 16:31:51 +0200231} &&
Ramkumar Ramachandra7b6c48e2013-07-04 18:11:27 +0530232
233complete -o bashdefault -o default -o nospace -F _perf perf 2>/dev/null \
234 || complete -o default -o nospace -F _perf perf