epitools 0.5.1 → 0.5.2

data/Guardfile

@@ -0,0 +1,16 @@
+# A sample Guardfile
+# More info at https://github.com/guard/guard#readme
+
+#guard 'spork', :wait => 50 do
+#  watch('Gemfile')
+#  watch('Gemfile.lock')
+#  watch('spec/spec_helper.rb')
+#end
+
+guard :rspec, :version => 2, :cli => "--color", :bundler => false, :all_after_pass => false, :all_on_start => false, :keep_failed => false do
+#guard 'rspec', :version => 2 do
+  watch(%r{^spec/.+_spec\.rb$})
+  watch(%r{^lib/epitools/(.+)\.rb$})     { |m| "spec/#{m[1]}_spec.rb" }
+  watch('spec/spec_helper.rb')  { "spec" }
+end
+

data/VERSION

@@ -1 +1 @@
-0.5.1
+0.5.2

data/epitools.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "epitools"
-  s.version = "0.5.1"
+  s.version = "0.5.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["epitron"]
-  s.date = "2011-11-23"
+  s.date = "2012-04-05"
   s.description = "Miscellaneous utility libraries to make my life easier."
   s.email = "chris@ill-logic.com"
   s.extra_rdoc_files = [
@@ -19,6 +19,7 @@ Gem::Specification.new do |s|
   ]
   s.files = [
     ".document",
+    "Guardfile",
     "LICENSE",
     "README.rdoc",
     "Rakefile",
@@ -27,12 +28,19 @@ Gem::Specification.new do |s|
     "epitools.gemspec",
     "lib/epitools.rb",
     "lib/epitools/autoloads.rb",
-    "lib/epitools/basetypes.rb",
     "lib/epitools/browser.rb",
     "lib/epitools/browser/cache.rb",
     "lib/epitools/browser/mechanize_progressbar.rb",
     "lib/epitools/clitools.rb",
     "lib/epitools/colored.rb",
+    "lib/epitools/core_ext.rb",
+    "lib/epitools/core_ext/array.rb",
+    "lib/epitools/core_ext/enumerable.rb",
+    "lib/epitools/core_ext/hash.rb",
+    "lib/epitools/core_ext/numbers.rb",
+    "lib/epitools/core_ext/object.rb",
+    "lib/epitools/core_ext/string.rb",
+    "lib/epitools/core_ext/truthiness.rb",
     "lib/epitools/ezdb.rb",
     "lib/epitools/hexdump.rb",
     "lib/epitools/iter.rb",
@@ -49,16 +57,15 @@ Gem::Specification.new do |s|
     "lib/epitools/rails.rb",
     "lib/epitools/rash.rb",
     "lib/epitools/ratio.rb",
-    "lib/epitools/string_to_proc.rb",
     "lib/epitools/sys.rb",
     "lib/epitools/term.rb",
     "lib/epitools/trie.rb",
     "lib/epitools/zopen.rb",
     "spec/autoreq_spec.rb",
-    "spec/basetypes_spec.rb",
     "spec/browser_spec.rb",
     "spec/clitools_spec.rb",
     "spec/colored_spec.rb",
+    "spec/core_ext_spec.rb",
     "spec/ezdb_spec.rb",
     "spec/iter_spec.rb",
     "spec/lcs_spec.rb",
@@ -76,7 +83,7 @@ Gem::Specification.new do |s|
   s.homepage = "http://github.com/epitron/epitools"
   s.licenses = ["WTFPL"]
   s.require_paths = ["lib"]
-  s.rubygems_version = "1.8.10"
+  s.rubygems_version = "1.8.21"
   s.summary = "NOT UTILS... METILS!"
 
   if s.respond_to? :specification_version then

data/lib/epitools.rb

@@ -35,20 +35,63 @@ class Object
       Module.autoreqs[const] = path
     end
   end
+
+  #
+  # Remove an object, method, constant, etc.
+  #
+  def del(x)
+    case x
+      when String
+        del(x.to_sym)
+      when Class, Module
+        Object.send(:remove_const, x.name)
+      when Method
+        x.owner.send(:undef_method, x.name)
+      when Symbol
+        if Object.const_get(x)
+          Object.send(:remove_const, x)
+        elsif method(x)
+          undef_method x
+        end
+      else
+        raise "Error: don't know how to 'del #{x.inspect}'"
+    end
+  end
   
+  # The hidden singleton lurks behind everyone
+  def metaclass
+    class << self
+      self
+    end
+  end
+
+  def meta_eval &blk
+    metaclass.instance_eval &blk
+  end
+
+  # Adds methods to a metaclass
+  def meta_def name, &blk
+    meta_eval { define_method name, &blk }
+  end
+
+  # Defines an instance method within a class
+  def class_def name, &blk
+    class_eval { define_method name, &blk }
+  end
+
 end
 
 #
-# Patch 'Module#const_missing' to support 'autoreq'
+# Patch 'Module#const_missing' to support 'autoreq' (which can autoload gems)
 #
 class Module
 
-  @@autoreq_searching_for = nil
+  @@autoreq_is_searching_for = nil
   
   alias const_missing_without_autoreq const_missing
   
   def const_missing(const)
-    return if const == @@autoreq_searching_for
+    return if const == @@autoreq_is_searching_for
     
     if thing = autoreqs[const]
       case thing
@@ -61,7 +104,7 @@ class Module
       end
     end
     
-    @@autoreq_searching_for = const
+    @@autoreq_is_searching_for = const
     const_get(const) || const_missing_without_autoreq(const)
   end
   
@@ -87,8 +130,7 @@ end
 #
 %w[
   autoloads
-  basetypes 
-  string_to_proc
+  core_ext 
   zopen
   colored
   clitools

data/lib/epitools/autoloads.rb

@@ -4,7 +4,6 @@ autoload :URI,        'uri'
 autoload :CGI,        'cgi'
 autoload :Base64,     'base64'
 autoload :JSON,       'json'
-autoload :YAML,       'yaml'
 autoload :Zlib,       'zlib'
 autoload :FileUtils,  'fileutils'
 autoload :Tempfile,   'tempfile'
@@ -14,8 +13,18 @@ autoload :Curses,     'curses'
 autoload :DateTime,   'date'
 autoload :Date,       'date'
 autoload :Open3,      'open3'
+autoload :Timeout,    'timeout'
+autoload :Find,       'find'
+autoload :Benchmark,  'benchmark'
 #autoload :DelegateClass, 'delegate'
 
+# YAML is sometimes loaded improperly.
+if defined? YAML and not defined? YAML.parse
+  del YAML  # remove the existing module
+end
+
+autoload :YAML, 'yaml'
+
 if RUBY_VERSION["1.8.7"]
   autoload :Prime,      'mathn'
 else
@@ -41,5 +50,7 @@ autoload :MimeMagic,    'epitools/mimemagic'
 autoload :Term,         'epitools/term'
 autoload :Iter,         'epitools/iter'
 
-## Gems
-autoreq  :ANSI,         'ansi' 
+## Gems (common)
+autoreq  :Nokogiri,     'nokogiri'
+autoreq  :ANSI,         'ansi'
+autoreq  :BSON,         'bson'

data/lib/epitools/browser.rb

@@ -140,6 +140,8 @@ class Browser
     puts "[ GET #{url} (using cache: #{use_cache}) ]"
     
     delay unless cached_already
+    max_retries = 4
+    retries = 0
 
     begin
       
@@ -154,9 +156,14 @@ class Browser
       puts
 
     rescue Net::HTTPBadResponse, Errno::ECONNRESET, SocketError, Timeout::Error, SOCKSError => e
+      raise if e.message == "getaddrinfo: Name or service not known"
+
+      retries += 1
+      return if retries >= max_retries
+
       puts "  |_ ERROR: #{e.inspect} -- retrying"
       delay(5)
-      retry
+      retry 
       
 =begin      
     rescue Mechanize::ResponseCodeError => e

data/lib/epitools/core_ext.rb

@@ -0,0 +1,207 @@
+require 'epitools'
+
+## Alias "Enumerator" to "Enum"
+
+if RUBY_VERSION["1.8"]
+  require 'enumerator' 
+  Enumerator = Enumerable::Enumerator unless defined? Enumerator
+end
+
+unless defined? Enum
+  if defined? Enumerator
+    Enum = Enumerator 
+  else
+    $stderr.puts "WARNING: Couldn't find the Enumerator class. Enum will not be available."
+  end
+end
+
+RbConfig = Config unless defined? RbConfig
+
+
+class Object
+  #
+  # Slightly gross hack to add a class method.
+  #
+  def self.alias_class_method(dest, src)
+    metaclass.send(:alias_method, dest, src)
+  end
+  
+end
+
+require 'epitools/core_ext/object'
+require 'epitools/core_ext/string'
+require 'epitools/core_ext/array'
+require 'epitools/core_ext/enumerable'
+require 'epitools/core_ext/hash'
+require 'epitools/core_ext/numbers'
+require 'epitools/core_ext/truthiness'
+
+
+class MatchData
+
+  #
+  # Return a hash of named matches
+  #
+  def to_hash
+    Hash[ names.zip(captures) ]
+  end
+  
+end
+
+
+class Binding
+
+  def [](key)
+    eval(key.to_s)
+  end
+  
+  def []=(key, val)
+    Thread.current[:_alter_binding_local_] = val
+    eval("#{key} = Thread.current[:_alter_binding_local_]")
+    Thread.current[:_alter_binding_local_] = nil
+  end
+
+  def local_variables
+    eval("local_variables")
+  end
+  alias_method :keys, :local_variables
+
+end
+
+
+class Proc
+
+  #
+  # Joins two procs together, returning a new proc.
+  #
+  # Example:
+  #   newproc = proc { 1 } & proc { 2 }
+  #   newproc.call #=> [1, 2]
+  #
+  def join(other=nil, &block)
+    other ||= block
+    proc { |*args| [self.call(*args), other.call(*args)] }
+  end
+  alias_method :&, :join
+  
+  #
+  # Chains two procs together, returning a new proc. The output from each proc is passed into
+  # the input of the next one.
+  #
+  # Example:
+  #   chain = proc { 1 } | proc { |input| input + 1 }
+  #   chain.call #=> 2
+  #
+  def chain(other=nil, &block)
+    other ||= block
+    proc { |*args| other.call( self.call(*args) ) }
+  end
+  alias_method :|, :chain 
+  
+end
+
+
+unless defined?(BasicObject)
+  #
+  # A BasicObject class for Ruby 1.8  
+  #
+  class BasicObject
+    instance_methods.each { |m| undef_method m unless m =~ /^__/ }
+  end
+end
+
+
+
+class NotWrapper < BasicObject # :nodoc:
+  def initialize(orig)
+    @orig = orig
+  end
+  
+  def inspect
+    "{NOT #{@orig.inspect}}"
+  end
+  
+  def method_missing(meth, *args, &block)
+    result = @orig.send(meth, *args, &block)
+    if result.is_a? ::TrueClass or result.is_a? ::FalseClass
+      !result
+    else
+      raise "Sorry, I don't know how to invert #{result.inspect}"
+    end
+  end
+end
+
+class Object
+  
+  #
+  # Negates a boolean, chained-method style.
+  #
+  # Example:
+  #   >> 10.even?
+  #   => true
+  #   >> 10.not.even?
+  #   => false
+  #
+  def not
+    NotWrapper.new(self)
+  end
+  
+end
+
+unless IO.respond_to? :copy_stream
+  
+  class IO
+    
+    def self.copy_stream(input, output)
+      while chunk = input.read(8192)
+        output.write(chunk)
+      end
+    end
+    
+  end
+  
+end
+
+
+class Range
+
+  #
+  # Pick a random number from the range.
+  #
+  def rand
+    magnitude = last-first
+    magnitude += 1 if not exclude_end?
+    Kernel.rand(magnitude)+first
+  end
+
+end
+
+
+class Struct
+
+  #
+  # Transform this struct into a JSON hash
+  #
+  def to_hash
+    hash = {}
+    each_pair { |k,v| hash[k] = v }
+    hash
+  end
+
+  #
+  # Transform the struct into a simple JSON hash.
+  #
+  def to_json(*args)
+    to_hash.to_json
+  end
+
+end
+
+
+module URI
+
+  def params
+    query.to_params
+  end
+
+end

data/lib/epitools/core_ext/array.rb

@@ -0,0 +1,98 @@
+
+class Array
+
+  #
+  # flatten.compact.uniq
+  #
+  def squash
+    flatten.compact.uniq
+  end
+  
+  #
+  # Removes the elements from the array for which the block evaluates to true. 
+  # In addition, return the removed elements.
+  #
+  # For example, if you wanted to split an array into evens and odds:
+  #
+  #   nums = [1,2,3,4,5,6,7,8,9,10,11,12]
+  #   even = nums.remove_if { |n| n.even? }   # remove all even numbers from the "nums" array and return them
+  #   odd = nums                              # "nums" now only contains odd numbers
+  #
+  def remove_if(&block)
+    removed = []
+    
+    delete_if do |x|
+      if block.call(x)
+        removed << x
+        true
+      else
+        false
+      end
+    end
+    
+    removed
+  end
+  
+  #
+  # zip from the right (or reversed zip.)
+  #
+  # eg:
+  #   >> [5,39].rzip([:hours, :mins, :secs]) 
+  #   => [ [:mins, 5], [:secs, 39] ]
+  #
+  def rzip(other)
+    # That's a lotta reverses!
+    reverse.zip(other.reverse).reverse
+  end  
+  
+  #
+  # Pick the middle element.
+  #
+  def middle
+    self[(size-1) / 2]
+  end
+  
+  #
+  # XOR operator
+  #
+  def ^(other)
+    (self | other) - (self & other)
+  end
+
+  #
+  # Shuffle the array
+  #
+  unless defined? shuffle
+    def shuffle
+      sort_by{rand}
+    end
+  end
+  
+  #
+  # Pick (a) random element(s).
+  #
+  unless defined? sample
+    def sample(n=1)
+      if n == 1
+        self[rand(size)]
+      else
+        shuffle[0...n]
+      end
+    end
+  end
+  alias_method :pick, :sample
+  
+  #
+  # Divide the array into n pieces.
+  #
+  def / pieces 
+    piece_size = (size.to_f / pieces).ceil
+    each_slice(piece_size).to_a
+  end
+  
+
+  alias_method :unzip, :transpose
+
+end
+
+