#
# Author:: Doug MacEachern (<dougm@vmware.com>)
# Copyright:: Copyright 2010-2019, VMware, Inc.
# License:: Apache License, Version 2.0
#
# 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.
#

require_relative "../user"
require_relative "../../exceptions"
require_relative "../../util/windows/net_user" if Chef::Platform.windows?

class Chef
  class Provider
    class User
      class Windows < Chef::Provider::User
        provides :windows_user
        provides :user, os: "windows"

        def initialize(new_resource, run_context)
          super
          @net_user = Chef::Util::Windows::NetUser.new(new_resource.username)
        end

        def load_current_resource
          if new_resource.gid
            logger.warn("The 'gid' (or 'group') property is not implemented on the Windows platform. Please use the `members` property of the  'group' resource to assign a user to a group.")
          end

          @current_resource = Chef::Resource::User::WindowsUser.new(new_resource.name)
          current_resource.username(new_resource.username)
          begin
            user_info = @net_user.get_info
            current_resource.uid(user_info[:user_id])
            current_resource.full_name(user_info[:full_name])
            current_resource.comment(user_info[:comment])
            current_resource.home(user_info[:home_dir])
            current_resource.shell(user_info[:script_path])
          rescue Chef::Exceptions::UserIDNotFound => e
            # e.message should be "The user name could not be found" but checking for that could cause a localization bug
            @user_exists = false
            logger.trace("#{new_resource} does not exist (#{e.message})")
          end

          current_resource
        end

        # Check to see if the user needs any changes
        #
        # === Returns
        # <true>:: If a change is required
        # <false>:: If the users are identical
        def compare_user
          @change_desc = []
          if new_resource.password && !@net_user.validate_credentials(new_resource.password)
            @change_desc << "update password"
          end

          %i{uid comment home shell full_name}.any? do |user_attrib|
            new_val = new_resource.send(user_attrib)
            cur_val = current_resource.send(user_attrib)
            if !new_val.nil? && new_val != cur_val
              @change_desc << "change #{user_attrib} from #{cur_val} to #{new_val}"
            end
          end

          !@change_desc.empty?
        end

        def create_user
          @net_user.add(**set_options)
        end

        def manage_user
          @net_user.update(**set_options)
        end

        def clear_account_rights(name)
          Chef::ReservedNames::Win32::Security.clear_account_rights(name)
        end

        def remove_user
          clear_account_rights(new_resource.username)
          @net_user.delete
        end

        def check_lock
          @net_user.check_enabled
        end

        def lock_user
          @net_user.disable_account
        end

        def unlock_user
          @net_user.enable_account
        end

        def set_options
          opts = { name: new_resource.username }

          field_list = {
            "full_name" => "full_name",
            "comment" => "comment",
            "home" => "home_dir",
            "uid" => "user_id",
            "shell" => "script_path",
            "password" => "password",
          }

          field_list.sort_by { |a| a[0] }.each do |field, option|
            field_symbol = field.to_sym
            next unless current_resource.send(field_symbol) != new_resource.send(field_symbol)
            next unless new_resource.send(field_symbol)

            unless field_symbol == :password
              logger.trace("#{new_resource} setting #{field} to #{new_resource.send(field_symbol)}")
            end
            opts[option.to_sym] = new_resource.send(field_symbol)
          end
          opts
        end

      end
    end
  end
end
