Module: YARDSorbet::SigToYARD

Extended by:
T::Sig
Defined in:
lib/yard-sorbet/sig_to_yard.rb

Overview

Translate sig type syntax to YARD type syntax.

Constant Summary collapse

REF_TYPES =

Map of common types to YARD conventions (in order to reduce allocations)

T.let({
  'T::Boolean' => ['Boolean'].freeze, # YARD convention for booleans
  # YARD convention is to use singleton objects when applicable:
  # https://www.rubydoc.info/gems/yard/file/docs/Tags.md#literals
  'FalseClass' => ['false'].freeze,
  'NilClass' => ['nil'].freeze,
  'TrueClass' => ['true'].freeze
}.freeze, T::Hash[String, [String]])

Class Method Summary collapse

Class Method Details

.build_generic_type(node) ⇒ String (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (String) 


58
59
60
61
62
63
64
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 58

def build_generic_type(node)
  return node.source if node.empty? || node.type != :aref

  collection_type = node.first.source
  member_type = node.last.children.map { build_generic_type(_1) }.join(', ')
  "#{collection_type}[#{member_type}]"
end

.convert(node) ⇒ Array<String>

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (Array<String>)

See Also:



23
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 23

def convert(node) = convert_node(node).map { _1.gsub(/\n\s*/, ' ') }

.convert_aref(node) ⇒ Array<String> (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (Array<String>) 


67
68
69
70
71
72
73
74
75
76
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 67

def convert_aref(node)
  # https://www.rubydoc.info/gems/yard/file/docs/Tags.md#parametrized-types
  case node.first.source
  when 'T::Array', 'T::Enumerable', 'T::Range', 'T::Set' then convert_collection(node)
  when 'T::Hash' then convert_hash(node)
  else
    log.info("Unsupported sig aref node #{node.source}")
    [build_generic_type(node)]
  end
end

.convert_array(node) ⇒ Array<(String)> (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (Array<(String)>)

See Also:



80
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 80

def convert_array(node) = ["Array<(#{sequence(node.first)})>"]

.convert_collection(node) ⇒ Array<(String)> (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (Array<(String)>) 


90
91
92
93
94
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 90

def convert_collection(node)
  collection_type = node.first.source.split('::').last
  member_type = convert_node(node[-1][0]).join(', ')
  ["#{collection_type}<#{member_type}>"]
end

.convert_hash(node) ⇒ Array<(String)> (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (Array<(String)>) 


97
98
99
100
101
102
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 97

def convert_hash(node)
  kv = node.last.children
  key_type = convert_node(kv.first).join(', ')
  value_type = convert_node(kv.last).join(', ')
  ["Hash{#{key_type} => #{value_type}}"]
end

.convert_list(node) ⇒ Array<String> (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (Array<String>) 


105
106
107
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 105

def convert_list(node)
  node.children.size == 1 ? convert_node(node.children.first) : [node.source]
end

.convert_node(node) ⇒ Array<String> (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (Array<String>) 


28
29
30
31
32
33
34
35
36
37
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 28

def convert_node(node)
  case node
  when YARD::Parser::Ruby::MethodCallNode
    node.namespace.source == 'T' ? convert_t_method(node) : [node.source]
  when YARD::Parser::Ruby::ReferenceNode
    node_source = node.source
    REF_TYPES[node_source] || [node_source]
  else convert_node_type(node)
  end
end

.convert_node_type(node) ⇒ Array<String> (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (Array<String>) 


40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 40

def convert_node_type(node)
  case node.type
  when :aref then convert_aref(node)
  when :arg_paren then convert_node(node.first)
  when :array then convert_array(node)
  # Fixed hashes as return values are unsupported:
  # https://github.com/lsegal/yard/issues/425
  #
  # Hash key params can be individually documented with `@option`, but
  # sig translation is currently unsupported.
  when :hash then ['Hash']
  # seen when sig methods omit parentheses
  when :list then convert_list(node)
  else convert_unknown(node)
  end
end

.convert_t_method(node) ⇒ Array<String> (private)

Parameters:

  • node (YARD::Parser::Ruby::MethodCallNode)

Returns:

  • (Array<String>) 


110
111
112
113
114
115
116
117
118
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 110

def convert_t_method(node)
  case node.method_name(true)
  # Order matters here, putting `nil` last results in a more concise return syntax in the UI (superscripted `?`):
  # https://github.com/lsegal/yard/blob/cfa62ae/lib/yard/templates/helpers/html_helper.rb#L499-L500
  when :nilable then convert_node(node.last) + REF_TYPES.fetch('NilClass')
  when :any then node[-1][0].children.flat_map { convert_node(_1) }
  else [node.source]
  end
end

.convert_unknown(node) ⇒ Array<(String)> (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode)

Returns:

  • (Array<(String)>) 


121
122
123
124
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 121

def convert_unknown(node)
  log.warn("Unsupported sig #{node.type} node #{node.source}")
  [node.source]
end

.sequence(node) ⇒ String (private)

Parameters:

  • node (YARD::Parser::Ruby::AstNode, nil)

Returns:

  • (String) 


83
84
85
86
87
 
# File 'lib/yard-sorbet/sig_to_yard.rb', line 83

def sequence(node)
  return '' unless node

  node.children.map { convert_node(_1) }.map { _1.size == 1 ? _1[0] : _1.to_s.tr('"', '') }.join(', ')
end