From 1f018d87f2660412b6cce7470388c58aa9772e12 Mon Sep 17 00:00:00 2001
From: syuilo <syuilotan@yahoo.co.jp>
Date: Mon, 7 May 2018 03:19:24 +0900
Subject: [PATCH] =?UTF-8?q?=E3=83=A1=E3=83=B3=E3=82=B7=E3=83=A7=E3=83=B3?=
 =?UTF-8?q?=E3=82=92=E5=87=A6=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/remote/resolve-user.ts  |  2 +-
 src/services/note/create.ts | 48 ++++++++++++++++++++++++++++++++-----
 2 files changed, 43 insertions(+), 7 deletions(-)

diff --git a/src/remote/resolve-user.ts b/src/remote/resolve-user.ts
index b6048842b..e59d19333 100644
--- a/src/remote/resolve-user.ts
+++ b/src/remote/resolve-user.ts
@@ -4,7 +4,7 @@ import webFinger from './webfinger';
 import config from '../config';
 import { createPerson } from './activitypub/models/person';
 
-export default async (username, _host, option) => {
+export default async (username, _host, option?) => {
 	const usernameLower = username.toLowerCase();
 	const hostAscii = toASCII(_host).toLowerCase();
 	const host = toUnicode(hostAscii);
diff --git a/src/services/note/create.ts b/src/services/note/create.ts
index c2c03516e..d146e5b7a 100644
--- a/src/services/note/create.ts
+++ b/src/services/note/create.ts
@@ -17,6 +17,7 @@ import event from '../../publishers/stream';
 import parse from '../../text/parse';
 import { IApp } from '../../models/app';
 import UserList from '../../models/user-list';
+import resolveUser from '../../remote/resolve-user';
 
 export default async (user: IUser, data: {
 	createdAt?: Date;
@@ -119,6 +120,13 @@ export default async (user: IUser, data: {
 	// Serialize
 	const noteObj = await pack(note);
 
+	const render = async () => {
+		const content = data.renote && data.text == null
+			? renderAnnounce(data.renote.uri ? data.renote.uri : await renderNote(data.renote))
+			: renderCreate(await renderNote(note));
+		return packAp(content);
+	};
+
 	// タイムラインへの投稿
 	if (note.channelId == null) {
 		if (!silent) {
@@ -190,12 +198,6 @@ export default async (user: IUser, data: {
 		}
 
 		//#region リプライとAnnounceのAP配送
-		const render = async () => {
-			const content = data.renote && data.text == null
-				? renderAnnounce(data.renote.uri ? data.renote.uri : await renderNote(data.renote))
-				: renderCreate(await renderNote(note));
-			return packAp(content);
-		};
 
 		// 投稿がリプライかつ投稿者がローカルユーザーかつリプライ先の投稿の投稿者がリモートユーザーなら配送
 		if (data.reply && isLocalUser(user) && isRemoteUser(data.reply._user)) {
@@ -243,6 +245,40 @@ export default async (user: IUser, data: {
 		);
 	}*/
 
+	//#region メンション
+	if (data.text) {
+		// TODO: Drop dupulicates
+		const mentions = tokens
+			.filter(t => t.type == 'mention');
+
+		mentions.forEach(async m => {
+			const u = await resolveUser(m.username, m.host);
+
+			if (isLocalUser(u)) {
+				// Fetch mentioned user
+				const mentionee = await User
+					.findOne({
+						usernameLower: m.username.toLowerCase()
+					}, { _id: true });
+
+				// When mentioned user not found
+				if (mentionee == null) return;
+
+				// 既に言及されたユーザーに対する返信や引用renoteの場合も無視
+				if (data.reply && data.reply.userId.equals(mentionee._id)) return;
+				if (data.renote && data.renote.userId.equals(mentionee._id)) return;
+
+				// Create notification
+				notify(mentionee._id, user._id, 'mention', {
+					noteId: note._id
+				});
+			} else {
+				deliver(user, await render(), u.inbox);
+			}
+		});
+	}
+	//#endregion
+
 	const mentions = [];
 
 	async function addMention(mentionee, reason) {