用 JavaScript 开发企业微信 ChatGPT 应用（含全部源码，免费托管，手把手教程）

1// @see https://docs.aircode.io/guide/functions/
2const xml2js = require('xml2js');
3const axios = require('axios');
4
5const { Configuration, OpenAIApi } = require("openai");
6const { decrypt } = require('@wecom/crypto');
7
8const { EncodingAESKey, OpenAIKey, CorpSecret, CorpId } = process.env;
9
10// 转换 xml 格式数据
11const parseXML = (data) => {
12
13  const parser = new xml2js.Parser();
14
15  return parser.parseStringPromise(data).then(function (result) {
16
17    console.dir(result);
18    return result;
19  })
20  .catch(function (err) {
21    console.error(err);
22  });    
23
24};
25
26// 从 buffer 格式数据获取消息内容
27const getMessageFromBuffer = async (buf) => {
28
29  try {
30
31    // 将 buffer 数据转换成 string 格式数据
32    const xmlData = buf.toString('utf8');
33    const result = await parseXML(xmlData);
34
35    const { xml } = result;
36    const { Encrypt } = xml;
37    // 解密数据
38    const data = decrypt(EncodingAESKey, Encrypt[0]);
39    const { message } = data;
40
41    const content = await parseXML(message);
42
43    console.log('转换 xml 格式数据: ', content.xml);
44    return content.xml;
45  
46  } catch (error) {
47    console.error('获取消息内容报错', error);
48    return {
49      message: `获取消息内容报错: ${error.message}`,
50    }
51  }
52}
53
54
55// 获取 access token https://developer.work.weixin.qq.com/document/path/91039
56const getAccessToken = async () => {
57
58  try {
59    const { data } = await axios(`https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=${CorpId}&corpsecret=${CorpSecret}`);
60    console.log('access token', data);
61    const { errcode, access_token, errmsg } = data;
62
63    if (errcode === 0 && errmsg === 'ok') {
64      return access_token;
65    } else {
66      console.error(`获取 access token 报错:: ${errmsg}`);
67      return {
68        code: errcode,
69        message: `获取 access token 报错: ${errmsg}`,
70      }
71    }
72    
73  } catch (error) {
74    console.log('error', error);
75    return {
76      message: `获取 access token 未知报错: ${error.message}`,
77    }
78  }
79  
80}
81
82// 回复信息 https://developer.work.weixin.qq.com/document/path/90236
83const sendMessage = async (accessToken, { touser, msgtype, agentid, text }) => {
84
85  try {
86    const { data } = await axios({
87      method: 'post',
88      url: `https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=${accessToken}`,
89      data: {
90        touser,
91        msgtype,
92        agentid,
93        text,
94      },
95    });
96    const { errcode, errmsg } = data;
97    console.log('回复消息数据: ', data);
98    if (errcode === 0 && errmsg === 'ok') {
99      console.log('成功回复消息');
100    } else {
101      console.error(`回复消息报错: ${errmsg}`);
102      return {
103        code: errcode,
104        message: `回复消息报错: ${errmsg}`,
105      }
106    }
107    
108  } catch (error) {
109    return {
110      message: `回复消息未知报错: ${error.message}`,
111    }
112  }
113}
114
115// 使用 openAI 接口调用 ChatGPT
116const getOpenAIChatCompletion = async (question) => {
117
118  try {
119
120    const configuration = new Configuration({
121      apiKey: OpenAIKey,
122    });
123
124    const openai = new OpenAIApi(configuration);
125
126    const completion = await openai.createChatCompletion({
127      // 使用当前 OpenAI 开放的最新 3.5 模型，如果后续 4 发布，则修改此处参数即可
128      // OpenAI models 参数列表 https://platform.openai.com/docs/models
129      model: "gpt-3.5-turbo",
130      messages: [{ role: "assistant", content: question }],
131    });
132
133    console.log('ChatGPT completion 数据:', completion.data.choices[0].message.content.trim());
134    return completion.data.choices[0].message.content.trim();
135
136  } catch(error) {
137  
138    console.error(`OpenAI 接口报错: ${error.message}`);
139    return {
140      code: 1,
141      message: `OpenAI 接口获取信息报错: ${error.message}`
142    }
143  }
144}
145
146module.exports = {
147  getAccessToken,
148  sendMessage,
149  getMessageFromBuffer,
150  getOpenAIChatCompletion,
151}
152

