上一章介绍了eggjs+wechaty搭建一个简单的微信聊天机器人，只是实现了简单的回复，这一章我们将对接ERNIE-Bot接口，实现智能回复。
access_token的请求
config.default.js配置
我们在上一章代码基础上进行调整，首先我们要加入access_token的请求逻辑。第一章我们说过，请求需要使用应用对应的API Key，Secret Key作为入参。eggjs规定了配置的写法，我们可以将其配置在config/config.default.js里面。
//配置ernie
config.ernie = {
client_id: '填入您的API Key', //API Key
client_secret: '填入您的Secret Key',//Secret Key
access_token: '',//先置空，后续由程序填充
expire_day: 30 //access_token过期时长（天）
}ernie.js定义getAccessToken
我们在app/service目录增加ernie.js。
ernie.js文件专门用来和ERNIE-Bot接口进行交互，代码如下：
const {
Service
} = require('egg');
class ErnieService extends Service {
async getAccessToken() {
console.log('===================ErnieService getAccessToken=====================');
let ctx = this.ctx;
try {
const res = await ctx.curl(
`https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=${ctx.app.config.ernie.client_id}&client_secret=${ctx.app.config.ernie.client_secret}`, {
method: 'GET',
rejectUnauthorized: false,
data: {},
headers: {},
timeout: 30000,
contentType: 'json',
dataType: 'json',
})
console.log(res)
if (res.data.access_token) {
ctx.app.config.ernie.access_token = res.data.access_token;
console.log('access_token', ctx.app.config.ernie.access_token)
}
} catch (error) {
console.log(error)
}
}
}
module.exports = ErnieService;我们按eggjs的service的写法，定义了一个方法，用来请求access_token，但是这样写并不能执行，我们需要有个逻辑，在应用启动完毕后会调用这个方法。
app.js执行getAccessToken
我们在根目录增加app.js。
app.js代码如下：
module.exports = app => {
app.beforeStart(async () => {
// 应用会等待这个函数执行完成才启动
console.log("==app beforeStart==");
});
app.ready(async () => {
console.log("==app ready==");
let ctx = app.createAnonymousContext();
ctx.service.ernie.getAccessToken(); //请求AccessToken
})
app.beforeClose(async () => {
console.log("==app beforeClose==");
})
};我们执行命令npm run dev启动项目，看到最终会打印出access_token，有了access_token我们继续开发接口请求函数sendMsg。
对话请求
ernie.js定义sendMsg
sendMsg函数用于调用ERNIE-Bot会话接口，传入对话内容并返回对话结果。在class ErnieService中加入以下代码：
async sendMsg(msg) {
console.log('===================ErnieService sendMsg=====================');
let ctx = this.ctx;
try {
const res = await ctx.curl(
`https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions?access_token=${ctx.app.config.ernie.access_token}`, {
method: 'POST',
rejectUnauthorized: false,
data: {
"messages": msg
},
timeout: 30000,
contentType: 'json',
dataType: 'json',
})
console.log(res)
if (res.data) {
return res.data;
}
return null;
} catch (error) {
console.log(error)
return null;
}
}wechat.js
我们在wechat.js中，将原先的逻辑调整一下，主要是改两个地方，
1、我们在之前的代码中经常有看到ctx这个变量，可以通过他获取配置文件的参数值。但Service文件中的ctx的获取只能通过app.js调用Service中函数的方式来获得。所以我们将机器人初始化的操作定义在startBot()方法中，由app.js来执行这个方法，就可以接收到ctx的值。
2、原先回复是直接扔回用户发的信息的，现在将其对接到ernie.js中的sendMsg方法。
wechat.js全部代码如下：
const {
Service
} = require('egg');
const {
WechatyBuilder,
ScanStatus
} = require("wechaty");
const qrcode = require("qrcode-terminal");
let ctx;
let wechaty;
let startStatus = false;
const onMessage = async (message) => {
console.log(`收到消息: ${message}`);
if (message.type() === wechaty.Message.Type.Text) {
const userMsg = await message.text();
try {
let msgRecord = [{
"role": "user",
"content": userMsg
}];
let res = await ctx.service.ernie.sendMsg(msgRecord);
if (res) {
if (res.error_code) {
message.say(JSON.stringify(res));
console.log(`报错: ${JSON.stringify(res)}`);
} else {
if (res.result) {
message.say(res.result);
console.log(`回复: ${res.result}`);
}
}
}
} catch (error) {
console.log(error);
message.say(JSON.stringify(error));
}
}
};
const onLogout = (user) => {
console.log(`用户 ${user} 退出成功`);
};
const onLogin = async (user) => {
console.log(`用户 ${user} 登录成功`);
};
const onError = console.error;
const onScan = (code, status) => {
// status: 2代表链接等待调用，3代表链接已打开，这个链接实际上是提供一个登录的二维码供扫描
if (status === ScanStatus.Waiting) {
// status: 2代表等待，3代表扫码完成
qrcode.generate(code, {
small: true
}, console.log)
}
};
class WechatService extends Service {
async startBot() {
console.log('===================WechatService startBot=====================');
ctx = this.ctx;
if (startStatus && wechaty) {
if (wechaty.isLoggedIn) {
await wechaty.logout();
}
await wechaty.stop();
startStatus = false;
wechaty = null;
}
wechaty = await WechatyBuilder.build();
wechaty
.on("scan", onScan)
.on("login", onLogin)
.on("logout", onLogout)
.on("error", onError)
.on("message", onMessage);
await wechaty.start();
startStatus = true;
}
}
module.exports = WechatService;app.js执行startBot
在AccessToken请求后加入初始化BOT
app.ready(async () => {
console.log("==app ready==");
let ctx = app.createAnonymousContext();
await ctx.service.ernie.getAccessToken(); //请求AccessToken
await ctx.service.wechat.startBot(); //初始化BOT
})
至此，我们已经实现了一次会话的功能。多轮对话的功能等下一章再详述。
本章完整代码在这里下载。运行前请配置好config/config.default.js里面config.ernie下的client_id和client_secret配置项。
文章出处登录后可见！
