darkext 0.12.0

data/lib/darkext.rb

@@ -0,0 +1,13 @@
+$:.unshift(File.dirname(__FILE__)) unless
+  $:.include?(File.dirname(__FILE__)) || $:.include?(File.expand_path(File.dirname(__FILE__)))
+
+libs = Dir[File.join(File.join(File.dirname(__FILE__), 'darkext'), '*.rb')]
+
+# Don't load these by default
+other_libs = libs.reject! { |lib| lib.include?('sinatra') || lib.include?('sitemap') }
+
+libs.reject! { |lib| lib.include?('beagle') }.first
+libs.each { |lib| require lib }
+
+# autoload beagle
+autoload(:Beagle, 'darkext/beagle')

data/lib/darkext/array.rb

@@ -0,0 +1,52 @@
+require 'darkext/numeric'
+require 'darkext/symbol'
+
+class Array
+  # Rotates the array left by n elements
+  def rotate(n = 1)
+    return if self.size.zero?
+    n.times { self.push(self.shift) }
+  end
+
+  # Rotates the array right by n elements
+  def rotate_reverse(n = 1)
+    return if self.size.zero?
+    n.times { self.unshift(self.pop) }
+  end
+
+  # Sums the array
+  def sum
+    self.inject(&:+)
+  end
+
+  # Finds the product of the array
+  def product
+    self.inject(&:*)
+  end
+
+  # Collects the squares of each value in the array
+  def squares
+    self.map(&:square)
+  end
+
+  # Destructively collects the squares
+  def squares!
+    self.map!(&:square)
+  end
+
+  # Picks a random value from the array
+  def random
+    self[rand(self.size)]
+  end
+  alias :pick :random
+
+  # Randomizes the array
+  def randomize
+    self.sort_by { rand }
+  end
+
+  # Destructively randomizes
+  def randomize!
+    self.replace(self.randomize)
+  end
+end

data/lib/darkext/beagle.rb

@@ -0,0 +1,88 @@
+require 'darkext/fiber'
+require 'darkext/string'
+require 'darkext/symbol'
+
+"Beagle.query is not secure! Use with caution".printn
+
+module Beagle
+  class BeagleError < RuntimeError; end
+
+  class BeagleResultsHelper
+    def initialize(io)
+      @io = io
+      @fiber = Fiber.new do
+        while !@io.eof?
+          result = Hash.new
+          loop do
+            line = @io.gets
+            break if line.nil?
+            line.chomp!
+            break if line.empty? && 0 < result.keys.size
+            if !line.include?('=') || line.starts_with?(' Snip')
+              parts = line.split(':')
+              k = parts.shift.strip
+              v = parts.join(':').strip
+              result[k] = v unless k.empty? || v.empty?
+            elsif line.include?('=')
+              k,v = line.split('=')
+              k = k.split(':').last.strip
+              v = v.gsub("'",'').strip
+              result[k] = v unless k.empty? || v.empty?
+            end
+          end
+          Fiber.yield(result)
+        end
+        @io.close
+      end
+    end
+
+    def next
+      begin
+        return @fiber.resume
+      rescue FiberError
+        return nil
+      end
+    end
+  end
+
+  def self.home
+    ENV['BEAGLE_HOME']
+  end
+
+  def self.home=(home)
+    ENV['BEAGLE_HOME'] = home
+  end
+
+  def self.start
+    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
+    system('beagled --backend Files')
+  end
+
+  def self.stop
+    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
+    system('beagle-shutdown')
+  end
+
+  def self.query(query, max_hits = 100)
+    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
+    args = Array.new
+    args << 'beagle-query'
+    args << '--verbose'
+    args << '--max-hits'
+    args << max_hits.to_s
+    args << "\"#{query.gsub('"','\"')}\""
+    return BeagleResultsHelper.new(IO.popen(args.join(' ')))
+  end
+
+  def self.running?
+    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
+    s = status
+    return false if s.include?("Could not connect") || s.empty?
+    return true
+  end
+
+  def self.status
+    raise BeagleError, "Beagle.home (BEAGLE_HOME) not set!" if home.nil?
+    DarkIO.capture_output(:stderr => false) { system('beagle-info','--status') }.strip
+  end
+end

data/lib/darkext/boolean.rb

@@ -0,0 +1,17 @@
+class Object
+  def is_boolean?
+    self.is_a?(TrueClass) || self.is_a?(FalseClass)
+  end
+end
+
+class TrueClass
+  def intern
+    return :true
+  end
+end
+
+class FalseClass
+  def intern
+    return :false
+  end
+end

