From b6eba4de033a7fcc003e57d67a46937b8396bd5f Mon Sep 17 00:00:00 2001
From: mei23 <m@m544.net>
Date: Mon, 21 Jan 2019 04:22:27 +0900
Subject: [PATCH] =?UTF-8?q?API=E3=81=AE=E5=AE=9F=E8=A3=85=E3=81=AF?=
 =?UTF-8?q?=E3=82=82=E3=81=A9=E3=81=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/server/api/endpoints/notes/polls/vote.ts | 84 +++++++++++++++++++-
 1 file changed, 82 insertions(+), 2 deletions(-)

diff --git a/src/server/api/endpoints/notes/polls/vote.ts b/src/server/api/endpoints/notes/polls/vote.ts
index 9cf65dcd7e..78dd1d1ff7 100644
--- a/src/server/api/endpoints/notes/polls/vote.ts
+++ b/src/server/api/endpoints/notes/polls/vote.ts
@@ -1,7 +1,12 @@
 import $ from 'cafy'; import ID, { transform } from '../../../../../misc/cafy-id';
+import Vote from '../../../../../models/poll-vote';
 import Note from '../../../../../models/note';
+import Watching from '../../../../../models/note-watching';
+import watch from '../../../../../services/note/watch';
+import { publishNoteStream } from '../../../../../stream';
+import notify from '../../../../../notify';
 import define from '../../../define';
-import vote from '../../../../../services/note/polls/vote';
+import createNote from '../../../../../services/note/create';
 
 export const meta = {
 	desc: {
@@ -43,5 +48,80 @@ export default define(meta, (ps, user) => new Promise(async (res, rej) => {
 		return rej('poll not found');
 	}
 
-	await vote(user, note, ps.choice).catch(e => rej(e));
+	if (!note.poll.choices.some(x => x.id == ps.choice)) return rej('invalid choice param');
+
+	// if already voted
+	const exist = await Vote.findOne({
+		noteId: note._id,
+		userId: user._id
+	});
+
+	if (exist !== null) {
+		return rej('already voted');
+	}
+
+	// Create vote
+	await Vote.insert({
+		createdAt: new Date(),
+		noteId: note._id,
+		userId: user._id,
+		choice: ps.choice
+	});
+
+	// Send response
+	res();
+
+	const inc: any = {};
+	inc[`poll.choices.${note.poll.choices.findIndex(c => c.id == ps.choice)}.votes`] = 1;
+
+	// Increment votes count
+	await Note.update({ _id: note._id }, {
+		$inc: inc
+	});
+
+	publishNoteStream(note._id, 'pollVoted', {
+		choice: ps.choice,
+		userId: user._id.toHexString()
+	});
+
+	// Notify
+	notify(note.userId, user._id, 'poll_vote', {
+		noteId: note._id,
+		choice: ps.choice
+	});
+
+	// Fetch watchers
+	Watching
+		.find({
+			noteId: note._id,
+			userId: { $ne: user._id },
+			// 削除されたドキュメントは除く
+			deletedAt: { $exists: false }
+		}, {
+			fields: {
+				userId: true
+			}
+		})
+		.then(watchers => {
+			for (const watcher of watchers) {
+				notify(watcher.userId, user._id, 'poll_vote', {
+					noteId: note._id,
+					choice: ps.choice
+				});
+			}
+		});
+
+	// この投稿をWatchする
+	if (user.settings.autoWatch !== false) {
+		watch(user._id, note);
+	}
+
+	// リモート投票の場合リプライ送信
+	if (note._user.host != null) {
+		createNote(user, {
+			createdAt: new Date(),
+			text: ps.choice.toString(),
+			reply: note,
+		});
+	}
 }));