class ErYamlModel

# File lib/yaml2erd/er_yaml_model.rb, line 2
def initialize(yaml_model)
  @yaml_model = yaml_model

  parse_columns
end

# File lib/yaml2erd/er_yaml_model.rb, line 8
def columns
  @yaml_model[:columns]
end

# File lib/yaml2erd/er_yaml_model.rb, line 25
def description
  @yaml_model[:description]
end

# File lib/yaml2erd/er_yaml_model.rb, line 21
def group_name
  @yaml_model[:group]
end

# File lib/yaml2erd/er_yaml_model.rb, line 12
def parsed_columns
  @yaml_model[:parsed_columns]
end

# File lib/yaml2erd/er_yaml_model.rb, line 16
def relations
  # TODO: 同じassociationのtypeまとめる形の方がいいかも yamlの構造を変えるかここで変換するか
  @yaml_model[:relations]
end

# File lib/yaml2erd/er_yaml_model.rb, line 31
def parse_columns
  @yaml_model[:parsed_columns] = @yaml_model[:columns].map do |column|
    { column_name: column[0], column_detail: column[1] }
  end
end