koishi-plugin-latex-render 1.0.1

package/lib/index.d.ts

@@ -0,0 +1,15 @@
+import { Context, Schema } from 'koishi';
+export declare const name = "latex-render";
+export declare const inject: string[];
+export interface Config {
+    /** 图片宽度 */
+    width?: number;
+    /** 背景色 */
+    backgroundColor?: string;
+    /** 文字颜色 */
+    textColor?: string;
+    /** 调试模式 */
+    debug?: boolean;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -0,0 +1,306 @@
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name2 in all)
+    __defProp(target, name2, { get: all[name2], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  Config: () => Config,
+  apply: () => apply,
+  inject: () => inject,
+  name: () => name
+});
+module.exports = __toCommonJS(src_exports);
+var import_koishi = require("koishi");
+
+// src/renderer.ts
+var import_katex = __toESM(require("katex"));
+function containsLatex(content) {
+  return /\$\$/.test(content) || /\\begin\{/.test(content) || /\\\[[\s\S]*?\\\]/.test(content) || /\\\([\s\S]*?\\\)/.test(content) || /\$[^\$\n]/.test(content);
+}
+__name(containsLatex, "containsLatex");
+function decodeHtmlEntities(text) {
+  return text.replace(/&lt;/g, "<").replace(/&gt;/g, ">").replace(/&amp;/g, "&").replace(/&quot;/g, '"').replace(/&#39;/g, "'").replace(/&nbsp;/g, " ");
+}
+__name(decodeHtmlEntities, "decodeHtmlEntities");
+function parseMessage(content) {
+  content = decodeHtmlEntities(content);
+  const result = [];
+  const regex = /(\$\$[\s\S]+?\$\$|\\\[[\s\S]*?\\\]|\\\([\s\S]*?\\\)|\$(?:[^\$\n]|\$(?!\$)|[\s\S])*?\$|\\begin\{[a-zA-Z*]+\}[\s\S]*?\\end\{[a-zA-Z*]+\})/g;
+  let lastIndex = 0;
+  let match;
+  while ((match = regex.exec(content)) !== null) {
+    if (match.index > lastIndex) {
+      const text = content.slice(lastIndex, match.index).trim();
+      if (text) {
+        result.push({ type: "text", content: text });
+      }
+    }
+    const latex = match[1];
+    let isDisplay = false;
+    let formula = latex;
+    if (latex.startsWith("$$") && latex.endsWith("$$")) {
+      isDisplay = true;
+      formula = latex.slice(2, -2).trim();
+    } else if (latex.startsWith("\\[") && latex.endsWith("\\]")) {
+      isDisplay = true;
+      formula = latex.slice(2, -2).trim();
+    } else if (latex.startsWith("\\(") && latex.endsWith("\\)")) {
+      formula = latex.slice(2, -2).trim();
+    } else if (latex.startsWith("$") && latex.endsWith("$")) {
+      formula = latex.slice(1, -1).trim();
+    } else if (latex.startsWith("\\begin")) {
+      isDisplay = true;
+    }
+    result.push({
+      type: "latex",
+      content: formula,
+      display: isDisplay
+    });
+    lastIndex = match.index + match[0].length;
+  }
+  if (lastIndex < content.length) {
+    const text = content.slice(lastIndex).trim();
+    if (text) {
+      result.push({ type: "text", content: text });
+    }
+  }
+  return result;
+}
+__name(parseMessage, "parseMessage");
+function generateHtml(parsed, config) {
+  const textColor = config.textColor || "#333333";
+  const bgColor = config.backgroundColor || "#ffffff";
+  const width = config.width || 800;
+  let html = `<!DOCTYPE html>
+<html>
+<head>
+  <meta charset="UTF-8">
+  <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/katex@0.16.9/dist/katex.min.css">
+  <style>
+    * { margin: 0; padding: 0; box-sizing: border-box; }
+    body {
+      font-family: "Microsoft YaHei", "PingFang SC", sans-serif;
+      font-size: 16px;
+      line-height: 1.8;
+      color: ${textColor};
+      background-color: ${bgColor};
+      padding: 24px;
+      width: ${width}px;
+      min-height: 100px;
+    }
+    .content {
+      word-wrap: break-word;
+      white-space: pre-wrap;
+    }
+    .latex-display {
+      margin: 12px 0;
+      text-align: center;
+      overflow-x: auto;
+    }
+    .latex-inline {
+      margin: 0 2px;
+    }
+    .text-line {
+      margin: 4px 0;
+    }
+  </style>
+</head>
+<body>
+  <div class="content">`;
+  for (const item of parsed) {
+    if (item.type === "text") {
+      const escaped = item.content.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;");
+      html += `<span class="text-line">${escaped}</span>`;
+    } else {
+      try {
+        const htmlContent = import_katex.default.renderToString(item.content, {
+          throwOnError: false,
+          displayMode: item.display || false,
+          trust: true,
+          strict: false
+        });
+        const className = item.display ? "latex-display" : "latex-inline";
+        html += `<span class="${className}">${htmlContent}</span>`;
+      } catch (e) {
+        html += `<span class="latex-display"><code>${item.content}</code></span>`;
+      }
+    }
+  }
+  html += `</div></body></html>`;
+  return html;
+}
+__name(generateHtml, "generateHtml");
+function estimateHeight(parsed) {
+  let charCount = 0;
+  for (const item of parsed) {
+    if (item.type === "text") {
+      charCount += item.content.length;
+    } else {
+      charCount += item.content.length * 1.5;
+    }
+  }
+  const lines = Math.ceil(charCount / 35);
+  return Math.max(100, lines * 24 + 48);
+}
+__name(estimateHeight, "estimateHeight");
+async function renderLatex(ctx, content, config) {
+  console.log("[latex-render] 开始渲染...");
+  let parsed;
+  try {
+    parsed = parseMessage(content);
+    console.log("[latex-render] 解析完成，共", parsed.length, "个片段");
+  } catch (error) {
+    console.error("[latex-render] 解析消息失败:", error);
+    throw new Error(`消息解析失败: ${error}`);
+  }
+  if (parsed.length === 0) {
+    throw new Error("No content to render");
+  }
+  const width = config.width || 800;
+  const height = estimateHeight(parsed);
+  let html;
+  try {
+    html = generateHtml(parsed, config);
+    console.log("[latex-render] HTML 生成完成，长度:", html.length);
+  } catch (error) {
+    console.error("[latex-render] HTML 生成失败:", error);
+    throw new Error(`HTML 生成失败: ${error}`);
+  }
+  let page = null;
+  try {
+    page = await ctx.puppeteer.page();
+    console.log("[latex-render] Puppeteer 页面获取成功");
+    await page.setContent(html, {
+      waitUntil: "networkidle2",
+      timeout: 3e4
+      // 30秒超时等待 CSS
+    });
+    console.log("[latex-render] HTML 内容设置完成");
+    await new Promise((resolve) => setTimeout(resolve, 500));
+    const actualHeight = await page.evaluate(() => {
+      const body = document.body;
+      return body ? body.scrollHeight : 0;
+    }).catch((e) => {
+      console.warn("[latex-render] 获取高度失败，使用预估高度:", e);
+      return height;
+    });
+    const finalHeight = Math.max(actualHeight + 20, height);
+    console.log("[latex-render] 实际高度:", finalHeight);
+    const buffer = await page.screenshot({
+      clip: {
+        x: 0,
+        y: 0,
+        width,
+        height: finalHeight
+      },
+      type: "png"
+    });
+    console.log("[latex-render] 截图完成，buffer 长度:", buffer.length);
+    await page.close().catch(() => {
+    });
+    page = null;
+    const base64 = Buffer.from(buffer).toString("base64");
+    const dataUrl = `data:image/png;base64,${base64}`;
+    const url = await ctx.assets.upload(dataUrl, "latex-render.png");
+    console.log("[latex-render] 图片上传完成，URL:", url);
+    return url;
+  } catch (error) {
+    console.error("[latex-render] Puppeteer 渲染失败:", error?.message || error);
+    console.error("[latex-render] 错误详情:", error?.stack || "无堆栈信息");
+    if (page) {
+      try {
+        await page.close();
+      } catch (e) {
+      }
+    }
+    throw new Error(`LaTeX 渲染失败: ${error?.message || error}`);
+  }
+}
+__name(renderLatex, "renderLatex");
+
+// src/index.ts
+var name = "latex-render";
+var inject = ["assets", "puppeteer"];
+var Config = import_koishi.Schema.object({
+  width: import_koishi.Schema.number().default(800).description("图片宽度"),
+  backgroundColor: import_koishi.Schema.string().default("#ffffff").description("背景色"),
+  textColor: import_koishi.Schema.string().default("#333333").description("文字颜色"),
+  debug: import_koishi.Schema.boolean().default(false).description("调试模式")
+});
+function apply(ctx, config) {
+  const debug = config.debug || false;
+  const handler = /* @__PURE__ */ __name(async (conversationId, sourceMessage, displayResponse, promptVariables, chatInterface, session) => {
+    try {
+      let content;
+      if (Array.isArray(displayResponse)) {
+        content = displayResponse[0]?.content;
+      } else if (displayResponse?.content) {
+        content = displayResponse.content;
+      } else if (typeof displayResponse === "string") {
+        content = displayResponse;
+      }
+      if (!content) return;
+      const contentStr = typeof content === "string" ? content : JSON.stringify(content);
+      if (containsLatex(contentStr)) {
+        if (debug) {
+          console.log("[latex-render] 检测到 LaTeX 公式，开始渲染...");
+        }
+        try {
+          const imageUrl = await renderLatex(ctx, contentStr, config);
+          if (session) {
+            await session.send(import_koishi.h.image(imageUrl));
+          }
+          const imageContent = import_koishi.h.image(imageUrl).toString();
+          if (Array.isArray(displayResponse)) {
+            displayResponse[0].content = imageContent;
+          } else {
+            displayResponse.content = imageContent;
+          }
+        } catch (error) {
+          if (debug) {
+            console.error("[latex-render] 渲染失败:", error);
+          }
+        }
+      }
+    } catch (error) {
+      if (debug) {
+        console.error("[latex-render] 处理失败:", error);
+      }
+    }
+  }, "handler");
+  ctx.on("chatluna/after-chat", handler);
+}
+__name(apply, "apply");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  inject,
+  name
+});

