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 commands 125 elif [[ $prev == @(kvm|kmem|mem|lock|sched) ]]; then 126 subcmds=$($cmd $prev --list-cmds) 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