@fanyue1117/koishi-plugin-trial-role 1.0.0

package/lib/index.d.ts

@@ -0,0 +1,23 @@
+declare const Schema: any;
+declare module 'koishi' {
+    interface Tables {
+        trial_record: {
+            id: number;
+            userId: string;
+            guildId: string;
+            lastTrialTime: number;
+            createdAt: Date;
+            updatedAt: Date;
+        };
+    }
+}
+declare const pluginName = "trial-role";
+interface Config {
+    trialRoleId: string;
+    successMessage: string;
+    errorMessage: string;
+    cooldownDays: number;
+    durationHours: number;
+}
+declare const Config: any;
+declare function apply(ctx: any, config: any): void;

package/lib/index.js

@@ -0,0 +1,278 @@
+const { Schema } = require('koishi');
+const pluginName = 'trial-role';
+const Config = Schema.object({
+    trialRoleId: Schema.string().description('试用角色的ID').required(),
+    successMessage: Schema.string().description('添加角色成功后的提示消息').default('已成功为您添加试用角色！有效期为1小时。'),
+    errorMessage: Schema.string().description('添加角色失败后的提示消息').default('添加试用角色失败，请联系管理员。'),
+    cooldownDays: Schema.number().description('重复试用的冷却天数').default(3),
+    durationHours: Schema.number().description('试用角色的有效期（小时）').default(1),
+});
+function apply(ctx, config) {
+    // 输出插件加载信息
+    console.log('=== 试用角色插件加载中 ===');
+    // 使用 ctx.model.extend 定义表结构
+    ctx.model.extend('trial_record', {
+        id: 'unsigned',
+        userId: 'string',
+        guildId: 'string',
+        lastTrialTime: 'integer',
+        createdAt: 'timestamp',
+        updatedAt: 'timestamp'
+    }, {
+        primary: 'id',
+        unique: [['userId', 'guildId']],
+        autoInc: true
+    });
+    console.log('=== 表结构定义完成 ===');
+    // 等待所有服务就绪
+    ctx.on('ready', async () => {
+        console.log('=== 所有服务就绪 ===');
+        if (!ctx.model) {
+            console.error('=== 模型对象不存在 ===');
+        }
+        else {
+            console.log('=== 模型对象已就绪 ===');
+        }
+    });
+    // 试用指令
+    console.log('=== 注册试用指令 ===');
+    ctx.command('试用', '申请试用角色')
+        .action(async ({ session }) => {
+        if (!session.guildId) {
+            return '请在服务器内使用此指令。';
+        }
+        if (session.bot.platform !== 'kook') {
+            return '此指令仅支持Kook平台。';
+        }
+        try {
+            console.log('=== 试用指令执行 ===', session.userId, session.guildId);
+            // 检查用户是否在冷却期内
+            const now = Date.now();
+            const cooldownMs = config.cooldownDays * 24 * 60 * 60 * 1000;
+            const durationMs = config.durationHours * 60 * 60 * 1000;
+            // 记录试用时间到数据库
+            try {
+                if (!ctx.model) {
+                    console.error('=== 模型对象不存在 ===');
+                    return '数据库服务未就绪，请稍后再试。';
+                }
+                // 尝试创建记录，会自动创建表
+                console.log('=== 尝试创建或更新试用记录 ===');
+                const existingRecords = await ctx.model.get('trial_record', {
+                    userId: { $eq: session.userId },
+                    guildId: { $eq: session.guildId }
+                });
+                if (existingRecords.length > 0) {
+                    const record = existingRecords[0];
+                    const lastTrialTime = typeof record.lastTrialTime === 'number' ? record.lastTrialTime : parseInt(record.lastTrialTime);
+                    console.log('=== 用户最后试用时间 ===', lastTrialTime);
+                    if (now - lastTrialTime < cooldownMs) {
+                        const remainingTime = Math.ceil((cooldownMs - (now - lastTrialTime)) / (24 * 60 * 60 * 1000));
+                        console.log('=== 用户在冷却期内 ===', remainingTime);
+                        return `您已经使用过试用角色，${remainingTime}天后可再次使用。`;
+                    }
+                    // 更新记录
+                    console.log('=== 更新试用记录 ===', record.id);
+                    await ctx.model.set('trial_record', record.id, {
+                        lastTrialTime: now,
+                        updatedAt: new Date()
+                    });
+                }
+                else {
+                    // 创建新记录
+                    console.log('=== 创建试用记录 ===');
+                    await ctx.model.create('trial_record', {
+                        userId: session.userId,
+                        guildId: session.guildId,
+                        lastTrialTime: now,
+                        createdAt: new Date(),
+                        updatedAt: new Date()
+                    });
+                }
+                console.log('=== 数据库记录成功 ===');
+            }
+            catch (error) {
+                console.error('=== 数据库操作失败 ===', error);
+                return '数据库操作失败，请联系管理员。';
+            }
+            // 添加角色
+            console.log('=== 添加试用角色 ===', config.trialRoleId);
+            await session.bot.setGuildMemberRole(session.guildId, session.userId, config.trialRoleId);
+            // 设置定时器，到期后移除角色
+            setTimeout(async () => {
+                try {
+                    console.log('=== 移除试用角色 ===');
+                    await session.bot.setGuildMemberRole(session.guildId, session.userId, config.trialRoleId, false);
+                }
+                catch (error) {
+                    console.error('=== 移除角色失败 ===', error);
+                }
+            }, durationMs);
+            return config.successMessage;
+        }
+        catch (error) {
+            console.error('=== 试用指令执行失败 ===', error);
+            return config.errorMessage;
+        }
+    });
+    // 清除冷却指令
+    console.log('=== 注册清除冷却指令 ===');
+    ctx.command('清除冷却', '清除指定用户的试用冷却时间')
+        .usage('清除冷却 @用户')
+        .action(async ({ session }, user) => {
+        if (!session.guildId) {
+            return '请在服务器内使用此指令。';
+        }
+        if (session.bot.platform !== 'kook') {
+            return '此指令仅支持Kook平台。';
+        }
+        if (!user) {
+            return '请正确提及用户，格式为：清除冷却 @用户';
+        }
+        // 提取用户ID
+        let targetUserId;
+        const match = user.match(/id="([^"]+)"/);
+        if (match) {
+            targetUserId = match[1];
+        }
+        else {
+            return '请正确提及用户，格式为：清除冷却 @用户';
+        }
+        try {
+            console.log('=== 清除冷却指令执行 ===', targetUserId, session.guildId);
+            if (!ctx.model) {
+                console.error('=== 模型对象不存在 ===');
+                return '数据库服务未就绪，请稍后再试。';
+            }
+            // 清除冷却
+            const records = await ctx.model.get('trial_record', {
+                userId: { $eq: targetUserId },
+                guildId: { $eq: session.guildId }
+            });
+            console.log('=== 查询冷却记录 ===', records.length);
+            if (records.length > 0) {
+                const record = records[0];
+                console.log('=== 移除冷却记录 ===', record.id);
+                await ctx.model.remove('trial_record', record.id);
+                return `已成功清除用户 ${user} 的试用冷却时间。`;
+            }
+            else {
+                return `用户 ${user} 目前没有冷却时间。`;
+            }
+        }
+        catch (error) {
+            console.error('=== 清除冷却失败 ===', error);
+            return '清除冷却失败，请联系管理员。';
+        }
+    });
+    // 检查数据库状态的调试指令
+    console.log('=== 注册检查数据库指令 ===');
+    ctx.command('检查数据库', '检查试用角色数据库状态')
+        .action(async ({ session }) => {
+        try {
+            console.log('=== 检查数据库状态 ===');
+            if (!ctx.model) {
+                console.error('=== 模型对象不存在 ===');
+                return '数据库服务未就绪，请稍后再试。';
+            }
+            // 尝试查询记录，会自动创建表
+            try {
+                console.log('=== 查询记录数量 ===');
+                const records = await ctx.model.get('trial_record', {});
+                console.log('=== 记录数量 ===', records.length);
+                return `数据库表存在，有 ${records.length} 条试用记录。`;
+            }
+            catch (error) {
+                console.error('=== 表不存在 ===', error);
+                return '数据库表不存在，请联系管理员。';
+            }
+        }
+        catch (error) {
+            console.error('=== 检查数据库失败 ===', error);
+            return '查询数据库失败，请联系管理员。';
+        }
+    });
+    // 发放角色指令
+    console.log('=== 注册发放角色指令 ===');
+    ctx.command('发放角色', '为指定用户发放试用角色')
+        .usage('发放角色 @用户')
+        .action(async ({ session }, user) => {
+        if (!session.guildId) {
+            return '请在服务器内使用此指令。';
+        }
+        if (session.bot.platform !== 'kook') {
+            return '此指令仅支持Kook平台。';
+        }
+        if (!user) {
+            return '请正确提及用户，格式为：发放角色 @用户';
+        }
+        // 提取用户ID
+        let targetUserId;
+        const match = user.match(/id="([^"]+)"/);
+        if (match) {
+            targetUserId = match[1];
+        }
+        else {
+            return '请正确提及用户，格式为：发放角色 @用户';
+        }
+        try {
+            console.log('=== 发放角色指令执行 ===', targetUserId, session.guildId);
+            const durationMs = config.durationHours * 60 * 60 * 1000;
+            // 添加角色
+            console.log('=== 为用户发放试用角色 ===', targetUserId, config.trialRoleId);
+            await session.bot.setGuildMemberRole(session.guildId, targetUserId, config.trialRoleId);
+            // 设置定时器，到期后移除角色
+            setTimeout(async () => {
+                try {
+                    console.log('=== 移除试用角色 ===', targetUserId);
+                    await session.bot.setGuildMemberRole(session.guildId, targetUserId, config.trialRoleId, false);
+                }
+                catch (error) {
+                    console.error('=== 移除角色失败 ===', error);
+                }
+            }, durationMs);
+            return `已成功为用户 ${user} 发放试用角色！有效期为 ${config.durationHours} 小时。`;
+        }
+        catch (error) {
+            console.error('=== 发放角色失败 ===', error);
+            return '发放角色失败，请联系管理员。';
+        }
+    });
+    // 删除角色指令
+    console.log('=== 注册删除角色指令 ===');
+    ctx.command('删除角色', '删除指定用户的试用角色')
+        .usage('删除角色 @用户')
+        .action(async ({ session }, user) => {
+        if (!session.guildId) {
+            return '请在服务器内使用此指令。';
+        }
+        if (session.bot.platform !== 'kook') {
+            return '此指令仅支持Kook平台。';
+        }
+        if (!user) {
+            return '请正确提及用户，格式为：删除角色 @用户';
+        }
+        // 提取用户ID
+        let targetUserId;
+        const match = user.match(/id="([^"]+)"/);
+        if (match) {
+            targetUserId = match[1];
+        }
+        else {
+            return '请正确提及用户，格式为：删除角色 @用户';
+        }
+        try {
+            console.log('=== 删除角色指令执行 ===', targetUserId, session.guildId);
+            // 移除角色
+            console.log('=== 为用户删除试用角色 ===', targetUserId, config.trialRoleId);
+            await session.bot.setGuildMemberRole(session.guildId, targetUserId, config.trialRoleId, false);
+            return `已成功删除用户 ${user} 的试用角色。`;
+        }
+        catch (error) {
+            console.error('=== 删除角色失败 ===', error);
+            return '删除角色失败，请联系管理员。';
+        }
+    });
+    console.log('=== 试用角色插件加载完成 ===');
+}
+module.exports = { name: pluginName, Config, apply };

package/package.json

@@ -0,0 +1,16 @@
+{
+  "name": "@fanyue1117/koishi-plugin-trial-role",
+  "version": "1.0.0",
+  "description": "Kook机器人试用角色插件",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "author": "",
+  "license": "MIT",
+  "peerDependencies": {
+    "koishi": "^4.0.0"
+  }
+}