From 7c630309cc3a2267f3a08eaf1c408ee4069ee5a7 Mon Sep 17 00:00:00 2001
From: Roman Kapitonov <RomanKapitonov@coherentsolutions.com>
Date: Wed, 2 Sep 2015 20:29:17 +0300
Subject: [PATCH] Add ability to autocomplete services for brew-services.

---
 plugins/brew/_brew | 31 ++++++++++++++++++++++++++++++-
 1 file changed, 30 insertions(+), 1 deletion(-)

diff --git a/plugins/brew/_brew b/plugins/brew/_brew
index 1f24bd67b..19cfb22cb 100644
--- a/plugins/brew/_brew
+++ b/plugins/brew/_brew
@@ -33,6 +33,10 @@ _brew_outdated_formulae() {
   outdated_formulae=(`brew outdated`)
 }
 
+_brew_installed_services() {
+  installed_services=(`brew services list | awk '{print $1}' | tail -n+2`)
+}
+
 local -a _1st_arguments
 _1st_arguments=(
   'audit:check formulae for Homebrew coding style'
@@ -64,6 +68,7 @@ _1st_arguments=(
   'prune:remove dead links'
   'remove:remove a formula'
   'search:search for a formula (/regex/ or string)'
+  'services:manage services'
   'switch:switch between different versions of a formula'
   'tap:tap a new formula repository from GitHub, or list existing taps'
   'tap-info:information about a tap'
@@ -82,7 +87,7 @@ _1st_arguments=(
 )
 
 local expl
-local -a formulae installed_formulae installed_taps official_taps outdated_formulae
+local -a formulae installed_formulae installed_taps official_taps outdated_formulae installed_services
 
 _arguments \
   '(-v)-v[verbose]' \
@@ -142,4 +147,28 @@ case "$words[1]" in
   upgrade)
     _brew_outdated_formulae
     _wanted outdated_formulae expl 'outdated formulae' compadd -a outdated_formulae ;;
+  services)
+     _arguments -C \
+      '1: :->command' \
+      '2: :->service' && return 0
+
+    local -a commands
+
+    commands=(
+      'cleanup:Get rid of stale services and unused plists'
+      'list:List all services managed by brew services'
+      'restart:Gracefully restart selected service'
+      'start:Start selected service'
+      'stop:Stop selected service'
+    )
+
+    case $state in
+      command)
+        _describe -t commands 'Action' commands
+        ;;
+      service)
+        _brew_installed_services
+       _describe -t subcommands 'Services' installed_services
+       ;;
+    esac
 esac