VS Code扩展中检测Git分支切换：通过文件系统监控HEAD文件

npm install chokidar
npm install --save-dev @types/chokidar # 如果使用TypeScript

import * as vscode from 'vscode';
import * as path from 'path';
import * as fs from 'fs';
import chokidar from 'chokidar';

/**
 * 获取当前工作区的Git仓库根目录。
 * 简化处理，只考虑第一个工作区文件夹。
 */
function getGitRoot(workspaceFolders: readonly vscode.WorkspaceFolder[] | undefined): string | undefined {
    if (!workspaceFolders || workspaceFolders.length === 0) {
        return undefined;
    }
    // 假设我们只关心第一个工作区文件夹的Git仓库
    const workspacePath = workspaceFolders[0].uri.fsPath;
    const gitPath = path.join(workspacePath, '.git');
    if (fs.existsSync(gitPath) && fs.lstatSync(gitPath).isDirectory()) {
        return workspacePath;
    }
    // 如果工作区本身不是Git仓库，可能需要向上查找父目录，这里简化处理
    return undefined;
}

/**
 * 解析.git/HEAD文件内容，提取分支名称。
 * 例如："ref: refs/heads/main" -> "main"
 * 如果是分离头指针，则返回undefined。
 */
function parseHeadFile(headContent: string): string | undefined {
    const branchRefMatch = headContent.match(/^ref: refs\/heads\/(.*)$/);
    if (branchRefMatch && branchRefMatch[1]) {
        return branchRefMatch[1].trim();
    }
    // 如果是分离头指针（HEAD直接指向一个提交哈希），则不返回分支名
    return undefined;
}

export function activate(context: vscode.ExtensionContext) {
    const gitRoot = getGitRoot(vscode.workspace.workspaceFolders);

    if (!gitRoot) {
        vscode.window.showWarningMessage('未找到Git仓库根目录，无法监控Git HEAD文件。');
        return;
    }

    const headFilePath = path.join(gitRoot, '.git', 'HEAD');

    // 初始读取当前分支
    let currentBranch: string | undefined;
    if (fs.existsSync(headFilePath)) {
        try {
            const initialHeadContent = fs.readFileSync(headFilePath, 'utf8');
            currentBranch = parseHeadFile(initialHeadContent);
            console.log(`初始Git分支: ${currentBranch || '分离头指针'}`);
        } catch (error) {
            console.error('读取初始Git HEAD文件失败:', error);
        }
    }

    // 创建chokidar文件观察器
    const watcher = chokidar.watch(headFilePath, {
        persistent: true,     // 保持进程活跃
        ignoreInitial: true   // 启动时不触发'add'或'change'事件
    });

    watcher.on('change', async (filePath) => {
        console.log(`Git HEAD文件已更改: ${filePath}`);
        try {
            const headContent = fs.readFileSync(filePath, 'utf8');
            const newBranch = parseHeadFile(headContent);

            if (newBranch && newBranch !== currentBranch) {
                console.log(`Git分支已切换至: ${newBranch}`);
                vscode.window.showInformationMessage(`Git分支已切换至: ${newBranch}`);
                // 在这里触发你的扩展特定功能，例如：
                // yourExtensionService.onBranchChanged(newBranch);
                currentBranch = newBranch;
            } else if (!newBranch && currentBranch) {
                // 从命名分支切换到分离头指针状态
                console.log('已切换至分离头指针状态。');
                vscode.window.showInformationMessage('已切换至分离头指针状态。');
                currentBranch = undefined; // 或根据需要处理分离头指针
            } else if (newBranch === currentBranch) {
                // HEAD文件内容可能因其他原因（如fast-forward merge）而改变，但分支名未变
                console.log(`分支仍为 ${newBranch}，但HEAD文件内容可能已更改。`);
            }

        } catch (error) {
            console.error('读取Git HEAD文件失败:', error);
        }
    });

    watcher.on('error', (error) => {
        console.error('Chokidar观察器错误:', error);
        vscode.window.showErrorMessage(`Git HEAD文件监控发生错误: ${error.message}`);
    });

    // 当扩展停用时，清理观察器资源
    context.subscriptions.push(new vscode.Disposable(() => {
        watcher.close();
        console.log('Git HEAD文件观察器已关闭。');
    }));
}

export function deactivate() {
    // 扩展停用时，资源清理已通过 context.subscriptions 处理
}