elasticsearch-ruby 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in elastic_search.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1,11 @@
+require 'bundler/gem_tasks'
+
+require 'rake'
+require 'rake/testtask'
+
+Rake::TestTask.new do |t|
+  t.pattern = 'spec/**/*_spec.rb'
+  t.libs << 'lib' << 'spec'
+end
+
+task :default => :test

data/elastic_search.gemspec

@@ -0,0 +1,29 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "elastic_search/version"
+
+Gem::Specification.new do |s|
+  s.name        = "elasticsearch-ruby"
+  s.version     = ElasticSearch::VERSION
+  s.authors     = ["Edmund Salvacion"]
+  s.email       = ["edmund@edmundatwork.com"]
+  s.homepage    = ""
+  s.summary     = %q{Ruby Driver for Elastic Search}
+  s.description = %q{Ruby Driver for Elastic Search}
+
+  s.rubyforge_project = "elasticsearch"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # = Library dependencies
+  #
+  s.add_dependency 'thrift'
+  s.add_dependency 'faraday'
+  s.add_dependency 'activesupport', ">= 3.0.0"
+  s.add_development_dependency "rake"
+  s.add_development_dependency "minitest"
+  s.add_development_dependency "turn"
+end

data/lib/elastic_search/client.rb

@@ -0,0 +1,44 @@
+module ElasticSearch
+  class Client
+    def initialize(transport)
+      @transport = transport
+      @indexes = {}
+    end
+
+    def get_index(index_name)
+      @indexes[index_name.to_sym] ||= ElasticSearch::Index.new(index_name, self)
+    end
+    alias :[] :get_index
+
+    def create_index(index_name, options = {})
+      body = { mappings: options[:mappings], settings: options[:settings] }.reject { |_, v| v.nil? }
+      request = Request.new({ method: :post, index: index_name, body: body })
+      status = self.execute(request).status
+      get_index(index_name) if status.eql?(200) || status.eql?(201)
+    end
+
+    def delete_index(index_name)
+      request = Request.new({ method: :delete, index: index_name })
+      execute(request)
+    end
+
+    def has_index?(index_name)
+      request = Request.new({ method: :head, index: index_name })
+      begin
+        execute(request).status.between?(200, 206)
+      rescue ElasticSearch::ResponseError
+        return false
+      end
+    end
+
+    def bulk(objects)
+      body = objects.flatten.join("\n") + "\n"
+      request = Request.new({ method: :post, action: :bulk, body: body })
+      execute(request)
+    end
+
+    def execute(request)
+      @transport.execute(request)
+    end
+  end
+end

data/lib/elastic_search/index.rb

@@ -0,0 +1,75 @@
+module ElasticSearch
+  class Index
+    attr_reader :index_name, :client
+
+    def initialize(index_name, client)
+      @index_name = index_name
+      @client = client
+      @types = {}
+    end
+
+    def get_type(type_name)
+      @types[type_name.to_sym] ||= ElasticSearch::Type.new(type_name, self)
+    end
+    alias :[] :get_type
+
+    def search(body, parameters)
+      request = build_request(method: :get, action: :search, parameters: parameters, body: body || {})
+      response = execute(request)
+      ElasticSearch::Search::Results.new(response)
+    end
+
+    def refresh
+      request = build_request(method: :post, action: :refresh)
+      execute(request)
+    end
+
+    def open
+      request = build_request(method: :post, action: :open)
+      execute(request)
+    end
+
+    def close
+      request = build_request(method: :post, action: :close)
+      execute(request)
+    end
+
+    def settings
+      request = build_request(method: :get, action: :settings)
+      execute(request).body
+    end
+
+    def settings=(settings)
+      request = build_request(method: :put, action: :settings, body: settings)
+      execute(request)
+    end
+
+    def mapping
+      request = build_request(method: :get, action: :mapping)
+      execute(request).body
+    end
+
+    def mapping=(mapping)
+      request = build_request(method: :put, action: :mapping, body: mapping)
+      execute(request).status.eql?(200)
+    end
+
+    def optimize(options = {})
+      request = build_request(method: :post, action: :optimize, parameters: options)
+      execute(request).status.eql?(200)
+    end
+
+    def exists?
+      @client.has_index?(@index_name)
+    end
+
+    def execute(request)
+      @client.execute(request)
+    end
+
+    private
+    def build_request(options = {})
+      Request.new({ index: @index_name }.merge(options))
+    end
+  end
+end