data/lib/darkext/fiber.rb

@@ -0,0 +1,48 @@
+# Poor Man's Fiber (API compatible Thread based Fiber implementation for Ruby 1.8)
+# (c) 2008 Aman Gupta (tmm1)
+
+unless defined? Fiber
+  require 'thread'
+
+  class FiberError < StandardError; end
+
+  class Fiber
+    def initialize
+      raise ArgumentError, 'new Fiber requires a block' unless block_given?
+
+      @yield = Queue.new
+      @resume = Queue.new
+
+      @thread = Thread.new{ @yield.push [ *yield(*@resume.pop) ] }
+      @thread.abort_on_exception = true
+      @thread[:fiber] = self
+    end
+    attr_reader :thread
+
+    def resume *args
+      raise FiberError, 'dead fiber called' unless @thread.alive?
+      @resume.push(args)
+      result = @yield.pop
+      result.size > 1 ? result : result.first
+    end
+
+    def yield *args
+      @yield.push(args)
+      result = @resume.pop
+      result.size > 1 ? result : result.first
+    end
+
+    def self.yield *args
+      raise FiberError, "can't yield from root fiber" unless fiber = Thread.current[:fiber]
+      fiber.yield(*args)
+    end
+
+    def self.current
+      Thread.current[:fiber] or raise FiberError, 'not inside a fiber'
+    end
+
+    def inspect
+      "#<#{self.class}:0x#{self.object_id.to_s(16)}>"
+    end
+  end
+end

data/lib/darkext/float.rb

@@ -0,0 +1,6 @@
+class Float
+  # Equals for floats with tolerance
+  def equals?(x, tolerance = 1e-6)
+    (self - x).abs < tolerance
+  end
+end

data/lib/darkext/hash.rb

