Sha256: 1d9ce6abf40b058f8b92d4482f453a29d9ef42b11355aae0c2460f946a4cf0cb
Contents?: true
Size: 1.85 KB
Versions: 31
Compression:
Stored size: 1.85 KB
Contents
require 'brakeman/util'
module Brakeman
class Collection
include Brakeman::Util
attr_reader :collection, :files, :includes, :name, :options, :parent, :src, :tracker
def initialize name, parent, file_name, src, tracker
@name = name
@parent = parent
@file_name = file_name
@files = [ file_name ]
@src = { file_name => src }
@includes = []
@methods = { :public => {}, :private => {}, :protected => {} }
@options = {}
@tracker = tracker
end
def ancestor? parent, seen={}
seen[self.name] = true
if self.parent == parent or seen[self.parent]
true
elsif parent_model = collection[self.parent]
parent_model.ancestor? parent, seen
else
false
end
end
def add_file file_name, src
@files << file_name unless @files.include? file_name
@src[file_name] = src
end
def add_include class_name
@includes << class_name
end
def add_option name, exp
@options[name] ||= []
@options[name] << exp
end
def add_method visibility, name, src, file_name
if src.node_type == :defs
name = :"#{src[1]}.#{name}"
end
@methods[visibility][name] = { :src => src, :file => file_name }
end
def each_method
@methods.each do |_vis, meths|
meths.each do |name, info|
yield name, info
end
end
end
def get_method name
each_method do |n, info|
if n == name
return info
end
end
nil
end
def file
@files.first
end
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
def methods_public
@methods[:public]
end
end
end
Version data entries
31 entries across 31 versions & 3 rubygems