lib/mobility/backends/sequel.rb in mobility-1.0.0.beta2 vs lib/mobility/backends/sequel.rb in mobility-1.0.0.rc1

@@ -28,9 +28,41 @@
         # @param [Symbol] locale Locale
         # @return [Sequel::Dataset] Prepared dataset
         def prepare_dataset(dataset, _predicate, _locale)
           dataset
         end
+
+        # Forces Sequel to notice changes when Mobility setter method is
+        # called.
+        # TODO: Find a better way to do this.
+        def define_column_changes(mod, attributes, column_affix: "%s")
+          mod.class_eval do
+            attributes.each do |attribute|
+              define_method "#{attribute}=" do |value, **options|
+                if !options[:super] && send(attribute) != value
+                  locale = options[:locale] || Mobility.locale
+                  column = (column_affix % attribute).to_sym
+                  attribute_with_locale = :"#{attribute}_#{Mobility.normalize_locale(locale)}"
+                  @changed_columns = changed_columns | [column, attribute.to_sym, attribute_with_locale]
+                end
+                super(value, **options)
+              end
+            end
+          end
+        end
+
+        # Initialize column value(s) by default to a hash.
+        # TODO: Find a better way to do this.
+        def define_hash_initializer(mod, columns)
+          mod.class_eval do
+            class_eval <<-EOM, __FILE__, __LINE__ + 1
+              def initialize_set(values)
+                #{columns.map { |c| "self[:#{c}] = {}" }.join(';')}
+                super
+              end
+            EOM
+          end
+        end
       end
     end
   end
 end