十分钟实现一个智能问答微信聊天机器人

最近开始写自己的公众号《前端布朗熊》了，注意到有的的公众号有微信机器人可以实现拉人进群、踢人等自动化操作，也想给自己的公众号开发一个机器人来帮我处理一些自动化的重复任务。因此就对微信机器人技术进行了调研，发现了以下几个情况：

1. 微信禁止了新用户和一部分老用户的网页微信登录权限
2. 市面上现有的微信机器人sdk大部分是基于网页微信的http协议实现的
3. wechaty是一个可以支持非http协议的Chat Bot SDK，并且是一个使用Typescript 构建的Node.js 应用，可以使用iPad协议实现手机和服务端同时登录微信，并且最重要的即使是新注册的微信号，也可以使用iPad协议登录微信

我本人手头上的常用微信号需要登录Mac微信工作使用，因此不能拿来做机器人，而另一个微信小号也没有登录网页微信的权限，因此wechaty就成了我的首选

准备工作

• 获取token：wechaty使用iPad协议时需要token，这个token可以购买也可以通过成为项目贡献者来得到一个长期可用的token。对于第一次接触wechaty的开发者，可以申请一个为期15天的token来试用，然后15天之内向官方提交你的wechaty成果，待官方审核通过之后就可以获得一个长期token
• 注册天行API（购买一个月的基础套餐）：我们要让机器人能够实现执行一些简单的交互，就必须需要数据的支持。比如，想让机器人帮我们查询天气，那么就需要有天气数据供我们使用；那么这个数据可以使用公开的免费API、自己写爬虫抓取等方式获得，我这边提供的方法是使用现成的天行API；价格不贵而且除了天气还有笑话、故事等等API的使用权限。

写代码

1. 登录
2. 接收好友请求
3. 接收文字消息并调用天行机器人接口回复

代码中用到的puppetPadplusToken和tianxinApiKey分别是wechaty的token和天行接口的api key需要自己申请，然后按照项目中的src/config-sample.ts的格式创建src/config.ts并填写wechaty的token和天行接口的api key

import { Contact, Message, Wechaty } from "wechaty";
import { ScanStatus } from "wechaty-puppet";
import { PuppetPadplus } from "wechaty-puppet-padplus";
import QrcodeTerminal from "qrcode-terminal";
import { puppetPadplusToken, tianxinApiKey } from "./config";
import { TXRobot } from "./cmds/tianxin";
import { error_text, hello_text } from "./common/constant";
import querystring from "querystring";
import axios from "axios";

const puppet = new PuppetPadplus({
    token: puppetPadplusToken,
});

const name = "BrownWeChatBot";

const bot = new Wechaty({
    puppet,
    name, // generate xxxx.memory-card.json and save login data for the next login
});

bot.on("scan", (qrcode, status) => {
    // 扫码
    if (status === ScanStatus.Waiting) {
        QrcodeTerminal.generate(qrcode, {
            small: true,
        });
    }
})
    .on("login", (user: Contact) => {
        // 登录成功
        console.log(`login success, user: ${user}`);
    })
    .on("friendship", async (friendship) => {
        // 接收到好友请求时，直接同意并说一句配置好的欢迎语
        try {
            switch (friendship.type()) {
                // 1. New Friend Request
                case bot.Friendship.Type.Receive:
                    await friendship.accept();
                    friendship.contact().say(hello_text);
                    break;

                // 2. Friend Ship Confirmed
                case bot.Friendship.Type.Confirm:
                    break;
            }
        } catch (e) {
            console.error(e);
        }
    })
    .on("message", (msg: Message) => {
        // 接收到文字消息时转给天行机器人，调用天行机器人接口并回复结果
        switch (msg.type()) {
            case Message.Type.Text: {
                TXRobot(msg.text(), msg.from()?.id)
                    .then((replay) => {
                        msg.from()?.say(replay);
                    })
                    .catch((e) => {
                        console.log("catch err", e);
                        msg.from()?.say(error_text);
                    });
            }
        }
    })
    .on("logout", (user: Contact, reason?: string) => {
        console.log(`logout user: ${user}, reason : ${reason}`);
    })
    .start()
    .catch((e) => {
        console.log(e);
    });

/**
 * 天行机器人
 * @param {string} question 提问
 * @param {number} [restype=0] 输入类型，文本0[默认]、语音1、人脸2、其他3
 * @param {number} [datatype=0] 返回类型，文本0[默认]、语音1
 * @returns
 */
const TXRobot = (
    question: string,
    userid?: string,
    restype?: number,
    datatype?: number
): Promise<string> => {
    // 使用axios请求天行机器人的接口就可以
    return axios({
        method: "post",
        url: "http://api.tianapi.com/txapi/robot/index",
        headers: {
            "content-type": "application/x-www-form-urlencoded",
        },
        data: querystring.stringify({
            key: tianxinApiKey,
            question,
            userid: restype ?? "",
            restype: restype ?? 0,
            datatype: datatype ?? 0,
            voc: 3,
        }),
    }).then((res: any) => {
        if (res?.data?.code === 200) {
            switch (res.data.datatype) {
                case "text": {
                    return res.data.newslist?.[0]?.reply ?? error_text;
                }
            }
        } else {
            return Promise.reject(res?.data?.msg ?? error_text);
        }
    });
};

项目源码

完整代码我已经提交到github上了，大家只需要准备好wechaty的token和天行的api key，然后再代码目录下执行yarn && yarn start就可以运行了。对了，需要提前安装好node环境和yarn

实现效果

后续

以上只是一个mvp的机器人，后期我会完善自己的机器人，实现一些可以帮我提高效率的功能，后面的代码我都会把代码同步到这个仓库，保证所有的代码开源。