wehub-resource-sync
105 lines
2.5 KiB
JavaScript
105 lines
2.5 KiB
JavaScript
import readline from 'node:readline/promises'
|
|
import { stdin as input, stdout as output } from 'node:process'
|
|
|
|
import { BUILT_IN_COMMAND_MANAGER } from '@/built-in-command'
|
|
|
|
const TERMINAL_ARG_START_INDEX = 2
|
|
const COMMAND_ARG_DELIMITER = ' '
|
|
const PNPM_ARGS_SEPARATOR = '--'
|
|
const COMMAND_PREFIX = '/'
|
|
const DISALLOWED_TERMINAL_COMMANDS = new Set(['skill', 's'])
|
|
const TOOL_TERMINAL_COMMANDS = new Set(['tool', 't'])
|
|
const SUPPORTED_TOOL_TERMINAL_SUBCOMMANDS = new Set([
|
|
'list',
|
|
'enable',
|
|
'disable'
|
|
])
|
|
|
|
function printResult(response) {
|
|
for (const line of response.result.plain_text) {
|
|
console.log(line)
|
|
}
|
|
}
|
|
|
|
async function run() {
|
|
const rawInput = process.argv
|
|
.slice(TERMINAL_ARG_START_INDEX)
|
|
.filter((argument) => argument !== PNPM_ARGS_SEPARATOR)
|
|
.join(COMMAND_ARG_DELIMITER)
|
|
.trim()
|
|
const [commandName] = rawInput.split(COMMAND_ARG_DELIMITER).filter(Boolean)
|
|
const [, subcommandName = ''] = rawInput
|
|
.split(COMMAND_ARG_DELIMITER)
|
|
.filter(Boolean)
|
|
|
|
if (!rawInput) {
|
|
console.error('Please provide a built-in command. Example: pnpm cmd help')
|
|
process.exitCode = 1
|
|
return
|
|
}
|
|
|
|
if (rawInput.startsWith(COMMAND_PREFIX)) {
|
|
console.error('Please run built-in commands without "/". Example: pnpm cmd help')
|
|
process.exitCode = 1
|
|
return
|
|
}
|
|
|
|
if (
|
|
commandName &&
|
|
DISALLOWED_TERMINAL_COMMANDS.has(commandName.toLowerCase())
|
|
) {
|
|
console.error(
|
|
'The /skill and /s built-in commands are only available from the chat UI.'
|
|
)
|
|
process.exitCode = 1
|
|
return
|
|
}
|
|
|
|
if (
|
|
commandName &&
|
|
TOOL_TERMINAL_COMMANDS.has(commandName.toLowerCase()) &&
|
|
subcommandName &&
|
|
!SUPPORTED_TOOL_TERMINAL_SUBCOMMANDS.has(subcommandName.toLowerCase())
|
|
) {
|
|
console.error(
|
|
'Direct /tool and /t invocations are only available from the chat UI.'
|
|
)
|
|
process.exitCode = 1
|
|
return
|
|
}
|
|
|
|
let response = await BUILT_IN_COMMAND_MANAGER.execute(
|
|
`${COMMAND_PREFIX}${rawInput}`
|
|
)
|
|
let prompt = null
|
|
|
|
printResult(response)
|
|
|
|
while (
|
|
response.status === 'awaiting_required_parameters' &&
|
|
response.session.pending_input
|
|
) {
|
|
if (!prompt) {
|
|
prompt = readline.createInterface({ input, output })
|
|
}
|
|
|
|
const nextInput = await prompt.question(
|
|
`${response.session.pending_input.prompt || response.session.pending_input.placeholder} `
|
|
)
|
|
|
|
response = await BUILT_IN_COMMAND_MANAGER.execute(
|
|
nextInput,
|
|
response.session.id
|
|
)
|
|
printResult(response)
|
|
}
|
|
|
|
await prompt?.close()
|
|
|
|
if (response.status === 'error') {
|
|
process.exitCode = 1
|
|
}
|
|
}
|
|
|
|
void run()
|