nuggets 0.9.9 → 1.0.0

data/lib/nuggets/integer/length.rb

@@ -0,0 +1,5 @@
+require 'nuggets/integer/length_mixin'
+
+class Integer
+  include Nuggets::Integer::LengthMixin
+end

data/lib/nuggets/integer/length_mixin.rb

@@ -0,0 +1,49 @@
+#--
+###############################################################################
+#                                                                             #
+# nuggets -- Extending Ruby                                                   #
+#                                                                             #
+# Copyright (C) 2007-2011 Jens Wille                                          #
+#                                                                             #
+# Authors:                                                                    #
+#     Jens Wille <jens.wille@gmail.com>                                       #
+#                                                                             #
+# nuggets is free software; you can redistribute it and/or modify it under    #
+# the terms of the GNU Affero General Public License as published by the Free #
+# Software Foundation; either version 3 of the License, or (at your option)   #
+# any later version.                                                          #
+#                                                                             #
+# nuggets is distributed in the hope that it will be useful, but WITHOUT ANY  #
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS   #
+# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for     #
+# more details.                                                               #
+#                                                                             #
+# You should have received a copy of the GNU Affero General Public License    #
+# along with nuggets. If not, see <http://www.gnu.org/licenses/>.             #
+#                                                                             #
+###############################################################################
+#++
+
+module Nuggets
+  class Integer
+    module LengthMixin
+
+  # call-seq:
+  #   int.length => anInteger
+  #
+  # Returns the length of _int_.
+  def length
+    to_s.length
+  end
+
+  # call-seq:
+  #   int.digit_count => anInteger
+  #
+  # Returns the digit count of _int_. [ruby-core:22383]
+  def digit_count
+    abs.length
+  end
+
+    end
+  end
+end

data/lib/nuggets/integer/map.rb

@@ -0,0 +1,5 @@
+require 'nuggets/integer/map_mixin'
+
+class Integer
+  include Nuggets::Integer::MapMixin
+end

data/lib/nuggets/integer/map_mixin.rb

@@ -0,0 +1,42 @@
+#--
+###############################################################################
+#                                                                             #
+# nuggets -- Extending Ruby                                                   #
+#                                                                             #
+# Copyright (C) 2007-2011 Jens Wille                                          #
+#                                                                             #
+# Authors:                                                                    #
+#     Jens Wille <jens.wille@gmail.com>                                       #
+#                                                                             #
+# nuggets is free software; you can redistribute it and/or modify it under    #
+# the terms of the GNU Affero General Public License as published by the Free #
+# Software Foundation; either version 3 of the License, or (at your option)   #
+# any later version.                                                          #
+#                                                                             #
+# nuggets is distributed in the hope that it will be useful, but WITHOUT ANY  #
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS   #
+# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for     #
+# more details.                                                               #
+#                                                                             #
+# You should have received a copy of the GNU Affero General Public License    #
+# along with nuggets. If not, see <http://www.gnu.org/licenses/>.             #
+#                                                                             #
+###############################################################################
+#++
+
+module Nuggets
+  class Integer
+    module MapMixin
+
+  # call-seq:
+  #   int.map_positive => anInteger
+  #
+  # Creates a bijection from *Z* to *N*, i.e., it maps _int_ to a positive
+  # integer.
+  def map_positive
+    2 * abs + (self > 0 ? 0 : 1)
+  end
+
+    end
+  end
+end

data/lib/nuggets/integer/to_binary_s.rb

