class TestLab
  module Utility

    # Misc Error Class
    class MiscError < UtilityError; end

    # Misc Module
    #
    # @author Zachary Patten <zachary AT jovelabs DOT com>
    module Misc

      def format_object(object, color)
        klass = object.class.to_s.split('::').last

        ["#{klass}:".upcase.send(color).bold, "#{object.id}".white.bold].join(' ')
      end

      def format_object_action(object, action, color)
        klass = object.class.to_s.split('::').last

        ["#{klass}".downcase.send(color).bold, "#{object.id}".white.bold, "#{action}".downcase.send(color).bold, ].join(' ')
      end

      def format_message(message)
        "[".blue + "TL".blue.bold + "]".blue + " " + message
      end

      def please_wait(options={}, &block)
        ui       = options[:ui]
        message  = options[:message]
        mark     = (options[:mark] || "Completed in %0.4f seconds!")

        !block_given? and raise MiscError, "You must supply a block!"
        ui.nil? and raise MiscError, "You must supply a ZTK::UI object!"
        message.nil? and raise MiscError, "You must supply a message!"

        use_spinner = true
        msg         = format_message("#{message} ")
        length      = msg.uncolor.length
        max         = (length >= 60 ? (length+1) : (60 - length))
        mrk         = ((' ' * max) + "# #{mark}\n".bold)

        if (ui.logger.respond_to?(:loggers) && ui.logger.loggers.is_a?(Array))
          if ui.logger.loggers.count > 1
            use_spinner = false
            mrk = format_message("#{message} / #{mark.bold}\n")
            msg = format_message("#{message}\n")
          end
        end

        ZTK::Benchmark.bench(:ui => ui, :message => msg, :mark => mrk, :use_spinner => use_spinner) do
          yield
        end
      end

      def do_provisioner_callbacks(object, action, ui)
        klass       = object.class.to_s.split('::').last
        method_name = %(on_#{klass.downcase}_#{action.to_s.downcase}).to_sym

        object.provisioners.each do |provisioner|
          ui.logger.info { ">>>>> #{object.id.to_s.upcase} #{klass.upcase} #{action.to_s.upcase} [#{method_name}] (#{provisioner}) <<<<<" }
          p = provisioner.new(object.config, ui)
          p.respond_to?(method_name) and p.send(method_name, object)
        end
      end

      def sudo
        %(sudo -p '#{sudo_prompt}')
      end

      def sudo_prompt
        %(sudo password for %u@%h: )
      end

    end

  end
end