Class: Annotato::EnumFormatter

Inherits:
Object
  • Object
show all
Defined in:
lib/annotato/enum_formatter.rb

Class Method Summary collapse

Class Method Details

.format(model) ⇒ Object 



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
# File 'lib/annotato/enum_formatter.rb', line 5

def self.format(model)
  return [] if model.defined_enums.empty?

  model.defined_enums.map do |attr, values|
    lines = ["#  #{attr}: {"]

    formatted_values = values.map do |key, val|
      if key.to_s == val.to_s
        "#{key}"
      else
        "#{key} (#{val})"
      end
    end

    lines += formatted_values.map { |v| "#    #{v}," }
    lines[-1] = lines[-1].chomp(',') # remove trailing comma from last line
    lines << "#  }"
    lines.join("\n")
  end
end