lb.plugin.zsh

# -*- sh -*-

# 2022-03-16 change from autoload to plugin
# printf "Initializing lb\n"


# if [[ $TERM_PROGRAM == *iTerm* ]]; then

  # Argument to line functions specifies line thickness
  function __lb_iyline() {
    [ $1 ] && ht=$1 || ht=1;
    printf "\e]1337;File=name=eWVsbG93;size=113;inline=1;width=100%%;height=%dpx;preserveAspectRatio=no:iVBORw0KGgoAAAANSUhEUgAAAAEAAAABAQMAAAAl21bKAAAAA1BMVEX//wCKxvRFAAAACklEQVQI12NgAAAAAgAB4iG8MwAAAABJRU5ErkJggg==\a\n" "$ht"
  }

  function __lb_irline() {
    [ $1 ] && ht=$1 || ht=1;
    printf "\e]1337;File=name=cmVk;size=113;inline=1;width=100%%;height=%dpx;preserveAspectRatio=no:iVBORw0KGgoAAAANSUhEUgAAAAEAAAABAQMAAAAl21bKAAAAA1BMVEX/AAAZ4gk3AAAACklEQVQI12NgAAAAAgAB4iG8MwAAAABJRU5ErkJggg==\a\n" "$ht"
    }

  function __lb_icline {
    [ $1 ] && ht=$1 || ht=1;
    printf "\e]1337;File=name=Y3lhbg==;size=113;inline=1;width=100%%;height=%dpx;preserveAspectRatio=no:iVBORw0KGgoAAAANSUhEUgAAAAEAAAABAQMAAAAl21bKAAAAA1BMVEUA//8ZXC8lAAAACklEQVQI12NgAAAAAgAB4iG8MwAAAABJRU5ErkJggg==\a\n" "$ht"
  }

