historiographer 4.1.7 → 4.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fdfddfba56fcb69bb91c73edcf7c1ddff76adc25f517d4b225bcf87bdd356540
-  data.tar.gz: 01e398edc375e2bcf17f22a81c4eb06ab2d507483b05db80b8014ea291f2ac82
+  metadata.gz: 28891ab31232a7eae8efccd015269069d992eb775259c6a6b9e738da1df69aed
+  data.tar.gz: 7a794f1a26351b47d194a0b8d18eb0222d49e9dd7d2b6cdcf90467d94cea62b0
 SHA512:
-  metadata.gz: 64b288645843cad4862b06724619ec8c98405745b4d6ba572d00baabbc1ce0d4f706cbf95b42c7e1e6214c870ac0c335fb37ea760f599830037f11a94cf5443e
-  data.tar.gz: '09663a6452efd27ce2f566b3764e7dd9f29499a61cb5eecc7b488e8fe31270d57a6ceb84489de983cbd4851ac8906cf6140127113ef06751fc6cb19839821906'
+  metadata.gz: 16ee4305c1aa8c6f09a13b042185afceea10fd18c0a19f7dba360bcab65a7f1a270dce78e24f15f2a30806fa4f4bcdf5ea6c10ecb81060b3dca02724918d7f5e
+  data.tar.gz: e66e7f2a31b5f65689f112ae6b43adeff5e286c42aeb6ed2ffc8931e1a75426dac3f4cb615b2fc06a537c74f070f4f2a245fe4668be8814b8382bd5084336651

data/lib/historiographer/history/instance_methods.rb

@@ -0,0 +1,98 @@
+module Historiographer
+  module History
+    module InstanceMethods
+      def destroy
+        false
+      end
+
+      def destroy!
+        false
+      end
+
+      def save(*args, **kwargs)
+        if persisted? && (changes.keys - %w(history_ended_at snapshot_id)).any?
+          false
+        else
+          super(*args, **kwargs)
+        end
+      end
+
+      def save!(*args, **kwargs)
+        if persisted? && (changes.keys - %w(history_ended_at snapshot_id)).any?
+          false
+        else
+          super(*args, **kwargs)
+        end
+      end
+
+      def snapshot
+        raise "Cannot snapshot a history model!"
+      end
+
+      def original_class
+        self.class.original_class
+      end
+
+      private
+
+      def dummy_instance
+        return @dummy_instance if @dummy_instance
+
+        cannot_keep_cols = %w(history_started_at history_ended_at history_user_id snapshot_id)
+        cannot_keep_cols += [self.class.inheritance_column.to_sym] if self.original_class.sti_enabled?
+        cannot_keep_cols += [self.class.history_foreign_key] 
+        cannot_keep_cols.map!(&:to_s)
+
+        attrs = attributes.clone
+        attrs[original_class.primary_key] = attrs[self.class.history_foreign_key]
+
+        instance = original_class.find_or_initialize_by(original_class.primary_key => attrs[original_class.primary_key])
+        instance.assign_attributes(attrs.except(*cannot_keep_cols))
+
+        # Create a module to hold methods from the history class
+        history_methods_module = Module.new
+
+        # Get methods defined directly in the history class, excluding baseline methods
+        history_methods = self.class.instance_methods(false) - baseline_methods
+
+        history_methods.each do |method_name|
+          next if instance.singleton_class.method_defined?(method_name)
+
+          method = self.class.instance_method(method_name)
+          history_methods_module.define_method(method_name) do |*args, &block|
+            method.bind(self.instance_variable_get(:@_history_instance)).call(*args, &block)
+          end
+        end
+
+        instance.singleton_class.prepend(history_methods_module)
+
+        self.class.reflect_on_all_associations.each do |reflection|
+          instance.singleton_class.class_eval do
+            define_method(reflection.name) do |*args, &block|
+              history_instance = instance.instance_variable_get(:@_history_instance)
+              history_instance.send(reflection.name, *args, &block)
+            end
+          end
+        end
+
+        instance.singleton_class.class_eval do
+          define_method(:class) do
+            history_instance = instance.instance_variable_get(:@_history_instance)
+            history_instance.class
+          end
+        end
+
+        instance.instance_variable_set(:@_history_instance, self)
+        @dummy_instance = instance
+      end
+
+      def forward_method(method_name, *args, &block)
+        if method_name == :class || method_name == 'class'
+          self.class
+        else
+          dummy_instance.send(method_name, *args, &block)
+        end
+      end
+    end
+  end
+end 