1// @see https://docs.aircode.io/guide/functions/
2const aircode = require('aircode');
3const { decrypt, getSignature } = require('@wecom/crypto');
4
5const { getAccessToken, sendMessage, getOpenAIChatCompletion, getMessageFromBuffer } = require('./api.js');
6
7let preMessageId = '';
8
9module.exports = async function(params, context) {
10
11  const { method } = context;
12  const { Token, OpenAIKey, EncodingAESKey } = process.env;
13
14  // 检查环境变量是否成功设置
15  const envVars = {
16    CorpId: '请检查你的环境变量 CorpId 值, 请在 https://work.weixin.qq.com/wework_admin/frame#profile/enterprise 获取到企业ID',
17    Token: '请检查你的环境变量 Token 值，请在后台应用详情页接收消息模块配置生成',
18    EncodingAESKey: '请检查你的环境变量 EncodingAESKey 值，请在后台应用详情页接收消息模块配置生成',
19    CorpSecret: '请检查你的环境变量 CorpSecret 值, 请在 https://work.weixin.qq.com/wework_admin/frame#apps) 获取到应用 Secret',
20  };
21
22  for (const [envVar, errorMessage] of Object.entries(envVars)) {
23    if (!process.env[envVar]) {
24      console.error(errorMessage);
25      return {
26        code: 1,
27        message: errorMessage,
28      };
29    }
30  };
31
32  if (method === 'GET') {
33
34    let { msg_signature, timestamp, nonce, echostr } = params;
35  
36    // 企业微信后台接收消息服务器地址配置验证
37    if (echostr) {
38  
39      try {
40
41        [msg_signature, timestamp, nonce, echostr] = [msg_signature, timestamp, nonce, echostr].map(decodeURIComponent);
42  
43        const signature = getSignature(Token, timestamp, nonce, echostr);
44
45        if (signature !== msg_signature) {
46          return {
47            code: 1,
48            message: `接收消息服务器地址配置验证错误: ${signature} !== ${msg_signature}`,
49          }
50        }
51  
52        const msg = decrypt(EncodingAESKey, echostr);
53        return msg['message'];
54      
55      } catch (error) {
56        console.error('接收消息服务器地址配置验证错误: ', error.message);
57        return {
58          code: 1,
59          message: `接收消息服务器地址配置验证报错: ${error.message}`,
60        }
61      }
62    }
63  }
64
65  // 回复用户信息
66  if (method === 'POST') {
67
68    const isBuffer = Buffer.isBuffer(params);
69
70    if (isBuffer) {
71
72      const message = await getMessageFromBuffer(params);
73
74      if (message) {
75        const { FromUserName, MsgType, AgentID, Content, MsgId } = message;
76
77        // 企业微信会重复发送信息，这里如果 message ID 相同则不再处理回复
78        const msgId = MsgId[0];
79        if (msgId === preMessageId) {
80          return
81        }
82
83        preMessageId = msgId;
84        
85        const accessToken = await getAccessToken();
86        let messageContent = Content[0];
87
88        if (OpenAIKey) {
89          messageContent = await getOpenAIChatCompletion(messageContent);
90        }
91
92        if (!messageContent) {
93          return {
94            code: 1,
95            message: '获取 ChatGPT 回答失败',
96          };
97        }
98    
99        const replyMessage = {
100          touser: FromUserName[0],
101          msgtype: MsgType[0],
102          agentid: AgentID[0],
103          text: {
104            content: messageContent
105          },
106        }
107
108        const res = await sendMessage(accessToken, replyMessage);
109        return {
110          code: 0,
111          message: '成功回复用户信息'
112        }
113      }
114    } else {
115      return {
116        code: 1,
117        message: `请跟随教程 https://aircode.cool/54fhemjpk2 进行配置，在企业微信应用聊天框输入信息调试代码, 或通过 Debug 功能使用 Buffer 类型数据进行调试`,
118      }
119    }
120  }
121};