@opensumi/ide-ai-native 3.7.2-next-1740107209.0 → 3.7.2-next-1740365741.0

package/src/browser/mcp/tools/components/SearchResult.tsx

@@ -0,0 +1,92 @@
+import React, { useEffect, useMemo, useState } from 'react';
+
+import { LabelService, URI, path, useInjectable } from '@opensumi/ide-core-browser';
+import { WorkbenchEditorService } from '@opensumi/ide-editor';
+import { IWorkspaceService } from '@opensumi/ide-workspace';
+
+import { IChatInternalService } from '../../../../common';
+import { ChatInternalService } from '../../../chat/chat.internal.service';
+
+import styles from './index.module.less';
+
+interface SearchResultProps {
+  args: any;
+  toolCallId: string;
+  messageId: string;
+  toolName: string;
+}
+
+export const FileSearchToolComponent: React.FC<SearchResultProps> = ({ args, toolCallId, messageId }) => (
+  <SearchResult args={args} toolCallId={toolCallId} messageId={messageId} toolName='fileSearch' />
+);
+
+export const GrepSearchToolComponent: React.FC<SearchResultProps> = ({ args, toolCallId, messageId }) => (
+  <SearchResult args={args} toolCallId={toolCallId} messageId={messageId} toolName='grepSearch' />
+);
+
+const SearchResult: React.FC<SearchResultProps> = ({ args, toolCallId, toolName, messageId }) => {
+  const [isExpanded, setIsExpanded] = useState(false);
+  const labelService = useInjectable<LabelService>(LabelService);
+  const editorService = useInjectable<WorkbenchEditorService>(WorkbenchEditorService);
+  const workspaceService = useInjectable<IWorkspaceService>(IWorkspaceService);
+  const workspaceRoot = useMemo(() => URI.parse(workspaceService.tryGetRoots()?.[0]?.uri), []);
+
+  const chatService = useInjectable<ChatInternalService>(IChatInternalService);
+  const [files, setFiles] = useState<string[]>(
+    chatService.sessionModel.history.getMessageAdditional(messageId)?.[toolCallId]?.files || [],
+  );
+  useEffect(() => {
+    const toDispose = chatService.sessionModel.history.onMessageAdditionalChange((additional) => {
+      setFiles(additional[toolCallId]?.files || []);
+    });
+    return () => {
+      toDispose.dispose();
+    };
+  }, []);
+
+  const handleFileClick = (uri: URI) => {
+    // 处理文件点击跳转
+    editorService.open(uri);
+  };
+
+  const parsedFiles = useMemo(
+    () =>
+      files.map((file) => {
+        const uri = URI.parse(file);
+        const iconClass = labelService.getIcon(uri);
+        return {
+          iconClass,
+          name: uri.path.base,
+          path: path.relative(workspaceRoot.codeUri.fsPath, uri.path.dir.toString()),
+        };
+      }),
+    [files],
+  );
+
+  return (
+    <div className={styles.container}>
+      <div className={styles.header} onClick={() => setIsExpanded(!isExpanded)}>
+        <span style={{ transform: `rotate(${isExpanded ? '90deg' : '0deg'})` }}>▶</span>
+        <span>
+          {toolName === 'fileSearch' ? `Searched files "${args.query}"` : `Grepped codebase "${args.query}"`} ·{' '}
+          {files.length} files
+        </span>
+      </div>
+      {isExpanded && (
+        <ul className={styles.fileList}>
+          {parsedFiles.map((file, index) => (
+            <li
+              key={index}
+              className={styles.fileItem}
+              onClick={() => handleFileClick(URI.file(path.join(workspaceRoot.codeUri.fsPath, file.path, file.name)))}
+            >
+              <span className={file.iconClass}></span>
+              <span style={{ flex: 1 }}>{file.name}</span>
+              <span className={styles.filePath}>{file.path}</span>
+            </li>
+          ))}
+        </ul>
+      )}
+    </div>
+  );
+};

package/src/browser/mcp/tools/components/Terminal.tsx

