class TOML::Parser

def initialize(content, options = {})
  @hash = {}
  @visited_keys = Set.new
  @current = @hash
  @symbolize_keys = options[:symbolize_keys]
  begin
    parsed = TOML::Document.parse(content)
    parsed.matches.map(&:value).compact.each { |m| m.accept_visitor(self) }
  rescue Citrus::ParseError => e
    raise TOML::ParseError.new(e.message)
  end
end

def visit_keygroup(keygroup)
  @current = keygroup.navigate_keys @hash, @visited_keys, @symbolize_keys
end

def visit_keyvalue(keyvalue)
  keyvalue.assign @current, @symbolize_keys
end

def visit_table_array(table_array)
  @current = table_array.navigate_keys @hash, @symbolize_keys
end