refining 0.0.4 → 0.0.5

data/lib/refining.rb

@@ -12,9 +12,30 @@
 
 class Object
   def refine_singleton_method (meth, &block)
-    class << self
-      self
-    end.refine_method meth, &block
+    return unless block
+
+    target = self
+
+    (method(meth) rescue nil).tap {|old|
+      old ||= proc {}
+
+      what = class << target
+        self
+      end
+
+      what.send :define_method, meth do |*args, &blk|
+        return if @__refining_defined__ && [:method_added, :singleton_method_added].member?(meth)
+        
+        @__refining_defined__ = true
+        what.send(:define_method, 'temporary method for refining', &block)
+        
+        target.send('temporary method for refining', old.is_a?(UnboundMethod) ? old.bind(self) : old, *args, &blk).tap {
+          what.send(:undef_method, 'temporary method for refining') rescue nil
+
+          @__refining_defined__ = false
+        }
+      end
+    }
   end
 end
 
@@ -22,13 +43,26 @@ class Module
   def refine_method (meth, &block)
     return unless block
 
-    old = instance_method(meth) rescue nil
+    (instance_method(meth) rescue nil).tap {|old|
+      old ||= proc {}
 
-    define_method(meth) {|*args|
-      instance_exec((old.is_a?(UnboundMethod) ? old.bind(self) : old) || proc {}, *args, &block)
-    }
+      define_method(meth) {|*args, &blk|
+        return if @__refining_defined__ && [:method_added, :singleton_method_added].member?(meth)
+
+        what = class << self
+          self
+        end
 
-    old
+        @__refining_defined__ = true
+        what.send(:define_method, 'temporary method for refining', &block)
+        
+        send('temporary method for refining', old.is_a?(UnboundMethod) ? old.bind(self) : old, *args, &blk).tap {
+          what.send(:undef_method, 'temporary method for refining') rescue nil
+
+          @__refining_defined__ = false
+        }
+      }
+    }
   end
 
   alias refine_module_method refine_singleton_method

metadata

@@ -1,45 +1,54 @@
---- !ruby/object:Gem::Specification
+--- !ruby/object:Gem::Specification 
 name: refining
-version: !ruby/object:Gem::Version
-  version: 0.0.4
+version: !ruby/object:Gem::Version 
   prerelease: 
+  version: 0.0.5
 platform: ruby
-authors:
+authors: 
 - meh.
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2011-08-14 00:00:00.000000000Z
+
+date: 2011-08-18 00:00:00 Z
 dependencies: []
+
 description: 
 email: meh@paranoici.org
 executables: []
+
 extensions: []
+
 extra_rdoc_files: []
-files:
+
+files: 
 - lib/refining.rb
 homepage: http://github.com/meh/refining
 licenses: []
+
 post_install_message: 
 rdoc_options: []
-require_paths:
+
+require_paths: 
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement
+required_ruby_version: !ruby/object:Gem::Requirement 
   none: false
-  requirements:
-  - - ! '>='
-    - !ruby/object:Gem::Version
-      version: '0'
-required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
   none: false
-  requirements:
-  - - ! '>='
-    - !ruby/object:Gem::Version
-      version: '0'
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
 requirements: []
+
 rubyforge_project: 
-rubygems_version: 1.8.6
+rubygems_version: 1.8.5
 signing_key: 
 specification_version: 3
 summary: Easily refine methods
 test_files: []
+