@@ -0,0 +1,38 @@
+class Hash
+  def nested_find(*keys)
+    v = self
+    keys.each do |key|
+      v = v[key]
+      return nil if v.nil?
+    end
+    return v
+  end
+
+  # http://snippets.dzone.com/posts/show/4146
+  def deep_merge!(second)
+    # From: http://www.ruby-forum.com/topic/142809
+    # Author: Stefan Rusterholz
+    merger = proc { |key,v1,v2| Hash === v1 && Hash === v2 ? v1.merge(v2, &merger) : v2 }
+    self.merge!(second, &merger)
+  end
+
+  def method_missing(m,*a)
+    if m.to_s =~ /=$/
+      self[$`] = a[0]
+    elsif a.empty?
+      self[m]
+    else
+      raise NoMethodError, "#{ m}"
+    end
+  end
+
+  # http://noobkit.com/show/ruby/gems/development/merb/hash/symbolize_keys-21.html
+  def symbolize_keys!
+    each do |k,v|
+      sym = k.respond_to?(:to_sym) ? k.to_sym : k
+      self[sym] = Hash === v ? v.symbolize_keys! : v
+      delete(k) unless k == sym
+    end
+    self
+  end
+end

data/lib/darkext/integer.rb

@@ -0,0 +1,10 @@
+require 'darkext/array'
+
+class Integer
+  # Computes the factorial of the Integer
+  def fact
+    raise ArgumentError.new('Cannot compute factorial of negative number') if 0 > self
+    return 1 if self.zero?
+    return (1..self).to_a.product
+  end
+end

data/lib/darkext/io.rb

@@ -0,0 +1,37 @@
+require 'darkext/hash'
+
+module DarkIO
+  # Runs a block and captures the output it generates
+  def self.capture_output(opts = { }) # yield e
+    opts = { :stdout => true, :stderr => false }.merge(opts)
+    cout,cerr = opts[:stdout],opts[:stderr]
+
+    yield and return if !cout && !cerr
+
+    old_stdout = STDOUT.dup if cout
+    old_stderr = STDERR.dup if cerr
+
+    r_stdout,w_stdout = IO.pipe if cout
+    r_stderr,w_stderr = IO.pipe if cerr
+
+    STDOUT.reopen(w_stdout) if cout
+    STDERR.reopen(w_stderr) if cerr
+
+    begin
+      yield
+    ensure
+      STDOUT.reopen(old_stdout) if cout
+      STDERR.reopen(old_stderr) if cerr
+      w_stdout.close if cout
+      w_stderr.close if cerr
+    end
+
+    ret_stdout = r_stdout.read if cout
+    ret_stderr = r_stderr.read if cerr
+    r_stdout.close if cout
+    r_stderr.close if cerr
+    return ret_stdout,ret_stderr if cout && cerr
+    return ret_stdout if cout
+    return ret_stderr if cerr
+  end
+end

data/lib/darkext/net.rb

@@ -0,0 +1,28 @@
+require 'net/http'
+
+module Net
+  def self.download(url,limit = 10)
+    raise ArgumentError, 'HTTP redirect too deep' if limit == 0
+    #url = URI.parse(url)
+    #req = Net::HTTP::Get.new(url.path)
+    #req['User-Agent'] = "Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.0.7) Gecko/2009030422 Ubuntu/8.10 (intrepid) Firefox/3.0.7"
+    #resp = Net::HTTP.new(url.host, url.port).start { |http| http.request(req) }
+    resp = Net::HTTP.get_response(URI.parse(url))
+    case resp
+    when Net::HTTPSuccess     then resp
+    when Net::HTTPRedirection then download(resp['location'], limit - 1)
+    else resp.error!
+    end
+  end
+
+  def self.download_and_save(url,path = nil)
+    if path.nil?
+      path = File.expand_path(url.split('/').last)
+    else
+      path = File.expand_path(path)
+    end
+    raise ArgumentError.new('Save path is a directory') if File.directory?(path)
+    resp = download(url)
+    open(path,'w') { |file| file.write(resp.body) } if resp.is_a?(Net::HTTPSuccess)
+  end
+end

data/lib/darkext/numeric.rb

@@ -0,0 +1,35 @@
+require 'mathn'
+
+class Numeric
+  # Squares the number
+  def square
+    self * self
+  end
+
+  # Cube the number
+  def cube
+    self.square * self
+  end
+
+  # Finds the square root of the number
+  def sqrt
+    Math.sqrt(self)
+  end
+
+  # Do some other roots
+  def root(n = 2)
+    return self if 1 == n
+    return self.sqrt if 2 == n
+    self ** (1 / n.to_f)
+  end
+
+  # Finds the log base e of the number
+  def ln
+    Math::log(self)
+  end
+
+  # Finds the log base 10 of the number
+  def log
+    Math::log10(self)
+  end
+end

data/lib/darkext/object.rb

@@ -0,0 +1,11 @@
+class Object
+  def try(*args, &block)
+    return nil if self.nil?
+    block_given? ? yield(self) : self.__send__(args.shift, *args)
+  end
+
+  def tap
+    yield(self)
+    return self
+  end
+end

data/lib/darkext/sinatra.rb

@@ -0,0 +1,70 @@
+module Sinatra
+  module DarkHelpers
+    module HelperMethods
+      def view(view, options = {})
+        html = haml(view,options)
+        flash.clear
+        html
+      end
+
+      def partial(view, options = {})
+        haml(view,options.merge(:layout => false))
+      end
+
+      def css_link_tag(sheet, media = 'screen,projection')
+        partial("%link{ :type => 'text/css', :href => '#{options.slash ? '/' : ''}stylesheets/#{sheet}.css', :rel => 'stylesheet', :media => '#{media}' }")
+      end
+
+      def js_script_tag(script)
+        partial("%script{ :type => 'text/javascript', :src => '#{options.slash ? '/' : ''}javascripts/#{script}.js' }")
+      end
+
+      def js_tag(script)
+        @dsh_script = script
+        temp = ["%script{ :type => 'text/javascript' }",
+                "  //<![CDATA[",
+                "  = @dsh_script",
+                "  //]]>"].join("\n")
+        partial(temp)
+      end
+
+      def flash
+        session[:flash] = {} if session[:flash] && session[:flash].class != Hash
+        session[:flash] ||= {}
+      end
+
+      def host
+        "#{protocol}://#{request.env['HTTP_HOST']}"
+      end
+
+      def protocol
+        request.env['rack.url_scheme']
+      end
+
+      def server_name
+        request.env['SERVER_NAME']
+      end
+
+      def set_title(extra)
+        @title = "#{options.site_name} - #{extra}"
+      end
+
+      def base
+        host + options.site_base
+      end
+
+      def redirect_back
+        redirect(request.env['HTTP_REFERER'])
+      end
+    end
+
+    def self.registered(app)
+      app.helpers(HelperMethods)
+      app.disable(:slash)
+      app.set(:site_base, '/')
+      app.enable(:sessions)
+    end
+  end
+
+  register DarkHelpers
+end