data/lib/elastic_search/item.rb

@@ -0,0 +1,14 @@
+module ElasticSearch
+  class Item < HashWithIndifferentAccess
+    attr_reader :index, :type, :item, :version, :exists
+
+    def initialize(request)
+      @index = request.body['_index']
+      @type = request.body['_type']
+      @item = request.body['_id']
+      @version = request.body['_version']
+      @exists = request.body['exists']
+      self.replace(request.body['_source'])
+    end
+  end
+end

data/lib/elastic_search/request.rb

@@ -0,0 +1,33 @@
+module ElasticSearch
+  class Request
+    attr_reader :method, :index, :type, :id, :action
+
+    def initialize(options = {})
+      @method = options[:method] ? options[:method].to_sym : :get
+      @index = options[:index]
+      @type = options[:type]
+      @id = options[:id]
+      @action = options[:action]
+      @parameters = options[:parameters]
+      @body = options[:body]
+    end
+
+    def parameters
+      @stringified_parameters ||= @parameters.inject({}) do |options, (key, value)|
+        options[key.to_s] = value
+        options
+      end if @parameters
+    end
+
+    def body
+      @encoded_body ||= @body.to_json
+    end
+
+    def path
+      components = [@index, @type]
+      components << @id if @id
+      components << "_#{@action.to_s}" if @action
+      components.compact.join("/")
+    end
+  end
+end

data/lib/elastic_search/response.rb

@@ -0,0 +1,38 @@
+module ElasticSearch
+  class Response
+    attr_reader :status
+
+    def initialize(options)
+      @body = options[:body].eql?("") ? "{}" : options[:body]
+      @status = options[:status]
+
+      case @status
+      when 200..206
+      when 404
+        if self.body['error'] =~ /^IndexMissingException/ || self.body.eql?(nil)
+          raise ElasticSearch::IndexMissingException, "#{self.body['status']}: #{self.body['error']}"
+        elsif self.body['exists'].eql?(false)
+          raise ElasticSearch::ItemMissingException, "_id: #{self.body['id']} NOT FOUND"
+        else
+          raise ElasticSearch::ResponseError
+        end
+      else
+        raise ElasticSearch::ResponseError, "#{self.body['status']}: #{self.body['error']}"
+      end
+    end
+
+    def body
+      @decoded_body ||= (@body ? ActiveSupport::JSON.decode(@body) : {})
+    end
+
+    private
+    def recursive_symbolize_keys!(hash)
+      hash.symbolize_keys!
+      hash.values.select{|v| v.is_a? Hash}.each{|h| recursive_symbolize_keys!(h)}
+    end
+  end
+
+  class ResponseError < StandardError; end
+  class IndexMissingException < StandardError; end
+  class ItemMissingException < StandardError; end
+end

data/lib/elastic_search/search/results.rb

@@ -0,0 +1,28 @@
+module ElasticSearch
+  module Search
+    class Results < Array
+      attr_reader :took, :timed_out, :shards, :total, :max_score
+
+      def initialize(response)
+        @took = response.body['took']
+        @timed_out = response.body['timed_out']
+        @shards = response.body['shards']
+        @total = response.body['hits']['total']
+        @max_score = response.body['hits']['max_score']
+        self.replace(response.body['hits']['hits'].collect { |h| ResultItem.new(h) })
+      end
+    end
+
+    class ResultItem < HashWithIndifferentAccess
+      attr_reader :index, :type, :id, :score
+
+      def initialize(hit)
+        @index = hit['_index']
+        @type = hit['_type']
+        @id = hit['_id']
+        @score = hit['_score']
+        self.replace(hit['_source'])
+      end
+    end
+  end
+end

data/lib/elastic_search/thrift/elastic_search_constants.rb

@@ -0,0 +1,10 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+module ElasticSearch
+  module Thrift
+  end
+end

data/lib/elastic_search/thrift/elastic_search_types.rb

