lib/rake_git/tasks/commit.rb in rake_git-0.1.0.pre.2 vs lib/rake_git/tasks/commit.rb in rake_git-0.1.0.pre.3

@@ -1,16 +1,56 @@
 # frozen_string_literal: true
 
 require 'rake_factory'
+require 'git'
 
 module RakeGit
   module Tasks
     class Commit < RakeFactory::Task
       default_name :commit
-      default_description 'Produce a commit on the current branch.'
+      default_description 'Commit changes to git.'
 
+      parameter :message, required: true
+
+      parameter :stage_tracked_files, default: true
+      parameter :stage_untracked_files, default: true
+
+      parameter :working_directory, default: '.'
+
       action do
-        puts('Committing...')
+        puts('Committing changes to git...')
+        base = Git.open(working_directory)
+
+        maybe_stage_tracked_files(base)
+        maybe_stage_untracked_files(base)
+        commit(base)
+      end
+
+      private
+
+      def maybe_stage_tracked_files(base)
+        return unless stage_tracked_files
+
+        status = base.status
+        base.add(
+          [
+            *status.added.keys,
+            *status.changed.keys,
+            *status.deleted.keys
+          ],
+          all: true
+        )
+      end
+
+      def maybe_stage_untracked_files(base)
+        return unless stage_untracked_files
+
+        status = base.status
+        base.add(status.untracked.keys, all: true)
+      end
+
+      def commit(base)
+        base.commit(message)
       end
     end
   end
 end