Contents

require "entraceable/version"
require "entraceable_preference"

module Entraceable
  @preference = Preference.new

  class << self
    attr_accessor :preference

    def logger
      @logger ||= Rails.logger
    end

    def logger=(arg)
      @logger = arg
    end

    def method_missing(method, *args, &block)
      raise NoMethodError.new %q(undefined method `#{method}' for #{preference}:#{preference.class}) unless accept? method
      preference.__send__ method, *args, &block
    end

    private
      def accept?(method)
        %w(default_level default_level= disable enable enabled?).map(&:intern).include? method
      end
  end

  def entraceable(method, tag: nil, level: nil)
    alias_name = alias_name_for method
    __send__ :alias_method, alias_name, method
    class_eval <<-EOS
      def #{method}(*args, &block)
        indent = " " * ((@indent_level ||= 0) * 2)
        level = (#{level.inspect} || Entraceable.default_level).intern
        puts = ->c{Entraceable.logger.tagged(%Q(#{tag})) {Entraceable.logger.__send__ level, indent + c} if Entraceable.enabled?}
        puts.call %Q(#{method} is called with arguments, \#\{args.map(&:inspect).join(", ")\})
        @indent_level += 1
        begin
          __send__(:#{alias_name}, *args, &block).tap{|result|puts.call %Q(#{method} returns \#\{result\})}
        ensure
          @indent_level -= 1
        end
      end
    EOS
  end

  def distraceable(method)
    alias_name = alias_name_for method
    __send__ :remove_method, method
    __send__ :alias_method, method, alias_name
    __send__ :remove_method, alias_name
  end

  private
    def alias_name_for(method)
      ["original", __id__, method.to_s.sub(/\[\]/, "indexer")].join("_").intern
    end
end

Object.extend Entraceable

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
entraceable-0.1.6	lib/entraceable.rb