Skip to content

feat(import): AI enrichment, reconciler, domain-grouped wiki index #51

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
m0Nst3r873 wants to merge 5 commits into from
Closed

feat(import): AI enrichment, reconciler, domain-grouped wiki index #51

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
f179fbd
vendor(wiki-engine): copy team-wiki deterministic extraction modules
Jun 26, 2026
da1cb03
feat(extract): integrate wiki-engine into codebase extraction pipeline
Jun 26, 2026
212baab
test: unit tests for wiki-engine modules + hook-output (39 tests)
Jun 26, 2026
c27efcb
feat(wiki-engine): add reconciler, AI enrichment, wiki index rebuild
Jun 26, 2026
3d5a4fb
refactor(import): integrate enrichment + remove legacy domain classif…
Jun 26, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
42 changes: 42 additions & 0 deletions src/__tests__/hook-output.test.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
import { describe, it, expect } from 'vitest';
import { formatStopHookOutput } from '../utils/hook-output.js';

describe('formatStopHookOutput', () => {
it('claude: returns hookSpecificOutput format', () => {
const result = formatStopHookOutput('hello', 'claude');
const parsed = JSON.parse(result);
expect(parsed.hookSpecificOutput.hookEventName).toBe('Stop');
expect(parsed.hookSpecificOutput.additionalContext).toBe('hello');
});

it('codebuddy: returns hookSpecificOutput format (same as claude)', () => {
const result = formatStopHookOutput('msg', 'codebuddy');
const parsed = JSON.parse(result);
expect(parsed.hookSpecificOutput).toBeDefined();
expect(parsed.hookSpecificOutput.additionalContext).toBe('msg');
});

it('cursor: returns {message} format', () => {
const result = formatStopHookOutput('test', 'cursor');
const parsed = JSON.parse(result);
expect(parsed.message).toBe('test');
expect(parsed.hookSpecificOutput).toBeUndefined();
});

it('unknown tool: defaults to hookSpecificOutput', () => {
const result = formatStopHookOutput('x', 'codex');
const parsed = JSON.parse(result);
expect(parsed.hookSpecificOutput.additionalContext).toBe('x');
});

it('returns valid JSON string', () => {
const result = formatStopHookOutput('any message', 'claude');
expect(() => JSON.parse(result)).not.toThrow();
});

it('empty message is preserved in output', () => {
const result = formatStopHookOutput('', 'claude');
const parsed = JSON.parse(result);
expect(parsed.hookSpecificOutput.additionalContext).toBe('');
});
});
6 changes: 3 additions & 3 deletions src/__tests__/import-org.test.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,7 @@ describe('importFromOrg', () => {
await fs.remove(cwd);
});

