respect rails application record setup

Author: thedumbtechguy

lib/chrono_forge.rb

@@ -11,4 +11,6 @@ module ChronoForge
   end
 
   class Error < StandardError; end
+  
+  def self.ApplicationRecord() = defined?(::ApplicationRecord) ? ::ApplicationRecord : ActiveRecord::Base
 end

lib/chrono_forge/error_log.rb

@@ -23,7 +23,7 @@
 #
 
 module ChronoForge
-  class ErrorLog < ActiveRecord::Base
+  class ErrorLog < ApplicationRecord()
     self.table_name = "chrono_forge_error_logs"
 
     belongs_to :workflow

lib/chrono_forge/execution_log.rb

@@ -28,7 +28,7 @@
 #  workflow_id  (workflow_id => chrono_forge_workflows.id)
 #
 module ChronoForge
-  class ExecutionLog < ActiveRecord::Base
+  class ExecutionLog < ApplicationRecord()
     self.table_name = "chrono_forge_execution_logs"
 
     belongs_to :workflow

lib/chrono_forge/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ChronoForge
-  VERSION = "0.4.0"
+  VERSION = "0.5.0"
 end

lib/chrono_forge/workflow.rb

@@ -22,7 +22,7 @@
 #  index_chrono_forge_workflows_on_key  (key) UNIQUE
 #
 module ChronoForge
-  class Workflow < ActiveRecord::Base
+  class Workflow < ApplicationRecord()
     self.table_name = "chrono_forge_workflows"
 
     has_many :execution_logs, -> { order(id: :asc) }, dependent: :destroy

lib/generators/chrono_forge/install/templates/install_chrono_forge.rb

@@ -2,7 +2,7 @@
 
 class InstallChronoForge < ActiveRecord::Migration[7.1]
   def change
-    create_table :chrono_forge_workflows do |t|
+    create_table :chrono_forge_workflows, id: primary_key_type do |t|
       t.string :key, null: false, index: true
       t.string :job_class, null: false
 
@@ -27,18 +27,23 @@ def change
       t.index %i[job_class key], unique: true
     end
 
-    create_table :chrono_forge_execution_logs do |t|
-      t.references :workflow, null: false, foreign_key: {to_table: :chrono_forge_workflows}
+    create_table :chrono_forge_execution_logs, id: primary_key_type do |t|
+      t.references :workflow, null: false, 
+                  foreign_key: {to_table: :chrono_forge_workflows},
+                  type: reference_type
+
       t.string :step_name, null: false
       t.integer :attempts, null: false, default: 0
       t.datetime :started_at
       t.datetime :last_executed_at
       t.datetime :completed_at
+      
       if t.respond_to?(:jsonb)
         t.jsonb :metadata
       else
         t.json :metadata
       end
+      
       t.integer :state, null: false, default: 0
       t.string :error_class
       t.text :error_message
@@ -47,11 +52,15 @@ def change
       t.index %i[workflow_id step_name], unique: true
     end
 
-    create_table :chrono_forge_error_logs do |t|
-      t.references :workflow, null: false, foreign_key: {to_table: :chrono_forge_workflows}
+    create_table :chrono_forge_error_logs, id: primary_key_type do |t|
+      t.references :workflow, null: false, 
+                  foreign_key: {to_table: :chrono_forge_workflows},
+                  type: reference_type
+                  
       t.string :error_class
       t.text :error_message
       t.text :backtrace
+      
       if t.respond_to?(:jsonb)
         t.jsonb :context
       else
@@ -61,4 +70,35 @@ def change
       t.timestamps
     end
   end
-end
+
+  private
+
+  def primary_key_type
+    # Check if the application is configured to use UUIDs
+    if ActiveRecord.respond_to?(:default_id) && ActiveRecord.default_id.respond_to?(:to_s) &&
+      ActiveRecord.default_id.to_s.include?('uuid')
+      return :uuid
+    end
+    
+    # Rails 6+ configuration style
+    if ActiveRecord.respond_to?(:primary_key_type) && 
+      ActiveRecord.primary_key_type.to_s == 'uuid'
+      return :uuid
+    end
+    
+    # Check application config
+    app_config = Rails.application.config.generators
+    if app_config.options.key?(:active_record) &&
+      app_config.options[:active_record].key?(:primary_key_type) &&
+      app_config.options[:active_record][:primary_key_type].to_s == 'uuid'
+      return :uuid
+    end
+    
+    # Default to traditional integer keys
+    return :bigint
+  end
+
+  def reference_type
+    primary_key_type == :uuid ? :uuid : nil
+  end
+end