mirror of https://github.com/vuejs/vitepress
strip changes from #3385
Yuxuan Zhang
2 years ago
parent
5493202cb2
commit
1dd35b2226
@ -1,80 +1,18 @@
|
||||
import ora from 'ora'
|
||||
import humanizeDuration from 'humanize-duration'
|
||||
import c from 'picocolors'
|
||||
import { workerMeta } from '../worker'
|
||||
|
||||
export const okMark = c.green('✓')
|
||||
export const failMark = c.red('✖')
|
||||
export const clearLine = '\x1b[2K\r'
|
||||
|
||||
export type UpdateHandle = (
|
||||
done?: number,
|
||||
total?: number,
|
||||
subtask?: string
|
||||
) => any
|
||||
|
||||
let updateHandle: UpdateHandle | null = null
|
||||
|
||||
export const updateCurrentTask: UpdateHandle = (...args) => {
|
||||
if (workerMeta) workerMeta.updateCurrentTask(...args)
|
||||
else 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> {
|
||||
if (workerMeta) {
|
||||
let retVal: T
|
||||
await workerMeta.task(taskName, async (handle: UpdateHandle) => {
|
||||
retVal = await task(handle)
|
||||
})
|
||||
return retVal!
|
||||
}
|
||||
export const okMark = '\x1b[32m✓\x1b[0m'
|
||||
export const failMark = '\x1b[31m✖\x1b[0m'
|
||||
|
||||
export async function task(taskName: string, task: () => Promise<void>) {
|
||||
const spinner = ora({ discardStdin: false })
|
||||
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 {
|
||||
return await task(updateHandle)
|
||||
await task()
|
||||
} catch (e) {
|
||||
success = false
|
||||
spinner.stopAndPersist({ symbol: failMark })
|
||||
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