From 04d66727675184c97f645ec9f76658c0e2023c23 Mon Sep 17 00:00:00 2001
From: Teddy Valente <teddy.valente@gmail.com>
Date: Fri, 12 May 2023 16:37:08 +0000
Subject: [PATCH] Add action text attributes

---
 lib/model_probe.rb                     | 4 ++++
 lib/model_probe/templates/model.rb.erb | 5 +++++
 2 files changed, 9 insertions(+)

diff --git a/lib/model_probe.rb b/lib/model_probe.rb
index 1b50ccb..590f2cc 100644
--- a/lib/model_probe.rb
+++ b/lib/model_probe.rb
@@ -112,6 +112,10 @@ def fixture_columns
     end
   end
 
+  def action_text_relations
+    name.constantize.rich_text_association_names
+  end
+
   def ddl
     config = connection_db_config.configuration_hash
     @ddl ||= begin
diff --git a/lib/model_probe/templates/model.rb.erb b/lib/model_probe/templates/model.rb.erb
index 44827d3..1a6fb8d 100644
--- a/lib/model_probe/templates/model.rb.erb
+++ b/lib/model_probe/templates/model.rb.erb
@@ -16,6 +16,11 @@ class <%= name -%> < <%= superclass_name %>
   <%= "belongs_to :#{column.name.sub(/_id\z/, "")}" %>
 <% end -%>
 
+  # Action text attributes ....................................................................................
+<% action_text_relations.each do |column| -%>
+  <%= "has_rich_text :#{column.name.sub(/_id\z/, "")}" %>
+<% end -%>
+
   # validations ..............................................................................................
 <% validation_columns.each do |column| -%>
   <%= "validates :#{column.name}, presence: true" unless column.null if required_columns.include?(column) %>