mirror of https://github.com/vuejs/vitepress
Merge c75b1c7b55
into 00bd73c8b6
commit
495a6a752d
@ -1,18 +1,70 @@
|
|||||||
import ora from 'ora'
|
import ora from 'ora'
|
||||||
|
import humanizeDuration from 'humanize-duration'
|
||||||
|
import c from 'picocolors'
|
||||||
|
|
||||||
export const okMark = '\x1b[32m✓\x1b[0m'
|
export const okMark = c.green('✓')
|
||||||
export const failMark = '\x1b[31m✖\x1b[0m'
|
export const failMark = c.red('✖')
|
||||||
|
export const clearLine = '\x1b[2K\r'
|
||||||
|
|
||||||
export async function task(taskName: string, task: () => Promise<void>) {
|
export type UpdateHandle = (
|
||||||
|
done?: number,
|
||||||
|
total?: number,
|
||||||
|
subtask?: string
|
||||||
|
) => any
|
||||||
|
|
||||||
|
let updateHandle: UpdateHandle | null = null
|
||||||
|
|
||||||
|
export const updateCurrentTask: UpdateHandle = (...args) => {
|
||||||
|
if (updateHandle) updateHandle(...args)
|
||||||
|
else if (!process.stderr.isTTY) {
|
||||||
|
return
|
||||||
|
} else if (args.length === 0) {
|
||||||
|
process.stderr.write(clearLine)
|
||||||
|
} else {
|
||||||
|
const name = args[2] || 'unknown task'
|
||||||
|
process.stderr.write(
|
||||||
|
`${clearLine}${name} [${args.slice(0, 2).join(' / ')}]`
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function task<T>(
|
||||||
|
taskName: string,
|
||||||
|
task: (update: UpdateHandle) => Promise<T>
|
||||||
|
): Promise<T> {
|
||||||
const spinner = ora({ discardStdin: false })
|
const spinner = ora({ discardStdin: false })
|
||||||
spinner.start(taskName + '...')
|
spinner.start(taskName + '...')
|
||||||
|
|
||||||
|
updateHandle = (done, total, subtask) => {
|
||||||
|
const taskFullName = subtask ? `${taskName} - ${subtask}` : taskName
|
||||||
|
if (done === undefined) {
|
||||||
|
spinner.text = taskFullName + '...'
|
||||||
|
} else if (total === undefined) {
|
||||||
|
spinner.text = `${taskFullName} [ ${done} ]`
|
||||||
|
} else {
|
||||||
|
// match length to display them in same width
|
||||||
|
const _total = `${total}`
|
||||||
|
const _done = `${done}`.padStart(_total.length, ' ')
|
||||||
|
spinner.text = `${taskFullName} [ ${_done} / ${_total} ]`
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const timeStart = performance.now()
|
||||||
|
let success = true
|
||||||
|
|
||||||
try {
|
try {
|
||||||
await task()
|
return await task(updateHandle)
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
spinner.stopAndPersist({ symbol: failMark })
|
success = false
|
||||||
throw e
|
throw e
|
||||||
|
} finally {
|
||||||
|
updateHandle = null
|
||||||
|
const timeEnd = performance.now()
|
||||||
|
const duration = humanizeDuration(timeEnd - timeStart, {
|
||||||
|
maxDecimalPoints: 2
|
||||||
|
})
|
||||||
|
const text = `${taskName} - ${duration}`
|
||||||
|
const symbol = success ? okMark : failMark
|
||||||
|
spinner.stopAndPersist({ symbol, text })
|
||||||
}
|
}
|
||||||
|
|
||||||
spinner.stopAndPersist({ symbol: okMark })
|
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in new issue