fix: 闭嘴功能

apps/chat.js

@@ -412,6 +412,11 @@ export class chatgpt extends plugin {
         return false
       }
     }
+    let groupId = e.isGroup ? e.group.group_id : ''
+    if (await redis.get('CHATGPT:SHUT_UP:ALL') || await redis.get(`CHATGPT:SHUT_UP:${groupId}`)) {
+      logger.info('chatgpt闭嘴中，不予理会')
+      return false
+    }
     let userSetting = await redis.get(`CHATGPT:USER:${e.sender.user_id}`)
     if (userSetting) {
       userSetting = JSON.parse(userSetting)

apps/help.js

@@ -61,6 +61,21 @@ let helpData = [
         title: '#chatgpt对话列表',
         desc: '查询当前哪些人正在与机器人聊天.目前API3模式下支持切换对话'
       },
+      {
+        icon: 'destroy',
+        title: '#chatgpt(本群)?(群xxx)?闭嘴(x秒/分钟/小时)',
+        desc: '让机器人在本群/某群闭嘴。不指定群时认为全局闭嘴。'
+      },
+      {
+        icon: 'chat',
+        title: '#chatgpt(本群)?(群xxx)?(张嘴|开口|说话|上班)',
+        desc: '让机器人在本群/某群重新可以说话。不指定群时认为全局开口。'
+      },
+      {
+        icon: 'chat',
+        title: '#chatgpt查看闭嘴',
+        desc: '查看当前闭嘴情况。'
+      },
       {
         icon: 'switch',
         title: '#chatgpt切换对话+对话id',

apps/management.js

@@ -2,7 +2,7 @@ import plugin from '../../../lib/plugins/plugin.js'
 import { Config } from '../utils/config.js'
 import { BingAIClient } from '@waylaidwanderer/chatgpt-api'
 import { exec } from 'child_process'
-import { checkPnpm } from '../utils/common.js'
+import { checkPnpm, formatDuration, parseDuration } from '../utils/common.js'
 
 export class ChatgptManagement extends plugin {
   constructor (e) {
@@ -64,6 +64,21 @@ export class ChatgptManagement extends plugin {
         {
           reg: '^#chatgpt(强制)?更新$',
           fnc: 'updateChatGPTPlugin'
+        },
+        {
+          reg: '^#chatgpt(本群)?(群\\d+)?闭嘴',
+          fnc: 'shutUp',
+          permission: 'master'
+        },
+        {
+          reg: '^#chatgpt(本群)?(群\\d+)?(张嘴|开口|说话|上班)',
+          fnc: 'openMouth',
+          permission: 'master'
+        },
+        {
+          reg: '^#chatgpt查看闭嘴',
+          fnc: 'listShutUp',
+          permission: 'master'
         }
       ]
     })
@@ -280,4 +295,119 @@ export class ChatgptManagement extends plugin {
 `
     await this.reply(message)
   }
+
+  async shutUp (e) {
+    let duration = e.msg.replace(/^#chatgpt(本群)?(群\d+)?闭嘴/, '')
+    let scope
+    let time = 3600000
+    if (duration === '永久') {
+      time = 0
+    } else if (duration) {
+      time = parseDuration(duration)
+    }
+    const match = e.msg.match(/#chatgpt群(\d+)闭嘴(.*)/)
+    if (e.msg.indexOf('本群') > -1) {
+      if (e.isGroup) {
+        scope = e.group.group_id
+        if (await redis.get(`CHATGPT:SHUT_UP:${scope}`)) {
+          await redis.del(`CHATGPT:SHUT_UP:${scope}`)
+          await redis.set(`CHATGPT:SHUT_UP:${scope}`, '1', { EX: time })
+          await e.reply(`好的，从现在开始我会在当前群聊闭嘴${formatDuration(time)}`)
+        } else {
+          await redis.set(`CHATGPT:SHUT_UP:${scope}`, '1', { EX: time })
+          await e.reply(`好的，从现在开始我会在当前群聊闭嘴${formatDuration(time)}`)
+        }
+      } else {
+        await e.reply('本群是指？你也没在群聊里让我闭嘴啊？')
+        return false
+      }
+    } else if (match) {
+      const groupId = parseInt(match[1], 10)
+      if (Bot.getGroupList().get(groupId)) {
+        if (await redis.get(`CHATGPT:SHUT_UP:${groupId}`)) {
+          await redis.del(`CHATGPT:SHUT_UP:${groupId}`)
+          await redis.set(`CHATGPT:SHUT_UP:${groupId}`, '1', { EX: time })
+          await e.reply(`好的，从现在开始我会在群聊${groupId}闭嘴${formatDuration(time)}`)
+        } else {
+          await redis.set(`CHATGPT:SHUT_UP:${groupId}`, '1', { EX: time })
+          await e.reply(`好的，从现在开始我会在群聊${groupId}闭嘴${formatDuration(time)}`)
+        }
+      } else {
+        await e.reply('这是什么群？')
+        return false
+      }
+    } else {
+      if (await redis.get('CHATGPT:SHUT_UP:ALL')) {
+        await redis.del('CHATGPT:SHUT_UP:ALL')
+        await redis.set('CHATGPT:SHUT_UP:ALL', '1', { EX: time })
+        await e.reply(`好的，我会再闭嘴${formatDuration(time)}`)
+      } else {
+        await redis.set('CHATGPT:SHUT_UP:ALL', '1', { EX: time })
+        await e.reply(`好的，我会闭嘴${formatDuration(time)}`)
+      }
+    }
+  }
+
+  async openMouth (e) {
+    const match = e.msg.match(/^#chatgpt群(\d+)/)
+    if (e.msg.indexOf('本群') > -1) {
+      if (await redis.get('CHATGPT:SHUT_UP:ALL')) {
+        await e.reply('主人，我现在全局闭嘴呢，你让我在这个群张嘴咱也不敢张啊')
+        return false
+      }
+      if (e.isGroup) {
+        let scope = e.group.group_id
+        if (await redis.get(`CHATGPT:SHUT_UP:${scope}`)) {
+          await redis.del(`CHATGPT:SHUT_UP:${scope}`)
+          await e.reply('好的主人，我终于又可以在本群说话了')
+        } else {
+          await e.reply('啊？我也没闭嘴啊？')
+        }
+      } else {
+        await e.reply('本群是指？你也没在群聊里让我张嘴啊？')
+        return false
+      }
+    } else if (match) {
+      if (await redis.get('CHATGPT:SHUT_UP:ALL')) {
+        await e.reply('主人，我现在全局闭嘴呢，你让我在那个群张嘴咱也不敢张啊')
+        return false
+      }
+      const groupId = parseInt(match[1], 10)
+      if (Bot.getGroupList().get(groupId)) {
+        if (await redis.get(`CHATGPT:SHUT_UP:${groupId}`)) {
+          await redis.del(`CHATGPT:SHUT_UP:${groupId}`)
+          await e.reply(`好的主人，我终于又可以在群${groupId}说话了`)
+        } else {
+          await e.reply(`啊？我也没在群${groupId}闭嘴啊？`)
+        }
+      } else {
+        await e.reply('这是什么群？')
+        return false
+      }
+    } else {
+      if (await redis.get('CHATGPT:SHUT_UP:ALL')) {
+        await redis.del('CHATGPT:SHUT_UP:ALL')
+        await e.reply('好的，我会结束全局闭嘴')
+      } else {
+        await e.reply('啊？我也没全局闭嘴啊？')
+      }
+    }
+  }
+
+  async listShutUp () {
+    let keys = await redis.keys('CHATGPT:SHUT_UP:*')
+    if (!keys || keys.length === 0) {
+      await this.reply('我没有在任何群闭嘴', true)
+    } else {
+      let list = []
+      for (let i = 0; i < keys.length; i++) {
+        let key = keys[i]
+        let groupId = key.replace('CHATGPT:SHUT_UP:', '')
+        let ttl = await redis.ttl(key)
+        let ttlFormat = formatDuration(ttl)
+        list.push({ groupId, ttlFormat })
+      }
+      await this.reply(list.map(item => item.groupId !== 'ALL' ? `群聊${item.groupId}: ${item.ttlFormat}` : `全局: ${item.ttlFormat}`).join('\n'))
+    }
+  }
 }

utils/common.js

@@ -5,7 +5,7 @@ import lodash from 'lodash'
 import fs from 'node:fs'
 import path from 'node:path'
 import puppeteer from '../../../lib/puppeteer/puppeteer.js'
-import {Config} from "./config.js";
+import { Config } from './config.js'
 // export function markdownToText (markdown) {
 //  return remark()
 //    .use(stripMarkdown)
@@ -296,3 +296,58 @@ export function getDefaultUserSetting () {
     ttsRole: Config.defaultTTSRole
   }
 }
+
+export function parseDuration (duration) {
+  const timeMap = {
+    秒: 1000,
+    分: 60 * 1000,
+    小时: 60 * 60 * 1000
+  }
+
+  // 去掉多余的空格并将单位转化为小写字母
+  duration = duration.trim().toLowerCase()
+
+  // 去掉末尾的 "钟" 字符
+  if (duration.endsWith('钟')) {
+    duration = duration.slice(0, -1)
+  }
+
+  // 提取数字和单位
+  const match = duration.match(/^(\d+)\s*([\u4e00-\u9fa5]+)$/)
+
+  if (!match) {
+    throw new Error('Invalid duration string: ' + duration)
+  }
+
+  const num = parseInt(match[1], 10)
+  const unit = match[2]
+
+  if (!(unit in timeMap)) {
+    throw new Error('Unknown time unit: ' + unit)
+  }
+
+  return num * timeMap[unit]
+}
+
+export function formatDuration (duration) {
+  const timeMap = {
+    小时: 60 * 60 * 1000,
+    分钟: 60 * 1000,
+    秒钟: 1000
+  }
+
+  const units = Object.keys(timeMap)
+  let result = ''
+
+  for (let i = 0; i < units.length; i++) {
+    const unit = units[i]
+    const value = Math.floor(duration / timeMap[unit])
+
+    if (value > 0) {
+      result += value + unit
+      duration -= value * timeMap[unit]
+    }
+  }
+
+  return result || '0秒钟'
+}

utils/config.js

@@ -46,7 +46,7 @@ const defaultConfig = {
   noiseScaleW: 0.668,
   lengthScale: 1.2,
   initiativeChatGroups: [],
-  version: 'v2.0.19'
+  version: 'v2.0.20'
 }
 const _path = process.cwd()
 let config = {}