| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424 |
- import process from 'node:process';
- import chalk from 'chalk';
- import cliCursor from 'cli-cursor';
- import cliSpinners from 'cli-spinners';
- import logSymbols from 'log-symbols';
- import stripAnsi from 'strip-ansi';
- import stringWidth from 'string-width';
- import isInteractive from 'is-interactive';
- import isUnicodeSupported from 'is-unicode-supported';
- import stdinDiscarder from 'stdin-discarder';
- class Ora {
- #linesToClear = 0;
- #isDiscardingStdin = false;
- #lineCount = 0;
- #frameIndex = -1;
- #lastSpinnerFrameTime = 0;
- #options;
- #spinner;
- #stream;
- #id;
- #initialInterval;
- #isEnabled;
- #isSilent;
- #indent;
- #text;
- #prefixText;
- #suffixText;
- color;
- constructor(options) {
- if (typeof options === 'string') {
- options = {
- text: options,
- };
- }
- this.#options = {
- color: 'cyan',
- stream: process.stderr,
- discardStdin: true,
- hideCursor: true,
- ...options,
- };
- // Public
- this.color = this.#options.color;
- // It's important that these use the public setters.
- this.spinner = this.#options.spinner;
- this.#initialInterval = this.#options.interval;
- this.#stream = this.#options.stream;
- this.#isEnabled = typeof this.#options.isEnabled === 'boolean' ? this.#options.isEnabled : isInteractive({stream: this.#stream});
- this.#isSilent = typeof this.#options.isSilent === 'boolean' ? this.#options.isSilent : false;
- // Set *after* `this.#stream`.
- // It's important that these use the public setters.
- this.text = this.#options.text;
- this.prefixText = this.#options.prefixText;
- this.suffixText = this.#options.suffixText;
- this.indent = this.#options.indent;
- if (process.env.NODE_ENV === 'test') {
- this._stream = this.#stream;
- this._isEnabled = this.#isEnabled;
- Object.defineProperty(this, '_linesToClear', {
- get() {
- return this.#linesToClear;
- },
- set(newValue) {
- this.#linesToClear = newValue;
- },
- });
- Object.defineProperty(this, '_frameIndex', {
- get() {
- return this.#frameIndex;
- },
- });
- Object.defineProperty(this, '_lineCount', {
- get() {
- return this.#lineCount;
- },
- });
- }
- }
- get indent() {
- return this.#indent;
- }
- set indent(indent = 0) {
- if (!(indent >= 0 && Number.isInteger(indent))) {
- throw new Error('The `indent` option must be an integer from 0 and up');
- }
- this.#indent = indent;
- this.#updateLineCount();
- }
- get interval() {
- return this.#initialInterval ?? this.#spinner.interval ?? 100;
- }
- get spinner() {
- return this.#spinner;
- }
- set spinner(spinner) {
- this.#frameIndex = -1;
- this.#initialInterval = undefined;
- if (typeof spinner === 'object') {
- if (spinner.frames === undefined) {
- throw new Error('The given spinner must have a `frames` property');
- }
- this.#spinner = spinner;
- } else if (!isUnicodeSupported()) {
- this.#spinner = cliSpinners.line;
- } else if (spinner === undefined) {
- // Set default spinner
- this.#spinner = cliSpinners.dots;
- } else if (spinner !== 'default' && cliSpinners[spinner]) {
- this.#spinner = cliSpinners[spinner];
- } else {
- throw new Error(`There is no built-in spinner named '${spinner}'. See https://github.com/sindresorhus/cli-spinners/blob/main/spinners.json for a full list.`);
- }
- }
- get text() {
- return this.#text;
- }
- set text(value = '') {
- this.#text = value;
- this.#updateLineCount();
- }
- get prefixText() {
- return this.#prefixText;
- }
- set prefixText(value = '') {
- this.#prefixText = value;
- this.#updateLineCount();
- }
- get suffixText() {
- return this.#suffixText;
- }
- set suffixText(value = '') {
- this.#suffixText = value;
- this.#updateLineCount();
- }
- get isSpinning() {
- return this.#id !== undefined;
- }
- #getFullPrefixText(prefixText = this.#prefixText, postfix = ' ') {
- if (typeof prefixText === 'string' && prefixText !== '') {
- return prefixText + postfix;
- }
- if (typeof prefixText === 'function') {
- return prefixText() + postfix;
- }
- return '';
- }
- #getFullSuffixText(suffixText = this.#suffixText, prefix = ' ') {
- if (typeof suffixText === 'string' && suffixText !== '') {
- return prefix + suffixText;
- }
- if (typeof suffixText === 'function') {
- return prefix + suffixText();
- }
- return '';
- }
- #updateLineCount() {
- const columns = this.#stream.columns ?? 80;
- const fullPrefixText = this.#getFullPrefixText(this.#prefixText, '-');
- const fullSuffixText = this.#getFullSuffixText(this.#suffixText, '-');
- const fullText = ' '.repeat(this.#indent) + fullPrefixText + '--' + this.#text + '--' + fullSuffixText;
- this.#lineCount = 0;
- for (const line of stripAnsi(fullText).split('\n')) {
- this.#lineCount += Math.max(1, Math.ceil(stringWidth(line, {countAnsiEscapeCodes: true}) / columns));
- }
- }
- get isEnabled() {
- return this.#isEnabled && !this.#isSilent;
- }
- set isEnabled(value) {
- if (typeof value !== 'boolean') {
- throw new TypeError('The `isEnabled` option must be a boolean');
- }
- this.#isEnabled = value;
- }
- get isSilent() {
- return this.#isSilent;
- }
- set isSilent(value) {
- if (typeof value !== 'boolean') {
- throw new TypeError('The `isSilent` option must be a boolean');
- }
- this.#isSilent = value;
- }
- frame() {
- // Ensure we only update the spinner frame at the wanted interval,
- // even if the render method is called more often.
- const now = Date.now();
- if (this.#frameIndex === -1 || now - this.#lastSpinnerFrameTime >= this.interval) {
- this.#frameIndex = ++this.#frameIndex % this.#spinner.frames.length;
- this.#lastSpinnerFrameTime = now;
- }
- const {frames} = this.#spinner;
- let frame = frames[this.#frameIndex];
- if (this.color) {
- frame = chalk[this.color](frame);
- }
- const fullPrefixText = (typeof this.#prefixText === 'string' && this.#prefixText !== '') ? this.#prefixText + ' ' : '';
- const fullText = typeof this.text === 'string' ? ' ' + this.text : '';
- const fullSuffixText = (typeof this.#suffixText === 'string' && this.#suffixText !== '') ? ' ' + this.#suffixText : '';
- return fullPrefixText + frame + fullText + fullSuffixText;
- }
- clear() {
- if (!this.#isEnabled || !this.#stream.isTTY) {
- return this;
- }
- this.#stream.cursorTo(0);
- for (let index = 0; index < this.#linesToClear; index++) {
- if (index > 0) {
- this.#stream.moveCursor(0, -1);
- }
- this.#stream.clearLine(1);
- }
- if (this.#indent || this.lastIndent !== this.#indent) {
- this.#stream.cursorTo(this.#indent);
- }
- this.lastIndent = this.#indent;
- this.#linesToClear = 0;
- return this;
- }
- render() {
- if (this.#isSilent) {
- return this;
- }
- this.clear();
- this.#stream.write(this.frame());
- this.#linesToClear = this.#lineCount;
- return this;
- }
- start(text) {
- if (text) {
- this.text = text;
- }
- if (this.#isSilent) {
- return this;
- }
- if (!this.#isEnabled) {
- if (this.text) {
- this.#stream.write(`- ${this.text}\n`);
- }
- return this;
- }
- if (this.isSpinning) {
- return this;
- }
- if (this.#options.hideCursor) {
- cliCursor.hide(this.#stream);
- }
- if (this.#options.discardStdin && process.stdin.isTTY) {
- this.#isDiscardingStdin = true;
- stdinDiscarder.start();
- }
- this.render();
- this.#id = setInterval(this.render.bind(this), this.interval);
- return this;
- }
- stop() {
- if (!this.#isEnabled) {
- return this;
- }
- clearInterval(this.#id);
- this.#id = undefined;
- this.#frameIndex = 0;
- this.clear();
- if (this.#options.hideCursor) {
- cliCursor.show(this.#stream);
- }
- if (this.#options.discardStdin && process.stdin.isTTY && this.#isDiscardingStdin) {
- stdinDiscarder.stop();
- this.#isDiscardingStdin = false;
- }
- return this;
- }
- succeed(text) {
- return this.stopAndPersist({symbol: logSymbols.success, text});
- }
- fail(text) {
- return this.stopAndPersist({symbol: logSymbols.error, text});
- }
- warn(text) {
- return this.stopAndPersist({symbol: logSymbols.warning, text});
- }
- info(text) {
- return this.stopAndPersist({symbol: logSymbols.info, text});
- }
- stopAndPersist(options = {}) {
- if (this.#isSilent) {
- return this;
- }
- const prefixText = options.prefixText ?? this.#prefixText;
- const fullPrefixText = this.#getFullPrefixText(prefixText, ' ');
- const symbolText = options.symbol ?? ' ';
- const text = options.text ?? this.text;
- const separatorText = symbolText ? ' ' : '';
- const fullText = (typeof text === 'string') ? separatorText + text : '';
- const suffixText = options.suffixText ?? this.#suffixText;
- const fullSuffixText = this.#getFullSuffixText(suffixText, ' ');
- const textToWrite = fullPrefixText + symbolText + fullText + fullSuffixText + '\n';
- this.stop();
- this.#stream.write(textToWrite);
- return this;
- }
- }
- export default function ora(options) {
- return new Ora(options);
- }
- export async function oraPromise(action, options) {
- const actionIsFunction = typeof action === 'function';
- const actionIsPromise = typeof action.then === 'function';
- if (!actionIsFunction && !actionIsPromise) {
- throw new TypeError('Parameter `action` must be a Function or a Promise');
- }
- const {successText, failText} = typeof options === 'object'
- ? options
- : {successText: undefined, failText: undefined};
- const spinner = ora(options).start();
- try {
- const promise = actionIsFunction ? action(spinner) : action;
- const result = await promise;
- spinner.succeed(
- successText === undefined
- ? undefined
- : (typeof successText === 'string' ? successText : successText(result)),
- );
- return result;
- } catch (error) {
- spinner.fail(
- failText === undefined
- ? undefined
- : (typeof failText === 'string' ? failText : failText(error)),
- );
- throw error;
- }
- }
- export {default as spinners} from 'cli-spinners';
|