lib/rubycritic/report_generators/line_generator.rb in rubycritic-0.0.1 vs lib/rubycritic/report_generators/line_generator.rb in rubycritic-0.0.2

@@ -1,13 +1,13 @@
-require "cgi"
 require "erb"
+require "rubycritic/report_generators/base_generator"
+require "cgi"
 
 module Rubycritic
 
-  class LineGenerator
+  class LineGenerator < BaseGenerator
     LINE_NUMBER_PADDING = 3
-    TEMPLATES_DIR = File.expand_path("../templates", __FILE__)
     NORMAL_TEMPLATE = ERB.new(File.read(File.join(TEMPLATES_DIR, "line.html.erb")))
     SMELLY_TEMPLATE = ERB.new(File.read(File.join(TEMPLATES_DIR, "smelly_line.html.erb")))
 
     def initialize(text, number, smells)
       @text = CGI::escapeHTML(text.chomp)
@@ -19,11 +19,11 @@
         else
           SMELLY_TEMPLATE
         end
     end
 
-    def output
-      @output ||= @template.result(binding).delete("\n") + "\n"
+    def render
+      @template.result(binding).delete("\n") + "\n"
     end
   end
 
 end