koishi-plugin-cfmrmod 1.1.4 → 1.1.5

package/README.md

@@ -19,6 +19,15 @@ Star就是我维护的动力🤤
 - `cnmc.mod/.data/.pack/.tutorial/.author/.user <关键词>`
 - `cf.help` / `mr.help` / `cnmc.help`
 
+#### AI 自然语言查询（可选）
+开启 `nlu.enabled` 并配置 OpenAI 兼容接口后，只有 `@机器人` 的消息会进入自然语言理解；不 @ 机器人时仍只响应上面的显式命令。
+
+示例：
+- `@机器人 查询钠模组`：默认在 MCMod 查询 Mod
+- `@机器人 在 cf 查一下 jei`：在 CurseForge 查询 Mod
+- `@机器人 在 mr 查一下 iris 光影`：在 Modrinth 查询 Shader
+- `@机器人 在 cnmc 查一下 Create: EasyFilling`：在 MCMod 查询 Mod
+
 #### 更新通知（notify）
 - `notify.add <platform> <projectId>` 添加订阅
 - `notify.remove <platform> <projectId>` 删除订阅
@@ -42,6 +51,20 @@ Star就是我维护的动力🤤
 - `timeouts`: 搜索会话超时（毫秒）
 - `debug`: 调试日志开关
 
+#### AI 自然语言理解（nlu）
+- `nlu.enabled`: 是否启用 `@机器人` 自然语言查询
+- `nlu.endpoint`: OpenAI 兼容 Chat Completions 接口地址，默认 `https://api.openai.com/v1/chat/completions`
+- `nlu.apiKey`: API Key
+- `nlu.model`: 模型名称，默认 `gpt-4o-mini`
+- `nlu.timeout`: AI 请求超时（毫秒）
+- `nlu.temperature`: AI 温度参数，默认 `0`
+
+#### MCMod（mcmod）
+- `mcmod.cookie`: 手动填写 mcmod.cn Cookie
+- `mcmod.autoCookie`: 自动从 `cookie-manager` 获取 Cookie（存在该模块时生效）
+- `mcmod.cookieCheckInterval`: Cookie / `MCMOD_SEED` 检查间隔（毫秒）
+- 未配置 Cookie 时，插件会自动访问 MCMod 首页获取 `MCMOD_SEED`，用于通过站点的基础 Cookie 校验。
+
 #### 更新通知（notify）
 - `notify.enabled`: 是否开启更新通知
 - `notify.interval`: 全局轮询间隔（毫秒）

package/dist/index.js

@@ -38,6 +38,7 @@ exports.apply = apply;
 const koishi_1 = require("koishi");
 const cfmr = __importStar(require("./plugins/cfmr"));
 const mcmod = __importStar(require("./plugins/mcmod"));
+const nlu = __importStar(require("./nlu"));
 const notify = __importStar(require("./notify"));
 exports.name = 'minecraft-search';
 exports.inject = ['database'];
@@ -65,6 +66,7 @@ exports.Config = koishi_1.Schema.object({
     }).description('—— 更新通知 ——'),
     timeouts: koishi_1.Schema.number().default(60000).description('搜索会话超时时间(ms)'),
     debug: koishi_1.Schema.boolean().default(false).description('开启调试日志'),
+    nlu: nlu.Config,
     cfmr: cfmr.Config.description('CurseForge/Modrinth 搜索与图片卡片'),
     mcmod: mcmod.Config.description('MCMod.cn 搜索与图片卡片'),
 });
