#
# Author:: Seth Chisamore (<schisamo@chef.io>)
# Cookbook Name:: windows
# Provider:: feature_servermanagercmd
#
# Copyright:: 2011-2016, Chef Software, Inc.
#
# 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.
#

use_inline_resources

include Chef::Provider::WindowsFeature::Base
include Chef::Mixin::ShellOut
include Windows::Helper

# Exit codes are listed at http://technet.microsoft.com/en-us/library/cc749128(v=ws.10).aspx

def check_reboot(result, feature)
  if result.exitstatus == 3010 # successful, but needs reboot
    node.run_state[:reboot_requested] = true
    Chef::Log.warn("Require reboot to install #{feature}")
  elsif result.exitstatus == 1001 # failure, but needs reboot before we can do anything else
    node.run_state[:reboot_requested] = true
    Chef::Log.warn("Failed installing #{feature} and need to reboot")
  end
  result.error! # throw for any other bad results. The above results will also get raised, and should cause a reboot via the handler.
end

def install_feature(_name)
  check_reboot(shell_out("#{servermanagercmd} -install #{@new_resource.feature_name}", returns: [0, 42, 127, 1003, 3010]), @new_resource.feature_name)
end

def remove_feature(_name)
  check_reboot(shell_out("#{servermanagercmd} -remove #{@new_resource.feature_name}", returns: [0, 42, 127, 1003, 3010]), @new_resource.feature_name)
end

def installed?
  @installed ||= begin
    cmd = shell_out("#{servermanagercmd} -query", returns: [0, 42, 127, 1003])
    cmd.stderr.empty? && (cmd.stdout =~ /^\s*?\[X\]\s.+?\s\[#{@new_resource.feature_name}\]\s*$/i)
  end
end

private

# account for File System Redirector
# http://msdn.microsoft.com/en-us/library/aa384187(v=vs.85).aspx
def servermanagercmd
  @servermanagercmd ||= begin
    locate_sysnative_cmd('servermanagercmd.exe')
  end
end
