-
Notifications
You must be signed in to change notification settings - Fork 12
Expand file tree
/
Copy pathindex.ts
More file actions
691 lines (602 loc) · 29.1 KB
/
index.ts
File metadata and controls
691 lines (602 loc) · 29.1 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
/* eslint-disable @typescript-eslint/no-unsafe-assignment */
import {Argument, Command as BaseCommand, InvalidArgumentError, InvalidOptionArgumentError} from 'commander'
import {Option as BaseOption} from 'commander'
import {JSONSchema7} from 'json-schema'
import {inspect} from 'util'
import {addCompletions} from './completions.js'
import {runWithCliContext} from './context.js'
import {FailedToExitError, CliValidationError} from './errors.js'
import {
flattenedProperties,
incompatiblePropertyPairs,
getDescription,
getSchemaTypes,
getEnumChoices,
getAllowedSchemas,
} from './json-schema.js'
import {commandToJSON} from './json.js'
import {lineByLineConsoleLogger} from './logging.js'
import {
type AnyRouter,
type CreateCallerFactoryLike,
getParsedProcedure,
isNorpcRouter,
isTrpcRouter,
parseRouter,
type ProcedureInfo,
} from './parse-router.js'
import {promptify} from './prompts.js'
import {prettifyStandardSchemaError} from './standard-schema/errors.js'
import {looksLikeStandardSchemaFailure} from './standard-schema/utils.js'
import {TrpcCli, TrpcCliParams, TrpcCliRunParams} from './types.js'
import {looksLikeInstanceof} from './util.js'
const orpcServerOrError = await import('@orpc/server').catch(String)
const getOrpcServerModule = () => {
if (typeof orpcServerOrError === 'string') {
throw new Error(`@orpc/server must be installed. Error loading: ${orpcServerOrError}`)
}
return orpcServerOrError
}
// @ts-ignore zod is an optional peer dependency so might not be installed. oh well, you still get this one interface
declare module 'zod/v4' {
export interface GlobalMeta {
/**
* If true, this property will be mapped to a positional CLI argument by trpc-cli. Only valid for string, number, or boolean types (or arrays of these types).
* Note: the order of positional arguments is determined by the order of properties in the schema.
* For example, the following are different:
* - `z.object({abc: z.string().meta({positional: true}), xyz: z.string().meta({positional: true})})`
* - `z.object({xyz: z.string().meta({positional: true}), abc: z.string().meta({positional: true})})`
*/
positional?: boolean
/**
* If set, this value will be used an alias for the option.
* Note: this is only valid for options, not positional arguments.
*/
alias?: string
/**
* If true, this option will not appear in the --help output but will still be functional.
*/
hidden?: boolean
}
}
// @ts-ignore zod is an optional peer dependency so might not be installed. oh well, you still get this one interface
declare module 'zod' {
export interface GlobalMeta {
/**
* If true, this property will be mapped to a positional CLI argument by trpc-cli. Only valid for string, number, or boolean types (or arrays of these types).
* Note: the order of positional arguments is determined by the order of properties in the schema.
* For example, the following are different:
* - `z.object({abc: z.string().meta({positional: true}), xyz: z.string().meta({positional: true})})`
* - `z.object({xyz: z.string().meta({positional: true}), abc: z.string().meta({positional: true})})`
*/
positional?: boolean
/**
* If set, this value will be used an alias for the option.
* Note: this is only valid for options, not positional arguments.
*/
alias?: string
/**
* If true, this option will not appear in the --help output but will still be functional.
*/
hidden?: boolean
}
}
export * from './types.js'
/** @deprecated use `import * as trpcServer from '@trpc/server'` instead */
export const trpcServer = "@deprecated use `import * as trpcServer from '@trpc/server'` instead"
/** @deprecated use `import {z} from "zod/v4"` instead (or use zod/v3 if you need to support use an old version of zod) */
export const z =
'@deprecated use `import {z} from "zod/v4"` instead (or use zod/v3 if you need to support use an old version of zod)'
/** @deprecated use `import {z} from "zod/v4"` instead (or use zod/v3 if you need to support use an old version of zod) */
export const zod =
'@deprecated use `import {z} from "zod/v4"` instead (or use zod/v3 if you need to support use an old version of zod)'
export class Command extends BaseCommand {
/** @internal track the commands that have been run, so that we can find the `__result` of the last command */
__ran: Command[] = []
__input?: unknown
/** @internal stash the return value of the underlying procedure on the command so to pass to `FailedToExitError` for use in a pinch */
__result?: unknown
/** @internal stash the argv so it's accessible from action handlers via cli context */
__argv?: string[]
}
/** re-export of the @trpc/server package, just to avoid needing to install manually when getting started */
export {type AnyRouter, type AnyProcedure, type NorpcProcedureLike, type NorpcRouterLike} from './parse-router.js'
export {parseRouter} from './parse-router.js'
/**
* Run a trpc router as a CLI.
*
* @param router A trpc router
* @param context The context to use when calling the procedures - needed if your router requires a context
* @param trpcServer The trpc server module to use. Only needed if using trpc v10.
* @returns A CLI object with a `run` method that can be called to run the CLI. The `run` method will parse the command line arguments, call the appropriate trpc procedure, log the result and exit the process. On error, it will log the error and exit with a non-zero exit code.
*/
export function createCli<R extends AnyRouter>({router, ...params}: TrpcCliParams<R>): TrpcCli {
const procedureEntries = parseRouter({router, ...params})
function buildProgram(runParams?: TrpcCliRunParams) {
const logger = {...lineByLineConsoleLogger, ...runParams?.logger}
const program = new Command(params.name)
if (params.version) program.version(params.version)
if (params.description) program.description(params.description)
if (params.usage) [params.usage].flat().forEach(usage => program.usage(usage))
program.showHelpAfterError()
program.showSuggestionAfterError()
// Organize commands in a tree structure for nested subcommands
const commandTree: Record<string, Command> = {
'': program, // Root level
}
// Keep track of default commands for each parent path
const defaultCommands: Record<
string,
{
procedurePath: string
config: (typeof procedureEntries)[0][1]
command: Command
}
> = {}
const _process = runParams?.process || process
const configureCommand = (command: Command, procedurePath: string, info: ProcedureInfo) => {
const {meta} = info
const parsedProcedure = getParsedProcedure(info)
const incompatiblePairs = incompatiblePropertyPairs(parsedProcedure.optionsJsonSchema)
// add meta to the commander command so we can access it in prompt.ts
Object.assign(command, {__trpcCli: {path: procedurePath, meta, originalInputSchema: info.originalInputSchema}})
const optionJsonSchemaProperties = flattenedProperties(parsedProcedure.optionsJsonSchema)
command.exitOverride(ec => {
_process.exit(ec.exitCode)
throw new FailedToExitError(`Command ${command.name()} exitOverride`, {exitCode: ec.exitCode, cause: ec})
})
command.configureOutput({
writeOut: str => {
logger.info?.(str)
},
writeErr: str => {
logger.error?.(str)
},
})
command.showHelpAfterError()
if (meta.usage) command.usage([meta.usage].flat().join('\n'))
if (meta.examples) command.addHelpText('after', `\nExamples:\n${[meta.examples].flat().join('\n')}`)
meta?.aliases?.command?.forEach(alias => {
command.alias(alias)
})
command.description(meta?.description || '')
parsedProcedure.positionalParameters.forEach(param => {
const descriptionParts = [
param.type === 'string' ? '' : param.type, // "string" is the default assumption, don't bother showing it
param.description,
param.required ? '(required)' : '',
]
const argument = new Argument(param.name, descriptionParts.filter(Boolean).join(' '))
if (param.type === 'number') {
argument.argParser(value => {
const number = numberParser(value, {fallback: null})
if (number == null) throw new InvalidArgumentError(`Invalid number: ${value}`)
return value
})
}
argument.required = param.required
argument.variadic = param.array
command.addArgument(argument)
})
const unusedOptionAliases: Record<string, string> = {...meta.aliases?.options}
const addOptionForProperty = ([propertyKey, propertyValue]: [string, JSONSchema7]) => {
class Option extends BaseOption {
attributeName() {
return propertyKey // by default commander uses camelcase(this.name()) which turns `use-mcp-server` into `useMcpServer` - when it might be `useMCPServer`
}
}
const description = getDescription(propertyValue)
const longOption = `--${kebabCase(propertyKey)}`
let flags = longOption
const alias =
propertyValue && 'alias' in propertyValue && typeof propertyValue.alias === 'string'
? propertyValue.alias
: meta.aliases?.options?.[propertyKey]
if (alias) {
let prefix = '-'
if (alias.startsWith('-')) prefix = ''
else if (alias.length > 1) prefix = '--'
flags = `${prefix}${alias}, ${flags}`
delete unusedOptionAliases[propertyKey]
}
const isHidden = 'hidden' in propertyValue && propertyValue.hidden === true
const addOption = (opt: InstanceType<typeof BaseOption>) => {
if (isHidden) opt.hideHelp()
command.addOption(opt)
}
const allowedSchemas = getAllowedSchemas(propertyValue)
const firstSchemaWithDefault = allowedSchemas.find(subSchema => 'default' in subSchema)
// Check for default value - first in the allowed schemas, then on the root property itself
const defaultValue =
firstSchemaWithDefault && 'default' in firstSchemaWithDefault
? ({exists: true, value: firstSchemaWithDefault.default} as const)
: 'default' in propertyValue
? ({exists: true, value: propertyValue.default} as const)
: ({exists: false} as const)
const rootTypes = getSchemaTypes(propertyValue).sort()
const propertyType = rootTypes[0]
const isValueRequired =
'required' in parsedProcedure.optionsJsonSchema &&
parsedProcedure.optionsJsonSchema.required?.includes(propertyKey)
const isCliOptionRequired = isValueRequired && propertyType !== 'boolean' && !defaultValue.exists
function negate() {
const shouldNegate = 'negatable' in propertyValue ? propertyValue.negatable : meta.negateBooleans
if (shouldNegate) {
const negation = new Option(longOption.replace('--', '--no-'), `Negate \`${longOption}\` option.`.trim())
command.addOption(negation)
}
}
const bracketise = (name: string) => (isCliOptionRequired ? `<${name}>` : `[${name}]`)
// Check if this is an enum (including union of literals like z.union([z.literal('foo'), z.literal('bar')]))
// If so, handle it as a string with choices, not as a multi-type union
const enumChoices = getEnumChoices(propertyValue)
if (enumChoices?.type === 'string_enum') {
const option = new Option(`${flags} ${bracketise('string')}`, description)
option.choices(enumChoices.choices)
if (defaultValue.exists) option.default(defaultValue.value)
addOption(option)
return
}
if (allowedSchemas.length > 1) {
const option = new Option(`${flags} [value]`, description)
if (defaultValue.exists) option.default(defaultValue.value)
else if (rootTypes.includes('boolean')) option.default(false)
option.argParser(getOptionValueParser(propertyValue))
addOption(option)
if (rootTypes.includes('boolean')) negate()
return
}
if (rootTypes.length !== 1) {
const option = new Option(`${flags} ${bracketise('json')}`, description)
option.argParser(getOptionValueParser(propertyValue))
addOption(option)
return
}
if (propertyType === 'boolean') {
const option = new Option(`${flags} [boolean]`, description)
option.argParser(value => booleanParser(value))
// don't set a default value of `false`, because `undefined` is accepted by the procedure
if (isValueRequired) option.default(false)
else if (defaultValue.exists) option.default(defaultValue.value)
addOption(option)
negate()
return
}
let option: Option | null = null
if (propertyType === 'string') {
option = new Option(`${flags} ${bracketise('string')}`, description)
} else if (propertyType === 'boolean') {
option = new Option(flags, description)
} else if (propertyType === 'number' || propertyType === 'integer') {
option = new Option(`${flags} ${bracketise('number')}`, description)
option.argParser(value => numberParser(value, {fallback: null}))
} else if (propertyType === 'array') {
option = new Option(`${flags} [values...]`, description)
if (defaultValue.exists) option.default(defaultValue.value)
else if (isValueRequired) option.default([])
const itemsSchema = 'items' in propertyValue ? (propertyValue.items as JSONSchema7) : {}
const itemEnumTypes = getEnumChoices(itemsSchema)
if (itemEnumTypes?.type === 'string_enum') {
option.choices(itemEnumTypes.choices)
}
const itemParser = getOptionValueParser(itemsSchema)
if (itemParser) {
option.argParser((value, previous): unknown[] => {
const parsed = itemParser(value)
return Array.isArray(previous) ? [...previous, parsed] : [parsed]
})
}
}
if (!option) {
option = new Option(`${flags} [json]`, description)
option.argParser(value => parseJson(value, InvalidOptionArgumentError))
}
if (defaultValue.exists && option.defaultValue !== defaultValue.value) {
option.default(defaultValue.value)
}
if (option.flags.includes('<')) {
option.makeOptionMandatory()
}
// Note: enum choices for union of literals are handled earlier (before allowedSchemas.length > 1 check)
// This handles z.enum() style enums which don't go through the multi-schema path
const propertyEnumChoices = getEnumChoices(propertyValue)
if (propertyEnumChoices?.type === 'string_enum') {
option.choices(propertyEnumChoices.choices)
}
option.conflicts(
incompatiblePairs.flatMap(pair => {
const filtered = pair.filter(p => p !== propertyKey)
if (filtered.length === pair.length) return []
return filtered
}),
)
addOption(option)
if (propertyType === 'boolean') negate() // just in case we refactor the code above and don't handle booleans as a special case
}
Object.entries(optionJsonSchemaProperties).forEach(addOptionForProperty)
const invalidOptionAliases = Object.entries(unusedOptionAliases).map(([option, alias]) => `${option}: ${alias}`)
if (invalidOptionAliases.length) {
throw new Error(`Invalid option aliases: ${invalidOptionAliases.join(', ')}`)
}
// Set the action for this command
command.action(async (...args) => {
program.__ran ||= []
program.__ran.push(command)
const options = command.opts()
if (args.at(-2) !== options) {
// This is a code bug and not recoverable. Will hopefully never happen but if commander totally changes their API this will break
throw new Error(`Unexpected args format, second last arg is not the options object`, {cause: args})
}
if (args.at(-1) !== command) {
// This is a code bug and not recoverable. Will hopefully never happen but if commander totally changes their API this will break
throw new Error(`Unexpected args format, last arg is not the Command instance`, {cause: args})
}
// the last arg is the Command instance itself, the second last is the options object, and the other args are positional
const positionalValues = args.slice(0, -2)
const input = parsedProcedure.getPojoInput({positionalValues, options})
let caller: Record<string, (input: unknown) => unknown>
const deprecatedCreateCaller = Reflect.get(params, 'createCallerFactory') as CreateCallerFactoryLike | undefined
if (deprecatedCreateCaller) {
const message = `Using deprecated \`createCallerFactory\` option. Use \`trpcServer\` instead. e.g. \`createCli({router: myRouter, trpcServer: import('@trpc/server')})\``
logger.error?.(message)
caller = deprecatedCreateCaller(router)(params.context)
} else if (isNorpcRouter(router)) {
// eslint-disable-next-line @typescript-eslint/no-explicit-any
let proc = router as any
for (const part of procedurePath.split('.')) proc = proc[part]
caller = {[procedurePath]: (_input: unknown) => proc.call(_input, params.context) as unknown}
} else if (isTrpcRouter(router)) {
const resolvedTrpcServer = await (params.trpcServer ||
(await import('@trpc/server').catch(e => {
throw new Error(`@trpc/server must be installed when using tRPC-style routers. Error loading: ${e}`)
})))
const createCallerFactor = resolvedTrpcServer.initTRPC.create().createCallerFactory as CreateCallerFactoryLike
caller = createCallerFactor(router)(params.context)
} else {
const {call} = getOrpcServerModule()
// create an object which acts enough like a trpc caller to be used for this specific procedure
const procedure = procedurePath
.split('.')
.reduce((acc, part) => acc[part] as Record<string, unknown>, router as Record<string, unknown>)
caller = {[procedurePath]: (_input: unknown) => call(procedure as never, _input, {context: params.context})}
}
// Derive leaf command's argv from the program's argv by stripping the routing segments.
// e.g. for procedurePath 'math.add' and program.__argv ['math', 'add', '--a', '2'],
// the leaf command gets ['--a', '2'].
const leafSegmentCount = procedurePath.split('.').length
command.__argv = (program.__argv ?? []).slice(leafSegmentCount)
const cliContext = {program, command}
const result = await runWithCliContext(cliContext, () =>
(caller[procedurePath](input) as Promise<unknown>).catch(err => {
throw transformError(err, command)
}),
)
command.__result = result
if (result != null) logger.info?.(result)
})
}
// Process each procedure and add as a command or subcommand
procedureEntries.forEach(([procedurePath, commandConfig]) => {
const segments = procedurePath.split('.')
// Create the command path and ensure parent commands exist
let currentPath = ''
for (let i = 0; i < segments.length - 1; i++) {
const segment = segments[i]
const parentPath = currentPath
currentPath = currentPath ? `${currentPath}.${segment}` : segment
// Create parent command if it doesn't exist
if (!commandTree[currentPath]) {
const parentCommand = commandTree[parentPath]
const newCommand = new Command(kebabCase(segment))
newCommand.showHelpAfterError()
parentCommand.addCommand(newCommand)
commandTree[currentPath] = newCommand
}
}
// Create the actual leaf command
const leafName = segments.at(-1)
const parentPath = segments.length > 1 ? segments.slice(0, -1).join('.') : ''
const parentCommand = commandTree[parentPath]
const leafCommand = new Command(leafName && kebabCase(leafName))
configureCommand(leafCommand, procedurePath, commandConfig)
parentCommand.addCommand(leafCommand)
// Check if this command should be the default for its parent
const meta = commandConfig.meta
if (meta.default === true) {
// the parent will pass on its args straight to the child, which will validate them. the parent just blindly accepts anything.
parentCommand.allowExcessArguments()
parentCommand.allowUnknownOption()
parentCommand.addHelpText('after', leafCommand.helpInformation())
parentCommand.action(async () => {
await leafCommand.parseAsync([...parentCommand.args], {from: 'user'})
})
// ancestors need to support positional options to pass through the positional args
// for (let ancestor = parentCommand.parent, i = 0; ancestor && i < 10; ancestor = ancestor.parent, i++) {
// ancestor.enablePositionalOptions()
// }
// parentCommand.passThroughOptions()
defaultCommands[parentPath] = {
procedurePath: procedurePath,
config: commandConfig,
command: leafCommand,
}
}
})
// After all commands are added, generate descriptions for parent commands
Object.entries(commandTree).forEach(([path, command]) => {
// Skip the root command and leaf commands (which already have descriptions)
// if (path === '' || command.commands.length === 0) return
if (command.commands.length === 0) return
// Get the names of all direct subcommands
const subcommandNames = command.commands.map(cmd => cmd.name())
// Check if there's a default command for this path
const defaultCommand = defaultCommands[path]?.command.name()
// Format the subcommand list, marking the default one
const formattedSubcommands = subcommandNames
.map(name => (name === defaultCommand ? `${name} (default)` : name))
.join(', ')
// Get the existing description (might have been set by a default command)
const existingDescription = command.description() || ''
// Only add the subcommand list if it's not already part of the description
const descriptionParts = [existingDescription, `Available subcommands: ${formattedSubcommands}`]
command.description(descriptionParts.filter(Boolean).join('\n'))
})
return program
}
const run: TrpcCli['run'] = async (runParams?: TrpcCliRunParams, program = buildProgram(runParams)) => {
if (!looksLikeInstanceof<Command>(program, 'Command')) throw new Error(`program is not a Command instance`)
const opts = runParams?.argv ? ({from: 'user'} as const) : undefined
const argv = [...(runParams?.argv || process.argv)]
const _process = runParams?.process || process
const logger = {...lineByLineConsoleLogger, ...runParams?.logger}
program.exitOverride(exit => {
_process.exit(exit.exitCode)
throw new FailedToExitError('Root command exitOverride', {exitCode: exit.exitCode, cause: exit})
})
program.configureOutput({
writeOut: str => logger.info?.(str),
writeErr: str => logger.error?.(str),
})
if (runParams?.completion) {
const completion =
typeof runParams.completion === 'function' ? await runParams.completion() : runParams.completion
addCompletions(program, completion)
}
const formatError =
runParams?.formatError ||
((err: unknown) => {
if (err instanceof CliValidationError) {
return err.message
}
return inspect(err)
})
if (runParams?.prompts) {
program = promptify(program, runParams.prompts) as Command
}
// Store the argv that commander will parse in "user" mode.
// When argv is provided via run({argv}), it's used as-is (already "user" args).
// When falling back to process.argv, strip the `node script.js` prefix.
;(program as Command).__argv = runParams?.argv ? argv : argv.slice(2)
await program.parseAsync(argv, opts).catch(err => {
if (err instanceof FailedToExitError) throw err
const logMessage = looksLikeInstanceof(err, Error)
? formatError(err) || err.message
: `Non-error of type ${typeof err} thrown: ${err}`
logger.error?.(logMessage)
_process.exit(1)
throw new FailedToExitError(`Program exit after failure`, {exitCode: 1, cause: err})
})
_process.exit(0)
throw new FailedToExitError('Program exit after success', {
exitCode: 0,
cause: (program as Command).__ran.at(-1)?.__result,
})
}
return {run, buildProgram, toJSON: (program = buildProgram()) => commandToJSON(program as Command)}
}
export const kebabCase = (str: string) =>
str
.replaceAll(/([\da-z])([A-Z])/g, '$1-$2')
.replaceAll(/([A-Z]+)([A-Z][a-z])/g, '$1-$2')
.toLowerCase()
/** @deprecated renamed to `createCli` */
export const trpcCli = createCli
function transformError(err: unknown, command: Command) {
if (looksLikeInstanceof(err, Error) && err.message.includes('This is a client-only function')) {
return new Error(
'Failed to create trpc caller. If using trpc v10, either upgrade to v11 or pass in the `@trpc/server` module to `createCli` explicitly',
)
}
type TRPCErrorLike = Error & {cause: Error; code: 'BAD_REQUEST' | 'INTERNAL_SERVER_ERROR' | (string & {})}
if (looksLikeInstanceof<TRPCErrorLike>(err, 'TRPCError') || looksLikeInstanceof<TRPCErrorLike>(err, 'ORPCError')) {
const cause = err.cause
if (looksLikeStandardSchemaFailure(cause)) {
const prettyMessage = prettifyStandardSchemaError(cause)
return new CliValidationError(prettyMessage + '\n\n' + command.helpInformation())
}
if (
err.code === 'BAD_REQUEST' &&
(err.cause?.constructor?.name === 'TraversalError' || // arktype error
err.cause?.constructor?.name === 'StandardSchemaV1Error') // valibot error
) {
return new CliValidationError(err.cause.message + '\n\n' + command.helpInformation())
}
if (err.code === 'INTERNAL_SERVER_ERROR') {
return cause
}
}
return err
}
export {FailedToExitError, CliValidationError} from './errors.js'
export {getCliContext, type CliContextValue, type CliCommand} from './context.js'
export {
autoTableConsoleLogger,
autoTableLogger,
lineByLineConsoleLogger,
lineByLineLogger,
yamlConsoleLogger,
yamlTableConsoleLogger,
yamlTableLogger,
yamlLogger,
} from './logging.js'
export {toYaml} from './yaml.js'
const numberParser = (val: string, {fallback = val as unknown} = {}) => {
const number = Number(val)
return Number.isNaN(number) ? fallback : number
}
const booleanParser = (val: string, {fallback = val as unknown} = {}) => {
if (val === 'true') return true
if (val === 'false') return false
return fallback
}
const getOptionValueParser = (schema: JSONSchema7) => {
const allowedSchemas = getAllowedSchemas(schema)
.slice()
.sort((a, b) => String(getSchemaTypes(a)[0]).localeCompare(String(getSchemaTypes(b)[0])))
const typesArray = allowedSchemas.flatMap(getSchemaTypes)
const types = new Set(typesArray)
return (value: string) => {
const definitelyPrimitive = typesArray.every(
t => t === 'boolean' || t === 'number' || t === 'integer' || t === 'string',
)
if (types.size === 0 || !definitelyPrimitive) {
// parse this as JSON - too risky to fall back to a string because that will probably do the wrong thing if someone passes malformed JSON like `{"foo": 1,}` (trailing comma)
const hint = `Malformed JSON. If passing a string, pass it as a valid JSON string with quotes (${JSON.stringify(value)})`
const parsed = parseJson(value, InvalidOptionArgumentError, hint)
if (!types.size) return parsed // if types is empty, it means any type is allowed - e.g. for json input
const jsonSchemaType = Array.isArray(parsed) ? 'array' : parsed === null ? 'null' : typeof parsed
if (!types.has(jsonSchemaType)) {
throw new InvalidOptionArgumentError(`Got ${jsonSchemaType} but expected ${[...types].join(' or ')}`)
}
return parsed
}
if (types.has('boolean')) {
const parsed = booleanParser(value, {fallback: null})
if (typeof parsed === 'boolean') return parsed
}
if (types.has('number')) {
const parsed = numberParser(value, {fallback: null})
if (typeof parsed === 'number') return parsed
}
if (types.has('integer')) {
const parsed = numberParser(value, {fallback: null})
if (typeof parsed === 'number' && Number.isInteger(parsed)) return parsed
}
if (types.has('string')) {
return value
}
throw new InvalidOptionArgumentError(`Got ${JSON.stringify(value)} but expected ${[...types].join(' or ')}`)
}
}
const parseJson = (
value: string,
ErrorClass: new (message: string) => Error = InvalidArgumentError,
hint = `Malformed JSON.`,
) => {
try {
return JSON.parse(value) as {}
} catch {
throw new ErrorClass(hint)
}
}
export {t, os} from './norpc.js'