piotrmurach/tty-prompt

# frozen_string_literal: true

module TTY
  class Prompt
    class Timer
      attr_reader :duration

      attr_reader :total

      attr_reader :interval

      def initialize(duration, interval)
        @duration = duration
        @interval = interval
        @total = 0.0
        @current = nil
        @events = []
      end

      def start
        return if @current

        @current = time_now
      end

      def stop
        return unless @current

        @current = nil
      end

      def runtime
        time_now - @current
      end

      def on_tick(&block)
        @events << block
      end

      def while_remaining
        start
        remaining = duration

        if @duration
          while remaining >= 0.0
            if runtime >= total
              tick = duration - @total
              @events.each { |block| block.(tick) }
              @total += @interval
            end

            yield(remaining)
            remaining = duration - runtime
          end
        else
          loop { yield }
        end
      ensure
        stop
      end

      if defined?(Process::CLOCK_MONOTONIC)
        # Object representing current time
        def time_now
          ::Process.clock_gettime(Process::CLOCK_MONOTONIC)
        end
      else
        # Object represeting current time
        def time_now
          ::Time.now
        end
      end
    end # Timer
  end # Prompt
end # TTY