Merge branch 'develop' into math-block

1 files changed, 42 insertions, 0 deletions

diff --git a/src/server/api/common/getters.ts b/src/server/api/common/getters.ts
index 1fce58b20a..1cd054cab1 100644
--- a/src/server/api/common/getters.ts
+++ b/src/server/api/common/getters.ts
@@ -1,5 +1,6 @@
 import * as mongo from 'mongodb';
 import Note from "../../../models/note";
+import User, { isRemoteUser, isLocalUser } from "../../../models/user";
 
 /**
  * Get valied note for API processing
@@ -16,3 +17,44 @@ export async function getValiedNote(noteId: mongo.ObjectID) {
 
 	return note;
 }
+
+/**
+ * Get user for API processing
+ */
+export async function getUser(userId: mongo.ObjectID) {
+	const user = await User.findOne({
+		_id: userId
+	});
+
+	if (user == null) {
+		throw 'user not found';
+	}
+
+	return user;
+}
+
+/**
+ * Get remote user for API processing
+ */
+export async function getRemoteUser(userId: mongo.ObjectID) {
+	const user = await getUser(userId);
+
+	if (!isRemoteUser(user)) {
+		throw 'user is not a remote user';
+	}
+
+	return user;
+}
+
+/**
+ * Get local user for API processing
+ */
+export async function getLocalUser(userId: mongo.ObjectID) {
+	const user = await getUser(userId);
+
+	if (!isLocalUser(user)) {
+		throw 'user is not a local user';
+	}
+
+	return user;
+}