@@ -98,6 +100,8 @@ function apply(ctx, config) {
         cfmr.apply(ctx, { ...((config === null || config === void 0 ? void 0 : config.cfmr) || {}), ...shared });
     if (mcmod.apply)
         mcmod.apply(ctx, { ...((config === null || config === void 0 ? void 0 : config.mcmod) || {}), ...shared });
+    if (nlu.apply)
+        nlu.apply(ctx, (config === null || config === void 0 ? void 0 : config.nlu) || {}, shared);
     if (notify.apply && canvasAdapter)
         notify.apply(ctx, (config === null || config === void 0 ? void 0 : config.notify) || {}, { cfmr: (config === null || config === void 0 ? void 0 : config.cfmr) || {} });
     if (!canvasAdapter)

package/dist/mcmod/http.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.setMcmodCookie = setMcmodCookie;
+exports.configureMcmodCookie = configureMcmodCookie;
 exports.getMcmodCookie = getMcmodCookie;
 exports.loadManagedCookie = loadManagedCookie;
 exports.fetchWithTimeout = fetchWithTimeout;
@@ -20,7 +21,8 @@ catch (e) {
 }
 let globalCookie = '';
 let cookieLastCheck = 0;
-const COOKIE_CHECK_INTERVAL = 30 * 60 * 1000;
+let cookieCheckInterval = 30 * 60 * 1000;
+let useManagedCookie = false;
 function mergeCookie(name, value) {
     if (!name || !value)
         return;
@@ -66,12 +68,22 @@ function setMcmodCookie(cookie, checkedAt = Date.now()) {
     globalCookie = String(cookie || '');
     cookieLastCheck = checkedAt;
 }
+function configureMcmodCookie(options = {}) {
+    useManagedCookie = !!options.autoCookie;
+    const interval = Number(options.checkInterval);
+    if (Number.isFinite(interval) && interval > 0)
+        cookieCheckInterval = interval;
+    if (options.cookie !== undefined) {
+        setMcmodCookie(options.cookie || '', options.cookie ? Date.now() : 0);
+    }
+}
 function getMcmodCookie() {
     return globalCookie;
 }
 function loadManagedCookie(logger) {
     if (!cookieManager)
         return;
+    useManagedCookie = true;
     cookieManager.getCookie().then(cookie => {
         var _a;
         if (cookie) {
@@ -125,10 +137,10 @@ function getImageHeaders(url, referer = `${constants_1.BASE_URL}/`) {
 }
 async function ensureValidCookie() {
     const now = Date.now();
-    if (hasCookie('MCMOD_SEED') && (now - cookieLastCheck) < COOKIE_CHECK_INTERVAL) {
+    if (hasCookie('MCMOD_SEED') && (now - cookieLastCheck) < cookieCheckInterval) {
         return;
     }
-    if (cookieManager) {
+    if (useManagedCookie && cookieManager) {
         try {
             const cookie = await cookieManager.getCookie();
             if (cookie) {

package/dist/mcmod/plugin.js

@@ -16,6 +16,8 @@ exports.name = 'mcmod-search';
 exports.Config = Schema.object({
     sendLink: Schema.boolean().default(true).description('发送卡片后是否附带链接'),
     cookie: Schema.string().description('【可选】手动填写 mcmod.cn 的 Cookie'),
+    autoCookie: Schema.boolean().default(false).description('自动从 cookie-manager 获取 mcmod.cn Cookie（存在该模块时生效）'),
+    cookieCheckInterval: Schema.number().default(30 * 60 * 1000).description('Cookie/Seed 检查间隔(ms)'),
     fontPath: Schema.string().role('path').description('可选：自定义字体文件路径'),
     debug: Schema.boolean().default(false).description('输出渲染调试日志'),
     render: Schema.object({
@@ -34,9 +36,13 @@ function apply(ctx, config) {
     if (!(0, rendering_1.configureRenderer)(config === null || config === void 0 ? void 0 : config.canvas, config, logger)) {
         return;
     }
+    (0, http_1.configureMcmodCookie)({
+        cookie: config.cookie,
+        autoCookie: config.autoCookie,
+        checkInterval: config.cookieCheckInterval,
+    });
     // 初始化 Cookie
     if (config.cookie) {
-        (0, http_1.setMcmodCookie)(config.cookie);
         logger.info('使用手动配置的 Cookie');
     }
     else if (config.autoCookie) {

package/dist/nlu.js

@@ -0,0 +1,239 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Config = void 0;
+exports.normalizeAiDecision = normalizeAiDecision;
+exports.buildCommand = buildCommand;
+exports.apply = apply;
+const { Schema } = require('koishi');
+const fetch = require('node-fetch');
+const DEFAULT_ENDPOINT = 'https://api.openai.com/v1/chat/completions';
+const DEFAULT_MODEL = 'gpt-4o-mini';
+exports.Config = Schema.object({
+    enabled: Schema.boolean().default(false).description('启用 @机器人 自然语言查询入口'),
+    endpoint: Schema.string().default(DEFAULT_ENDPOINT).description('OpenAI 兼容 Chat Completions 接口地址'),
+    apiKey: Schema.string().role('secret').description('OpenAI 兼容接口 API Key'),
+    model: Schema.string().default(DEFAULT_MODEL).description('模型名称'),
+    timeout: Schema.number().default(15000).description('AI 请求超时(ms)'),
+    temperature: Schema.number().default(0).description('AI 温度参数'),
+}).description('—— AI 自然语言理解 ——');
+const PLATFORM_ALIASES = {
+    mcmod: 'mcmod', cnmc: 'mcmod', mc: 'mcmod', 'mcmod.cn': 'mcmod', 'mc百科': 'mcmod',
+    cf: 'cf', curseforge: 'cf', curse: 'cf',
+    mr: 'mr', modrinth: 'mr',
+};
+const TYPE_ALIASES = {
+    mod: 'mod', mods: 'mod', 模组: 'mod',
+    pack: 'pack', modpack: 'pack', 整合包: 'pack',
+    resource: 'resource', resourcepack: 'resource', 材质: 'resource', 资源包: 'resource', 材质包: 'resource',
+    shader: 'shader', 光影: 'shader',
+    plugin: 'plugin', 插件: 'plugin',
+    data: 'data', item: 'data', 资料: 'data', 物品: 'data',
+    tutorial: 'tutorial', post: 'tutorial', 教程: 'tutorial',
+    author: 'author', 作者: 'author',
+    user: 'user', 用户: 'user',
+};
+const PLATFORM_TYPES = {
+    mcmod: new Set(['mod', 'pack', 'data', 'tutorial', 'author', 'user']),
+    cf: new Set(['mod', 'pack', 'resource', 'shader', 'plugin']),
+    mr: new Set(['mod', 'pack', 'resource', 'shader', 'plugin']),
+};
+function normalizeEndpoint(endpoint) {
+    const value = String(endpoint || DEFAULT_ENDPOINT).trim().replace(/\/+$/, '');
+    if (/\/chat\/completions$/i.test(value))
+        return value;
+    if (/\/v\d+$/i.test(value))
+        return `${value}/chat/completions`;
+    return value;
+}
+function withTimeout(timeout) {
+    const controller = new AbortController();
+    const timer = setTimeout(() => controller.abort(), Math.max(1000, Number(timeout) || 15000));
+    return { controller, done: () => clearTimeout(timer) };
+}
+function extractJsonObject(text) {
+    const value = String(text || '').trim();
+    try {
+        return JSON.parse(value);
+    }
+    catch { }
+    const match = value.match(/\{[\s\S]*\}/);
+    if (!match)
+        throw new Error('AI 返回内容不是 JSON');
+    return JSON.parse(match[0]);
+}
+function normalizePlatform(value) {
+    const key = String(value || '').trim().toLowerCase();
+    return PLATFORM_ALIASES[key] || 'mcmod';
+}
+function normalizeType(value, platform) {
+    var _a;
+    const key = String(value || '').trim().toLowerCase();
+    const type = TYPE_ALIASES[key] || 'mod';
+    return ((_a = PLATFORM_TYPES[platform]) === null || _a === void 0 ? void 0 : _a.has(type)) ? type : 'mod';
+}
+function normalizeAiDecision(raw) {
+    const action = String((raw === null || raw === void 0 ? void 0 : raw.action) || '').trim().toLowerCase();
+    if (action && action !== 'search')
+        return { action: 'ignore' };
+    const query = String((raw === null || raw === void 0 ? void 0 : raw.query) || (raw === null || raw === void 0 ? void 0 : raw.keyword) || '').replace(/[\r\n]+/g, ' ').trim();
+    if (!query)
+        return { action: 'ignore' };
+    const platform = normalizePlatform(raw === null || raw === void 0 ? void 0 : raw.platform);
+    const type = normalizeType(raw === null || raw === void 0 ? void 0 : raw.type, platform);
+    return { action: 'search', platform, type, query };
+}
+async function requestAi(config, text) {
+    var _a, _b, _c, _d, _e, _f;
+    const endpoint = normalizeEndpoint(config === null || config === void 0 ? void 0 : config.endpoint);
+    const headers = { 'Content-Type': 'application/json' };
+    if (config === null || config === void 0 ? void 0 : config.apiKey)
+        headers['Authorization'] = `Bearer ${config.apiKey}`;
+    const body = {
+        model: (config === null || config === void 0 ? void 0 : config.model) || DEFAULT_MODEL,
+        temperature: Number((_a = config === null || config === void 0 ? void 0 : config.temperature) !== null && _a !== void 0 ? _a : 0) || 0,
+        response_format: { type: 'json_object' },
+        messages: [
+            {
+                role: 'system',
+                content: [
+                    '你是 Minecraft 模组搜索意图解析器，只返回 JSON，不要解释。',
+                    'JSON 格式: {"action":"search|ignore","platform":"mcmod|cf|mr","type":"mod|pack|data|tutorial|author|user|resource|shader|plugin","query":"关键词"}',
+                    '默认 platform 为 mcmod，默认 type 为 mod。',
+                    'cf/curseforge 表示 CurseForge，mr/modrinth 表示 Modrinth，cnmc/mcmod/MC百科 表示 mcmod.cn。',
+                    '“查询/搜索/查一下/找一下/模组”等只是意图词，不要放进 query；保留具体模组名、英文名、ID 或关键词。',
+                    '如果不是搜索请求或没有明确关键词，返回 {"action":"ignore"}。',
+                ].join('\n'),
+            },
+            { role: 'user', content: text },
+        ],
+    };
+    const run = async (payload) => {
+        const { controller, done } = withTimeout(config === null || config === void 0 ? void 0 : config.timeout);
+        try {
+            const res = await fetch(endpoint, {
+                method: 'POST',
+                headers,
+                body: JSON.stringify(payload),
+                signal: controller.signal,
+            });
+            const responseText = await res.text();
+            if (!res.ok) {
+                const err = new Error(`AI 请求失败: HTTP ${res.status} ${responseText.slice(0, 300)}`);
+                err.status = res.status;
+                throw err;
+            }
+            return JSON.parse(responseText);
+        }
+        finally {
+            done();
+        }
+    };
+    let json;
+    try {
+        json = await run(body);
+    }
+    catch (e) {
+        if ((e === null || e === void 0 ? void 0 : e.status) !== 400 || !body.response_format)
+            throw e;
+        const retryBody = { ...body };
+        delete retryBody.response_format;
+        json = await run(retryBody);
+    }
+    const content = ((_d = (_c = (_b = json === null || json === void 0 ? void 0 : json.choices) === null || _b === void 0 ? void 0 : _b[0]) === null || _c === void 0 ? void 0 : _c.message) === null || _d === void 0 ? void 0 : _d.content) || ((_f = (_e = json === null || json === void 0 ? void 0 : json.choices) === null || _e === void 0 ? void 0 : _e[0]) === null || _f === void 0 ? void 0 : _f.text) || (json === null || json === void 0 ? void 0 : json.output_text);
+    if (!content)
+        throw new Error('AI 返回中没有 message.content');
+    return normalizeAiDecision(extractJsonObject(content));
+}
+function botIds(session) {
+    var _a, _b;
+    return [session === null || session === void 0 ? void 0 : session.selfId, (_a = session === null || session === void 0 ? void 0 : session.bot) === null || _a === void 0 ? void 0 : _a.selfId, (_b = session === null || session === void 0 ? void 0 : session.bot) === null || _b === void 0 ? void 0 : _b.userId]
+        .filter(Boolean)
+        .map(id => String(id));
+}
+function hasAtSelf(session) {
+    var _a, _b, _c, _d;
+    const ids = botIds(session);
+    const elements = Array.isArray(session === null || session === void 0 ? void 0 : session.elements) ? session.elements : [];
+    for (const element of elements) {
+        if ((element === null || element === void 0 ? void 0 : element.type) !== 'at')
+            continue;
+        const id = String(((_a = element === null || element === void 0 ? void 0 : element.attrs) === null || _a === void 0 ? void 0 : _a.id) || ((_b = element === null || element === void 0 ? void 0 : element.attrs) === null || _b === void 0 ? void 0 : _b.userId) || ((_c = element === null || element === void 0 ? void 0 : element.attrs) === null || _c === void 0 ? void 0 : _c.qq) || '');
+        if (id && (!ids.length || ids.includes(id)))
+            return true;
+    }
+    const content = String((session === null || session === void 0 ? void 0 : session.content) || '');
+    const atRegex = /<at\s+([^>]*?)\/?>(?:<\/at>)?/gi;
+    let match;
+    while ((match = atRegex.exec(content))) {
+        const attrs = match[1] || '';
+        const id = (_d = attrs.match(/(?:id|user-id|qq)=(['"]?)([^'"\s/>]+)\1/i)) === null || _d === void 0 ? void 0 : _d[2];
+        if (id && (!ids.length || ids.includes(String(id))))
+            return true;
+    }
+    return false;
+}
+function stripAt(text) {
+    return String(text || '')
+        .replace(/<at\s+[^>]*\/?>(?:<\/at>)?/gi, ' ')
+        .replace(/\[CQ:at,[^\]]+\]/gi, ' ')
+        .replace(/\s+/g, ' ')
+        .trim();
+}
+function getMentionText(session) {
+    var _a;
+    const stripped = String(((_a = session === null || session === void 0 ? void 0 : session.stripped) === null || _a === void 0 ? void 0 : _a.content) || '').trim();
+    const raw = stripAt((session === null || session === void 0 ? void 0 : session.content) || '');
+    return (stripped && stripped !== String((session === null || session === void 0 ? void 0 : session.content) || '').trim()) ? stripped : raw;
+}
+function isExplicitCommand(text, prefixes) {
+    const value = String(text || '').trim().toLowerCase();
+    return Object.values(prefixes || {}).some(prefix => {
+        const p = String(prefix || '').trim().toLowerCase();
+        return p && (value === p || value.startsWith(`${p} `) || value.startsWith(`${p}.`));
+    });
+}
+function buildCommand(decision, prefixes) {
+    if (!decision || decision.action !== 'search')
+        return '';
+    const platform = normalizePlatform(decision.platform);
+    const type = normalizeType(decision.type, platform);
+    const query = String(decision.query || '').replace(/[\r\n]+/g, ' ').trim();
+    if (!query)
+        return '';
+    const prefix = platform === 'mcmod'
+        ? ((prefixes === null || prefixes === void 0 ? void 0 : prefixes.cnmc) || 'cnmc')
+        : platform === 'cf'
+            ? ((prefixes === null || prefixes === void 0 ? void 0 : prefixes.cf) || 'cf')
+            : ((prefixes === null || prefixes === void 0 ? void 0 : prefixes.mr) || 'mr');
+    return `${prefix}.${type} ${query}`;
+}
+function apply(ctx, config, shared = {}) {
+    if (!(config === null || config === void 0 ? void 0 : config.enabled))
+        return;
+    const logger = ctx.logger('minecraft-nlu');
+    const prefixes = (shared === null || shared === void 0 ? void 0 : shared.prefixes) || {};
+    ctx.middleware(async (session, next) => {
+        if (!hasAtSelf(session))
+            return next();
+        const text = getMentionText(session);
+        if (!text || isExplicitCommand(text, prefixes))
+            return next();
+        let decision;
+        try {
+            decision = await requestAi(config, text);
+        }
+        catch (e) {
+            logger.warn(`AI 自然语言解析失败: ${(e === null || e === void 0 ? void 0 : e.message) || e}`);
+            await session.send(`AI 自然语言解析失败: ${(e === null || e === void 0 ? void 0 : e.message) || e}`);
+            return;
+        }
+        const command = buildCommand(decision, prefixes);
+        if (!command)
+            return next();
+        if ((shared === null || shared === void 0 ? void 0 : shared.debug) || (config === null || config === void 0 ? void 0 : config.debug))
+            logger.info(`NLU: ${text} -> ${command}`);
+        const result = await session.execute(command);
+        if (result)
+            await session.send(result);
+    });
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-cfmrmod",
-  "version": "1.1.4",
+  "version": "1.1.5",
   "description": "Koishi 插件：搜索 CurseForge/Modrinth/MCMod 并渲染图片卡片",
   "main": "dist/index.js",
   "files": [