@@ -0,0 +1,97 @@
+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 { localize } from '@opensumi/ide-core-common';
+
+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: {
+      content: { type: string; text: string }[];
+      isError?: boolean;
+    } = JSON.parse(raw);
+    if (data.isError) {
+      result.isError = data.isError;
+    }
+
+    if (data.content) {
+      result.text = data.content.map((item) => item.text).join('\n');
+    }
+
+    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) => {
+    if (!toolCallId) {
+      return;
+    }
+    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>{localize('ai.native.mcp.terminal.output')}</span>
+          </div>
+          {output ? (
+            <div className={styles.command_content}>
+              <code>{output.text}</code>
+            </div>
+          ) : (
+            ''
+          )}
+        </div>
+      )}
+
+      {props.state === 'complete' && args?.require_user_approval && (
+        <div>
+          <div className={styles.command_title}>
+            <Icon icon='terminal' />
+            <span>{localize('ai.native.mcp.terminal.allow-question')}</span>
+          </div>
+          <p className={styles.command_content}>
+            <code>$ {args.command}</code>
+          </p>
+          <p className={styles.comand_description}>{args.explanation}</p>
+          <div className={styles.cmmand_footer}>
+            <Button type='link' size='small' disabled={disabled} onClick={() => handleClick(true)}>
+              {localize('ai.native.mcp.terminal.allow')}
+            </Button>
+            <Button type='link' size='small' disabled={disabled} onClick={() => handleClick(false)}>
+              {localize('ai.native.mcp.terminal.deny')}
+            </Button>
+          </div>
+        </div>
+      )}
+    </div>
+  );
+});

package/src/browser/mcp/tools/components/index.module.less

@@ -65,3 +65,104 @@
 .warning > span {
   color: var(--debugConsole-warningForeground);
 }
+
+.container {
+  border: 1px solid var(--vscode-commandCenter-inactiveBorder);
+  border-radius: 8px;
+  margin: 8px 0;
+  overflow: hidden;
+}
+
+.header {
+  padding: 8px 12px;
+  background-color: var(--design-block-background);
+  border-bottom: 1px solid var(--vscode-commandCenter-inactiveBorder);
+  cursor: pointer;
+  display: flex;
+  align-items: center;
+  gap: 4px;
+  color: var(--design-text-foreground);
+  font-size: 12px;
+}
+
+.fileList {
+  margin: 0;
+  padding: 0;
+  list-style: none;
+  display: block;
+  background-color: var(--design-block-background);
+}
+
+.fileList.collapsed {
+  display: none;
+}
+
+.fileItem {
+  padding: 2px 6px;
+  font-size: 12px;
+  margin: 0px 6px;
+  border-radius: 4px;
+  display: flex;
+  align-items: center;
+  gap: 8px;
+  cursor: pointer;
+  color: var(--design-text-primary);
+  &:hover {
+    background-color: var(--design-block-background);
+  }
+  > span {
+    overflow: hidden;
+    text-overflow: ellipsis;
+    white-space: nowrap;
+  }
+}
+
+.fileIcon {
+  color: var(--design-text-secondary);
+  font-size: 12px;
+  width: 16px;
+}
+
+.filePath {
+  color: var(--design-text-secondary);
+  font-size: 12px;
+  margin-left: auto;
+  flex-basis: 0px;
+  flex-grow: 1;
+}
+
+.run_cmd_tool {
+  .command_title {
+    display: flex;
+    align-items: center;
+    span {
+      margin-left: 5px;
+    }
+  }
+
+  .command_content {
+    padding: 4px;
+    font-size: 12px;
+    color: var(--design-text-foreground);
+    margin: 0px;
+    background-color: var(--terminal-background);
+    margin: 10px 0px;
+    border-radius: 4px;
+    overflow: auto;
+
+    code {
+      font-size: 12px;
+      white-space: pre;
+    }
+  }
+
+  .comand_description {
+    font-size: 11px;
+    color: var(--descriptionForeground);
+  }
+
+  .cmmand_footer {
+    display: flex;
+    justify-content: flex-end;
+  }
+}

package/src/browser/mcp/tools/fileSearch.ts

