code_file.rb in rubycritic-0.0.15

- old
+ new

@@ -1,6 +1,5 @@
-require "erb"
 require "rubycritic/report_generators/base"
 require "rubycritic/report_generators/line"
 
 module Rubycritic
   module Generator
@@ -10,11 +9,10 @@
       TEMPLATE = erb_template("code_file.html.erb")
 
       def initialize(analysed_file)
         @analysed_file = analysed_file
         @pathname = @analysed_file.pathname
-        @smells = @analysed_file.smells
       end
 
       def file_directory
         File.join(root_directory, @pathname.dirname)
       end
@@ -25,10 +23,10 @@
 
       def render
         file_code = ""
         File.readlines(@pathname).each.with_index(LINE_NUMBER_OFFSET) do |line_text, line_number|
           location = Location.new(@pathname, line_number)
-          line_smells = @smells.select { |smell| smell.at_location?(location) }
+          line_smells = @analysed_file.smells_at_location(location)
           file_code << Line.new(line_text, line_smells).render
         end
 
         file_body = TEMPLATE.result(get_binding { file_code })
         LAYOUT_TEMPLATE.result(get_binding { file_body })