chimera 0.0.1

data/lib/typhoeus/request.rb

@@ -0,0 +1,124 @@
+module Typhoeus
+  class Request
+    attr_accessor :method, :params, :body, :headers, :timeout, :user_agent, :response, :cache_timeout, :follow_location, :max_redirects, :proxy, :disable_ssl_peer_verification
+    attr_reader   :url
+
+    def initialize(url, options = {})
+      @method           = options[:method] || :get
+      @params           = options[:params]
+      @body             = options[:body]
+      @timeout          = options[:timeout]
+      @headers          = options[:headers] || {}
+      @user_agent       = options[:user_agent] || Typhoeus::USER_AGENT
+      @cache_timeout    = options[:cache_timeout]
+      @follow_location  = options[:follow_location]
+      @max_redirects    = options[:max_redirects]
+      @proxy            = options[:proxy]
+      @disable_ssl_peer_verification = options[:disable_ssl_peer_verification]
+
+      if @method == :post
+        @url = url
+      else
+        @url = @params ? "#{url}?#{params_string}" : url
+      end
+      @on_complete      = nil
+      @after_complete   = nil
+      @handled_response = nil
+    end
+
+    def host
+      slash_location = @url.index('/', 8)
+      if slash_location
+        @url.slice(0, slash_location)
+      else
+        query_string_location = @url.index('?')
+        return query_string_location ? @url.slice(0, query_string_location) : @url
+      end
+    end
+
+    def headers
+      @headers["User-Agent"] = @user_agent
+      @headers
+    end
+
+    def params_string
+      params.keys.sort.collect do |k|
+        value = params[k]
+        if value.is_a? Hash
+          value.keys.collect {|sk| Rack::Utils.escape("#{k}[#{sk}]") + "=" + Rack::Utils.escape(value[sk].to_s)}
+        elsif value.is_a? Array
+          key = Rack::Utils.escape(k.to_s)
+          value.collect { |v| "#{key}=#{Rack::Utils.escape(v.to_s)}" }.join('&')
+        else
+          "#{Rack::Utils.escape(k.to_s)}=#{Rack::Utils.escape(params[k].to_s)}"
+        end
+      end.flatten.join("&")
+    end
+
+    def on_complete(&block)
+      @on_complete = block
+    end
+
+    def on_complete=(proc)
+      @on_complete = proc
+    end
+
+    def after_complete(&block)
+      @after_complete = block
+    end
+
+    def after_complete=(proc)
+      @after_complete = proc
+    end
+
+    def call_handlers
+      if @on_complete
+        @handled_response = @on_complete.call(response)
+        call_after_complete
+      end
+    end
+
+    def call_after_complete
+       @after_complete.call(@handled_response) if @after_complete
+    end
+
+    def handled_response=(val)
+      @handled_response = val
+    end
+
+    def handled_response
+      @handled_response || response
+    end
+
+    def cache_key
+      Digest::SHA1.hexdigest(url)
+    end
+
+    def self.run(url, params)
+      r = new(url, params)
+      Typhoeus::Hydra.hydra.queue r
+      Typhoeus::Hydra.hydra.run
+      r.response
+    end
+
+    def self.get(url, params = {})
+      run(url, params.merge(:method => :get))
+    end
+
+    def self.post(url, params = {})
+      run(url, params.merge(:method => :post))
+    end
+
+    def self.put(url, params = {})
+      run(url, params.merge(:method => :put))
+    end
+
+    def self.delete(url, params = {})
+      run(url, params.merge(:method => :delete))
+    end
+
+    def self.head(url, params = {})
+      run(url, params.merge(:method => :head))
+    end
+  end
+end

data/lib/typhoeus/response.rb

