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 ora from 'ora'
|
||||||
import humanizeDuration from 'humanize-duration'
|
|
||||||
import c from 'picocolors'
|
|
||||||
import { workerMeta } from '../worker'
|
|
||||||
|
|
||||||
export const okMark = c.green('✓')
|
export const okMark = '\x1b[32m✓\x1b[0m'
|
||||||
export const failMark = c.red('✖')
|
export const failMark = '\x1b[31m✖\x1b[0m'
|
||||||
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 async function task(taskName: string, task: () => Promise<void>) {
|
||||||
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 {
|
||||||
return await task(updateHandle)
|
await task()
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
success = false
|
spinner.stopAndPersist({ symbol: failMark })
|
||||||
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