@@ -0,0 +1,99 @@
+import { z } from 'zod';
+
+import { Autowired } from '@opensumi/di';
+import { Domain, URI } from '@opensumi/ide-core-common';
+import { FileSearchServicePath, IFileSearchService } from '@opensumi/ide-file-search/lib/common';
+import { IWorkspaceService } from '@opensumi/ide-workspace';
+
+import { IChatInternalService } from '../../../common';
+import { ChatInternalService } from '../../chat/chat.internal.service';
+import { IMCPServerRegistry, MCPLogger, MCPServerContribution, MCPToolDefinition } from '../../types';
+
+import { FileSearchToolComponent } from './components/SearchResult';
+
+const inputSchema = z.object({
+  query: z.string().describe('Fuzzy filename to search for'),
+  explanation: z
+    .string()
+    .describe('One sentence explanation as to why this tool is being used, and how it contributes to the goal.'),
+});
+
+const MAX_RESULTS = 10;
+
+@Domain(MCPServerContribution)
+export class FileSearchTool implements MCPServerContribution {
+  @Autowired(IWorkspaceService)
+  private readonly workspaceService: IWorkspaceService;
+
+  @Autowired(FileSearchServicePath)
+  private readonly fileSearchService: IFileSearchService;
+
+  @Autowired(IChatInternalService)
+  private readonly chatInternalService: ChatInternalService;
+
+  registerMCPServer(registry: IMCPServerRegistry): void {
+    registry.registerMCPTool(this.getToolDefinition());
+    registry.registerToolComponent('file_search', FileSearchToolComponent);
+  }
+
+  getToolDefinition(): MCPToolDefinition {
+    return {
+      name: 'file_search',
+      label: 'Search Files',
+      description:
+        "Fast file search based on fuzzy matching against file path. Use if you know part of the file path but don't know where it's located exactly. Response will be capped to 10 results. Make your query more specific if need to filter results further.",
+      inputSchema,
+      handler: this.handler.bind(this),
+    };
+  }
+
+  private async handler(args: z.infer<typeof inputSchema> & { toolCallId: string }, logger: MCPLogger) {
+    if (!args.query) {
+      throw new Error('No fileSearch parameters provided. Need to give a query.');
+    }
+    // 获取工作区根目录
+    const workspaceRoots = this.workspaceService.tryGetRoots();
+    if (!workspaceRoots || workspaceRoots.length === 0) {
+      throw new Error('Cannot determine project directory');
+    }
+
+    // 使用 OpenSumi 的文件搜索 API
+    const searchPattern = args.query;
+    const searchResults = await this.fileSearchService.find(searchPattern, {
+      rootUris: [new URI(workspaceRoots[0].uri).codeUri.fsPath],
+      // TODO: 忽略配置
+      excludePatterns: ['**/node_modules/**'],
+      limit: 100,
+      useGitIgnore: true,
+      noIgnoreParent: true,
+      fuzzyMatch: true,
+    });
+
+    const files = searchResults.slice(0, MAX_RESULTS).map((file) => {
+      const uri = URI.parse(file);
+      return uri.codeUri.fsPath;
+    });
+
+    const messages = this.chatInternalService.sessionModel.history.getMessages();
+    this.chatInternalService.sessionModel.history.setMessageAdditional(messages[messages.length - 1].id, {
+      [args.toolCallId]: {
+        files,
+      },
+    });
+
+    logger.appendLine(`Found ${files.length} files matching "${args.query}"`);
+
+    return {
+      content: [
+        {
+          type: 'text',
+          text: `${files.join('\n')}\n${
+            searchResults.length > MAX_RESULTS
+              ? `\nFound ${searchResults.length} files matching "${args.query}", only return the first ${MAX_RESULTS} results`
+              : ''
+          }`,
+        },
+      ],
+    };
+  }
+}

package/src/browser/mcp/tools/grepSearch.ts

