feat: add fake random

Author: pynickle

README.md

@@ -20,21 +20,27 @@
 - 🔒 **管理员保护**：可配置管理员消息保护
 - 📏 **媒体限制**：支持配置媒体文件大小限制
 - 👥 **转发消息用户选择**：支持从转发消息中选择相关用户进行绑定
+- ⚖️ **加权随机抽取**：根据消息被抽取次数动态调整抽取概率，被抽取次数越多，概率越低。使用公式：
+  ```
+  权重 = 1 / (1 + drawCount * α)
+  概率 = 权重 / 总权重
+  ```
+  其中，`drawCount` 为消息被抽取次数，`α` 为调整因子（可配置，默认0.2）
 
 ## 📋 命令列表
 
-| 命令                   | 说明                      | 权限要求            |
-|----------------------|-------------------------|-----------------|
-| `cave`               | 随机获取一条回声洞消息             | 所有人             |
-| `cave <id>`          | 获取特定 ID 的回声洞消息          | 所有人             |
-| `cave.echo [...userIds]` | 将引用的消息存入回声洞           | 所有人             |
-| `cave.drop <id>`     | 删除特定 ID 的回声洞消息          | 消息存储者、原始发送者或管理员 |
-| `cave.listen`        | 获取自己投稿的回声洞列表            | 所有人             |
-| `cave.trace`         | 获取自己发言被投稿的回声洞列表         | 所有人             |
-| `cave.purge <...ids>` | 批量删除多个 ID 的回声洞消息        | 消息存储者、原始发送者或管理员 |
-| `cave.search <id>`   | 搜索特定 ID 的回声洞消息          | 所有人             |
+| 命令                            | 说明               | 权限要求            |
+|-------------------------------|------------------|-----------------|
+| `cave`                        | 随机获取一条回声洞消息      | 所有人             |
+| `cave <id>`                   | 获取特定 ID 的回声洞消息   | 所有人             |
+| `cave.echo [...userIds]`      | 将引用的消息存入回声洞      | 所有人             |
+| `cave.drop <id>`              | 删除特定 ID 的回声洞消息   | 消息存储者、原始发送者或管理员 |
+| `cave.listen`                 | 获取自己投稿的回声洞列表     | 所有人             |
+| `cave.trace`                  | 获取自己发言被投稿的回声洞列表  | 所有人             |
+| `cave.purge <...ids>`         | 批量删除多个 ID 的回声洞消息 | 消息存储者、原始发送者或管理员 |
+| `cave.search <id>`            | 搜索特定 ID 的回声洞消息   | 所有人             |
 | `cave.bind <id> <...userIds>` | 将用户绑定到特定 ID 的回声洞 | 所有人             |
-| `cave.rank [period]` | 查看回声洞排行榜，支持多种时间段 | 所有人             |
+| `cave.rank [period]`          | 查看回声洞排行榜，支持多种时间段 | 所有人             |
 
 ## 🚀 使用指南
 
@@ -64,6 +70,7 @@ npm install koishi-plugin-echo-cave
 | `rankingTopCount`            | number  | `10`    | 排行榜显示的用户数量                    |
 | `forwardSelectTimeout`       | number  | `20`    | 转发消息用户选择超时时间（秒）              |
 | `enableForwardUserSelection` | boolean | `true`  | 启用转发消息用户选择功能                  |
+| `alpha`                      | number  | `0.2`   | 加权随机抽取的调整因子，控制抽取次数对概率的影响程度，值越大影响越明显 |
 
 ## 📝 注意事项
 

src/config/config.ts

@@ -16,6 +16,7 @@ export interface Config {
     rankingTopCount?: number;
     forwardSelectTimeout?: number;
     enableForwardUserSelection?: boolean;
+    alpha?: number;
 }
 
 export const Config: Schema<Config> = Schema.object({
@@ -33,6 +34,7 @@ export const Config: Schema<Config> = Schema.object({
     rankingTopCount: Schema.number().default(10),
     forwardSelectTimeout: Schema.number().default(20),
     enableForwardUserSelection: Schema.boolean().default(true),
+    alpha: Schema.number().default(0.2).min(0).max(2),
 }).i18n({
     'zh-CN': zhCN,
 });

src/config/locales/zh-CN.json

@@ -12,5 +12,6 @@
   "sendAllAsForwardMsg": "是否将所有消息以转发消息形式发送，开启后普通消息也会转换为转发消息格式",
   "rankingTopCount": "排行榜显示的用户数量",
   "forwardSelectTimeout": "合并转发消息选择关联用户的超时时间 (秒)",
-  "enableForwardUserSelection": "是否启用合并转发消息的关联用户选择功能"
+  "enableForwardUserSelection": "是否启用合并转发消息的关联用户选择功能",
+  "alpha": "加权随机抽取的调整因子，控制抽取次数对概率的影响程度，值越大影响越明显"
 }

src/core/command/get-cave.ts

@@ -69,7 +69,38 @@ export async function getCave(ctx: Context, session: Session, cfg: Config, id: n
             return session.text('.noMsgInCave');
         }
 
-        caveMsg = caves[Math.floor(Math.random() * caves.length)];
+        // Use weighted random selection based on drawCount
+        const alpha = cfg.alpha || 0.2;
+
+        // Calculate weights for each cave
+        const weights = caves.map((cave) => 1 / (1 + cave.drawCount * alpha));
+
+        // Calculate total weight
+        const totalWeight = weights.reduce((sum, weight) => sum + weight, 0);
+
+        // Generate a random number between 0 and totalWeight
+        let random = Math.random() * totalWeight;
+
+        // Select cave based on weights
+        let selectedIndex = 0;
+        while (random > weights[selectedIndex]) {
+            random -= weights[selectedIndex];
+            selectedIndex++;
+        }
+
+        caveMsg = caves[selectedIndex];
+
+        // Update drawCount for the selected cave
+        await ctx.database.set(
+            'echo_cave_v2',
+            {
+                id: caveMsg.id,
+                channelId,
+            },
+            {
+                drawCount: caveMsg.drawCount + 1,
+            }
+        );
     } else {
         const caves = await ctx.database.get('echo_cave_v2', {
             id,
@@ -81,6 +112,18 @@ export async function getCave(ctx: Context, session: Session, cfg: Config, id: n
         }
 
         caveMsg = caves[0];
+
+        // Update drawCount for the specified cave
+        await ctx.database.set(
+            'echo_cave_v2',
+            {
+                id,
+                channelId,
+            },
+            {
+                drawCount: caveMsg.drawCount + 1,
+            }
+        );
     }
 
     await sendCaveMsg(ctx, session, caveMsg, cfg);

src/index.ts

@@ -23,6 +23,7 @@ export interface EchoCave {
     type: 'forward' | 'msg';
     content: string;
     relatedUsers: string[];
+    drawCount: number;
 }
 
 declare module 'koishi' {
@@ -46,6 +47,7 @@ export function apply(ctx: Context, cfg: Config) {
             type: 'string',
             content: 'text',
             relatedUsers: 'list',
+            drawCount: { type: 'unsigned', default: 0 },
         },
         {
             primary: 'id',
@@ -64,6 +66,7 @@ export function apply(ctx: Context, cfg: Config) {
             type: 'string',
             content: 'text',
             relatedUsers: 'list',
+            drawCount: { type: 'unsigned', default: 0 },
         },
         {
             primary: 'id',