wehub-resource-sync
59 lines
1.8 KiB
Ruby
59 lines
1.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "progressbar"
|
|
require "bytesize"
|
|
|
|
# Extends progressbar with tqdm-like output:
|
|
# %s / %z → human-readable byte sizes: current size / total siZe
|
|
# %d / %o → compact time: elapsed Duration / Outstanding remaining
|
|
# %r → rate_scale lambda may return a ByteSize string
|
|
|
|
# Override rate_of_change so rate_scale can return a string.
|
|
# The default format '%i' coerces the result to integer, breaking string values.
|
|
ProgressBar::Components::Rate.prepend(Module.new do
|
|
def rate_of_change(format_string = "%s")
|
|
return "0" if elapsed_seconds <= 0
|
|
|
|
format_string % scaled_rate
|
|
end
|
|
end)
|
|
|
|
class ProgressBar # :nodoc: all
|
|
# Add ByteSize-formatted progress methods to the Progress component.
|
|
class Progress
|
|
def progress_with_precision
|
|
ByteSize.new(progress).to_s
|
|
end
|
|
|
|
def total_with_unknown_indicator_with_precision
|
|
total ? ByteSize.new(total).to_s : total_with_unknown_indicator
|
|
end
|
|
end
|
|
|
|
module Components
|
|
# Add label-free time methods to the Time component.
|
|
class Time
|
|
def elapsed_no_label
|
|
val = elapsed
|
|
val.start_with?("00:") ? val[3..-1] : val
|
|
end
|
|
|
|
def estimated_no_label
|
|
val = estimated_with_friendly_oob.split(": ", 2).last
|
|
val.start_with?("00:") ? val[3..-1] : val
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
# Extend MOLECULES with new non-overlapping keys — originals are untouched.
|
|
# remove_const avoids the "already initialized constant" warning.
|
|
molecules = ProgressBar::Format::Molecule::MOLECULES
|
|
ProgressBar::Format::Molecule.send(:remove_const, :MOLECULES)
|
|
ProgressBar::Format::Molecule::MOLECULES = molecules.merge(
|
|
s: %i[progressable progress_with_precision],
|
|
z: %i[progressable total_with_unknown_indicator_with_precision],
|
|
d: %i[time_component elapsed_no_label],
|
|
o: %i[time_component estimated_no_label],
|
|
).freeze
|