indices 0.0.1

data/lib/indices/pagination.rb

@@ -0,0 +1,33 @@
+module Indices
+  module Pagination
+
+    def total_pages
+      @total_pages ||= [(total_count.to_f / page_length).ceil, 1].max
+    end
+
+    def previous_page
+      @previous_page ||= (current_page > 1 ? (current_page - 1) : nil)
+    end
+
+    def next_page
+      @next_page ||= (current_page < total_pages ? (current_page + 1) : nil)
+    end
+
+    def first_page
+      1
+    end
+
+    def last_page
+      total_pages
+    end
+
+    def out_of_bounds?
+      @out_of_bounds ||= (current_page > total_pages || current_page < first_page)
+    end
+
+    def total_count
+      @total_count ||= (response['hits']['total'].to_i - padding)
+    end
+
+  end
+end

data/lib/indices/proxy.rb

@@ -0,0 +1,17 @@
+module Indices
+  class Proxy
+
+    def initialize(name, options={}, &block)
+      @options = options
+      instance_eval &block
+      Indices.add name, @options
+    end
+
+    %i(mappings serializer search).each do |name|
+      define_method name do |&block|
+        @options[name] = block
+      end
+    end
+
+  end
+end

data/lib/indices/railtie.rb

@@ -0,0 +1,15 @@
+module Indices
+  class Railtie < Rails::Railtie
+
+    initializer :indices do
+      Dir[Rails.root.join('app/indices/*')].each do |index|
+        load index
+      end
+    end
+
+    rake_tasks do
+      load 'tasks/indices.rake'
+    end
+
+  end
+end

data/lib/indices/version.rb

@@ -0,0 +1,5 @@
+module Indices
+
+  VERSION = '0.0.1'
+
+end

data/lib/tasks/indices.rake

@@ -0,0 +1,11 @@
+namespace :indices do
+  desc 'Build all indices.'
+  task build: :environment do
+    Indices.build
+  end
+
+  desc 'Rebuild all indices.'
+  task rebuild: :environment do
+    Indices.rebuild
+  end
+end

data/test/dsl_test.rb

@@ -0,0 +1,92 @@
+require 'test_helper'
+
+class DslTest < ActiveSupport::TestCase
+
+  test 'search' do
+    seed = rand
+    days = [1,2,3]
+    opens_at = Time.now.to_i
+    closes_at = (Time.now + 1.day).to_i
+    assert_equal(
+      {
+        query: {
+          function_score: {
+            functions: [
+              { random_score: { seed: seed } },
+              {
+                weight: 400,
+                filter: {
+                  bool: {
+                    must: [
+                      { term: {} }
+                    ]
+                  }
+                }
+              }
+            ]
+          },
+          query: {
+            filtered: {
+              filter: {
+                bool: {
+                  should: [
+                    { term: { :'schedules.days' => days } }
+                  ],
+                  must: [
+                    { range: { :'schedules.opens_at' => { lte: opens_at } } }
+                  ],
+                  must_not: []
+                }
+              }
+            }
+          }
+        },
+        size: {}
+      },
+      build_request do
+        query do
+          function_score do
+            functions do
+              random_score do
+                seed seed
+              end
+              filter weight: 400 do
+                bool do
+                  must do
+                    term
+                  end
+                end
+              end
+            end
+          end
+          query do
+            filtered do
+              filter do
+                bool do
+                  should do
+                    term 'schedules.days' => days
+                  end
+                  must do
+                    range 'schedules.opens_at' do
+                      lte opens_at
+                    end
+                  end
+                  must_not do
+                  end
+                end
+              end
+            end
+          end
+        end
+        size
+      end
+    )
+  end
+
+  private
+
+  def build_request(&block)
+    Indices::Dsl::Search.new(&block).to_h
+  end
+
+end

data/test/dummy/Rakefile

@@ -0,0 +1,5 @@
+# Add your own tasks in files placed in lib/tasks ending in .rake,
+# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
+
+require File.expand_path('../config/application', __FILE__)
+Rails.application.load_tasks

