mirror of
https://github.com/microsoft/vscode.git
synced 2026-05-03 06:51:53 +01:00
f1a5fb082b
* Removes HTTP logic * enable/disables automation tools based on if app is opened * kill and re-create the playwright server based on if app is opened
84 lines
2.7 KiB
TypeScript
84 lines
2.7 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the MIT License. See License.txt in the project root for license information.
|
|
*--------------------------------------------------------------------------------------------*/
|
|
|
|
import { McpServer, RegisteredTool } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
import { ApplicationService } from '../application';
|
|
import { z } from 'zod';
|
|
|
|
/**
|
|
* Task Tools
|
|
*/
|
|
export function applyTaskTools(server: McpServer, appService: ApplicationService): RegisteredTool[] {
|
|
const tools: RegisteredTool[] = [];
|
|
|
|
// Seems too niche
|
|
// server.tool(
|
|
// 'vscode_automation_task_assert_tasks',
|
|
// 'Assert that specific tasks exist with given properties',
|
|
// {
|
|
// filter: z.string().describe('Filter string for tasks'),
|
|
// expected: z.array(z.object({
|
|
// label: z.string().optional(),
|
|
// type: z.string().optional(),
|
|
// command: z.string().optional(),
|
|
// identifier: z.string().optional(),
|
|
// group: z.string().optional(),
|
|
// isBackground: z.boolean().optional(),
|
|
// promptOnClose: z.boolean().optional(),
|
|
// icon: z.object({
|
|
// id: z.string().optional(),
|
|
// color: z.string().optional()
|
|
// }).optional(),
|
|
// hide: z.boolean().optional()
|
|
// })).describe('Array of expected task properties'),
|
|
// type: z.enum(['run', 'configure']).describe('Type of task operation')
|
|
// },
|
|
// async (args) => {
|
|
// const { filter, expected, type } = args;
|
|
// await app.workbench.task.assertTasks(filter, expected, type);
|
|
// return {
|
|
// content: [{
|
|
// type: 'text' as const,
|
|
// text: `Asserted ${expected.length} tasks for ${type} with filter: "${filter}"`
|
|
// }]
|
|
// };
|
|
// }
|
|
// );
|
|
|
|
tools.push(server.tool(
|
|
'vscode_automation_task_configure',
|
|
'Configure a task with specific properties',
|
|
{
|
|
properties: z.object({
|
|
label: z.string().optional(),
|
|
type: z.string().optional(),
|
|
command: z.string().optional(),
|
|
identifier: z.string().optional(),
|
|
group: z.string().optional(),
|
|
isBackground: z.boolean().optional(),
|
|
promptOnClose: z.boolean().optional(),
|
|
icon: z.object({
|
|
id: z.string().optional(),
|
|
color: z.string().optional()
|
|
}).optional(),
|
|
hide: z.boolean().optional()
|
|
}).describe('Task configuration properties')
|
|
},
|
|
async (args) => {
|
|
const { properties } = args;
|
|
const app = await appService.getOrCreateApplication();
|
|
await app.workbench.task.configureTask(properties);
|
|
return {
|
|
content: [{
|
|
type: 'text' as const,
|
|
text: `Configured task: ${properties.label || properties.identifier || 'unnamed task'}`
|
|
}]
|
|
};
|
|
}
|
|
));
|
|
|
|
return tools;
|
|
}
|