new hierarchy decorators

1 files changed, 44 insertions, 8 deletions

diff --git a/api/hierarchy.py b/api/hierarchy.py
index f75613c..20dcc45 100644
--- a/api/hierarchy.py
+++ b/api/hierarchy.py
@@ -1,29 +1,65 @@
 from flask import request
 from auth.login_token import token_login
+from user.info import valid_user_id
 from db import cursor
 
 ranks = ["none", "user", "moderator", "admin", "bot"]
 
 
+# This decorator doesn't check for hierarchy constraints, but does
+# make sure that token_id or explicit_id are valid user_id's
+def util_two_person(func):
+	def wrapper():
+		token_id = None
+		explicit_id = None
+
+		token = request.cookies.get("token") or ""
+		if token: token_id = token_login(token)
+
+		data = request.get_json()
+		if data: explicit_id = data.get("id")
+
+		if explicit_id and not valid_user_id(explicit_id): explicit_id = None
+
+		return func(token_id, explicit_id)
+
+	wrapper.__name__ = func.__name__
+	return wrapper
+
+
+# no authentication, just runs endpoint() if both token_id and
+# explicit_id are present from @util_two_person.
+def two_person(func):
+	@util_two_person
+	def wrapper(token_id, explicit_id):
+		if not token_id or \
+           not explicit_id:
+			return "", 400
+
+		return func(token_id, explicit_id)
+
+	wrapper.__name__ = func.__name__
+	return wrapper
+
+
 # @auth_required function decorator (use after @flask.Blueprint.route() decorator)
+# This decorator only runs endpoint() if token_id from
+# @util_two_person is not None and passes hierarchy constraints
 def auth_required(level):
 	def decorator(func):
-		def wrapper():
-			token = request.cookies.get("token") or ""
-			if not token: return "", 403
-
-			user_id = token_login(token)
-			if not user_id: return "", 403
+		@util_two_person
+		def wrapper(token_id, explicit_id):
+			if not token_id: return "", 400
 
 			user_rank_text = cursor.execute(
-				"select type from users where user_id = ?", [user_id]
+				"select type from users where user_id = ?", [token_id]
 			).fetchone()[0]
 
 			required_rank = ranks.index(level)
 			user_rank = ranks.index(user_rank_text)
 			if required_rank > user_rank: return "", 403
 
-			return func(user_id)
+			return func(token_id)
 
 		wrapper.__name__ = func.__name__
 		return wrapper