@@ -0,0 +1,123 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+module ElasticSearch
+  module Thrift
+    module Method
+      GET = 0
+      PUT = 1
+      POST = 2
+      DELETE = 3
+      HEAD = 4
+      OPTIONS = 5
+      VALUE_MAP = {0 => "GET", 1 => "PUT", 2 => "POST", 3 => "DELETE", 4 => "HEAD", 5 => "OPTIONS"}
+      VALID_VALUES = Set.new([GET, PUT, POST, DELETE, HEAD, OPTIONS]).freeze
+    end
+
+    module Status
+      CONT = 100
+      SWITCHING_PROTOCOLS = 101
+      OK = 200
+      CREATED = 201
+      ACCEPTED = 202
+      NON_AUTHORITATIVE_INFORMATION = 203
+      NO_CONTENT = 204
+      RESET_CONTENT = 205
+      PARTIAL_CONTENT = 206
+      MULTI_STATUS = 207
+      MULTIPLE_CHOICES = 300
+      MOVED_PERMANENTLY = 301
+      FOUND = 302
+      SEE_OTHER = 303
+      NOT_MODIFIED = 304
+      USE_PROXY = 305
+      TEMPORARY_REDIRECT = 307
+      BAD_REQUEST = 400
+      UNAUTHORIZED = 401
+      PAYMENT_REQUIRED = 402
+      FORBIDDEN = 403
+      NOT_FOUND = 404
+      METHOD_NOT_ALLOWED = 405
+      NOT_ACCEPTABLE = 406
+      PROXY_AUTHENTICATION = 407
+      REQUEST_TIMEOUT = 408
+      CONFLICT = 409
+      GONE = 410
+      LENGTH_REQUIRED = 411
+      PRECONDITION_FAILED = 412
+      REQUEST_ENTITY_TOO_LARGE = 413
+      REQUEST_URI_TOO_LONG = 414
+      UNSUPPORTED_MEDIA_TYPE = 415
+      REQUESTED_RANGE_NOT_SATISFIED = 416
+      EXPECTATION_FAILED = 417
+      UNPROCESSABLE_ENTITY = 422
+      LOCKED = 423
+      FAILED_DEPENDENCY = 424
+      INTERNAL_SERVER_ERROR = 500
+      NOT_IMPLEMENTED = 501
+      BAD_GATEWAY = 502
+      SERVICE_UNAVAILABLE = 503
+      GATEWAY_TIMEOUT = 504
+      INSUFFICIENT_STORAGE = 506
+      VALUE_MAP = {100 => "CONT", 101 => "SWITCHING_PROTOCOLS", 200 => "OK", 201 => "CREATED", 202 => "ACCEPTED", 203 => "NON_AUTHORITATIVE_INFORMATION", 204 => "NO_CONTENT", 205 => "RESET_CONTENT", 206 => "PARTIAL_CONTENT", 207 => "MULTI_STATUS", 300 => "MULTIPLE_CHOICES", 301 => "MOVED_PERMANENTLY", 302 => "FOUND", 303 => "SEE_OTHER", 304 => "NOT_MODIFIED", 305 => "USE_PROXY", 307 => "TEMPORARY_REDIRECT", 400 => "BAD_REQUEST", 401 => "UNAUTHORIZED", 402 => "PAYMENT_REQUIRED", 403 => "FORBIDDEN", 404 => "NOT_FOUND", 405 => "METHOD_NOT_ALLOWED", 406 => "NOT_ACCEPTABLE", 407 => "PROXY_AUTHENTICATION", 408 => "REQUEST_TIMEOUT", 409 => "CONFLICT", 410 => "GONE", 411 => "LENGTH_REQUIRED", 412 => "PRECONDITION_FAILED", 413 => "REQUEST_ENTITY_TOO_LARGE", 414 => "REQUEST_URI_TOO_LONG", 415 => "UNSUPPORTED_MEDIA_TYPE", 416 => "REQUESTED_RANGE_NOT_SATISFIED", 417 => "EXPECTATION_FAILED", 422 => "UNPROCESSABLE_ENTITY", 423 => "LOCKED", 424 => "FAILED_DEPENDENCY", 500 => "INTERNAL_SERVER_ERROR", 501 => "NOT_IMPLEMENTED", 502 => "BAD_GATEWAY", 503 => "SERVICE_UNAVAILABLE", 504 => "GATEWAY_TIMEOUT", 506 => "INSUFFICIENT_STORAGE"}
+      VALID_VALUES = Set.new([CONT, SWITCHING_PROTOCOLS, OK, CREATED, ACCEPTED, NON_AUTHORITATIVE_INFORMATION, NO_CONTENT, RESET_CONTENT, PARTIAL_CONTENT, MULTI_STATUS, MULTIPLE_CHOICES, MOVED_PERMANENTLY, FOUND, SEE_OTHER, NOT_MODIFIED, USE_PROXY, TEMPORARY_REDIRECT, BAD_REQUEST, UNAUTHORIZED, PAYMENT_REQUIRED, FORBIDDEN, NOT_FOUND, METHOD_NOT_ALLOWED, NOT_ACCEPTABLE, PROXY_AUTHENTICATION, REQUEST_TIMEOUT, CONFLICT, GONE, LENGTH_REQUIRED, PRECONDITION_FAILED, REQUEST_ENTITY_TOO_LARGE, REQUEST_URI_TOO_LONG, UNSUPPORTED_MEDIA_TYPE, REQUESTED_RANGE_NOT_SATISFIED, EXPECTATION_FAILED, UNPROCESSABLE_ENTITY, LOCKED, FAILED_DEPENDENCY, INTERNAL_SERVER_ERROR, NOT_IMPLEMENTED, BAD_GATEWAY, SERVICE_UNAVAILABLE, GATEWAY_TIMEOUT, INSUFFICIENT_STORAGE]).freeze
+    end
+
+    class RestRequest
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      METHOD = 1
+      URI = 2
+      PARAMETERS = 3
+      HEADERS = 4
+      BODY = 5
+
+      FIELDS = {
+        METHOD => {:type => ::Thrift::Types::I32, :name => 'method', :enum_class => ElasticSearch::Thrift::Method},
+        URI => {:type => ::Thrift::Types::STRING, :name => 'uri'},
+        PARAMETERS => {:type => ::Thrift::Types::MAP, :name => 'parameters', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}, :optional => true},
+        HEADERS => {:type => ::Thrift::Types::MAP, :name => 'headers', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}, :optional => true},
+        BODY => {:type => ::Thrift::Types::STRING, :name => 'body', :binary => true, :optional => true}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field method is unset!') unless @method
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field uri is unset!') unless @uri
+        unless @method.nil? || ElasticSearch::Thrift::Method::VALID_VALUES.include?(@method)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field method!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class RestResponse
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      STATUS = 1
+      HEADERS = 2
+      BODY = 3
+
+      FIELDS = {
+        STATUS => {:type => ::Thrift::Types::I32, :name => 'status', :enum_class => ElasticSearch::Thrift::Status},
+        HEADERS => {:type => ::Thrift::Types::MAP, :name => 'headers', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}, :optional => true},
+        BODY => {:type => ::Thrift::Types::STRING, :name => 'body', :binary => true, :optional => true}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field status is unset!') unless @status
+        unless @status.nil? || ElasticSearch::Thrift::Status::VALID_VALUES.include?(@status)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field status!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+  end
+end

