Class: RBI::Formatter

Inherits:
Object
  • Object
show all
Extended by:
T::Sig
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) ⇒ Formatter

Returns a new instance of Formatter.



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

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
)
  @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
end

Instance Attribute Details

#max_line_length ⇒ Object

Returns the value of attribute max_line_length.



9
10
11
 
# File 'lib/rbi/formatter.rb', line 9

def max_line_length
  @max_line_length
end

Instance Method Details

#format_file(file) ⇒ Object 



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

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

#format_tree(tree) ⇒ Object 



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

def format_tree(tree)
  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 



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

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