data/lib/historiographer/history.rb

@@ -61,7 +61,7 @@ module Historiographer
     extend ActiveSupport::Concern
 
     included do |base|
-      clear_validators!
+      clear_validators! if respond_to?(:clear_validators!)
       #
       # A History class (e.g. RetailerProductHistory) will gain
       # access to a current scope, returning
@@ -94,6 +94,7 @@ module Historiographer
 
       # Store the original class for method delegation
       class_variable_set(:@@original_class, foreign_class)
+      class_variable_set(:@@method_map, {})
 
       # Add method_added hook to the original class
       foreign_class.singleton_class.class_eval do
@@ -102,7 +103,6 @@ module Historiographer
           alias_method :original_method_added, :method_added
         end
 
-        method_map = Hash.new(0)
         define_method(:method_added) do |method_name|
           # Skip if we're already in the process of defining a method
           return if Thread.current[:defining_historiographer_method]
@@ -121,6 +121,7 @@ module Historiographer
             return if foreign_class.history_class.method_defined?(method_name)
 
             # Define the method in the history class
+            foreign_class.history_class.set_method_map(method_name, false)
             foreign_class.history_class.class_eval do
               define_method(method_name) do |*args, &block|
                 forward_method(method_name, *args, &block)
@@ -239,9 +240,30 @@ module Historiographer
         define_history_association(association)
       end
 
+      def snapshot
+        raise "Cannot snapshot a history model!"
+      end
+
     end
 
     class_methods do
+      def method_added(method_name)
+        set_method_map(method_name, true)
+      end
+
+      def set_method_map(method_name, is_overridden)
+        mm = method_map
+        mm[method_name.to_sym] = is_overridden
+        class_variable_set(:@@method_map, mm)
+      end
+
+      def method_map
+        unless class_variable_defined?(:@@method_map)
+          class_variable_set(:@@method_map, {})
+        end
+        class_variable_get(:@@method_map) || {}
+      end
+
       def original_class
         unless class_variable_defined?(:@@original_class)
           class_variable_set(:@@original_class, self.name.gsub(/History$/, '').constantize)
@@ -255,38 +277,38 @@ module Historiographer
           association = original_class.reflect_on_association(association)
         end
         assoc_name = association.name
+        assoc_module = association.active_record.module_parent
         assoc_history_class_name = "#{association.class_name}History"
-        assoc_foreign_key = association.foreign_key
 
-        # Skip if the association is already defined
-        return if method_defined?(assoc_name)
+        begin
+          assoc_module.const_get(assoc_history_class_name)
+          assoc_history_class_name = "#{assoc_module}::#{assoc_history_class_name}" unless assoc_history_class_name.match?(Regexp.new("#{assoc_module}::"))
+        rescue
+        end
+
+        assoc_foreign_key = association.foreign_key
 
         # Skip through associations to history classes to avoid infinite loops
         return if association.class_name.end_with?('History')
 
-        # We're writing a belongs_to
-        # The dataset belongs_to the datasource
-        # dataset#datasource_id => datasource.id
-        # 
-        # For the history class, we're writing a belongs_to
-        # the DatasetHistory belongs_to the DatasourceHistory
-        # dataset_history#datasource_id => datasource_history.easy_ml_datasource_id
-        #
-        # The missing piece for us here is whatever DatasourceHistory would call easy_ml_datasource_id (history foreign key?)
+        # Always use the history class if it exists
+        assoc_class = assoc_history_class_name.safe_constantize || OpenStruct.new(name: association.class_name)
+        assoc_class_name = assoc_class.name
+
+        # Define the scope to filter by snapshot_id for history associations
+        scope = if assoc_class_name.match?(/History/)
+                  ->(history_instance) { where(snapshot_id: history_instance.snapshot_id) }
+                else
+                  ->(history_instance) { all }
+                end
 
         case association.macro
         when :belongs_to
-          belongs_to assoc_name, ->(history_instance) {
-            where(snapshot_id: history_instance.snapshot_id)
-          }, class_name: assoc_history_class_name, foreign_key: assoc_foreign_key, primary_key: assoc_foreign_key
+          belongs_to assoc_name, scope, class_name: assoc_class_name, foreign_key: assoc_foreign_key, primary_key: assoc_foreign_key
         when :has_one
