mirror of
https://github.com/ckaczor/azuredatastudio.git
synced 2026-02-13 19:48:37 -05:00
85 lines
3.0 KiB
TypeScript
85 lines
3.0 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the Source EULA. See License.txt in the project root for license information.
|
|
*--------------------------------------------------------------------------------------------*/
|
|
|
|
import * as assert from 'assert';
|
|
import { firstIndex } from 'vs/base/common/arrays';
|
|
import { localize } from 'vs/nls';
|
|
import { MIN_MAX_MEMORY_SIZE_MB } from 'vs/platform/files/node/files';
|
|
import { parseArgs, ErrorReporter, OPTIONS, ParsedArgs } from 'vs/platform/environment/node/argv';
|
|
|
|
function parseAndValidate(cmdLineArgs: string[], reportWarnings: boolean): ParsedArgs {
|
|
const errorReporter: ErrorReporter = {
|
|
onUnknownOption: (id) => {
|
|
console.warn(localize('unknownOption', "Warning: '{0}' is not in the list of known options, but still passed to Electron/Chromium.", id));
|
|
},
|
|
onMultipleValues: (id, val) => {
|
|
console.warn(localize('multipleValues', "Option '{0}' is defined more than once. Using value '{1}.'", id, val));
|
|
}
|
|
};
|
|
|
|
const args = parseArgs(cmdLineArgs, OPTIONS, reportWarnings ? errorReporter : undefined);
|
|
if (args.goto) {
|
|
args._.forEach(arg => assert(/^(\w:)?[^:]+(:\d*){0,2}$/.test(arg), localize('gotoValidation', "Arguments in `--goto` mode should be in the format of `FILE(:LINE(:CHARACTER))`.")));
|
|
}
|
|
|
|
if (args['max-memory']) {
|
|
assert(parseInt(args['max-memory']) >= MIN_MAX_MEMORY_SIZE_MB, `The max-memory argument cannot be specified lower than ${MIN_MAX_MEMORY_SIZE_MB} MB.`);
|
|
}
|
|
|
|
return args;
|
|
}
|
|
|
|
function stripAppPath(argv: string[]): string[] | undefined {
|
|
const index = firstIndex(argv, a => !/^-/.test(a));
|
|
|
|
if (index > -1) {
|
|
return [...argv.slice(0, index), ...argv.slice(index + 1)];
|
|
}
|
|
return undefined;
|
|
}
|
|
|
|
/**
|
|
* Use this to parse raw code process.argv such as: `Electron . --verbose --wait`
|
|
*/
|
|
export function parseMainProcessArgv(processArgv: string[]): ParsedArgs {
|
|
let [, ...args] = processArgv;
|
|
|
|
// If dev, remove the first non-option argument: it's the app location
|
|
if (process.env['VSCODE_DEV']) {
|
|
args = stripAppPath(args) || [];
|
|
}
|
|
|
|
// If called from CLI, don't report warnings as they are already reported.
|
|
let reportWarnings = !process.env['VSCODE_CLI'];
|
|
return parseAndValidate(args, reportWarnings);
|
|
}
|
|
|
|
/**
|
|
* Use this to parse raw code CLI process.argv such as: `Electron cli.js . --verbose --wait`
|
|
*/
|
|
export function parseCLIProcessArgv(processArgv: string[]): ParsedArgs {
|
|
let [, , ...args] = processArgv;
|
|
|
|
if (process.env['VSCODE_DEV']) {
|
|
args = stripAppPath(args) || [];
|
|
}
|
|
|
|
return parseAndValidate(args, true);
|
|
}
|
|
|
|
export function addArg(argv: string[], ...args: string[]): string[] {
|
|
const endOfArgsMarkerIndex = argv.indexOf('--');
|
|
if (endOfArgsMarkerIndex === -1) {
|
|
argv.push(...args);
|
|
} else {
|
|
// if the we have an argument "--" (end of argument marker)
|
|
// we cannot add arguments at the end. rather, we add
|
|
// arguments before the "--" marker.
|
|
argv.splice(endOfArgsMarkerIndex, 0, ...args);
|
|
}
|
|
|
|
return argv;
|
|
}
|