77 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
		
		
			
		
	
	
			77 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
|   | # typed: false | ||
|  | # 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 |