Class: RBI::Formatter

Inherits:
Object
  • Object
show all
Defined in:
lib/rbi/formatter.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(add_sig_templates: false, group_nodes: false, max_line_length: nil, nest_singleton_methods: false, nest_non_public_members: false, sort_nodes: false, replace_attributes_with_methods: false) ⇒ Formatter

: ( | ?add_sig_templates: bool, | ?group_nodes: bool, | ?max_line_length: Integer?, | ?nest_singleton_methods: bool, | ?nest_non_public_members: bool, | ?sort_nodes: bool, | ?replace_attributes_with_methods: bool | ) -> void



18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/rbi/formatter.rb', line 18

def initialize(
  add_sig_templates: false,
  group_nodes: false,
  max_line_length: nil,
  nest_singleton_methods: false,
  nest_non_public_members: false,
  sort_nodes: false,
  replace_attributes_with_methods: false
)
  @add_sig_templates = add_sig_templates
  @group_nodes = group_nodes
  @max_line_length = max_line_length
  @nest_singleton_methods = nest_singleton_methods
  @nest_non_public_members = nest_non_public_members
  @sort_nodes = sort_nodes
  @replace_attributes_with_methods = replace_attributes_with_methods
end

Instance Attribute Details

#max_line_length ⇒ Object

: Integer?



7
8
9
 
# File 'lib/rbi/formatter.rb', line 7

def max_line_length
  @max_line_length
end

Instance Method Details

#format_file(file) ⇒ Object

: (RBI::File file) -> void



43
44
45
 
# File 'lib/rbi/formatter.rb', line 43

def format_file(file)
  format_tree(file.root)
end

#format_tree(tree) ⇒ Object

: (RBI::Tree tree) -> void



48
49
50
51
52
53
54
55
 
# File 'lib/rbi/formatter.rb', line 48

def format_tree(tree)
  tree.replace_attributes_with_methods! if @replace_attributes_with_methods
  tree.add_sig_templates! if @add_sig_templates
  tree.nest_singleton_methods! if @nest_singleton_methods
  tree.nest_non_public_members! if @nest_non_public_members
  tree.group_nodes! if @group_nodes
  tree.sort_nodes! if @sort_nodes
end

#print_file(file) ⇒ Object

: (RBI::File file) -> String



37
38
39
40
 
# File 'lib/rbi/formatter.rb', line 37

def print_file(file)
  format_file(file)
  file.string(max_line_length: @max_line_length)
end