@@ -0,0 +1,38 @@
+#--
+###############################################################################
+#                                                                             #
+# nuggets -- Extending Ruby                                                   #
+#                                                                             #
+# Copyright (C) 2007-2011 Jens Wille                                          #
+#                                                                             #
+# Authors:                                                                    #
+#     Jens Wille <jens.wille@gmail.com>                                       #
+#                                                                             #
+# nuggets is free software; you can redistribute it and/or modify it under    #
+# the terms of the GNU Affero General Public License as published by the Free #
+# Software Foundation; either version 3 of the License, or (at your option)   #
+# any later version.                                                          #
+#                                                                             #
+# nuggets is distributed in the hope that it will be useful, but WITHOUT ANY  #
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS   #
+# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for     #
+# more details.                                                               #
+#                                                                             #
+# You should have received a copy of the GNU Affero General Public License    #
+# along with nuggets. If not, see <http://www.gnu.org/licenses/>.             #
+#                                                                             #
+###############################################################################
+#++
+
+class Integer
+
+  # call-seq:
+  #   int.to_binary_s => aString
+  #   int.to_binary_s(length) => aString
+  #
+  # Returns _int_ as binary number string; optionally zero-padded to +length+.
+  def to_binary_s(length = nil)
+    "%0#{length}d" % to_s(2)
+  end
+
+end

data/lib/nuggets/io/agrep.rb

@@ -0,0 +1,43 @@
+#--
+###############################################################################
+#                                                                             #
+# nuggets -- Extending Ruby                                                   #
+#                                                                             #
+# Copyright (C) 2007-2011 Jens Wille                                          #
+#                                                                             #
+# Authors:                                                                    #
+#     Jens Wille <jens.wille@gmail.com>                                       #
+#                                                                             #
+# nuggets is free software; you can redistribute it and/or modify it under    #
+# the terms of the GNU Affero General Public License as published by the Free #
+# Software Foundation; either version 3 of the License, or (at your option)   #
+# any later version.                                                          #
+#                                                                             #
+# nuggets is distributed in the hope that it will be useful, but WITHOUT ANY  #
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS   #
+# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for     #
+# more details.                                                               #
+#                                                                             #
+# You should have received a copy of the GNU Affero General Public License    #
+# along with nuggets. If not, see <http://www.gnu.org/licenses/>.             #
+#                                                                             #
+###############################################################################
+#++
+
+require 'nuggets/enumerable/agrep'
+
+class IO
+
+  # call-seq:
+  #   IO.agrep(fd, pattern[, distance]) -> anArray
+  #   IO.agrep(fd, pattern[, distance]) { |line| ... } -> io
+  #
+  def self.agrep(fd, pattern, distance = 0, &block)
+    open(fd) { |io| io.agrep(pattern, distance, &block) }
+  end
+
+end
+
+if $0 == __FILE__
+  puts ::File.agrep(__FILE__, /calls/, 2)
+end

data/lib/nuggets/io/interact.rb

@@ -0,0 +1,5 @@
+require 'nuggets/io/interact_mixin'
+
+class IO
+  extend Nuggets::IO::InteractMixin
+end

data/lib/nuggets/io/interact_mixin.rb

