dazzl-tv/validates_timeliness-mongoid

# frozen_string_literal: true

require 'validates_timeliness'

module ValidatesTimeliness
  module ORM
    module Mongoid
      extend ActiveSupport::Concern
      # You need define the fields before you define the validations.
      # It is best to use the plugin parser to avoid errors on a bad
      # field value in Mongoid. Parser will return nil rather than error.

      module ClassMethods
        # Mongoid has no bulk attribute method definition hook. It defines
        # them with each field definition. So we likewise define them after
        # each validation is defined.
        #
        def timeliness_validation_for(attr_names, type)
          super
          attr_names.each do |attr_name|
            define_timeliness_write_method(attr_name)
          end
        end

        def timeliness_attribute_type(attr_name)
          {
            Date => :date,
            Time => :time,
            DateTime => :datetime
          }[fields[database_field_name(attr_name)].type] || :datetime
        end

        protected

        def timeliness_type_cast_code(attr_name, var_name)
          type = timeliness_attribute_type(attr_name)

          "#{var_name} = Timeliness::Parser.parse(value, :#{type})"
        end

        def define_timeliness_write_method(attr_name)
          generated_timeliness_methods.module_eval <<-STR, __FILE__, __LINE__ + 1
            def #{attr_name}=(value)                      # def publish_date=(value)
              @timeliness_cache ||= {}                    #   @timeliness_cache ||= {}
              @timeliness_cache['#{attr_name}'] = value   #   @timeliness_cache['publish_date'] = value
              @attributes['#{attr_name}'] = super         #   @attributes['publish_date']
            end                                           # end
          STR
        end

        def generated_timeliness_methods
          @generated_timeliness_methods ||= Module.new do |_m|
            extend Mutex_m
          end
          @generated_timeliness_methods.tap { |mod| include mod }
        end
      end

      # This is needed in order to mark the attribute as changed;
      # otherwise, mongoid won't save it to the database.
      def write_timeliness_attribute(attr_name, value)
        attribute_will_change!(database_field_name(attr_name))
        super
      end

      def reload(*args)
        _clear_timeliness_cache
        super
      end

      def read_timeliness_attribute_before_type_cast(attr_name)
        @timeliness_cache && @timeliness_cache[attr_name] || @attributes[attr_name]
      end

      def _clear_timeliness_cache
        @timeliness_cache = {}
      end
    end
  end
end

module Mongoid
  module Document
    include ValidatesTimeliness::AttributeMethods
    include ValidatesTimeliness::ORM::Mongoid
  end
end