bin/method_log in method_log-0.0.3 vs bin/method_log in method_log-0.0.4

@@ -2,18 +2,26 @@
 
 require 'bundler/setup'
 require 'method_log'
 require 'method_log/repository'
 require 'method_log/api'
+require 'trollop'
 
+options = Trollop::options do
+  opt :patch, 'Generate patch.', short: 'p'
+  opt :max_count, 'Limit the number of commits to output.', type: :integer, short: 'n'
+end
+
 repository = MethodLog::Repository.new(path: Dir.pwd)
 api = MethodLog::API.new(repository: repository)
-api.diffs(ARGV[0]).each do |method_commit, method_diff|
+api.diffs(ARGV[0], max_count: options[:max_count]).each do |method_commit, method_diff|
   puts "commit #{method_commit.sha}"
   puts "Author: #{method_commit.author[:name]} <#{method_commit.author[:email]}>"
   puts "Date:   #{method_commit.author[:time].strftime('%a %b %-e %T %Y %z')}"
   puts
   puts method_commit.message
   puts
-  puts method_diff.to_s(:color)
-  puts
+  if options[:patch]
+    puts method_diff.to_s(:color)
+    puts
+  end
 end