typesense-rails 1.0.0.rc1

data/lib/typesense/pagination/pagy.rb

@@ -0,0 +1,29 @@
+unless defined? Pagy
+  raise(Typesense::BadConfiguration, "Typesense: Please add 'pagy' to your Gemfile to use Pagy pagination backend")
+end
+
+module Typesense
+  module Pagination
+    class Pagy
+
+      def self.create(results, total_hits, options = {})
+        vars = {
+          count: total_hits,
+          page: options[:page],
+          items: options[:per_page]
+        }
+
+        pagy_version = Gem::Version.new(::Pagy::VERSION)
+        pagy = if pagy_version >= Gem::Version.new('9.0')
+                 ::Pagy.new(**vars)
+               else
+                 ::Pagy.new(vars)
+               end
+
+        [pagy, results]
+      end
+    end
+
+  end
+end
+

data/lib/typesense/pagination/will_paginate.rb

@@ -0,0 +1,17 @@
+begin
+  require 'will_paginate/collection'
+rescue LoadError
+  raise(Typesense::BadConfiguration, "Typesense: Please add 'will_paginate' to your Gemfile to use will_paginate pagination backend")
+end
+
+module Typesense
+  module Pagination
+    class WillPaginate
+      def self.create(results, total_hits, options = {})
+        ::WillPaginate::Collection.create(options[:page], options[:per_page], total_hits) do |pager|
+          pager.replace results
+        end
+      end
+    end
+  end
+end

data/lib/typesense/pagination.rb

@@ -0,0 +1,20 @@
+module Typesense
+  module Pagination
+
+    autoload :WillPaginate, 'typesense/pagination/will_paginate'
+    autoload :Kaminari, 'typesense/pagination/kaminari'
+    autoload :Pagy, 'typesense/pagination/pagy'
+
+    def self.create(results, total_hits, options = {})
+      return results if Typesense.configuration[:pagination_backend].nil?
+      begin
+        backend = Typesense.configuration[:pagination_backend].to_s.gsub(/\/(.?)/) { "::#{$1.upcase}" }.gsub(/(?:^|_)(.)/) { $1.upcase } # classify pagination backend name
+        page = Object.const_get(:Typesense).const_get(:Pagination).const_get(backend).create(results, total_hits, options)
+        page
+      rescue NameError
+        raise(BadConfiguration, "Unknown pagination backend")
+      end
+    end
+    
+  end
+end

data/lib/typesense/railtie.rb

@@ -0,0 +1,12 @@
+require 'rails'
+
+module Typesense
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      load "typesense/tasks/typesense.rake"
+    end
+  end
+
+  class Engine < Rails::Engine
+  end
+end

data/lib/typesense/tasks/typesense.rake

@@ -0,0 +1,17 @@
+namespace :typesense do
+  desc "Reindex all models"
+  task :reindex => :environment do
+    Typesense::Utilities.reindex_all_models
+  end
+
+  desc "Set settings to all indexes"
+  task :set_all_settings => :environment do
+    Typesense::Utilities.set_settings_all_models
+  end
+
+  desc "Clear all indexes"
+  task :clear_indexes => :environment do
+    puts "clearing all indexes"
+    Typesense::Utilities.clear_all_indexes
+  end
+end

data/lib/typesense/typesense_job.rb

@@ -0,0 +1,9 @@
+module Typesense
+  class TypesenseJob < ::ActiveJob::Base
+    queue_as { Typesense.configuration[:queue_name] }
+
+    def perform(record, method)
+      record.send(method)
+    end
+  end
+end

data/lib/typesense/utilities.rb

@@ -0,0 +1,47 @@
+module Typesense
+  module Utilities
+    class << self
+      def get_model_classes
+        if Rails.application && defined?(Rails.autoloaders) && Rails.autoloaders.zeitwerk_enabled?
+          Zeitwerk::Loader.eager_load_all
+        elsif Rails.application
+          Rails.application.eager_load!
+        end
+        Typesense.instance_variable_get :@included_in
+      end
+
+      def clear_all_indexes
+        get_model_classes.each do |klass|
+          klass.clear_index!
+        end
+      end
+
+      def reindex_all_models
+        klasses = get_model_classes
+
+        puts ""
+        puts "Reindexing #{klasses.count} models: #{klasses.to_sentence}."
+        puts ""
+
+        klasses.each do |klass|
+          puts klass
+          puts "Reindexing #{klass.count} records..."
+          klass.typesense_reindex
+        end
+      end
+
+      def set_settings_all_models
+        klasses = get_model_classes
+
+        puts ""
+        puts "Pushing settings for #{klasses.count} models: #{klasses.to_sentence}."
+        puts ""
+
+        klasses.each do |klass|
+          puts "Pushing #{klass} settings..."
+          klass.typesense_set_settings
+        end
+      end
+    end
+  end
+end

data/lib/typesense/version.rb

@@ -0,0 +1,3 @@
+module Typesense
+  GEM_VERSION = "1.0.0.rc1"
+end