Contents

Sequel.extension :pg_json, :pg_json_ops

module Mobility
  module Backend
    class Sequel::Jsonb::QueryMethods < Sequel::QueryMethods
      def initialize(attributes, **options)
        super
        attributes_extractor = @attributes_extractor

        define_method :_filter_or_exclude do |invert, clause, *conds, &block|
          if (clause == :where) && i18n_keys = attributes_extractor.call(conds.first)
            locale = Mobility.locale.to_s
            table_name = model.table_name
            cond = conds.first

            i18n_query = i18n_keys.inject(::Sequel.expr(!invert)) do |expr, attr|
              value = cond.delete(attr)
              attr_jsonb = ::Sequel.pg_jsonb_op(attr)
              contains_value = attr_jsonb.contains({ locale => value })
              has_key = attr_jsonb.has_key?(locale)
              if invert
                expr.|(has_key & ~contains_value)
              else
                expr.&(value.nil? ? ~has_key : contains_value)
              end
            end
            super(invert, clause, *conds, &block).where(i18n_query)
          else
            super(invert, clause, *conds, &block)
          end
        end

        attributes.each do |attribute|
          define_method :"first_by_#{attribute}" do |value|
            where(::Sequel.pg_jsonb_op(attribute).contains({ Mobility.locale => value })).
              select_all(model.table_name).first
          end
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
mobility-0.1.3	lib/mobility/backend/sequel/jsonb/query_methods.rb
mobility-0.1.2	lib/mobility/backend/sequel/jsonb/query_methods.rb
mobility-0.1.1	lib/mobility/backend/sequel/jsonb/query_methods.rb
mobility-0.1.0	lib/mobility/backend/sequel/jsonb/query_methods.rb