linux/ohmyzsh
1
0
Fork 0
mirror of https://github.com/ohmyzsh/ohmyzsh.git synced 2024-11-12 21:39:48 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

colorize: add style environment variable (#7635)

Browse source
This commit is contained in:
Marc Cornellà 2019-06-15 20:48:45 +02:00 committed by GitHub
commit fb43df128f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 21 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
plugins/colorize/README.md
View file

@ -11,6 +11,12 @@ To use it, add colorize to the plugins array of your zshrc file:
plugins=(... colorize) plugins=(... colorize)
``` ```
## Styles
Pygments offers multiple styles. By default, the `default` style is used, but you can choose another theme by setting the `ZSH_COLORIZE_STYLE` environment variable:
`ZSH_COLORIZE_STYLE="colorful"`
## Usage ## Usage
* `ccat <file> [files]`: colorize the contents of the file (or files, if more than one are provided). * `ccat <file> [files]`: colorize the contents of the file (or files, if more than one are provided).

22
plugins/colorize/colorize.plugin.zsh
View file

@ -1,6 +1,6 @@
# easier alias to use the plugin # easier alias to use the plugin
alias ccat='colorize_via_pygmentize' alias ccat='colorize_via_pygmentize'
alias cless='colorize_via_pygmentize_less' alias cless='colorize_via_pygmentize_less'
colorize_via_pygmentize() { colorize_via_pygmentize() {
if ! (( $+commands[pygmentize] )); then if ! (( $+commands[pygmentize] )); then
@ -8,36 +8,44 @@ colorize_via_pygmentize() {
return 1 return 1
fi fi
# 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
# pygmentize stdin if no arguments passed # pygmentize stdin if no arguments passed
if [ $# -eq 0 ]; then if [ $# -eq 0 ]; then
pygmentize -g pygmentize -O style="$ZSH_COLORIZE_STYLE" -g
return $? return $?
fi fi
# guess lexer from file extension, or # guess lexer from file extension, or
# guess it from file contents if unsuccessful # guess it from file contents if unsuccessful
local FNAME lexer local FNAME lexer
for FNAME in "$@" for FNAME in "$@"
do do
lexer=$(pygmentize -N "$FNAME") lexer=$(pygmentize -N "$FNAME")
if [[ $lexer != text ]]; then if [[ $lexer != text ]]; then
pygmentize -l "$lexer" "$FNAME" pygmentize -O style="$ZSH_COLORIZE_STYLE" -l "$lexer" "$FNAME"
else else
pygmentize -g "$FNAME" pygmentize -O style="$ZSH_COLORIZE_STYLE" -g "$FNAME"
fi fi
done done
} }
colorize_via_pygmentize_less() ( colorize_via_pygmentize_less() (
# this function is a subshell so tmp_files can be shared to cleanup function # this function is a subshell so tmp_files can be shared to cleanup function
declare -a tmp_files declare -a tmp_files
cleanup () { cleanup () {
[[ ${#tmp_files} -gt 0 ]] && rm -f "${tmp_files[@]}" [[ ${#tmp_files} -gt 0 ]] && rm -f "${tmp_files[@]}"
exit exit
} }
trap 'cleanup' EXIT HUP TERM INT trap 'cleanup' EXIT HUP TERM INT
while (( $# != 0 )); do #TODO: filter out less opts while (( $# != 0 )); do #TODO: filter out less opts
tmp_file="$(mktemp -t "tmp.colorize.XXXX.$(sed 's/\//./g' <<< "$1")")" tmp_file="$(mktemp -t "tmp.colorize.XXXX.$(sed 's/\//./g' <<< "$1")")"
tmp_files+=("$tmp_file") tmp_files+=("$tmp_file")