@waline/vercel 1.17.0 → 1.17.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@waline/vercel",
-  "version": "1.17.0",
+  "version": "1.17.3",
   "description": "vercel server for waline comment system",
   "keywords": [
     "waline",
@@ -31,7 +31,7 @@
     "markdown-it-sub": "^1.0.0",
     "markdown-it-sup": "^1.0.0",
     "mathjax-full": "^3.2.0",
-    "nodemailer": "^6.7.4",
+    "nodemailer": "^6.7.5",
     "nunjucks": "^3.2.3",
     "phpass": "^0.1.1",
     "prismjs": "^1.28.0",

package/src/controller/comment.js

@@ -365,7 +365,7 @@ module.exports = class extends BaseRest {
           });
           comments.forEach((cmt) => {
             const countItem = (counts || []).find(({ mail, user_id }) => {
-              if (user_id) {
+              if (cmt.user_id) {
                 return user_id === cmt.user_id;
               }
               return mail === cmt.mail;

package/src/controller/token.js

@@ -47,13 +47,18 @@ module.exports = class extends BaseRest {
       }
     }
 
-    let avatar = user[0].avatar;
-    if (/(github)/i.test(avatar)) {
-      avatar =
-        this.config('avatarProxy') + '?url=' + encodeURIComponent(avatar);
+    let avatarUrl = user[0].avatar
+      ? user[0].avatar
+      : await think.service('avatar').stringify({
+          mail: user[0].email,
+          nick: user[0].display_name,
+          link: user[0].url,
+        });
+    const { avatarProxy } = think.config();
+    if (avatarProxy) {
+      avatarUrl = avatarProxy + '?url=' + encodeURIComponent(avatarUrl);
     }
-    user[0].avatar = avatar;
-
+    user[0].avatar = avatarUrl;
     return this.success({
       ...user[0],
       password: null,

package/src/controller/user.js

@@ -18,7 +18,7 @@ module.exports = class extends BaseRest {
     const users = await this.modelInstance.select(
       {},
       {
-        desc: 'insertedAt',
+        desc: 'createdAt',
         limit: pageSize,
         offset: Math.max((page - 1) * pageSize, 0),
       }

package/src/extend/think.js

@@ -24,6 +24,10 @@ module.exports = {
   },
   promiseAllQueue(promises, taskNum) {
     return new Promise((resolve, reject) => {
+      if (!promises.length) {
+        return resolve();
+      }
+
       const ret = [];
       let index = 0;
       let count = 0;
@@ -55,7 +59,11 @@ module.exports = {
 
     try {
       const search = helper.promisify(regionSearch.btreeSearch, regionSearch);
-      const { region } = await search(ip);
+      const result = await search(ip);
+      if (!result) {
+        return '';
+      }
+      const { region } = result;
       const [, , province, city, isp] = region.split('|');
       const address = Array.from(new Set([province, city, isp]));
       return address.slice(0, depth).join(' ');

package/src/logic/base.js

@@ -65,6 +65,7 @@ module.exports = class extends think.Logic {
           'qq',
           'avatar',
           '2fa',
+          'label',
         ],
       }
     );

package/src/service/storage/github.js

@@ -30,6 +30,7 @@ const CSV_HEADERS = {
     'type',
     'url',
     'avatar',
+    'label',
     'github',
     'twitter',
     'facebook',

package/src/service/storage/leancloud.js

@@ -126,6 +126,95 @@ module.exports = class extends Base {
     return data;
   }
 
+  async _getCmtGroupByMailUserIdCache(key, where) {
+    if (this.tableName !== 'Comment' || key !== 'user_id_mail') {
+      return [];
+    }
+
+    const cacheTableName = `cache_group_count_${key}`;
+    const currentTableName = this.tableName;
+    this.tableName = cacheTableName;
+    const cacheData = await this.select({ _complex: where._complex });
+    this.tableName = currentTableName;
+    return cacheData;
+  }
+
+  async _setCmtGroupByMailUserIdCache(key, data) {
+    if (this.tableName !== 'Comment' || key !== 'user_id_mail') {
+      return;
+    }
+
+    const cacheTableName = `cache_group_count_${key}`;
+    const currentTableName = this.tableName;
+    this.tableName = cacheTableName;
+
+    await think.promiseAllQueue(
+      data.map((item) => {
+        if (item.user_id && !think.isString(item.user_id)) {
+          item.user_id = item.user_id.toString();
+        }
+        return this.add(item);
+      }),
+      1
+    );
+    this.tableName = currentTableName;
+  }
+
+  async _updateCmtGroupByMailUserIdCache(data, method) {
+    if (this.tableName !== 'Comment') {
+      return;
+    }
+
+    if (!data.user_id && !data.mail) {
+      return;
+    }
+
+    const cacheTableName = `cache_group_count_user_id_mail`;
+    const cacheData = await this.select({
+      _complex: {
+        _logic: 'or',
+        user_id: think.isObject(data.user_id)
+          ? data.user_id.toString()
+          : data.user_id,
+        mail: data.mail,
+      },
+    });
+    if (think.isEmpty(data)) {
+      return;
+    }
+
+    let count = cacheData[0].count;
+    switch (method) {
+      case 'add':
+        if (data.status === 'approved') {
+          count += 1;
+        }
+        break;
+      case 'udpate_status':
+        if (data.status === 'approved') {
+          count += 1;
+        } else {
+          count -= 1;
+        }
+        break;
+      case 'delete':
+        count -= 1;
+        break;
+    }
+
+    const currentTableName = this.tableName;
+    this.tableName = cacheTableName;
+    await this.update({ count }, { objectId: cacheData[0].objectId }).catch(
+      (e) => {
+        if (e.code === 101) {
+          return;
+        }
+        throw e;
+      }
+    );
+    this.tableName = currentTableName;
+  }
+
   async count(where = {}, options = {}) {
     const instance = this.where(this.tableName, where);
     if (!options.group) {
@@ -137,7 +226,19 @@ module.exports = class extends Base {
       });
     }
 
-    // todo: query optimize
+    // get group count cache by group field where data
+    const cacheData = await this._getCmtGroupByMailUserIdCache(
+      options.group.join('_'),
+      where
+    );
+    const cacheDataMap = {};
+    for (let i = 0; i < cacheData.length; i++) {
+      const key = options.group
+        .map((item) => cacheData[i][item] || undefined)
+        .join('_');
+      cacheDataMap[key] = cacheData[i];
+    }
+
     const counts = [];
     const countsPromise = [];
     for (let i = 0; i < options.group.length; i++) {
@@ -148,10 +249,23 @@ module.exports = class extends Base {
 
       const groupFlatValue = {};
       options.group.slice(0, i).forEach((group) => {
-        groupFlatValue[group] = null;
+        groupFlatValue[group] = undefined;
       });
 
       for (let j = 0; j < where._complex[groupName][1].length; j++) {
+        const cacheKey = options.group
+          .map(
+            (item) =>
+              ({
+                ...groupFlatValue,
+                [groupName]: where._complex[groupName][1][j],
+              }[item] || undefined)
+          )
+          .join('_');
+        if (cacheDataMap[cacheKey]) {
+          continue;
+        }
+
         const groupWhere = {
           ...where,
           ...groupFlatValue,
@@ -172,8 +286,10 @@ module.exports = class extends Base {
       }
     }
 
-    await think.promiseAllQueue(countsPromise, 3);
-    return counts;
+    await think.promiseAllQueue(countsPromise, 1);
+    // cache data
+    await this._setCmtGroupByMailUserIdCache(options.group.join('_'), counts);
+    return [...cacheData, ...counts];
   }
 
   async add(
@@ -190,6 +306,7 @@ module.exports = class extends Base {
     instance.setACL(acl);
 
     const resp = await instance.save();
+    await this._updateCmtGroupByMailUserIdCache(data, 'add');
     return resp.toJSON();
   }
 
@@ -199,11 +316,16 @@ module.exports = class extends Base {
 
     return Promise.all(
       ret.map(async (item) => {
+        const _oldStatus = item.get('status');
         if (think.isFunction(data)) {
           item.set(data(item.toJSON()));
         } else {
           item.set(data);
         }
+        const _newStatus = item.get('status');
+        if (_newStatus && _oldStatus !== _newStatus) {
+          await this._updateCmtGroupByMailUserIdCache(data, 'update_status');
+        }
 
         const resp = await item.save();
         return resp.toJSON();
@@ -214,6 +336,7 @@ module.exports = class extends Base {
   async delete(where) {
     const instance = this.where(this.tableName, where);
     const data = await instance.find();
+    await this._updateCmtGroupByMailUserIdCache(data, 'delete');
 
     return AV.Object.destroyAll(data);
   }