ohmyzsh/plugins/colorize/colorize.plugin.zsh

83 lines
2.5 KiB
Bash
Raw Normal View History

# easier alias to use the plugin
alias ccat='colorize_via_pygmentize'
alias cless='colorize_via_pygmentize_less'
2013-04-12 16:19:42 +02:00
colorize_via_pygmentize() {
local available_tools=("chroma" "pygmentize")
2013-04-12 16:19:42 +02:00
2019-11-15 20:11:50 +01:00
if [ -z $ZSH_COLORIZE_TOOL ]; then
if (( $+commands[pygmentize] )); then
ZSH_COLORIZE_TOOL="pygmentize"
elif (( $+commands[chroma] )); then
ZSH_COLORIZE_TOOL="chroma"
else
echo "Neither 'pygments' nor 'chroma' is installed!"
2019-11-15 20:11:50 +01:00
return 1
fi
fi
if [[ ${available_tools[(Ie)$ZSH_COLORIZE_TOOL]} -eq 0 ]]; then
echo "ZSH_COLORIZE_TOOL '$ZSH_COLORIZE_TOOL' not recognized. Available options are 'pygmentize' and 'chroma'."
return 1
elif (( $+commands["$ZSH_COLORIZE_TOOL"] )); then
echo "Package '$ZSH_COLORIZE_TOOL' is not installed!"
return 1
fi
2019-11-15 20:11:50 +01:00
# If the environment variable ZSH_COLORIZE_STYLE
# is set, use that theme instead. Otherwise,
# use the default.
if [ -z $ZSH_COLORIZE_STYLE ]; then
# Both pygmentize & chroma support 'emacs'
ZSH_COLORIZE_STYLE="emacs"
fi
# pygmentize stdin if no arguments passed
2013-04-12 16:19:42 +02:00
if [ $# -eq 0 ]; then
2019-11-15 20:11:50 +01:00
if [[ $ZSH_COLORIZE_TOOL == "pygmentize" ]]; then
pygmentize -O style="$ZSH_COLORIZE_STYLE" -g
else
chroma --style="$ZSH_COLORIZE_STYLE"
fi
return $?
2013-04-12 16:19:42 +02:00
fi
# guess lexer from file extension, or
# guess it from file contents if unsuccessful
local FNAME lexer
for FNAME in "$@"
2013-04-12 16:19:42 +02:00
do
2019-11-15 20:11:50 +01:00
if [[ $ZSH_COLORIZE_TOOL == "pygmentize" ]]; then
lexer=$(pygmentize -N "$FNAME")
if [[ $lexer != text ]]; then
pygmentize -O style="$ZSH_COLORIZE_STYLE" -l "$lexer" "$FNAME"
else
pygmentize -O style="$ZSH_COLORIZE_STYLE" -g "$FNAME"
fi
2013-04-12 16:19:42 +02:00
else
2019-11-15 20:11:50 +01:00
chroma --style="$ZSH_COLORIZE_STYLE" "$FNAME"
2013-04-12 16:19:42 +02:00
fi
done
2014-05-08 20:19:37 +02:00
}
colorize_via_pygmentize_less() (
# this function is a subshell so tmp_files can be shared to cleanup function
declare -a tmp_files
cleanup () {
[[ ${#tmp_files} -gt 0 ]] && rm -f "${tmp_files[@]}"
exit
}
trap 'cleanup' EXIT HUP TERM INT
while (( $# != 0 )); do #TODO: filter out less opts
tmp_file="$(mktemp -t "tmp.colorize.XXXX.$(sed 's/\//./g' <<< "$1")")"
tmp_files+=("$tmp_file")
colorize_via_pygmentize "$1" > "$tmp_file"
shift 1
done
less -f "${tmp_files[@]}"
)