File: attribute_query_methods.rb

package info (click to toggle)
ruby-jsonb-accessor 1.4-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 204 kB
  • sloc: ruby: 371; makefile: 10; sh: 5
file content (42 lines) | stat: -rw-r--r-- 1,621 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# frozen_string_literal: true

module JsonbAccessor
  class AttributeQueryMethods
    def initialize(klass)
      @klass = klass
    end

    def define(store_key_mapping_method_name, jsonb_attribute)
      return if klass.superclass.respond_to? store_key_mapping_method_name

      # <jsonb_attribute>_where scope
      klass.define_singleton_method "#{jsonb_attribute}_where" do |attributes|
        store_key_attributes = JsonbAccessor::Helpers.convert_keys_to_store_keys(attributes, all.model.public_send(store_key_mapping_method_name))
        jsonb_where(jsonb_attribute, store_key_attributes)
      end

      # <jsonb_attribute>_where_not scope
      klass.define_singleton_method "#{jsonb_attribute}_where_not" do |attributes|
        store_key_attributes = JsonbAccessor::Helpers.convert_keys_to_store_keys(attributes, all.model.public_send(store_key_mapping_method_name))
        jsonb_where_not(jsonb_attribute, store_key_attributes)
      end

      # <jsonb_attribute>_order scope
      klass.define_singleton_method "#{jsonb_attribute}_order" do |*args|
        ordering_options = args.extract_options!
        order_by_defaults = args.each_with_object({}) { |attribute, config| config[attribute] = :asc }
        store_key_mapping = all.model.public_send(store_key_mapping_method_name)

        order_by_defaults.merge(ordering_options).reduce(all) do |query, (name, direction)|
          key = store_key_mapping[name.to_s]
          order_query = jsonb_order(jsonb_attribute, key, direction)
          query.merge(order_query)
        end
      end
    end

    private

    attr_reader :klass
  end
end