Bash/Shell通过配置文件.bashrc劫持命令行,实现输入数字前缀即可重复多次执行某条命令,替代手敲for循环

stevehe 2023年02月23日 93次浏览

要实现的效果

输入数字N + 空格 + 任意命令行:重复执行该条命令N次:
da531c4451488ad672f1e682ba45e8cd_176373202302231548107981610109854.png
常规做法:

for i in {1..5};do pwd;done   #此种做法命令略显冗长,特别命令和参数较长的话不方便光标移动和高效操作

96a584d7f24f8e1d146ad8fe7fe46537_17637320230223155832832529472516.png

改造方式:

通过Bash Shell配置文件(.bashrc)文件动态判断输入的命令行是否是以一个纯数字打头,如果是,执行我们预定义的操作来实现目的;

先决条件:

安装Bash Shell第三方开源拓展工具库bash-preexec:( https://github.com/rcaloras/bash-preexec
此库封装了两个回调函数preexecprecmd,分别对应命令执行前和命令执行后要执行的hook操作;
利用此库提供的这俩回调函数功能,可以实现更多骚操作(例如:动态更改Prompt提示符,计算命令的执行时间、终端输入算术表达式直接计算结果等等),此处不再赘述,具体用法参看官方Github文档说明;

Hook代码实现:

1、在preexec回调函数中加入以下代码块:

preexec() {
        exec 7>&1 8>&2
	export COMMAND_EXEC_BEGINNING=$(date +'%s')   #开始执行命令时记录时间戳
	#echo "$BASH_COMMAND"  #<---当前执行的完整命令行
	local _prefixCommand=$(awk '{print $1;exit}' <<<"$BASH_COMMAND")
	local _commandPartCount=$(awk '{print NF;exit}' <<<"$BASH_COMMAND")
	[ $_commandPartCount -gt 1 ] && [ ! -z "$_prefixCommand" ] && ! type -t "$_prefixCommand" &>/dev/null && expr "$_prefixCommand" + 0 &>/dev/null && {
	    #(注意执行命令时如果需要保存输出,需要使用追加重定向符号,覆盖重定向符号不起作用,疑似被覆盖?)
		#eg:5 date>>datelog.log【有效】
		#    5 date>datelog.log 【无效】
		#echo "执行多次命令..."
		local _rCount="$_prefixCommand"
		local _command=$(awk '{sub($1" ","");print;}' <<<"$BASH_COMMAND")
		for _i in `seq 1 $_rCount`
		do
			#echo "执行第 $_i 次:$_command"
			eval "$_command"
		done
		#unset PROMPT_COMMAND
		#unset BASH_COMMAND
		#_no_error() {
		#	eval ""
		#}
		#trap _no_error 0
		#exec 1>/dev/null 2>&1
		exec 2>/dev/null   #<---屏蔽掉数字开头的命令的错误输出
		return
		}
}

2、在precmd回调函数中加入以下代码块:

precmd() { 
 #exec 1>/dev/tty 2>&1
 [ -t 7 ] && exec 1>&7 7>&-   #恢复标准输出
 [ -t 8 ] && exec 2>&8 8>&-   #恢复错误输出
 [ -n "$COMP_LINE" ] && return  # do nothing if completing
 [ "$BASH_COMMAND" = "$PROMPT_COMMAND" ] && return # don't cause a preexec for $PROMPT_COMMAND
 local this_command=`HISTTIMEFORMAT= history 1 | sed -e "s/^[ ]*[0-9]*[ ]*//"`;
 #echo "printing the prompt $this_command"; 
 #echo $this_command
}

使用效果: