Add --workdir option

Allows for both staged and unstaged files to be checked.

Author: tylerhunt

lib/pronto/cli.rb

@@ -33,6 +33,11 @@ def is_thor_reserved_word?(word, type)
                   type: :boolean,
                   desc: 'Analyze changes in git staging area'
 
+    method_option :workdir,
+                  type: :boolean,
+                  aliases: ['-w'],
+                  desc: 'Analyze both staged and unstaged changes'
+
     method_option :runner,
                   type: :array,
                   default: [],
@@ -55,7 +60,7 @@ def run(path = '.')
 
       formatters = ::Pronto::Formatter.get(options[:formatters])
 
-      commit_options = %i[staged unstaged index]
+      commit_options = %i[workdir staged unstaged index]
       commit = commit_options.find { |o| options[o] } || options[:commit]
 
       repo_workdir = ::Rugged::Repository.discover(path).workdir

lib/pronto/git/repository.rb

@@ -13,6 +13,18 @@ def diff(commit, options = nil)
                             [head_commit_sha, @repo.index.diff(options)]
                           when :staged
                             [head_commit_sha, head.diff(@repo.index, options)]
+                          when :workdir
+                            [
+                              head_commit_sha,
+                              @repo.diff_workdir(
+                                head,
+                                {
+                                  include_untracked: true,
+                                  include_untracked_content: true,
+                                  recurse_untracked_dirs: true
+                                }.merge(options || {})
+                              )
+                            ]
                           else
                             merge_base = merge_base(commit)
                             patches = @repo.diff(merge_base, head, options)

spec/pronto/git/repository_spec.rb

@@ -92,6 +92,15 @@ module Git
           let(:sha) { :staged }
           it { should be_one }
         end
+
+        context 'workdir' do
+          let(:sha) { :workdir }
+
+          it do
+            # this count includes all the files from the repositories (*.git)
+            subject.count.should eq 48
+          end
+        end
       end
 
       describe '#blame' do