# encoding: UTF-8
require 'powerbar'
require 'ansi/code'

module Kafo
# Progress bar base class
#
# To define new progress bar you can inherit from this class and implement
# #finite_template and #infinite_template methods. Also you may find useful to
# change more methods like #done_message or #print_error
  class ProgressBar
    def initialize
      @lines                                    = 0
      @all_lines                                = 0
      @total                                    = :unknown
      @term_width                               = HighLine::SystemExtensions.terminal_size[0] || 0
      @bar                                      = PowerBar.new
      @bar.settings.tty.infinite.template.main  = infinite_template
      @bar.settings.tty.finite.template.main    = finite_template
      @bar.settings.tty.finite.template.padchar = ' '
      @bar.settings.tty.finite.template.barchar = '.'
      @bar.settings.tty.finite.output           = Proc.new { |s| $stderr.print s }
    end

    def update(line)
      @total     = $1.to_i if line =~ /\w*START (\d+)/
      @lines     += 1 if line.include?('RESOURCE') && @lines < @total - 1
      @all_lines += 1

      # we print every 20th line during installation preparing otherwise we update every line
      if @all_lines % 20 == 0 || @total != :unknown
        @bar.show({ :msg   => format(line),
                    :done  => @lines,
                    :total => @total })
      end
    end

    def close
      @bar.show({ :msg   => done_message,
                  :done  => @total == :unknown ? @bar.done + 1 : @total,
                  :total => @total }, true)
      @bar.close
    end

    def print(line)
      @bar.print line
    end

    def print_error(line)
      print line
    end

    private

    def done_message
      text = 'Done'
      text + (' ' * (50 - text.length))
    end

    def format(line)
      (line.tr("\r\n", '') + (' ' * 50))[0..49]
    end

    def finite_template
      'Installing... [${<percent>%}]'
    end

    def infinite_template
      'Installing...'
    end

  end
end

require 'kafo/progress_bars/colored'
require 'kafo/progress_bars/black_white'