Top Level Namespace

Defined Under Namespace

Classes: Symbol

Instance Method Summary collapse

Instance Method Details

#import(array, from: "") ⇒ Object 



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
 
# File 'lib/import_rb.rb', line 11

def import(array, from: "")
  raise unless array.is_a? Array

  path = File.expand_path(from)
  const_map = {}
  array.each do |const|
    if const.is_a? Array
      const_map[const[0]] = const[1]
    elsif const.is_a? Symbol
      const_map[const] = const
    end
  end

  program_node = Prism.parse_file(path)
  statements = program_node.value.child_nodes[0]

  statements.body.each do |node|
    next unless (node.is_a?(Prism::ClassNode) || node.is_a?(Prism::ModuleNode))
    next unless const_map.key?(node.name)

    start_offset = node.start_offset
    end_offset = node.end_offset

    node_source = statements.source_lines.join()[start_offset..end_offset]
    node_source = node_source.sub(node.name.to_s, const_map[node.name].to_s)

    if self.is_a?(Module) || self.is_a?(Class)
      self.module_eval(node_source)
    else
      eval(node_source)
    end
  end
end