Module: CLI::Kit::Util

Extended by:
T::Sig
Defined in:
lib/cli/kit/util.rb

Class Method Summary collapse

Methods included from T::Sig

sig

Class Method Details

.begin(&block_that_might_raise) ⇒ Object 



92
93
94
 
# File 'lib/cli/kit/util.rb', line 92

def begin(&block_that_might_raise)
  Retrier.new(block_that_might_raise)
end

.to_filesize(bytes, precision: 2, space: false) ⇒ Object 



14
15
16
 
# File 'lib/cli/kit/util.rb', line 14

def to_filesize(bytes, precision: 2, space: false)
  to_si_scale(bytes, 'B', precision: precision, space: space, factor: 1024)
end

.to_si_scale(number, unit = '', factor: 1000, precision: 2, space: false) ⇒ Object

Raises:

  • (ArgumentError) 


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
 
# File 'lib/cli/kit/util.rb', line 24

def to_si_scale(number, unit = '', factor: 1000, precision: 2, space: false)
  raise ArgumentError, 'factor should only be 1000 or 1024' unless [1000, 1024].include?(factor)

  small_scale = ['m', 'ยต', 'n', 'p', 'f', 'a', 'z', 'y']
  big_scale = ['k', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y']
  negative = number < 0
  number = number.abs.to_f

  if number == 0.0 || number.between?(1, factor)
    prefix = ''
    scale = 0
  else
    scale = Math.log(number, factor).floor
    if number < 1
      index = [-scale - 1, small_scale.length].min
      scale = -(index + 1)
      prefix = T.must(small_scale[index])
    else
      index = [scale - 1, big_scale.length].min
      scale = index + 1
      prefix = T.must(big_scale[index])
    end
  end

  divider = (factor**scale)
  fnum = (number / divider.to_f).round(precision)

  # Trim useless decimal
  fnum = fnum.to_i if (fnum.to_i.to_f * divider.to_f) == number

  fnum = -fnum if negative
  if space
    prefix = ' ' + prefix
  end

  "#{fnum}#{prefix}#{unit}"
end

.with_dir(dir, &block) ⇒ Object 



69
70
71
72
73
74
75
76
77
 
# File 'lib/cli/kit/util.rb', line 69

def with_dir(dir, &block)
  prev = Dir.pwd
  begin
    Dir.chdir(dir)
    yield
  ensure
    Dir.chdir(prev)
  end
end