perf-completion.sh 6.3 KB

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