@waline/vercel 1.5.1 → 1.6.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@waline/vercel",
-  "version": "1.5.1",
+  "version": "1.6.0",
   "description": "vercel server for waline comment system",
   "repository": "https://github.com/walinejs/waline",
   "license": "MIT",

package/src/config/config.js

@@ -36,6 +36,7 @@ const {
   MAIL_TEMPLATE_ADMIN,
   QQ_TEMPLATE,
   TG_TEMPLATE,
+  WX_TEMPLATE,
 } = process.env;
 
 let storage = 'leancloud';
@@ -115,4 +116,5 @@ module.exports = {
   mailTemplateAdmin: MAIL_TEMPLATE_ADMIN,
   QQTemplate: QQ_TEMPLATE,
   TGTemplate: TG_TEMPLATE,
+  WXTemplate: WX_TEMPLATE,
 };

package/src/service/notify.js

@@ -91,6 +91,81 @@ module.exports = class extends think.Service {
     });
   }
 
+  async qywxAmWechat({ title, content }, self, parent) {
+    const { QYWX_AM, SITE_NAME, SITE_URL } = process.env;
+    if (!QYWX_AM) {
+      return false;
+    }
+
+    const QYWX_AM_AY = QYWX_AM.split(',');
+    const comment = self.comment
+      .replace(/<a href="(.*?)">(.*?)<\/a>/g, '\n[$2] $1\n')
+      .replace(/<[^>]+>/g, '');
+    const postName = self.url;
+
+    const data = {
+      self: {
+        ...self,
+        comment,
+      },
+      postName,
+      parent,
+      site: {
+        name: SITE_NAME,
+        url: SITE_URL,
+        postUrl: SITE_URL + self.url + '#' + self.objectId,
+      },
+    };
+    const contentWechat =
+      think.config('WXTemplate') ||
+      `💬 {{site.name|safe}}的文章《{{postName}}》有新评论啦 
+【评论者昵称】：{{self.nick}}
+【评论者邮箱】：{{self.mail}} 
+【内容】：{{self.comment}} 
+<a href='{{site.postUrl}}'>查看详情</a>`;
+
+    title = nunjucks.renderString(title, data);
+    const desp = nunjucks.renderString(contentWechat, data);
+    content = desp.replace(/\n/g, '<br/>');
+
+    const { access_token } = await request({
+      uri: `https://qyapi.weixin.qq.com/cgi-bin/gettoken`,
+      qs: {
+        corpid: `${QYWX_AM_AY[0]}`,
+        corpsecret: `${QYWX_AM_AY[1]}`,
+      },
+      headers: {
+        'Content-Type': 'application/json',
+      },
+      json: true,
+    });
+    return request({
+      url: `https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=${access_token}`,
+      body: {
+        touser: `${QYWX_AM_AY[2]}`,
+        agentid: `${QYWX_AM_AY[3]}`,
+        msgtype: 'mpnews',
+        mpnews: {
+          articles: [
+            {
+              title,
+              thumb_media_id: `${QYWX_AM_AY[4]}`,
+              author: `Waline Comment`,
+              content_source_url: `${data.site.postUrl}`,
+              content: `${content}`,
+              digest: `${desp}`,
+            },
+          ],
+        },
+      },
+      method: 'POST',
+      json: true,
+      headers: {
+        'Content-Type': 'application/json',
+      },
+    });
+  }
+
   async qq(self, parent) {
     const { QMSG_KEY, QQ_ID, SITE_NAME, SITE_URL } = process.env;
     if (!QMSG_KEY) {
@@ -230,12 +305,18 @@ module.exports = class extends think.Service {
 
     if (!isAuthorComment && !disableAuthorNotify) {
       const wechat = await this.wechat({ title, content }, comment, parent);
+      const qywxAmWechat = await this.qywxAmWechat(
+        { title, content },
+        comment,
+        parent
+      );
       const qq = await this.qq(comment, parent);
       const telegram = await this.telegram(comment, parent);
       if (
         think.isEmpty(wechat) &&
         think.isEmpty(qq) &&
         think.isEmpty(telegram) &&
+        think.isEmpty(qywxAmWechat) &&
         !isReplyAuthor
       ) {
         mailList.push({ to: AUTHOR, title, content });