package/lib/renderer.d.ts

@@ -0,0 +1,15 @@
+import { Context } from 'koishi';
+interface Config {
+    width?: number;
+    backgroundColor?: string;
+    textColor?: string;
+}
+/**
+ * 检测是否为 LaTeX 公式（增强版）
+ */
+export declare function containsLatex(content: string): boolean;
+/**
+ * 主渲染函数 - 使用 Puppeteer + KaTeX
+ */
+export declare function renderLatex(ctx: Context, content: string, config: Config): Promise<string>;
+export {};

package/package.json

@@ -0,0 +1,45 @@
+{
+  "name": "koishi-plugin-latex-render",
+  "description": "ChatLuna 专用 - 将 AI 返回的 LaTeX 公式渲染为图片",
+  "version": "1.0.1",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "homepage": "https://github.com/gt4404gb/koishi-plugin-latex-render",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/gt4404gb/koishi-plugin-latex-render.git"
+  },
+  "keywords": [
+    "koishi",
+    "plugin",
+    "latex",
+    "chatluna",
+    "ai",
+    "katex"
+  ],
+  "peerDependencies": {
+    "koishi": "^4.18.7"
+  },
+  "koishi": {
+    "description": {
+      "zh": "ChatLuna 专用 - 将 AI 返回的 LaTeX 公式渲染为图片",
+      "en": "ChatLuna exclusive - Render LaTeX formulas returned by AI as images"
+    },
+    "service": {
+      "optional": [
+        "assets"
+      ],
+      "required": [
+        "puppeteer"
+      ]
+    }
+  },
+  "dependencies": {
+    "katex": "^0.16.0"
+  }
+}

