kotovalexarian-likes-github/rails--rails
1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
Code Releases Activity

Add attachment and attachments field generators

Browse source
This commit is contained in:
Chris Oliver 2019-03-30 11:54:47 -05:00
parent 4e076b03b6
commit ecdcf06cb2
13 changed files with 130 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
activerecord/lib/rails/generators/active_record/model/templates/model.rb.tt
View file

@ -6,6 +6,12 @@ class <%= class_name %> < <%= parent_class_name.classify %>
<% attributes.select(&:rich_text?).each do |attribute| -%>
has_rich_text :<%= attribute.name %>
<% end -%>
<% attributes.select(&:attachment?).each do |attribute| -%>
has_one_attached :<%= attribute.name %>
<% end -%>
<% attributes.select(&:attachments?).each do |attribute| -%>
has_many_attached :<%= attribute.name %>
<% end -%>
<% attributes.select(&:token?).each do |attribute| -%>
has_secure_token<% if attribute.name != "token" %> :<%= attribute.name %><% end %>
<% end -%>

3
railties/lib/rails/generators/erb/scaffold/templates/_form.html.erb.tt
View file

@ -21,6 +21,9 @@
<div class="field">
<%%= form.label :password_confirmation %>
<%%= form.password_field :password_confirmation %>
<% elsif attribute.attachments? -%>
<%%= form.label :<%= attribute.column_name %> %>
<%%= form.<%= attribute.field_type %> :<%= attribute.column_name %>, multiple: true %>
<% else -%>
<%%= form.label :<%= attribute.column_name %> %>
<%%= form.<%= attribute.field_type %> :<%= attribute.column_name %> %>

8
railties/lib/rails/generators/erb/scaffold/templates/show.html.erb.tt
View file

@ -3,7 +3,15 @@
<% attributes.reject(&:password_digest?).each do |attribute| -%>
<p>
<strong><%= attribute.human_name %>:</strong>
<% if attribute.attachment? -%>
<%%= link_to @<%= singular_table_name %>.<%= attribute.column_name %>.filename, @<%= singular_table_name %>.<%= attribute.column_name %> %>
<% elsif attribute.attachments? -%>
<%% @<%= singular_table_name %>.<%= attribute.column_name %>.each do |<%= attribute.singular_name %>| %>
<div><%%= link_to <%= attribute.singular_name %>.filename, <%= attribute.singular_name %> %></div>
<%% end %>
<% else -%>
<%%= @<%= singular_table_name %>.<%= attribute.column_name %> %>
<% end -%>
</p>
<% end -%>

47
railties/lib/rails/generators/generated_attribute.rb
View file

@ -68,14 +68,15 @@ module Rails
def field_type
@field_type ||= case type
when :integer then :number_field
when :float, :decimal then :text_field
when :time then :time_select
when :datetime, :timestamp then :datetime_select
when :date then :date_select
when :text then :text_area
when :rich_text then :rich_text_area
when :boolean then :check_box
when :integer then :number_field
when :float, :decimal then :text_field
when :time then :time_select
when :datetime, :timestamp then :datetime_select
when :date then :date_select
when :text then :text_area
when :rich_text then :rich_text_area
when :boolean then :check_box
when :attachment, :attachments then :file_field
else
:text_field
end
@ -83,15 +84,17 @@ module Rails
def default
@default ||= case type
when :integer then 1
when :float then 1.5
when :decimal then "9.99"
when :datetime, :timestamp, :time then Time.now.to_s(:db)
when :date then Date.today.to_s(:db)
when :string then name == "type" ? "" : "MyString"
when :text then "MyText"
when :boolean then false
when :references, :belongs_to, :rich_text then nil
when :integer then 1
when :float then 1.5
when :decimal then "9.99"
when :datetime, :timestamp, :time then Time.now.to_s(:db)
when :date then Date.today.to_s(:db)
when :string then name == "type" ? "" : "MyString"
when :text then "MyText"
when :boolean then false
when :references, :belongs_to,
:attachment, :attachments,
:rich_text then nil
else
""
end
@ -157,8 +160,16 @@ module Rails
type == :rich_text
end
def attachment?
type == :attachment
end
def attachments?
type == :attachments
end
def virtual?
rich_text?
rich_text? || attachment? || attachments?
end
def inject_options

