feat: add message api (#126)

Author: sinchang

app/controller/api/message.js

@@ -0,0 +1,82 @@
+'use strict';
+
+const Controller = require('egg').Controller;
+
+class MessageController extends Controller {
+  async count() {
+    const { ctx } = this;
+    const userId = ctx.request.user._id;
+    const count = await ctx.service.message.getMessagesCount(userId);
+    ctx.body = { count };
+  }
+
+  async markAll() {
+    const { ctx } = this;
+    const userId = ctx.request.user._id;
+    const messages = await ctx.service.message.getUnreadMessagesByUserId(userId);
+    const result = await ctx.service.message.updateMessagesToRead(userId, messages);
+    ctx.body = {
+      success: true,
+      marked_msgs: result ? messages.map(message => { return { id: message._id }; }) : [],
+    };
+  }
+
+  async markOne() {
+    const { ctx } = this;
+    const messageId = ctx.params.msg_id;
+    await ctx.service.message.updateOneMessageToRead(messageId);
+
+    ctx.body = {
+      success: true,
+      marked_msg_id: messageId,
+    };
+  }
+
+  async list() {
+    const { ctx } = this;
+    const userId = ctx.request.user._id;
+    const msgService = ctx.service.message;
+    const mdrender = ctx.request.query.mdrender !== 'false';
+    const messages = await Promise.all([
+      msgService.getReadMessagesByUserId(userId),
+      msgService.getUnreadMessagesByUserId(userId),
+    ]);
+
+    let hasReadMessages = await Promise.all(messages[0].map(async message => await msgService.getMessageRelations(message)));
+    let hasUnReadMessages = await Promise.all(messages[1].map(async message => await msgService.getMessageRelations(message)));
+
+    const formatMessage = message => {
+      return {
+        id: message._id,
+        type: message.type,
+        has_read: message.has_read,
+        create_at: message.create_at,
+        author: {
+          loginname: message.author.loginname,
+          avatar_url: message.author.avatar_url,
+        },
+        topic: {
+          id: message.topic.topic._id,
+          title: message.topic.topic.title,
+          last_reply_at: message.topic.topic.last_reply_at,
+        },
+        reply: message.reply ? {
+          content: mdrender ? ctx.helper.markdown(message.reply.content) : message.reply.content,
+        } : {},
+      };
+    };
+
+    hasReadMessages = hasReadMessages.map(message => formatMessage(message));
+    hasUnReadMessages = hasUnReadMessages.map(message => formatMessage(message));
+
+    ctx.body = {
+      success: true,
+      data: {
+        has_read_messages: hasReadMessages,
+        hasnot_read_messages: hasUnReadMessages,
+      },
+    };
+  }
+}
+
+module.exports = MessageController;

app/router/api.js

@@ -6,13 +6,19 @@
 module.exports = app => {
   const { router, controller, middleware } = app;
 
-  const { user } = controller.api;
+  const { user, message } = controller.api;
 
   const tokenRequired = middleware.tokenRequired();
-  // const adminRequired = middleware.adminRequired();
   // const createTopicLimit = middleware.createTopicLimit(config.topic);
   // const createUserLimit = middleware.createUserLimit(config.create_user_per_ip);
 
+  // 用户
   router.get('/api/v1/user/:loginname', user.show);
   router.post('/api/v1/accesstoken', tokenRequired, user.verify);
+
+  // 消息通知
+  router.get('/api/v1/message/count', tokenRequired, message.count);
+  router.get('/api/v1/messages', tokenRequired, message.list);
+  router.post('/api/v1/message/mark_all', tokenRequired, message.markAll);
+  router.post('/api/v1/message/mark_one/:msg_id', tokenRequired, message.markOne);
 };

test/app/controller/api/message.test.js

@@ -0,0 +1,57 @@
+'use strict';
+
+const { app, assert } = require('egg-mock/bootstrap');
+
+describe('test/app/controller/api/message.test.js', () => {
+  let user1,
+    user2,
+    token,
+    messageId,
+    topicId;
+
+  before(async function() {
+    const ctx = app.mockContext();
+    const loginname1 = `loginname1_${Date.now()}`;
+    const loginname2 = `loginname2_${Date.now()}`;
+    const email1 = `${loginname1}@test.com`;
+    const email2 = `${loginname2}@test.com`;
+    user1 = await ctx.service.user.newAndSave('name', loginname1, 'pass', email1, 'avatar_url', 'active');
+    user2 = await ctx.service.user.newAndSave('name', loginname2, 'pass', email2, 'avatar_url', 'active');
+    token = user2.accessToken;
+    assert(user1.loginname === loginname1);
+    assert(user2.loginname === loginname2);
+
+    const title = 'first post';
+    const content = 'hello world';
+    const tab = 'share';
+    const topic = await ctx.service.topic.newAndSave(title, content, tab, user1._id);
+    topicId = topic._id;
+    assert(topic.title === title);
+    assert(topic.content === content);
+    assert(topic.tab === tab);
+    assert.equal(topic.author_id, user1._id);
+
+    const message = await ctx.service.message.sendAtMessage(user2._id, user1._id, topicId);
+    messageId = message._id;
+  });
+
+  it('get /message/count should ok', async () => {
+    const result = await app.httpRequest().get(`/api/v1/message/count?accesstoken=${token}`);
+    assert(result.body.count >= 1);
+  });
+
+  it('get /messages should ok', async () => {
+    const result = await app.httpRequest().get(`/api/v1/messages?accesstoken=${token}`);
+    assert(result.body.data.hasnot_read_messages.length >= 1);
+  });
+
+  it('post /message/mark_all should ok', async () => {
+    const result = await app.httpRequest().post(`/api/v1/message/mark_all?accesstoken=${token}`);
+    assert(result.body.marked_msgs.length >= 1);
+  });
+
+  it('post message/mark_one/:msg_id should ok', async () => {
+    const result = await app.httpRequest().post(`/api/v1/message/mark_one/${messageId}?accesstoken=${token}`);
+    assert.equal(result.body.marked_msg_id, messageId);
+  });
+});