nakajima-booty-call 0.0.1

data/lib/booty_call/callbacker.rb

@@ -0,0 +1,89 @@
+module BootyCall
+  class Callbacker
+    attr_reader :klass
+    
+    def initialize(klass)
+      @klass = klass
+      extend_klass
+    end
+    
+    def before(method_id, *symbols, &block)
+      add_callback(:before, method_id, *symbols, &block)
+    end
+    
+    def after(method_id, *symbols, &block)
+      add_callback(:after, method_id, *symbols, &block)
+    end
+    
+    private
+    
+    def extend_klass
+      klass.class_eval do
+        meta_eval { attr_reader :pristine_cache, :callback_cache }
+        @pristine_cache = Hash.new
+        @callback_cache = { :after => Hash.new([]), :before => Hash.new([]) }
+        extend ClassMethods
+        include InstanceMethods
+      end
+    end
+    
+    def add_callback(position, method_id, *symbols, &block)
+      klass.callback_cache[position][method_id].tap do |slot|
+        slot.push(block) if block_given?
+        slot.concat(symbols)
+        slot.tap.compact!.uniq!
+      end
+      
+      klass.pristine_cache[method_id] ||= begin
+        klass.instance_method(method_id).tap do
+          redefine_method method_id
+        end
+      end
+    end
+    
+    def redefine_method(method_id)
+      klass.class_eval(<<-EOS, "(__DELEGATION__)", 1)
+        def #{method_id}(*args, &block)
+          catch(#{method_id.to_sym.inspect}) do
+            return unless self.class.run_callbacks_for(self, :before, #{method_id.inspect})
+            result = __PRISTINE__(#{method_id.inspect}, *args, &block)
+            self.class.run_callbacks_for(self, :after, #{method_id.inspect}, result)
+            return result
+          end
+        end
+        
+        def #{method_id}_without_callbacks(*args, &block)
+          __PRISTINE__(#{method_id.inspect}, *args, &block)
+        end
+      EOS
+    end
+    
+    module ClassMethods
+      def run_callbacks_for(target, position, method_id, *results)
+        callbacks = callback_cache[position][method_id.to_sym]
+        
+        handler = proc do |fn|
+          fn.is_a?(Proc) ? fn : begin
+            target.method(fn).arity.abs == results.length ?
+              proc { send(fn, *results) } :
+              proc { send(fn) }
+          end
+        end
+        
+        callbacks.empty? ? true : callbacks.map { |fn|
+          target.instance_exec(*results, &handler.call(fn))
+        }.all?
+      end
+    end
+    
+    module InstanceMethods
+      def __PRISTINE__(method_id, *args, &block)
+        if method = self.class.pristine_cache[method_id]
+          method.bind(self).call(*args, &block)
+        else
+          raise NoMethodError, "No method named #{method_id.inspect}"
+        end
+      end
+    end
+  end
+end

data/lib/booty_call/hook.rb

@@ -0,0 +1,31 @@
+module BootyCall
+  module Hook
+    def self.included(klass)
+      klass.class_eval do
+        extend(ClassMethods)
+        @callbacker = BootyCall::Callbacker.new(self)
+        @introspector = BootyCall::Introspector.new(self)
+      end
+    end
+    
+    module ClassMethods
+      def before(method_id, *args, &block)
+        callback(:before, method_id, *args, &block)
+      end
+      
+      def after(method_id, *args, &block)
+        callback(:after, method_id, *args, &block)
+      end
+      
+      def observe(method_id, &block)
+        @introspector.observe(method_id, &block)
+      end
+      
+      def callback(position, method_id, *args, &block)
+        @introspector.defined_methods.include?(method_id) ?
+          @callbacker.send(position, method_id, *args, &block) :
+          observe(method_id) { send(position, method_id, *args, &block) }
+      end
+    end
+  end
+end

data/lib/booty_call/introspector.rb

@@ -0,0 +1,41 @@
+module BootyCall
+  class Introspector
+    attr_reader :klass
+    
+    def initialize(klass)
+      @klass = klass
+      @observed_methods = { }
+    end
+    
+    def observe_klass!
+      @observed ||= begin
+        this = self
+        klass.meta_def(:method_added) do |m|
+          this.check_method(m)
+        end and true
+      end
+    end
+    
+    def observing?(method_id)
+      not not @observed_methods[method_id]
+    end
+    
+    def observe(method_id, &block)
+      observe_klass!
+      @observed_methods[method_id] ||= []
+      @observed_methods[method_id].tap do |set|
+        set.push(block) if block_given?
+        set.tap.compact!.uniq!
+      end
+    end
+    
+    def check_method(method_id)
+      handlers = @observed_methods.delete(method_id)
+      handlers.each(&:call) rescue nil
+    end
+    
+    def defined_methods
+      (klass.instance_methods - Object.instance_methods).map(&:to_sym)
+    end
+  end
+end

data/lib/booty_call.rb

@@ -0,0 +1,11 @@
+$LOAD_PATH << File.dirname(__FILE__) + '/booty_call'
+
+module BootyCall
+  VERSION = '0.0.1'
+end
+
+require 'rubygems'
+require 'nakajima'
+require 'callbacker'
+require 'introspector'
+require 'hook'

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification 
+name: nakajima-booty-call
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Pat Nakajima
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2008-11-12 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: nakajima-nakajima
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: 
+email: patnakajima@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/booty_call
+- lib/booty_call/callbacker.rb
+- lib/booty_call/hook.rb
+- lib/booty_call/introspector.rb
+- lib/booty_call.rb
+has_rdoc: false
+homepage: http://github.com/nakajima/booty-call
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Callbacks for your Ruby
+test_files: []
+