Added primary key type to generated migration - resolves #57

Author: ankane

lib/generators/authtrail/install_generator.rb

@@ -78,6 +78,18 @@ def mysql?
       def adapter
         ActiveRecord::Base.connection_db_config.adapter.to_s
       end
+
+      def primary_key_type
+        ", id: :#{key_type}" if key_type
+      end
+
+      def foreign_key_type
+        ", type: :#{key_type}" if key_type
+      end
+
+      def key_type
+        Rails.configuration.generators.options.dig(:active_record, :primary_key_type)
+      end
     end
   end
 end

lib/generators/authtrail/templates/login_activities_migration.rb.tt

@@ -1,12 +1,12 @@
 class <%= migration_class_name %> < ActiveRecord::Migration<%= migration_version %>
   def change
-    create_table :login_activities do |t|
+    create_table :login_activities<%= primary_key_type %> do |t|
       t.string :scope
       t.string :strategy
       <%= identity_column %>
       t.boolean :success
       t.string :failure_reason
-      t.references :user, polymorphic: true
+      t.references :user<%= foreign_key_type %>, polymorphic: true
       t.string :context
       <%= ip_column %>
       t.text :user_agent

test/install_generator_test.rb

@@ -28,4 +28,12 @@ def test_encryption_none
     assert_file "app/models/login_activity.rb", /LoginActivity < ApplicationRecord/
     assert_migration "db/migrate/create_login_activities.rb", /t.string :identity, index: true/
   end
+
+  def test_primary_key_type
+    Rails.configuration.generators.stub(:options, {active_record: {primary_key_type: :uuid}}) do
+      run_generator ["--encryption=lockbox"]
+    end
+    assert_migration "db/migrate/create_login_activities.rb", /id: :uuid/
+    assert_migration "db/migrate/create_login_activities.rb", /type: :uuid/
+  end
 end