Defer inclusion until after active record has fully loaded

Author: westonganger

CHANGELOG.md

@@ -1,7 +1,8 @@
 # CHANGELOG
 
 ### Unreleased - [View Diff](https://github.com/westonganger/paper_trail-association_tracking/compare/v2.3.0...master)
-- [#54](https://github.com/westonganger/paper_trail-association_tracking/pull/54) - Change migration for `versions.transaction_id` to BIGINT 
+- [#56](https://github.com/westonganger/paper_trail-association_tracking/pull/56) - Defer inclusion until after active record has fully loaded
+- [#54](https://github.com/westonganger/paper_trail-association_tracking/pull/54) - Change migration for `versions.transaction_id` to BIGINT
 
 ### v2.3.0 - 2025-04-17 - [View Diff](https://github.com/westonganger/paper_trail-association_tracking/compare/v2.2.1...v2.3.0)
 - [#46](https://github.com/westonganger/paper_trail-association_tracking/pull/46) - Add support for custom version association class with separate database connection

lib/paper_trail-association_tracking.rb

@@ -7,14 +7,14 @@
 require "paper_trail_association_tracking/record_trail"
 require "paper_trail_association_tracking/request"
 require "paper_trail_association_tracking/paper_trail"
-require "paper_trail_association_tracking/version_concern"
 
 if defined?(Rails)
-  require "paper_trail/frameworks/active_record"
-  require "paper_trail_association_tracking/frameworks/rails"
-elsif defined?(ActiveRecord)
-  require "paper_trail/frameworks/active_record"
-  require "paper_trail_association_tracking/frameworks/active_record"
+  require "paper_trail_association_tracking/frameworks/rails/railtie"
+else
+  ActiveSupport.on_load(:active_record) do
+    require "paper_trail/frameworks/active_record"
+    require "paper_trail_association_tracking/frameworks/active_record"
+  end
 end
 
 module PaperTrailAssociationTracking

lib/paper_trail_association_tracking/frameworks/active_record.rb

@@ -1,3 +1,5 @@
 # frozen_string_literal: true
 
 require "paper_trail_association_tracking/frameworks/active_record/models/paper_trail/version_association"
+
+require "paper_trail_association_tracking/version_concern"

lib/paper_trail_association_tracking/frameworks/rails.rb

@@ -5,13 +5,10 @@ class Railtie < ::Rails::Railtie
 
     initializer "paper_trail_association_tracking", after: "paper_trail" do
       ActiveSupport.on_load(:active_record) do
+        require "paper_trail/frameworks/active_record"
         require "paper_trail_association_tracking/frameworks/active_record"
       end
     end
 
-    config.to_prepare do
-      ::PaperTrail::Version.include(::PaperTrailAssociationTracking::VersionConcern)
-    end
-
   end
 end

lib/paper_trail_association_tracking/frameworks/rails/railtie.rb

@@ -16,3 +16,5 @@ module VersionConcern
     end
   end
 end
+
+PaperTrail::Version.include(PaperTrailAssociationTracking::VersionConcern)