file-toolkit-mcp 1.0.0

package/src/lib/result.js

@@ -0,0 +1,86 @@
+/**
+ * 统一的结果输出格式。
+ *
+ * 每个工具都返回 Markdown 摘要，而不是裸 JSON。原因：
+ *   1. 模型读 Markdown 表格比读嵌套 JSON 更不容易看错行列
+ *   2. 模型可以直接把这段文本转述给用户，不需要自己重新组织
+ *   3. 附带的结构化数据放在代码块里，模型需要精确数值时可直接取用
+ */
+
+/** 人类可读的体积格式 */
+export function formatBytes(bytes) {
+  if (!Number.isFinite(bytes) || bytes < 0) return '未知';
+  if (bytes < 1024) return `${bytes} B`;
+  const units = ['KB', 'MB', 'GB', 'TB'];
+  let value = bytes / 1024;
+  let idx = 0;
+  while (value >= 1024 && idx < units.length - 1) {
+    value /= 1024;
+    idx += 1;
+  }
+  return `${value.toFixed(value >= 100 ? 0 : value >= 10 ? 1 : 2)} ${units[idx]}`;
+}
+
+/** 压缩率展示：负数表示变小了 */
+export function formatDelta(before, after) {
+  if (!Number.isFinite(before) || before <= 0 || !Number.isFinite(after)) return '—';
+  const ratio = (1 - after / before) * 100;
+  if (ratio >= 0) return `↓ ${ratio.toFixed(1)}%`;
+  return `↑ ${Math.abs(ratio).toFixed(1)}%`;
+}
+
+/**
+ * 生成标准结果对象。
+ *
+ * @param {object} params
+ * @param {string} params.title 一句话结论
+ * @param {string[]} [params.table] Markdown 表格的行（含表头与分隔行）
+ * @param {string[]} [params.notes] 附加说明
+ * @param {object} [params.data] 结构化数据
+ * @param {string} [params.outputDir] 输出目录
+ */
+export function buildResult({ title, table = [], notes = [], data = null, outputDir = null }) {
+  const lines = [title];
+
+  if (table.length > 0) {
+    lines.push('', ...table);
+  }
+
+  if (outputDir) {
+    lines.push('', `输出目录：\`${outputDir}\``);
+  }
+
+  for (const note of notes) {
+    if (note) lines.push('', note);
+  }
+
+  if (data) {
+    lines.push('', '```json', JSON.stringify(data, null, 2), '```');
+  }
+
+  return lines.join('\n');
+}
+
+/** 组装 Markdown 表格 */
+export function mdTable(headers, rows) {
+  const escapeCell = (v) => String(v ?? '').replace(/\|/g, '\\|').replace(/\n/g, ' ');
+  const head = `| ${headers.map(escapeCell).join(' | ')} |`;
+  const sep = `| ${headers.map(() => '---').join(' | ')} |`;
+  const body = rows.map((r) => `| ${r.map(escapeCell).join(' | ')} |`);
+  return [head, sep, ...body];
+}
+
+/**
+ * 统计成功/失败，生成「N 个成功，M 个失败」这样的结论句。
+ */
+export function summarize(action, results) {
+  const ok = results.filter((r) => r.ok).length;
+  const failed = results.length - ok;
+  if (failed === 0) {
+    return `${action}完成：${ok} 个文件全部成功。`;
+  }
+  if (ok === 0) {
+    return `${action}失败：${failed} 个文件都没有处理成功。`;
+  }
+  return `${action}完成：${ok} 个成功，${failed} 个失败。`;
+}

package/src/lib/text.js

