npm - koishi-plugin-chatluna-affinity - Versions diffs - 0.0.4 → 0.0.6 - Mend

koishi-plugin-chatluna-affinity 0.0.4 → 0.0.6

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (8) hide show

package/lib/affinity-store.js CHANGED Viewed

@@ -318,4 +318,4 @@ function createAffinityStore(ctx, config, log) {
   };
 }
-module.exports = { createAffinityStore };
+module.exports = { createAffinityStore, MODEL_NAME };

package/lib/history.js CHANGED Viewed

@@ -42,25 +42,6 @@ function createHistoryManager(ctx, config, log) {
     cache.set(key, list);
   }
-  async function readFromService(session, count) {
-    const service = ctx.chatluna_group_analysis_message;
-    if (!service?.getHistoricalMessages) return [];
-    try {
-      const messages = await service.getHistoricalMessages({
-        guildId: session.guildId,
-        channelId: session.channelId,
-        limit: count,
-        selfId: session.selfId,
-        platform: session.platform,
-        endTime: new Date()
-      });
-      return normalizeEntries(messages, count);
-    } catch (error) {
-      log('warn', '调用 chatluna_group_analysis_message 获取历史消息失败', error);
-      return [];
-    }
-  }
   async function readFromCache(session, count) {
     const cached = cache.get(makeKey(session));
     if (cached?.length) return normalizeEntries(cached, count);
@@ -81,8 +62,6 @@ function createHistoryManager(ctx, config, log) {
   async function fetch(session) {
     const count = config.historyMessageCount || 0;
     if (count <= 0) return [];
-    const serviceEntries = await readFromService(session, count);
-    if (serviceEntries.length) return serviceEntries;
     return readFromCache(session, count);
   }

package/lib/index.d.ts CHANGED Viewed

@@ -38,7 +38,10 @@ export interface Config {
     note: string;
   }>;
   registerAffinityTool: boolean;
+  registerBlacklistTool: boolean;
   registerRelationshipTool: boolean;
+  rankDefaultLimit: number;
+  rankRenderAsImage: boolean;
 }
 export declare const Config: Schema<Config>;
 export declare const inject: string[];

package/lib/index.js CHANGED Viewed

@@ -1,3 +1,4 @@
+const { h } = require('koishi');
 const { modelSchema } = require('koishi-plugin-chatluna/utils/schema');
 const { ChatLunaPlugin } = require('koishi-plugin-chatluna/services/chat');
 const { getMessageContent } = require('koishi-plugin-chatluna/utils/string');
@@ -5,7 +6,7 @@ const { getMessageContent } = require('koishi-plugin-chatluna/utils/string');
 const { Config, inject, name } = require('./schema');
 const { createLogger } = require('./logger');
 const { renderTemplate } = require('./template');
-const { createAffinityStore } = require('./affinity-store');
+const { createAffinityStore, MODEL_NAME } = require('./affinity-store');
 const { createHistoryManager } = require('./history');
 const { createAffinityCache } = require('./cache');
 const { createAffinityProvider, createRelationshipProvider } = require('./providers');
@@ -63,6 +64,12 @@ function apply(ctx, config) {
         createTool: registry.createRelationshipTool
       });
     }
