class ActiveAdmin::Views::TableFor::Column

def initialize(*args, &block)
  @options = args.extract_options!
  @title = args[0]
  html_classes = [:col]
  if @options.has_key?(:class)
    html_classes << @options.delete(:class)
  elsif @title.present?
    html_classes << "col-#{@title.to_s.parameterize(separator: "_")}"
  end
  @html_class = html_classes.join(' ')
  @data = args[1] || args[0]
  @data = block if block
  @resource_class = args[2]
end

def pretty_title
  if @title.is_a? Symbol
    default = @title.to_s.titleize
    if @options[:i18n].respond_to? :human_attribute_name
      @title = @options[:i18n].human_attribute_name @title, default: default
    else
      default
    end
  else
    @title
  end
end

def sort_column_name
  @data.is_a?(Symbol) ? @data.to_s : @title.to_s
end

def sort_key
  # If boolean or nil, use the default sort key.
  if @options[:sortable].nil? || @options[:sortable] == true || @options[:sortable] == false
    sort_column_name
  else
    @options[:sortable].to_s
  end
end

def sortable?
  if @options.has_key?(:sortable)
    !!@options[:sortable]
  elsif @resource_class
    @resource_class.column_names.include?(sort_column_name)
  else
    @title.present?
  end
end