module Compony
  module ModelFields
    class Association < Base
      def initialize(...)
        super
        resolve_association!
      end

      def value_for(data, link_to_component: nil, link_opts: {}, controller: nil, **_)
        if link_to_component
          fail('Must pass controller if link_to_component is given.') unless controller
          return transform_and_join(data.send(@name), controller:) do |el|
            next nil if el.nil?
            if Compony.comp_class_for(link_to_component, el)
              next controller.helpers.compony_link(link_to_component, el, **link_opts)
            else
              next el.label
            end
          end
        else
          return transform_and_join(data.send(@name), controller:) { |el| el&.label }
        end
      end

      def schema_line
        local_schema_key = @schema_key # Capture schema_key as it will not be available within the lambda
        target_primary_key_type = @target_class.primary_key_type_key
        if multi?
          return proc do
            ary? local_schema_key do
              if target_primary_key_type == :integer
                list :integer, cast_str: true
              elsif target_primary_key_type == :string
                list :string
              else
                fail("Unsupported target primary_key_type_key #{target_primary_key_type}")
              end
            end
          end
        else
          return proc do
            if target_primary_key_type == :integer
              int? local_schema_key, cast_str: true
            elsif target_primary_key_type == :string
              str? local_schema_key
            else
              fail("Unsupported target primary_key_type_key #{target_primary_key_type}")
            end
          end
        end
      end

      def simpleform_input(form, _component, name: nil, **input_opts)
        return form.association name || @name, **input_opts
      end

      def simpleform_input_hidden(form, _component, name: nil, **input_opts)
        return form.input name || @schema_key, as: :hidden, **input_opts
      end

      protected

      # Uses Rails methods to figure out the arity, schema key etc. and store them.
      # This can be auto-inferred without accessing the database.
      def resolve_association!
        @association = true
        association_info = @model_class.reflect_on_association(@name) || fail("Association #{@name.inspect} does not exist for #{@model_class.inspect}.")
        @multi = association_info.macro == :has_many
        @target_class = association_info.klass
        id_name = "#{@name.to_s.singularize}_id"
        @schema_key = @multi ? id_name.pluralize.to_sym : id_name.to_sym
      rescue ActiveRecord::NoDatabaseError
        Rails.logger.warn('Warning: Compony could not auto-detect fields due to missing database. This is ok when running db:create.')
      end
    end
  end
end