+    if (config.registerBlacklistTool) {
+      plugin.registerTool('adjust_blacklist', {
+        selector: registry.blacklistSelector,
+        createTool: registry.createBlacklistTool
+      });
+    }
   });
   const middleware = createAnalysisMiddleware(ctx, config, {
@@ -76,6 +83,138 @@ function apply(ctx, config) {
   });
   ctx.middleware(middleware);
+  const renderRankImage = async (lines) => {
+    const puppeteer = ctx.puppeteer;
+    if (!puppeteer?.page) return null;
+    const html = `<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+  <meta charset="utf-8" />
+  <style>
+    body {
+      margin: 0;
+      font-family: "Segoe UI", "Helvetica Neue", PingFangSC, "Microsoft Yahei", sans-serif;
+      background: #ffffff;
+      color: #111111;
+    }
+    .container {
+      padding: 20px 24px;
+    }
+    h1 {
+      font-size: 18px;
+      margin: 0 0 16px;
+      font-weight: 600;
+    }
+    table {
+      border-collapse: collapse;
+      width: 100%;
+      min-width: 360px;
+      font-size: 14px;
+    }
+    th, td {
+      padding: 8px 12px;
+      border-bottom: 1px solid #e5e5e5;
+      text-align: left;
+      white-space: nowrap;
+    }
+    th {
+      background: #f5f7fa;
+      font-weight: 600;
+    }
+    tr:nth-child(odd) td {
+      background: #fbfcfe;
+    }
+  </style>
+</head>
+<body>
+  <div class="container" id="rank-root">
+    <h1>好感度排行</h1>
+    <table>
+      <thead>
+        <tr><th>排名</th><th>用户名</th><th>关系</th><th>好感度</th></tr>
+      </thead>
+      <tbody>
+        ${lines.map((line, index) => {
+          const [name, relation, affinity] = line;
+          return `<tr><td>${index + 1}</td><td>${name}</td><td>${relation}</td><td>${affinity}</td></tr>`;
+        }).join('')}
+      </tbody>
+    </table>
+  </div>
+</body>
+</html>`;
+    let page;
+    try {
+      page = await puppeteer.page();
+      await page.setViewport({ width: 600, height: 200 + lines.length * 32 });
+      await page.setContent(html, { waitUntil: 'networkidle0' });
+      const element = await page.$('#rank-root');
+      if (!element) return null;
+      const buffer = await element.screenshot({ omitBackground: false });
+      return buffer;
+    } catch (error) {
+      ctx.logger?.warn?.('排行榜图片渲染失败', error);
+      return null;
+    } finally {
+      try {
+        await page?.close();
+      } catch (_) {
+        // ignore
+      }
+    }
+  };
+  ctx.command('affinity.rank [limit:number] [platform:string] [image]', '查看当前好感度排行', { authority: 1 })
+    .alias('好感度排行')
+    .usage('affinity.rank [人数] [平台] [image|text]')
+    .action(async ({ session }, limitArg, platformArg, imageArg) => {
+      const parsedLimit = Number(limitArg);
+      const limit = Math.max(1, Math.min(Number.isFinite(parsedLimit) ? parsedLimit : config.rankDefaultLimit, 50));
+      const shouldRenderImage = (() => {
+        if (imageArg === undefined) return !!config.rankRenderAsImage;
+        const lower = String(imageArg).toLowerCase();
+        if (['0', 'false', 'text', 'no', 'n'].includes(lower)) return false;
+        if (['1', 'true', 'image', 'img', 'yes', 'y'].includes(lower)) return true;
+        return !!config.rankRenderAsImage;
+      })();
+      if (shouldRenderImage && (!ctx.puppeteer || typeof ctx.puppeteer.page !== 'function')) {
+        return '当前环境未启用 puppeteer，已改为文本模式（可安装 koishi-plugin-puppeteer 或传入 text）。';
+      }
+      const conditions = {};
+      const platform = platformArg || session?.platform;
+      if (platform) conditions.platform = platform;
+      if (session?.selfId) conditions.selfId = session.selfId;
+      const query = ctx.database
+        .select(MODEL_NAME)
+        .project(['platform', 'selfId', 'userId', 'nickname', 'affinity', 'relation'])
+        .orderBy('affinity', 'desc')
+        .limit(limit);
+      if (Object.keys(conditions).length) query.where(conditions);
+      const rows = await query.execute();
+      if (!rows.length) return '当前暂无好感度记录。';
+      const lines = rows.map((row) => {
+        const name = row.nickname || row.userId;
+        const relation = row.relation || '——';
+        return [name, relation, row.affinity];
+      });
+      const textLines = ['用户名 关系 好感度', ...lines.map((item, index) => `${index + 1}. ${item[0]} ${item[1]} ${item[2]}`)];
+      if (shouldRenderImage) {
+        const buffer = await renderRankImage(lines);
+        if (buffer) return h.image(buffer, 'image/png');
+        ctx.logger?.warn?.('排行榜图片渲染失败或服务缺失，已改为文本输出');
+        return textLines.join('\n');
+      }
+      return textLines.join('\n');
+    });
 }
 module.exports = { apply, Config, inject, name };

package/lib/schema.js CHANGED Viewed

