convert proxy to service code

Author: JacksonTian

.eslintrc

@@ -1,3 +1,9 @@
 {
-  "extends": "eslint-config-egg"
+  "extends": "eslint-config-egg",
+  "parserOptions": {
+    "ecmaVersion": 8,
+    "sourceType": "script",
+    "ecmaFeatures": {
+    }
+  }
 }

app/common/tools.js

@@ -0,0 +1,29 @@
+'use strict';
+
+const bcrypt = require('bcryptjs');
+const moment = require('moment');
+
+moment.locale('zh-cn'); // 使用中文
+
+// 格式化时间
+exports.formatDate = function(date, friendly) {
+  date = moment(date);
+
+  if (friendly) {
+    return date.fromNow();
+  }
+
+  return date.format('YYYY-MM-DD HH:mm');
+};
+
+exports.validateId = function(str) {
+  return (/^[a-zA-Z0-9\-_]+$/i).test(str);
+};
+
+exports.bhash = function(str, callback) {
+  bcrypt.hash(str, 10, callback);
+};
+
+exports.bcompare = function(str, hash, callback) {
+  bcrypt.compare(str, hash, callback);
+};

app/extend/application.js

@@ -3,7 +3,6 @@
 const mongoose = require('mongoose');
 
 const MONGODB = Symbol('Application#mongodb');
