Sha256: 31eaac172a6045b4982a06a899d624ffaec3b56e102933210e802b12fc9368f6
Contents?: true
Size: 1.35 KB
Versions: 16
Compression:
Stored size: 1.35 KB
Contents
module ActiveMocker
class RubyParse
attr_reader :source
def initialize(source)
@source = source
end
def is_class?
ast.type == :class
end
def class_name
Unparser.unparse(find_class.to_a[0])
end
def parent_class_name
Unparser.unparse(find_class.to_a[1])
end
def has_parent_class?
return false if find_class.nil?
find_class.to_a[1].try(:type) == :const
end
def change_class_name(class_name)
reset_nodes
nodes[0] = Parser::CurrentRuby.parse(class_name)
new_ast = ast.updated(nil, nodes, nil)
Unparser.unparse(new_ast)
end
def modify_parent_class(parent_class)
reset_nodes
if has_parent_class?
class_node = find_class.to_a.dup
class_node[1] = Parser::CurrentRuby.parse(parent_class)
new_ast = find_class.updated(nil, class_node, nil)
else
nodes[1] = nodes[0].updated(:const, [nil, parent_class.to_sym])
new_ast = ast.updated(nil, nodes, nil)
end
Unparser.unparse(new_ast)
end
def find_class
return ast if ast.try(:type) == :class
ast.to_a.select {|n| n.try(:type) == :class}.first
end
def ast
@ast ||= Parser::CurrentRuby.parse(source)
end
def nodes
@nodes ||= ast.to_a.dup
end
def reset_nodes
@nodes = nil
end
end
end
Version data entries
16 entries across 16 versions & 1 rubygems