Contents

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

class ProgressBar
  def initialize(title, max_progress)
    @progress = 0
    @max_progress = max_progress
    @title = title
    @start_time = Time.now
    @interrupt = false
  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

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
doro-0.2.3	lib/doro/progressbar.rb