351 lines
10 KiB
JavaScript
351 lines
10 KiB
JavaScript
/**
|
||
* requirement-generator-v1 Skill Web 服务器
|
||
*
|
||
* 功能：
|
||
* 1. 通过 SDK 调用 requirement-generator-v1 Skill
|
||
* 2. 支持 AskUserQuestion 交互
|
||
* 3. 支持多轮对话
|
||
* 4. 文档展示和下载
|
||
*/
|
||
|
||
import express from 'express';
|
||
import { query } from '@anthropic-ai/claude-agent-sdk';
|
||
import path from 'path';
|
||
import { fileURLToPath } from 'url';
|
||
import fs from 'fs';
|
||
|
||
const __dirname = path.dirname(fileURLToPath(import.meta.url));
|
||
|
||
// Skills 项目根目录（包含 .claude/skills/ 和 .claude/agents/）
|
||
const SKILLS_PROJECT_DIR = path.resolve(__dirname, '..');
|
||
|
||
const app = express();
|
||
const PORT = 3457;
|
||
|
||
app.use(express.json());
|
||
app.use(express.static(path.join(__dirname, 'public')));
|
||
|
||
// 存储待处理的 AskUserQuestion 请求
|
||
const pendingQuestions = new Map();
|
||
|
||
// 存储会话信息
|
||
const sessions = new Map();
|
||
|
||
/**
|
||
* SSE 流式执行 Skill
|
||
* GET /api/stream?prompt=xxx&sessionId=xxx
|
||
*/
|
||
app.get('/api/stream', async (req, res) => {
|
||
const { prompt, sessionId } = req.query;
|
||
const requestId = Date.now().toString();
|
||
|
||
const existingClaudeSessionId = sessionId ? sessions.get(sessionId) : null;
|
||
|
||
console.log('\n========================================');
|
||
console.log('[Server] 收到请求');
|
||
console.log('[Server] Prompt:', prompt);
|
||
console.log('[Server] SessionId:', existingClaudeSessionId || '(新会话)');
|
||
console.log('[Server] CWD:', SKILLS_PROJECT_DIR);
|
||
console.log('========================================\n');
|
||
|
||
res.setHeader('Content-Type', 'text/event-stream');
|
||
res.setHeader('Cache-Control', 'no-cache');
|
||
res.setHeader('Connection', 'keep-alive');
|
||
res.setHeader('X-Request-Id', requestId);
|
||
|
||
const sendEvent = (type, data) => {
|
||
res.write(`data: ${JSON.stringify({ type, ...data })}\n\n`);
|
||
};
|
||
|
||
try {
|
||
const options = {
|
||
// 项目目录 - 必须包含 .claude/skills/ 和 .claude/agents/
|
||
cwd: SKILLS_PROJECT_DIR,
|
||
|
||
// 权限模式 - 自动执行所有工具
|
||
permissionMode: 'bypassPermissions',
|
||
|
||
// 启用 Skill 和 Agent 加载
|
||
settingSources: ['user', 'project'],
|
||
|
||
// 不指定 allowedTools，让 SDK 使用默认值（允许所有工具）
|
||
|
||
// 继续会话
|
||
...(existingClaudeSessionId && { resume: existingClaudeSessionId }),
|
||
|
||
// 处理 AskUserQuestion
|
||
canUseTool: async (toolName, input) => {
|
||
console.log(`[SDK] canUseTool: ${toolName}`);
|
||
|
||
if (toolName === 'AskUserQuestion') {
|
||
sendEvent('ask_user_question', {
|
||
requestId,
|
||
questions: input.questions
|
||
});
|
||
|
||
return new Promise((resolve) => {
|
||
pendingQuestions.set(requestId, (answers) => {
|
||
console.log('[SDK] 收到用户回答:', answers);
|
||
resolve({
|
||
behavior: 'allow',
|
||
updatedInput: {
|
||
questions: input.questions,
|
||
answers: answers
|
||
}
|
||
});
|
||
});
|
||
|
||
// 60分钟超时
|
||
setTimeout(() => {
|
||
if (pendingQuestions.has(requestId)) {
|
||
console.log('[SDK] 等待用户回答超时 (60分钟)');
|
||
pendingQuestions.delete(requestId);
|
||
resolve({ behavior: 'deny' });
|
||
}
|
||
}, 3600000);
|
||
});
|
||
}
|
||
|
||
return { behavior: 'allow' };
|
||
}
|
||
};
|
||
|
||
sendEvent('status', {
|
||
message: existingClaudeSessionId
|
||
? `继续会话 ${existingClaudeSessionId.substring(0, 8)}...`
|
||
: '正在启动 Skill...'
|
||
});
|
||
|
||
const response = query({
|
||
prompt: prompt || '生成需求文档',
|
||
options
|
||
});
|
||
|
||
let currentClaudeSessionId = existingClaudeSessionId;
|
||
|
||
for await (const message of response) {
|
||
console.log('\n[SDK] ────────────────────────────');
|
||
console.log('[SDK] Message type:', message.type);
|
||
if (message.subtype) console.log('[SDK] Subtype:', message.subtype);
|
||
|
||
// 详细日志
|
||
if (message.type === 'assistant' && message.message?.content) {
|
||
const content = message.message.content;
|
||
if (Array.isArray(content)) {
|
||
content.forEach((block, i) => {
|
||
if (block.type === 'text') {
|
||
const text = block.text.substring(0, 200);
|
||
console.log(`[SDK] Content[${i}] text:`, text + (block.text.length > 200 ? '...' : ''));
|
||
} else if (block.type === 'tool_use') {
|
||
console.log(`[SDK] Content[${i}] tool_use:`, block.name);
|
||
console.log('[SDK] Input:', JSON.stringify(block.input, null, 2).substring(0, 500));
|
||
}
|
||
});
|
||
} else if (typeof content === 'string') {
|
||
console.log('[SDK] Content:', content.substring(0, 200));
|
||
}
|
||
}
|
||
|
||
if (message.type === 'user' && message.message?.content) {
|
||
const content = message.message.content;
|
||
if (Array.isArray(content)) {
|
||
content.forEach((block, i) => {
|
||
if (block.type === 'tool_result') {
|
||
const result = typeof block.content === 'string'
|
||
? block.content.substring(0, 300)
|
||
: JSON.stringify(block.content).substring(0, 300);
|
||
console.log(`[SDK] ToolResult[${i}]:`, result + '...');
|
||
}
|
||
});
|
||
}
|
||
}
|
||
|
||
if (message.type === 'system' && message.subtype === 'init') {
|
||
currentClaudeSessionId = message.session_id;
|
||
console.log('[SDK] Session ID:', currentClaudeSessionId);
|
||
|
||
const clientSessionId = sessionId || currentClaudeSessionId;
|
||
sessions.set(clientSessionId, currentClaudeSessionId);
|
||
|
||
sendEvent('session', {
|
||
clientSessionId,
|
||
claudeSessionId: currentClaudeSessionId,
|
||
isNewSession: !existingClaudeSessionId
|
||
});
|
||
}
|
||
else if (message.type === 'stream_event') {
|
||
const event = message.event || {};
|
||
|
||
if (event.type === 'content_block_delta') {
|
||
const delta = event.delta;
|
||
let text = '';
|
||
if (typeof delta === 'string') {
|
||
text = delta;
|
||
} else if (delta?.text) {
|
||
text = delta.text;
|
||
}
|
||
if (text) {
|
||
sendEvent('delta', { text });
|
||
}
|
||
}
|
||
else if (event.type === 'content_block_start') {
|
||
const block = event.content_block;
|
||
if (block?.type === 'tool_use') {
|
||
sendEvent('tool_start', {
|
||
toolName: block.name,
|
||
toolId: block.id
|
||
});
|
||
}
|
||
}
|
||
}
|
||
else if (message.type === 'assistant') {
|
||
let content = '';
|
||
const toolCalls = [];
|
||
|
||
if (typeof message.message?.content === 'string') {
|
||
content = message.message.content;
|
||
} else if (Array.isArray(message.message?.content)) {
|
||
for (const block of message.message.content) {
|
||
if (block.type === 'text') {
|
||
content += block.text;
|
||
} else if (block.type === 'tool_use') {
|
||
toolCalls.push({
|
||
name: block.name,
|
||
id: block.id,
|
||
input: block.input
|
||
});
|
||
}
|
||
}
|
||
}
|
||
|
||
if (content) {
|
||
sendEvent('assistant', { content });
|
||
}
|
||
if (toolCalls.length > 0) {
|
||
sendEvent('tool_calls', { tools: toolCalls });
|
||
}
|
||
}
|
||
else if (message.type === 'result') {
|
||
sendEvent('result', {
|
||
subtype: message.subtype,
|
||
sessionId: currentClaudeSessionId
|
||
});
|
||
}
|
||
}
|
||
|
||
sendEvent('done', {
|
||
message: '执行完成',
|
||
sessionId: currentClaudeSessionId
|
||
});
|
||
|
||
} catch (error) {
|
||
console.error('[Server] 错误:', error);
|
||
sendEvent('error', { message: error.message });
|
||
} finally {
|
||
pendingQuestions.delete(requestId);
|
||
res.end();
|
||
}
|
||
});
|
||
|
||
/**
|
||
* 提交用户回答
|
||
* POST /api/answer
|
||
*/
|
||
app.post('/api/answer', (req, res) => {
|
||
const { requestId, answers } = req.body;
|
||
console.log('[Server] 收到用户回答:', { requestId, answers });
|
||
|
||
const resolver = pendingQuestions.get(requestId);
|
||
if (resolver) {
|
||
resolver(answers);
|
||
pendingQuestions.delete(requestId);
|
||
res.json({ success: true });
|
||
} else {
|
||
res.status(404).json({ error: '未找到待处理的问题' });
|
||
}
|
||
});
|
||
|
||
/**
|
||
* 获取文档内容
|
||
* GET /api/document?type=final|draft
|
||
*/
|
||
app.get('/api/document', (req, res) => {
|
||
const type = req.query.type || 'final';
|
||
|
||
// 优先查找 requirement_final.md，否则 requirement.md
|
||
const finalPath = path.join(SKILLS_PROJECT_DIR, 'requirement_final.md');
|
||
const draftPath = path.join(SKILLS_PROJECT_DIR, 'requirement.md');
|
||
|
||
let filePath;
|
||
let docType;
|
||
|
||
if (type === 'final' && fs.existsSync(finalPath)) {
|
||
filePath = finalPath;
|
||
docType = 'final';
|
||
} else if (fs.existsSync(draftPath)) {
|
||
filePath = draftPath;
|
||
docType = 'draft';
|
||
} else {
|
||
return res.status(404).json({ error: '文档不存在' });
|
||
}
|
||
|
||
try {
|
||
const content = fs.readFileSync(filePath, 'utf-8');
|
||
res.json({
|
||
type: docType,
|
||
filename: path.basename(filePath),
|
||
content
|
||
});
|
||
} catch (error) {
|
||
res.status(500).json({ error: '读取文档失败: ' + error.message });
|
||
}
|
||
});
|
||
|
||
/**
|
||
* 下载文档
|
||
* GET /api/download?type=final|draft
|
||
*/
|
||
app.get('/api/download', (req, res) => {
|
||
const type = req.query.type || 'final';
|
||
|
||
const finalPath = path.join(SKILLS_PROJECT_DIR, 'requirement_final.md');
|
||
const draftPath = path.join(SKILLS_PROJECT_DIR, 'requirement.md');
|
||
|
||
let filePath;
|
||
|
||
if (type === 'final' && fs.existsSync(finalPath)) {
|
||
filePath = finalPath;
|
||
} else if (fs.existsSync(draftPath)) {
|
||
filePath = draftPath;
|
||
} else {
|
||
return res.status(404).json({ error: '文档不存在' });
|
||
}
|
||
|
||
res.download(filePath);
|
||
});
|
||
|
||
/**
|
||
* 获取会话列表
|
||
*/
|
||
app.get('/api/sessions', (req, res) => {
|
||
const sessionList = [];
|
||
for (const [clientId, claudeId] of sessions.entries()) {
|
||
sessionList.push({ clientId, claudeId });
|
||
}
|
||
res.json({ sessions: sessionList });
|
||
});
|
||
|
||
app.listen(PORT, () => {
|
||
console.log(`
|
||
========================================
|
||
需求文档生成器 Web 服务已启动
|
||
地址: http://localhost:${PORT}
|
||
========================================
|
||
|
||
项目目录: ${SKILLS_PROJECT_DIR}
|
||
Skill: requirement-generator-v1
|
||
|
||
使用前请确保已安装依赖: npm install
|
||
`);
|
||
});
|