Contents

# frozen_string_literal: true

require "rails/generators/active_record/model/model_generator"

module Koi
  class ActiveRecordGenerator < ActiveRecord::Generators::ModelGenerator
    source_root ActiveRecord::Generators::ModelGenerator.source_root

    def admin_search
      "PgSearch::Model".safe_constantize ? pg_search : sql_search
    end

    private

    def pg_search
      insert_into_file "app/models/#{file_name}.rb", after: "class #{class_name} < ApplicationRecord\n" do
        <<~RUBY
          include PgSearch::Model
        RUBY
      end

      insert_into_file "app/models/#{file_name}.rb", before: "end\n" do
        <<~RUBY
          pg_search :admin_search, against: %i[#{search_fields.join(' ')}], using: { tsearch: { prefix: true } }
        RUBY
      end
    end

    def sql_search
      insert_into_file "app/models/#{file_name}.rb", before: "end\n" do
        <<~RUBY
          scope :admin_search, ->(query) do
            where("#{search_fields.map { |f| "#{f} LIKE :query" }.join(' OR ')}", query: "%\#{query}%")
          end
        RUBY
      end
    end

    def search_fields
      attributes.select { |attr| attr.type == :string }.map(&:name)
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
katalyst-koi-4.0.3	lib/generators/koi/active_record/active_record_generator.rb
katalyst-koi-4.0.2	lib/generators/koi/active_record/active_record_generator.rb
katalyst-koi-4.0.1	lib/generators/koi/active_record/active_record_generator.rb
katalyst-koi-4.0.0	lib/generators/koi/active_record/active_record_generator.rb