#
# Copyright:: Copyright (c) Chef Software 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"

class Chef
  class Provider
    class User
      class Linux < Chef::Provider::User
        provides :linux_user, target_mode: true
        provides :user, os: "linux", target_mode: true

        def load_current_resource
          super
          load_shadow_options
        end

        def supports_ruby_shadow?
          # For target mode, ruby-shadow is redirected to a file-based implementation
          true unless ChefConfig::Config.target_mode?
        end

        def compare_user
          user_changed = super

          @change_desc ||= []

          %i{expire_date inactive}.each do |user_attrib|
            new_val = new_resource.send(user_attrib)
            cur_val = current_resource.send(user_attrib)
            if !new_val.nil? && new_val.to_s != cur_val.to_s
              @change_desc << "change #{user_attrib} from #{cur_val} to #{new_val}"
            end
          end

          user_changed || !@change_desc.empty?
        end

        def create_user
          shell_out!("useradd", universal_options, useradd_options, new_resource.username)
        end

        def manage_user
          manage_u = shell_out("usermod", universal_options, usermod_options, new_resource.username, returns: [0, 12])
          if manage_u.exitstatus == 12 && manage_u.stderr !~ /exists/
            raise Chef::Exceptions::User, "Unable to modify home directory for #{new_resource.username}"
          end

          manage_u.error!
        end

        def remove_user
          shell_out!("userdel", userdel_options, new_resource.username)
        end

        def lock_user
          shell_out!("usermod", "-L", new_resource.username)
        end

        def unlock_user
          shell_out!("usermod", "-U", new_resource.username)
        end

        # common to usermod and useradd
        def universal_options
          opts = []
          opts << "-c" << new_resource.comment if should_set?(:comment)
          opts << "-e" << new_resource.expire_date if prop_is_set?(:expire_date)
          opts << "-g" << new_resource.gid if should_set?(:gid)
          opts << "-f" << new_resource.inactive if prop_is_set?(:inactive)
          opts << "-p" << new_resource.password if should_set?(:password)
          opts << "-s" << new_resource.shell if should_set?(:shell)
          opts << "-u" << new_resource.uid if should_set?(:uid)
          opts << "-d" << new_resource.home if updating_home?
          opts << "-o" if new_resource.non_unique
          opts
        end

        def usermod_options
          opts = []
          opts += [ "-u", new_resource.uid ] if new_resource.non_unique
          if updating_home?
            if new_resource.manage_home
              opts << "-m"
            end
          end
          opts
        end

        def useradd_options
          opts = []
          opts << "-r" if new_resource.system
          opts << if new_resource.manage_home
                    "-m"
                  else
                    "-M"
                  end
          opts
        end

        def userdel_options
          opts = []
          opts << "-r" if new_resource.manage_home
          opts << "-f" if new_resource.force
          opts
        end

        def check_lock
          # there's an old bug in rhel (https://bugzilla.redhat.com/show_bug.cgi?id=578534)
          # which means that both 0 and 1 can be success.
          passwd_s = shell_out("passwd", "-S", new_resource.username, returns: [ 0, 1 ])

          # checking "does not exist" has to come before exit code handling since centos and ubuntu differ in exit codes
          if /does not exist/.match?(passwd_s.stderr)
            return false if whyrun_mode?

            raise Chef::Exceptions::User, "User #{new_resource.username} does not exist when checking lock status for #{new_resource}"
          end

          # now raise if we didn't get a 0 or 1 (see above)
          passwd_s.error!

          # now the actual output parsing
          @locked = nil
          status_line = passwd_s.stdout.split(" ")
          @locked = false if /^[PN]/.match?(status_line[1])
          @locked = true if /^L/.match?(status_line[1])

          raise Chef::Exceptions::User, "Cannot determine if user #{new_resource.username} is locked for #{new_resource}" if @locked.nil?

          # FIXME: should probably go on the current_resource
          @locked
        end

        def prop_is_set?(prop)
          v = new_resource.send(prop.to_sym)

          !v.nil? && v != ""
        end
      end
    end
  end
end
