feat(permission): 实现 WebSocket 权限确认机制

重构权限系统,将终端 UI 代码从 core 模块移除,实现基于 WebSocket 的权限确认流程:

Core 模块清理:
- 删除 permission/prompt.ts 和 file-prompt.ts(终端交互)
- 删除 diff.ts 中的 chalk 渲染函数
- 删除 config.ts 中的 inquirer 交互
- 移除 chalk 依赖

Server 权限处理:
- 新增 permission/handler.ts,实现 WebSocket 权限请求/响应
- 更新 agent/adapter.ts 设置权限回调
- 更新 ws.ts 处理 permission_response 消息

Web 权限组件:
- 新增 PermissionDialog 组件,显示权限请求详情和 Diff
- 更新 useChat hook 管理权限状态
- 更新 Chat 页面集成权限弹窗
This commit is contained in:
2025-12-13 01:09:35 +08:00
parent 5d4afecd48
commit 1d69fd876d
20 changed files with 739 additions and 1560 deletions
@@ -1,39 +1,5 @@
import { describe, it, expect, vi, beforeEach } from 'vitest';
import {
computeDiff,
formatDiff,
countChanges,
formatEditDiff,
confirmFileChange,
} from '../../../src/utils/diff.js';
// Mock readline
vi.mock('readline', () => ({
createInterface: vi.fn(() => ({
question: vi.fn(),
close: vi.fn(),
})),
}));
// Mock fs/promises
vi.mock('fs/promises', () => ({
readFile: vi.fn(),
}));
// Mock chalk (保留原始功能以便测试输出格式)
vi.mock('chalk', () => ({
default: {
yellow: (s: string) => `[yellow]${s}[/yellow]`,
cyan: (s: string) => `[cyan]${s}[/cyan]`,
white: (s: string) => `[white]${s}[/white]`,
gray: (s: string) => `[gray]${s}[/gray]`,
red: (s: string) => `[red]${s}[/red]`,
green: (s: string) => `[green]${s}[/green]`,
},
}));
import * as readline from 'readline';
import * as fs from 'fs/promises';
import { describe, it, expect } from 'vitest';
import { computeDiff, countChanges } from '../../../src/utils/diff.js';
describe('Diff - 差异比较扩展测试', () => {
describe('computeDiff - 计算 diff', () => {
@@ -120,45 +86,6 @@ describe('Diff - 差异比较扩展测试', () => {
});
});
describe('formatDiff - 格式化 diff', () => {
it('新文件显示 +++ 新文件标记', () => {
const diff = computeDiff(null, 'new content');
const formatted = formatDiff(diff, '/test/file.ts');
expect(formatted).toContain('新文件');
expect(formatted).toContain('/test/file.ts');
});
it('修改文件显示 --- 和 +++ 标记', () => {
const diff = computeDiff('old', 'new');
const formatted = formatDiff(diff, '/test/file.ts');
expect(formatted).toContain('原文件');
expect(formatted).toContain('修改后');
});
it('显示 hunk 头部 @@ 信息', () => {
const diff = computeDiff('old', 'new');
const formatted = formatDiff(diff, '/test/file.ts');
expect(formatted).toContain('@@');
});
it('添加行使用 + 前缀', () => {
const diff = computeDiff('', 'added line');
const formatted = formatDiff(diff, '/test/file.ts');
expect(formatted).toContain('+ added line');
});
it('删除行使用 - 前缀', () => {
const diff = computeDiff('removed line', '');
const formatted = formatDiff(diff, '/test/file.ts');
expect(formatted).toContain('- removed line');
});
});
describe('countChanges - 统计变更', () => {
it('统计添加行数', () => {
const diff = computeDiff(null, 'line1\nline2\nline3');
@@ -191,173 +118,4 @@ describe('Diff - 差异比较扩展测试', () => {
expect(changes.additions + changes.deletions).toBe(0);
});
});
describe('formatEditDiff - 编辑 diff 格式化', () => {
it('显示删除和添加内容', () => {
const formatted = formatEditDiff('old text', 'new text');
expect(formatted).toContain('old text');
expect(formatted).toContain('new text');
expect(formatted).toContain('-');
expect(formatted).toContain('+');
});
it('处理多行内容', () => {
const formatted = formatEditDiff('old1\nold2', 'new1\nnew2');
expect(formatted).toContain('old1');
expect(formatted).toContain('old2');
expect(formatted).toContain('new1');
expect(formatted).toContain('new2');
});
it('包含变更内容标题', () => {
const formatted = formatEditDiff('old', 'new');
expect(formatted).toContain('变更内容');
});
});
describe('confirmFileChange - 文件变更确认', () => {
let consoleLogSpy: ReturnType<typeof vi.spyOn>;
beforeEach(() => {
vi.clearAllMocks();
consoleLogSpy = vi.spyOn(console, 'log').mockImplementation(() => {});
});
it('内容相同直接返回确认', async () => {
vi.mocked(fs.readFile).mockResolvedValue('same content');
const result = await confirmFileChange('/test/file.ts', 'same content', 'write');
expect(result.confirmed).toBe(true);
expect(result.remember).toBe(false);
});
it('新文件(读取失败)显示 diff', async () => {
vi.mocked(fs.readFile).mockRejectedValue(new Error('ENOENT'));
const mockRl = {
question: vi.fn((_, callback) => callback('y')),
close: vi.fn(),
};
vi.mocked(readline.createInterface).mockReturnValue(mockRl as any);
const result = await confirmFileChange('/test/file.ts', 'new content', 'write');
expect(result.confirmed).toBe(true);
});
it('用户输入 y 确认写入', async () => {
vi.mocked(fs.readFile).mockResolvedValue('old content');
const mockRl = {
question: vi.fn((_, callback) => callback('y')),
close: vi.fn(),
};
vi.mocked(readline.createInterface).mockReturnValue(mockRl as any);
const result = await confirmFileChange('/test/file.ts', 'new content', 'write');
expect(result.confirmed).toBe(true);
expect(result.remember).toBe(false);
});
it('用户输入 Y 确认并记住', async () => {
vi.mocked(fs.readFile).mockResolvedValue('old content');
const mockRl = {
question: vi.fn((_, callback) => callback('Y')),
close: vi.fn(),
};
vi.mocked(readline.createInterface).mockReturnValue(mockRl as any);
const result = await confirmFileChange('/test/file.ts', 'new content', 'write');
expect(result.confirmed).toBe(true);
expect(result.remember).toBe(true);
});
it('用户输入 n 取消操作', async () => {
vi.mocked(fs.readFile).mockResolvedValue('old content');
const mockRl = {
question: vi.fn((_, callback) => callback('n')),
close: vi.fn(),
};
vi.mocked(readline.createInterface).mockReturnValue(mockRl as any);
const result = await confirmFileChange('/test/file.ts', 'new content', 'write');
expect(result.confirmed).toBe(false);
expect(result.remember).toBe(false);
});
it('用户输入 N 取消并记住', async () => {
vi.mocked(fs.readFile).mockResolvedValue('old content');
const mockRl = {
question: vi.fn((_, callback) => callback('N')),
close: vi.fn(),
};
vi.mocked(readline.createInterface).mockReturnValue(mockRl as any);
const result = await confirmFileChange('/test/file.ts', 'new content', 'write');
expect(result.confirmed).toBe(false);
expect(result.remember).toBe(true);
});
it('无效输入默认取消', async () => {
vi.mocked(fs.readFile).mockResolvedValue('old content');
const mockRl = {
question: vi.fn((_, callback) => callback('invalid')),
close: vi.fn(),
};
vi.mocked(readline.createInterface).mockReturnValue(mockRl as any);
const result = await confirmFileChange('/test/file.ts', 'new content', 'write');
expect(result.confirmed).toBe(false);
expect(result.remember).toBe(false);
});
it('显示变更预览信息', async () => {
vi.mocked(fs.readFile).mockResolvedValue('old content');
const mockRl = {
question: vi.fn((_, callback) => callback('y')),
close: vi.fn(),
};
vi.mocked(readline.createInterface).mockReturnValue(mockRl as any);
await confirmFileChange('/test/file.ts', 'new content', 'write');
const output = consoleLogSpy.mock.calls.flat().join('\n');
expect(output).toContain('文件变更预览');
expect(output).toContain('写入文件');
expect(output).toContain('/test/file.ts');
});
it('显示编辑操作类型', async () => {
vi.mocked(fs.readFile).mockResolvedValue('old content');
const mockRl = {
question: vi.fn((_, callback) => callback('y')),
close: vi.fn(),
};
vi.mocked(readline.createInterface).mockReturnValue(mockRl as any);
await confirmFileChange('/test/file.ts', 'new content', 'edit');
const output = consoleLogSpy.mock.calls.flat().join('\n');
expect(output).toContain('编辑文件');
});
afterEach(() => {
consoleLogSpy.mockRestore();
});
});
});