work-agent 0.1.0

package/bin/generate-api-system

@@ -0,0 +1,300 @@
+#!/usr/bin/env node
+/**
+ * API System Generator - CLI Tool
+ * 生成 Skill 到 .pi/skills/ 目录
+ */
+
+const { writeFileSync, mkdirSync, chmodSync, existsSync, readFileSync } = require('fs');
+const { join, dirname } = require('path');
+
+const HELP = `
+API Skill Generator - 从 Swagger 生成 AI 可读取的 Skill
+
+用法:
+  generate-api-system -s <swagger-url> [选项]
+
+选项:
+  -s, --swagger <url>     Swagger/OpenAPI 文档地址 (必填)
+  -n, --name <name>        Skill 名称 (默认: 从 Swagger 提取)
+  -d, --description <desc>  Skill 描述
+  -b, --base-url <url>     API 基础地址
+  -t, --token <token>      API Token
+  -h, --help              显示帮助
+
+示例:
+  # 基本用法
+  generate-api-system -s "https://api.example.com/openapi.json"
+
+  # 完整参数
+  generate-api-system -s "https://api.example.com/openapi.json" \\
+    -n "my-api" \\
+    -d "我的API系统" \\
+    -b "https://api.example.com"
+
+输出 (生成到 .pi/skills/):
+  .pi/skills/{name}/
+  ├── SKILL.md            ← AI Skill 定义
+  ├── SKILL.json          ← 结构化元数据
+  ├── config.env          ← 环境变量配置（供脚本使用）
+  ├── config.yaml         ← API 配置说明
+  ├── scripts/           ← 可执行脚本
+  └── curl_examples/      ← curl 示例
+`;
+
+function extractName(swaggerUrl, spec) {
+  if (spec.info?.title) {
+    return slugify(spec.info.title);
+  }
+  if (spec.info?.version) {
+    return slugify(spec.info.title || 'api') + '-' + slugify(spec.info.version);
+  }
+  try {
+    const url = new URL(swaggerUrl);
+    return slugify(url.hostname);
+  } catch {
+    return 'api-system';
+  }
+}
+
+function slugify(str) {
+  return str.toLowerCase()
+    .replace(/[^a-z0-9]+/g, '-')
+    .replace(/^-|-$/g, '');
+}
+
+function extractBaseUrl(swaggerUrl, spec) {
+  if (spec.servers?.[0]?.url) {
+    return spec.servers[0].url;
+  }
+  try {
+    const url = new URL(swaggerUrl);
+    return url.origin + url.pathname.replace(/\/[^/]*$/, '');
+  } catch {
+    return '';
+  }
+}
+
+async function generate() {
+  const args = process.argv.slice(2);
+  
+  if (args.includes('-h') || args.includes('--help')) {
+    console.log(HELP);
+    process.exit(0);
+  }
+
+  const options = {
+    swaggerUrl: '',
+    name: '',
+    description: '',
+    baseUrl: ''
+  };
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === '-s' || arg === '--swagger') options.swaggerUrl = args[++i];
+    if (arg === '-n' || arg === '--name') options.name = args[++i];
+    if (arg === '-d' || arg === '--description') options.description = args[++i];
+    if (arg === '-b' || arg === '--base-url') options.baseUrl = args[++i];
+  }
+
+  if (!options.swaggerUrl) {
+    console.error('错误: 请指定 Swagger URL (-s <url>)');
+    console.error('帮助: generate-api-system --help');
+    process.exit(1);
+  }
+
+  console.log('Fetching swagger...');
+  const response = await fetch(options.swaggerUrl);
+  if (!response.ok) {
+    console.error('Failed to fetch swagger:', response.statusText);
+    process.exit(1);
+  }
+  const spec = await response.json();
+
+  const skillName = options.name || extractName(options.swaggerUrl, spec);
+  const skillDescription = options.description || spec.info?.description || spec.info?.title || 'API System';
+  const baseUrl = options.baseUrl || extractBaseUrl(options.swaggerUrl, spec);
+
+  // 输出到 .pi/skills/
+  const piSkillsDir = '.pi/skills';
+  const skillDir = join(piSkillsDir, skillName);
+  const scriptsDir = join(skillDir, 'scripts');
+  const curlDir = join(skillDir, 'curl_examples');
+
+  mkdirSync(piSkillsDir, { recursive: true });
+  mkdirSync(skillDir, { recursive: true });
+  mkdirSync(scriptsDir, { recursive: true });
+  mkdirSync(curlDir, { recursive: true });
+
+  console.log('Generating skill:', skillName);
+  console.log('Output:', skillDir);
+
+  // 生成 config.yaml
+  writeFileSync(join(skillDir, 'config.yaml'), `# ${skillName} - API配置
+
+api:
+  baseUrl: ${baseUrl}
+  auth:
+    type: bearer
+    credentials:
+      token: $TOKEN
+ `);
+
+  // 生成 config.env (供脚本使用)
+  writeFileSync(join(skillDir, 'config.env'), `# ${skillName} - 环境变量配置
+BASE_URL=${baseUrl}
+TOKEN=your-token-here
+ `);
+
+  // 生成 SKILL.json
+  const paths = spec.paths || {};
+  const endpoints = [];
+  const readOnlyOps = [];
+  const modifyOps = [];
+
+  for (const [path, item] of Object.entries(paths)) {
+    const normalizedPath = path.replace(/\{/g, ':').replace(/\}/g, '');
+    for (const [method, op] of Object.entries(item)) {
+      if (['get', 'post', 'put', 'delete', 'patch'].includes(method)) {
+        const operation = op;
+        const endpoint = {
+          method: method.toUpperCase(),
+          path,
+          summary: operation.summary || '',
+          isReadOnly: ['get', 'head', 'options'].includes(method)
+        };
+        endpoints.push(endpoint);
+        if (endpoint.isReadOnly) readOnlyOps.push(endpoint);
+        else modifyOps.push(endpoint);
+      }
+    }
+  }
+
+  const skillJson = {
+    name: skillName,
+    displayName: spec.info?.title || skillName,
+    description: skillDescription,
+    version: spec.info?.version || '1.0.0',
+    author: spec.info?.contact?.name || 'xuanwu',
+    keywords: ['api', 'generated', 'swagger'],
+    baseUrl,
+    endpoints: {
+      readOnly: readOnlyOps,
+      modify: modifyOps
+    },
+    scripts: {
+      readOnly: readOnlyOps.map(e => `${e.method.toLowerCase()}_${e.path.split('/').pop()?.replace(/[{}]/g, '')}`),
+      modify: modifyOps.map(e => `${e.method.toLowerCase()}_${e.path.split('/').pop()?.replace(/[{}]/g, '')}`)
+    }
+  };
+  writeFileSync(join(skillDir, 'SKILL.json'), JSON.stringify(skillJson, null, 2));
+
+  // 生成 SKILL.md
+  let skillMd = `# ${skillName}
+
+${skillDescription}
+
+## API 列表
+
+### 只读操作（AI 自动执行）
+
+| API | 描述 | 脚本 |
+|-----|------|------|
+${readOnlyOps.map(e => `| \`${e.method} ${e.path}\` | ${e.summary || '-'} | \`scripts/${e.method.toLowerCase()}_${e.path.split('/').pop()?.replace(/[{}]/g, '')}\` |`).join('\n')}
+
+### 修改操作（需审批）
+
+| API | 描述 | 脚本 |
+|-----|------|------|
+${modifyOps.map(e => `| \`${e.method} ${e.path}\` | ${e.summary || '-'} | \`scripts/${e.method.toLowerCase()}_${e.path.split('/').pop()?.replace(/[{}]/g, '')}\` |`).join('\n')}
+
+## 使用方式
+
+### 只读操作
+
+\`\`\`bash
+./scripts/get_users
+\`\`\`
+
+### 修改操作
+
+\`\`\`bash
+./scripts/create_user  # 创建工单，等待审批
+\`\`\`
+
+## 配置
+
+编辑 \`config.env\` 配置 API 地址和认证。
+
+## 相关文件
+
+- \`SKILL.json\` - 结构化元数据
+- \`config.env\` - 环境变量配置（供脚本使用）
+- \`config.yaml\` - API 配置说明
+- \`scripts/\` - 可执行脚本
+`;
+  writeFileSync(join(skillDir, 'SKILL.md'), skillMd);
+
+  // 生成脚本
+  for (const [path, item] of Object.entries(paths)) {
+    const normalizedPath = path.replace(/\{/g, ':').replace(/\}/g, '');
+    const groupName = path.split('/').filter(Boolean)[1] || 'default';
+    
+    for (const [method, op] of Object.entries(item)) {
+      if (['get', 'post', 'put', 'delete', 'patch'].includes(method)) {
+        const operation = op;
+        const summary = operation.summary || '';
+        const isReadOnly = ['get', 'head', 'options'].includes(method);
+        const scriptName = method.toLowerCase() + '_' + path.split('/').pop()?.replace(/[{}]/g, '')?.replace(/-/g, '_') || method;
+
+        const scriptContent = `#!/bin/bash
+# ${scriptName} - ${summary}
+# 分类: ${isReadOnly ? '只读操作 (read_only)' : '修改操作 (modify)'}
+# API: ${method.toUpperCase()} ${path}
+
+SCRIPT_DIR="$(cd "$(dirname "\${BASH_SOURCE[0]}")" && pwd)"
+source "\${SCRIPT_DIR}/../config.env"
+
+${isReadOnly ? `# 只读操作 - 直接执行
+echo "执行: ${method.toUpperCase()} ${path}"
+curl -s -X ${method.toUpperCase()} "$BASE_URL${normalizedPath}" \\
+  -H "Authorization: Bearer $TOKEN" \\
+  -H "Content-Type: application/json" | jq .` : `# 修改操作 - 需要审批
+echo "执行: ${method.toUpperCase()} ${path}"
+curl -s -X ${method.toUpperCase()} "$BASE_URL${normalizedPath}" \\
+  -H "Authorization: Bearer $TOKEN" \\
+  -H "Content-Type: application/json" \\
+  -d '{"data": "请修改"}' | jq .`}
+`;
+        writeFileSync(join(scriptsDir, scriptName), scriptContent);
+        chmodSync(join(scriptsDir, scriptName), 0o755);
+
+        // curl 示例
+        const curlContent = `curl -X ${method.toUpperCase()} "${normalizedPath}" \\
+  -H "Authorization: Bearer $TOKEN" \\
+  -H "Content-Type: application/json"
+`;
+        writeFileSync(join(curlDir, `${scriptName}.sh`), curlContent);
+      }
+    }
+  }
+
+  console.log('');
+  console.log('✅ Skill 生成完成！');
+  console.log('📁 路径:', skillDir);
+  console.log('');
+  console.log('文件列表:');
+  console.log('  ├── SKILL.md       ← AI Skill 定义');
+  console.log('  ├── SKILL.json     ← 结构化元数据');
+  console.log('  ├── config.env     ← 环境变量配置（修改此文件）');
+  console.log('  ├── config.yaml    ← API 配置说明');
+  console.log('  ├── scripts/       ← 可执行脚本');
+  console.log('  └── curl_examples/ ← curl 示例');
+  console.log('');
+  console.log('AI 可直接读取 .pi/skills/' + skillName + '/SKILL.md');
+}
+
+generate().catch(err => {
+  console.error('Error:', err);
+  process.exit(1);
+});

