From 65304ec3f2199ed4c482fbe8a068f07881c3be5b Mon Sep 17 00:00:00 2001
From: Andrew Janke <andrew@apjanke.net>
Date: Mon, 11 Jun 2018 15:35:55 -0400
Subject: [PATCH] brew style zsh completion: complete files

---
 completions/zsh/_brew | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/completions/zsh/_brew b/completions/zsh/_brew
index a83ef2ec35..e339fa03d5 100644
--- a/completions/zsh/_brew
+++ b/completions/zsh/_brew
@@ -32,6 +32,13 @@ __brew_list_aliases() {
   echo "${aliases}"
 }
 
+__brew_formulae_or_ruby_files() {
+  local -a formulae
+  formulae=($(brew search))
+  _alternative 'formulae::__brew_formulae' \
+    'files:files:{_files -g *.rb}'
+}
+
 __brew_formulae() {
   local -a formulae
   formulae=($(brew search))
@@ -571,8 +578,7 @@ _brew_style() {
   _arguments \
     '(--fix)--fix[fix style violations automatically]' \
     '(--display-cop-names)--display-cop-names[include RuboCop name for each violation in output]' \
-    '*::formula:__brew_formulae'
-    # TODO add files to completion
+    '*::formula:__brew_formulae_or_ruby_files'
 }
 
 # brew switch name version: