Issy Long 6de61e4994
Ensure that stanza grouping works for nested stanzas with comments
- Since moving `comments_hash` to `Stanza`, we've been using the wrong
  kind of "comments": the comments for the _stanza_, not the comments
  for the entire Cask.
- Add a test to ensure this actually works. There was previously an
  infinite loop here due to the bad `comments`, visible in a `StanzaOrder`
  cop test, which I speculatively added a failing test for. Turns out
  that supporting nested stanza _ordering_ (vs. just grouping) is a
  whole separate piece of work (there are multiple TODOs there already),
  so I've backed that out and will do that separately.
2023-04-13 15:59:18 +01:00

82 lines
2.1 KiB
Ruby

# typed: true
# frozen_string_literal: true
require "forwardable"
module RuboCop
module Cask
module AST
# This class wraps the AST send/block node that encapsulates the method
# call that comprises the stanza. It includes various helper methods to
# aid cops in their analysis.
class Stanza
extend Forwardable
def initialize(method_node, all_comments)
@method_node = method_node
@all_comments = all_comments
end
attr_reader :method_node, :all_comments
alias stanza_node method_node
def_delegator :stanza_node, :parent, :parent_node
def_delegator :stanza_node, :arch_variable?
def source_range
stanza_node.location_expression
end
def source_range_with_comments
comments.reduce(source_range) do |range, comment|
range.join(comment.loc.expression)
end
end
def_delegator :source_range, :source
def_delegator :source_range_with_comments, :source,
:source_with_comments
def stanza_name
return :on_arch_conditional if arch_variable?
stanza_node.method_name
end
def stanza_group
Constants::STANZA_GROUP_HASH[stanza_name]
end
def same_group?(other)
stanza_group == other.stanza_group
end
def comments
@comments ||= stanza_node.each_node.reduce([]) do |comments, node|
comments | comments_hash[node.loc]
end
end
def comments_hash
@comments_hash ||= Parser::Source::Comment.associate_locations(stanza_node.parent, all_comments)
end
def ==(other)
self.class == other.class && stanza_node == other.stanza_node
end
alias eql? ==
Constants::STANZA_ORDER.each do |stanza_name|
class_eval <<-RUBY, __FILE__, __LINE__ + 1
def #{stanza_name}? # def url?
stanza_name == :#{stanza_name} # stanza_name == :url
end # end
RUBY
end
end
end
end
end