-
Notifications
You must be signed in to change notification settings - Fork 403
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: add terminal command execution tool with user approval
- Loading branch information
Showing
4 changed files
with
291 additions
and
68 deletions.
There are no files selected for viewing
82 changes: 82 additions & 0 deletions
82
packages/ai-native/src/browser/mcp/tools/components/Terminal.tsx
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,82 @@ | ||
| import React, { memo, useCallback, useMemo, useState } from 'react'; | ||
|
|
||
| import { useInjectable } from '@opensumi/ide-core-browser'; | ||
| import { Button, Icon } from '@opensumi/ide-core-browser/lib/components'; | ||
|
|
||
| import { IMCPServerToolComponentProps } from '../../../types'; | ||
| import { RunCommandHandler } from '../handlers/RunCommand'; | ||
|
|
||
| import styles from './index.module.less'; | ||
|
|
||
| function getResult(raw: string) { | ||
| const result: { | ||
| isError?: boolean; | ||
| text?: string; | ||
| } = {}; | ||
|
|
||
| try { | ||
| const data = JSON.parse(raw); | ||
| if (data.isError) { | ||
| result.isError = data.isError; | ||
| } | ||
|
|
||
| if (data.content) { | ||
| result.text = data.content; | ||
| } | ||
|
|
||
| return result; | ||
| } catch { | ||
| return null; | ||
| } | ||
| } | ||
|
|
||
| export const TerminalToolComponent = memo((props: IMCPServerToolComponentProps) => { | ||
| const { args, toolCallId } = props; | ||
| const handler = useInjectable<RunCommandHandler>(RunCommandHandler); | ||
| const [disabled, toggleDisabled] = useState(false); | ||
|
|
||
| const handleClick = useCallback((approval: boolean) => { | ||
| handler.handleApproval(toolCallId!, approval); | ||
| toggleDisabled(true); | ||
| }, []); | ||
|
|
||
| const output = useMemo(() => { | ||
| if (props.result) { | ||
| return getResult(props.result); | ||
| } | ||
| return null; | ||
| }, [props]); | ||
|
|
||
| return ( | ||
| <div className={styles.run_cmd_tool}> | ||
| {props.state === 'result' && ( | ||
| <div> | ||
| <div className={styles.command_title}> | ||
| <Icon icon='terminal' /> | ||
| <span>输出</span> | ||
| </div> | ||
| {output ? <div className={styles.command_content}>{output.text}</div> : ''} | ||
| </div> | ||
| )} | ||
|
|
||
| {props.state === 'complete' && args?.require_user_approval && ( | ||
| <div> | ||
| <div className={styles.command_title}> | ||
| <Icon icon='terminal' /> | ||
| <span>是否允许运行命令?</span> | ||
| </div> | ||
| <p className={styles.command_content}>{args.command}</p> | ||
| <p className={styles.comand_description}>{args.explanation}</p> | ||
| <div className={styles.cmmand_footer}> | ||
| <Button type='link' size='small' disabled={disabled} onClick={() => handleClick(true)}> | ||
| 允许 | ||
| </Button> | ||
| <Button type='link' size='small' disabled={disabled} onClick={() => handleClick(false)}> | ||
| 不允许 | ||
| </Button> | ||
| </div> | ||
| </div> | ||
| )} | ||
| </div> | ||
| ); | ||
| }); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
107 changes: 107 additions & 0 deletions
107
packages/ai-native/src/browser/mcp/tools/handlers/RunCommand.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,107 @@ | ||
| import z from 'zod'; | ||
|
|
||
| import { Autowired, Injectable } from '@opensumi/di'; | ||
| import { AppConfig } from '@opensumi/ide-core-browser'; | ||
| import { ITerminalController, ITerminalGroupViewService } from '@opensumi/ide-terminal-next'; | ||
| import { Deferred } from '@opensumi/ide-utils/lib/promises'; | ||
|
|
||
| import { MCPLogger } from '../../../types'; | ||
|
|
||
| const color = { | ||
| italic: '\x1b[3m', | ||
| reset: '\x1b[0m', | ||
| }; | ||
|
|
||
| export const inputSchema = z.object({ | ||
| command: z.string().describe('The terminal command to execute'), | ||
| is_background: z.boolean().describe('Whether the command should be run in the background'), | ||
| explanation: z | ||
| .string() | ||
| .describe('One sentence explanation as to why this command needs to be run and how it contributes to the goal.'), | ||
| require_user_approval: z | ||
| .boolean() | ||
| .describe( | ||
| "Whether the user must approve the command before it is executed. Only set this to false if the command is safe and if it matches the user's requirements for commands that should be executed automatically.", | ||
| ), | ||
| }); | ||
|
|
||
| @Injectable() | ||
| export class RunCommandHandler { | ||
| @Autowired(ITerminalController) | ||
| protected readonly terminalController: ITerminalController; | ||
|
|
||
| @Autowired(AppConfig) | ||
| protected readonly appConfig: AppConfig; | ||
|
|
||
| @Autowired(ITerminalGroupViewService) | ||
| protected readonly terminalView: ITerminalGroupViewService; | ||
|
|
||
| private approvalDeferredMap = new Map<string, Deferred<boolean>>(); | ||
|
|
||
| private terminalId = 0; | ||
|
|
||
| getShellLaunchConfig(command: string) { | ||
| return { | ||
| name: `MCP:Terminal_${this.terminalId++}`, | ||
| cwd: this.appConfig.workspaceDir, | ||
| args: ['-c', command], | ||
| }; | ||
| } | ||
|
|
||
| async handler(args: z.infer<typeof inputSchema> & { toolCallId: string }, logger: MCPLogger) { | ||
| if (args.require_user_approval) { | ||
| const def = new Deferred<boolean>(); | ||
| this.approvalDeferredMap.set(args.toolCallId, def); | ||
| const approval = await def.promise; | ||
| if (!approval) { | ||
| return { | ||
| isError: false, | ||
| content: 'User rejection', | ||
| }; | ||
| } | ||
| } | ||
| const terminalClient = await this.terminalController.createTerminalWithWidget({ | ||
| config: this.getShellLaunchConfig(args.command), | ||
| closeWhenExited: false, | ||
| }); | ||
|
|
||
| this.terminalController.showTerminalPanel(); | ||
|
|
||
| const result: string[] = []; | ||
| const def = new Deferred<{ isError?: boolean; content: string[] }>(); | ||
|
|
||
| terminalClient.onOutput((e) => { | ||
| result.push(e.data.toString()); | ||
| }); | ||
|
|
||
| terminalClient.onExit((e) => { | ||
| const isError = e.code !== 0; | ||
| def.resolve({ | ||
| isError, | ||
| content: result, | ||
| }); | ||
|
|
||
| terminalClient.term.writeln( | ||
| `\n${color.italic}> Command ${args.command} executed successfully. Terminal will close in ${ | ||
| 3000 / 1000 | ||
| } seconds.${color.reset}\n`, | ||
| ); | ||
|
|
||
| setTimeout(() => { | ||
| terminalClient.dispose(); | ||
| this.terminalView.removeWidget(terminalClient.id); | ||
| }, 3000); | ||
| }); | ||
|
|
||
| return def.promise; | ||
| } | ||
|
|
||
| handleApproval(callId: string, approval: boolean) { | ||
| if (!this.approvalDeferredMap.has(callId)) { | ||
| return; | ||
| } | ||
|
|
||
| const def = this.approvalDeferredMap.get(callId); | ||
| def?.resolve(approval); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters