lib/mobility/backends/sequel/jsonb/query_methods.rb in mobility-0.3.6 vs lib/mobility/backends/sequel/jsonb/query_methods.rb in mobility-0.4.0

@@ -1,25 +1,39 @@
-require 'mobility/backends/sequel/postgres_query_methods'
+require 'mobility/backends/sequel/pg_query_methods'
 require "mobility/backends/sequel/query_methods"
 
 Sequel.extension :pg_json, :pg_json_ops
 
 module Mobility
   module Backends
     class Sequel::Jsonb::QueryMethods < Sequel::QueryMethods
-      include PostgresQueryMethods
+      include Sequel::PgQueryMethods
 
       def initialize(attributes, _)
         super
 
-        define_query_methods("pg_jsonb")
+        define_query_methods
 
         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
+
+      private
+
+      def contains_value(key, value, locale)
+        build_op(key).contains({ locale => value }.to_json)
+      end
+
+      def has_locale(key, locale)
+        build_op(key).has_key?(locale)
+      end
+
+      def build_op(key)
+        ::Sequel.pg_jsonb_op(key)
       end
     end
   end
 end