require 'wright/util/file'
require 'wright/util/user'

module Wright
  module Util
    # Helper class to manage file permissions.
    class FilePermissions
      # Creates a FilePermissions object from a
      # {Wright::Resource::File} or {Wright::Resource::Directory}.
      #
      # @param resource [Wright::Resource::File,
      #   Wright::Resource::Directory] the resource object
      # @param filetype [Symbol] the file's type (+:file+ or +:directory+)
      #
      # @return [Wright::Util::FilePermissions] the FilePermissions
      #   object
      def self.create_from_resource(resource, filetype)
        filepath = ::File.expand_path(resource.name)
        p = Wright::Util::FilePermissions.new(filepath, filetype)
        p.owner = resource.owner
        p.group = resource.group
        p.mode = resource.mode
        p
      end

      # @return [String] the target file's name
      attr_accessor :filename

      # @return [Integer] the file's target group id
      attr_reader :group

      # @return [Integer] the file's target mode
      attr_reader :mode

      # @return [Integer] the file's target owner uid
      attr_reader :owner

      VALID_FILETYPES = [:file, :directory]
      private_constant :VALID_FILETYPES

      # Initializes a FilePermissions object.
      #
      # @param filename [String] the file's name
      # @param filetype [Symbol] the file's type (+:file+ or +:directory+)
      def initialize(filename, filetype)
        unless VALID_FILETYPES.include?(filetype)
          fail ArgumentError, "Invalid filetype '#{filetype}'"
        end
        @filename = filename
        @filetype = filetype
      end

      # Sets the file's owner
      def owner=(owner)
        @owner = Util::User.user_to_uid(owner)
      end

      # Sets the file's group
      def group=(group)
        @group = Util::User.group_to_gid(group)
      end

      # Sets the file's mode
      def mode=(mode)
        if mode.nil?
          @mode = nil
          return
        end

        mode_i = File.numeric_mode_to_i(mode)
        unless mode_i
          base_mode_i = ::File.exist?(@filename) ? current_mode : default_mode
          mode_i = File.symbolic_mode_to_i(mode, base_mode_i, @filetype)
        end
        @mode = mode_i
      end

      # Checks if the file's owner, group and mode are up-to-date
      # @return [Bool] +true+ if the file is up to date, +false+
      #   otherwise
      def uptodate?
        if ::File.exist?(@filename)
          owner_uptodate? && group_uptodate? && mode_uptodate?
        else
          false
        end
      end

      # Updates the file's owner, group and mode.
      #
      # @return [void]
      def update
        ::File.chmod(@mode, @filename) if @mode
        ::File.chown(@owner, @group, @filename) if @owner || @group
      end

      # @return [Integer] the file's current mode
      def current_mode
        Wright::Util::File.file_mode(@filename)
      end

      # @return [Integer] the file's current owner's uid
      def current_owner
        Wright::Util::File.file_owner(@filename)
      end

      # @return [Integer] the file's current group's gid
      def current_group
        Wright::Util::File.file_group(@filename)
      end

      private

      def owner_uptodate?
        @owner.nil? || current_owner == @owner
      end

      def group_uptodate?
        @group.nil? || current_group == @group
      end

      def mode_uptodate?
        @mode.nil? || current_mode == @mode
      end

      def default_mode
        case @filetype
        when :file
          ~::File.umask & 0666
        when :directory
          ~::File.umask & 0777
        end
      end
    end
  end
end