DFE-Digital/govuk-formbuilder

module GOVUKDesignSystemFormBuilder
  module Elements
    module Radios
      class Collection < Base
        include Traits::Error
        include Traits::Hint
        include Traits::Supplemental

        def initialize(builder, object_name, attribute_name, collection, value_method:, text_method:, hint_method:, hint:, legend:, caption:, inline:, small:, bold_labels:, form_group:, include_hidden:, **kwargs, &block)
          super(builder, object_name, attribute_name, &block)

          @collection      = collection
          @value_method    = value_method
          @text_method     = text_method
          @hint_method     = hint_method
          @inline          = inline
          @small           = small
          @legend          = legend
          @caption         = caption
          @hint            = hint
          @form_group      = form_group
          @include_hidden  = include_hidden
          @bold_labels     = (bold_labels.nil? && auto_bold_labels?) || bold_labels
          @html_attributes = kwargs
        end

        def html
          Containers::FormGroup.new(*bound, **@form_group).html do
            Containers::Fieldset.new(*bound, **fieldset_options).html do
              safe_join([hidden_field, supplemental_content, hint_element, error_element, radios])
            end
          end
        end

      private

        def auto_bold_labels?
          (config.default_collection_radio_buttons_auto_bold_labels &&
            @hint_method.present?)
        end

        def fieldset_options
          {
            legend: @legend,
            caption: @caption,
            described_by: [error_id, hint_id, supplemental_id]
          }
        end

        def radios
          Containers::Radios.new(@builder, inline: @inline, small: @small, **@html_attributes).html do
            safe_join(collection)
          end
        end

        def hidden_field
          return unless @include_hidden

          @builder.hidden_field(@attribute_name, value: "")
        end

        def collection
          @collection.map.with_index do |item, i|
            Elements::Radios::CollectionRadioButton.new(*bound, item, **collection_options(i)).html
          end
        end

        def collection_options(index)
          {
            value_method: @value_method,
            text_method: @text_method,
            hint_method: @hint_method,
            link_errors: link_errors?(index),
            bold_labels: @bold_labels
          }
        end

        def link_errors?(index)
          has_errors? && index.zero?
        end
      end
    end
  end
end