#
# Copyright:: Copyright 2020, Chef Software Inc.
# Author:: Tim Smith (<tsmith@chef.io>)
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
module RuboCop
  module Cop
    module Chef
      module ChefCorrectness
        # Use valid platform family values in case statements.
        #
        # @example
        #
        #   # bad
        #   case node['platform_family']
        #   when 'redhat'
        #     puts "I'm on a RHEL-like system"
        #   end
        #
        class InvalidPlatformFamilyInCase < Cop
          include RangeHelp
          include ::RuboCop::Chef::PlatformHelpers

          MSG = 'Use valid platform family values in case statements.'.freeze

          def_node_matcher :node_platform_family?, <<-PATTERN
            (send (send nil? :node) :[] (str "platform_family") )
          PATTERN

          def on_case(node)
            node_platform_family?(node.condition) do
              node.each_when do |when_node|
                when_node.each_condition do |con|
                  next unless con.str_type? # if the condition isn't a string we can't check so skip

                  if INVALID_PLATFORM_FAMILIES[con.str_content]
                    add_offense(con, location: :expression, message: MSG, severity: :refactor)
                  end
                end
              end
            end
          end

          def autocorrect(node)
            new_value = INVALID_PLATFORM_FAMILIES[node.str_content]

            # some invalid platform families have no direct correction value and return nil instead
            return unless new_value

            # if the correct value already exists in the when statement then we just want to delete this node
            if node.parent.conditions.any? { |x| x.str_content == new_value }
              lambda do |corrector|
                range = range_with_surrounding_comma(range_with_surrounding_space(range: node.loc.expression, side: :left), :both)
                corrector.remove(range)
              end
            else
              lambda do |corrector|
                corrector.replace(node.loc.expression, "'#{new_value}'")
              end
            end
          end
        end
      end
    end
  end
end