module ActiveSupport::Memoizable::InstanceMethods

def self.included(base)
  base.class_eval do
    unless base.method_defined?(:freeze_without_memoizable)
      alias_method_chain :freeze, :memoizable
    end
  end
end

def flush_cache(*syms)
  syms.each do |sym|
    (methods + private_methods + protected_methods).each do |m|
      if m.to_s =~ /^_unmemoized_(#{sym.to_s.gsub(/\?\Z/, '\?')})/
        ivar = ActiveSupport::Memoizable.memoized_ivar_for($1)
        instance_variable_get(ivar).clear if instance_variable_defined?(ivar)
      end
    end
  end
end

def freeze_with_memoizable
  memoize_all unless frozen?
  freeze_without_memoizable
end

def memoize_all
  prime_cache ".*"
end

def prime_cache(*syms)
  syms.each do |sym|
    methods.each do |m|
      if m.to_s =~ /^_unmemoized_(#{sym})/
        if method(m).arity == 0
          __send__($1)
        else
          ivar = ActiveSupport::Memoizable.memoized_ivar_for($1)
          instance_variable_set(ivar, {})
        end
      end
    end
  end
end

def unmemoize_all
  flush_cache ".*"
end