82 lines
1.9 KiB
Ruby
82 lines
1.9 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
|
|
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 = Hash.new([])
|
|
@class_uses = Hash.new([])
|
|
@types = Hash.new(0)
|
|
@not_found = []
|
|
@walker = Walker.new(self)
|
|
@files = []
|
|
@current = nil
|
|
end
|
|
attr_reader :class_defs , :class_uses , :types , :current
|
|
|
|
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
|
|
puts "Class defs #{@class_defs.length}"
|
|
puts "Class uses #{@class_uses.length}"
|
|
#puts "Types #{@types.to_yaml}"
|
|
#puts "files #{@files.to_yaml}"
|
|
#puts "Not found #{@not_found.length} #{@not_found.join(' ')}"
|
|
end
|
|
end
|
|
|
|
Collector.new.run
|