brew/Library/Homebrew/rubocops/cask/stanza_grouping.rb
2020-11-05 23:50:55 -05:00

101 lines
2.8 KiB
Ruby

# typed: true
# frozen_string_literal: true
require "forwardable"
module RuboCop
module Cop
module Cask
# This cop checks that a cask's stanzas are grouped correctly.
# @see https://github.com/Homebrew/homebrew-cask/blob/HEAD/doc/cask_language_reference/readme.md#stanza-order
class StanzaGrouping < Cop
extend Forwardable
include CaskHelp
include RangeHelp
MISSING_LINE_MSG = "stanza groups should be separated by a single " \
"empty line"
EXTRA_LINE_MSG = "stanzas within the same group should have no lines " \
"between them"
def on_cask(cask_block)
@cask_block = cask_block
@line_ops = {}
add_offenses
end
def autocorrect(range)
lambda do |corrector|
case line_ops[range.line - 1]
when :insert
corrector.insert_before(range, "\n")
when :remove
corrector.remove(range)
end
end
end
private
attr_reader :cask_block, :line_ops
def_delegators :cask_block, :cask_node, :toplevel_stanzas
def add_offenses
toplevel_stanzas.each_cons(2) do |stanza, next_stanza|
next unless next_stanza
if missing_line_after?(stanza, next_stanza)
add_offense_missing_line(stanza)
elsif extra_line_after?(stanza, next_stanza)
add_offense_extra_line(stanza)
end
end
end
def missing_line_after?(stanza, next_stanza)
!(stanza.same_group?(next_stanza) ||
empty_line_after?(stanza))
end
def extra_line_after?(stanza, next_stanza)
stanza.same_group?(next_stanza) &&
empty_line_after?(stanza)
end
def empty_line_after?(stanza)
source_line_after(stanza).empty?
end
def source_line_after(stanza)
processed_source[index_of_line_after(stanza)]
end
def index_of_line_after(stanza)
stanza.source_range.last_line
end
def add_offense_missing_line(stanza)
line_index = index_of_line_after(stanza)
line_ops[line_index] = :insert
add_offense(line_index, message: MISSING_LINE_MSG)
end
def add_offense_extra_line(stanza)
line_index = index_of_line_after(stanza)
line_ops[line_index] = :remove
add_offense(line_index, message: EXTRA_LINE_MSG)
end
def add_offense(line_index, message:)
line_length = [processed_source[line_index].size, 1].max
range = source_range(processed_source.buffer, line_index + 1, 0,
line_length)
super(range, location: range, message: message)
end
end
end
end
end