1
railties/lib/rails/generators/named_base.rb
View file

@ -187,6 +187,7 @@ module Rails
def attributes_names # :doc:
@attributes_names ||= attributes.each_with_object([]) do |a, names|
next if a.attachments?
names << a.column_name
names << "password_confirmation" if a.password_digest?
names << "#{a.name}_type" if a.polymorphic?

8
railties/lib/rails/generators/rails/scaffold_controller/scaffold_controller_generator.rb
View file

@ -32,6 +32,14 @@ module Rails
hook_for :helper, as: :scaffold do |invoked|
invoke invoked, [ controller_name ]
end
private
def permitted_params
params = attributes_names.map { |name| ":#{name}" }.join(", ")
params += attributes.select(&:attachments?).map { |a| ", #{a.name}: []" }.join
params
end
end
end
end

2
railties/lib/rails/generators/rails/scaffold_controller/templates/api_controller.rb.tt
View file

@ -54,7 +54,7 @@ class <%= controller_class_name %>Controller < ApplicationController
<%- if attributes_names.empty? -%>
params.fetch(:<%= singular_table_name %>, {})
<%- else -%>
params.require(:<%= singular_table_name %>).permit(<%= attributes_names.map { |name| ":#{name}" }.join(', ') %>)
params.require(:<%= singular_table_name %>).permit(<%= permitted_params %>)
<%- end -%>
end
end

2
railties/lib/rails/generators/rails/scaffold_controller/templates/controller.rb.tt
View file

@ -61,7 +61,7 @@ class <%= controller_class_name %>Controller < ApplicationController
<%- if attributes_names.empty? -%>
params.fetch(:<%= singular_table_name %>, {})
<%- else -%>
params.require(:<%= singular_table_name %>).permit(<%= attributes_names.map { |name| ":#{name}" }.join(', ') %>)
params.require(:<%= singular_table_name %>).permit(<%= permitted_params %>)
<%- end -%>
end
end

8
railties/test/generators/generated_attribute_test.rb
View file

@ -42,6 +42,12 @@ class GeneratedAttributeTest < Rails::Generators::TestCase
assert_field_type :rich_text, :rich_text_area
end
def test_field_type_returns_file_field
%w(attachment attachments).each do |attribute_type|
assert_field_type attribute_type, :file_field
end
end
def test_field_type_with_unknown_type_returns_text_field
%w(foo bar baz).each do |attribute_type|
assert_field_type attribute_type, :text_field
@ -88,7 +94,7 @@ class GeneratedAttributeTest < Rails::Generators::TestCase
end
def test_default_value_is_nil
%w(references belongs_to rich_text).each do |attribute_type|
%w(references belongs_to rich_text attachment attachments).each do |attribute_type|
assert_field_default_value attribute_type, nil
end
end

12
railties/test/generators/migration_generator_test.rb
View file

@ -368,32 +368,38 @@ class MigrationGeneratorTest < Rails::Generators::TestCase
def test_add_migration_ignores_virtual_attributes
migration = "add_rich_text_content_to_messages"
run_generator [migration, "content:rich_text"]
run_generator [migration, "content:rich_text", "video:attachment", "photos:attachments"]
assert_migration "db/migrate/#{migration}.rb" do |content|
assert_method :change, content do |change|
assert_no_match(/add_column :messages, :content, :rich_text/, change)
assert_no_match(/add_column :messages, :video, :attachment/, change)
assert_no_match(/add_column :messages, :photos, :attachments/, change)
end
end
end
def test_create_table_migration_ignores_virtual_attributes
run_generator ["create_messages", "content:rich_text"]
run_generator ["create_messages", "content:rich_text", "video:attachment", "photos:attachments"]
assert_migration "db/migrate/create_messages.rb" do |content|
assert_method :change, content do |change|
assert_match(/create_table :messages/, change)
assert_no_match(/ t\.rich_text :content/, change)
assert_no_match(/ t\.attachment :video/, change)
assert_no_match(/ t\.attachments :photos/, change)
end
end
end
def test_remove_migration_with_virtual_attributes
migration = "remove_content_from_messages"
run_generator [migration, "content:rich_text"]
run_generator [migration, "content:rich_text", "video:attachment", "photos:attachments"]
assert_migration "db/migrate/#{migration}.rb" do |content|
assert_method :change, content do |change|
assert_no_match(/remove_column :messages, :content, :rich_text/, change)
assert_no_match(/remove_column :messages, :video, :attachment/, change)
assert_no_match(/remove_column :messages, :photos, :attachments/, change)
end
end
end

