77 lines
2.0 KiB
Ruby
77 lines
2.0 KiB
Ruby
# typed: true
|
|
# frozen_string_literal: true
|
|
|
|
require "forwardable"
|
|
|
|
module RuboCop
|
|
module Cop
|
|
module Cask
|
|
# This cop audits variables in casks.
|
|
#
|
|
# @example
|
|
# # bad
|
|
# cask do
|
|
# arch = Hardware::CPU.intel? ? "darwin" : "darwin-arm64"
|
|
# end
|
|
#
|
|
# # good
|
|
# cask 'foo' do
|
|
# arch arm: "darwin-arm64", intel: "darwin"
|
|
# end
|
|
class Variables < Base
|
|
extend Forwardable
|
|
extend AutoCorrector
|
|
include CaskHelp
|
|
|
|
def on_cask(cask_block)
|
|
@cask_block = cask_block
|
|
add_offenses
|
|
end
|
|
|
|
private
|
|
|
|
def_delegator :@cask_block, :cask_node
|
|
|
|
def add_offenses
|
|
variable_assignment(cask_node) do |node, var_name, arch_condition, true_node, false_node|
|
|
arm_node, intel_node = if arch_condition == :arm?
|
|
[true_node, false_node]
|
|
else
|
|
[false_node, true_node]
|
|
end
|
|
|
|
replacement_string = if var_name == :arch
|
|
"arch "
|
|
else
|
|
"#{var_name} = on_arch_conditional "
|
|
end
|
|
replacement_parameters = []
|
|
replacement_parameters << "arm: #{arm_node.source}" unless blank_node?(arm_node)
|
|
replacement_parameters << "intel: #{intel_node.source}" unless blank_node?(intel_node)
|
|
replacement_string += replacement_parameters.join(", ")
|
|
|
|
add_offense(node, message: "Use `#{replacement_string}` instead of `#{node.source}`.") do |corrector|
|
|
corrector.replace(node, replacement_string)
|
|
end
|
|
end
|
|
end
|
|
|
|
def blank_node?(node)
|
|
case node.type
|
|
when :str
|
|
node.value.empty?
|
|
when :nil
|
|
true
|
|
else
|
|
false
|
|
end
|
|
end
|
|
|
|
def_node_search :variable_assignment, <<~PATTERN
|
|
$(lvasgn $_ (if (send (const (const nil? :Hardware) :CPU) ${:arm? :intel?}) $_ $_))
|
|
PATTERN
|
|
end
|
|
end
|
|
end
|
|
end
|