Contents

require 'brakeman/processors/base_processor'
require 'brakeman/tracker/template'

#Base Processor for templates/views
class Brakeman::TemplateProcessor < Brakeman::BaseProcessor

  #Initializes template information.
  def initialize tracker, template_name, called_from = nil, file_name = nil
    super(tracker) 
    @current_template = Brakeman::Template.new template_name, called_from, file_name, tracker
    @file_name = file_name

    if called_from
      template_name = (template_name.to_s + "." + called_from.to_s).to_sym
    end

    tracker.templates[template_name] = @current_template

    @inside_concat = false
  end

  #Process the template Sexp.
  def process exp
    begin
      super
    rescue => e
      except = e.exception("Error when processing #{@current_template.name}: #{e.message}")
      except.set_backtrace(e.backtrace)
      raise except
    end
  end

  #Ignore initial variable assignment
  def process_lasgn exp
    if exp.lhs == :_erbout and exp.rhs.node_type == :str  #ignore
      ignore
    elsif exp.lhs == :_buf and exp.rhs.node_type == :str
      ignore
    else
      exp.rhs = process exp.rhs
      exp
    end
  end

  #Adds output to the list of outputs.
  def process_output exp
    exp.value = process exp.value
    @current_template.add_output exp unless exp.original_line
    exp
  end

  def process_escaped_output exp
    process_output exp
  end

  # Pull out actual output value from template
  def normalize_output arg
    if call? arg and [:to_s, :html_safe!, :freeze].include? arg.method
      arg.target
    elsif node_type? arg, :if
      branches = [arg.then_clause, arg.else_clause].compact

      if branches.empty?
        s(:nil)
      elsif branches.length == 2
        Sexp.new(:or, *branches)
      else
        branches.first
      end
    else
      arg
    end
  end
end

Version data entries

6 entries across 6 versions & 3 rubygems

Version	Path
brakeman-3.3.2	lib/brakeman/processors/template_processor.rb
brakeman-lib-3.3.2	lib/brakeman/processors/template_processor.rb
brakeman-min-3.3.2	lib/brakeman/processors/template_processor.rb
brakeman-3.3.1	lib/brakeman/processors/template_processor.rb
brakeman-min-3.3.1	lib/brakeman/processors/template_processor.rb
brakeman-lib-3.3.1	lib/brakeman/processors/template_processor.rb