module Devise
  module Models
    module EmailAuthenticatable
      USER_ASSOCIATION = :user

      def devise_scope
        user_association = self.class.reflect_on_association(USER_ASSOCIATION)
        if user_association
          user_association.class_name.constantize
        else
          raise "#{self.class.name}: Association :#{USER_ASSOCIATION} not found: Have you declared that ?"
        end
      end
    end

    module MultiEmailAuthenticatable
      extend ActiveSupport::Concern
      EMAILS_ASSOCIATION = :emails

      included do
        devise :database_authenticatable

        attr_accessor :current_login_email

        email_class.send :include, EmailAuthenticatable
      end

      def self.required_fields(klass)
        []
      end

      # Gets the primary email record.
      def primary_email_record
        valid_emails = emails.each.select do |email_record|
          !email_record.destroyed? && !email_record.marked_for_destruction?
        end

        result = valid_emails.find(&:primary?)
        result ||= valid_emails.first
        result
      end

      # Gets the primary email address of the user.
      def email
        primary_email_record.try(:email)
      end

      # Sets the default email address of the user.
      def email=(email)
        record = primary_email_record
        if email
          record ||= emails.build
          record.email = email
          record.primary = true
        elsif email.nil? && record
          record.mark_for_destruction
        end
      end

      module ClassMethods
        def find_first_by_auth_conditions(tainted_conditions, opts = {})
          tainted_conditions = tainted_conditions.dup
          email = tainted_conditions.delete(:email)
          if email && email.is_a?(String)
            conditions = devise_parameter_filter.filter(tainted_conditions).to_h.merge(opts).
                reverse_merge(emails: { email: email })

            resource = joins(:emails).find_by(conditions)
            resource.current_login_email = email if resource.respond_to?(:current_login_email=)
            resource
          else
            super(tainted_conditions, opts)
          end
        end

        def email_class
          email_association = reflect_on_association(EMAILS_ASSOCIATION)
          if email_association
            email_association.class_name.constantize
          else
            raise "#{self.class.name}: Association :#{EMAILS_ASSOCIATION} not found: It might because your declaration is after `devise :multi_email_confirmable`."
          end
        end
      end
    end
  end
end