Function runCommand

src/spec-common/commonUtils.ts:182–251

(options: {
	ptyExec: PtyExecFunction;
	cmd: string;
	args?: string[];
	cwd?: string;
	env?: NodeJS.ProcessEnv;
	output: Log;
	resolveOn?: RegExp;
	onDidInput?: Event<string>;
	stdin?: string;
	print?: 'off' | 'continuous' | 'end';
})

}

export async function runCommand(options: {
	ptyExec: PtyExecFunction;
	cmd: string;
	args?: string[];
	cwd?: string;
	env?: NodeJS.ProcessEnv;
	output: Log;
	resolveOn?: RegExp;
	onDidInput?: Event<string>;
	stdin?: string;
	print?: 'off' | 'continuous' | 'end';
}) {
	const { ptyExec, cmd, args, cwd, env, output, resolveOn, onDidInput, stdin } = options;
	const print = options.print || 'continuous';

	const p = await ptyExec({
		cmd,
		args,
		cwd,
		env,
		output: output,
	});

	return new Promise<{ cmdOutput: string }>((resolve, reject) => {
		let cmdOutput = '';

		const subs: Disposable[] = [];
		if (p.write) {
			if (stdin) {
				p.write(stdin);
			}
			if (onDidInput) {
				subs.push(onDidInput(data => p.write!(data)));
			}
		}

		p.onData(chunk => {
			cmdOutput += chunk;
			if (print === 'continuous') {
				output.raw(chunk);
			}
			if (resolveOn && resolveOn.exec(cmdOutput)) {
				resolve({ cmdOutput });
			}
		});
		p.exit.then(({ code, signal }) => {
			try {
				if (print === 'end') {
					output.raw(cmdOutput);
				}
				subs.forEach(sub => sub?.dispose());
				if (code || signal) {
					reject({
						message: `Command failed: ${cmd} ${(args || []).join(' ')}`,
						cmdOutput,
						code,
						signal,
					});