/
collection.rb
144 lines (118 loc) · 3.06 KB
/
collection.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
require 'brakeman/util'
require 'brakeman/tracker/method_info'
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
@files = []
@src = {}
@includes = []
@methods = { :public => {}, :private => {}, :protected => {} }
@class_methods = {}
@simple_methods = { :class => {}, instance: {} }
@options = {}
@tracker = tracker
add_file file_name, src
end
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
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 unless ancestor?(class_name)
end
def add_option name, exp
@options[name] ||= []
@options[name] << exp
end
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
def each_method
@methods.each do |_vis, meths|
meths.each do |name, info|
yield name, info
end
end
end
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
def get_instance_method name
@methods.each do |_vis, meths|
if meths[name]
return meths[name]
end
end
nil
end
def get_class_method name
@class_methods[name]
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
def get_simple_method_return_value type, name
@simple_methods[type][name]
end
private
def add_simple_method_maybe meth_info
if meth_info.very_simple_method?
add_simple_method meth_info
end
end
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
end
end