From 412f2755468146a03a9ccbcd950e2e0231a81af1 Mon Sep 17 00:00:00 2001
From: Eduardo Bizarro <edbizarro@gmail.com>
Date: Sun, 16 Oct 2016 23:41:26 -0200
Subject: [PATCH] yarn completion

---
 plugins/yarn/_yarn           | 57 ++++++++++++++++++++++
 plugins/yarn/yarn.plugin.zsh | 95 ++++++++++++++++++++++++++++++++++++
 2 files changed, 152 insertions(+)
 create mode 100644 plugins/yarn/_yarn
 create mode 100644 plugins/yarn/yarn.plugin.zsh

diff --git a/plugins/yarn/_yarn b/plugins/yarn/_yarn
new file mode 100644
index 000000000..6bf8e51f0
--- /dev/null
+++ b/plugins/yarn/_yarn
@@ -0,0 +1,57 @@
+
+
+# Credits to bower's awesome completion utility.
+#
+# yarn completion script, based on bower completion script.
+
+###-begin-yarn-completion-###
+#
+# Installation: yarn completion >> ~/.bashrc  (or ~/.zshrc)
+# Or, maybe: yarn completion > /usr/local/etc/bash_completion.d/yarn
+#
+
+COMP_WORDBREAKS=${COMP_WORDBREAKS/=/}
+COMP_WORDBREAKS=${COMP_WORDBREAKS/@/}
+export COMP_WORDBREAKS
+
+if type complete &>/dev/null; then
+  _yarn_completion () {
+    local si="$IFS"
+    IFS=$'\n' COMPREPLY=($(COMP_CWORD="$COMP_CWORD" \
+                           COMP_LINE="$COMP_LINE" \
+                           COMP_POINT="$COMP_POINT" \
+                           yarn completion -- "${COMP_WORDS[@]}" \
+                           2>/dev/null)) || return $?
+    IFS="$si"
+  }
+  complete -F _yarn_completion yarn --help
+elif type compdef &>/dev/null; then
+  _yarn_completion() {
+    si=$IFS
+    compadd -- $(COMP_CWORD=$((CURRENT-1)) \
+                 COMP_LINE=$BUFFER \
+                 COMP_POINT=0 \
+                 yarn completion -- "${words[@]}" \
+                 2>/dev/null)
+    IFS=$si
+  }
+  compdef _yarn_completion yarn --help
+elif type compctl &>/dev/null; then
+  _yarn_completion () {
+    local cword line point words si
+    read -Ac words
+    read -cn cword
+    let cword-=1
+    read -l line
+    read -ln point
+    si="$IFS"
+    IFS=$'\n' reply=($(COMP_CWORD="$cword" \
+                       COMP_LINE="$line" \
+                       COMP_POINT="$point" \
+                       yarn completion -- "${words[@]}" \
+                       2>/dev/null)) || return $?
+    IFS="$si"
+  }
+  compctl -K _yarn_completion yarn --help
+fi
+###-end-yarn-completion-###
diff --git a/plugins/yarn/yarn.plugin.zsh b/plugins/yarn/yarn.plugin.zsh
new file mode 100644
index 000000000..349e081d7
--- /dev/null
+++ b/plugins/yarn/yarn.plugin.zsh
@@ -0,0 +1,95 @@
+alias yi="yarn install"
+
+_yarn_installed_packages () {
+    yarn_package_list=$(yarn ls --no-color 2>/dev/null| awk 'NR>3{print p}{p=$0}'| cut -d ' ' -f 2|sed 's/#.*//')
+}
+_yarn ()
+{
+    local -a _1st_arguments _no_color _dopts _dev _production
+    local expl
+    typeset -A opt_args
+
+    _no_color=('--no-color[Do not print colors (available in all commands)]')
+
+    _dopts=(
+        '(--force)--force[This refetches all packages, even ones that were previously installed.]'
+    )
+
+    _installopts=(
+        '(--flat)--flat[Only allow one version of a package. On the first run this will prompt you to choose a single version for each package that is depended on at multiple version ranges.]'
+        '(--har)--har[Outputs an HTTP archive from all the network requests performed during the installation.]'
+        '(--no-lockfile)--no-lockfile[Don’t read or generate a yarn.lock lockfile.]'
+        '(--pure-lockfile)--pure-lockfile[Don’t generate a yarn.lock lockfile.]'
+    )
+
+    _dev=('(--dev)--dev[Save installed packages into the project"s package.json devDependencies]')
+
+    _production=('(--production)--production[Do not install project devDependencies]')
+
+    _1st_arguments=(
+    'help:Display help information about yarn' \
+    'init:Initialize for the development of a package.' \
+    'add:Add a package to use in your current package.' \
+    'install:Install all the dependencies listed within package.json in the local node_modules folder.' \
+    'publish:Publish a package to a package manager.' \
+    'remove:Remove a package that will no longer be used in your current package.' \
+    'cache:Clear the local cache. It will be populated again the next time yarn or yarn install is run.' \
+    'clean:Frees up space by removing unnecessary files and folders from dependencies.' \
+    'check:Verifies that versions of the package dependencies in the current project’s package.json matches that of yarn’s lock file.' \
+    'ls:List all installed packages.' \
+    'global:Makes binaries available to use on your operating system.' \
+    'info:<package> [<field>] - fetch information about a package and return it in a tree format.' \
+    'outdated:Checks for outdated package dependencies.' \
+    'run:Runs a defined package script.' \
+    'self-update:Updates Yarn to the latest version.' \
+    'upgrade:Upgrades packages to their latest version based on the specified range.' \
+    'why:<query> - Show information about why a package is installed.'
+    )
+    _arguments \
+    $_no_color \
+    '*:: :->subcmds' && return 0
+
+    if (( CURRENT == 1 )); then
+        _describe -t commands "yarn subcommand" _1st_arguments
+        return
+    fi
+
+    case "$words[1]" in
+        add)
+        _arguments \
+        $_dopts \
+        $_dev \
+        $_no_color \
+        $_production
+        ;;
+        install)
+        _arguments \
+        $_installopts \
+        $_dopts \
+        $_dev \
+        $_no_color \
+        $_production
+        ;;
+        update)
+        _arguments \
+        $_dopts \
+        $_no_color \
+        _yarn_installed_packages
+        compadd "$@" $(echo $yarn_package_list)
+        ;;
+        uninstall)
+        _arguments \
+        $_no_color \
+        $_dopts
+        _yarn_installed_packages
+        compadd "$@" $(echo $yarn_package_list)
+        ;;
+        *)
+        _arguments \
+        $_no_color \
+        ;;
+    esac
+
+}
+
+compdef _yarn yarn