mirror of
https://github.com/balena-io/balena-cli.git
synced 2024-12-30 10:38:50 +00:00
86 lines
2.3 KiB
TypeScript
86 lines
2.3 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2019 Balena Ltd.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import { Main } from '@oclif/command';
|
|
import type * as Config from '@oclif/config';
|
|
|
|
/**
|
|
* This class is a partial copy-and-paste of
|
|
* @oclif/plugin-help/command/CommandHelp, which is used to generate oclif's
|
|
* command help output.
|
|
*/
|
|
export class CommandHelp {
|
|
constructor(public command: { args?: any[] }) {}
|
|
|
|
protected arg(arg: Config.Command['args'][0]): string {
|
|
const name = arg.name.toUpperCase();
|
|
if (arg.required) {
|
|
return `${name}`;
|
|
}
|
|
return `[${name}]`;
|
|
}
|
|
|
|
public defaultUsage(): string {
|
|
return CommandHelp.compact([
|
|
// this.command.id,
|
|
(this.command.args || [])
|
|
.filter((a) => !a.hidden)
|
|
.map((a) => this.arg(a))
|
|
.join(' '),
|
|
]).join(' ');
|
|
}
|
|
|
|
public static compact<T>(array: Array<T | undefined>): T[] {
|
|
return array.filter((a): a is T => !!a);
|
|
}
|
|
}
|
|
|
|
export class CustomMain extends Main {
|
|
protected _helpOverride(): boolean {
|
|
// Disable oclif's default handler for the 'version' command
|
|
if (['-v', '--version', 'version'].includes(this.argv[0])) {
|
|
return false;
|
|
} else {
|
|
return super._helpOverride();
|
|
}
|
|
}
|
|
}
|
|
|
|
/** Convert e.g. 'env add NAME [VALUE]' to 'env add <name> [value]' */
|
|
export function capitanoizeOclifUsage(
|
|
oclifUsage: string | string[] | undefined,
|
|
): string {
|
|
return (oclifUsage || '')
|
|
.toString()
|
|
.replace(/(?<=\s)[A-Z]+(?=(\s|$))/g, (match) => `<${match}>`)
|
|
.toLowerCase();
|
|
}
|
|
|
|
export async function getCommandsFromManifest() {
|
|
const manifest = require('../../oclif.manifest.json');
|
|
|
|
if (manifest.commands == null) {
|
|
throw new Error('Commands section not found in manifest.');
|
|
}
|
|
return manifest.commands;
|
|
}
|
|
|
|
export async function getCommandIdsFromManifest() {
|
|
const commands = await getCommandsFromManifest();
|
|
return Object.keys(commands);
|
|
}
|