@@ -0,0 +1,80 @@
+/**
+ * 文本格式转换：Markdown <-> HTML。
+ * 走 marked（MIT）与 turndown（MIT），都是纯 JS，无系统依赖。
+ */
+
+import { marked } from 'marked';
+import TurndownService from 'turndown';
+
+const PAGE_TEMPLATE = (title, body) => `<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+<meta charset="utf-8">
+<meta name="viewport" content="width=device-width, initial-scale=1">
+<title>${escapeHtml(title)}</title>
+<style>
+  body { max-width: 46rem; margin: 2rem auto; padding: 0 1rem;
+         font-family: -apple-system, "PingFang SC", "Microsoft YaHei", sans-serif;
+         line-height: 1.75; color: #24292f; }
+  pre { background: #f6f8fa; padding: 1rem; overflow: auto; border-radius: 6px; }
+  code { background: #f6f8fa; padding: .15em .35em; border-radius: 4px; font-size: .92em; }
+  pre code { background: none; padding: 0; }
+  table { border-collapse: collapse; width: 100%; }
+  th, td { border: 1px solid #d0d7de; padding: .5rem .75rem; text-align: left; }
+  th { background: #f6f8fa; }
+  img { max-width: 100%; }
+  blockquote { margin: 0; padding: 0 1rem; color: #57606a; border-left: .25rem solid #d0d7de; }
+</style>
+</head>
+<body>
+${body}
+</body>
+</html>
+`;
+
+function escapeHtml(text) {
+  return String(text ?? '')
+    .replace(/&/g, '&amp;')
+    .replace(/</g, '&lt;')
+    .replace(/>/g, '&gt;')
+    .replace(/"/g, '&quot;');
+}
+
+/**
+ * Markdown -> HTML
+ * @param {string} markdown
+ * @param {{ fullPage?: boolean, title?: string }} [options]
+ */
+export function markdownToHtml(markdown, options = {}) {
+  const { fullPage = true, title = 'Document' } = options;
+  const source = String(markdown ?? '');
+  const body = marked.parse(source, { gfm: true, breaks: false });
+  return fullPage ? PAGE_TEMPLATE(title, body) : body;
+}
+
+/**
+ * HTML -> Markdown
+ */
+export function htmlToMarkdown(html) {
+  const service = new TurndownService({
+    headingStyle: 'atx',
+    codeBlockStyle: 'fenced',
+    bulletListMarker: '-',
+    emDelimiter: '*',
+  });
+
+  // 保留表格结构（turndown 默认把 table 丢掉）
+  service.keep(['table', 'thead', 'tbody', 'tr', 'th', 'td']);
+
+  return service.turndown(String(html ?? ''));
+}
+
+/**
+ * 从 HTML 里猜一个标题，用作输出的 PDF/HTML 标题
+ */
+export function guessTitle(text) {
+  const match = String(text ?? '').match(/^\s*#\s+(.+)$/m);
+  if (match) return match[1].trim().slice(0, 120);
+  const firstLine = String(text ?? '').split('\n').find((l) => l.trim() !== '');
+  return (firstLine ?? 'Document').trim().slice(0, 120) || 'Document';
+}

package/src/server.js

@@ -0,0 +1,60 @@
+/**
+ * File Toolkit MCP Server 入口。
+ *
+ * 传输方式：stdio（本地进程，由 WorkBuddy 拉起）。
+ * 注意：stdio 模式下 stdout 被 MCP 协议独占，
+ * 任何日志都必须走 stderr，否则会破坏协议。
+ */
+
+import { readFile } from 'node:fs/promises';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+
+import { registerImageTools } from './tools/image-tools.js';
+import { registerPdfTools } from './tools/pdf-tools.js';
+import { registerDataTools } from './tools/data-tools.js';
+import { registerFileTools } from './tools/file-tools.js';
+
+const pkg = JSON.parse(
+  await readFile(new URL('../package.json', import.meta.url), 'utf8'),
+);
+
+const INSTRUCTIONS = `File Toolkit 是一个完全在本机运行的文件处理工具箱。
+
+- 所有处理都在用户自己的电脑上完成，不会把文件上传到任何服务器。
+- 默认不覆盖原文件：结果统一写到「第一个输入文件所在目录」下的 file-toolkit-output 文件夹；
+  如果该目录里已有同名文件，会自动加序号，绝不覆盖。
+- 工具既可以接收单个文件，也可以接收整个文件夹（会自动收集里面的文件并批量处理）。
+- 批量处理时单个文件失败不会中断整批，结果里会分别列出成功和失败。
+- batch_rename 默认只返回预览，必须先让用户确认，再用 confirm=true 真正执行；
+  执行后可用 rename_undo 回滚。
+
+支持的图片格式：jpg / png / webp。支持处理 PDF、CSV、TSV、JSON、Markdown、HTML。`;
+
+async function main() {
+  const server = new McpServer(
+    { name: pkg.name, version: pkg.version },
+    { instructions: INSTRUCTIONS },
+  );
+
+  registerImageTools(server);
+  registerPdfTools(server);
+  registerDataTools(server);
+  registerFileTools(server);
+
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+
+  // 这些错误不应该让已经建立的服务连接直接消失
+  process.on('uncaughtException', (error) => {
+    console.error('[file-toolkit] 未捕获异常：', error?.stack ?? error);
+  });
+  process.on('unhandledRejection', (reason) => {
+    console.error('[file-toolkit] 未处理的 Promise 拒绝：', reason);
+  });
+}
+
+main().catch((error) => {
+  console.error('[file-toolkit] 启动失败：', error?.stack ?? error);
+  process.exit(1);
+});