Merge pull request microsoft#253040 from microsoft/copilot/fix-253029

Add Fig completion spec for `code chat` CLI subcommand

Author: Tyriar

extensions/terminal-suggest/src/completions/code.ts

@@ -549,7 +549,7 @@ export const extTunnelSubcommand = {
 };
 
 
-export const codeTunnelSubcommands = [
+export const codeTunnelSubcommands: Fig.Subcommand[] = [
 	{
 		name: 'tunnel',
 		description: 'Create a tunnel that\'s accessible on vscode.dev from anywhere. Run`code tunnel --help` for more usage info',
@@ -797,6 +797,39 @@ export const codeTunnelSubcommands = [
 			},
 		],
 	},
+	{
+		name: 'chat',
+		description: 'Pass in a prompt to run in a chat session in the current working directory.',
+		args: {
+			name: 'prompt',
+			description: 'The prompt to use as chat',
+			isVariadic: true,
+			isOptional: true,
+		},
+		options: [
+			{
+				name: ['-m', '--mode'],
+				description: 'The mode to use for the chat session. Defaults to \'agent\'',
+				args: {
+					name: 'mode',
+					suggestions: ['agent', 'ask', 'edit'],
+				},
+			},
+			{
+				name: ['-a', '--add-file'],
+				description: 'Add files as context to the chat session',
+				isRepeatable: true,
+				args: {
+					name: 'file',
+					template: 'filepaths',
+				},
+			},
+			{
+				name: ['-h', '--help'],
+				description: 'Print usage',
+			},
+		],
+	},
 	{
 		name: 'status',
 		description: 'Print process usage and diagnostics information',
@@ -953,6 +986,10 @@ export const codeTunnelSubcommands = [
 					}
 				],
 			},
+			{
+				name: 'chat',
+				description: 'Pass in a prompt to run in a chat session in the current working directory.',
+			},
 			extTunnelSubcommand,
 			{
 				name: 'status',

extensions/terminal-suggest/src/test/completions/code.test.ts

@@ -60,6 +60,7 @@ export const codeSpecOptionsAndSubcommands = [
 	'--version',
 	'--wait',
 	'tunnel',
+	'chat [<prompt>]',
 	'serve-web',
 	'help',
 	'status',
@@ -71,6 +72,7 @@ export function createCodeTestSpecs(executable: string): ITestSpec[] {
 	const categoryOptions = ['azure', 'data science', 'debuggers', 'extension packs', 'education', 'formatters', 'keymaps', 'language packs', 'linters', 'machine learning', 'notebooks', 'programming languages', 'scm providers', 'snippets', 'testing', 'themes', 'visualization', 'other'];
 	const logOptions = ['critical', 'error', 'warn', 'info', 'debug', 'trace', 'off'];
 	const syncOptions = ['on', 'off'];
+	const chatOptions = ['--add-file <file>', '--help', '--mode <mode>', '-a <file>', '-h', '-m <mode>'];
 
 	const typingTests: ITestSpec[] = [];
 	for (let i = 1; i < executable.length; i++) {
@@ -119,6 +121,11 @@ export function createCodeTestSpecs(executable: string): ITestSpec[] {
 		{ input: `${executable} --category |`, expectedCompletions: categoryOptions },
 		{ input: `${executable} --category a|`, expectedCompletions: categoryOptions },
 
+		// Chat subcommand tests
+		{ input: `${executable} chat |`, expectedCompletions: chatOptions },
+		{ input: `${executable} chat --mode |`, expectedCompletions: ['agent', 'ask', 'edit'] },
+		{ input: `${executable} chat --add-file |`, expectedResourceRequests: { type: 'files', cwd: testPaths.cwd } },
+
 		// Middle of command
 		{ input: `${executable} | --locale`, expectedCompletions: codeSpecOptionsAndSubcommands, expectedResourceRequests: { type: 'both', cwd: testPaths.cwd } },
 	];
@@ -177,6 +184,7 @@ export function createCodeTunnelTestSpecs(executable: string): ITestSpec[] {
 		'-s',
 		'-v',
 		'-w',
+		'chat [<prompt>]',
 		'ext',
 		'help',
 		'serve-web',
@@ -273,6 +281,9 @@ export function createCodeTunnelTestSpecs(executable: string): ITestSpec[] {
 		{ input: `${executable} tunnel unregister |`, expectedCompletions: [...commonFlags] },
 		{ input: `${executable} tunnel service |`, expectedCompletions: [...commonFlags, 'help', 'install', 'log', 'uninstall'] },
 		{ input: `${executable} tunnel help |`, expectedCompletions: helpSubcommands },
+		{ input: `${executable} chat |`, expectedCompletions: ['--mode <mode>', '--add-file <file>', '--help', '-m <mode>', '-a <file>', '-h'] },
+		{ input: `${executable} chat --mode |`, expectedCompletions: ['agent', 'ask', 'edit'] },
+		{ input: `${executable} chat --add-file |`, expectedResourceRequests: { type: 'files', cwd: testPaths.cwd } },
 		{ input: `${executable} serve-web |`, expectedCompletions: serveWebSubcommandsAndFlags },
 		{ input: `${executable} ext |`, expectedCompletions: extSubcommands },
 		{ input: `${executable} ext list |`, expectedCompletions: [...commonFlags, '--category [<category>]', '--show-versions'] },