lib/metric_fu/metrics/flog/generator.rb in metric_fu-4.12.0 vs lib/metric_fu/metrics/flog/generator.rb in metric_fu-4.13.0

@@ -11,11 +11,11 @@
       parse_options = FlogCLI.parse_options [
         "--all",
         options[:continue] ? "--continue" : nil,
       ].compact
       @flogger = FlogCLI.new parse_options
-      @flogger.flog *options[:dirs_to_flog]
+      @flogger.flog *files_to_flog
     end
 
     def analyze
       @method_containers = {}
       @flogger.calculate
@@ -48,9 +48,21 @@
           file, line = data[:path].split(":")
 
           out[file][line] << { type: :flog, description: "Score of %.2f" % data[:score] }
         end
       end
+    end
+
+    private
+
+    def files_to_flog
+      options[:dirs_to_flog].flatten.map do |p|
+        if File.directory? p then
+          Dir[File.join(p, '**/*.{rb,rake}')]
+        else
+          p
+        end
+      end.flatten
     end
   end
 
   class MethodContainer
     attr_reader :methods