75 lines
2.3 KiB
JavaScript
75 lines
2.3 KiB
JavaScript
import * as Log from "../build/output/log";
|
|
import createSpinner from "./spinner";
|
|
function divideSegments(number, segments) {
|
|
const result = [];
|
|
while(number > 0 && segments > 0){
|
|
const dividedNumber = number < segments ? number : Math.floor(number / segments);
|
|
number -= dividedNumber;
|
|
segments--;
|
|
result.push(dividedNumber);
|
|
}
|
|
return result;
|
|
}
|
|
export const createProgress = (total, label)=>{
|
|
const segments = divideSegments(total, 4);
|
|
if (total === 0) {
|
|
throw new Error("invariant: progress total can not be zero");
|
|
}
|
|
let currentSegmentTotal = segments.shift();
|
|
let currentSegmentCount = 0;
|
|
let lastProgressOutput = Date.now();
|
|
let curProgress = 0;
|
|
let progressSpinner = createSpinner(`${label} (${curProgress}/${total})`, {
|
|
spinner: {
|
|
frames: [
|
|
"[ ]",
|
|
"[= ]",
|
|
"[== ]",
|
|
"[=== ]",
|
|
"[ ===]",
|
|
"[ ==]",
|
|
"[ =]",
|
|
"[ ]",
|
|
"[ =]",
|
|
"[ ==]",
|
|
"[ ===]",
|
|
"[====]",
|
|
"[=== ]",
|
|
"[== ]",
|
|
"[= ]"
|
|
],
|
|
interval: 200
|
|
}
|
|
});
|
|
return ()=>{
|
|
curProgress++;
|
|
// Make sure we only log once
|
|
// - per fully generated segment, or
|
|
// - per minute
|
|
// when not showing the spinner
|
|
if (!progressSpinner) {
|
|
currentSegmentCount++;
|
|
if (currentSegmentCount === currentSegmentTotal) {
|
|
currentSegmentTotal = segments.shift();
|
|
currentSegmentCount = 0;
|
|
} else if (lastProgressOutput + 60000 > Date.now()) {
|
|
return;
|
|
}
|
|
lastProgressOutput = Date.now();
|
|
}
|
|
const isFinished = curProgress === total;
|
|
const message = `${label} (${curProgress}/${total})`;
|
|
if (progressSpinner && !isFinished) {
|
|
progressSpinner.setText(message);
|
|
} else {
|
|
progressSpinner == null ? void 0 : progressSpinner.stop();
|
|
if (isFinished) {
|
|
Log.event(message);
|
|
} else {
|
|
Log.info(`${message} ${process.stdout.isTTY ? "\n" : "\r"}`);
|
|
}
|
|
}
|
|
};
|
|
};
|
|
|
|
//# sourceMappingURL=progress.js.map
|