data/lib/elastic_search/thrift/elasticsearch.thrift

@@ -0,0 +1,80 @@
+namespace java org.elasticsearch.thrift
+namespace csharp ElasticSearch.Thrift
+namespace cpp  elasticsearch.thrift
+namespace rb ElasticSearch.Thrift
+namespace py elasticsearch
+namespace perl Elasticsearch
+
+enum Method {
+    GET = 0,
+    PUT = 1,
+    POST = 2,
+    DELETE = 3,
+    HEAD = 4,
+    OPTIONS = 5
+}
+
+struct RestRequest {
+    1: required Method method,
+    2: required string uri
+    3: optional map<string, string> parameters
+    4: optional map<string, string> headers
+    5: optional binary body
+}
+
+enum Status {
+    CONT = 100,
+    SWITCHING_PROTOCOLS = 101,
+    OK = 200,
+    CREATED = 201,
+    ACCEPTED = 202,
+    NON_AUTHORITATIVE_INFORMATION = 203,
+    NO_CONTENT = 204,
+    RESET_CONTENT = 205,
+    PARTIAL_CONTENT = 206,
+    MULTI_STATUS = 207,
+    MULTIPLE_CHOICES = 300,
+    MOVED_PERMANENTLY = 301,
+    FOUND = 302,
+    SEE_OTHER = 303,
+    NOT_MODIFIED = 304,
+    USE_PROXY = 305,
+    TEMPORARY_REDIRECT = 307,
+    BAD_REQUEST = 400,
+    UNAUTHORIZED = 401,
+    PAYMENT_REQUIRED = 402,
+    FORBIDDEN = 403,
+    NOT_FOUND = 404,
+    METHOD_NOT_ALLOWED = 405,
+    NOT_ACCEPTABLE = 406,
+    PROXY_AUTHENTICATION = 407,
+    REQUEST_TIMEOUT = 408,
+    CONFLICT = 409,
+    GONE = 410,
+    LENGTH_REQUIRED = 411,
+    PRECONDITION_FAILED = 412,
+    REQUEST_ENTITY_TOO_LARGE = 413,
+    REQUEST_URI_TOO_LONG = 414,
+    UNSUPPORTED_MEDIA_TYPE = 415,
+    REQUESTED_RANGE_NOT_SATISFIED = 416,
+    EXPECTATION_FAILED = 417,
+    UNPROCESSABLE_ENTITY = 422,
+    LOCKED = 423,
+    FAILED_DEPENDENCY = 424,
+    INTERNAL_SERVER_ERROR = 500,
+    NOT_IMPLEMENTED = 501,
+    BAD_GATEWAY = 502,
+    SERVICE_UNAVAILABLE = 503,
+    GATEWAY_TIMEOUT = 504,
+    INSUFFICIENT_STORAGE = 506
+}
+
+struct RestResponse {
+    1: required Status status,
+    2: optional map<string, string> headers,
+    3: optional binary body
+}
+
+service Rest {
+    RestResponse execute(1:required RestRequest request)
+}

