dilute 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rspec

@@ -0,0 +1,2 @@
+--color
+--format progress

data/.ruby-gemset

@@ -0,0 +1 @@
+dilute

data/.ruby-version

@@ -0,0 +1 @@
+ruby-1.9.3-p374

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 Sam Schenkman-Moore
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,38 @@
+# Dilute
+
+It's an ORM for ElasticSearch.
+
+Disclosure: I'm learning ElasticSearch, this is the product of some of my experiments. This has been helpful for my needs, but is very new.
+
+## Installation
+
+    gem 'dilute'
+
+## Usage
+
+In a Rails app called "MyRailsApp" ...
+
+    class Note
+      include Dilute::Modelize
+
+      define_type do
+        attribute(:data).not_indexed
+        attribute(:header).not_indexed
+        attribute(:keys).index_analyzer("keyword")
+        attribute(:created_at, :date)
+      end
+    end
+
+This will create a new index called "MyRailsApp" with a new type mapping "notes".
+
+define_type can some options: `server_url`, `index_name`, `type_name`
+
+    define_type(server_url: "http://example.com/", index_name: "special_name", type_name: "something_else")
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/dilute.gemspec

@@ -0,0 +1,29 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'dilute/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "dilute"
+  spec.version       = Dilute::VERSION
+  spec.authors       = ["Sam Schenkman-Moore"]
+  spec.email         = ["samsm@samsm.com"]
+  spec.description   = %q{A basic ORM-sort of thing for Stretcher.}
+  spec.summary       = %q{An ActiveRecord-ish thing for using ElasticSearch through Stretcher.}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "stretcher"
+  spec.add_dependency "activesupport"
+  spec.add_dependency "activemodel"
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
+  spec.add_development_dependency "pry"
+end

data/lib/dilute/attribute.rb

@@ -0,0 +1,54 @@
+require "active_support/hash_with_indifferent_access"
+
+class Dilute::Attribute
+  attr_reader :options
+  def initialize(options = {})
+    @options = HashWithIndifferentAccess.new.merge(options)
+  end
+
+  %w(index_name store index term_vector boost null_value omit_norms
+     omit_term_freq_and_positions index_options analyzer index_analyzer
+     search_analyzer include_in_all ignore_above position_offset_gap
+     precision_step ignore_malformed format).each do |option|
+    define_method option do |new_value|
+      raise "invalid option #{option} for #{type}" unless acceptable_mapping_options[type_mapping].include?(option)
+      chain_add(option => new_value)
+    end
+  end
+
+  def not_indexed
+    index("not_analyzed")
+  end
+
+  def to_config
+    options
+  end
+
+  private
+
+  def chain_add(hsh)
+    # self.class.new(options.merge(hsh))
+    options.merge!(hsh)
+    self
+  end
+
+  def type(new_val = nil)
+    new_val ? options[:type] = new_val.to_sym : options[:type].to_sym
+  end
+
+  def type_mapping
+    return :number if [:float, :double, :integer, :long, :short, :byte].include?(type)
+    type
+  end
+
+
+  def acceptable_mapping_options
+    {
+      string: %w(index_name store index term_vector boost null_value omit_norms omit_term_freq_and_positions index_options analyzer index_analyzer search_analyzer include_in_all ignore_above position_offset_gap),
+      number: %w(type index_name store index precision_step boost null_value include_in_all ignore_malformed),
+      date: %w(index_name format store index precision_step boost null_value include_in_all ignore_malformed),
+      boolean: %w(index_name store index boost null_value include_in_all),
+      binary: %w(index_name)
+    }
+  end
+end

data/lib/dilute/modelize.rb

@@ -0,0 +1,103 @@
+require "active_support/concern"
+require "active_model/naming"
+require 'active_support/core_ext/string'
+
+module Dilute::Modelize
+  extend ActiveSupport::Concern
+
+  included do
+    attr_reader :attributes
+    extend ActiveModel::Naming
+  end
+
+  module ClassMethods
+    def name
+      super || "DefaultDilutedModel"
+    end
+
+    def find(id)
+      new(type.get(id, {}, true))
+    end
+
+    def all(query = {}, page_size = 10)
+      search_results = type.search(size: page_size, query: {match_all: {}})
+      search_results.raw["hits"]["hits"].collect {|r| new(r) }
+    end
+
+    def define_type(options = {}, &blk)
+      @type ||= Dilute::Type.new(type_defaults.merge(options), &blk)
+      add_attribute_accessors
+    end
+
+    def application_name
+      if Kernel.const_defined? :Rails
+        Rails.application.class.parent_name
+      else
+        "default_dilute_index_name"
+      end
+    end
+
+    def type_defaults
+      HashWithIndifferentAccess.new({
+        index_name: application_name,
+        type_name: model_name.plural
+      })
+    end
+
+    def add_attribute_accessors
+      type.properties.keys.each do |k|
+        define_method k do
+          attributes["_source"] && attributes["_source"][k]
+        end
+        define_method "#{k}=" do |new_var|
+          attributes["_source"] && attributes["_source"][k] = new_var
+        end
+      end
+    end
+
+    def type
+      @type or raise "No type defined!"
+    end
+  end
+
+  def id
+    attributes["_id"]
+  end
+
+  def attributes_without_elasticsearch_vars
+    # elasticsearch_vars = %w(ok _index _type _id _version)
+    attributes["_source"]
+  end
+
+  def save
+    if id
+      type.put(id, attributes_without_elasticsearch_vars)
+    else
+      results = type.post(attributes_without_elasticsearch_vars)
+      if results["ok"] == true
+        attributes.merge!(results)
+      else
+        puts "Something went wrong!"
+        puts "attributes: #{attributes}"
+      end
+    end
+  end
+
+  def destroy
+    raise "id can't be nil" unless id
+    type.delete(id)
+  end
+
+  def initialize(attributes = {})
+    @attributes = if attributes.has_key?("_source")
+      attributes
+    else
+      {"_source" => HashWithIndifferentAccess.new.merge(attributes)}
+    end
+  end
+
+  def type
+    self.class.type.type
+  end
+
+end

