class SyntaxTree::For

def accept(visitor)
  visitor.visit_for(self)
end

def child_nodes
  [index, collection, statements]
end

def deconstruct_keys(_keys)
  {
    index: index,
    collection: collection,
    statements: statements,
    location: location,
    comments: comments
  }
end

def format(q)
  q.group do
    q.text("for ")
    q.group { q.format(index) }
    q.text(" in ")
    q.format(collection)
    unless statements.empty?
      q.indent do
        q.breakable(force: true)
        q.format(statements)
      end
    end
    q.breakable(force: true)
    q.text("end")
  end
end

def initialize(index:, collection:, statements:, location:, comments: [])
  @index = index
  @collection = collection
  @statements = statements
  @location = location
  @comments = comments
end