module ClassProfiler

# File lib/class_profiler.rb, line 72
def self._get_methods_for(_caller, inherited = false)
  _caller.constantize if _caller.is_a? String #use const_get

  array = []
  array.concat(
    _caller.instance_methods(inherited)
  ).concat(
    _caller.protected_instance_methods(inherited)
  ).concat(
    _caller.private_instance_methods(inherited)
  )

  return array unless inherited
  return (array - IGNORED_METHODS).flatten.uniq
end

# File lib/class_profiler.rb, line 29
def self.for(options = {})
  methods = options[:instance_methods] || []
  _caller = CONSTANTIZE.call(caller_locations(1,1)[0].label)

  if options[:modules]
    methods.concat(
      options[:modules].map{|m|
        _get_methods_for(m)
      }
    )
  end

  @__cp_instance_methods = methods.flatten.uniq

  return self
end

# File lib/class_profiler.rb, line 46
def self.included(base)
  @__cp_instance_methods = _get_methods_for(base, true) if @__cp_instance_methods.nil?

  default_protected_methods = self.protected_instance_methods
  default_private_methods = self.private_instance_methods

  (@__cp_instance_methods).each do |method_name|
    base.send(:alias_method, "__#{method_name}", method_name)

    base.send(:define_method, method_name) do |*args, &block|
      Benchmark.instance.start(label(__method__)){
        if block
          self.send("__#{method_name}", *args, &block)
        else
          self.send("__#{method_name}", *args)
        end
      }
    end

    protected(method_name) if default_protected_methods.include?(method_name)
    private(method_name) if default_private_methods.include?(method_name)
  end

  @__cp_instance_methods = nil
end