Skip to content

[Fix #1083] Add new Rails/BeforeDestroy cop #1451

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions changelog/new_rails_before_destroy_cop.md
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
* [#1083](https://github.com/rubocop/rubocop-rails/issues/1083): Add `Rails/BeforeDestroy` cop. ([@ecbrodie][], [@ydakuka][])
6 changes: 6 additions & 0 deletions config/default.yml
Original file line number Diff line number Diff line change
Expand Up @@ -240,6 +240,12 @@ Rails/AttributeDefaultBlockValue:
Include:
- 'app/models/**/*'

Rails/BeforeDestroy:
Description: 'Ensure the correct usage of `prepend: true` in `before_destroy` callbacks.'
Enabled: pending
Reference: 'https://guides.rubyonrails.org/active_record_callbacks.html#destroying-an-object'
VersionAdded: '<<next>>'

Rails/BelongsTo:
Description: >-
Use `optional: true` instead of `required: false` for
Expand Down
169 changes: 169 additions & 0 deletions lib/rubocop/cop/rails/before_destroy.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,169 @@
# frozen_string_literal: true

module RuboCop
module Cop
module Rails
# Ensures that `before_destroy` callbacks are executed
# before `dependent: :destroy` associations by requiring
# the `prepend: true` option.
#
# Without `prepend: true`, `before_destroy` callbacks may run
# after associated records are already deleted, leading to
# unintended behavior.
#
# @example
# # bad
# has_many :entities, dependent: :destroy
# before_destroy { do_something }
#
# # good
# has_many :entities, dependent: :destroy
# before_destroy(prepend: true) { do_something }
#
# @example
# # bad
# belongs_to :entity, dependent: :destroy
# before_destroy :some_method
#
# # good
# belongs_to :entity, dependent: :destroy
# before_destroy :some_method, prepend: true
#
# @example
# # bad
# has_one :entity, dependent: :destroy
# before_destroy MyClass.new
#
# # good
# has_one :entity, dependent: :destroy
# before_destroy MyClass.new, prepend: true
#
# @example
# # bad
# has_many :entities, dependent: :destroy
# before_destroy -> { do_something }
#
# # good
# has_many :entities, dependent: :destroy
# before_destroy -> { do_something }, prepend: true
#
class BeforeDestroy < Base
extend AutoCorrector

MSG = '"before_destroy" callbacks must be declared before "dependent: :destroy" associations ' \
'or use `prepend: true`.'
RESTRICT_ON_SEND = %i[before_destroy].freeze

def_node_search :association_nodes, <<~PATTERN
(send nil? {:belongs_to :has_one :has_many} _ (hash ...))
PATTERN

def_node_matcher :hash_options, <<~PATTERN
`(hash $...)
PATTERN

def_node_matcher :dependent_destroy?, <<~PATTERN
(pair (sym :dependent) (sym :destroy))
PATTERN

def_node_matcher :prepend_true?, <<~PATTERN
(pair (sym :prepend) true)
PATTERN

def on_send(node)
check_add_prepend_true(node)
check_remove_prepend_true(node)
end

private

def check_add_prepend_true(node)
return if contains_prepend_true?(node)
return unless before_association_with_dependent_destroy?(node)

add_offense(node) { |corrector| autocorrect_add_prepend(corrector, node) }
end

def check_remove_prepend_true(node)
return unless contains_prepend_true?(node)
return if before_any_association?(node)

add_offense(node) { |corrector| autocorrect_remove_prepend(corrector, node) }
end

def autocorrect_remove_prepend(corrector, node)
prepend_pair = hash_options(node).find { |pair| prepend_true?(pair) }
prepend_range = prepend_pair.source_range
start_pos, end_pos = adjust_removal_range(prepend_range)

corrector.remove(prepend_range.with(begin_pos: start_pos, end_pos: end_pos))
remove_block_delimiters(corrector, node)
end

def autocorrect_add_prepend(corrector, node)
hash_node = node.arguments.find(&:hash_type?)

if hash_node
corrector.insert_before(hash_node.children.first, 'prepend: true, ')
elsif node.arguments.empty?
corrector.insert_after(node.loc.selector, '(prepend: true)')
else
corrector.insert_after(node.last_argument, ', prepend: true')
end
end

def adjust_removal_range(prepend_range)
start_pos = prepend_range.begin_pos
end_pos = prepend_range.end_pos

source = processed_source.buffer.source

prev_match = source[0...start_pos].match(/,\s*$/)
next_match = source[end_pos..].match(/^\s*,?/)

if prev_match
start_pos = prev_match.begin(0)
elsif next_match
end_pos += next_match.end(0)
end

[start_pos, end_pos]
end

def remove_block_delimiters(corrector, node)
return unless node.block_literal?
return unless node.loc.begin

corrector.remove(node.loc.begin)
corrector.remove(node.loc.end)
end

def before_association_with_dependent_destroy?(node)
root_class_node = find_root_class(node)
association_nodes(root_class_node).any? do |assoc|
contains_dependent_destroy?(assoc) && assoc.first_line < node.first_line
end
end

def before_any_association?(node)
root_class_node = find_root_class(node)
association_nodes(root_class_node).any? do |assoc|
assoc.first_line < node.first_line
end
end

def find_root_class(node)
node.each_ancestor(:class, :module).first
end

def contains_prepend_true?(node)
hash_options(node)&.any? { |pair| prepend_true?(pair) }
end

def contains_dependent_destroy?(node)
hash_options(node).any? { |pair| dependent_destroy?(pair) }
end
end
end
end
end
1 change: 1 addition & 0 deletions lib/rubocop/cop/rails_cops.rb
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@
require_relative 'rails/arel_star'
require_relative 'rails/assert_not'
require_relative 'rails/attribute_default_block_value'
require_relative 'rails/before_destroy'
require_relative 'rails/belongs_to'
require_relative 'rails/blank'
require_relative 'rails/bulk_change_table'
Expand Down
Loading