Implement custom audit model generator within install script

Author: amkisko

lib/generators/audited/install_generator.rb

@@ -23,6 +23,12 @@ class InstallGenerator < Rails::Generators::Base
       def copy_migration
         migration_template "install.rb", "db/migrate/install_audited.rb"
       end
+
+      def create_custom_audit
+        return if options[:audited_table_name] == "audits"
+
+        template "custom_audit.rb", "app/models/#{options[:audited_table_name].singularize.underscore}.rb"
+      end
     end
   end
 end

lib/generators/audited/templates/custom_audit.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+<% table_name = options[:audited_table_name].underscore.pluralize %>
+class <%= table_name.singularize.classify %> < Audited::Audit
+  self.table_name = "<%= table_name %>"
+end

lib/generators/audited/templates/install.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-<% table_name = options[:audited_table_name] %>
+<% table_name = options[:audited_table_name].underscore.pluralize %>
 class <%= migration_class_name %> < <%= migration_parent %>
   def self.up
     create_table :<%= table_name %>, :force => true do |t|

test/install_generator_test.rb

@@ -59,4 +59,12 @@ class InstallGeneratorTest < Rails::Generators::TestCase
       assert_includes(content, "class InstallAudited < ActiveRecord::Migration[#{ActiveRecord::Migration.current_version}]\n")
     end
   end
+
+  test "generate custom audit" do
+    run_generator %w[--audited-table-name custom_audits]
+
+    assert_file "app/models/custom_audit.rb" do |content|
+      assert_includes(content, "class CustomAudit < Audited::Audit")
+    end
+  end
 end