data/lib/dilute/type.rb

@@ -0,0 +1,75 @@
+require "active_support/hash_with_indifferent_access"
+require "stretcher"
+
+class Dilute::Type
+  attr_reader :options, :attributes
+  def initialize(options = {}, &block)
+    @options = HashWithIndifferentAccess.new.merge(options)
+    @attributes = HashWithIndifferentAccess.new
+    instance_eval(&block) if block_given?
+  end
+
+  def server_url
+    options[:server_url] || "http://localhost:9200"
+  end
+
+  def server
+    @server ||= Stretcher::Server.new(server_url, log_level: :info)
+  end
+
+  def index_name
+    options[:index_name] # :qknt
+  end
+
+  def type_name
+    options[:type_name] # :notes
+  end
+
+  def index
+    @index ||= begin
+      the_index = server.index(index_name)
+      the_index.exists? ? the_index : (the_index.create && the_index)
+    end
+  end
+
+  def type
+    @type ||= begin
+      the_type = index.type(type_name)
+      the_type.exists? ? the_type : (the_type.put_mapping(generate_mapping) && the_type)
+    end
+  end
+
+  def delete
+    result = type.delete_mapping
+    @type = nil
+    result
+  end
+
+  def generate_mapping
+    { :"#{type_name}" => { properties: properties } }
+  end
+
+  def properties
+    attributes.inject({}) do |sum, pair|
+      k, v = pair
+      sum[k] = v.to_config ; sum
+    end
+  end
+
+  def mapping
+    type.get_mapping
+  end
+
+  def attribute(key, attribute_type = :string)
+    attributes[key] = Dilute::Attribute.new(type: attribute_type)
+  end
+
+  def search(*args)
+    type.search *args
+  end
+
+  def get(*args)
+    type.get *args
+  end
+
+end

data/lib/dilute/version.rb

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

data/lib/dilute.rb

@@ -0,0 +1,9 @@
+module Dilute
+
+  # autoload all the stuff in filters under this namespace
+  Dir.glob("#{File.dirname(__FILE__)}/dilute/*.rb").each do |filename|
+    class_name = File.basename(filename, '.*').split('_').collect(&:capitalize).join.to_sym
+    autoload class_name, filename
+  end
+
+end

data/spec/modelize_integration_spec.rb

@@ -0,0 +1,47 @@
+require_relative "spec_helper"
+require "pry"
+
+describe Dilute::Modelize do
+  let(:dilute_class) do
+    Class.new do
+      include Dilute::Modelize
+
+      define_type(index_name: :dilute_test_suite, type_name: :abstract) do
+        attribute(:data).not_indexed
+        attribute(:header).not_indexed
+        attribute(:keys).index_analyzer("keyword")
+        attribute(:created_at, :date)
+      end
+    end
+  end
+
+  let(:test_data) { "foo city" }
+  let(:new_diluted) { dilute_class.new({data: test_data}) }
+  let(:saved) { new_diluted.save ; new_diluted }
+
+  it "should have accessors for new object" do
+    expect(new_diluted.data).to eq(test_data)
+  end
+
+  it "should save data" do
+    expect(new_diluted.id).to be_nil
+    new_diluted.save
+    expect(new_diluted.id).not_to be_nil
+  end
+
+  it "should retrieve saved data" do
+    expect(dilute_class.find(saved.id).data).to eq(test_data)
+  end
+
+  it "should retreive many saved documents" do
+    destroy_test_index!
+    10.times { dilute_class.new({data: test_data}).save }
+    sleep 1
+    expect(dilute_class.all({}, 5)).to have(5).items
+  end
+
+  it "I have no idea what I'm doing." do
+    true
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,26 @@
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.run_all_when_everything_filtered = true
+  # config.filter_run :focus
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'random'
+end
+
+require "dilute"
+
+def destroy_test_index!
+  require "stretcher"
+  server = Stretcher::Server.new("http://localhost:9200", log_level: :info)
+  index  = server.index("dilute_test_suite")
+  index.delete
+end

metadata

@@ -0,0 +1,182 @@
+--- !ruby/object:Gem::Specification
+name: dilute
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Sam Schenkman-Moore
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-06-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: stretcher
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: activemodel
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A basic ORM-sort of thing for Stretcher.
+email:
+- samsm@samsm.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- .ruby-gemset
+- .ruby-version
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- dilute.gemspec
+- lib/dilute.rb
+- lib/dilute/attribute.rb
+- lib/dilute/modelize.rb
+- lib/dilute/type.rb
+- lib/dilute/version.rb
+- spec/modelize_integration_spec.rb
+- spec/spec_helper.rb
+homepage: ''
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -1974874539562345975
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -1974874539562345975
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: An ActiveRecord-ish thing for using ElasticSearch through Stretcher.
+test_files:
+- spec/modelize_integration_spec.rb
+- spec/spec_helper.rb