# frozen_string_literal: true

module ActionPolicy
  module Ext
    # Adds #_policy_cache_key method to Object,
    # which just call #policy_cache_key or #cache_key
    # or #object_id (if `use_object_id` parameter is set to true).
    #
    # For other core classes returns string representation.
    #
    # Raises ArgumentError otherwise.
    module PolicyCacheKey # :nodoc: all
      module ObjectExt
        def _policy_cache_key(use_object_id: false)
          return policy_cache_key if respond_to?(:policy_cache_key)
          return cache_key if respond_to?(:cache_key)

          return object_id if use_object_id == true

          raise ArgumentError, "object is not cacheable"
        end
      end

      # JRuby doesn't support _global_ modules refinements (see https://github.com/jruby/jruby/issues/5220)
      # Fallback to monkey-patching.
      ::Object.include(ObjectExt) if RUBY_PLATFORM =~ /java/i

      refine Object do
        include ObjectExt
      end

      refine NilClass do
        def _policy_cache_key(*)
          ""
        end
      end

      refine TrueClass do
        def _policy_cache_key(*)
          "t"
        end
      end

      refine FalseClass do
        def _policy_cache_key(*)
          "f"
        end
      end

      refine String do
        def _policy_cache_key(*)
          self
        end
      end

      refine Symbol do
        def _policy_cache_key(*)
          to_s
        end
      end

      if RUBY_PLATFORM =~ /java/i
        refine Integer do
          def _policy_cache_key(*)
            to_s
          end
        end

        refine Float do
          def _policy_cache_key(*)
            to_s
          end
        end
      else
        refine Numeric do
          def _policy_cache_key(*)
            to_s
          end
        end
      end

      refine Time do
        def _policy_cache_key(*)
          to_s
        end
      end
    end
  end
end