presidentbeef/brakeman

View on GitHub
lib/brakeman/processors/lib/processor_helper.rb

Summary

Maintainability
A
0 mins
Test Coverage
A
93%
#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

  # e.g. private defn
  def process_call_defn? exp
    if call? exp and exp.target.nil? and node_type? exp.first_arg, :defn, :defs and [:private, :public, :protected].include? exp.method
      prev_visibility = @visibility
      @visibility = exp.method
      process exp.first_arg
      @visibility = prev_visibility
      exp
    else
      false
    end
  end

  def current_file
    case
    when @current_file
      @current_file
    when @current_class.is_a?(Brakeman::Collection)
      @current_class.file
    when @current_module.is_a?(Brakeman::Collection)
      @current_module.file
    else
      nil
    end
  end
end