class Brakeman::Collection

Attributes

collection[R]
files[R]
includes[R]
name[R]
options[R]
parent[R]
src[R]
tracker[R]

Public Class Methods

new(name, parent, file_name, src, tracker) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 10
def initialize name, parent, file_name, src, tracker
  @name = name
  @parent = parent
  @files = []
  @src = {}
  @includes = []
  @methods = { :public => {}, :private => {}, :protected => {} }
  @class_methods = {}
  @simple_methods = { :class => {}, instance: {} }
  @options = {}
  @tracker = tracker

  add_file file_name, src
end

Public Instance Methods

add_file(file_name, src) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 37
def add_file file_name, src
  @files << file_name unless @files.include? file_name
  @src[file_name] = src
end
add_include(class_name) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 42
def add_include class_name
  @includes << class_name unless ancestor?(class_name)
end
add_method(visibility, name, src, file_name) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 51
def add_method visibility, name, src, file_name
  meth_info = Brakeman::MethodInfo.new(name, src, self, file_name)
  add_simple_method_maybe meth_info

  if src.node_type == :defs
    @class_methods[name] = meth_info

    # TODO fix this weirdness
    name = :"#{src[1]}.#{name}"
  end

  @methods[visibility][name] = meth_info
end
add_option(name, exp) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 46
def add_option name, exp
  @options[name] ||= []
  @options[name] << exp
end
ancestor?(parent, seen={}) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 25
def ancestor? parent, seen={}
  seen[self.name] = true

  if self.parent == parent or self.name == parent or seen[self.parent]
    true
  elsif parent_model = collection[self.parent]
    parent_model.ancestor? parent, seen
  else
    false
  end
end
each_method() { |name, info| ... } click to toggle source
# File lib/brakeman/tracker/collection.rb, line 65
def each_method
  @methods.each do |_vis, meths|
    meths.each do |name, info|
      yield name, info
    end
  end
end
file() click to toggle source
# File lib/brakeman/tracker/collection.rb, line 98
def file
  @files.first
end
get_class_method(name) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 94
def get_class_method name
  @class_methods[name]
end
get_instance_method(name) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 84
def get_instance_method name
  @methods.each do |_vis, meths|
    if meths[name]
      return meths[name]
    end
  end

  nil
end
get_method(name, type = :instance) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 73
def get_method name, type = :instance
  case type
  when :class
    get_class_method name
  when :instance
    get_instance_method name
  else
    raise "Unexpected method type: #{type.inspect}"
  end
end
get_simple_method_return_value(type, name) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 118
def get_simple_method_return_value type, name
  @simple_methods[type][name]
end
methods_public() click to toggle source
# File lib/brakeman/tracker/collection.rb, line 114
def methods_public
  @methods[:public]
end
top_line() click to toggle source
# File lib/brakeman/tracker/collection.rb, line 102
def top_line
  if sexp? @src[file]
    @src[file].line
  else
    @src.each_value do |source|
      if sexp? source
        return source.line
      end
    end
  end
end

Private Instance Methods

add_simple_method(meth_info) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 130
def add_simple_method meth_info
  name = meth_info.name
  value = meth_info.return_value

  case meth_info.src.node_type
  when :defn
    @simple_methods[:instance][name] = value
  when :defs
    @simple_methods[:class][name] = value
  else
    raise "Expected sexp type: #{src.node_type}"
  end
end
add_simple_method_maybe(meth_info) click to toggle source
# File lib/brakeman/tracker/collection.rb, line 124
def add_simple_method_maybe meth_info
  if meth_info.very_simple_method?
    add_simple_method meth_info
  end
end