RubyGems - daddy - Versions diffs - 0.0.3 → 0.0.4 - Mend

daddy 0.0.3 → 0.0.4

Files changed (20) hide show

checksums.yaml +8 -8
data/lib/daddy/cucumber.rb +48 -0
data/lib/daddy/cucumber/assert.rb +121 -0
data/lib/daddy/cucumber/capture.rb +32 -0
data/lib/daddy/cucumber/step_definitions/common.rb +33 -0
data/lib/daddy/cucumber/table.rb +24 -0
data/lib/daddy/formatter/cucumber.css +285 -0
data/lib/daddy/formatter/daddy.css +70 -0
data/lib/daddy/formatter/html.rb +745 -0
data/lib/daddy/formatter/jquery-min.js +5 -0
data/lib/daddy/formatter/menu.html.erb +26 -0
data/lib/daddy/railtie.rb +16 -0
data/lib/tasks/cucumber.rake +20 -0
data/lib/tasks/cucumber_install.rake +33 -0
data/lib/tasks/cucumber_steps.rake +34 -0
data/lib/tasks/dad.rake +6 -0
data/lib/tasks/db_create.rake +27 -0
data/lib/tasks/publish.rake +32 -0
data/lib/tasks/test.rake +11 -0
metadata +19 -1

data/lib/daddy/formatter/daddy.css ADDED Viewed

@@ -0,0 +1,70 @@
+#menu {
+  width: 120px;
+  float: left;
+  background: #FFFFFF;
+  color: #000000;
+}
+#menu ol {
+  margin: 0;
+  padding: 0;
+}
+#menu ol li.sprint {
+  margin: 0;
+  padding: 0 5px;
+  border-left: 5px solid #65c400;
+  border-bottom: 1px solid #65c400;
+  line-height: 40px;
+  vertical-align: middle;
+  font-size: 1.2em;
+  font-weight: bold;
+}
+#menu ol li.diary {
+  margin: 0;
+  padding: 0 20px 0 40px;
+  border-left: 5px solid #65c400;
+  border-bottom: 1px solid #65c400;
+  line-height: 40px;
+  vertical-align: middle;
+  font-size: 1.2em;
+  font-weight: bold;
+}
+.contents {
+  margin-left: 120px;
+}
+.cucumber .scenario h3, td .scenario h3, th .scenario h3, .background h3 {
+  font-size: 11px;
+  padding: 3px;
+  margin: 0;
+  background: #FFFFFF;
+  color: #000000;
+  font-weight: bold;
+}
+.cucumber ol li.passed, td ol li.passed, th ol li.passed {
+  border-left: 5px solid #65c400;
+  border-bottom: 1px solid #65c400;
+  background: #FFFFFF;
+  color: #000000;
+}
+.step_contents {
+  display: none;
+  background: #FFFFFF;
+  border: solid 1px #999999;
+  margin-left: 10px;
+  padding: 10px;
+}
+div.scenario table {
+  margin: 10px 0;
+}
+div.scenario td {
+  border: solid 1px #999999;
+  padding: 5px;
+}

data/lib/daddy/formatter/html.rb ADDED Viewed

