Sha256: d1558d70a26809d0e1a5b5c39f54dea3ed6aec5a8a97c25f64aa63cb9d650c63
Contents?: true
Size: 1.53 KB
Versions: 14
Compression:
Stored size: 1.53 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
#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.
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 :call
process exp
when :self
@current_class || @current_module || nil
else
raise "Error: Cannot get class name from #{exp}"
end
when Symbol
exp
when nil
nil
else
raise "Error: Cannot get class name from #{exp}"
end
end
end
Version data entries
14 entries across 14 versions & 2 rubygems