Class: Spoom::Sorbet::Translate::RBSCommentsToSorbetSigs

Inherits:
Translator
  • Object
show all
Includes:
RBS::ExtractRBSComments
Defined in:
lib/spoom/sorbet/translate/rbs_comments_to_sorbet_sigs.rb

Instance Method Summary collapse

Methods included from RBS::ExtractRBSComments

#node_rbs_comments

Methods inherited from Translator

#rewrite

Methods inherited from Visitor

#visit_alias_global_variable_node, #visit_alias_method_node, #visit_alternation_pattern_node, #visit_and_node, #visit_arguments_node, #visit_array_node, #visit_array_pattern_node, #visit_assoc_node, #visit_assoc_splat_node, #visit_back_reference_read_node, #visit_begin_node, #visit_block_argument_node, #visit_block_local_variable_node, #visit_block_node, #visit_block_parameter_node, #visit_block_parameters_node, #visit_break_node, #visit_call_and_write_node, #visit_call_operator_write_node, #visit_call_or_write_node, #visit_call_target_node, #visit_capture_pattern_node, #visit_case_match_node, #visit_case_node, #visit_child_nodes, #visit_class_variable_and_write_node, #visit_class_variable_operator_write_node, #visit_class_variable_or_write_node, #visit_class_variable_read_node, #visit_class_variable_target_node, #visit_class_variable_write_node, #visit_constant_and_write_node, #visit_constant_operator_write_node, #visit_constant_or_write_node, #visit_constant_path_and_write_node, #visit_constant_path_node, #visit_constant_path_operator_write_node, #visit_constant_path_or_write_node, #visit_constant_path_target_node, #visit_constant_path_write_node, #visit_constant_read_node, #visit_constant_target_node, #visit_constant_write_node, #visit_defined_node, #visit_else_node, #visit_embedded_statements_node, #visit_embedded_variable_node, #visit_ensure_node, #visit_false_node, #visit_find_pattern_node, #visit_flip_flop_node, #visit_float_node, #visit_for_node, #visit_forwarding_arguments_node, #visit_forwarding_parameter_node, #visit_forwarding_super_node, #visit_global_variable_and_write_node, #visit_global_variable_operator_write_node, #visit_global_variable_or_write_node, #visit_global_variable_read_node, #visit_global_variable_target_node, #visit_global_variable_write_node, #visit_hash_node, #visit_hash_pattern_node, #visit_if_node, #visit_imaginary_node, #visit_implicit_node, #visit_implicit_rest_node, #visit_in_node, #visit_index_and_write_node, #visit_index_operator_write_node, #visit_index_or_write_node, #visit_index_target_node, #visit_instance_variable_and_write_node, #visit_instance_variable_operator_write_node, #visit_instance_variable_or_write_node, #visit_instance_variable_read_node, #visit_instance_variable_target_node, #visit_instance_variable_write_node, #visit_integer_node, #visit_interpolated_match_last_line_node, #visit_interpolated_regular_expression_node, #visit_interpolated_string_node, #visit_interpolated_symbol_node, #visit_interpolated_x_string_node, #visit_keyword_hash_node, #visit_keyword_rest_parameter_node, #visit_lambda_node, #visit_local_variable_and_write_node, #visit_local_variable_operator_write_node, #visit_local_variable_or_write_node, #visit_local_variable_read_node, #visit_local_variable_target_node, #visit_local_variable_write_node, #visit_match_last_line_node, #visit_match_predicate_node, #visit_match_required_node, #visit_match_write_node, #visit_missing_node, #visit_multi_target_node, #visit_multi_write_node, #visit_next_node, #visit_nil_node, #visit_no_keywords_parameter_node, #visit_numbered_parameters_node, #visit_numbered_reference_read_node, #visit_optional_keyword_parameter_node, #visit_optional_parameter_node, #visit_or_node, #visit_parameters_node, #visit_parentheses_node, #visit_pinned_expression_node, #visit_pinned_variable_node, #visit_post_execution_node, #visit_pre_execution_node, #visit_program_node, #visit_range_node, #visit_rational_node, #visit_redo_node, #visit_regular_expression_node, #visit_required_keyword_parameter_node, #visit_required_parameter_node, #visit_rescue_modifier_node, #visit_rescue_node, #visit_rest_parameter_node, #visit_retry_node, #visit_return_node, #visit_self_node, #visit_source_encoding_node, #visit_source_file_node, #visit_source_line_node, #visit_splat_node, #visit_statements_node, #visit_string_node, #visit_super_node, #visit_symbol_node, #visit_true_node, #visit_undef_node, #visit_unless_node, #visit_until_node, #visit_when_node, #visit_while_node, #visit_x_string_node, #visit_yield_node

Constructor Details

#initialize(ruby_contents, file:, max_line_length: nil) ⇒ RBSCommentsToSorbetSigs

: (String, file: String, ?max_line_length: Integer?) -> void



11
12
13
14
15
 
# File 'lib/spoom/sorbet/translate/rbs_comments_to_sorbet_sigs.rb', line 11

def initialize(ruby_contents, file:, max_line_length: nil)
  super(ruby_contents, file: file)

  @max_line_length = max_line_length
end

Instance Method Details

#visit_call_node(node) ⇒ Object

: (Prism::CallNode node) -> void



49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/spoom/sorbet/translate/rbs_comments_to_sorbet_sigs.rb', line 49

def visit_call_node(node)
  case node.message
  when "attr_reader", "attr_writer", "attr_accessor"
    visit_attr(node)
  else
    def_node = node.arguments&.arguments&.first
    if def_node&.is_a?(Prism::DefNode)
      rewrite_def(def_node, node_rbs_comments(node))
      return
    end

    super
  end
end

#visit_class_node(node) ⇒ Object

: (Prism::ClassNode node) -> void



19
20
21
22
23
 
# File 'lib/spoom/sorbet/translate/rbs_comments_to_sorbet_sigs.rb', line 19

def visit_class_node(node)
  apply_class_annotations(node)

  super
end

#visit_def_node(node) ⇒ Object

: (Prism::DefNode node) -> void



43
44
45
 
# File 'lib/spoom/sorbet/translate/rbs_comments_to_sorbet_sigs.rb', line 43

def visit_def_node(node)
  rewrite_def(node, node_rbs_comments(node))
end

#visit_module_node(node) ⇒ Object

: (Prism::ModuleNode node) -> void



27
28
29
30
31
 
# File 'lib/spoom/sorbet/translate/rbs_comments_to_sorbet_sigs.rb', line 27

def visit_module_node(node)
  apply_class_annotations(node)

  super
end

#visit_singleton_class_node(node) ⇒ Object

: (Prism::SingletonClassNode node) -> void



35
36
37
38
39
 
# File 'lib/spoom/sorbet/translate/rbs_comments_to_sorbet_sigs.rb', line 35

def visit_singleton_class_node(node)
  apply_class_annotations(node)

  super
end