@@ -0,0 +1,159 @@
+#--
+###############################################################################
+#                                                                             #
+# nuggets -- Extending Ruby                                                   #
+#                                                                             #
+# Copyright (C) 2007-2011 Jens Wille                                          #
+#                                                                             #
+# Authors:                                                                    #
+#     Jens Wille <jens.wille@gmail.com>                                       #
+#                                                                             #
+# nuggets is free software; you can redistribute it and/or modify it under    #
+# the terms of the GNU Affero General Public License as published by the Free #
+# Software Foundation; either version 3 of the License, or (at your option)   #
+# any later version.                                                          #
+#                                                                             #
+# nuggets is distributed in the hope that it will be useful, but WITHOUT ANY  #
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS   #
+# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for     #
+# more details.                                                               #
+#                                                                             #
+# You should have received a copy of the GNU Affero General Public License    #
+# along with nuggets. If not, see <http://www.gnu.org/licenses/>.             #
+#                                                                             #
+###############################################################################
+#++
+
+module Nuggets
+  class IO
+    module InteractMixin
+
+  # call-seq:
+  #   IO.interact(input, output[, timeout[, maxlen]]) -> anArray | nil
+  #
+  # Interact with both ends of a pipe in a non-blocking manner.
+  #
+  # +input+ represents the sending end and is a mapping from the actual
+  # input IO (to send into the pipe) to the pipe's input handle (_stdin_).
+  # The input IO must support +read_nonblock+. If it's a StringIO it will
+  # be extended appropriately. If it's a String it will be converted to
+  # a StringIO.
+  #
+  # +output+ represents the receiving end and is a mapping from the pipe's
+  # output handle (_stdout_) to the designated output IO (to receive data
+  # from the pipe), and, optionally, from the pipe's error handle (_stderr_)
+  # to the designated error IO. The output and error IO must support <tt><<</tt>
+  # with a string argument. If either of them is a Proc it will be extended such
+  # that <tt><<</tt> delegates to +call+.
+  #
+  # +timeout+, if given, will be passed to IO::select and +nil+ is returned
+  # if the select call times out; in all other cases an empty array is returned.
+  #
+  # +maxlen+ is the chunk size for +read_nonblock+.
+  #
+  # Examples:
+  #
+  #   require 'open3'
+  #
+  #   # simply prints 'input string' on STDOUT, ignores +stderr+
+  #   Open3.popen3('cat') { |stdin, stdout, stderr|
+  #     IO.interact({ "input string\n" => stdin }, { stdout => STDOUT })
+  #   }
+  #
+  #   # prints lines you type in reverse order to a string
+  #   str = ''
+  #   Open3.popen3('tac') { |stdin, stdout, stderr|
+  #     IO.interact({ STDIN => stdin }, { stdout => str })
+  #   }
+  #   puts str
+  #
+  #   # prints the IP adresses from /etc/hosts on STDOUT and their lengths
+  #   # on STDERR
+  #   cmd = %q{ruby -ne 'i = $_.split.first or next; warn i.length; puts i'}
+  #   Open3.popen3(cmd) { |stdin, stdout, stderr|
+  #     File.open('/etc/hosts') { |f|
+  #       IO.interact({ f => stdin }, { stdout => STDOUT, stderr => STDERR })
+  #     }
+  #   }
+  def interact(input, output, timeout = nil, maxlen = 2 ** 16)
+    readers, writers = {}, {}
+
+    output.each { |key, val|
+      if val.is_a?(::Proc) && !val.respond_to?(:<<)
+        class << val; alias_method :<<, :call; end
+      end
+
+      readers[key] = val
+    }
+
+    input.each { |key, val|
+      if key.is_a?(::String)
+        require 'stringio'
+        key = ::StringIO.new(key)
+      end
+
+      unless key.respond_to?(:read_nonblock)
+        def key.read_nonblock(*args)
+          read(*args) or raise ::EOFError, 'end of string reached'
+        end
+      end
+
+      writers[val] = [key, '']
+    }
+
+    close = lambda { |*args|
+      container, item, read = args
+
+      container.delete(item)
+      read ? item.close_read : item.close_write
+    }
+
+    read = lambda { |*args|
+      reader, buffer, writer = args
+
+      container = writer ? writers : readers
+      buffer ||= container[reader]
+
+      begin
+        buffer << reader.read_nonblock(maxlen)
+      rescue ::Errno::EAGAIN
+      rescue ::EOFError
+        buffer.force_encoding(
+          reader.internal_encoding  ||
+          reader.external_encoding  ||
+          Encoding.default_internal ||
+          Encoding.default_external
+        ) if buffer.respond_to?(:force_encoding)
+
+        close[container, writer || reader, !writer]
+      end
+    }
+
+    until readers.empty? && writers.empty?
+      fhs = select(readers.keys, writers.keys, nil, timeout) or return
+
+      fhs[0].each { |reader| read[reader] }
+
+      fhs[1].each { |writer|
+        reader, buffer = writers[writer]
+        read[reader, buffer, writer] or next if buffer.empty?
+
+        begin
+          bytes = writer.write_nonblock(buffer)
+        rescue ::Errno::EPIPE
+          close[writers, writer]
+        end
+
+        if bytes
+          buffer.force_encoding('BINARY') if buffer.respond_to?(:force_encoding)
+          buffer.slice!(0, bytes)
+        end
+      }
+    end
+
+    []
+  end
+
+    end
+  end
+end

data/lib/nuggets/io/modes.rb

