koishi-plugin-chatluna-affinity 0.0.3 → 0.0.5

package/lib/affinity-store.js

@@ -46,6 +46,7 @@ function extendModel(ctx) {
       platform: 'string',
       selfId: { type: 'string', nullable: true },
       userId: 'string',
+      nickname: { type: 'string', nullable: true },
       affinity: 'integer',
       affinityInited: 'boolean',
       relation: { type: 'string', nullable: true },
@@ -142,7 +143,15 @@ function createAffinityStore(ctx, config, log) {
     const existing = await ctx.database.get(MODEL_NAME, { id });
     const current = existing?.[0];
     const now = new Date();
-    const relationText = relation !== undefined && relation !== null ? String(relation).trim() : undefined;
+    const manual = findManualRelationship(identity.platform, identity.userId);
+    let relationText;
+    if (relation === undefined || relation === null) {
+      relationText = manual?.relation ?? current?.relation ?? '';
+    } else {
+      const trimmed = String(relation).trim();
+      relationText = trimmed ? trimmed : manual?.relation ?? current?.relation ?? '';
+    }
+    const nickname = seed?.nickname || seed?.authorNickname || seed?.session?.nickname || seed?.session?.author?.nickname || seed?.session?.author?.name || '';
 
     const payload = {
       id,
@@ -151,11 +160,16 @@ function createAffinityStore(ctx, config, log) {
       userId: identity.userId,
       affinity: typeof value === 'number' ? value : current?.affinity ?? defaultInitialValue(),
       affinityInited: inited ?? current?.affinityInited ?? false,
-      relation: relationText !== undefined ? relationText : current?.relation ?? '',
+      relation: relationText,
+      nickname: nickname || current?.nickname || manual?.relationNickname || '',
       updatedAt: now,
       relationUpdatedAt: relationText !== undefined ? now : current?.relationUpdatedAt ?? now
     };
 
+    if (nickname && (!current || nickname !== current.nickname)) {
+      payload.nickname = nickname;
+    }
+
     await ctx.database.upsert(MODEL_NAME, [payload]);
     return payload;
   };
@@ -271,14 +285,17 @@ function createAffinityStore(ctx, config, log) {
         : manual?.initialAffinity ?? rollInitial();
       const initial = clampFn(base, config.min, config.max);
       const level = resolveLevelByAffinity(initial);
-      await save({ platform, userId, selfId: session?.selfId }, initial, true, level?.relation ?? '');
+      const desiredRelation = manual?.relation ?? level?.relation ?? '';
+      await save({ platform, userId, selfId: session?.selfId, session }, initial, true, desiredRelation);
       return { affinity: initial, isNew: true };
     }
 
     const normalized = clampFn(record.affinity ?? defaultInitialValue(), config.min, config.max);
     if (normalized !== record.affinity) {
       const level = resolveLevelByAffinity(normalized);
-      await save({ platform, userId, selfId: session?.selfId }, normalized, record.affinityInited, level?.relation ?? record.relation ?? '');
+      const manual = findManualRelationship(platform, userId);
+      const desiredRelation = manual?.relation ?? level?.relation ?? record.relation ?? '';
+      await save({ platform, userId, selfId: session?.selfId, session }, normalized, record.affinityInited, desiredRelation);
     }
     return { affinity: normalized, isNew: false };
   };
@@ -301,4 +318,4 @@ function createAffinityStore(ctx, config, log) {
   };
 }
 
-module.exports = { createAffinityStore };
+module.exports = { createAffinityStore, MODEL_NAME };

package/lib/index.d.ts

@@ -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

@@ -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/middleware.js