-          has_one assoc_name, ->(history_instance) {
-            where(snapshot_id: history_instance.snapshot_id)
-          }, class_name: assoc_history_class_name, foreign_key: assoc_foreign_key, primary_key: history_foreign_key
+          has_one assoc_name, scope, class_name: assoc_class_name, foreign_key: assoc_foreign_key, primary_key: history_foreign_key
         when :has_many
-          has_many assoc_name, ->(history_instance) {
-            where(snapshot_id: history_instance.snapshot_id)
-          }, class_name: assoc_history_class_name, foreign_key: assoc_foreign_key, primary_key: history_foreign_key
+          has_many assoc_name, scope, class_name: assoc_class_name, foreign_key: assoc_foreign_key, primary_key: history_foreign_key
         end
       end
       #
@@ -331,11 +353,52 @@ module Historiographer
 
       instance = original_class.find_or_initialize_by(original_class.primary_key => attrs[original_class.primary_key])
       instance.assign_attributes(attrs.except(*cannot_keep_cols))
+
+      # Filter out any methods that are not overridden on the history class
+      history_methods = self.class.instance_methods(false)
+      history_class_location = Module.const_source_location(self.class.name).first
+      history_methods.select! do |method| 
+        self.class.instance_method(method).source_location.first == history_class_location
+      end
+
+      history_methods.each do |method_name|
+        instance.singleton_class.class_eval do 
+          define_method(method_name) do |*args, &block|
+            history_instance = instance.instance_variable_get(:@_history_instance)
+            history_instance.send(method_name, *args, &block)
+          end
+        end
+      end
+
+      # For each association in the history class
+      self.class.reflect_on_all_associations.each do |reflection|
+        # Define a method that forwards to the history association
+        instance.singleton_class.class_eval do
+          define_method(reflection.name) do |*args, &block|
+            history_instance = instance.instance_variable_get(:@_history_instance)
+            history_instance.send(reflection.name, *args, &block)
+          end
+        end
+      end
+
+      # Override class method to return history class
+      instance.singleton_class.class_eval do
+        define_method(:class) do
+          history_instance = instance.instance_variable_get(:@_history_instance)
+          history_instance.class
+        end
+      end
+
+      instance.instance_variable_set(:@_history_instance, self)
       @dummy_instance = instance
     end
 
     def forward_method(method_name, *args, &block)
-      dummy_instance.send(method_name, *args, &block)
+      if method_name == :class || method_name == 'class'
+        self.class
+      else
+        dummy_instance.send(method_name, *args, &block)
+      end
     end
   end
 end

data/lib/historiographer/version.rb

@@ -1,3 +1,3 @@
 module Historiographer
-  VERSION = "4.1.7"
+  VERSION = "4.1.8"
 end

data/lib/historiographer.rb

@@ -216,21 +216,25 @@ module Historiographer
     base.singleton_class.prepend(Module.new do
       def belongs_to(name, scope = nil, **options, &extension)
         super
+        return if is_history_class?
         history_class.define_history_association(name)
       end
 
       def has_one(name, scope = nil, **options, &extension)
         super
+        return if is_history_class?
         history_class.define_history_association(name)
       end
 
       def has_many(name, scope = nil, **options, &extension)
         super
+        return if is_history_class?
         history_class.define_history_association(name)
       end
 
       def has_and_belongs_to_many(name, scope = nil, **options, &extension)
         super
+        return if is_history_class?
         history_class.define_history_association(name)
       end
     end)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: historiographer
 version: !ruby/object:Gem::Version
-  version: 4.1.7
+  version: 4.1.8
 platform: ruby
 authors:
 - brettshollenberger
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-11-20 00:00:00.000000000 Z
+date: 2024-11-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -220,6 +220,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: zeitwerk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Append-only histories + chained snapshots of your ActiveRecord tables
 email: brett.shollenberger@gmail.com
 executables: []
@@ -233,6 +247,7 @@ files:
 - lib/historiographer.rb
 - lib/historiographer/configuration.rb
 - lib/historiographer/history.rb
+- lib/historiographer/history/instance_methods.rb
 - lib/historiographer/history_migration.rb
 - lib/historiographer/history_migration_mysql.rb
 - lib/historiographer/mysql_migration.rb