data/test/dummy/app/assets/javascripts/application.js

@@ -0,0 +1,13 @@
+// This is a manifest file that'll be compiled into application.js, which will include all the files
+// listed below.
+//
+// Any JavaScript/Coffee file within this directory, lib/assets/javascripts, vendor/assets/javascripts,
+// or any plugin's vendor/assets/javascripts directory can be referenced here using a relative path.
+//
+// It's not advisable to add code directly here, but if you do, it'll appear at the bottom of the
+// compiled file.
+//
+// Read Sprockets README (https://github.com/rails/sprockets#sprockets-directives) for details
+// about supported directives.
+//
+//= require_tree .

data/test/dummy/app/assets/stylesheets/application.css

@@ -0,0 +1,15 @@
+/*
+ * This is a manifest file that'll be compiled into application.css, which will include all the files
+ * listed below.
+ *
+ * Any CSS and SCSS file within this directory, lib/assets/stylesheets, vendor/assets/stylesheets,
+ * or any plugin's vendor/assets/stylesheets directory can be referenced here using a relative path.
+ *
+ * You're free to add application-wide styles to this file and they'll appear at the bottom of the
+ * compiled file so the styles you add here take precedence over styles defined in any styles
+ * defined in the other CSS/SCSS files in this directory. It is generally better to create a new
+ * file per style scope.
+ *
+ *= require_tree .
+ *= require_self
+ */

data/test/dummy/app/controllers/application_controller.rb

@@ -0,0 +1,5 @@
+class ApplicationController < ActionController::Base
+  # Prevent CSRF attacks by raising an exception.
+  # For APIs, you may want to use :null_session instead.
+  protect_from_forgery with: :exception
+end

data/test/dummy/app/helpers/application_helper.rb

@@ -0,0 +1,2 @@
+module ApplicationHelper
+end

data/test/dummy/app/indices/products_index.rb

@@ -0,0 +1,51 @@
+Indices.define :products do
+
+  mappings do
+    properties :name, :category, :shop_id, :price, :currency, :product_suggestions
+    _parent type: 'shop'
+  end
+
+  serializer do |record|
+    set record, :name, :category, :shop_id, :price, :currency
+    product_suggestions do
+      input [record.name, transliterate(record.name)].uniq
+      output record.name
+      context do
+        shop_id [record.shop_id.to_s, 'all'].compact
+      end
+    end
+  end
+
+  search do |*args|
+    options = args.extract_options!
+    shop = options[:shop]
+    term = args.first
+    query do
+      filtered do
+        filter do
+          bool do
+            must do
+              if shop
+                term do
+                  _parent shop.id
+                end
+              end
+            end
+          end
+        end
+        query do
+          if term.present?
+            multi_match do
+              query term
+              type 'phrase_prefix'
+              fields %w(name category)
+            end
+          else
+            match_all
+          end
+        end
+      end
+    end
+  end
+
+end

data/test/dummy/app/indices/shops_index.rb

@@ -0,0 +1,28 @@
+Indices.define :other, class_name: 'Shop' do
+
+  mappings do
+    properties :name
+  end
+
+  serializer do |record|
+    set record, :name
+  end
+
+  search do |*args|
+    options = args.extract_options!
+    term = args.first
+    query do
+      if term.present?
+        match do
+          name do
+            query term
+            type 'phrase_prefix'
+          end
+        end
+      else
+        match_all
+      end
+    end
+  end
+
+end

data/test/dummy/app/models/product.rb

@@ -0,0 +1,5 @@
+class Product < ActiveRecord::Base
+
+  belongs_to :shop
+
+end

data/test/dummy/app/models/shop.rb

@@ -0,0 +1,5 @@
+class Shop < ActiveRecord::Base
+
+  has_many :products
+
+end

data/test/dummy/app/views/layouts/application.html.erb

@@ -0,0 +1,12 @@
+<!DOCTYPE html>
+<html>
+  <head>
+    <title>Dummy</title>
+    <%= stylesheet_link_tag 'application', media: 'all', 'data-turbolinks-track' => true %>
+    <%= javascript_include_tag 'application', 'data-turbolinks-track' => true %>
+    <%= csrf_meta_tags %>
+  </head>
+  <body>
+    <%= yield %>
+  </body>
+</html>

