module RBS::Prototype::Helpers
Private Instance Methods
any_node?(node, nodes: []) { |node| ... }
click to toggle source
# File rbs-3.4.0/lib/rbs/prototype/helpers.rb, line 97 def any_node?(node, nodes: [], &block) if yield(node) nodes << node end each_child node do |child| any_node? child, nodes: nodes, &block end nodes.empty? ? nil : nodes end
args_from_node(args_node)
click to toggle source
NOTE: args_node may be a nil by a bug
https://bugs.ruby-lang.org/issues/17495
# File rbs-3.4.0/lib/rbs/prototype/helpers.rb, line 121 def args_from_node(args_node) args_node&.children || [0, nil, nil, nil, 0, nil, nil, nil, nil, nil] end
block_from_body(node)
click to toggle source
# File rbs-3.4.0/lib/rbs/prototype/helpers.rb, line 8 def block_from_body(node) _, args_node, body_node = node.children _pre_num, _pre_init, _opt, _first_post, _post_num, _post_init, _rest, _kw, _kwrest, block_var = args_from_node(args_node) # @type var body_node: node? if body_node yields = any_node?(body_node) {|n| n.type == :YIELD } end if yields || block_var required = true if body_node if any_node?(body_node) {|n| n.type == :FCALL && n.children[0] == :block_given? && !n.children[1] } required = false end end if _rest == :* && block_var == :& # ... is given required = false end if block_var if body_node usage = NodeUsage.new(body_node) if usage.each_conditional_node.any? {|n| n.type == :LVAR && n.children[0] == block_var } required = false end end end method_block = Types::Block.new( required: required, type: Types::Function.empty(untyped), self_type: nil ) if yields yields.each do |yield_node| array_content = yield_node.children[0]&.children&.compact || [] # @type var keywords: node? positionals, keywords = if keyword_hash?(array_content.last) [array_content.take(array_content.size - 1), array_content.last] else [array_content, nil] end if (diff = positionals.size - method_block.type.required_positionals.size) > 0 diff.times do method_block.type.required_positionals << Types::Function::Param.new( type: untyped, name: nil ) end end if keywords keywords.children[0].children.each_slice(2) do |key_node, value_node| if key_node key = key_node.children[0] method_block.type.required_keywords[key] ||= Types::Function::Param.new( type: untyped, name: nil ) end end end end end end method_block end
each_child(node, &block)
click to toggle source
# File rbs-3.4.0/lib/rbs/prototype/helpers.rb, line 85 def each_child(node, &block) each_node node.children, &block end
each_node(nodes) { |child| ... }
click to toggle source
# File rbs-3.4.0/lib/rbs/prototype/helpers.rb, line 89 def each_node(nodes) nodes.each do |child| if child.is_a?(RubyVM::AbstractSyntaxTree::Node) yield child end end end
keyword_hash?(node)
click to toggle source
# File rbs-3.4.0/lib/rbs/prototype/helpers.rb, line 109 def keyword_hash?(node) if node && node.type == :HASH node.children[0].children.compact.each_slice(2).all? {|key, _| key.type == :LIT && key.children[0].is_a?(Symbol) } else false end end
untyped()
click to toggle source
# File rbs-3.4.0/lib/rbs/prototype/helpers.rb, line 125 def untyped @untyped ||= Types::Bases::Any.new(location: nil) end