added source_location to procs for rbx, all tests passing on rbx

1 files changed, 21 insertions, 16 deletions

diff --git a/lib/method_source.rb b/lib/method_source.rb
index 878ddbe..33eeb1f 100644
--- a/lib/method_source.rb
+++ b/lib/method_source.rb
@@ -47,7 +47,7 @@ module MethodSource
   # @return [File] The opened source file
   def self.source_helper(source_location)
     return nil if !source_location.is_a?(Array)
-    
+
     file_name, line = source_location
     File.open(file_name) do |file|
       (line - 1).times { file.readline }
@@ -56,26 +56,26 @@ module MethodSource
       loop do
         val = file.readline
         code << val
-        
+
         return code if valid_expression?(code)
       end
-    end    
+    end
   end
-  
+
   # Helper method responsible for opening source file and buffering up
-  # the comments for a specified method. Defined here to avoid polluting 
+  # the comments for a specified method. Defined here to avoid polluting
   # `Method` class.
   # @param [Array] source_location The array returned by Method#source_location
   # @return [String] The comments up to the point of the method.
   def self.comment_helper(source_location)
     return nil if !source_location.is_a?(Array)
-    
+
     file_name, line = source_location
     File.open(file_name) do |file|
       buffer = ""
       (line - 1).times do
         line = file.readline
-        # Add any line that is a valid ruby comment, 
+        # Add any line that is a valid ruby comment,
         # but clear as soon as we hit a non comment line.
         if (line =~ /^\s*#/) || (line =~ /^\s*$/)
           buffer << line.lstrip
@@ -83,11 +83,11 @@ module MethodSource
           buffer.replace("")
         end
       end
-      
+
       buffer
     end
   end
-  
+
   # This module is to be included by `Method` and `UnboundMethod` and
   # provides the `#source` functionality
   module MethodExtensions
@@ -103,7 +103,7 @@ module MethodSource
 
         klass.class_eval do
           orig_source = instance_method(:source)
-          
+
           define_method(:source) do
             begin
               super
@@ -111,11 +111,11 @@ module MethodSource
               orig_source.bind(self).call
             end
           end
-          
+
         end
       end
     end
-    
+
     # Return the sourcecode for the method as a string
     # (This functionality is only supported in Ruby 1.9 and above)
     # @return [String] The method sourcecode as a string
@@ -129,15 +129,15 @@ module MethodSource
     def source
       if respond_to?(:source_location)
         source = MethodSource.source_helper(source_location)
-        
+
         raise "Cannot locate source for this method: #{name}" if !source
       else
         raise "#{self.class}#source not supported by this Ruby version (#{RUBY_VERSION})"
       end
-      
+
       source
     end
-    
+
     # Return the comments associated with the method as a string.
     # (This functionality is only supported in Ruby 1.9 and above)
     # @return [String] The method's comments as a string
@@ -148,7 +148,7 @@ module MethodSource
     def comment
       if respond_to?(:source_location)
         comment = MethodSource.comment_helper(source_location)
-        
+
         raise "Cannot locate source for this method: #{name}" if !comment
       else
         raise "#{self.class}#comment not supported by this Ruby version (#{RUBY_VERSION})"
@@ -171,4 +171,9 @@ end
 
 class Proc
   include MethodSource::MethodExtensions
+
+  if defined?(RUBY_ENGINE) && RUBY_ENGINE =~ /rbx/
+    include MethodSource::SourceLocation::ProcExtensions
+  end
 end
+