data/lib/elastic_search/thrift/rest.rb

@@ -0,0 +1,79 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+
+module ElasticSearch
+  module Thrift
+    module Rest
+      class Client
+        include ::Thrift::Client
+
+        def execute(request)
+          send_execute(request)
+          return recv_execute()
+        end
+
+        def send_execute(request)
+          send_message('execute', Execute_args, :request => request)
+        end
+
+        def recv_execute()
+          result = receive_message(Execute_result)
+          return result.success unless result.success.nil?
+          raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'execute failed: unknown result')
+        end
+
+      end
+
+      class Processor
+        include ::Thrift::Processor
+
+        def process_execute(seqid, iprot, oprot)
+          args = read_args(iprot, Execute_args)
+          result = Execute_result.new()
+          result.success = @handler.execute(args.request)
+          write_result(result, oprot, 'execute', seqid)
+        end
+      end
+
+      # HELPER FUNCTIONS AND STRUCTURES
+
+      class Execute_args
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        REQUEST = 1
+
+        FIELDS = {
+          REQUEST => {:type => ::Thrift::Types::STRUCT, :name => 'request', :class => ElasticSearch::Thrift::RestRequest}
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Required field request is unset!') unless @request
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+
+      class Execute_result
+        include ::Thrift::Struct, ::Thrift::Struct_Union
+        SUCCESS = 0
+
+        FIELDS = {
+          SUCCESS => {:type => ::Thrift::Types::STRUCT, :name => 'success', :class => ElasticSearch::Thrift::RestResponse}
+        }
+
+        def struct_fields; FIELDS; end
+
+        def validate
+        end
+
+        ::Thrift::Struct.generate_accessors self
+      end
+    end
+  end
+end

data/lib/elastic_search/transport/base.rb

@@ -0,0 +1,21 @@
+module ElasticSearch
+  class TransportBase
+    def initialize(hosts, options = {})
+      @options = options
+      @connections =  hosts.collect { |host| get_connection(host) }
+    end
+
+    def current_connection
+      @connections.sample
+    end
+
+    def execute(request)
+      raise NotImplementedError
+    end
+
+    private
+    def get_connection(host)
+      raise NotImplementedError
+    end
+  end
+end

data/lib/elastic_search/transport/http.rb

@@ -0,0 +1,18 @@
+require 'faraday'
+
+module ElasticSearch
+  class HTTPTransport < TransportBase
+    def execute(request)
+      response = current_connection.send(request.method, request.path) do |req|
+        req.body = request.body unless request.method.eql?(:delete)
+        req.params = request.parameters if request.parameters
+      end
+      ElasticSearch::Response.new(body: response.body, status: response.status)
+    end
+
+    private
+    def get_connection(host)
+      Faraday.new(:url => host)
+    end
+  end
+end

data/lib/elastic_search/transport/thrift.rb

@@ -0,0 +1,38 @@
+require 'thrift'
+require 'elastic_search/thrift/elastic_search_constants'
+require 'elastic_search/thrift/elastic_search_types'
+require 'elastic_search/thrift/rest'
+
+module ElasticSearch
+  class ThriftTransport < TransportBase
+    def execute(request)
+      thrift_request = ElasticSearch::Thrift::RestRequest.new
+      thrift_request.method = get_method(request.method)
+      thrift_request.uri = request.path
+      thrift_request.body = request.body
+      thrift_request.parameters = request.parameters
+      response = current_connection.execute(thrift_request)
+      ElasticSearch::Response.new(body: response.body, status: response.status)
+    end
+
+    private
+    def get_method(method)
+      case method
+      when :get then ElasticSearch::Thrift::Method::GET
+      when :post then ElasticSearch::Thrift::Method::POST
+      when :put then ElasticSearch::Thrift::Method::PUT
+      when :delete then ElasticSearch::Thrift::Method::DELETE
+      when :head then ElasticSearch::Thrift::Method::HEAD
+      end
+    end
+
+    def get_connection(host)
+      host, port = host.to_s.split(":")
+      transport = ::Thrift::Socket.new(host, port.to_i, @options[:timeout])
+      buffered_transport = ::Thrift::BufferedTransport.new(transport)
+      buffered_transport.open
+      protocol = ::Thrift::BinaryProtocol.new(buffered_transport)
+      ElasticSearch::Thrift::Rest::Client.new(protocol)
+    end
+  end
+end

data/lib/elastic_search/type.rb

@@ -0,0 +1,64 @@
+module ElasticSearch
+  class Type
+    def initialize(type_name, index)
+      @type_name = type_name
+      @index = index
+    end
+
+    def search(body, parameters = {})
+      request = build_request({ method: :get, action: :search, parameters: parameters, body: body || {} })
+      response = execute(request)
+      ElasticSearch::Search::Results.new(response)
+    end
+
+    def put(id, doc)
+      request = build_request({ method: :put, id: id.to_s, body: doc })
+      execute(request)
+    end
+    alias :index :put
+
+    def get(id)
+      request = build_request({ method: :get, id: id.to_s })
+      response = execute(request)
+      ElasticSearch::Item.new(response)
+    end
+
+    def delete(id)
+      request = build_request({ method: :delete, id: id.to_s })
+      execute(request)
+    end
+
+    def delete_by_query(body, parameters = {})
+      request = build_request({ method: :delete, action: :search, parameters: parameters, body: body || {} })
+      execute(request)
+    end
+
+    def update(id, body, parameters = {})
+      request = build_request({ method: :post, id: id.to_s, action: :update, parameters: parameters, body: doc })
+      execute(request)
+    end
+
+    def count(parameters = {})
+      request = build_request({ method: :post, action: :update, id: id.to_s, body: doc })
+      execute(request).body['count']
+    end
+
+    def more_like_this(id, parameters = {})
+      request = build_request({ method: :get, id: id.to_s, action: :mlt, parameters: parameters, body: doc })
+    end
+
+    def delete_mapping
+      request = build_request({ method: :delete })
+      execute(request).status.eql?(200)
+    end
+
+    def execute(request)
+      @index.client.execute(request)
+    end
+
+    private
+    def build_request(parameters = {})
+      Request.new({ index: @index.index_name, type: @type_name }.merge(parameters))
+    end
+  end
+end

data/lib/elastic_search/version.rb

@@ -0,0 +1,3 @@
+module ElasticSearch
+  VERSION = "0.0.1"
+end

data/lib/elastic_search.rb

@@ -0,0 +1,16 @@
+require "active_support/all"
+
+require "elastic_search/client"
+require "elastic_search/index"
+require "elastic_search/item"
+require "elastic_search/request"
+require "elastic_search/response"
+require "elastic_search/search/results"
+require "elastic_search/transport/base"
+require "elastic_search/transport/thrift"
+require "elastic_search/transport/http"
+require "elastic_search/type"
+require "elastic_search/version"
+
+module ElasticSearch
+end

data/spec/elasticsearch/client_spec.rb

@@ -0,0 +1,53 @@
+require 'spec_helper'
+
+describe ElasticSearch::Client do
+  TRANSPORT_CONTEXTS.each do |transport, context|
+    describe context do
+      let :client do
+        ElasticSearch::Client.new(transport)
+      end
+
+      after do
+        begin
+          client.delete_index(TEST_INDEX)
+        rescue
+        end
+      end
+
+      describe :create_index do
+        it "should create a new index" do
+          index = client.create_index(TEST_INDEX)
+          index.must_be_instance_of ElasticSearch::Index
+          index.refresh
+          client.has_index?(TEST_INDEX).must_equal true
+        end
+
+        it "should create a new index with a mapping" do
+          mapping = {
+            test_type: {
+              properties: {
+                foo: { type: :string, analyzer: :keyword }
+              }
+            }
+          }
+          index = client.create_index(TEST_INDEX, mappings: mapping)
+          index.mapping['test_index']['test_type']['properties']['foo']['analyzer'].must_equal "keyword"
+          index.mapping['test_index']['test_type']['properties']['foo']['type'].must_equal "string"
+        end
+      end
+
+      describe :delete_index do
+        it "should delete an index" do
+          client.has_index?(TEST_INDEX).must_equal false
+        end
+      end
+
+      describe :has_index? do
+        it "should know if an index exists" do
+          client.create_index(TEST_INDEX).refresh
+          client.has_index?(TEST_INDEX).must_equal true
+        end
+      end
+    end
+  end
+end

data/spec/elasticsearch/index_spec.rb

@@ -0,0 +1,106 @@
+require 'spec_helper'
+
+describe ElasticSearch::Index do
+  TRANSPORT_CONTEXTS.each do |transport, context|
+    describe context do
+      let :client do
+        ElasticSearch::Client.new(transport)
+      end
+
+      let :index do
+        client[TEST_INDEX]
+      end
+
+      before do
+        client.create_index(TEST_INDEX).refresh
+      end
+
+      after do
+        begin
+          client.delete_index(TEST_INDEX)
+        rescue ElasticSearch::IndexMissingException
+        end
+      end
+
+      describe :search do
+        it "should return results" do
+          index['type_1'].put(1, { foo: 'bar' })
+          index['type_2'].put(2, { foo: 'baz' })
+          index.refresh
+          results = index.search(nil, q: '*')
+          results.must_be_instance_of ElasticSearch::Search::Results
+        end
+
+        it "should return the total" do
+          (1..20).each { |i| index['type'].put(i, { foo: 'bar' }) }
+          index.refresh
+          results = index.search(nil, q: '*')
+          results.total.must_equal 20
+          results.count.must_equal 10
+        end
+
+        it "should know the type of results" do
+          index['type_1'].put(1, { foo: 'bar' })
+          index.refresh
+          results = index.search(nil, q: '*')
+          results[0].type.must_equal 'type_1'
+        end
+      end
+
+      describe :refresh do
+        it "should refresh the index" do
+          index.refresh
+        end
+      end
+
+      describe :open do
+        it "should not raise an error if index is open" do
+          index.close
+          index.open
+          index[TEST_TYPE].put(1, { foo: 'bar' })
+        end
+      end
+
+      describe :close do
+        it "should throw an error when trying to put into a closed index" do
+          index.close
+          proc { index[TEST_TYPE].put(1, { foo: 'bar' }) }.must_raise ElasticSearch::ResponseError
+        end
+      end
+
+      describe :settings do
+        it "should return the index settings"
+      end
+
+      describe :settings= do
+        it "should set the index settings"
+      end
+
+      describe :mapping do
+        it "should return the mapping" do
+          index.mapping
+        end
+      end
+
+      describe :mapping= do
+        it "should set the index mapping"
+      end
+
+      describe :delete_mapping do
+        it "should set delete the index mapping"
+      end
+
+      describe :optimize do
+        it "should optimize the index"
+      end
+
+      describe :exists? do
+        it "should check if the index exists" do
+          index.exists?.must_equal true
+          client.delete_index(TEST_INDEX)
+          index.exists?.must_equal false
+        end
+      end
+    end
+  end
+end

data/spec/elasticsearch/type_spec.rb

@@ -0,0 +1,57 @@
+require 'spec_helper'
+
+describe ElasticSearch::Type do
+  TRANSPORT_CONTEXTS.each do |transport, context|
+    describe context do
+      let :client do
+        ElasticSearch::Client.new(transport)
+      end
+
+      let :index do
+        client[TEST_INDEX]
+      end
+
+      let :type do
+        index[TEST_TYPE]
+      end
+
+      before do
+        client.create_index(TEST_INDEX).refresh
+      end
+
+      after do
+        begin client.delete_index(TEST_INDEX) end
+      end
+
+      describe :put do
+        it "should index a document" do
+          type.put(1, { foo: 'bar' })
+          index.refresh
+          type.get(1)['foo'].must_equal 'bar'
+        end
+      end
+
+      describe :get do
+        it "should return an error if document does not exist"
+      end
+
+      describe :search do
+        it "should return results" do
+          type.put(1, { foo: 'bar' })
+          type.put(2, { foo: 'baz' })
+          index.refresh
+          results = type.search(nil, q: '*')
+          results.must_be_instance_of ElasticSearch::Search::Results
+        end
+
+        it "should return the total" do
+          (1..20).each { |i| type.put(i, { foo: 'bar' }) }
+          index.refresh
+          results = type.search(nil, q: '*')
+          results.total.must_equal 20
+          results.count.must_equal 10
+        end
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,26 @@
+require "bundler/setup"
+require 'minitest/spec'
+require 'minitest/mock'
+require 'turn/autorun'
+require 'active_support/all'
+
+require 'elastic_search'
+
+class MiniTest::Spec
+  alias :method_name :__name__ if defined? :__name__
+
+  class << self
+    alias :context :describe
+  end
+end
+
+Turn.config.format = :progress
+Turn.config.natural = true
+
+TEST_INDEX = 'test_index'
+TEST_TYPE = 'test_type'
+
+TRANSPORT_CONTEXTS = [
+  [ElasticSearch::ThriftTransport.new(['localhost:9500']), 'Using Thrift'],
+  [ElasticSearch::HTTPTransport.new(['http://localhost:9200']), 'Using HTTP']
+]

metadata

@@ -0,0 +1,139 @@
+--- !ruby/object:Gem::Specification
+name: elasticsearch-ruby
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Edmund Salvacion
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-16 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thrift
+  requirement: &70333353079280 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70333353079280
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: &70333353078860 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70333353078860
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: &70333353078200 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70333353078200
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: &70333353077540 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70333353077540
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: &70333353077040 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70333353077040
+- !ruby/object:Gem::Dependency
+  name: turn
+  requirement: &70333353076580 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70333353076580
+description: Ruby Driver for Elastic Search
+email:
+- edmund@edmundatwork.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- elastic_search.gemspec
+- lib/elastic_search.rb
+- lib/elastic_search/client.rb
+- lib/elastic_search/index.rb
+- lib/elastic_search/item.rb
+- lib/elastic_search/request.rb
+- lib/elastic_search/response.rb
+- lib/elastic_search/search/results.rb
+- lib/elastic_search/thrift/elastic_search_constants.rb
+- lib/elastic_search/thrift/elastic_search_types.rb
+- lib/elastic_search/thrift/elasticsearch.thrift
+- lib/elastic_search/thrift/rest.rb
+- lib/elastic_search/transport/base.rb
+- lib/elastic_search/transport/http.rb
+- lib/elastic_search/transport/thrift.rb
+- lib/elastic_search/type.rb
+- lib/elastic_search/version.rb
+- spec/elasticsearch/client_spec.rb
+- spec/elasticsearch/index_spec.rb
+- spec/elasticsearch/type_spec.rb
+- spec/spec_helper.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: elasticsearch
+rubygems_version: 1.8.17
+signing_key: 
+specification_version: 3
+summary: Ruby Driver for Elastic Search
+test_files:
+- spec/elasticsearch/client_spec.rb
+- spec/elasticsearch/index_spec.rb
+- spec/elasticsearch/type_spec.rb
+- spec/spec_helper.rb