data/test/dummy/bin/bundle

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../../Gemfile', __FILE__)
+load Gem.bin_path('bundler', 'bundle')

data/test/dummy/bin/rails

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+APP_PATH = File.expand_path('../../config/application', __FILE__)
+require_relative '../config/boot'
+require 'rails/commands'

data/test/dummy/bin/rake

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require_relative '../config/boot'
+require 'rake'
+Rake.application.run

data/test/dummy/bin/setup

@@ -0,0 +1,30 @@
+#!/usr/bin/env ruby
+
+require 'pathname'
+
+# path to your application root.
+APP_ROOT = Pathname.new File.expand_path('../../',  __FILE__)
+
+Dir.chdir APP_ROOT do
+  # This script is a starting point to setup your application.
+  # Add necessary setup steps to this file:
+
+  puts "== Installing dependencies =="
+  system 'gem install bundler --conservative'
+  system 'bundle check || bundle install'
+
+  # puts "\n== Copying sample files =="
+  # unless File.exist?('config/database.yml')
+  #   system 'cp config/database.yml.sample config/database.yml'
+  # end
+
+  puts "\n== Preparing database =="
+  system 'bin/rake db:setup'
+
+  puts "\n== Removing old logs and tempfiles =="
+  system 'rm -f log/*'
+  system 'rm -rf tmp/cache'
+
+  puts "\n== Restarting application server =="
+  system 'touch tmp/restart.txt'
+end

data/test/dummy/config.ru

@@ -0,0 +1,4 @@
+# This file is used by Rack-based servers to start the application.
+
+require ::File.expand_path('../config/environment', __FILE__)
+run Rails.application

data/test/dummy/config/application.rb

@@ -0,0 +1,25 @@
+require File.expand_path('../boot', __FILE__)
+
+require 'rails/all'
+
+Bundler.require(*Rails.groups)
+require 'indices'
+
+module Dummy
+  class Application < Rails::Application
+    # Settings in config/environments/* take precedence over those specified here.
+    # Application configuration should go into files in config/initializers
+    # -- all .rb files in that directory are automatically loaded.
+
+    # Set Time.zone default to the specified zone and make Active Record auto-convert to this zone.
+    # Run "rake -D time" for a list of tasks for finding time zone names. Default is UTC.
+    # config.time_zone = 'Central Time (US & Canada)'
+
+    # The default locale is :en and all translations from config/locales/*.rb,yml are auto loaded.
+    # config.i18n.load_path += Dir[Rails.root.join('my', 'locales', '*.{rb,yml}').to_s]
+    # config.i18n.default_locale = :de
+
+    # Do not swallow errors in after_commit/after_rollback callbacks.
+    config.active_record.raise_in_transactional_callbacks = true
+  end
+end

data/test/dummy/config/boot.rb

@@ -0,0 +1,5 @@
+# Set up gems listed in the Gemfile.
+ENV['BUNDLE_GEMFILE'] ||= File.expand_path('../../../../Gemfile', __FILE__)
+
+require 'bundler/setup' if File.exist?(ENV['BUNDLE_GEMFILE'])
+$LOAD_PATH.unshift File.expand_path('../../../../lib', __FILE__)

data/test/dummy/config/database.yml

@@ -0,0 +1,7 @@
+development:
+  adapter: postgresql
+  database: indices_development
+
+test:
+  adapter: postgresql
+  database: indices_test

data/test/dummy/config/database.yml.travis

@@ -0,0 +1,3 @@
+test:
+  adapter: <%= 'jdbc' if RUBY_ENGINE == 'jruby' %>postgresql
+  database: travis

data/test/dummy/config/environment.rb

@@ -0,0 +1,5 @@
+# Load the Rails application.
+require File.expand_path('../application', __FILE__)
+
+# Initialize the Rails application.
+Rails.application.initialize!