22
railties/test/generators/model_generator_test.rb
View file

@ -509,8 +509,28 @@ class ModelGeneratorTest < Rails::Generators::TestCase
assert_file "app/models/message.rb", expected_file
end
def test_model_with_attachment_attribute_adds_has_one_attached
run_generator ["message", "video:attachment"]
expected_file = <<~FILE
class Message < ApplicationRecord
has_one_attached :video
end
FILE
assert_file "app/models/message.rb", expected_file
end
def test_model_with_attachments_attribute_adds_has_many_attached
run_generator ["message", "photos:attachments"]
expected_file = <<~FILE
class Message < ApplicationRecord
has_many_attached :photos
end
FILE
assert_file "app/models/message.rb", expected_file
end
def test_skip_virtual_fields_in_fixtures
run_generator ["message", "content:rich_text"]
run_generator ["message", "content:rich_text", "video:attachment", "photos:attachments"]
assert_generated_fixture("test/fixtures/messages.yml",
"one" => nil, "two" => nil)

18
railties/test/generators/scaffold_controller_generator_test.rb
View file

@ -80,6 +80,15 @@ class ScaffoldControllerGeneratorTest < Rails::Generators::TestCase
end
end
def test_controller_permit_attachment_attributes
run_generator ["Message", "video:attachment", "photos:attachments"]
assert_file "app/controllers/messages_controller.rb" do |content|
assert_match(/def message_params/, content)
assert_match(/params\.require\(:message\)\.permit\(:video, photos: \[\]\)/, content)
end
end
def test_helper_are_invoked_with_a_pluralized_name
run_generator
assert_file "app/helpers/users_helper.rb", /module UsersHelper/
@ -276,4 +285,13 @@ class ScaffoldControllerGeneratorTest < Rails::Generators::TestCase
assert_no_match(/assert_redirected_to/, content)
end
end
def test_api_only_generates_params_for_attachments
run_generator ["Message", "video:attachment", "photos:attachments", "--api"]
assert_file "app/controllers/messages_controller.rb" do |content|
assert_match(/def message_params/, content)
assert_match(/params\.require\(:message\)\.permit\(:video, photos: \[\]\)/, content)
end
end
end

18
railties/test/generators/scaffold_generator_test.rb
View file

@ -471,6 +471,24 @@ class ScaffoldGeneratorTest < Rails::Generators::TestCase
end
end
def test_scaffold_generator_attachments
run_generator ["message", "video:attachment", "photos:attachments", "images:attachments"]
assert_file "app/models/message.rb", /has_one_attached :video/
assert_file "app/models/message.rb", /has_many_attached :photos/
assert_file "app/controllers/messages_controller.rb" do |content|
assert_instance_method :message_params, content do |m|
assert_match(/permit\(:video, photos: \[\], images: \[\]\)/, m)
end
end
assert_file "app/views/messages/_form.html.erb" do |content|
assert_match(/^\W{4}<%= form\.file_field :video %>/, content)
assert_match(/^\W{4}<%= form\.file_field :photos, multiple: true %>/, content)
end
end
def test_scaffold_generator_database
with_secondary_database_configuration do
run_generator ["posts", "--database=secondary"]