require 'io/console'
require 'time'
require 'notifier'

##
# This class handles the incrementing and display of the progress bar.
class ProgressBar
  ##
  # Title can be any string, max_progress is in seconds.
  def initialize(title, max_progress)
    @progress = 0
    @max_progress = max_progress
    @title = title
    @start_time = Time.now
    @interrupt = false
  end

  ##
  # This starts the timer. It accepts a block which allows you plug in other
  # behaviors (such as incrementing another timer outside of the class).
  # You'd do something like:
  #   ProgressBar.new("Example", 60).start do |t|
  #     p 'this fires every second'
  #   end
  def start
    Signal.trap("INT") { @interrupt = true }

    while (@progress <= @max_progress && @interrupt == false )
      render_progress
      @progress += 1
      yield
      sleep 1
    end

    print("\r")

    display_notification
  end

  private

  def render_progress
    print @title
    print ' ['
    print '=' * progress_bar_ticks
    print ' ' * remaining_bar_ticks rescue 0
    print '] '
    print "\e[32m#{display_time}\e[0m"
    print "\r"
  end

  # Methods for determing time
  def display_time
    Time.at((Time.now - @start_time)).utc.strftime("%M:%S")
  end

  # Methods for determing progress bar length
  def progress_bar_ticks
    (percent * (terminal_length - other_items_length)).to_i
  end

  def remaining_bar_ticks
    terminal_length - progress_bar_ticks - other_items_length
  end

  def percent
    (@progress.to_f / @max_progress)
  end

  def other_items_length
    @title.size + 10
  end

  # Methods for determining terminal length
  def terminal_length
    return 80 unless unix?

    result = dynamic_width
    result < 20 ? 80 : result

  rescue
    80
  end

  def dynamic_width
    _rows, columns = IO.console.winsize
    columns
  end

  def unix?
    RUBY_PLATFORM =~ /(aix|darwin|linux|(net|free|open)bsd|cygwin|solaris|irix|hpux)/i
  end

  # Methods for notifications
  def display_notification
    unless @interrupt
      Notifier.notify(image: "logo.png", title: @title, message: "")
    end
  end

end