VSCode扩展开发：如何拦截修改搜索值并为全场景搜索应用正则？

完全可以开发这类扩展，通过VSCode提供的API拦截各类搜索操作，修改搜索值并应用自定义正则规则，覆盖全局文件夹搜索、编辑器内联搜索等所有场景。以下是具体实现思路和代码示例：

核心实现思路 #

通过覆盖原生搜索命令和监听搜索配置/状态变化两种方式，拦截用户的搜索输入，应用正则规则修改后，再传递给VSCode原生搜索逻辑。

关键实现步骤与代码示例 #

1. 覆盖原生搜索触发命令 #

针对全局搜索（文件夹搜索）和内联搜索，分别替换原生命令，在触发时修改搜索值：

import * as vscode from 'vscode';

export function activate(context: vscode.ExtensionContext) {
    // 覆盖全局搜索命令（Ctrl+Shift+F）
    const findInFilesDisposable = vscode.commands.registerCommand('workbench.action.findInFiles', async () => {
        const currentQuery = vscode.workspace.getConfiguration('search').get('query') as string || '';
        const modifiedQuery = applyRegexRules(currentQuery);
        await vscode.commands.executeCommand('workbench.action.findInFiles', { query: modifiedQuery });
    });

    // 覆盖内联搜索命令（Ctrl+F）
    const inlineSearchDisposable = vscode.commands.registerCommand('editor.action.startFindReplaceAction', async () => {
        const activeEditor = vscode.window.activeTextEditor;
        if (!activeEditor) return;
        
        const selectedText = activeEditor.document.getText(activeEditor.selection);
        const modifiedQuery = applyRegexRules(selectedText || '');
        await vscode.commands.executeCommand('editor.action.startFindReplaceAction', { query: modifiedQuery });
    });

    context.subscriptions.push(findInFilesDisposable, inlineSearchDisposable);
}

// 自定义正则规则应用函数
function applyRegexRules(original: string): string {
    // 示例：将空格替换为.*实现模糊匹配，可按需修改规则
    return original.replace(/\s+/g, '.*');
}

2. 实时修改搜索框输入 #

如果需要在用户输入过程中实时修改搜索内容，可监听搜索配置或编辑器状态变化：

// 监听全局搜索框内容变化，实时修改
const configChangeDisposable = vscode.workspace.onDidChangeConfiguration(e => {
    if (e.affectsConfiguration('search.query')) {
        const currentQuery = vscode.workspace.getConfiguration('search').get('query') as string || '';
        const modifiedQuery = applyRegexRules(currentQuery);
        if (modifiedQuery !== currentQuery) {
            vscode.workspace.getConfiguration('search').update('query', modifiedQuery, vscode.ConfigurationTarget.Workspace);
        }
    }
});

// 监听内联搜索框内容变化，实时修改
const inlineSearchChangeDisposable = vscode.commands.registerCommand('editor.action.changeFindString', (args) => {
    const modified = applyRegexRules(args);
    vscode.commands.executeCommand('editor.action.changeFindString', modified);
});

context.subscriptions.push(configChangeDisposable, inlineSearchChangeDisposable);

3. 支持用户自定义正则规则 #

在扩展配置中添加自定义规则项，让用户自行配置正则替换逻辑：
在扩展的package.json中添加配置声明：

"contributes": {
    "configuration": {
        "title": "Search Modifier",
        "properties": {
            "searchModifier.regexRules": {
                "type": "array",
                "items": {
                    "type": "object",
                    "properties": {
                        "pattern": { "type": "string", "description": "正则匹配模式" },
                        "replacement": { "type": "string", "description": "替换内容" }
                    }
                },
                "description": "自定义搜索查询的正则替换规则"
            }
        }
    }
}

修改applyRegexRules函数读取配置：

function applyRegexRules(original: string): string {
    const rules = vscode.workspace.getConfiguration('searchModifier').get('regexRules') as Array<{pattern: string, replacement: string}> || [];
    let modified = original;
    rules.forEach(rule => {
        modified = modified.replace(new RegExp(rule.pattern, 'g'), rule.replacement);
    });
    return modified;
}

注意事项 #

• 命令兼容性：不同VSCode版本的原生搜索命令ID可能略有差异，需在目标版本中测试验证。
• 循环触发处理：实时修改搜索内容时，必须判断修改前后的内容是否一致，避免无限触发配置/状态变化事件。
• 替换场景支持：如果需要覆盖替换操作（Ctrl+H），可采用类似逻辑拦截workbench.action.replaceInFiles和editor.action.startFindReplaceAction命令。

内容的提问来源于stack exchange，提问作者Good guy