@@ -0,0 +1,121 @@
+import { z } from 'zod';
+
+import { Autowired } from '@opensumi/di';
+import { CancellationToken, Deferred, Domain } from '@opensumi/ide-core-common';
+import { ContentSearchResult, IContentSearchClientService } from '@opensumi/ide-search';
+import { ContentSearchClientService } from '@opensumi/ide-search/lib/browser/search.service';
+import { IWorkspaceService } from '@opensumi/ide-workspace';
+
+import { IChatInternalService } from '../../../common';
+import { ChatInternalService } from '../../chat/chat.internal.service';
+import { IMCPServerRegistry, MCPLogger, MCPServerContribution, MCPToolDefinition } from '../../types';
+
+import { GrepSearchToolComponent } from './components/SearchResult';
+
+const inputSchema = z.object({
+  query: z.string().describe('The regex pattern to search for'),
+  case_sensitive: z.boolean().optional().describe('Whether the search should be case sensitive'),
+  include_pattern: z
+    .string()
+    .optional()
+    .describe('Glob pattern for files to include (e.g. "*.ts" for TypeScript files)'),
+  exclude_pattern: z.string().optional().describe('Glob pattern for files to exclude'),
+  explanation: z
+    .string()
+    .optional()
+    .describe('One sentence explanation as to why this tool is being used, and how it contributes to the goal.'),
+});
+
+const MAX_RESULTS = 50;
+
+@Domain(MCPServerContribution)
+export class GrepSearchTool implements MCPServerContribution {
+  @Autowired(IWorkspaceService)
+  private readonly workspaceService: IWorkspaceService;
+
+  @Autowired(IContentSearchClientService)
+  private readonly searchService: ContentSearchClientService;
+
+  @Autowired(IChatInternalService)
+  private readonly chatInternalService: ChatInternalService;
+
+  registerMCPServer(registry: IMCPServerRegistry): void {
+    registry.registerMCPTool(this.getToolDefinition());
+    registry.registerToolComponent('grep_search', GrepSearchToolComponent);
+  }
+
+  getToolDefinition(): MCPToolDefinition {
+    return {
+      name: 'grep_search',
+      label: 'Search Contents',
+      description:
+        // TODO: 支持语义化搜索后需要描述清楚优劣势
+        'Fast text-based regex search that finds exact pattern matches within files or directories, utilizing the ripgrep command for efficient searching.\nResults will be formatted in the style of ripgrep and can be configured to include line numbers and content.\nTo avoid overwhelming output, the results are capped at 50 matches.\nUse the include or exclude patterns to filter the search scope by file type or specific paths.\n\nThis is best for finding exact text matches or regex patterns.',
+      inputSchema,
+      handler: this.handler.bind(this),
+    };
+  }
+
+  private async handler(args: z.infer<typeof inputSchema> & { toolCallId: string }, logger: MCPLogger) {
+    if (!args.query) {
+      throw new Error('No ripgrep search parameters provided. Need to give at least a query.');
+    }
+    // 获取工作区根目录
+    const workspaceRoots = this.workspaceService.tryGetRoots();
+    if (!workspaceRoots || workspaceRoots.length === 0) {
+      throw new Error('Cannot determine project directory');
+    }
+
+    // 使用 OpenSumi 的文件搜索 API
+    const searchPattern = args.query;
+    await this.searchService.doSearch(
+      searchPattern,
+      {
+        isMatchCase: !!args.case_sensitive,
+        include: args.include_pattern?.split(','),
+        exclude: args.exclude_pattern?.split(','),
+        maxResults: MAX_RESULTS,
+        isUseRegexp: true,
+        isToggleOpen: false,
+        isDetailOpen: false,
+        isWholeWord: false,
+        isOnlyOpenEditors: false,
+        isIncludeIgnored: false,
+      },
+      CancellationToken.None,
+    );
+    const deferred = new Deferred<string>();
+    this.searchService.onDidChange(() => {
+      if (this.searchService.isSearching) {
+        return;
+      }
+      const results: string[] = [];
+      const files: string[] = [];
+      for (const [fileUri, result] of this.searchService.searchResults.entries()) {
+        results.push(
+          `File: ${fileUri}\n${result
+            .reduce((acc, r) => {
+              if (acc.find((a) => a.line === r.line)) {
+                return acc;
+              }
+              return [...acc, r];
+            }, [] as ContentSearchResult[])
+            .map((r) => `Line: ${r.line}\nContent: ${r.lineText || r.renderLineText}`)
+            .join('\n')}`,
+        );
+        files.push(fileUri);
+      }
+      deferred.resolve(results.join('\n\n'));
+      const messages = this.chatInternalService.sessionModel.history.getMessages();
+      this.chatInternalService.sessionModel.history.setMessageAdditional(messages[messages.length - 1].id, {
+        [args.toolCallId]: {
+          files,
+        },
+      });
+    });
+    const text = await deferred.promise;
+    return {
+      content: [{ type: 'text', text }],
+    };
+  }
+}

package/src/browser/mcp/tools/handlers/RunCommand.ts

@@ -0,0 +1,115 @@
+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: [
+            {
+              type: 'text',
+              text: 'User rejection',
+            },
+          ],
+        };
+      }
+    }
+    const terminalClient = await this.terminalController.createTerminalWithWidget({
+      config: this.getShellLaunchConfig(args.command),
+      closeWhenExited: false,
+    });
+
+    this.terminalController.showTerminalPanel();
+
+    const result: { type: string; text: string }[] = [];
+    const def = new Deferred<{ isError?: boolean; content: { type: string; text: string }[] }>();
+
+    terminalClient.onOutput((e) => {
+      result.push({
+        type: 'text',
+        text: 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);
+  }
+}

package/src/browser/mcp/tools/listDir.ts

@@ -14,6 +14,7 @@ const inputSchema = z
       .describe("Path to list contents of, relative to the workspace root. Ex: './' is the root of the workspace"),
     explanation: z
       .string()
+      .optional()
       .describe('One sentence explanation as to why this tool is being used, and how it contributes to the goal.'),
   })
   .transform((data) => ({

package/src/browser/mcp/tools/readFile.ts

@@ -15,6 +15,7 @@ const inputSchema = z
     end_line_one_indexed_inclusive: z.number().describe('The one-indexed line number to end reading at (inclusive).'),
     explanation: z
       .string()
+      .optional()
       .describe('One sentence explanation as to why this tool is being used, and how it contributes to the goal.'),
   })
   .transform((data) => ({