forked from choria-legacy/marionette-collective
/
progress.rb
63 lines (53 loc) · 1.68 KB
/
progress.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
module MCollective
module RPC
# Class that shows a progress bar, currently only supports a twirling
# progress bar.
#
# You can specify a size for the progress bar if you want if you dont
# it will use the helper functions to figure out terminal dimensions
# and draw an appropriately sized bar
#
# p = Progress.new
# 100.times {|i| print p.twirl(i+1, 100) + "\r"};puts
#
# * [ ==================================================> ] 100 / 100
class Progress
def initialize(size=nil)
@twirl = ['|', '/', '-', "\\", '|', '/', '-', "\\"]
@twirldex = 0
if size
@size = size
else
cols = Helpers.terminal_dimensions[0] - 22
# Defaults back to old behavior if it
# couldn't figure out the size or if
# its more than 60 wide
if cols <= 0
@size = 0
elsif cols > 60
@size = 60
else
@size = cols
end
end
end
def twirl(current, total)
# if the size is negative there is just not enough
# space on the terminal, return a simpler version
return "\r#{current} / #{total}" if @size == 0
if current == total
txt = "\r %s [" % Util.colorize(:green, "*")
else
txt = "\r %s [" % Util.colorize(:red, @twirl[@twirldex])
end
dashes = ((current.to_f / total) * @size).round
dashes.times { txt << "=" }
txt << ">"
(@size - dashes).times { txt << " " }
txt << " ] #{current} / #{total}"
@twirldex == 7 ? @twirldex = 0 : @twirldex += 1
return txt
end
end
end
end