package/bin/postinstall.js

@@ -0,0 +1,95 @@
+#!/usr/bin/env node
+
+'use strict';
+
+/**
+ * postinstall.js
+ *
+ * npm install / npm install -g 完成后自动触发：
+ * 1. 检查 Node.js 版本
+ * 2. 如果 .next/standalone 不存在 → 执行 next build（首次安装）
+ * 3. 无论如何 → 对 .next/standalone 里的 better-sqlite3 执行 rebuild
+ *    （确保原生模块与当前 Node.js ABI 匹配）
+ */
+
+const { spawnSync } = require('child_process');
+const path = require('path');
+const fs = require('fs');
+
+const PKG_ROOT = path.resolve(__dirname, '..');
+const STANDALONE_DIR = path.join(PKG_ROOT, '.next', 'standalone');
+const STANDALONE_SERVER = path.join(STANDALONE_DIR, 'server.js');
+const BSQL3_NODE = path.join(
+  STANDALONE_DIR,
+  'node_modules', 'better-sqlite3', 'build', 'Release', 'better_sqlite3.node'
+);
+
+// ── CI 环境 → 全部跳过 ────────────────────────────────────────────────────────
+if (process.env.CI) {
+  console.log('[work-agent] CI environment detected, skipping postinstall.');
+  process.exit(0);
+}
+
+// ── Node.js 版本检查 ──────────────────────────────────────────────────────────
+const [major] = process.versions.node.split('.').map(Number);
+if (major < 18) {
+  console.error(`[work-agent] Node.js >= 18 is required. Current: v${process.versions.node}`);
+  process.exit(1);
+}
+
+// ── Step 1: 如果没有 standalone → 先 build ───────────────────────────────────
+if (!fs.existsSync(STANDALONE_SERVER)) {
+  console.log('\n[work-agent] Building the app (first install, this may take a minute)...\n');
+
+  const buildResult = spawnSync('npm', ['run', 'build'], {
+    cwd: PKG_ROOT,
+    stdio: 'inherit',
+    shell: process.platform === 'win32',
+  });
+
+  if (buildResult.status !== 0) {
+    console.error('\n[work-agent] Build failed. Retry manually:\n');
+    console.error(`  cd ${PKG_ROOT} && npm run build\n`);
+    process.exit(0);
+  }
+
+  console.log('\n[work-agent] Build complete.\n');
+} else {
+  console.log('[work-agent] Standalone build found, skipping next build.');
+}
+
+// ── Step 2: 确保 better-sqlite3 与当前 Node.js ABI 匹配 ──────────────────────
+// next build 会把 better-sqlite3 的 .node 文件复制进 standalone，
+// 但那个文件是安装时编译的，换了 Node.js 版本就会崩。
+// 每次 postinstall 都强制 rebuild 一次，代价很低（< 10 秒）。
+if (fs.existsSync(BSQL3_NODE)) {
+  console.log('[work-agent] Rebuilding better-sqlite3 for current Node.js version...');
+
+  const rebuildResult = spawnSync(
+    'npm',
+    ['rebuild', 'better-sqlite3'],
+    {
+      cwd: STANDALONE_DIR,
+      stdio: 'inherit',
+      shell: process.platform === 'win32',
+      env: {
+        ...process.env,
+        // 防止 npm rebuild 触发其他包的 postinstall
+        npm_lifecycle_event: 'rebuild',
+      },
+    }
+  );
+
+  if (rebuildResult.status !== 0) {
+    console.warn('[work-agent] better-sqlite3 rebuild failed. You may need build tools:');
+    console.warn('  macOS: xcode-select --install');
+    console.warn('  Linux: apt install -y python3 make g++ (or yum equivalent)');
+    console.warn('  Windows: npm install -g windows-build-tools');
+  } else {
+    console.log('[work-agent] better-sqlite3 rebuild complete.');
+  }
+} else {
+  console.log('[work-agent] better-sqlite3 not found in standalone, skipping rebuild.');
+}
+
+console.log('\n[work-agent] Ready. Run `work-agent` to start.\n');