@@ -0,0 +1,39 @@
+module Typhoeus
+  class Response
+    attr_accessor :request
+    attr_reader :code, :headers, :body, :time,
+                :requested_url, :requested_remote_method,
+                :requested_http_method, :start_time
+    
+    def initialize(params = {})
+      @code                  = params[:code]
+      @headers               = params[:headers]
+      @body                  = params[:body]
+      @time                  = params[:time]
+      @requested_url         = params[:requested_url]
+      @requested_http_method = params[:requested_http_method]
+      @start_time            = params[:start_time]
+      @request               = params[:request]
+    end
+    
+    def headers_hash
+      headers.split("\n").map {|o| o.strip}.inject({}) do |hash, o|
+        if o.empty?
+          hash
+        else
+          i = o.index(":") || o.size
+          key = o.slice(0, i)
+          value = o.slice(i + 1, o.size)
+          value = value.strip unless value.nil?
+          if hash.has_key? key
+            hash[key] = [hash[key], value].flatten
+          else
+            hash[key] = value
+          end
+          
+          hash
+        end
+      end
+    end
+  end
+end

data/lib/typhoeus/service.rb

@@ -0,0 +1,20 @@
+module Typhoeus
+  class Service
+    def initialize(host, port)
+      @host = host
+      @port = port
+    end
+    
+    def get(resource, params)
+    end
+    
+    def put(resource, params)
+    end
+    
+    def post(resource, params)
+    end
+    
+    def delete(resource, params)
+    end
+  end
+end

data/script/console

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+# File: script/console
+irb = RUBY_PLATFORM =~ /(:?mswin|mingw)/ ? 'irb.bat' : 'irb'
+
+libs =  " -r irb/completion"
+# Perhaps use a console_lib to store any extra methods I may want available in the cosole
+# libs << " -r #{File.dirname(__FILE__) + '/../lib/console_lib/console_logger.rb'}"
+libs <<  " -r #{File.dirname(__FILE__) + '/../lib/chimera.rb'}"
+puts "Loading chimera gem"
+exec "#{irb} #{libs} --simple-prompt"

data/script/destroy

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..'))
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/destroy'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Destroy.new.run(ARGV)

data/script/generate

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+APP_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..'))
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/scripts/generate'
+
+ARGV.shift if ['--help', '-h'].include?(ARGV[0])
+RubiGen::Base.use_component_sources! [:rubygems, :newgem, :newgem_theme, :test_unit]
+RubiGen::Scripts::Generate.new.run(ARGV)

data/test/models.rb

@@ -0,0 +1,49 @@
+class User < Chimera::Base
+  use_config :default # this is implied even if not here
+  
+  attribute :name
+  attribute :age
+  attribute :occupation
+  attribute :interests
+  attribute :home_coordinate # [37.2,122.1]
+  attribute :ssn
+  attribute :updated_at
+  attribute :favorite_car, :model, :class => :car
+  
+  # User.find_with_index(:home_coordinate, {:coordinate => [37.2,122.1], :steps => 5})
+  index :home_coordinate, :type => :geo, :step_size => 0.05
+  
+  # User.find_with_index(:occupation, { :q => "developer", :type => :intersect } ) # fuzzy search. :intersect or :union
+  index :occupation, :type => :search
+  
+  # User.find_with_index(:ssn, "12345") # exact search, enforces unique constraint
+  index :ssn, :type => :unique
+  
+  # User.find_with_index(:name, "Ben") # like :search but exact
+  index :name, :type => :find
+  
+  association :friends, :user
+  association :cars, :car
+  
+  redis_object :num_logins, :counter
+  
+  validates_presence_of :name
+end
+
+class Car < Chimera::Base
+  attribute :color
+  attribute :make
+  attribute :model
+  attribute :year
+  attribute :mileage
+  attribute :comments
+  attribute :sku
+  attribute :curr_location
+  
+  index :year, :type => :find
+  index :comments, :type => :search
+  index :sku, :type => :unique
+  index :curr_location, :type => :geo, :step_size => 0.05
+  
+  validates_presence_of :make, :model, :year
+end

