（附源码）nodejs微信机器人(可以自动回复)

使用js实现一个微信机器人，可以实现自动回复。扩展技能还有连接chatgpt或者你想接的语言模型、淘宝联盟或者其他平台的返利。

效果如下：

下面是机器人自动回复的开发步骤：

一、安装需要的包

npm install wechaty 或者 yarn add wechaty
npm install wechaty-puppet-wechat或者 yarn add wechaty-puppet-wechat
npm install openai或者 yarn add openai
npm install remark或者 yarn add remark
npm install strip-markdown或者 yarn add strip-markdown
npm install qrcode-terminal或者 yarn add qrcode-terminal


//作者使用的版本是
// "wechaty": "^1.20.2"
// "openai": "^3.1.0"
// "qrcode-terminal": "^0.12.0"
// "wechaty-puppet-wechat": "^1.18.4"
//"remark": "^14.0.2",
//"strip-markdown": "^5.0.0",

二、引入并使用

import { WechatyBuilder, ScanStatus, log } from 'wechaty'
import qrTerminal from 'qrcode-terminal'
import { defaultMessage, shardingMessage } from './sendMessage.js'
// 扫码
function onScan(qrcode, status) {
  if (status === ScanStatus.Waiting || status === ScanStatus.Timeout) {
    // 在控制台显示二维码
    qrTerminal.generate(qrcode, { small: true })
    const qrcodeImageUrl = ['https://api.qrserver.com/v1/create-qr-code/?data=', encodeURIComponent(qrcode)].join('')
    console.log('onScan:', qrcodeImageUrl, ScanStatus[status], status)
  } else {
    log.info('onScan: %s(%s)', ScanStatus[status], status)
  }
}

// 登录
function onLogin(user) {
  console.log(`${user} has logged in`)
  const date = new Date()
  console.log(`Current time:${date}`)
  console.log(`Automatic robot chat mode has been activated`)
}

// 登出
function onLogout(user) {
  console.log(`${user} has logged out`)
}

// 收到好友请求
async function onFriendShip(friendship) {
  const frienddShipRe = /chatgpt|chat/
  if (friendship.type() === 2) {
    if (frienddShipRe.test(friendship.hello())) {
      await friendship.accept()
    }
  }
}

/**
 * 消息发送
 * @param msg
 * @param isSharding
 * @returns {Promise<void>}
 */
async function onMessage(msg) {
  // 默认消息回复
  await defaultMessage(msg, bot)
  // 消息分片
  // await shardingMessage(msg,bot)
}

/**
*1、初始化机器人
*
*这里做了环境变量的处理、如果你的是本地运行、则不需要配置CHROME_BIN
*如果配置docker,则需要配置路劲CHROME_BIN 
**/
const CHROME_BIN = process.env.CHROME_BIN ? { endpoint: process.env.CHROME_BIN } : {}
export const bot = WechatyBuilder.build({
  name: 'WechatEveryDay',
  // puppet: 'wechaty-puppet-wechat4u', // 如果有token，记得更换对应的puppet
  puppet: 'wechaty-puppet-wechat', // 如果 wechaty-puppet-wechat 存在问题，也可以尝试使用上面的 wechaty-puppet-wechat4u ，记得安装 wechaty-puppet-wechat4u
  puppetOptions: {
    uos: true,
    ...CHROME_BIN
  },
})

// 扫码
bot.on('scan', onScan)
// 登录
bot.on('login', onLogin)
// 登出
bot.on('logout', onLogout)
// 收到消息
bot.on('message', onMessage)
// 添加好友
bot.on('friendship', onFriendShip)

// 启动微信机器人
bot
  .start()
  .then(() => console.log('Start to log in wechat...'))
  .catch((e) => console.error(e))

三、配置发送消息的逻辑

sendMessage.js文件

import { getOpenAiReply as getReply } from '../openai/index.js'
import { botName, roomWhiteList, aliasWhiteList } from '../../config.js'


/**
 * 默认消息发送
 * @param msg
 * @param bot
 * @returns {Promise<void>}
 */

export async function defaultMessage(msg, bot) {
  const contact = msg.talker() // 发消息人
  // console.log('contact', contact)
  const receiver = msg.to() // 消息接收人
  const content = msg.text() // 消息内容
  const room = msg.room() // 是否是群消息
  const roomName = (await room?.topic()) || null // 群名称
  const alias = (await contact.alias()) || (await contact.name()) // 发消息人昵称
  const remarkName = await contact.alias() // 备注名称
  const name = await contact.name() // 微信名称
  const isText = msg.type() === bot.Message.Type.Text // 消息类型是否为文本
  const isRoom = roomWhiteList.includes(roomName) && content.includes(`${botName}`) // 是否在群聊白名单内并且艾特了机器人
  const isAlias = aliasWhiteList.includes(remarkName) || aliasWhiteList.includes(name) // 发消息的人是否在联系人白名单内
  const isBotSelf = botName === remarkName || botName === name // 是否是机器人自己
  // TODO 你们可以根据自己的需求修改这里的逻辑
}


/**
 * 分片消息发送
 * @param message
 * @param bot
 * @returns {Promise<void>}
 */
export async function shardingMessage(message, bot) {
  const talker = message.talker()
  const isText = message.type() === bot.Message.Type.Text // 消息类型是否为文本
  if (talker.self() || message.type() > 10 || (talker.name() === '微信团队' && isText)) {
    return
  }
  const text = message.text()
  const room = message.room()
  if (!room) {
    console.log(`Chat GPT Enabled User: ${talker.name()}`)
    const response = await getChatGPTReply(text)
    await trySay(talker, response)
    return
  }
  let realText = splitMessage(text)
  // 如果是群聊但不是指定艾特人那么就不进行发送消息
  if (text.indexOf(`${botName}`) === -1) {
    return
  }
  realText = text.replace(`${botName}`, '')
  const topic = await room.topic()
  const response = await getChatGPTReply(realText)
  const result = `${realText}\n ---------------- \n ${response}`
  await trySay(room, result)
}

config.js做一些白名单处理

// 定义机器人的名称，这里是为了防止群聊消息太多，所以只有艾特机器人才会回复，
export const botName = '小旅人'

// 群聊白名单，白名单内的群聊才会自动回复
export const roomWhiteList = ['群名称']

// 联系人白名单，白名单内的联系人才会自动回复
export const aliasWhiteList = []

openai/index.js配置chatgpt

import { remark } from 'remark'
import stripMarkdown from 'strip-markdown'
import { Configuration, OpenAIApi } from 'openai'


const configuration = new Configuration({
  apiKey: 'OPENAI_API_KEY',//你的openaikey
})
const openai = new OpenAIApi(configuration)

export async function getOpenAiReply(prompt) {
const response = await openai.createChatCompletion(
      {
        model: 'gpt-3.5-turbo',
        messages: [
          { role: 'user', content: prompt },
        ],
      },
/*如果你本地报错可以尝试打开，proxy这里的代理是本地的，如果你是不是本地运行环境可以打开代理
 *     {
 *       proxy: {
 *         host: '127.0.0.1',
 *         port: 7890,
 *       },
 *     },*/
    )

    reply = markdownToText(response.data.choices[0].message.content)

}
function markdownToText(markdown) {
  return remark()
    .use(stripMarkdown)
    .processSync(markdown ?? '')
    .toString()
}

最后直接node你的第一个文件，就可以在控制台扫码登陆了

源码：点击下载