Sha256: 83b7f96ba82a43753c1fdfecc265f6826098c6f312804594690e107f711fa554

Contents?: true

Size: 1.63 KB

Versions: 14

Compression:

Stored size: 1.63 KB

Contents

#Contains a couple shared methods for Processors.
module Brakeman::ProcessorHelper
  def process_all exp
    exp.each_sexp do |e|
      process e
    end
    exp
  end

  def process_all! exp
    exp.map! do |e|
      if sexp? e
        process e
      else
        e
      end
    end

    exp
  end

  #Process the arguments of a method call. Does not store results.
  #
  #This method is used because Sexp#args and Sexp#arglist create new objects.
  def process_call_args exp
    exp.each_arg do |a|
      process a if sexp? a
    end

    exp
  end

  def process_class exp
    current_class = @current_class
    @current_class = class_name exp[1]
    process_all exp.body
    @current_class = current_class
    exp
  end

  #Sets the current module.
  def process_module exp
    module_name = class_name(exp.class_name).to_s
    prev_module = @current_module

    if prev_module
      @current_module = "#{prev_module}::#{module_name}"
    else
      @current_module = module_name
    end

    if block_given?
      yield
    else
      process_all exp.body
    end

    @current_module = prev_module

    exp
  end

  #Returns a class name as a Symbol.
  #If class name cannot be determined, returns _exp_.
  def class_name exp
    case exp
    when Sexp
      case exp.node_type
      when :const
        exp.value
      when :lvar
        exp.value.to_sym
      when :colon2
        "#{class_name(exp.lhs)}::#{exp.rhs}".to_sym
      when :colon3
        "::#{exp.value}".to_sym
      when :self
        @current_class || @current_module || nil
      else
        exp
      end
    when Symbol
      exp
    when nil
      nil
    else
      exp
    end
  end
end

Version data entries

14 entries across 14 versions & 2 rubygems

Version Path
brakeman-3.1.0 lib/brakeman/processors/lib/processor_helper.rb
brakeman-min-3.1.0 lib/brakeman/processors/lib/processor_helper.rb
brakeman-3.0.5 lib/brakeman/processors/lib/processor_helper.rb
brakeman-min-3.0.5 lib/brakeman/processors/lib/processor_helper.rb
brakeman-3.0.4 lib/brakeman/processors/lib/processor_helper.rb
brakeman-min-3.0.4 lib/brakeman/processors/lib/processor_helper.rb
brakeman-min-3.0.3 lib/brakeman/processors/lib/processor_helper.rb
brakeman-3.0.3 lib/brakeman/processors/lib/processor_helper.rb
brakeman-min-3.0.2 lib/brakeman/processors/lib/processor_helper.rb
brakeman-3.0.2 lib/brakeman/processors/lib/processor_helper.rb
brakeman-min-3.0.1 lib/brakeman/processors/lib/processor_helper.rb
brakeman-3.0.1 lib/brakeman/processors/lib/processor_helper.rb
brakeman-3.0.0 lib/brakeman/processors/lib/processor_helper.rb
brakeman-min-3.0.0 lib/brakeman/processors/lib/processor_helper.rb