-const REDIS = Symbol('Application#redis');
 
 // 扩展一些框架便利的方法
 module.exports = {

app/models/base_model.js

@@ -0,0 +1,17 @@
+'use strict';
+
+/**
+ * 给所有的 Model 扩展功能
+ * http://mongoosejs.com/docs/plugins.html
+ */
+const tools = require('../common/tools');
+
+module.exports = function(schema) {
+  schema.methods.create_at_ago = function() {
+    return tools.formatDate(this.create_at, true);
+  };
+
+  schema.methods.update_at_ago = function() {
+    return tools.formatDate(this.update_at, true);
+  };
+};

app/service/at.js

@@ -0,0 +1,77 @@
+'use strict';
+
+module.exports = app => {
+
+  class AtService extends app.Service {
+    /**
+     * 从文本中提取出@username 标记的用户名数组
+     * @param {String} text 文本内容
+     * @return {Array} 用户名数组
+     */
+    fetchUsers(text) {
+      if (!text) {
+        return [];
+      }
+
+      const ignoreRegexs = [
+        /```.+?```/g, // 去除单行的 ```
+        /^```[\s\S]+?^```/gm, // ``` 里面的是 pre 标签内容
+        /`[\s\S]+?`/g, // 同一行中，`some code` 中内容也不该被解析
+        /^ {4}.*/gm, // 4个空格也是 pre 标签，在这里 . 不会匹配换行
+        /\b\S*?@[^\s]*?\..+?\b/g, // somebody@gmail.com 会被去除
+        /\[@.+?\]\(\/.+?\)/g, // 已经被 link 的 username
+      ];
+
+      ignoreRegexs.forEach(ignore_regex => {
+        text = text.replace(ignore_regex, '');
+      });
+
+      const results = text.match(/@[a-z0-9\-_]+\b/igm);
+      const names = [];
+      if (results) {
+        for (let i = 0, l = results.length; i < l; i++) {
+          let s = results[i];
+          // remove leading char @
+          s = s.slice(1);
+          names.push(s);
+        }
+      }
+      return [ ...new Set(names) ];
+    }
+
+    /*
+     * 根据文本内容中读取用户，并发送消息给提到的用户
+     * @param {String} text 文本内容
+     * @param {String} topicId 主题ID
+     * @param {String} authorId 作者ID
+     * @param {String} reply_id 回复ID
+     */
+    async sendMessageToMentionUsers(text, topicId, authorId, reply_id = null) {
+      let users = await this.service.user.getUsersByNames(this.fetchUsers(text));
+
+      users = users.filter(user => {
+        return !user._id.equals(authorId);
+      });
+
+      return Promise.all(users.map(user => {
+        return this.service.message.sendAtMessage(user._id, authorId, topicId, reply_id);
+      }));
+    }
+
+    /**
+     * 根据文本内容，替换为数据库中的数据
+     * @param {String} text 文本内容
+     * @return {String} 替换后的文本内容
+     */
+    linkUsers(text) {
+      const users = AtService.fetchUsers(text);
+      for (let i = 0; i < users.length; i++) {
+        const name = users[i];
+        text = text.replace(new RegExp('@' + name + '\\b(?!\\])', 'g'), '[@' + name + '](/user/' + name + ')');
+      }
+      return text;
+    }
+  }
+
+  return AtService;
+};

app/service/message.js

@@ -0,0 +1,131 @@
+'use strict';
+
+const mongoose = require('mongoose');
+
+const BaseModel = require('./base_model');
+const Schema = mongoose.Schema;
+const ObjectId = Schema.ObjectId;
+
+module.exports = app => {
+
+  /*
+   * type:
+   * reply: xx 回复了你的话题
+   * reply2: xx 在话题中回复了你
+   * follow: xx 关注了你
+   * at: xx ＠了你
+   */
+  const MessageSchema = new Schema({
+    type: { type: String },
+    master_id: { type: ObjectId },
+    author_id: { type: ObjectId },
+    topic_id: { type: ObjectId },
+    reply_id: { type: ObjectId },
+    has_read: { type: Boolean, default: false },
+    create_at: { type: Date, default: Date.now },
+  });
+  MessageSchema.plugin(BaseModel);
+  MessageSchema.index({ master_id: 1, has_read: -1, create_at: -1 });
+  const Message = mongoose.model('Message', MessageSchema);
+
+  return class extends app.Service {
+
+    /*
+     * 根据用户ID，获取未读消息的数量
+     * Callback:
+     * @param {String} id 用户ID
+     * @return {Promise[messagesCount]} 承载消息列表的 Promise 对象
+     */
+    getMessagesCount(id) {
+      return Message.count({ master_id: id, has_read: false }).exec();
+    }
+
+    async getMessageRelations(message) {
+      if (message.type === 'reply' || message.type === 'reply2' || message.type === 'at') {
+        const [ author, topic, reply ] = await Promise.all([
+          this.services.user.getUserById(message.author_id),
+          this.services.topic.getTopicById(message.topic_id),
+          this.services.reply.getReplyById(message.reply_id),
+        ]);
+
+        message.author = author;
+        message.topic = topic;
+        message.reply = reply;
+
+        if (!author || !topic) {
+          message.is_invalid = true;
+        }
+
+        return message;
+      }
+
+      return { is_invalid: true };
+    }
+
+    /*
+     * 根据消息Id获取消息
+     * @param {String} id 消息ID
+     * @return {Promise[message]} 承载消息的 Promise 对象
+     */
+    async getMessageById(id) {
+      const message = await Message.findOne({ _id: id }).exec();
+      return this.getMessageRelations(message);
+    }
+
+    /*
+     * 根据用户ID，获取已读消息列表
+     * @param {String} userId 用户ID
+     * @return {Promise[messages]} 承载消息列表的 Promise 对象
+     */
+    getReadMessagesByUserId(userId) {
+      const query = { master_id: userId, has_read: true };
+      return Message.find(query, null,
+        { sort: '-create_at', limit: 20 }).exec();
+    }
+
+    /*
+     * 根据用户ID，获取未读消息列表
+     * @param {String} userId 用户ID
+     * @return {Promise[messages]} 承载消息列表的 Promise 对象
+     */
+    getUnreadMessagesByUserId(userId) {
+      const query = { master_id: userId, has_read: false };
+      return Message.find(query, null,
+        { sort: '-create_at' }).exec();
+    }
+
+    /*
+     * 将消息设置成已读
+     * @return {Promise[messages]} 承载消息列表的 Promise 对象
+     */
+    async updateMessagesToRead(userId, messages) {
+      if (messages.length === 0) {
+        return;
+      }
+
+      const ids = messages.map(function(m) {
+        return m.id;
+      });
+
+      const query = { master_id: userId, _id: { $in: ids } };
+      const update = { $set: { has_read: true } };
+      const opts = { multi: true };
+      return Message.update(query, update, opts).exec();
+    }
+
+    /**
+     * 将单个消息设置成已读
+     * @param {String} msgId 消息 ID
+     * @return {Promise} 更新消息返回的 Promise 对象
+     */
+    async updateOneMessageToRead(msgId) {
+      if (!msgId) {
+        return;
+      }
+      const query = { _id: msgId };
+      const update = { $set: { has_read: true } };
+      return Message.update(query, update, { multi: true }).exec();
+    }
+  };
+
+};