Files
claude-code/src/commands/skill-search/skillSearchPanel.tsx
2026-04-22 22:38:09 +08:00

170 lines
5.1 KiB
TypeScript
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
import React, { useMemo, useState } from 'react';
import { Box, Text, useInput } from '@anthropic/ink';
import { Dialog } from '@anthropic/ink';
import { useRegisterOverlay } from '../../context/overlayContext.js';
import type { LocalJSXCommandOnDone } from '../../types/command.js';
import { isSkillSearchEnabled } from '../../services/skillSearch/featureCheck.js';
type SkillSearchAction = {
label: string;
description: string;
run: () => Promise<string>;
};
const ACTION_LABEL_COLUMN_WIDTH = 28;
const ABOUT_TEXT = `# Skill Search (自动技能匹配)
Skill Search 控制对话中的自动技能匹配功能。
启用后Claude Code 会在每轮对话中自动搜索并加载与当前任务最相关的 skill 文件,
无需手动指定。搜索基于 TF-IDF 向量余弦相似度,支持英文词干化和 CJK bi-gram 分词。
## 工作原理
1. 对话开始时,自动索引 .claude/skills/ 和 ~/.claude/skills/ 下的 Markdown 文件
2. 每轮对话根据上下文自动匹配最相关的 skill
3. 匹配到的 skill 内容会作为上下文注入,指导 Claude Code 的行为
## 控制方式
- /skill-search start — 启用自动匹配
- /skill-search stop — 禁用自动匹配
- /skill-search status — 查看当前状态
当前状态: ${isSkillSearchEnabled() ? '已启用' : '未启用'}
`;
function getStatusText(): string {
return [
'Skill Search (自动技能匹配)',
`Status: ${isSkillSearchEnabled() ? 'enabled' : 'disabled'}`,
'',
'When enabled, relevant skills are automatically matched and',
'injected into conversation context each turn.',
].join('\n');
}
async function startSkillSearch(): Promise<string> {
if (isSkillSearchEnabled() && process.env.SKILL_SEARCH_ENABLED !== '0') {
return 'Skill Search: already enabled';
}
process.env.SKILL_SEARCH_ENABLED = '1';
const lines = ['Skill Search: enabled (SKILL_SEARCH_ENABLED=1)'];
try {
const { clearSkillIndexCache } = await import('../../services/skillSearch/localSearch.js');
clearSkillIndexCache();
lines.push('Skill index cache: cleared (will rebuild on next search)');
} catch {
lines.push('Skill index cache: clear skipped');
}
return lines.join('\n');
}
async function stopSkillSearch(): Promise<string> {
if (!isSkillSearchEnabled()) {
return 'Skill Search: already disabled';
}
process.env.SKILL_SEARCH_ENABLED = '0';
return 'Skill Search: disabled (SKILL_SEARCH_ENABLED=0)';
}
function SkillSearchPanel({ onDone }: { onDone: LocalJSXCommandOnDone }): React.ReactNode {
useRegisterOverlay('skill-search-panel');
const [selectedIndex, setSelectedIndex] = useState(0);
const actions = useMemo<SkillSearchAction[]>(
() => [
{
label: 'Status',
description: 'Show whether automatic skill matching is active',
run: () => Promise.resolve(getStatusText()),
},
{
label: 'Start',
description: 'Enable automatic skill matching for this session',
run: startSkillSearch,
},
{
label: 'Stop',
description: 'Disable automatic skill matching for this session',
run: stopSkillSearch,
},
{
label: 'About',
description: 'How automatic skill matching works',
run: () => Promise.resolve(ABOUT_TEXT),
},
],
[],
);
const selectCurrent = () => {
const action = actions[selectedIndex];
if (!action) return;
void action.run().then(result => {
onDone(result, { display: 'system' });
});
};
useInput((_input, key) => {
if (key.upArrow) {
setSelectedIndex(index => Math.max(0, index - 1));
return;
}
if (key.downArrow) {
setSelectedIndex(index => Math.min(actions.length - 1, index + 1));
return;
}
if (key.return) {
selectCurrent();
}
});
return (
<Dialog
title="Skill Search"
subtitle={`${actions.length} actions`}
onCancel={() => onDone('Skill search panel dismissed', { display: 'system' })}
color="background"
hideInputGuide
>
<Box flexDirection="column">
{actions.map((action, index) => (
<Box key={action.label} flexDirection="row">
<Text>{`${index === selectedIndex ? '' : ' '} ${action.label}`.padEnd(ACTION_LABEL_COLUMN_WIDTH)}</Text>
<Text dimColor>{action.description}</Text>
</Box>
))}
<Box marginTop={1}>
<Text dimColor>/ select · Enter run · Esc close</Text>
</Box>
</Box>
</Dialog>
);
}
export async function call(onDone: LocalJSXCommandOnDone, _context: unknown, args?: string): Promise<React.ReactNode> {
const trimmed = args?.trim() ?? '';
if (trimmed === 'start') {
onDone(await startSkillSearch(), { display: 'system' });
return null;
}
if (trimmed === 'stop') {
onDone(await stopSkillSearch(), { display: 'system' });
return null;
}
if (trimmed === 'about') {
onDone(ABOUT_TEXT, { display: 'system' });
return null;
}
if (trimmed === 'status') {
onDone(getStatusText(), { display: 'system' });
return null;
}
return <SkillSearchPanel onDone={onDone} />;
}