data/test/test_chimera.rb

@@ -0,0 +1,238 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestChimera < Test::Unit::TestCase
+  def setup
+    Car.each { |c| c.destroy }
+    Car.connection(:redis).flush_all
+  end
+  
+  def test_geo_indexes
+    c = Car.new
+    c.make = "Porsche"
+    c.model = "911"
+    c.year = 2010
+    c.sku = 1000
+    c.id = Car.new_uuid
+    c.curr_location = [37.12122, 121.43392]
+    assert c.save
+    
+    c2 = Car.new
+    c2.make = "Toyota"
+    c2.model = "Hilux"
+    c2.year = 2010
+    c2.sku = 1001
+    c2.curr_location = [37.12222, 121.43792]
+    c2.id = Car.new_uuid
+    assert c2.save
+    
+    found = Car.find_with_index(:curr_location, {:coordinate => [37.12222, 121.43792], :steps => 5})
+    assert_equal [c,c2].sort, found.sort
+    
+    c2.curr_location = [38.0, 122.0]
+    assert c2.save
+    
+    found = Car.find_with_index(:curr_location, {:coordinate => [37.12222, 121.43792], :steps => 5})
+    assert_equal [c].sort, found.sort
+    
+    found = Car.find_with_index(:curr_location, {:coordinate => [38.0-0.05, 122.0+0.05], :steps => 5})
+    assert_equal [c2].sort, found.sort
+  end
+  
+  def test_search_indexes
+    c = Car.new
+    c.make = "Porsche"
+    c.model = "911"
+    c.year = 2010
+    c.sku = 1000
+    c.comments = "cat dog chicken dolphin whale panther"
+    c.id = Car.new_uuid
+    assert c.save
+    
+    c2 = Car.new
+    c2.make = "Porsche"
+    c2.model = "911"
+    c2.year = 2010
+    c2.sku = 1001
+    c2.comments = "cat dog chicken"
+    c2.id = Car.new_uuid
+    assert c2.save
+
+    c3 = Car.new
+    c3.make = "Porsche"
+    c3.model = "911"
+    c3.year = 2010
+    c3.sku = 1002
+    c3.comments = "dog chicken dolphin whale"
+    c3.id = Car.new_uuid
+    assert c3.save
+    
+    assert_equal [c,c2,c3].sort, Car.find_with_index(:comments, "dog").sort
+    assert_equal [c,c2].sort, Car.find_with_index(:comments, "cat").sort
+    assert_equal [c,c2].sort, Car.find_with_index(:comments, "cat").sort
+
+    assert_equal [c,c2,c3].sort, Car.find_with_index(:comments, {:q => "dog dolphin", :type => :union}).sort
+    assert_equal [c,c3].sort, Car.find_with_index(:comments, {:q => "dog dolphin", :type => :intersect}).sort
+  end
+  
+  def test_indexes
+    c = Car.new
+    c.make = "Nissan"
+    c.model = "RX7"
+    c.year = 2010
+    c.sku = 1001
+    c.comments = "really fast car. it's purple too!"
+    c.id = Car.new_uuid
+    assert c.save
+    
+    assert !c.new?
+    
+    assert_equal [c], Car.find_with_index(:comments, "fast")
+    assert_equal [c], Car.find_with_index(:comments, "purple")
+    assert_equal [], Car.find_with_index(:comments, "blue")
+    
+    assert_equal [c], Car.find_with_index(:year, 2010)
+    assert_equal [c], Car.find_with_index(:sku, 1001)
+    
+    c2 = Car.new
+    c2.make = "Honda"
+    c2.model = "Accord"
+    c2.year = 2010
+    c2.sku = 1001
+    c2.id = Car.new_uuid
+    assert_raise(Chimera::Error::UniqueConstraintViolation) { c2.save }
+    c2.sku = 1002
+    assert c2.save
+    
+    c3 = Car.new
+    c3.make = "Honda"
+    c3.model = "Civic"
+    c3.year = 2010
+    c3.sku = 1003
+    c3.id = Car.new_uuid
+    assert c3.save
+    
+    assert_equal 3, Car.find_with_index(:year, 2010).size
+    assert Car.find_with_index(:year, 2010).include?(c)
+    assert Car.find_with_index(:year, 2010).include?(c2)
+    assert Car.find_with_index(:year, 2010).include?(c3)
+    
+    count = 0
+    Car.find_with_index(:all) { |car| count += 1 }
+    assert_equal 3, count
+    
+    count = 0
+    Car.each { |car| count += 1 }
+    assert_equal 3, count
+    
+    c2.destroy
+    
+    count = 0
+    Car.find_with_index(:all) { |car| count += 1 }
+    assert_equal 2, count
+    
+    count = 0
+    Car.each { |car| count += 1 }
+    assert_equal 2, count
+  end
+  
+  def test_associations
+    u = User.new
+    u.id = User.new_uuid
+    u.name = "Ben"
+    assert u.save
+    
+    assert_equal 0, u.friends.size
+    
+    chris = User.new
+    chris.id = User.new_uuid
+    chris.name = "Chris"
+    assert chris.save
+    
+    assert_equal 0, u.friends.size
+    u.friends << chris
+    assert_equal 1, u.friends.size
+    chris.destroy
+    assert_equal 0, u.friends.size
+    
+    c = Car.new
+    c.make = "Nissan"
+    c.model = "RX7"
+    c.year = 2010
+    c.sku = 1001
+    c.comments = "really fast car. it's purple too!"
+    c.id = Car.new_uuid
+    assert c.save
+    
+    assert_equal 0, u.cars.size
+    u.cars << c
+    assert_equal 1, u.cars.size
+    assert_equal [c], u.cars.all
+    assert_equal 1, c.association_memberships.all_associations.size
+    u.cars.remove(c)
+    assert_equal 0, c.association_memberships.all_associations.size
+  end
+  
+  def test_model_attribute
+    u = User.new
+    u.id = User.new_uuid
+    u.name = "Ben"
+    assert u.save
+    assert_nil u.favorite_car
+    
+    c = Car.new
+    c.make = "Nissan"
+    c.model = "RX7"
+    c.year = 2010
+    c.sku = 1001
+    c.comments = "really fast car. it's purple too!"
+    c.id = Car.new_uuid
+    assert c.save
+    
+    u.favorite_car = c
+    assert u.save
+    assert_equal c, u.favorite_car
+    u = User.find(u.id)
+    assert_equal c, u.favorite_car
+    u.favorite_car = nil
+    assert u.save
+    assert_nil u.favorite_car
+    
+    u.favorite_car = c
+    assert u.save
+    assert_equal c, u.favorite_car
+    c.destroy
+    assert_equal c, u.favorite_car
+    u = User.find(u.id)
+    assert_nil u.favorite_car
+  end
+  
+  def test_redis_objects
+    u = User.new
+    u.id = User.new_uuid
+    u.name = "Ben"
+    assert u.save
+    
+    assert_equal false, User.connection(:redis).exists(u.num_logins.key)
+    assert_equal 0, u.num_logins.count
+    u.num_logins.incr
+    assert_equal 1, u.num_logins.count
+    assert_equal true, User.connection(:redis).exists(u.num_logins.key)
+    u.num_logins.incr_by 10
+    assert_equal 11, u.num_logins.count
+    
+    u.destroy
+    
+    assert_equal false, User.connection(:redis).exists(u.num_logins.key)
+  end
+  
+  def test_rich_attributes
+    u = User.new
+    u.id = User.new_uuid
+    u.updated_at = Time.now.utc
+    assert u.updated_at.is_a?(Time)
+    u.name = "ben"
+    assert u.save
+    u = User.find(u.id)
+    assert u.updated_at.is_a?(Time)
+  end
+end