@@ -0,0 +1,121 @@
+#--
+###############################################################################
+#                                                                             #
+# nuggets -- Extending Ruby                                                   #
+#                                                                             #
+# Copyright (C) 2007-2011 Jens Wille                                          #
+#                                                                             #
+# Authors:                                                                    #
+#     Jens Wille <jens.wille@gmail.com>                                       #
+#                                                                             #
+# nuggets is free software; you can redistribute it and/or modify it under    #
+# the terms of the GNU Affero General Public License as published by the Free #
+# Software Foundation; either version 3 of the License, or (at your option)   #
+# any later version.                                                          #
+#                                                                             #
+# nuggets is distributed in the hope that it will be useful, but WITHOUT ANY  #
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS   #
+# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for     #
+# more details.                                                               #
+#                                                                             #
+# You should have received a copy of the GNU Affero General Public License    #
+# along with nuggets. If not, see <http://www.gnu.org/licenses/>.             #
+#                                                                             #
+###############################################################################
+#++
+
+class IO
+
+  # Inspired by Martin DeMello, [ruby-talk:307782] -- thx ;-)
+
+  class << self
+
+    alias_method :_nuggets_original_read, :read
+
+    # call-seq:
+    #   IO.read(name[, length[, offset]]]) => aString
+    #   IO.read(name[, binary]) { |io| ... } => anObject
+    #
+    # Opens +name+ with mode +r+. NOTE: With no associated block,
+    # acts like the original IO::read, not like IO::new.
+    def read(name, *args)
+      return _nuggets_original_read(name, *args) unless block_given?
+
+      case args.size
+        when 0
+          # ok
+        when 1
+          case binary = args.first
+            when true, false, nil
+              # ok
+            else
+              raise ::TypeError, "wrong argument type #{binary.class} (expected boolean)"
+          end
+        else
+          raise ::ArgumentError, "wrong number of arguments (#{args.size + 1} for 1-2)"
+      end
+
+      open_with_mode(name, 'r', binary, &::Proc.new)
+    end
+
+    # call-seq:
+    #   IO.write(name[, binary]) => anIO
+    #   IO.write(name[, binary]) { |io| ... } => anObject
+    #
+    # Opens +name+ with mode +w+.
+    def write(name, binary = false)
+      open_with_mode(name, 'w', binary, &block_given? ? ::Proc.new : nil)
+    end
+
+    # call-seq:
+    #   IO.append(name[, binary]) => anIO
+    #   IO.append(name[, binary]) { |io| ... } => anObject
+    #
+    # Opens +name+ with mode +a+.
+    def append(name, binary = false)
+      open_with_mode(name, 'a', binary, &block_given? ? ::Proc.new : nil)
+    end
+
+    # call-seq:
+    #   IO.read_write(name[, binary]) => anIO
+    #   IO.read_write(name[, binary]) { |io| ... } => anObject
+    #
+    # Opens +name+ with mode <tt>r+</tt>.
+    def read_write(name, binary = false)
+      open_with_mode(name, 'r+', binary, &block_given? ? ::Proc.new : nil)
+    end
+
+    # call-seq:
+    #   IO.write_read(name[, binary]) => anIO
+    #   IO.write_read(name[, binary]) { |io| ... } => anObject
+    #
+    # Opens +name+ with mode <tt>w+</tt>.
+    def write_read(name, binary = false)
+      open_with_mode(name, 'w+', binary, &block_given? ? ::Proc.new : nil)
+    end
+
+    # call-seq:
+    #   IO.append_read(name[, binary]) => anIO
+    #   IO.append_read(name[, binary]) { |io| ... } => anObject
+    #
+    # Opens +name+ with mode <tt>a+</tt>.
+    def append_read(name, binary = false)
+      open_with_mode(name, 'a+', binary, &block_given? ? ::Proc.new : nil)
+    end
+
+    private
+
+    # Just a helper to DRY things up.
+    def open_with_mode(name, mode, binary = false)
+      open(name, "#{mode}#{'b' if binary}", &block_given? ? ::Proc.new : nil)
+    end
+
+  end
+
+end
+
+if $0 == __FILE__
+  # ::File.read(__FILE__) { |f| ... }
+  # ::File.write(__FILE__) { |f| ... }
+  # ::File.append(__FILE__) { |f| ... }
+end