1
0
Fork 0
management/front/dkha-web-sz-main/node_modules/yargs/lib/completion.js

116 lines
3.8 KiB
JavaScript
Raw Normal View History

2023-12-28 23:41:32 +08:00
'use strict'
2023-12-18 13:12:25 +08:00
const path = require('path')
// add bash completions to your
// yargs-powered applications.
2023-12-28 23:41:32 +08:00
module.exports = function completion (yargs, usage, command) {
2023-12-18 13:12:25 +08:00
const self = {
completionKey: 'get-yargs-completions'
}
2023-12-28 23:41:32 +08:00
const zshShell = process.env.SHELL && process.env.SHELL.indexOf('zsh') !== -1
2023-12-18 13:12:25 +08:00
// get a list of completion commands.
// 'args' is the array of strings from the line to be completed
2023-12-28 23:41:32 +08:00
self.getCompletion = function getCompletion (args, done) {
2023-12-18 13:12:25 +08:00
const completions = []
const current = args.length ? args[args.length - 1] : ''
const argv = yargs.parse(args, true)
const aliases = yargs.parsed.aliases
2023-12-28 23:41:32 +08:00
const parentCommands = yargs.getContext().commands
2023-12-18 13:12:25 +08:00
// a custom completion function can be provided
// to completion().
if (completionFunction) {
if (completionFunction.length < 3) {
2023-12-28 23:41:32 +08:00
const result = completionFunction(current, argv)
2023-12-18 13:12:25 +08:00
// promise based completion function.
if (typeof result.then === 'function') {
2023-12-28 23:41:32 +08:00
return result.then((list) => {
process.nextTick(() => { done(list) })
}).catch((err) => {
process.nextTick(() => { throw err })
2023-12-18 13:12:25 +08:00
})
}
// synchronous completion function.
return done(result)
} else {
// asynchronous completion function
2023-12-28 23:41:32 +08:00
return completionFunction(current, argv, (completions) => {
2023-12-18 13:12:25 +08:00
done(completions)
})
}
}
2023-12-28 23:41:32 +08:00
const handlers = command.getCommandHandlers()
for (let i = 0, ii = args.length; i < ii; ++i) {
2023-12-18 13:12:25 +08:00
if (handlers[args[i]] && handlers[args[i]].builder) {
const builder = handlers[args[i]].builder
if (typeof builder === 'function') {
const y = yargs.reset()
builder(y)
return y.argv
}
}
}
2023-12-28 23:41:32 +08:00
if (!current.match(/^-/) && parentCommands[parentCommands.length - 1] !== current) {
usage.getCommands().forEach((usageCommand) => {
const commandName = command.parseCommand(usageCommand[0]).cmd
if (args.indexOf(commandName) === -1) {
if (!zshShell) {
completions.push(commandName)
} else {
const desc = usageCommand[1] || ''
completions.push(commandName.replace(/:/g, '\\:') + ':' + desc)
}
2023-12-18 13:12:25 +08:00
}
})
}
2023-12-28 23:41:32 +08:00
if (current.match(/^-/) || (current === '' && completions.length === 0)) {
const descs = usage.getDescriptions()
Object.keys(yargs.getOptions().key).forEach((key) => {
2023-12-18 13:12:25 +08:00
// If the key and its aliases aren't in 'args', add the key to 'completions'
2023-12-28 23:41:32 +08:00
const keyAndAliases = [key].concat(aliases[key] || [])
const notInArgs = keyAndAliases.every(val => args.indexOf(`--${val}`) === -1)
2023-12-18 13:12:25 +08:00
if (notInArgs) {
2023-12-28 23:41:32 +08:00
if (!zshShell) {
completions.push(`--${key}`)
} else {
const desc = descs[key] || ''
completions.push(`--${key.replace(/:/g, '\\:')}:${desc.replace('__yargsString__:', '')}`)
}
2023-12-18 13:12:25 +08:00
}
})
}
done(completions)
}
// generate the completion script to add to your .bashrc.
2023-12-28 23:41:32 +08:00
self.generateCompletionScript = function generateCompletionScript ($0, cmd) {
const templates = require('./completion-templates')
let script = zshShell ? templates.completionZshTemplate : templates.completionShTemplate
const name = path.basename($0)
2023-12-18 13:12:25 +08:00
// add ./to applications not yet installed as bin.
2023-12-28 23:41:32 +08:00
if ($0.match(/\.js$/)) $0 = `./${$0}`
2023-12-18 13:12:25 +08:00
script = script.replace(/{{app_name}}/g, name)
2023-12-28 23:41:32 +08:00
script = script.replace(/{{completion_command}}/g, cmd)
2023-12-18 13:12:25 +08:00
return script.replace(/{{app_path}}/g, $0)
}
// register a function to perform your own custom
// completions., this function can be either
// synchrnous or asynchronous.
2023-12-28 23:41:32 +08:00
let completionFunction = null
self.registerFunction = (fn) => {
2023-12-18 13:12:25 +08:00
completionFunction = fn
}
return self
}