#compdef ag
# ------------------------------------------------------------------------------
# Description
# -----------
#
#  Completion script for ag (https://github.com/ggreer/the_silver_searcher)
#
# ------------------------------------------------------------------------------
# Authors
# -------
#
#  * Akira Maeda <https://github.com/glidenote>
#
# ------------------------------------------------------------------------------
# -*- mode: zsh; sh-indentation: 2; indent-tabs-mode: nil; sh-basic-offset: 2; -*-
# vim: ft=zsh sw=2 ts=2 et
# ------------------------------------------------------------------------------
_ag_version() {
  local version cnt=0
  for i in $(ag --version); do
    if [[ $cnt -eq 2 ]]; then
      version=$i
      break
    fi
    ((cnt += 1))
  done

  version=( "${(@s/./)version}" )
  printf "${version[2]}"
}

# Dynamically build the file type completion
# Modifies the global $__AG_OPTS array
_ag_add_file_types() {
  local ttype exts
  for i in $(ag --list-file-types); do
    if [[ "$i" =~ '--' ]]; then
      if [[ "${ttype}x" != "x" ]]; then
        __AG_OPTS+="(${ttype})${ttype}[${exts%% }]"
      fi
      ttype=$i
      exts=
    else
      exts+="$i "
    fi
  done
  __AG_OPTS+="(${ttype})${ttype}[${exts%% }]"
}

# Add version appropriate options above base
# Modifies the global $__AG_OPTS array
_ag_add_version_opts() {
  local minor
  minor=$(_ag_version)

  if [[ $minor -gt 15 ]];then
    __AG_OPTS+=(
      '(--color-line-number)--color-line-number[Color codes for line numbers. Default is 1;33.]'
      '(--color-match)--color-match[Color codes for result match numbers. Default is 30;43.]'
      '(--color-path)--color-path[Color codes for path names. Default is 1;32.]'
    )
  fi

  if [[ $minor -gt 21 ]];then
    _ag_add_file_types
    __AG_OPTS+=(
      '(--list-file-types)--list-file-types[list supported filetypes to search]'
      '(--silent)--silent[suppress all log messages, including errors]'
    )
  fi

  if [[ $minor -gt 22 ]];then
    __AG_OPTS+=(
      '(-z --search-zip)'{-z,--search-zip}'[search contents of compressed files]'
    )
  fi

  if [[ $minor -le 24 ]];then
    __AG_OPTS+=(
      '(--noheading --heading)'{--noheading,--heading}'[print file names above matching contents]'
      '(-s --case-sensitive)'{-s,--case-sensitive}'[match case sensitively]'
    )
  fi
  if [[ $minor -gt 24 ]];then
    __AG_OPTS+=(
      '(-H --noheading --heading)'{-H,--noheading,--heading}'[print file names above matching contents]'
      '(-s --case-sensitive)'{-s,--case-sensitive}'[Match case sensitively. Default on.]'
      '(--vimgrep)--vimgrep[output results like vim''s, :vimgrep /pattern/g would (report every match on the line)]'
    )
  fi

  if [[ $minor -gt 26 ]];then
    __AG_OPTS+=(
      '(-0 --null --print0)'{-0,--null,--print0}'[separate the filenames with \\0, rather than \\n]'
    )
  fi

  if [[ $minor -le 27 ]];then
    __AG_OPTS+=(
      '(--depth)--depth[Search up to NUM directories deep. Default is 25.]:NUM'
    )
  fi
  if [[ $minor -gt 27 ]];then
    __AG_OPTS+=(
      '(-c --count)'{-c,--count}'[only print the number of matches in each file]'
      '(--depth)--depth[Search up to NUM directories deep, -1 for unlimited. Default is 25.]:NUM'
      '(-F --fixed-strings)'{-F,--fixed-strings}'[alias for --literal for compatibility with grep]'
    )
  fi

  if [[ $minor -le 28 ]];then
    __AG_OPTS+=(
      '(--no-numbers)--no-numbers[don´t show line numbers]'
    )
  fi
  if [[ $minor -gt 28 ]];then
    __AG_OPTS+=(
      '(--nofilename --filename)'{--nofilename,--filename}'[Print file names. Default on, except when searching a single file.]'
      '(--nonumbers --numbers)'{--nonumbers,--numbers}'[Print line numbers. Default is to omit line numbers when searching streams]'
      '(-o --only-matching)'{-o,--only-matching}'[print only the matching part of the lines]'
    )
  fi
}

