Module: RBS::Prototype::Helpers
Relationships & Source Files | |
Extension / Inclusion / Inheritance Descendants | |
Included In:
| |
Defined in: | lib/rbs/prototype/helpers.rb |
Instance Method Summary
- #any_node?(node, nodes: [], &block) ⇒ Boolean private
-
#args_from_node(args_node)
private
NOTE: args_node may be a nil by a bug bugs.ruby-lang.org/issues/17495.
- #block_from_body(node) private
- #each_child(node, &block) private
- #each_node(nodes) private
- #keyword_hash?(node) ⇒ Boolean private
- #symbol_literal_node?(node) ⇒ Boolean private
- #untyped private
Instance Method Details
#any_node?(node, nodes: [], &block) ⇒ Boolean
(private)
# File 'lib/rbs/prototype/helpers.rb', line 96
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) (private)
NOTE: args_node may be a nil by a bug bugs.ruby-lang.org/issues/17495
# File 'lib/rbs/prototype/helpers.rb', line 120
def args_from_node(args_node) args_node&.children || [0, nil, nil, nil, 0, nil, nil, nil, nil, nil] end
#block_from_body(node) (private)
[ GitHub ]# File '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 if yields function = Types::Function.empty(untyped) 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 - function.required_positionals.size) > 0 diff.times do function.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] function.required_keywords[key] ||= Types::Function::Param.new( type: untyped, name: nil ) end end end end else function = Types::UntypedFunction.new(return_type: untyped) end Types::Block.new(required: required, type: function, self_type: nil) end end
#each_child(node, &block) (private)
[ GitHub ]# File 'lib/rbs/prototype/helpers.rb', line 84
def each_child(node, &block) each_node node.children, &block end
#each_node(nodes) (private)
[ GitHub ]# File 'lib/rbs/prototype/helpers.rb', line 88
def each_node(nodes) nodes.each do |child| if child.is_a?(RubyVM::AbstractSyntaxTree::Node) yield child end end end
#keyword_hash?(node) ⇒ Boolean
(private)
# File 'lib/rbs/prototype/helpers.rb', line 108
def keyword_hash?(node) if node && node.type == :HASH node.children[0].children.compact.each_slice(2).all? {|key, _| symbol_literal_node?(key) } else false end end
#symbol_literal_node?(node) ⇒ Boolean
(private)
# File 'lib/rbs/prototype/helpers.rb', line 124
def symbol_literal_node?(node) case node.type when :LIT if node.children[0].is_a?(Symbol) node.children[0] end when :SYM node.children[0] end end