@@ -4,7 +4,7 @@ const name = 'chatluna-affinity';
 const inject = {
   required: ['chatluna', 'database'],
-  optional: ['chatluna_group_analysis_message', 'chatluna_group_analysis']
+  optional: ['puppeteer']
 };
 const AffinitySchema = Schema.object({
@@ -44,7 +44,10 @@ const AffinitySchema = Schema.object({
     .default('你是一位温暖可靠的伙伴，会根据好感度高低调整语气：好感度越高越亲近，越低越保持礼貌。')
     .description('补充的人设提示词，会注入到分析提示词中'),
   debugLogging: Schema.boolean().default(false).description('输出调试日志'),
-  registerAffinityTool: Schema.boolean().default(false).description('注册 ChatLuna 工具：调整好感度')
+  registerAffinityTool: Schema.boolean().default(false).description('注册 ChatLuna 工具：调整好感度'),
+  registerBlacklistTool: Schema.boolean().default(false).description('注册 ChatLuna 工具：管理黑名单'),
+  rankDefaultLimit: Schema.number().default(10).min(1).max(50).description('排行榜默认展示人数'),
+  rankRenderAsImage: Schema.boolean().default(false).description('排行榜默认渲染为图片')
 }).description('好感度设置');
 const RelationshipSchema = Schema.object({

package/lib/tools.js CHANGED Viewed

@@ -69,6 +69,38 @@ function createRelationshipTool(options) {
   })();
 }
+function createBlacklistTool(options) {
+  return new (class extends StructuredTool {
+    constructor() {
+      super({});
+      this.name = 'adjust_blacklist';
+      this.description = 'Add or remove a user from the affinity blacklist.';
+      this.schema = z.object({
+        action: z.enum(['add', 'remove']).describe('Add or remove the user from blacklist'),
+        targetUserId: z.string().describe('Target user ID'),
+        platform: z.string().optional().describe('Target platform; defaults to current session'),
+        note: z.string().optional().describe('Optional note when adding to blacklist')
+      });
+    }
+    async _call(input, _manager, runnable) {
+      const session = runnable?.configurable?.session;
+      const platform = input.platform || session?.platform;
+      const userId = input.targetUserId;
+      if (!platform || !userId) return 'Missing platform or user ID. Unable to adjust blacklist.';
+      if (input.action === 'add') {
+        options.store.recordBlacklist(platform, userId, input.note ?? 'tool');
+        options.cache.clear(platform, userId);
+        return `User ${platform}/${userId} added to blacklist.`;
+      }
+      const removed = options.store.removeBlacklist(platform, userId);
+      options.cache.clear(platform, userId);
+      return removed
+        ? `User ${platform}/${userId} removed from blacklist.`
+        : `User ${platform}/${userId} not found in blacklist.`;
+    }
+  })();
+}
 function createToolRegistry(config, store, cache) {
   const options = {
     clamp: store.clamp,
@@ -80,6 +112,7 @@ function createToolRegistry(config, store, cache) {
     save: store.save,
     cache,
     updateRelationshipConfig: store.updateRelationshipConfig,
+    store,
     min: config.min,
     max: config.max
   };
@@ -87,8 +120,10 @@ function createToolRegistry(config, store, cache) {
   return {
     affinitySelector: () => true,
     relationshipSelector: () => true,
+    blacklistSelector: () => true,
     createAffinityTool: () => createAffinityTool(options),
-    createRelationshipTool: () => createRelationshipTool(options)
+    createRelationshipTool: () => createRelationshipTool(options),
+    createBlacklistTool: () => createBlacklistTool(options)
   };
 }

package/package.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chatluna-affinity",
   "description": "为 ChatLuna 提供{好感度}与{关系}变量并提供对应的工具调用和低好感自动拉黑功能。仅测试过伪装插件。",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [

package/readme.md CHANGED Viewed

@@ -13,6 +13,7 @@
 - 调整工具：
   - `adjust_affinity` 将好感度设为指定数值，并同步关系区间。
   - `adjust_relationship` 将关系切换为自定义称谓，同时把好感度调至对应区间的下限。
+  - `adjust_blacklist` 管理自动拉黑名单，可快速拉黑或解除指定用户。
 - 特殊关系：可为指定用户记录“初始好感度 + 自定义称谓”，工具操作会实时写回配置。
 ## 主要配置
@@ -35,7 +36,13 @@
 | `autoBlacklist` | 自动拉黑记录列表（自动维护，可在控制台查看与编辑） |
 | `relationships` | 特殊关系配置：`userId`、`initialAffinity`、`relation`、`note` |
 | `relationshipAffinityLevels` | 区间 → 称谓映射，默认提供“陌生人/友好/亲近/挚友” |
-| `registerAffinityTool` / `registerRelationshipTool` | 是否注册对应 ChatLuna 工具 |
+| `registerAffinityTool` / `registerRelationshipTool` / `registerBlacklistTool` | 是否注册对应 ChatLuna 工具 |
+| `rankDefaultLimit` | 指令默认展示的排行榜人数 |
+| `rankRenderAsImage` | 指令默认是否渲染排行榜为图片（需 `koishi-plugin-puppeteer`） |
+## 指令
+- `affinity.rank [limit] [platform] [image|text]`：查看好感度排行，可指定人数、平台，并选择文本或图片输出（图片模式需 `koishi-plugin-puppeteer`）。
 ## 工具调用