2018-10-20 01:27:49 +08:00
|
|
|
# easier alias to use the plugin
|
2018-01-23 11:18:03 +08:00
|
|
|
alias ccat='colorize_via_pygmentize'
|
2019-03-06 14:34:12 +08:00
|
|
|
alias cless='colorize_via_pygmentize_less'
|
2013-04-12 22:19:42 +08:00
|
|
|
|
|
|
|
colorize_via_pygmentize() {
|
2018-10-08 23:56:44 +08:00
|
|
|
if ! (( $+commands[pygmentize] )); then
|
|
|
|
echo "package 'Pygments' is not installed!"
|
|
|
|
return 1
|
2013-04-12 22:19:42 +08:00
|
|
|
fi
|
|
|
|
|
2019-03-06 14:34:12 +08:00
|
|
|
# If the environment varianle ZSH_COLORIZE_STYLE
|
|
|
|
# is set, use that theme instead. Otherwise,
|
|
|
|
# use the default.
|
|
|
|
if [ -z $ZSH_COLORIZE_STYLE ]; then
|
|
|
|
ZSH_COLORIZE_STYLE="default"
|
|
|
|
fi
|
|
|
|
|
2018-10-20 01:27:49 +08:00
|
|
|
# pygmentize stdin if no arguments passed
|
2013-04-12 22:19:42 +08:00
|
|
|
if [ $# -eq 0 ]; then
|
2019-03-06 14:34:12 +08:00
|
|
|
pygmentize -O style="$ZSH_COLORIZE_STYLE" -g
|
2018-10-20 01:27:49 +08:00
|
|
|
return $?
|
2013-04-12 22:19:42 +08:00
|
|
|
fi
|
|
|
|
|
2018-10-20 01:27:49 +08:00
|
|
|
# guess lexer from file extension, or
|
|
|
|
# guess it from file contents if unsuccessful
|
2019-03-06 14:34:12 +08:00
|
|
|
|
2018-10-20 01:27:49 +08:00
|
|
|
local FNAME lexer
|
2019-04-23 03:06:47 +08:00
|
|
|
for FNAME in "$@"
|
2013-04-12 22:19:42 +08:00
|
|
|
do
|
2018-10-20 01:27:49 +08:00
|
|
|
lexer=$(pygmentize -N "$FNAME")
|
|
|
|
if [[ $lexer != text ]]; then
|
2019-03-06 14:34:12 +08:00
|
|
|
pygmentize -O style="$ZSH_COLORIZE_STYLE" -l "$lexer" "$FNAME"
|
2013-04-12 22:19:42 +08:00
|
|
|
else
|
2019-03-06 14:34:12 +08:00
|
|
|
pygmentize -O style="$ZSH_COLORIZE_STYLE" -g "$FNAME"
|
2013-04-12 22:19:42 +08:00
|
|
|
fi
|
|
|
|
done
|
2014-05-09 02:19:37 +08:00
|
|
|
}
|
2019-04-23 03:06:47 +08:00
|
|
|
|
2019-03-06 14:34:12 +08:00
|
|
|
colorize_via_pygmentize_less() (
|
2019-04-23 03:06:47 +08:00
|
|
|
# this function is a subshell so tmp_files can be shared to cleanup function
|
|
|
|
declare -a tmp_files
|
2019-03-06 14:34:12 +08:00
|
|
|
|
2019-04-23 03:06:47 +08:00
|
|
|
cleanup () {
|
|
|
|
[[ ${#tmp_files} -gt 0 ]] && rm -f "${tmp_files[@]}"
|
|
|
|
exit
|
|
|
|
}
|
|
|
|
trap 'cleanup' EXIT HUP TERM INT
|
2019-03-06 14:34:12 +08:00
|
|
|
|
2019-04-23 03:06:47 +08:00
|
|
|
while (( $# != 0 )); do #TODO: filter out less opts
|
2019-04-30 02:51:29 +08:00
|
|
|
tmp_file="$(mktemp -t "tmp.colorize.XXXX.$(sed 's/\//./g' <<< "$1")")"
|
2019-04-23 03:06:47 +08:00
|
|
|
tmp_files+=("$tmp_file")
|
|
|
|
colorize_via_pygmentize "$1" > "$tmp_file"
|
|
|
|
shift 1
|
|
|
|
done
|
|
|
|
|
|
|
|
less -f "${tmp_files[@]}"
|
|
|
|
)
|