require 'fileutils' require 'coderay' MetricFu.metrics_require { 'base_template' } MetricFu.lib_require { 'utility' } class AwesomeTemplate < MetricFu::Template def write # Getting rid of the crap before and after the project name from integrity # @name = File.basename(MetricFu.run_dir).gsub(/^\w+-|-\w+$/, "") @name = Pathname.new(MetricFu.run_dir).basename # Copy Bluff javascripts to output directory Dir[File.join(template_directory, '..', 'javascripts', '*')].each do |f| FileUtils.cp(f, File.join(self.output_directory, File.basename(f))) end @metrics = {} result.each_pair do |section, contents| if template_exists?(section) create_instance_var(section, contents) @metrics[section] = contents create_instance_var(:per_file_data, per_file_data) mf_debug "Generating html for section #{section} with #{template(section)} for result #{result.class}" @html = erbify(section) html = erbify('layout') fn = output_filename(section) formatter.write_template(html, fn) else mf_debug "no template for section #{section} with #{template(section)} for result #{result.class}" end end # Instance variables we need should already be created from above if template_exists?('index') @html = erbify('index') html = erbify('layout') fn = output_filename('index') formatter.write_template(html, fn) else mf_debug "no template for section index for result #{result.class}" end write_file_data end def convert_ruby_to_html(ruby_text, line_number) tokens = CodeRay.scan(MetricFu::Utility.clean_ascii_text(ruby_text), :ruby) options = { :css => :class, :style => :alpha } if line_number.to_i > 0 options = options.merge({:line_numbers => :inline, :line_number_start => line_number.to_i }) end tokens.div(options) # CodeRay options # used to analyze source code, because object Tokens is a list of tokens with specified types. # :tab_width – tabulation width in spaces. Default: 8 # :css – how to include the styles (:class и :style). Default: :class) # # :wrap – wrap result in html tag :page, :div, :span or not to wrap (nil) # # :line_numbers – how render line numbers (:table, :inline, :list or nil) # # :line_number_start – first line number # # :bold_every – make every n-th line number bold. Default: 10 end def write_file_data per_file_data.each_pair do |file, lines| next if file.to_s.empty? next unless File.file?(file) data = File.readlines(file) fn = "#{file.gsub(%r{/}, '_')}.html" out = <<-HTML HTML out << "" data.each_with_index do |line, idx| line_number = (idx + 1).to_s out << "" out << "" if MetricFu::Formatter::Templates.option('syntax_highlighting') line_for_display = convert_ruby_to_html(line, line_number) else line_for_display = "#{line_number}#{line}" end out << "" out << "" end out << "

" if lines.has_key?(line_number) out << "

#{problem[:description]} » #{problem[:type]}

" else out << " " end out << "

#{line_for_display}

" formatter.write_template(out, fn) end end def template_directory File.dirname(__FILE__) end end