How to display numeric arg right away

[edshamis]

I'd like the numeric arg to be displayed at prompt right away when used.
Is it possible?

[romkatv]

What's numeric arg?

[edshamis]

Pressing Alt-{number} repeats the following pressed char or zle command {number} times.
It is accessible in user defined widget in $NUMERIC

[romkatv]

I think it's a great idea to display the numeric argument in prompt. Do you know of a zsh prompt/theme that does it?

[edshamis]

Nice you liked the idea.
Your great theme is the only one I know and use

[edshamis]

Here's my implementation:

.p10k.zsh

function prompt_numeric() {
  [[ -n $NUMERIC ]] && p10k segment -f $_P10K_COLOR_GREEN_BRIGHT -i $NUMERIC
}

.zshrc

digit_argument() {
  zle .digit-argument
  NUMERIC=$NUMERIC${KEYS[-1]}
  -z4h-redraw-prompt
}
zle -N digit_argument
for i in {0..9}; do
  bindkey "^[$i" digit_argument
done

neg-argument() {
  zle .neg-argument
  (( NUMERIC = ${NUMERIC:-1} * -1 ))
  -z4h-redraw-prompt
}
zle -N neg-argument

numeric-clear-hook() {
  if [[ $LASTWIDGET == (digit_argument|neg-argument) ]] then
    LASTNUMERIC=1
  elif [[ -n $LASTNUMERIC ]] then
    unset LASTNUMERIC NUMERIC
    -z4h-redraw-prompt
  fi
}
add-zle-hook-widget line-pre-redraw numeric-clear-hook