module GovukPublishingComponents
  module AppHelpers
    class TableHelper
      def self.helper(context, caption = nil, opt = {})
        builder = TableBuilder.new(context.tag)

        classes = %w(gem-c-table govuk-table)
        classes << "govuk-table--sortable" if opt[:sortable]

        caption_classes = %w(govuk-table__caption)
        caption_classes << opt[:caption_classes] if opt[:caption_classes]

        context.tag.table class: classes do
          context.concat context.tag.caption caption, class: caption_classes
          yield(builder)
        end
      end
      class TableBuilder
        include ActionView::Helpers::UrlHelper
        include ActionView::Helpers::TagHelper

        attr_reader :tag

        def initialize(tag)
          @tag = tag
        end

        def head
          tag.thead class: "govuk-table__head" do
            tag.tr class: "govuk-table__row" do
              yield(self)
            end
          end
        end

        def body
          tag.tbody class: "govuk-table__body" do
            yield(self)
          end
        end

        def row
          tag.tr class: "govuk-table__row" do
            yield(self)
          end
        end

        def header(str, opt = {})
          classes = %w(govuk-table__header)
          classes << "govuk-table__header--#{opt[:format]}" if opt[:format]
          classes << "govuk-table__header--active" if opt[:sort_direction]
          link_classes = %w(app-table__sort-link)
          link_classes << "app-table__sort-link--#{opt[:sort_direction]}" if opt[:sort_direction]
          str = link_to str, opt[:href], class: link_classes, data: opt[:data_attributes] if opt[:href]
          tag.th str, class: classes, scope: opt[:scope] || "col"
        end

        def cell(str, opt = {})
          classes = %w(govuk-table__cell)
          classes << "govuk-table__cell--" + opt[:format] if opt[:format]
          classes << "govuk-table__cell--empty" unless str
          str ||= "Not set"
          tag.td str, class: classes
        end
      end
    end
  end
end