Sha256: 997249f61254a41e5b7ffb4419603dcefc17f59012862c2a3757770ee0cc0bad
Contents?: true
Size: 1.53 KB
Versions: 34
Compression:
Stored size: 1.53 KB
Contents
require 'brakeman/processors/base_processor'
require 'brakeman/processors/alias_processor'
require 'brakeman/processors/lib/module_helper'
require 'brakeman/tracker/library'
#Process generic library and stores it in Tracker.libs
class Brakeman::LibraryProcessor < Brakeman::BaseProcessor
include Brakeman::ModuleHelper
def initialize tracker
super
@file_name = nil
@alias_processor = Brakeman::AliasProcessor.new tracker
@current_module = nil
@current_class = nil
end
def process_library src, file_name = nil
@file_name = file_name
process src
end
def process_class exp
handle_class exp, @tracker.libs, Brakeman::Library
end
def process_module exp
handle_module exp, Brakeman::Library
end
def process_defn exp
exp = @alias_processor.process exp
if @current_class
exp.body = process_all! exp.body
@current_class.add_method :public, exp.method_name, exp, @file_name
elsif @current_module
exp.body = process_all! exp.body
@current_module.add_method :public, exp.method_name, exp, @file_name
end
exp
end
alias process_defs process_defn
def process_call exp
if process_call_defn? exp
exp
else
process_default exp
end
end
def process_iter exp
res = process_default exp
if node_type? res, :iter and call? exp.block_call # sometimes this changes after processing
if exp.block_call.method == :included
(@current_module || @current_class).options[:included] = res.block
end
end
res
end
end
Version data entries
34 entries across 34 versions & 3 rubygems