class ActiveAdmin::Views::TableFor

def build(obj, *attrs)
  options = attrs.extract_options!
  @sortable = options.delete(:sortable)
  @collection = obj.respond_to?(:each) && !obj.is_a?(Hash) ? obj : [obj]
  @resource_class = options.delete(:i18n)
  @resource_class ||= @collection.klass if @collection.respond_to? :klass
  @columns = []
  @row_class = options.delete(:row_class)
  build_table
  super(options)
  columns(*attrs)
end

def build_table
  build_table_head
  build_table_body
end

def build_table_body
  @tbody = tbody do
    # Build enough rows for our collection
    @collection.each do |elem|
      classes = [helpers.cycle("odd", "even")]
      if @row_class
        classes << @row_class.call(elem)
      end
      tr(class: classes.flatten.join(" "), id: dom_id_for(elem))
    end
  end
end

def build_table_cell(col, resource)
  td class: col.html_class do
    html = helpers.format_attribute(resource, col.data)
    # Don't add the same Arbre twice, while still allowing format_attribute to call status_tag
    current_arbre_element << html unless current_arbre_element.children.include? html
  end
end

def build_table_head
  @thead = thead do
    @header_row = tr
  end
end

def build_table_header(col)
  classes = Arbre::HTML::ClassList.new
  sort_key = sortable? && col.sortable? && col.sort_key
  params = request.query_parameters.except :page, :order, :commit, :format
  classes << "sortable" if sort_key
  classes << "sorted-#{current_sort[1]}" if sort_key && current_sort[0] == sort_key
  classes << col.html_class
  if sort_key
    th class: classes do
      link_to col.pretty_title, params: params, order: "#{sort_key}_#{order_for_sort_key(sort_key)}"
    end
  else
    th col.pretty_title, class: classes
  end
end

def column(*args, &block)
  options = default_options.merge(args.extract_options!)
  title = args[0]
  data = args[1] || args[0]
  col = Column.new(title, data, @resource_class, options, &block)
  @columns << col
  # Build our header item
  within @header_row do
    build_table_header(col)
  end
  # Add a table cell for each item
  @collection.each_with_index do |resource, index|
    within @tbody.children[index] do
      build_table_cell col, resource
    end
  end
end

def columns(*attrs)
  attrs.each { |attr| column(attr) }
end

def current_sort
  @current_sort ||= begin
    order_clause = active_admin_config.order_clause.new(active_admin_config, params[:order])
    if order_clause.valid?
      [order_clause.field, order_clause.order]
    else
      []
    end
  end
end

def default_options
  {
    i18n: @resource_class
  }
end

def order_for_sort_key(sort_key)
  current_key, current_order = current_sort
  return "desc" unless current_key == sort_key
  current_order == "desc" ? "asc" : "desc"
end

def sortable?
  !!@sortable
end

def tag_name
  "table"
end