95 lines
2.2 KiB
Ruby
95 lines
2.2 KiB
Ruby
require_relative '../helper'
|
|
require "register/interpreter"
|
|
require "parser/ruby22"
|
|
require "yaml"
|
|
|
|
class Walker < AST::Processor
|
|
def initialize collector
|
|
@collector = collector
|
|
end
|
|
|
|
def on_send node
|
|
_ , method , file_node = *node
|
|
if method == :require
|
|
@collector.load file_node.children[0] + ".rb"
|
|
end
|
|
if method == :require_relative
|
|
@collector.load File.dirname(@collector.current) + "/" + file_node.children[0] + ".rb"
|
|
end
|
|
if method.to_s.include?("eval")
|
|
@collector.evals << method
|
|
end
|
|
handler_missing(node)
|
|
end
|
|
|
|
def on_class node
|
|
@collector.class_defs << node.children[0].children[1]
|
|
handler_missing(node)
|
|
end
|
|
def on_const node
|
|
@collector.const_uses[node.children[1]] += 1
|
|
handler_missing(node)
|
|
end
|
|
def handler_missing node
|
|
type = node.type
|
|
@collector.types[type] += 1
|
|
node.children.each do |kid|
|
|
process(kid) if kid.is_a? AST::Node
|
|
end
|
|
end
|
|
end
|
|
|
|
class Collector
|
|
|
|
def initialize
|
|
@parser = Parser::Ruby22
|
|
@paths = Bundler.load.specs.collect { |s| s.gem_dir + "/lib/" }
|
|
@class_defs = []
|
|
@const_uses = Hash.new(0)
|
|
@types = Hash.new(0)
|
|
@not_found = []
|
|
@walker = Walker.new(self)
|
|
@files = []
|
|
@evals = []
|
|
@current = nil
|
|
end
|
|
attr_reader :class_defs , :const_uses , :types , :current , :evals
|
|
|
|
def file_content file_name
|
|
return nil if @files.include? file_name
|
|
@paths.each do |name|
|
|
next unless File.exist? name + file_name
|
|
@files << file_name
|
|
return File.open(name + file_name).read
|
|
end
|
|
nil
|
|
end
|
|
|
|
def run
|
|
load "salama.rb"
|
|
load "parser/ruby22.rb"
|
|
load "../../../.rbenv/versions/2.2.3/lib/ruby/2.2.0/racc/parser.rb"
|
|
print
|
|
end
|
|
|
|
def load file
|
|
str = file_content(file)
|
|
return @not_found.push(file) unless str
|
|
ast = @parser.parse str
|
|
was = @current
|
|
@current = file
|
|
@walker.process ast
|
|
@current = was
|
|
end
|
|
def print
|
|
@class_defs.uniq!
|
|
@files.uniq!
|
|
puts "Class defs #{@class_defs.length}"
|
|
puts "Types #{@types.to_yaml}"
|
|
puts "evals=#{@evals.length} #{@evals.uniq}"
|
|
#puts "Not found #{@not_found.length} #{@not_found}"
|
|
end
|
|
end
|
|
|
|
Collector.new.run
|