_ag() {
  local curcontext="$curcontext" state line cmds update_policy ret=1

  zstyle -s ":completion:${curcontext}:" cache-policy update_policy
  [[ -z "$update_policy" ]] && zstyle ":completion:${curcontext}:" cache-policy _ag_types_caching_policy

  if ( [[ ${+__AG_OPTS} -eq 0 ]] || _cache_invalid "AG_OPTS" ) && ! _retrieve_cache "AG_OPTS"; then
    # Base opts starts at ag version 0.14
    __AG_OPTS=(
      '(- 1 *)--help[print a short help statement]'
      '(- 1 *)--man[print the manual page]'
      '(- 1 *)--version[display version and copyright information]'
      '(--ackmate)--ackmate[output results in a format parseable by AckMate]'
      '(--column)--column[print column numbers in results]'
      '(--hidden)--hidden[search hidden files, still obeys ignore files.]'
      '(--ignore)--ignore[Ignore files/directories matching this pattern. Literal file and directory names are also allowed.]'
      '(--ignore-dir)--ignore-dir[alias for --ignore for compatability with ack]'
      '(--nobreak --break)'{--nobreak,--break}'[Print a newline between matches in different files. Default on.]'
      '(--nocolor --color)'{--nocolor,--color}'[Print color codes in results. Default on.]'
      '(--nogroup --group)'{--nogroup,--group}'[same as --\[no\]break --\[no\]heading]'
      '(--pager --nopager)'{--pager,--nopager}'[Display results with PAGER. Disabled by default.]'
      '(--passthrough)--passthrough[when searching a stream, print all lines even if they don''t match]'
      '(--print-long-lines)--print-long-lines[Print matches on very long lines (> 2k characters by default)]'
      '(--search-binary)--search-binary[search binary files for matches]'
      '(--stats)--stats[print stats (files scanned, time taken, etc)]'
      '(-A --after)'{-A,--after}'[Print NUM lines before match. Default is 2]:LINES'
      '(-B --before)'{-B,--before}'[Print NUM lines after match. Defaults is 2]:LINES'
      '(-C --context)'{-C,--context}'[Print NUM lines before and after matches. Default is 2.]:LINES'
      '(-D --debug)'{-D,--debug}'[enable debug logging]'
      '(-G --file-search-regex)'{-G,--file-search-regex}'[only search file names matching PATTERN]:PATTERN'
      '(-L --files-without-matches)'{-L,--files-without-matches}"[only print filenames that don't contain matches]"
      '(-Q --literal)'{-Q,--literal}'[Do not parse PATTERN as a regular expression. Try to match it literally.]'
      '(-S --smart-case)'{-S,--smart-case}'[match case sensitively if PATTERN contains any uppercase letters, else match case insensitively]'
      '(-U --skip-vcs-ignores)'{-U,--skip-vcs-ignores}'[ignore VCS ignore files (.gitigore, .hgignore, svn:ignore), but still use .agignore]'
      '(-a --all-types)'{-a,--all-types}"[Search all files. This doesn't include hidden files, and also doesn't respect any ignore files.]"
      '(-f --follow)'{-f,--follow}'[follow symlinks]'
      '(-g)-g[print filenames that match PATTERN]:PATTERN'
      '(-i --ignore-case)'{-i,--ignore-case}'[match case insensitively]:PATTERN'
      '(-l --files-with-matches)'{-l,--files-with-matches}'[only print filenames containing matches, not matching lines]'
      '(-m --max-count)'{-m,--max-count}'[Skip the rest of a file after NUM matches. Default is 10,000.]:NUM'
      '(-p --path-to-agignore)'{-p,--path-to-agignore}'[provide a path to a specific .agignore file]:STRING'
      '(-t --all-text)'{-t,--all-text}"[search all text files, not including hidden]"
      '(-u --unrestricted)'{-u,--unrestricted}'[Search *all* files. This ignores .agignore, .gitignore, etc. It searches binary and hidden files as well.]'
      '(-v --invert-match)'{-v,--invert-match}'[invert match]'
      '(-w --word-regexp)'{-w,--word-regexp}'[only match whole words]'
      '*: :_files'
      '1: :->patterns'
    )
    _ag_add_version_opts
    [[ $#__AG_OPTS -gt 0 ]] && _store_cache 'AG_OPTS' __AG_OPTS
  fi

  _arguments -C ${__AG_OPTS} && ret=0
  unset __AG_OPTS

  case $state in
    patterns)
      _message -e patterns 'pattern' && ret=0
      ;;
  esac

  return ret
}

_ag_types_caching_policy() {
  # Rebuild if .agignore more recent than cache.
  [[ -f $HOME/.agignore && $$HOME/.agignore -nt "$1" ]] && return 0

  # Rebuild if cache is older than one week.
  local -a oldp
  oldp=( "$1"(Nmw+1) )
  (( $#oldp )) && return 0

  return 1
}

_ag "$@"