lib/backends/flog_backend.rb

   1 require 'flog'
   2
   3 class FlogBackend < Backend
   4   class << self
   5     def run(file)
   6       @file = file
   7       @layers = {}
   8
   9       flogger = Flog.new
  10       flogger.flog_files @file
  11       flogger.totals.each { |method, score| record method, score }
  12       write_layers
  13     end
  14
  15     def record(method, score)
  16       return if method =~ /\#none$/
  17       color = 'red' # TODO
  18       message = "#{method} flogs at #{score}"
  19       (@layers[@file] ||= []) << Layer.new(method, color, message, self, @file)
  20     end
  21   end
  22
  23   Augment::BACKENDS['flog'] = self
  24 end