# else
  # Argument is ignored as a repeating number is ugly
  function __lb_tyline () { local S='-'; printf -v _hr "%*s" $(tput cols) && echo ${_hr// /${S--}} }
  function __lb_trline () { local S='-'; printf -v _hr "%*s" $(tput cols) && echo ${_hr// /${S--}} }
  function __lb_tcline () { local S='-'; printf -v _hr "%*s" $(tput cols) && echo ${_hr// /${S--}} }
# fi

function _lb_ident () {
  local cmd pad

  for cmd in $@; do
    printf "%b%s\n" $pad "$cmd"
    strings $cmd | command egrep '^ *\$.*\$$|^@\(#\)' | sed 's/^/  /'
    pad="\n"
  done
}

function lb__l {
  # NOTE: changing order here does not change ls order, still need -rt
  local -a ta=( $(printf "%s\n" $@ | sort -u ) )

  type els > /dev/null
  [[ $? == 1 ]] && /bin/ls -lrt $ta || els +T^NY-M-DT +G~Atp~ugsmnL -rt $ta  # l

  if (( lb_diff )) && {
    ta=( $( /bin/ls -rt $ta ) )
    printf "\n"
    mdiff -W $ta
  }
}
function lb_ls {

  # NOTE: changing order here does not change ls order, still need -rt
  local -a ta=( $(printf "%s\n" $@ | sort -u ) )

  type els > /dev/null
  [[ $? == 0 ]] && /bin/ls -rt $ta    || els +T^NY-M-DT +G~At~smn -rt $ta      # ll

  if (( lb_diff )) && {
    ta=( $( /bin/ls -rt $ta ) )
    printf "\n"
    mdiff -W $ta
  }
}

function lb_hl {                   # hl -I -g $cmd
  while IFS= read -r; do
    GREP_COLOR="07;32" egrep --color=always "$1|\$" <<< "$REPLY"
  done
}

function lb_ccol {                 # ccol -y 4 -g -s 6 20
  local ccols=( 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 )
  local cmd

  on=""                   # yellow bold
  off=""
  cmd="sed 's/[^[:blank:]]\{1,\}/${on}&${off}/4; "    # set first cmd

  on=""                   # green bold

  for col in $ccols; do
    cmd="$cmd s/[^[:blank:]]\{1,\}/${on}&${off}/$col;"
  done
  cmd="$cmd '"

  while read -r line; do
    echo "$line" | ( eval $cmd )
  done
}

function lb_prism {                # prism -F:;
  local cmd

  on=""        # red normal
  off=""

  cmd="sed 's/[^:]\{1,\}/${on}&${off}/1;"             # set first cmd

  on=""                   # red bold

  cmd="$cmd s/[^:]\{1,\}/${on}&${off}/2 '"

  while read -r line; do
    echo "$line" | ( eval $cmd )
  done

}

function lb_exe {
  local l_cmd cmd txt flg
  if (( lb_long )); then
    l_cmd=lb__l
  else
    l_cmd=lb_ls
  fi


  cmd=$( type $1 )
  flg=$2

  setopt localoptions no_nomatch
  local -a o_cmd=()
  [[ -e $HOME/Build ]] && o_cmd=( $(command ls $HOME/Build/**/$cmd:t(.) 2>/dev/null) )     # find all occurances under Build

  a_cmd=$(type $1 | egrep -v "alias |shell " | sed 's/^.* \([^ ][^ ]\)/\1/g')
  [[ $a_cmd == "" ]] && a_cmd=$cmd && printf "\n"

  if [[ $flg == 1 ]]; then           # executable
    if (( lb_verb )) || (( lb_file )); then
      txt=$( file $a_cmd )
      printf "%s\n" $txt | lb_prism
    fi
  fi

  $l_cmd $o_cmd $(\type -a $1 | egrep -v "alias|shell" | sed 's/^.* \([^ ][^ ]\)/\1/g') | lb_hl $a_cmd

  if (( lb_verb )); then
    echo $txt | grep -q text;
    if ! (( $? )); then
      $__lb_cline 5
      $cat $a_cmd
      $__lb_cline 5
    fi
  fi

  if (( lb_ident )); then
    $__lb_yline 2
    _lb_ident $(\type -a $1 | egrep -v "alias|shell" | sed 's/^.* \([^ ][^ ]\)/\1/g') | lb_hl $a_cmd
    $__lb_yline 2
  fi

  if (( lb_comp  )); then
    $__lb_yline 2
    printf "%s Cmd Completion: %s\n" $1 $_comps[$1]
    if (( lb_verb )); then
      lb -v $_comps[$1]
    fi
    $__lb_yline 2
  fi

  if (( lb_path )); then
    printf "Removing %s from path\n" $1
    eval $(pathtool -z -p $PATH $1)
  fi
}

function lb_alias {
  local oRT=$REPORTTIME
  REPORTTIME=-1
  zsh -ixc : 2>&1 | grep "\<$1\>=" | head -1
  REPORTTIME=$oRT
}

function lb_global {
  local oRT=$REPORTTIME
  REPORTTIME=-1

  local ga
  foreach ga in ${(ok)galiases[@]}; do
    lb -lv "$ga"
  done | align -F" is a " " is "

  REPORTTIME=$oRT

}

function lb_usage() {
  for (( i=1; i<=${#myopts}; i++ )); do
    printf "-%c\n" ${myopts[i]}
  done
}

function lb_help {
  printf "%s -%s CMD [CMD]\n\n" $0 $myopts
  printf "Find location of executable, function, or alias\n"
  printf "  -C: colorize source using colorize_cat\n"
  printf "  -L: colorize source using colorize_less\n"
  printf "  -c: show assigned completion command\n"
  printf "  -f: show 'file' output for executables\n"
  printf "  -g: show all global aliases\n"
  printf "  -i: show 'ident' information\n"
  printf "  -l: long ls output\n"
  printf "  -r: reload function\n"
  printf "  -e: edit function\n"
  printf "  -a: show where alias is defined\n"
  printf "  -p: remove path containing program\n"
  printf "  -t: force text for line separators\n"
  printf "  -d: diff multiple locations of same item\n"
  printf "  -v: show script and function source\n"
  printf "\n"
}

function lb {
  local cat=cat;
  local lb_comp=0;            # show completion command
  local lb_diff=0;            # diff multiple matches
  local lb_file=0;
  local lb_long=0;
  local lb_verb=0;
  local lb_ident=0;
  local lb_rload=0;           # resource file containing function
  local lb_edit=0;
  local lb_find=0;
  local lb_path=0;
  local lb_galias=0;

  if [[ $TERM_PROGRAM == *iTerm* ]]; then
    __lb_yline=__lb_iyline
    __lb_rline=__lb_irline
    __lb_cline=__lb_icline
  else
    __lb_yline=__lb_tyline
    __lb_rline=__lb_trline
    __lb_cline=__lb_tcline
  fi

  local myopts="acdefgilprtuABCFLvh"
  while getopts $myopts opt; do
    case $opt in
      C) cat=colorize_cat ; ;;
      L) cat=colorize_less; ;;  # colorize_cat uses default tab stops
      A|B|C|F|G);    ;;         # ignore, used by completion expansion
      c) lb_comp=1;  ;;
      d) lb_diff=1;  ;;
      f) lb_file=1;  ;;
      g) lb_galias=1;;;
      i) lb_ident=1; ;;
      l) lb_long=1;  ;;
      r) lb_rload=1; ;;
      e) lb_edit=1;  ;;
      a) lb_find=1;  ;;         # find alias
      p) lb_path=1;  ;;         # delete path to program
      v) lb_verb=1;  ;;
      t) __lb_yline=__lb_tyline;
         __lb_rline=__lb_trline;
         __lb_cline=__lb_tcline;
         printf "Texting!\n";
         ;;

      u) lb_usage;
         ;;
      ?) lb_help;
         ;;
    esac
  done
  shift $((OPTIND -1))

  [[ $lb_galias == 1 ]] && lb_global && return 0

  while test "$#" -gt 0; do
    cmd="$1"; shift;
    x=$( whence -v $cmd )
    a=("${(@s/ /)x}")
    c=$( type -a $cmd | wc -l )

    if [[ $a[2] == "not" ]]; then
      printf "%s\n" $x
    elif [[ $#a -eq 3 ]]; then           # executable
      lb_exe $cmd 1
    else                                 # alias or shell function
      type -a $cmd | lb_ccol
#     printf "%s\n" $x | lb_ccol

      if [[ $a[4] == "shell" ]]; then    # shell
        if (( lb_verb )); then
          printf "\n"
          $__lb_yline 5
          which $cmd | $cat # --         # -- borked: lb -C -v gp
          printf "\n"
          $__lb_rline 5
        fi

        if (( lb_rload )); then
          printf "Reload from: %s" $a[7]
          [[ -e $a[7] ]] && source $a[7]
          [[ $? ]] && printf " - \e[01;32mSuccess\e[m\n" || printf " - \e[01;31mFAIL]e[m\n"
        fi

        if (( lb_edit )); then
          type zed 2>&1 > /dev/null || autoload zed
          zed -f $cmd
        fi
      else                               # alias
        [[ $lb_find -eq 1 ]] && printf "\n" && lb_alias $cmd
      fi

      if (( lb_comp  )); then
        $__lb_yline 2
        printf "%s Completion: %s\n" $cmd $_comps[$cmd]
        if (( lb_verb )); then
          lb -v $_comps[$cmd]
        fi
        $__lb_yline 2
      fi

      # 2023-05-10: This breaks with multiply defined aliases/functions
      # why is it needed?
#     if [[ $c -gt 1 ]] lb_exe $cmd 0
    fi

    if [[ $# -gt 0 ]]; then
      printf "\n"
    fi
  done

}

_lbparam() {
  lb -u | while read option; do
    _wanted Foo expl Option compadd -- "$option"
  done
}

_lbcmds () {

  f=0
  for w in ${words[@]}; do
    case $w in
      (-C|-B|-F|-G|-A) f=1; ;;
    esac
  done

  [[ $f -eq 0 ]] && words=(-A -B -C -F)

  for w in ${words[@]}; do
    case $w in
      (-C) _wanted Boz expl Option _path_commands;       ;;
      (-B) _wanted Boz expl Option compadd -k builtins;  ;;
      (-F) _wanted Boz expl Option compadd -k functions; ;;
      (-G)
          zstyle ':completion::(^approximate*):*:functions' ignored-patterns '_*'
          _wanted Boz expl Option _functions;           ;;
      (-A) _wanted Boz expl Option compadd -k aliases;   ;;
    esac
  done
}


# _arguments: the '*' allows the pattern to be repeated on the cmdline

function _lb () {
  _alternative \
    "*:Foo      :_lbparam" \
    "*:Commands :_lbcmds"  \
}

compdef _lb lb

# This is only for an autoload function
# lb $*