Adds command line argument for adjusting timeout

lib/flog.rb

@@ -20,6 +20,11 @@ class Flog < MethodBasedSexpProcessor
 
   THRESHOLD = DEFAULT_THRESHOLD # :nodoc:
 
+  ##
+  # Timeout in seconds, can be set with --timeout.
+
+  DEFAULT_TIMEOUT = 10
+
   ##
   # The scoring system hash. Maps node type to score.
 
@@ -101,6 +106,7 @@ class Flog < MethodBasedSexpProcessor
   attr_reader :method_scores, :scores
   attr_reader :total_score, :totals
   attr_writer :threshold
+  attr_reader :timeout
 
   # :startdoc:
 
@@ -190,8 +196,8 @@ def flog(*files)
   # for methods. Smart. Handles syntax errors and timeouts so you
   # don't have to.
 
-  def flog_ruby ruby, file="-", timeout = 10
-    flog_ruby! ruby, file, timeout
+  def flog_ruby ruby, file="-", timeout_override = nil
+    flog_ruby! ruby, file, timeout_override || timeout
   rescue Timeout::Error
     warn "TIMEOUT parsing #{file}. Skipping."
   rescue RubyParser::SyntaxError, Racc::ParseError => e
@@ -216,12 +222,12 @@ def flog_ruby ruby, file="-", timeout = 10
   # Flog the given ruby source, optionally using file to provide paths for
   # methods. Does not handle timeouts or syntax errors. See #flog_ruby.
 
-  def flog_ruby! ruby, file="-", timeout = 10
+  def flog_ruby! ruby, file="-", timeout_override = nil
     @parser = (option[:parser] || RubyParser).new
 
     warn "** flogging #{file}" if option[:verbose]
 
-    ast = @parser.process ruby, file, timeout
+    ast = @parser.process ruby, file, timeout_override ||timeout
 
     return unless ast
 
@@ -238,6 +244,7 @@ def initialize option = {}
     @mass                = {}
     @parser              = nil
     @threshold           = option[:threshold] || DEFAULT_THRESHOLD
+    @timeout             = option[:timeout] || DEFAULT_TIMEOUT
     self.auto_shift_type = true
     self.reset
   end

lib/flog_cli.rb

@@ -131,6 +131,10 @@ def self.parse_options args = ARGV
         option[:verbose] = true
       end
 
+      opts.on("--timeout=N", Integer, "Set the timeout (in sec) for parsing (default: 10s)") do |n|
+        option[:timeout] = n
+      end
+
       next if self.plugins.empty?
       opts.separator "Plugin options:"
 

test/test_flog_cli.rb

@@ -39,6 +39,7 @@ def test_cls_parse_options
       "--score"        => :score,
       "-t"             => [:threshold, "75", 0.75],
       "--threshold"    => [:threshold, "75", 0.75],
+      "--timeout"      => [:timeout, "60", 60],
       "-v"             => :verbose,
       "--verbose"      => :verbose,
       # TODO: (maybe)