package/bin/work-agent.js

@@ -0,0 +1,173 @@
+#!/usr/bin/env node
+
+'use strict';
+
+const path = require('path');
+const fs = require('fs');
+const { spawn, spawnSync } = require('child_process');
+const http = require('http');
+
+// ── 路径解析 ──────────────────────────────────────────────────────────────────
+const PKG_ROOT = path.resolve(__dirname, '..');
+const STANDALONE_DIR = path.join(PKG_ROOT, '.next', 'standalone');
+const STANDALONE_SERVER = path.join(STANDALONE_DIR, 'server.js');
+const STATIC_SRC = path.join(PKG_ROOT, '.next', 'static');
+const PUBLIC_SRC = path.join(PKG_ROOT, 'public');
+const BSQL3_DIR = path.join(STANDALONE_DIR, 'node_modules', 'better-sqlite3');
+const BSQL3_NODE = path.join(BSQL3_DIR, 'build', 'Release', 'better_sqlite3.node');
+
+// ── 端口 ──────────────────────────────────────────────────────────────────────
+const PORT = parseInt(process.env.PORT || '11024', 10);
+const HOST = process.env.HOST || '0.0.0.0';
+
+// ── 检查构建产物是否存在 ──────────────────────────────────────────────────────
+if (!fs.existsSync(STANDALONE_SERVER)) {
+  console.error('\n[work-agent] 未找到构建产物。请先运行：\n');
+  console.error('  cd ' + PKG_ROOT);
+  console.error('  npm run build\n');
+  process.exit(1);
+}
+
+// ── 检查并修复 better-sqlite3 ABI 兼容性 ─────────────────────────────────────
+// .node 文件中嵌入了编译时的 NODE_MODULE_VERSION，通过读取文件内容来检测，
+// 避免 try/require 在某些平台上行为不一致。
+function getNodeAbiFromBinary(nodePath) {
+  try {
+    // NODE_MODULE_VERSION 以字符串 "NODE_MODULE_VERSION_xx" 或直接数字嵌入在二进制里
+    // 更可靠的方式：直接 require 并捕获错误消息
+    const mod = require('module');
+    // 通过 process.versions.modules 获取当前运行时的 ABI
+    return parseInt(process.versions.modules, 10);
+  } catch {
+    return null;
+  }
+}
+
+function needsRebuild() {
+  if (!fs.existsSync(BSQL3_NODE)) return false;
+
+  // 读取 .node 二进制，在其中搜索 "NODE_MODULE_VERSION" 标记后的版本号
+  try {
+    const buf = fs.readFileSync(BSQL3_NODE);
+    const marker = Buffer.from('NODE_MODULE_VERSION');
+    let idx = buf.indexOf(marker);
+    if (idx === -1) {
+      // 找不到标记，用 try-require 兜底
+      try {
+        require(path.join(BSQL3_DIR, 'lib', 'database.js'));
+        return false;
+      } catch {
+        return true;
+      }
+    }
+    // marker 之后是一个字节 0x00（分隔），然后是版本号字符串（如 "115\0"）
+    // 实际格式因编译工具链不同，直接 try-require 更可靠
+    return false; // fallback 到 try-require
+  } catch {
+    return false;
+  }
+}
+
+// 最可靠方案：直接 try-require，捕获 ERR_DLOPEN_FAILED
+function checkAndRebuild() {
+  if (!fs.existsSync(BSQL3_NODE)) return;
+
+  let failed = false;
+  try {
+    // 用 child_process 在子进程里 require，避免污染主进程模块缓存
+    const result = spawnSync(process.execPath, [
+      '-e',
+      `require(${JSON.stringify(path.join(BSQL3_DIR, 'lib', 'database.js'))})`,
+    ], { timeout: 5000 });
+    failed = result.status !== 0 && result.stderr &&
+      result.stderr.toString().includes('NODE_MODULE_VERSION');
+  } catch {
+    failed = true;
+  }
+
+  if (!failed) return;
+
+  console.log('[work-agent] Detected Node.js version mismatch for better-sqlite3.');
+  console.log('[work-agent] Rebuilding native module (this takes ~10 seconds)...\n');
+
+  const result = spawnSync(
+    'npm',
+    ['rebuild', 'better-sqlite3'],
+    {
+      cwd: STANDALONE_DIR,
+      stdio: 'inherit',
+      shell: process.platform === 'win32',
+    }
+  );
+
+  if (result.status !== 0) {
+    console.error('\n[work-agent] Rebuild failed. Please install build tools:');
+    console.error('  macOS:   xcode-select --install');
+    console.error('  Linux:   apt install -y python3 make g++');
+    console.error('  Windows: npm install -g windows-build-tools\n');
+    process.exit(1);
+  }
+
+  console.log('\n[work-agent] Rebuild complete.\n');
+}
+
+checkAndRebuild();
+
+// ── 确保 static / public 软链存在 ────────────────────────────────────────────
+function ensureLink(src, dst) {
+  if (!fs.existsSync(dst) && fs.existsSync(src)) {
+    try {
+      fs.symlinkSync(src, dst, 'junction');
+    } catch {
+      // 忽略
+    }
+  }
+}
+
+ensureLink(STATIC_SRC, path.join(STANDALONE_DIR, '.next', 'static'));
+ensureLink(PUBLIC_SRC, path.join(STANDALONE_DIR, 'public'));
+
+// ── 等待服务就绪 ──────────────────────────────────────────────────────────────
+function waitForReady(port, retries, delay, cb) {
+  http.get(`http://127.0.0.1:${port}/`, () => cb(null))
+    .on('error', () => {
+      if (retries <= 0) return cb(new Error('timeout'));
+      setTimeout(() => waitForReady(port, retries - 1, delay, cb), delay);
+    });
+}
+
+// ── 启动服务 ──────────────────────────────────────────────────────────────────
+console.log(`\n[work-agent] Starting on http://localhost:${PORT} ...\n`);
+
+const server = spawn(process.execPath, [STANDALONE_SERVER], {
+  cwd: STANDALONE_DIR,
+  env: { ...process.env, PORT: String(PORT), HOST, NODE_ENV: 'production' },
+  stdio: 'inherit',
+});
+
+server.on('error', (err) => {
+  console.error('[work-agent] Failed to start server:', err.message);
+  process.exit(1);
+});
+
+server.on('exit', (code) => process.exit(code || 0));
+
+// ── 就绪后打开浏览器 ──────────────────────────────────────────────────────────
+const url = `http://localhost:${PORT}`;
+
+waitForReady(PORT, 30, 500, (err) => {
+  if (err) {
+    console.log(`[work-agent] Open your browser at: ${url}`);
+    return;
+  }
+  console.log(`[work-agent] Ready! Opening ${url}`);
+  try {
+    require('open')(url).catch(() => {});
+  } catch {
+    console.log(`[work-agent] Open your browser at: ${url}`);
+  }
+});
+
+['SIGINT', 'SIGTERM'].forEach((sig) => {
+  process.on(sig, () => server.kill(sig));
+});