From 583b64331b4d3d36f642801c847109b8634df1d9 Mon Sep 17 00:00:00 2001
From: syuilo <syuilotan@yahoo.co.jp>
Date: Fri, 3 Mar 2017 08:00:10 +0900
Subject: wip

---
 src/api/endpoints/posts/polls/vote.js | 100 ----------------------------------
 src/api/endpoints/posts/polls/vote.ts |  88 ++++++++++++++++++++++++++++++
 2 files changed, 88 insertions(+), 100 deletions(-)
 delete mode 100644 src/api/endpoints/posts/polls/vote.js
 create mode 100644 src/api/endpoints/posts/polls/vote.ts

(limited to 'src/api/endpoints/posts/polls')

diff --git a/src/api/endpoints/posts/polls/vote.js b/src/api/endpoints/posts/polls/vote.js
deleted file mode 100644
index 9f9a5171a0..0000000000
--- a/src/api/endpoints/posts/polls/vote.js
+++ /dev/null
@@ -1,100 +0,0 @@
-'use strict';
-
-/**
- * Module dependencies
- */
-import * as mongo from 'mongodb';
-import Vote from '../../../models/poll-vote';
-import Post from '../../../models/post';
-import notify from '../../../common/notify';
-
-/**
- * Vote poll of a post
- *
- * @param {any} params
- * @param {any} user
- * @return {Promise<any>}
- */
-module.exports = (params, user) =>
-	new Promise(async (res, rej) => {
-		// Get 'post_id' parameter
-		const postId = params.post_id;
-		if (postId === undefined || postId === null) {
-			return rej('post_id is required');
-		}
-
-		// Validate id
-		if (!mongo.ObjectID.isValid(postId)) {
-			return rej('incorrect post_id');
-		}
-
-		// Get votee
-		const post = await Post.findOne({
-			_id: new mongo.ObjectID(postId)
-		});
-
-		if (post === null) {
-			return rej('post not found');
-		}
-
-		if (post.poll == null) {
-			return rej('poll not found');
-		}
-
-		// Get 'choice' parameter
-		const choice = params.choice;
-		if (choice == null) {
-			return rej('choice is required');
-		}
-
-		// Validate choice
-		if (!post.poll.choices.some(x => x.id == choice)) {
-			return rej('invalid choice');
-		}
-
-		// if already voted
-		const exist = await Vote.findOne({
-			post_id: post._id,
-			user_id: user._id
-		});
-
-		if (exist !== null) {
-			return rej('already voted');
-		}
-
-		// Create vote
-		await Vote.insert({
-			created_at: new Date(),
-			post_id: post._id,
-			user_id: user._id,
-			choice: choice
-		});
-
-		// Send response
-		res();
-
-		const inc = {};
-		inc[`poll.choices.${findWithAttr(post.poll.choices, 'id', choice)}.votes`] = 1;
-
-		console.log(inc);
-
-		// Increment likes count
-		Post.update({ _id: post._id }, {
-			$inc: inc
-		});
-
-		// Notify
-		notify(post.user_id, user._id, 'poll_vote', {
-			post_id: post._id,
-			choice: choice
-		});
-	});
-
-function findWithAttr(array, attr, value) {
-	for (let i = 0; i < array.length; i += 1) {
-		if (array[i][attr] === value) {
-			return i;
-		}
-	}
-	return -1;
-}
diff --git a/src/api/endpoints/posts/polls/vote.ts b/src/api/endpoints/posts/polls/vote.ts
new file mode 100644
index 0000000000..d0caf7da95
--- /dev/null
+++ b/src/api/endpoints/posts/polls/vote.ts
@@ -0,0 +1,88 @@
+'use strict';
+
+/**
+ * Module dependencies
+ */
+import it from '../../../it';
+import Vote from '../../../models/poll-vote';
+import Post from '../../../models/post';
+import notify from '../../../common/notify';
+
+/**
+ * Vote poll of a post
+ *
+ * @param {any} params
+ * @param {any} user
+ * @return {Promise<any>}
+ */
+module.exports = (params, user) =>
+	new Promise(async (res, rej) => {
+		// Get 'post_id' parameter
+		const [postId, postIdErr] = it(params.post_id, 'id', true);
+		if (postIdErr) return rej('invalid post_id param');
+
+		// Get votee
+		const post = await Post.findOne({
+			_id: postId
+		});
+
+		if (post === null) {
+			return rej('post not found');
+		}
+
+		if (post.poll == null) {
+			return rej('poll not found');
+		}
+
+		// Get 'choice' parameter
+		const [choice, choiceError] =
+			it(params.choice).expect.string()
+				.required()
+				.validate(c => post.poll.choices.some(x => x.id == c))
+				.qed();
+		if (choiceError) return rej('invalid choice param');
+
+		// if already voted
+		const exist = await Vote.findOne({
+			post_id: post._id,
+			user_id: user._id
+		});
+
+		if (exist !== null) {
+			return rej('already voted');
+		}
+
+		// Create vote
+		await Vote.insert({
+			created_at: new Date(),
+			post_id: post._id,
+			user_id: user._id,
+			choice: choice
+		});
+
+		// Send response
+		res();
+
+		const inc = {};
+		inc[`poll.choices.${findWithAttr(post.poll.choices, 'id', choice)}.votes`] = 1;
+
+		// Increment likes count
+		Post.update({ _id: post._id }, {
+			$inc: inc
+		});
+
+		// Notify
+		notify(post.user_id, user._id, 'poll_vote', {
+			post_id: post._id,
+			choice: choice
+		});
+	});
+
+function findWithAttr(array, attr, value) {
+	for (let i = 0; i < array.length; i += 1) {
+		if (array[i][attr] === value) {
+			return i;
+		}
+	}
+	return -1;
+}
-- 
cgit v1.2.3-freya