@@ -128,6 +128,7 @@ function createAnalysisMiddleware(ctx, config, { store, history, cache, getModel
       const jsonCandidate = typeof text === 'string' ? text : String(text ?? '');
       const match = jsonCandidate.match(/\{[\s\S]*\}/);
       let delta = 0;
+      const nickname = session?.author?.nickname || session?.author?.name || session?.user?.nickname || session?.user?.name || session?.username || session?.nickname || '';
       if (match) {
         try {
           const parsed = JSON.parse(match[0]);
@@ -137,7 +138,7 @@ function createAnalysisMiddleware(ctx, config, { store, history, cache, getModel
           if (action === 'increase' && delta <= 0) delta = Math.max(1, Math.abs(delta));
           if (action === 'decrease' && delta >= 0) delta = -Math.max(1, Math.abs(delta));
           if (action === 'hold') delta = 0;
-          if (debugEnabled) log('info', '模型返回', { raw: parsed, parsedDelta: delta, action, userId: session.userId, platform: session.platform });
+          if (debugEnabled) log('info', '模型返回', { raw: parsed, parsedDelta: delta, action, userId: session.userId, platform: session.platform, nickname });
         } catch (error) {
           log('warn', '解析模型响应失败', { text: jsonCandidate, error });
         }
@@ -149,11 +150,12 @@ function createAnalysisMiddleware(ctx, config, { store, history, cache, getModel
         ? Math.min(delta, positiveLimit)
         : Math.max(delta, -negativeLimit);
       const nextAffinity = clampValue(oldAffinity + limitedDelta, config.min, config.max);
+
       if (nextAffinity !== oldAffinity) {
         const level = store.resolveLevelByAffinity(nextAffinity);
-        await store.save({ platform: session.platform, userId: session.userId, selfId: session?.selfId }, nextAffinity, true, level?.relation ?? '');
+        await store.save({ platform: session.platform, userId: session.userId, selfId: session?.selfId, session }, nextAffinity, true, level?.relation ?? '');
         cache.set(session.platform, session.userId, nextAffinity);
-        log('info', '好感度已更新', { oldAffinity, delta: limitedDelta, nextAffinity, userId: session.userId, platform: session.platform });
+        log('info', '好感度已更新', { oldAffinity, delta: limitedDelta, nextAffinity, userId: session.userId, platform: session.platform, nickname });
       } else if (config.useLastAffinity) {
         cache.set(session.platform, session.userId, oldAffinity);
       }

package/lib/schema.js

@@ -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({
@@ -36,7 +36,7 @@ const AffinitySchema = Schema.object({
   analysisPrompt: Schema.string()
     .role('textarea')
     .default(
-      '你是好感度管家，需要根据上下文评估是否调整好感度。\n- 关注最近若干条群聊消息，判断整体语气与语境；\n- 当用户友善、感谢、积极互动时，适度增加；\n- 当用户正常交流且无明显倾向时，保持不变；\n- 当用户冒犯、敷衍、重复打扰时，减少；\n- 增加幅度不超过 {{maxIncreasePerMessage}} ，减少幅度不超过 {{maxDecreasePerMessage}} ，并保持在提供的范围内；\n- 使用 action 表示行为：increase 增加、decrease 减少、hold 保持。\n\n角色设定：{{personaPrompt}}\n\n当前好感度：{{currentAffinity}} （范围 {{minAffinity}} ~ {{maxAffinity}}）\n最近 {{historyCount}} 条消息（旧 -> 新）：\n{{historyText}}\n\n本次用户消息：\n{{userMessage}}\n\n请仅输出 JSON：{"delta": 整数, "action": "increase|decrease|hold", "reason": "简短中文原因"}。'
+      '你是好感度管家，需要根据上下文评估是否调整好感度。\n- 仔细阅读最近若干条群聊消息，聚焦语气、内容、语境与互动意图；\n- 只有当用户展现出明确、具体的善意或贡献时（如感谢、称赞、帮助解决问题、分享价值信息、表达真诚关心），才考虑增加；\n- 若用户只是礼貌问候、例行回应、轻度调侃或重复旧话题，可保持不变；\n- 对于冒犯、敷衍、抄袭、频繁打扰、传播负面情绪或破坏氛围的行为，应减少；\n- 增加幅度不超过 {{maxIncreasePerMessage}} ，减少幅度不超过 {{maxDecreasePerMessage}} ，并保持在提供的范围内；\n- 在总结理由时，指明触发增减的具体言辞或行为；\n- 使用 action 表示行为：increase 增加、decrease 减少、hold 保持。\n\n角色设定：{{personaPrompt}}\n\n当前好感度：{{currentAffinity}} （范围 {{minAffinity}} ~ {{maxAffinity}}）\n最近 {{historyCount}} 条消息（旧 -> 新）：\n{{historyText}}\n\n本次用户消息：\n{{userMessage}}\n\n请仅输出 JSON：{"delta": 整数, "action": "increase|decrease|hold", "reason": "简短中文原因"}。'
     )
     .description('好感度分析主提示词'),
   personaPrompt: Schema.string()
@@ -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

@@ -20,7 +20,7 @@ function createAffinityTool(options) {
       if (!platform || !userId) return 'Missing platform or user ID. Unable to adjust affinity.';
       const value = options.clamp(input.affinity);
       const level = options.resolveLevelByAffinity(value);
-      await options.save({ platform, userId, selfId: session?.selfId }, value, true, level?.relation ?? options.defaultRelation);
+      await options.save({ platform, userId, selfId: session?.selfId, session }, value, true, level?.relation ?? options.defaultRelation);
       options.cache.set(platform, userId, value);
       if (level?.relation) {
         return `Affinity for ${platform}/${userId} set to ${value}. Relationship updated to ${level.relation}.`;
@@ -58,7 +58,7 @@ function createRelationshipTool(options) {
       }
       const baseValue = level ? level.min : options.defaultInitial();
       const base = options.clamp(baseValue);
-      await options.save({ platform, userId, selfId: session?.selfId }, base, true, relationName);
+      await options.save({ platform, userId, selfId: session?.selfId, session }, base, true, relationName);
       options.cache.set(platform, userId, base);
       options.updateRelationshipConfig(userId, relationName, base);
       if (level) {
@@ -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

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

package/readme.md

@@ -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`）。
 
 ## 工具调用