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
|
RSpec::Support.require_rspec_support 'encoded_string'
RSpec::Support.require_rspec_support 'ruby_features'
module RSpec
module Support
# @private
# Represents a Ruby source file and provides access to AST and tokens.
class Source
attr_reader :source, :path
def self.from_file(path)
source = File.read(path)
new(source, path)
end
if String.method_defined?(:encoding)
def initialize(source_string, path=nil)
@source = RSpec::Support::EncodedString.new(source_string, Encoding.default_external)
@path = path ? File.expand_path(path) : '(string)'
end
else # for 1.8.7
# :nocov:
def initialize(source_string, path=nil)
@source = RSpec::Support::EncodedString.new(source_string)
@path = path ? File.expand_path(path) : '(string)'
end
# :nocov:
end
def lines
@lines ||= source.split("\n")
end
def inspect
"#<#{self.class} #{path}>"
end
if RSpec::Support::RubyFeatures.ripper_supported?
RSpec::Support.require_rspec_support 'source/node'
RSpec::Support.require_rspec_support 'source/token'
def ast
@ast ||= begin
require 'ripper'
sexp = Ripper.sexp(source)
raise SyntaxError unless sexp
Node.new(sexp)
end
end
def tokens
@tokens ||= begin
require 'ripper'
tokens = Ripper.lex(source)
Token.tokens_from_ripper_tokens(tokens)
end
end
def nodes_by_line_number
@nodes_by_line_number ||= begin
nodes_by_line_number = ast.select(&:location).group_by { |node| node.location.line }
Hash.new { |hash, key| hash[key] = [] }.merge(nodes_by_line_number)
end
end
def tokens_by_line_number
@tokens_by_line_number ||= begin
nodes_by_line_number = tokens.group_by { |token| token.location.line }
Hash.new { |hash, key| hash[key] = [] }.merge(nodes_by_line_number)
end
end
end
end
end
end
|