arraysugar 0.1.0

data/lib/array_sugar.rb

@@ -0,0 +1,124 @@
+## array_sugar.rb -- sweet, sweet, syntax for ruby arrays
+## Copyright (c) David Salamon, 2007, all rights reserved
+#
+# Redistribution and use of this software in source and binary forms, with or without modification, are
+# permitted provided that the following conditions are met:
+#
+# * Redistributions of source code must retain the above
+#   copyright notice, this list of conditions and the
+#   following disclaimer.
+#
+# * Redistributions in binary form must reproduce the above
+#   copyright notice, this list of conditions and the
+#   following disclaimer in the documentation and/or other
+#   materials provided with the distribution.
+#
+# * Neither the name of David Salamon nor the names of his
+#   contributors may be used to endorse or promote products
+#   derived from this software without specific prior
+#   written permission from David Salamon.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED
+# WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
+# PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
+# TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+module ArraySugar
+  class IterationProxy
+    instance_methods.each do |m|
+      undef_method(m) unless m =~ /(^__|^nil\\?|^send)/
+    end
+
+    def initialize(array, mode)
+      @array = array
+      @mode = mode
+      @depth = 0
+    end
+
+    def __do_call(e, depth = @depth)
+      if (depth != 0) && (e.is_a?(Array) || e.is_a?(Range))
+        e.send(@mode) {|e| __do_call(e, depth - 1) }
+      else
+        if @nils_okay && e.nil?
+          nil
+        else
+          e.send(@s, *@a, &@b)
+        end
+      end
+    end
+
+    def method_missing(s, *a, &b)
+      @s = s; @a = a; @b = b
+      @array = @array.send(@mode) {|e| __do_call(e) }
+      self
+    end
+
+    def to_a
+      @array
+    end
+
+    def leafs
+      @depth = -1
+      self
+    end
+    def deep(depth = -1)
+      @depth = -1
+    end
+    def shallow(depth = 0)
+      @depth = depth
+    end
+
+    def nils_okay
+      @nils_okay = true
+      self
+    end
+    alias_method :nils_ok, :nils_okay
+
+    def inspect
+      to_a.inspect
+    end
+
+    [:all?, :any?, :map, :map!, :collect, :collect!, :detect, :find, :find_all,
+     :reject, :reject!, :reverse_each, :select, :sort_by, :each].each do |m|
+      class_eval <<-CODE
+      def #{m}(*args, &b)
+        if args.size == 0 && !block_given?
+          @mode = :#{m}
+          self
+        else
+          @array.orig_#{m}(*args, &b)
+        end
+      end
+      CODE
+    end
+  end
+end
+
+class Array
+  [:all?, :any?, :map, :map!, :collect, :collect!, :detect, :find, :find_all,
+   :reject, :reject!, :reverse_each, :select, :sort_by, :each].each do |m|
+    class_eval <<-CODE
+      alias_method :orig_#{m}, :#{m}
+      def #{m}(*args, &b)
+        return ArraySugar::IterationProxy.new(self, :#{m}) if args.size == 0 && !block_given?
+        orig_#{m}(*args, &b)
+      end
+    CODE
+  end
+end
+
+class Range
+  [:all?, :any?, :map, :collect, :detect, :find, :find_all, :reject, :select, :sort_by, :each].each do |m|
+    class_eval <<-CODE
+      alias_method :orig_#{m}, :#{m}
+      def #{m}(*args, &b)
+        return ArraySugar::IterationProxy.new(self, :#{m}) if args.size == 0 && !block_given?
+        orig_#{m}(*args, &b)
+      end
+    CODE
+  end
+end

metadata

@@ -0,0 +1,46 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.0
+specification_version: 1
+name: arraysugar
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+date: 2007-10-06 00:00:00 +00:00
+summary: Nice Ruby Array manipulation syntax
+require_paths: 
+- lib
+email: 
+homepage: 
+rubyforge_project: 
+description: 
+autorequire: 
+default_executable: 
+bindir: bin
+has_rdoc: false
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: 1.8.1
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: []
+
+files: 
+- lib/array_sugar.rb
+test_files: []
+
+rdoc_options: []
+
+extra_rdoc_files: []
+
+executables: []
+
+extensions: []
+
+requirements: []
+
+dependencies: []
+