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)
  add_class "data-table"
  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|
      tr(id: dom_id_for(elem), class: @row_class&.call(elem))
    end
  end
end

def build_table_cell(col, resource)
  td class: col.html_class, "data-column": col.title_id.presence 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)
  sort_key = sortable? && col.sortable? && col.sort_key
  params = request.query_parameters.except :page, :order, :commit, :format
  attributes = {
    class: col.html_class,
    "data-column": col.title_id.presence,
    "data-sortable": (sort_key.present?) ? "" : nil,
    "data-sort-direction": (sort_key && current_sort[0] == sort_key) ? current_sort[1] : nil
  }
  if sort_key
    th(attributes) do
      link_to params: params, order: "#{sort_key}_#{order_for_sort_key(sort_key)}" do
        svg = '<svg class="data-table-sorted-icon" fill="none" viewBox="0 0 10 6"><path stroke="currentColor" stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="m1 1 4 4 4-4"/></svg>'
        (col.pretty_title + svg).html_safe
      end
    end
  else
    th col.pretty_title, attributes
  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