package/readme.md

@@ -0,0 +1,93 @@
+# koishi-plugin-latex-render
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-latex-render?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-latex-render)
+
+## 简介
+
+**本插件专为 [ChatLuna](https://github.com/ChatLuna/ChatLuna) 设计**，用于将 AI 返回的 LaTeX 公式渲染为图片并发送。
+
+当 ChatLuna 的 AI 返回包含 LaTeX 公式的消息时，本插件会自动：
+1. 拦截消息
+2. 使用 KaTeX 解析并渲染公式
+3. 使用 Puppeteer 截图生成图片
+4. 将图片发送给用户
+
+## 特性
+
+- **整条消息渲染为一张图片** - 不是每个公式单独一张
+- **支持多种 LaTeX 格式** - `$$...$$`, `\[...\]`, `\(...\)`, `$...$`, `\begin{}...\end{}`
+- **中文支持** - 使用微软雅黑字体
+- **可自定义样式** - 支持配置背景色、文字颜色、图片宽度
+
+## 依赖
+
+本插件依赖以下 Koishi 插件：
+- `koishi-plugin-puppeteer` - 用于浏览器截图
+- `koishi-plugin-assets-local` - 用于图片上传托管（可选）
+
+## 安装
+
+```bash
+# 使用 yarn
+yarn add koishi-plugin-latex-render
+
+# 或使用 npm
+npm install koishi-plugin-latex-render
+```
+
+## 配置
+
+在 `koishi.yml` 中配置插件：
+
+```yaml
+latex-render:
+  width: 800          # 图片宽度，默认 800
+  backgroundColor: "#ffffff"  # 背景色，默认白色
+  textColor: "#333333"        # 文字颜色，默认深灰色
+
+# 必须启用 puppeteer 插件
+puppeteer: {}
+
+# 建议配置 assets-local 的 selfUrl
+assets-local:5fyoiw:
+  selfUrl: http://127.0.0.1:5140
+```
+
+## 工作原理
+
+1. 监听 `chatluna/after-chat` 事件
+2. 检测消息中是否包含 LaTeX 公式
+3. 使用 KaTeX 将公式渲染为 HTML
+4. 使用 Puppeteer 打开 HTML 并截图
+5. 通过 assets 服务上传图片
+6. 发送图片消息给用户
+
+## 支持的 LaTeX 格式
+
+| 格式 | 示例 | 类型 |
+|------|------|------|
+| 行内公式 | `$x^2$` | inline |
+| 行内公式 | `\(x^2\)` | inline |
+| 块级公式 | `$$x^2$$` | display |
+| 块级公式 | `\[x^2\]` | display |
+| 环境 | `\begin{cases}...\end{cases}` | display |
+
+## 注意事项
+
+- **本插件极度特化于 ChatLuna** - 仅监听 ChatLuna 的消息事件
+- 需要网络访问 `cdn.jsdelivr.net` 加载 KaTeX CSS
+- Puppeteer 插件必须在插件列表中启用
+
+## 开发
+
+```bash
+# 构建
+yarn build
+
+# 或在项目根目录
+npm run build latex-render
+```
+
+## License
+
+MIT