Class: RBI::Rewriters::SortNodes

Inherits:
Visitor
  • Object
show all
Extended by:
T::Sig
Defined in:
lib/rbi/rewriters/sort_nodes.rb

Instance Method Summary collapse

Methods inherited from Visitor

#visit_all

Instance Method Details

#visit(node) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/rbi/rewriters/sort_nodes.rb', line 10

def visit(node)
  return unless node.is_a?(Tree)
  visit_all(node.nodes)
  node.nodes.sort! do |a, b|
    return 0 if a.is_a?(Mixin) || b.is_a?(Mixin)

    res = node_rank(a) <=> node_rank(b)
    res = node_name(a) <=> node_name(b) if res == 0
    res || 0
  end
end