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) %>