Sha256: 0b9ebc62b4d6ce2218930de9b7c5aaf6204377e030e6f89a46e9b72dd6109537
Contents?: true
Size: 1.48 KB
Versions: 8
Compression:
Stored size: 1.48 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
process_all exp.body
@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
8 entries across 8 versions & 1 rubygems