openclaw
Version:
Multi-channel AI gateway with extensible messaging integrations
154 lines (153 loc) • 5.23 kB
JavaScript
import { F as resolveTimerTimeoutMs } from "./number-coercion-CLj0HTDM.js";
import { n as registerActiveProgressLine, r as unregisterActiveProgressLine, t as clearActiveProgressLine } from "./progress-line-DiTuCPbL.js";
import { r as theme } from "./theme-vjDs9tao.js";
import { n as supportsOscProgress, t as createOscProgressController } from "./osc-progress-D3edmCLy.js";
import { spinner } from "@clack/prompts";
//#region src/cli/progress.ts
const DEFAULT_DELAY_MS = 0;
let activeProgress = 0;
/** Decide whether the interactive spinner is safe for the current terminal state. */
function shouldUseInteractiveProgressSpinner(params) {
return (params.fallback === void 0 || params.fallback === "spinner") && params.streamIsTty === true && params.stdinIsRaw !== true;
}
const noopReporter = {
setLabel: () => {},
setPercent: () => {},
tick: () => {},
done: () => {}
};
/** Create a no-op, spinner, line, log, and OSC-capable progress reporter. */
function createCliProgress(options) {
if (options.enabled === false) return noopReporter;
if (activeProgress > 0) return noopReporter;
const stream = options.stream ?? process.stderr;
const isTty = stream.isTTY;
const allowLog = !isTty && options.fallback === "log";
if (!isTty && !allowLog) return noopReporter;
const delayMs = resolveTimerTimeoutMs(options.delayMs, DEFAULT_DELAY_MS, 0);
const canOsc = isTty && supportsOscProgress(process.env, isTty);
const stdinIsRaw = process.stdin.isRaw;
const allowSpinner = shouldUseInteractiveProgressSpinner({
fallback: options.fallback,
streamIsTty: isTty,
stdinIsRaw
});
const allowLine = isTty && options.fallback === "line";
if (isTty && stdinIsRaw && (options.fallback === void 0 || options.fallback === "spinner")) return noopReporter;
let started = false;
let finished = false;
let label = options.label;
const total = options.total ?? null;
let completed = 0;
let percent = 0;
let indeterminate = options.indeterminate ?? (options.total === void 0 || options.total === null);
activeProgress += 1;
if (isTty) registerActiveProgressLine(stream);
const controller = canOsc ? createOscProgressController({
env: process.env,
isTty: stream.isTTY,
write: (chunk) => stream.write(chunk)
}) : null;
const spin = allowSpinner ? spinner({ output: stream }) : null;
const renderLine = allowLine ? () => {
if (!started) return;
const suffix = indeterminate ? "" : ` ${percent}%`;
clearActiveProgressLine();
stream.write(`${theme.accent(label)}${suffix}`);
} : null;
const renderLog = allowLog ? (() => {
let lastLine = "";
let lastAt = 0;
const throttleMs = 250;
return () => {
if (!started) return;
const suffix = indeterminate ? "" : ` ${percent}%`;
const nextLine = `${label}${suffix}`;
const now = Date.now();
if (nextLine === lastLine && now - lastAt < throttleMs) return;
lastLine = nextLine;
lastAt = now;
stream.write(`${nextLine}\n`);
};
})() : null;
let timer = null;
const applyState = () => {
if (!started || finished) return;
if (controller) {
if (indeterminate) controller.setIndeterminate(label);
else controller.setPercent(label, percent);
}
if (spin) spin.message(theme.accent(label));
if (renderLine) renderLine();
if (renderLog) renderLog();
};
const start = () => {
if (started) return;
started = true;
if (spin) spin.start(theme.accent(label));
applyState();
};
if (delayMs === 0) start();
else timer = setTimeout(start, delayMs);
const setLabel = (next) => {
label = next;
applyState();
};
const setPercent = (nextPercent) => {
percent = Math.max(0, Math.min(100, Math.round(nextPercent)));
indeterminate = false;
applyState();
};
const tick = (delta = 1) => {
if (!total) return;
completed = Math.min(total, completed + delta);
const nextPercent = total > 0 ? Math.round(completed / total * 100) : 0;
setPercent(nextPercent);
};
const done = () => {
if (finished) return;
finished = true;
if (timer) {
clearTimeout(timer);
timer = null;
}
if (!started) {
if (isTty) unregisterActiveProgressLine(stream);
activeProgress = Math.max(0, activeProgress - 1);
return;
}
if (controller) controller.clear();
if (spin) spin.stop();
clearActiveProgressLine();
if (isTty) unregisterActiveProgressLine(stream);
activeProgress = Math.max(0, activeProgress - 1);
};
return {
setLabel,
setPercent,
tick,
done
};
}
/** Run async work with a progress reporter that is always stopped in finally. */
async function withProgress(options, work) {
const progress = createCliProgress(options);
try {
return await work(progress);
} finally {
progress.done();
}
}
/** Run async work with a progress reporter plus a completed/total update adapter. */
async function withProgressTotals(options, work) {
return await withProgress(options, async (progress) => {
const update = ({ completed, total, label }) => {
if (label) progress.setLabel(label);
if (!Number.isFinite(total) || total <= 0) return;
progress.setPercent(completed / total * 100);
};
return await work(update, progress);
});
}
//#endregion
export { withProgressTotals as i, shouldUseInteractiveProgressSpinner as n, withProgress as r, createCliProgress as t };