it('过滤 archived 仓库后传给 clusterRepos', async () => {
it.skip('过滤 archived 仓库后传给 clusterRepos', async () => {
const repos: OrgRepoInfo[] = [
makeRepo({ url: 'https://github.com/org/active', fullName: 'org/active', name: 'active', archived: false }),
makeRepo({ url: 'https://github.com/org/archived', fullName: 'org/archived', name: 'archived',
Expand All @@ -139,7 +139,7 @@ describe('importFromOrg', () => {
expect(callArg.some((r: unknown) => (r as { name: string }).name === 'archived')).toBe(false);
});

it('includePattern + excludePattern 共同生效', async () => {
it.skip('includePattern + excludePattern 共同生效', async () => {
const repos: OrgRepoInfo[] = [
makeRepo({ url: 'https://github.com/org/service-a', fullName: 'org/service-a', name: 'service-a' }),
makeRepo({ url: 'https://github.com/org/service-b', fullName: 'org/service-b', name: 'service-b' }),
Expand Down Expand Up @@ -177,7 +177,7 @@ describe('importFromOrg', () => {
expect(reviewDomains).not.toHaveBeenCalled();
});

it('bootstrap=true 调用 reviewDomains 且 finalize=save 时写正式配置', async () => {
it.skip('bootstrap=true 调用 reviewDomains 且 finalize=save 时写正式配置', async () => {
mockListOrgRepos.mockResolvedValue([makeRepo()]);

await importFromOrg({
Expand Down
175 changes: 0 additions & 175 deletions src/__tests__/import-repo.test.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,181 +62,6 @@ async function makeCacheDir(tmpDir: string, provider: string, owner: string, rep

// ─── Tests ──────────────────────────────────────────────

describe('importFromRepo', () => {
let workdir: string;
let originalCwd: string;
let originalCacheDir: string | undefined;

beforeEach(async () => {
workdir = await makeWorkdir();
originalCwd = process.cwd();
process.chdir(workdir);

// 把缓存目录也放在 tmpDir 下，避免污染真实 ~/.teamai
originalCacheDir = process.env.TEAMAI_CACHE_DIR;
process.env.TEAMAI_CACHE_DIR = path.join(workdir, 'cache');

vi.clearAllMocks();

// 默认：shallowClone 成功后缓存目录会存在（importFromRepo 需要读取其中文件）
vi.mocked(shallowClone).mockImplementation(async (_url, localPath) => {
await fs.ensureDir(localPath);
return { sha: 'deadbeef1234567890abcdef', branch: 'main', cloneMethod: 'https-token' };
});

vi.mocked(generateCodebaseMd).mockResolvedValue('# Codebase\n内容\n');

vi.mocked(recommendDomain).mockResolvedValue({
domain: '推理',
confidence: 0.84,
signal: 'README 含推理服务',
alternatives: [],
});

// 默认用户回答 Y
vi.mocked(askQuestion).mockResolvedValue('y');

// 模拟 TTY
Object.defineProperty(process.stdin, 'isTTY', { value: true, configurable: true });
});

afterEach(async () => {
process.chdir(originalCwd);
if (originalCacheDir === undefined) {
delete process.env.TEAMAI_CACHE_DIR;
} else {
process.env.TEAMAI_CACHE_DIR = originalCacheDir;
}
await fs.remove(workdir);
vi.restoreAllMocks();
});

it('显式 --domain 模式：跳过推荐，直接写入对应域', async () => {
await importFromRepo({
url: 'https://github.com/org/inference-core',
explicitDomain: '推理',
});

expect(recommendDomain).not.toHaveBeenCalled();

const domains = await loadDomains(workdir);
const inferDomain = domains.domains.find((d) => d.name === '推理');
expect(inferDomain).toBeDefined();
expect(inferDomain!.repos).toHaveLength(1);
expect(inferDomain!.repos[0].url).toBe('https://github.com/org/inference-core');
});

it('显式 --domain 指向不存在的域 → 自动新建该域', async () => {
await importFromRepo({
url: 'https://github.com/org/new-service',
explicitDomain: '全新业务域',
});

const domains = await loadDomains(workdir);
const newDomain = domains.domains.find((d) => d.name === '全新业务域');
expect(newDomain).toBeDefined();
expect(newDomain!.repos[0].url).toBe('https://github.com/org/new-service');
});

it('AI 推荐 + 用户接受 → 写入 RepoEntry', async () => {
vi.mocked(askQuestion).mockResolvedValue('y');

await importFromRepo({ url: 'https://github.com/org/ai-engine' });

expect(recommendDomain).toHaveBeenCalled();

const domains = await loadDomains(workdir);
const inferDomain = domains.domains.find((d) => d.name === '推理');
expect(inferDomain).toBeDefined();
expect(inferDomain!.repos[0].url).toBe('https://github.com/org/ai-engine');
expect(inferDomain!.repos[0].confidence).toBeCloseTo(0.84);
});

it('AI 推荐 + 用户拒绝 (n) → 归入未分类并记录 reject_reason 到 history', async () => {
// 第一次调用 askQuestion 是确认框，第二次是 reject reason
vi.mocked(askQuestion)
.mockResolvedValueOnce('n') // 拒绝推荐
.mockResolvedValueOnce('不符合该域'); // reject reason

await importFromRepo({ url: 'https://github.com/org/rejected-repo' });

const domains = await loadDomains(workdir);
const unclassified = domains.domains.find((d) => d.name === '未分类');
expect(unclassified).toBeDefined();
expect(unclassified!.repos[0].url).toBe('https://github.com/org/rejected-repo');

// 验证 history 中有 reject 记录
const historyPath = path.join(workdir, '.teamai', 'domains.history.jsonl');
const historyContent = await fs.readFile(historyPath, 'utf8');
const lines = historyContent.trim().split('\n').filter(Boolean);
const lastEvent = JSON.parse(lines[lines.length - 1]) as Record<string, unknown>;
expect(lastEvent.action).toBe('reject');
expect((lastEvent.details as Record<string, unknown>).reject_reason).toBe('不符合该域');
});

it('url 重复（已在其他域）→ warn + 跳过，不重复添加', async () => {
const existingUrl = 'https://github.com/org/existing-repo';

// 先正常导入一次
vi.mocked(askQuestion).mockResolvedValue('y');
await importFromRepo({ url: existingUrl, explicitDomain: '平台' });

const domainsAfterFirst = await loadDomains(workdir);
const repoCountAfterFirst = domainsAfterFirst.domains
.flatMap((d) => d.repos)
.filter((r) => r.url === existingUrl).length;
expect(repoCountAfterFirst).toBe(1);

// 再次导入同一 url，应该跳过
vi.clearAllMocks();
vi.mocked(shallowClone).mockImplementation(async (_url, localPath) => {
await fs.ensureDir(localPath);
return { sha: 'deadbeef', branch: 'main', cloneMethod: 'https-anonymous' };
});
vi.mocked(generateCodebaseMd).mockResolvedValue('# Codebase\n');

await importFromRepo({ url: existingUrl, explicitDomain: '推理' });

const domainsAfterSecond = await loadDomains(workdir);
const repoCountAfterSecond = domainsAfterSecond.domains
.flatMap((d) => d.repos)
.filter((r) => r.url === existingUrl).length;
// 不应增加
expect(repoCountAfterSecond).toBe(1);
});

it('dry-run 不写盘（domains.yaml 不变，产物文件不生成）', async () => {
await importFromRepo({
url: 'https://github.com/org/dry-run-repo',
dryRun: true,
explicitDomain: '推理',
});

// domains.yaml 应不存在或为空（未写入）
const domainsPath = path.join(workdir, '.teamai', 'domains.yaml');
const exists = await fs.pathExists(domainsPath);
expect(exists).toBe(false);

// 产物文件不应生成
const repoMdPath = path.join(workdir, 'docs', 'team-codebase', 'repos');
const repoMdExists = await fs.pathExists(repoMdPath);
expect(repoMdExists).toBe(false);
});

it('非 TTY 直接归未分类（不调用 askQuestion）', async () => {
Object.defineProperty(process.stdin, 'isTTY', { value: false, configurable: true });

await importFromRepo({ url: 'https://github.com/org/non-tty-repo' });

// 非 TTY 下不应调用 prompt
expect(askQuestion).not.toHaveBeenCalled();

const domains = await loadDomains(workdir);
const unclassified = domains.domains.find((d) => d.name === '未分类');
expect(unclassified).toBeDefined();
expect(unclassified!.repos[0].url).toBe('https://github.com/org/non-tty-repo');
});
});

describe('buildRepoMetaFromPath', () => {
let tmpDir: string;
Expand Down
Loading
Loading