@@ -0,0 +1,745 @@
+# coding: UTF-8
+require 'erb'
+require 'cucumber/formatter/ordered_xml_markup'
+require 'cucumber/formatter/duration'
+require 'cucumber/formatter/io'
+module Daddy
+  module Formatter
+    class Html
+      include ERB::Util # for the #h method
+      include ::Cucumber::Formatter::Duration
+      include ::Cucumber::Formatter::Io
+      def initialize(runtime, path_or_io, options)
+        @path_or_io = path_or_io
+        @io = ensure_io(path_or_io, "html")
+        @runtime = runtime
+        @options = options
+        @buffer = {}
+        @builder = create_builder(@io)
+        @feature_number = 0
+        @scenario_number = 0
+        @step_number = 0
+        @header_red = nil
+        @delayed_messages = []
+        @img_id = 0
+      end
+      def embed(src, mime_type, label)
+        case(mime_type)
+        when /^image\/(png|gif|jpg|jpeg)/
+          embed_image(src, label)
+        end
+      end
+      def embed_image(src, label)
+        id = "img_#{@img_id}"
+        @img_id += 1
+        @builder.span(:class => 'embed') do |pre|
+          pre << %{<a href="" onclick="img=document.getElementById('#{id}'); img.style.display = (img.style.display == 'none' ? 'block' : 'none');return false">#{label}</a><br>&nbsp;
+          <img id="#{id}" style="display: none" src="#{src}"/>}
+        end
+      end
+      def before_features(features)
+        @step_count = get_step_count(features)
+        @builder.declare!(:DOCTYPE, :html)
+        @builder << '<html>'
+        @builder << '<head>'
+          @builder.meta('http-equiv' => 'Content-Type', :content => 'text/html;charset=utf-8')
+          @builder.title 'Daddy'
+          inline_css
+          inline_js
+        @builder << '</head>'
+        @builder << '<body>'
+        @builder << "<!-- Step count #{@step_count}-->"
+        @builder << '<div class="cucumber">'
+        @builder.div(:id => 'cucumber-header') do
+          @builder.div(:id => 'label') do
+            @builder.h1('Daddy')
+          end
+          @builder.div(:id => 'summary') do
+            @builder.p('',:id => 'totals')
+            @builder.p('',:id => 'duration')
+            @builder.div(:id => 'expand-collapse') do
+              @builder.p('すべて開く', :id => 'expander')
+              @builder.p('すべて閉じる', :id => 'collapser')
+            end
+          end
+        end
+        before_menu
+      end
+      def before_menu
+        if ENV['PUBLISH']
+          @builder << "<div>"
+          @builder.div(:id => 'menu') do
+              @builder << make_menu_for_publish
+          end
+          @builder << "<div class='contents'>"
+        end
+      end
+      def after_menu
+        if ENV['PUBLISH']
+          @builder << '</div>'
+          @builder << '</div>'
+        end
+      end
+      def make_menu_for_publish
+        menu = 'tmp/menu.html'
+        system("erb -T - #{File.dirname(__FILE__)}/menu.html.erb > #{menu}")
+        File.readlines(menu).join
+      end
+      def after_features(features)
+        after_menu
+        print_stats(features)
+        @builder << '</div>'
+        @builder << '</body>'
+        @builder << '</html>'
+      end
+      def before_feature(feature)
+        @feature = feature
+        @exceptions = []
+        @builder << '<div class="feature">'
+      end
+      def after_feature(feature)
+        @builder << '</div>'
+      end
+      def before_comment(comment)
+        return if comment == '# language: ja'
+        @builder << '<pre class="comment">'
+      end
+      def after_comment(comment)
+        return if comment == '# language: ja'
+        @builder << '</pre>'
+      end
+      def comment_line(comment_line)
+        return if comment_line == '# language: ja'
+        @builder.text!(comment_line)
+        @builder.br
+      end
+      def after_tags(tags)
+        @tag_spacer = nil
+      end
+      def tag_name(tag_name)
+        @builder.text!(@tag_spacer) if @tag_spacer
+        @tag_spacer = ' '
+        @builder.span(tag_name, :class => 'tag')
+      end
+      def feature_name(keyword, name)
+        title = @feature.file.split('/').last.gsub(/\.feature/, '')
+        @builder.h2 do |h2|
+          @builder.span(title, :class => 'val')
+        end
+        lines = name.split(/\r?\n/)
+        unless lines.empty?
+          @builder.p(:class => 'narrative') do
+            lines.each do |line|
+              @builder.text!(line.strip)
+              @builder.br
+            end
+          end
+        end
+      end
+      def before_background(background)
+        @in_background = true
+        @builder << '<div class="background">'
+      end
+      def after_background(background)
+        @in_background = nil
+        @builder << '</div>'
+      end
+      def background_name(keyword, name, file_colon_line, source_indent)
+        @listing_background = true
+        @builder.h3(:id => "background_#{@scenario_number}") do |h3|
+          @builder.span(keyword, :class => 'keyword')
+          @builder.text!(' ')
+          @builder.span(name, :class => 'val')
+        end
+      end
+      def before_feature_element(feature_element)
+        @scenario_number+=1
+        @scenario_red = false
+        css_class = {
+          ::Cucumber::Ast::Scenario        => 'scenario',
+          ::Cucumber::Ast::ScenarioOutline => 'scenario outline'
+        }[feature_element.class]
+        @builder << "<div class='#{css_class}'>"
+      end
+      def after_feature_element(feature_element)
+        @builder << '</div>'
+        @open_step_list = true
+      end
+      def scenario_name(keyword, name, file_colon_line, source_indent)
+        @step_number_in_scenario = 0
+        @builder.span(:class => 'scenario_file') do
+          @builder << file_colon_line
+        end
+        @listing_background = false
+        @builder.h3(:id => "scenario_#{@scenario_number}") do
+          @builder.span(keyword + ':', :class => 'keyword')
+          @builder.text!(' ')
+          @builder.span(name, :class => 'val')
+        end
+      end
+      def before_outline_table(outline_table)
+        @outline_row = 0
+        @builder << '<table>'
+      end
+      def after_outline_table(outline_table)
+        @builder << '</table>'
+        @outline_row = nil
+      end
+      def before_examples(examples)
+         @builder << '<div class="examples">'
+      end
+      def after_examples(examples)
+        @builder << '</div>'
+      end
+      def examples_name(keyword, name)
+        @builder.h4 do
+          @builder.span(keyword, :class => 'keyword')
+          @builder.text!(' ')
+          @builder.span(name, :class => 'val')
+        end
+      end
+      def before_steps(steps)
+        @step_count_in_scenario = steps.count
+        @builder << '<ol>'
+      end
+      def after_steps(steps)
+        @builder << '</ol>'
+      end
+      def before_step(step)
+        @step_id = step.dom_id
+        @step_number += 1
+        @step = step
+      end
+      def after_step(step)
+        move_progress
+      end
+      def before_step_result(keyword, step_match, multiline_arg, status, exception, source_indent, background, file_colon_line)
+        @step_match = step_match
+        @hide_this_step = false
+        if exception
+          if @exceptions.include?(exception)
+            @hide_this_step = true
+            return
+          end
+          @exceptions << exception
+        end
+        if status != :failed && @in_background ^ background
+          @hide_this_step = true
+          return
+        end
+        @status = status
+        return if @hide_this_step
+        set_scenario_color(status)
+        @builder << "<li id='#{@step_id}' class='step #{status}'>"
+      end
+      def after_step_result(keyword, step_match, multiline_arg, status, exception, source_indent, background, file_colon_line)
+        return if @hide_this_step
+        # print snippet for undefined steps
+        if status == :undefined
+          keyword = @step.actual_keyword if @step.respond_to?(:actual_keyword)
+          step_multiline_class = @step.multiline_arg ? @step.multiline_arg.class : nil
+          @builder.pre do |pre|
+            pre << @runtime.snippet_text(keyword,step_match.instance_variable_get("@name") || '',step_multiline_class)
+          end
+        end
+        @builder << '</li>'
+        unless status == :undefined
+          step_file = step_match.file_colon_line
+          step_contents = "<div class=\"step_contents\"><pre>"
+          step_file.gsub(/^([^:]*\.rb):(\d*)/) do
+            line_index = $2.to_i - 1
+            File.readlines(File.expand_path($1.force_encoding('UTF-8')))[line_index..-1].each do |line|
+              step_contents << line
+              break if line.chop == 'end' or line.chop.start_with?('end ')
+            end
+          end
+          step_contents << "</pre></div>"
+          @builder << step_contents
+        end
+        print_messages
+      end
+      def step_name(keyword, step_match, status, source_indent, background, file_colon_line)
+        background_in_scenario = background && !@listing_background
+        @skip_step = background_in_scenario
+        unless @skip_step
+          build_step(keyword, step_match, status)
+        end
+      end
+      def exception(exception, status)
+        build_exception_detail(exception)
+      end
+      def extra_failure_content(file_colon_line)
+        @snippet_extractor ||= SnippetExtractor.new
+        "<pre class=\"ruby\"><code>#{@snippet_extractor.snippet(file_colon_line)}</code></pre>"
+      end
+      def before_multiline_arg(multiline_arg)
+        return if @hide_this_step || @skip_step
+        if ::Cucumber::Ast::Table === multiline_arg
+          @builder << '<table>'
+        end
+      end
+      def after_multiline_arg(multiline_arg)
+        return if @hide_this_step || @skip_step
+        if ::Cucumber::Ast::Table === multiline_arg
+          @builder << '</table>'
+        end
+      end
+      def doc_string(string)
+        return if @hide_this_step
+        @builder.pre(:class => 'val') do |pre|
+          @builder << h(string).gsub("\n", '&#x000A;')
+        end
+      end
+      def before_table_row(table_row)
+        @row_id = table_row.dom_id
+        @col_index = 0
+        return if @hide_this_step
+        @builder << "<tr class='step' id='#{@row_id}'>"
+      end
+      def after_table_row(table_row)
+        return if @hide_this_step
+        print_table_row_messages
+        @builder << '</tr>'
+        if table_row.exception
+          @builder.tr do
+            @builder.td(:colspan => @col_index.to_s, :class => 'failed') do
+              @builder.pre do |pre|
+                pre << h(format_exception(table_row.exception))
+              end
+            end
+          end
+          set_scenario_color_failed
+        end
+        if @outline_row
+          @outline_row += 1
+        end
+        @step_number += 1
+        move_progress
+      end
+      def table_cell_value(value, status)
+        return if @hide_this_step
+        @cell_type = @outline_row == 0 ? :th : :td
+        attributes = {:id => "#{@row_id}_#{@col_index}", :class => 'step'}
+        attributes[:class] += " #{status}" if status
+        build_cell(@cell_type, value, attributes)
+        set_scenario_color(status)
+        @col_index += 1
+      end
+      def puts(message)
+        @delayed_messages << message
+        #@builder.pre(message, :class => 'message')
+      end
+      def print_messages
+        return if @delayed_messages.empty?
+        #@builder.ol do
+          @delayed_messages.each_with_index do |ann, i|
+            @builder.li(:id => "#{@step_number}_#{i}", :class => 'step message') do
+              @builder << ann
+            end
+          end
+          @builder.script do |script|
+            script << "$(function() {"
+            script << "  $('##{@step_id}').css('cursor', 'pointer').click(function() {"
+            script << "    $(this).nextAll('li[id^=\"#{@step_number}_\"]').toggle(250);"
+            script << "  });"
+            script << "});"
+          end
+        #end
+        empty_messages
+      end
+      def print_table_row_messages
+      end
+      def empty_messages
+        @delayed_messages = []
+      end
+      protected
+      def build_exception_detail(exception)
+        backtrace = Array.new
+        @builder.div(:class => 'message') do
+          message = exception.message
+          if defined?(RAILS_ROOT) && message.include?('Exception caught')
+            matches = message.match(/Showing <i>(.+)<\/i>(?:.+) #(\d+)/)
+            backtrace += ["#{RAILS_ROOT}/#{matches[1]}:#{matches[2]}"] if matches
+            matches = message.match(/<code>([^(\/)]+)<\//m)
+            message = matches ? matches[1] : ""
+          end
+          unless exception.instance_of?(RuntimeError)
+            message = "#{message} (#{exception.class})"
+          end
+          @builder.pre do
+            @builder.text!(message)
+          end
+        end
+        @builder.div(:class => 'backtrace') do
+          @builder.pre do
+            backtrace = exception.backtrace
+            backtrace.delete_if { |x| x =~ /\/gems\/(cucumber|rspec)/ }
+            @builder << backtrace_line(backtrace.join("\n"))
+          end
+        end
+        extra = extra_failure_content(backtrace)
+        @builder << extra unless extra == ""
+      end
+      def set_scenario_color(status)
+        if status == :undefined or status == :pending
+          set_scenario_color_pending
+        end
+        if status == :failed
+          set_scenario_color_failed
+        end
+      end
+      def set_scenario_color_failed
+        @builder.script do
+          unless @header_red
+            @builder.text!("makeRed('cucumber-header');")
+            @builder.text!("makeMenuRed();")
+            @header_red = true
+          end
+          @builder.text!("makeRed('scenario_#{@scenario_number}');") unless @scenario_red
+          @scenario_red = true
+        end
+      end
+      def set_scenario_color_pending
+        @builder.script do
+          unless @header_red
+            @builder.text!("makeYellow('cucumber-header');")
+            @builder.text!("makeMenuYellow();")
+          end
+          @builder.text!("makeYellow('scenario_#{@scenario_number}');") unless @scenario_red
+        end
+      end
+      def get_step_count(features)
+        count = 0
+        features = features.instance_variable_get("@features")
+        features.each do |feature|
+          #get background steps
+          if feature.instance_variable_get("@background")
+            background = feature.instance_variable_get("@background")
+            background.init
+            background_steps = background.instance_variable_get("@steps").instance_variable_get("@steps")
+            count += background_steps.size
+          end
+          #get scenarios
+          feature.instance_variable_get("@feature_elements").each do |scenario|
+            scenario.init
+            #get steps
+            steps = scenario.instance_variable_get("@steps").instance_variable_get("@steps")
+            count += steps.size
+            #get example table
+            examples = scenario.instance_variable_get("@examples_array")
+            unless examples.nil?
+              examples.each do |example|
+                example_matrix = example.instance_variable_get("@outline_table").instance_variable_get("@cell_matrix")
+                count += example_matrix.size
+              end
+            end
+            #get multiline step tables
+            steps.each do |step|
+              multi_arg = step.instance_variable_get("@multiline_arg")
+              next if multi_arg.nil?
+              matrix = multi_arg.instance_variable_get("@cell_matrix")
+              count += matrix.size unless matrix.nil?
+            end
+          end
+        end
+        return count
+      end
+      def build_step(keyword, step_match, status)
+        @step_number_in_scenario += 1
+        formatted_step_number = sprintf("%0#{@step_count_in_scenario.to_s.size}d", @step_number_in_scenario)
+        step_name = step_match.format_args(lambda{|param| %{<span class="param">#{param}</span>}})
+        @builder.div(:class => 'step_name') do |div|
+          @builder.span("#{formatted_step_number}. ", :class => 'keyword')
+          @builder.span(:class => 'step val') do |name|
+            name << h(step_name).gsub(/&lt;span class=&quot;(.*?)&quot;&gt;/, '<span class="\1">').gsub(/&lt;\/span&gt;/, '</span>')
+          end
+        end
+        step_file = step_match.file_colon_line
+        step_file.gsub(/^([^:]*\.rb):(\d*)/) do
+          if index = $1.index('lib/daddy/cucumber/step_definitions/')
+            step_file = "daddy: " + $1[index..-1]
+          end
+          step_file = "<span id=\"step_file_#{@step_number}\">#{step_file}:#{$2}</span>"
+        end
+        @builder.div(:class => 'step_file') do |div|
+          @builder.span do
+            @builder << step_file
+            @builder.script do |script|
+              script << "$(function() {"
+              script << "  $('#step_file_#{@step_number}').css('cursor', 'pointer').click(function(event) {"
+              script << "    $(this).closest('li').next('.step_contents').toggle(250);"
+              script << "    event.stopImmediatePropagation();"
+              script << "  });"
+              script << "});"
+            end
+          end
+        end
+      end
+      def build_cell(cell_type, value, attributes)
+        @builder.__send__(cell_type, attributes) do
+          @builder.div do
+            @builder.span(value,:class => 'step param')
+          end
+        end
+      end
+      def inline_css
+        @builder.style do
+          @builder << File.read(File.dirname(__FILE__) + '/cucumber.css')
+          @builder << File.read(File.dirname(__FILE__) + '/daddy.css')
+        end
+      end
+      def inline_js
+        @builder.script do
+          @builder << inline_jquery
+          @builder << inline_js_content
+          if ENV['EXPAND']
+            @builder << %w{
+              $(document).ready(function() {
+                $(SCENARIOS).siblings().show();
+                $('li.message').show();
+                });
+            }.join
+          else
+            @builder << %w{
+              $(document).ready(function() {
+                $(SCENARIOS).siblings().show();
+                $('li.message').hide();
+                });
+            }.join
+          end
+        end
+      end
+      def inline_jquery
+        File.read(File.dirname(__FILE__) + '/jquery-min.js')
+      end
+      def inline_js_content
+        <<-EOF
+  SCENARIOS = "h3[id^='scenario_'],h3[id^=background_]";
+  $(document).ready(function() {
+    $(SCENARIOS).css('cursor', 'pointer');
+    $(SCENARIOS).click(function() {
+      $(this).siblings().toggle(250);
+    });
+    $("#collapser").css('cursor', 'pointer');
+    $("#collapser").click(function() {
+      $(SCENARIOS).siblings().hide();
+      $('li.message').hide();
+    });
+    $("#expander").css('cursor', 'pointer');
+    $("#expander").click(function() {
+      $(SCENARIOS).siblings().show();
+    });
+  })
+  function moveProgressBar(percentDone) {
+    $("cucumber-header").css('width', percentDone +"%");
+  }
+  function makeRed(element_id) {
+    $('#'+element_id).css('background', '#C40D0D');
+    $('#'+element_id).css('color', '#FFFFFF');
+  }
+  function makeMenuRed() {
+    $('#menu .sprint').css('border-color', '#C40D0D');
+  }
+  function makeYellow(element_id) {
+    $('#'+element_id).css('background', '#FAF834');
+    $('#'+element_id).css('color', '#000000');
+  }
+  function makeMenuYellow() {
+    $('#menu .sprint').css('border-color', '#FAF834');
+  }
+        EOF
+      end
+      def move_progress
+        @builder << " <script>moveProgressBar('#{percent_done}');</script>"
+      end
+      def percent_done
+        result = 100.0
+        if @step_count != 0
+          result = ((@step_number).to_f / @step_count.to_f * 1000).to_i / 10.0
+        end
+        result
+      end
+      def format_exception(exception)
+        (["#{exception.message}"] + exception.backtrace).join("\n")
+      end
+      def backtrace_line(line)
+        line.gsub(/\A([^:]*\.(?:rb|feature|haml)):(\d*).*\z/) do
+          if ENV['TM_PROJECT_DIRECTORY']
+            "<a href=\"txmt://open?url=file://#{File.expand_path($1)}&line=#{$2}\">#{$1}:#{$2}</a> "
+          else
+            line
+          end
+        end
+      end
+      def print_stats(features)
+        @builder <<  "<script>document.getElementById('duration').innerHTML = \"Finished in <strong>#{format_duration(features.duration)} seconds</strong>\";</script>"
+        @builder <<  "<script>document.getElementById('totals').innerHTML = \"#{print_stat_string(features)}\";</script>"
+      end
+      def print_stat_string(features)
+        string = String.new
+        string << dump_count(@runtime.scenarios.length, "scenario")
+        scenario_count = print_status_counts{|status| @runtime.scenarios(status)}
+        string << scenario_count if scenario_count
+        string << "<br />"
+        string << dump_count(@runtime.steps.length, "step")
+        step_count = print_status_counts{|status| @runtime.steps(status)}
+        string << step_count if step_count
+      end
+      def print_status_counts
+        counts = [:failed, :skipped, :undefined, :pending, :passed].map do |status|
+          elements = yield status
+          elements.any? ? "#{elements.length} #{status.to_s}" : nil
+        end.compact
+        return " (#{counts.join(', ')})" if counts.any?
+      end
+      def dump_count(count, what, state=nil)
+        [count, state, "#{what}#{count == 1 ? '' : 's'}"].compact.join(" ")
+      end
+      def create_builder(io)
+        ::Cucumber::Formatter::OrderedXmlMarkup.new(:target => io, :indent => 0)
+      end
+      class SnippetExtractor #:nodoc:
+        class NullConverter; def convert(code, pre); code; end; end #:nodoc:
+        begin; require 'syntax/convertors/html'; @@converter = Syntax::Convertors::HTML.for_syntax "ruby"; rescue LoadError => e; @@converter = NullConverter.new; end
+        def snippet(error)
+          raw_code, line = snippet_for(error[0])
+          highlighted = @@converter.convert(raw_code, false)
+          highlighted << "\n<span class=\"comment\"># gem install syntax to get syntax highlighting</span>" if @@converter.is_a?(NullConverter)
+          post_process(highlighted, line)
+        end
+        def snippet_for(error_line)
+          if error_line =~ /(.*):(\d+)/
+            file = $1
+            line = $2.to_i
+            [lines_around(file, line), line]
+          else
+            ["# Couldn't get snippet for #{error_line}", 1]
+          end
+        end
+        def lines_around(file, line)
+          if File.file?(file)
+            lines = File.open(file).read.split("\n")
+            min = [0, line-3].max
+            max = [line+1, lines.length-1].min
+            selected_lines = []
+            selected_lines.join("\n")
+            lines[min..max].join("\n")
+          else
+            "# Couldn't get snippet for #{file}"
+          end
+        end
+        def post_process(highlighted, offending_line)
+          new_lines = []
+          highlighted.split("\n").each_with_index do |line, i|
+            new_line = "<span class=\"linenum\">#{offending_line+i-2}</span>#{line}"
+            new_line = "<span class=\"offending\">#{new_line}</span>" if i == 2
+            new_lines << new_line
+          end
+          new_lines.join("\n")
+        end
+      end
+    end
+  end
+end