chatgpt-plugin/utils/slack/slackClient.js

import { Config } from '../config.js'
import slack from '@slack/bolt'
import delay from 'delay'
import { limitString } from '../common.js'
let proxy
if (Config.proxy) {
  try {
    proxy = (await import('https-proxy-agent')).default
  } catch (e) {
    console.warn('未安装https-proxy-agent，请在插件目录下执行pnpm add https-proxy-agent')
  }
}
export class SlackClaudeClient {
  constructor (props) {
    this.config = props
    if (Config.slackSigningSecret && Config.slackBotUserToken && Config.slackUserToken) {
      let option = {
        signingSecret: Config.slackSigningSecret,
        token: Config.slackBotUserToken,
        // socketMode: true,
        appToken: Config.slackUserToken
        // port: 45912
      }
      if (Config.proxy) {
        option.agent = proxy(Config.proxy)
      }
      option.logLevel = Config.debug ? 'debug': 'info'
      this.app = new slack.App(option)
    } else {
      throw new Error('未配置Slack信息')
    }
  }

  async sendMessage (prompt, e, t = 0) {
    if (t > 10) {
      return 'claude 未响应'
    }
    if (prompt.length > 3990) {
      logger.warn('消息长度大于slack限制，长度剪切至3990')
      prompt = limitString(prompt, 3990, false)
    }
    let channel
    let qq = e.sender.user_id
    if (Config.slackClaudeSpecifiedChannel) {
      channel = { id: Config.slackClaudeSpecifiedChannel }
    } else {
      let channels = await this.app.client.conversations.list({
        token: this.config.slackUserToken,
        types: 'public_channel,private_channel'
      })
      channel = channels.channels.filter(c => c.name === '' + qq)
      if (!channel || channel.length === 0) {
        let createChannelResponse = await this.app.client.conversations.create({
          token: this.config.slackUserToken,
          name: qq + '',
          is_private: true
        })
        channel = createChannelResponse.channel
        await this.app.client.conversations.invite({
          token: this.config.slackUserToken,
          channel: channel.id,
          users: Config.slackClaudeUserId
        })
        await delay(1000)
      } else {
        channel = channel[0]
      }
    }
    let conversationId = await redis.get(`CHATGPT:SLACK_CONVERSATION:${qq}`)
    if (!conversationId) {
      let sendResponse = await this.app.client.chat.postMessage({
        as_user: true,
        text: `<@${Config.slackClaudeUserId}> ${prompt}`,
        token: this.config.slackUserToken,
        channel: channel.id
      })
      let ts = sendResponse.ts
      let response = '_Typing…_'
      let tryTimes = 0
      // 发完先等3喵
      await delay(3000)
      while (response.trim().endsWith('_Typing…_')) {
        let replies = await this.app.client.conversations.replies({
          token: this.config.slackUserToken,
          channel: channel.id,
          limit: 1000,
          ts
        })
        await await redis.set(`CHATGPT:SLACK_CONVERSATION:${qq}`, `${ts}`)
        if (replies.messages.length > 0) {
          let formalMessages = replies.messages
            .filter(m => m.metadata?.event_type !== 'claude_moderation')
            .filter(m => !m.text.startsWith('_'))
          if (!formalMessages[formalMessages.length - 1].bot_profile) {
            // 问题的下一句不是bot回复的，这属于意料之外的问题，可能是多人同时问问题导致 再问一次吧
            return await this.sendMessage(prompt, e, t + 1)
          }
          let reply = formalMessages[formalMessages.length - 1]
          if (!reply.text.startsWith(`<@${Config.slackClaudeUserId}>`)) {
            response = reply.text
            if (Config.debug) {
              let text = response.replace('_Typing…_', '')
              if (text) {
                logger.info(response.replace('_Typing…_', ''))
              }
            }
          }
        }
        await delay(2000)
        tryTimes++
        if (tryTimes > 3 && response === '_Typing…_') {
          // 过了6秒还没任何回复，就重新发一下试试
          logger.warn('claude没有响应，重试中')
          return await this.sendMessage(prompt, e, t + 1)
        }
      }
      return response
    } else {
      let postResponse = await this.app.client.chat.postMessage({
        as_user: true,
        text: `<@${Config.slackClaudeUserId}> ${prompt}`,
        token: this.config.slackUserToken,
        channel: channel.id,
        thread_ts: conversationId
      })
      let postTs = postResponse.ts
      let response = '_Typing…_'
      let tryTimes = 0
      // 发完先等3喵
      await delay(3000)
      while (response.trim().endsWith('_Typing…_')) {
        let replies = await this.app.client.conversations.replies({
          token: this.config.slackUserToken,
          channel: channel.id,
          limit: 1000,
          ts: conversationId,
          oldest: postTs
        })

        if (replies.messages.length > 0) {
          let formalMessages = replies.messages
            .filter(m => m.metadata?.event_type !== 'claude_moderation')
            .filter(m => !m.text.startsWith('_'))
          if (!formalMessages[formalMessages.length - 1].bot_profile) {
            // 问题的下一句不是bot回复的，这属于意料之外的问题，可能是多人同时问问题导致 再问一次吧
            return await this.sendMessage(prompt, e, t + 1)
          }
          let reply = formalMessages[formalMessages.length - 1]
          if (!reply.text.startsWith(`<@${Config.slackClaudeUserId}>`)) {
            response = reply.text
            if (Config.debug) {
              let text = response.replace('_Typing…_', '')
              if (text) {
                logger.info(response.replace('_Typing…_', ''))
              }
            }
          }
        }
        await delay(2000)
        tryTimes++
        if (tryTimes > 3 && response === '_Typing…_') {
          // 过了6秒还没任何回复，就重新发一下试试
          logger.warn('claude没有响应，重试中')
          return await this.sendMessage(prompt, e, t + 1)
        }
      }
      return response
    }
  }
}