# -*- mode: sh; sh-indentation: 4; indent-tabs-mode: nil; sh-basic-offset: 4; -*- # Copyright (c) 2018 Sebastian Gniazdowski # Copyright (c) 2018 plexigras # # The old chroma function for command `git'. It colorizes the part of command # line that holds `git' invocation. # # $1 - 0 or 1, denoting if it's first call to the chroma, or following one # $2 - the current token, also accessible by $__arg from the above scope - # basically a private copy of $__arg # $3 - a private copy of $_start_pos, i.e. the position of the token in the # command line buffer, used to add region_highlight entry (see man), # because Zsh colorizes by *ranges* in command line buffer # $4 - a private copy of $_end_pos from the above scope # (( next_word = 2 | 8192 )) local __first_call="$1" __wrd="$2" __start_pos="$3" __end_pos="$4" local __style integer __idx1 __idx2 local -a __lines_list chroma_git_remote_subcommands chroma_git_remote_subcommands=(add rename remove set-head set-branches get-url set-url set-url set-url show prune update) if (( __first_call )); then # Called for the first time - new command # FAST_HIGHLIGHT is used because it survives between calls, and # allows to use a single global hash only, instead of multiple # global variables FAST_HIGHLIGHT[chroma-git-counter]=0 FAST_HIGHLIGHT[chroma-git-got-subcommand]=0 FAST_HIGHLIGHT[chroma-git-subcommand]="" FAST_HIGHLIGHT[chrome-git-got-msg1]=0 FAST_HIGHLIGHT[chrome-git-got-anymsg]=0 FAST_HIGHLIGHT[chrome-git-occurred-double-hyphen]=0 FAST_HIGHLIGHT[chroma-git-checkout-new]=0 FAST_HIGHLIGHT[chroma-git-fetch-multiple]=0 FAST_HIGHLIGHT[chroma-git-branch-change]=0 FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=0 return 1 else # Following call, i.e. not the first one # Check if chroma should end – test if token is of type # "starts new command", if so pass-through – chroma ends [[ "$__arg_type" = 3 ]] && return 2 if [[ "$__wrd" = "--" ]]; then FAST_HIGHLIGHT[chrome-git-occurred-double-hyphen]=1 __style=${FAST_THEME_NAME}double-hyphen-option elif [[ "$__wrd" = -* && ${FAST_HIGHLIGHT[chroma-git-got-subcommand]} -eq 0 ]]; then # Options occuring before a subcommand if (( FAST_HIGHLIGHT[chroma-git-option-with-argument-active] == 0 )); then if [[ "$__wrd" = -[^[:space:]-]#C ]]; then FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=2 elif [[ "$__wrd" = -[^[:space:]-]#c ]]; then FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=1 fi fi return 1 else # If at e.g. '>' or destination/source spec (of the redirection) if (( in_redirection > 0 || this_word & 128 )) || [[ $__wrd == "<<<" ]]; then return 1 # If at main git option taking argument in a separate word (-C and -c) elif (( FAST_HIGHLIGHT[chroma-git-option-with-argument-active] > 0 && \ 0 == FAST_HIGHLIGHT[chroma-git-got-subcommand] )) then # Remember the value __idx2=${FAST_HIGHLIGHT[chroma-git-option-with-argument-active]} # Reset the is-argument mark-field FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=0 (( __idx2 == 2 )) && return 1 # Other options' args (i.e. arg of -c) aren't routed to the big-loop # as they aren't paths and aren't handled in any special way there elif (( FAST_HIGHLIGHT[chroma-git-got-subcommand] == 0 )); then FAST_HIGHLIGHT[chroma-git-got-subcommand]=1 # Check if the command is an alias - we want to highlight the # aliased command just like the target command of the alias .fast-run-command "git config --get-regexp 'alias.*'" chroma-git-alias-list "" $(( 10 * 60 )) # Grep for line: alias.{user-entered-subcmd}[[:space:]], and remove alias. prefix __lines_list=( ${${(M)__lines_list[@]:#alias.${__wrd}[[:space:]]##*}#alias.} ) if (( ${#__lines_list} > 0 )); then # (*) # First remove alias name (#*[[:space:]]) and the space after it, then # remove any leading spaces from what's left (##[[:space:]]##), then # remove everything except the first word that's in the left line # (%%[[:space:]]##*, i.e.: "everything from right side up to any space") FAST_HIGHLIGHT[chroma-git-subcommand]="${${${__lines_list[1]#*[[:space:]]}##[[:space:]]##}%%[[:space:]]##*}" else FAST_HIGHLIGHT[chroma-git-subcommand]="$__wrd" fi if (( __start_pos >= 0 )); then # if subcommand exists LANG=C .fast-run-command "git help -a" chroma-git-subcmd-list "" $(( 10 * 60 )) # (s: :) will split on every space, but because the expression # isn't double-quoted, the empty elements will be eradicated # Some further knowledge-base: s-flag is special, it skips # empty elements and creates an array (not a concatenated # string) even when double-quoted. The normally needed @-flag # that logically breaks the concaetnated string back into array # in case of double-quoting has additional effect for s-flag: # it finally blocks empty-elements eradication. if [[ "${__lines_list[1]}" = See* ]]; then # (**) # git >= v2.20 __lines_list=( ${(M)${${${(M)__lines_list[@]:# [[:blank:]]#[a-z]*}##[[:blank:]]##}%%[[:blank:]]##*}:#${FAST_HIGHLIGHT[chroma-git-subcommand]}} ) else # (**) # git < v2.20 __lines_list=( ${(M)${(s: :)${(M)__lines_list[@]:# [a-z]*}}:#${FAST_HIGHLIGHT[chroma-git-subcommand]}} ) fi # Above we've checked: # 1) If given subcommand is an alias (*) # 2) If the command, or command pointed by the alias, exists (**) # 3) There's little problem, git v2.20 outputs aliases in git help -a, # which means that alias will be recognized as correct if it will # point at another alias or on itself. That's a minor problem, a # TODO for future planned optimization for v2.20 Git # 4) Notice that the above situation is better than the previous - the # alias is being verified to point to a valid git subcommand # That's all that's needed to decide on the correctnes: if (( ${#__lines_list} > 0 )); then __style=${FAST_THEME_NAME}subcommand else __style=${FAST_THEME_NAME}incorrect-subtle fi fi # The counter includes the subcommand itself (( FAST_HIGHLIGHT[chroma-git-counter] += 1 )) else __wrd="${__wrd//\`/x}" __arg="${__arg//\`/x}" __wrd="${(Q)__wrd}" if [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "push" \ || "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "pull" \ || "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "fetch" ]] \ && (( ${FAST_HIGHLIGHT[chroma-git-fetch-multiple]} == 0 )); then # if not option if [[ "$__wrd" != -* || "${FAST_HIGHLIGHT[chrome-git-occurred-double-hyphen]}" -eq 1 ]]; then (( FAST_HIGHLIGHT[chroma-git-counter] += 1, __idx1 = FAST_HIGHLIGHT[chroma-git-counter] )) if (( __idx1 == 2 )); then .fast-run-git-command "git remote" "chroma-git-remotes" "" else __wrd="${__wrd%%:*}" .fast-run-git-command "git for-each-ref --format='%(refname:short)' refs/heads" "chroma-git-branches" "refs/heads" fi # if remote/ref exists if [[ -n ${__lines_list[(r)$__wrd]} ]]; then (( __start=__start_pos-${#PREBUFFER}, __end=__start_pos+${#__wrd}-${#PREBUFFER}, __start >= 0 )) && \ reply+=("$__start $__end ${FAST_HIGHLIGHT_STYLES[${FAST_THEME_NAME}correct-subtle]}") # if ref (__idx1 == 3) does not exist and subcommand is push elif (( __idx1 != 2 )) && [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "push" ]]; then (( __start=__start_pos-${#PREBUFFER}, __end=__start_pos+${#__wrd}-${#PREBUFFER}, __start >= 0 )) && \ reply+=("$__start $__end ${FAST_HIGHLIGHT_STYLES[${FAST_THEME_NAME}incorrect-subtle]}") # if not existing remote name, because not an URL (i.e. no colon) elif [[ $__idx1 -eq 2 && $__wrd != *:* ]]; then (( __start=__start_pos-${#PREBUFFER}, __end=__start_pos+${#__wrd}-${#PREBUFFER}, __start >= 0 )) && \ reply+=("$__start $__end ${FAST_HIGHLIGHT_STYLES[${FAST_THEME_NAME}incorrect-subtle]}") fi # if option else if [[ "$__wrd" = "--multiple" && "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "fetch" ]]; then FAST_HIGHLIGHT[chroma-git-fetch-multiple]=1 __style=${FAST_THEME_NAME}double-hyphen-option else return 1 fi fi elif (( ${FAST_HIGHLIGHT[chroma-git-fetch-multiple]} )) \ && [[ "$__wrd" != -* || "${FAST_HIGHLIGHT[chrome-git-occurred-double-hyphen]}" -eq 1 ]]; then .fast-run-git-command "git remote" "chroma-git-remotes" "" if [[ -n ${__lines_list[(r)$__wrd]} ]]; then __style=${FAST_THEME_NAME}correct-subtle fi elif [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "commit" ]]; then match[1]="" match[2]="" # if previous argument is -m or current argument is --message=something if (( FAST_HIGHLIGHT[chrome-git-got-msg1] == 1 && ! FAST_HIGHLIGHT[chrome-git-got-anymsg] )) \ || [[ "$__wrd" = (#b)(--message=)(*) && "${FAST_HIGHLIGHT[chrome-git-occurred-double-hyphen]}" = 0 ]]; then FAST_HIGHLIGHT[chrome-git-got-msg1]=0 FAST_HIGHLIGHT[chrome-git-got-anymsg]=1 if [[ -n "${match[1]}" ]]; then __wrd="${(Q)${match[2]//\`/x}}" # highlight (--message=)something (( __start=__start_pos-${#PREBUFFER}, __end=__start_pos-${#PREBUFFER}+10, __start >= 0 )) && \ reply+=("$__start $__end ${FAST_HIGHLIGHT_STYLES[${FAST_THEME_NAME}double-hyphen-option]}") # highlight --message=(something) (( __start=__start_pos-${#PREBUFFER}+10, __end=__end_pos-${#PREBUFFER}, __start >= 0 )) && \ reply+=("$__start $__end ${FAST_HIGHLIGHT_STYLES[${FAST_THEME_NAME}double-quoted-argument]}") else (( __start=__start_pos-${#PREBUFFER}, __end=__end_pos-${#PREBUFFER}, __start >= 0 )) && \ reply+=("$__start $__end ${FAST_HIGHLIGHT_STYLES[${FAST_THEME_NAME}double-quoted-argument]}") fi local __firstline=${__wrd%%$'\n'*} if (( ${#__firstline} > 50 )); then for (( __idx1 = 1, __idx2 = 1; __idx1 <= 50; ++ __idx1, ++ __idx2 )); do while [[ "${__arg[__idx2]}" != "${__firstline[__idx1]}" ]]; do (( ++ __idx2 )) (( __idx2 > __asize )) && { __idx2=-1; break; } done (( __idx2 == -1 )) && break done if (( __idx2 != -1 )); then if [[ -n "${match[1]}" ]]; then (( __start=__start_pos-${#PREBUFFER}+__idx2, __end=__end_pos-${#PREBUFFER}-$#__wrd+$#__firstline-1, __start >= 0 )) && \ reply+=("$__start $__end ${FAST_HIGHLIGHT_STYLES[${FAST_THEME_NAME}incorrect-subtle]}") else (( __start=__start_pos-${#PREBUFFER}+__idx2-1, __end=__end_pos-${#PREBUFFER}-$#__wrd+$#__firstline-1, __start >= 0 )) && \ reply+=("$__start $__end ${FAST_HIGHLIGHT_STYLES[${FAST_THEME_NAME}incorrect-subtle]}") fi fi fi # if before -- elif [[ "${FAST_HIGHLIGHT[chrome-git-occurred-double-hyphen]}" = 0 ]]; then if [[ "$__wrd" = -[^[:space:]-]#m ]]; then FAST_HIGHLIGHT[chrome-git-got-msg1]=1 __style=${FAST_THEME_NAME}single-hyphen-option else return 1 fi # if after -- is file elif [[ -e "$__wrd" ]]; then __style=${FAST_THEME_NAME}path else __style=${FAST_THEME_NAME}incorrect-subtle fi elif [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "checkout" ]] \ || [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "revert" ]] \ || [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "merge" ]] \ || [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "diff" ]] \ || [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "reset" ]] \ || [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "rebase" ]]; then # if doing `git checkout -b ...' if [[ "$__wrd" = -[^[:space:]-]#b && "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "checkout" ]]; then FAST_HIGHLIGHT[chroma-git-checkout-new]=1 __style=${FAST_THEME_NAME}single-hyphen-option # if command is not checkout -b something elif [[ "${FAST_HIGHLIGHT[chroma-git-checkout-new]}" = 0 ]]; then # if not option if [[ "$__wrd" != -* || "${FAST_HIGHLIGHT[chrome-git-occurred-double-hyphen]}" = 1 ]]; then (( FAST_HIGHLIGHT[chroma-git-counter] += 1, __idx1 = FAST_HIGHLIGHT[chroma-git-counter] )) if (( __idx1 == 2 )) || \ [[ "$__idx1" = 3 && "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "diff" ]]; then # if is ref if command git rev-parse --verify --quiet "$__wrd" >/dev/null 2>&1; then __style=${FAST_THEME_NAME}correct-subtle # if is file and subcommand is checkout or diff elif [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "checkout" \ || "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "reset" \ || "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "diff" ]] && [[ -e ${~__wrd} ]]; then __style=${FAST_THEME_NAME}path elif [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "checkout" && \ "1" = "$(command git rev-list --count --no-walk --glob="refs/remotes/${$(git \ config --get checkout.defaultRemote):-*}/$__wrd")" ]] then __style=${FAST_THEME_NAME}correct-subtle else __style=${FAST_THEME_NAME}incorrect-subtle fi fi # if option else return 1 fi # if option elif [[ "${FAST_HIGHLIGHT[chrome-git-occurred-double-hyphen]}" = 0 && "$__wrd" = -* ]]; then return 1 fi elif [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "remote" && "$__wrd" != -* ]]; then (( FAST_HIGHLIGHT[chroma-git-counter] += 1, __idx1 = FAST_HIGHLIGHT[chroma-git-counter] )) if [[ "$__idx1" = 2 ]]; then if (( ${chroma_git_remote_subcommands[(I)$__wrd]} )); then FAST_HIGHLIGHT[chroma-git-remote-subcommand]="$__wrd" __style=${FAST_THEME_NAME}subcommand else __style=${FAST_THEME_NAME}incorrect-subtle fi elif [[ "$__idx1" = 3 && "$FAST_HIGHLIGHT[chroma-git-remote-subcommand]" = "add" ]]; then .fast-run-git-command "git remote" "chroma-git-remotes" "" if [[ -n ${__lines_list[(r)$__wrd]} ]]; then __style=${FAST_THEME_NAME}incorrect-subtle fi elif [[ "$__idx1" = 3 && -n "$FAST_HIGHLIGHT[chroma-git-remote-subcommand]" ]]; then .fast-run-git-command "git remote" "chroma-git-remotes" "" if [[ -n ${__lines_list[(r)$__wrd]} ]]; then __style=${FAST_THEME_NAME}correct-subtle else __style=${FAST_THEME_NAME}incorrect-subtle fi fi elif [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "branch" ]]; then if [[ "$__wrd" = --delete \ || "$__wrd" = --edit-description \ || "$__wrd" = --set-upstream-to=* \ || "$__wrd" = --unset-upstream \ || "$__wrd" = -[^[:space:]-]#d \ || "$__wrd" = -[^[:space:]-]#D ]]; then FAST_HIGHLIGHT[chroma-git-branch-change]=1 return 1 elif [[ "$__wrd" != -* ]]; then .fast-run-git-command "git for-each-ref --format='%(refname:short)' refs/heads" "chroma-git-branches" "refs/heads" if [[ -n ${__lines_list[(r)$__wrd]} ]]; then __style=${FAST_THEME_NAME}correct-subtle elif (( FAST_HIGHLIGHT[chroma-git-branch-change] )); then __style=${FAST_THEME_NAME}incorrect-subtle fi else return 1 fi elif [[ "${FAST_HIGHLIGHT[chroma-git-subcommand]}" = "tag" ]]; then if [[ "${FAST_HIGHLIGHT[chroma-git-option-with-argument-active]}" -le 0 ]]; then if [[ "$__wrd" = -[^[:space:]-]#(u|m) ]]; then FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=1 elif [[ "$__wrd" = -[^[:space:]-]#F ]]; then FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=2 elif [[ "$__wrd" = -[^[:space:]-]#d ]]; then FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=3 elif [[ "$__wrd" = (--contains|--no-contains|--points-at|--merged|--no-merged) ]]; then FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=4 fi if [[ "$__wrd" != -* ]]; then (( FAST_HIGHLIGHT[chroma-git-counter] += 1, __idx1 = FAST_HIGHLIGHT[chroma-git-counter] )) if [[ ${FAST_HIGHLIGHT[chroma-git-counter]} -eq 2 ]]; then .fast-run-git-command "git for-each-ref --format='%(refname:short)' refs/heads" "chroma-git-branches" "refs/heads" .fast-run-git-command "+git tag" "chroma-git-tags" "" [[ -n ${__lines_list[(r)$__wrd]} ]] && __style=${FAST_THEME_NAME}incorrect-subtle elif [[ ${FAST_HIGHLIGHT[chroma-git-counter]} -eq 3 ]]; then fi else return 1 fi else case "${FAST_HIGHLIGHT[chroma-git-option-with-argument-active]}" in (1) __style=${FAST_THEME_NAME}optarg-string ;; (2) FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=0 return 1; ;; (3) .fast-run-git-command "git tag" "chroma-git-tags" "" [[ -n ${__lines_list[(r)$__wrd]} ]] && \ __style=${FAST_THEME_NAME}correct-subtle || \ __style=${FAST_THEME_NAME}incorrect-subtle ;; (4) if git rev-parse --verify --quiet "$__wrd" >/dev/null 2>&1; then __style=${FAST_THEME_NAME}correct-subtle else __style=${FAST_THEME_NAME}incorrect-subtle fi ;; esac FAST_HIGHLIGHT[chroma-git-option-with-argument-active]=0 fi else return 1 fi fi fi fi # Add region_highlight entry (via `reply' array) if [[ -n "$__style" ]]; then (( __start=__start_pos-${#PREBUFFER}, __end=__end_pos-${#PREBUFFER}, __start >= 0 )) \ && reply+=("$__start $__end ${FAST_HIGHLIGHT_STYLES[$__style]}") fi # We aren't passing-through, do obligatory things ourselves (( this_word = next_word )) _start_pos=$_end_pos return 0 # vim:ft=zsh:et:sw=4