active_model_serializers/lib/active_model/serializer/adapter/cached_serializer.rb

module ActiveModel
  class Serializer
    module Adapter
      class CachedSerializer
        def initialize(serializer)
          @cached_serializer = serializer
          @klass             = @cached_serializer.class
        end

        def cache_check(adapter_instance)
          if cached?
            @klass._cache.fetch(cache_key, @klass._cache_options) do
              yield
            end
          elsif fragment_cached?
            FragmentCache.new(adapter_instance, @cached_serializer, adapter_instance.instance_options).fetch
          else
            yield
          end
        end

        def cached?
          @klass._cache && !@klass._cache_only && !@klass._cache_except
        end

        def fragment_cached?
          @klass._cache && (@klass._cache_only && !@klass._cache_except || !@klass._cache_only && @klass._cache_except)
        end

        def cache_key
          return @cache_key if defined?(@cache_key)

          parts = []
          parts << object_cache_key
          parts << @klass._cache_digest unless @klass._cache_options && @klass._cache_options[:skip_digest]
          @cache_key = parts.join('/')
        end

        def object_cache_key
          object_time_safe = @cached_serializer.object.updated_at
          object_time_safe = object_time_safe.strftime('%Y%m%d%H%M%S%9N') if object_time_safe.respond_to?(:strftime)
          (@klass._cache_key) ? "#{@klass._cache_key}/#{@cached_serializer.object.id}-#{object_time_safe}" : @cached_serializer.object.cache_key
        end

        # find all cache_key for the collection_serializer
        # @param collection_serializer
        # @param include_tree
        # @return [Array] all cache_key of collection_serializer
        def self.object_cache_keys(serializers, include_tree)
          cache_keys = []

          serializers.each do |serializer|
            cache_keys << object_cache_key(serializer)

            serializer.associations(include_tree).each do |association|
              if association.serializer.respond_to?(:each)
                association.serializer.each do |sub_serializer|
                  cache_keys << object_cache_key(sub_serializer)
                end
              else
                cache_keys << object_cache_key(association.serializer)
              end
            end
          end

          cache_keys.compact.uniq
        end

        # @return [String, nil] the cache_key of the serializer or nil
        def self.object_cache_key(serializer)
          return unless serializer.present? && serializer.object.present?

          cached_serializer = new(serializer)
          cached_serializer.cached? ? cached_serializer.cache_key : nil
        end
      end
    end
  end
end