SAE-A2-TruthInquiry/truthseeker/routes/routes_api.py
2022-12-01 10:54:30 +01:00

92 lines
2.6 KiB
Python

import flask
import jwt
import truthseeker
from truthseeker.logic import game_logic
from functools import wraps
routes_api = flask.Blueprint("api", __name__)
# Auth decorator
def jwt_required(f):
@wraps(f)
def decorator(*args, **kwargs):
jwt_str = flask.request.values.get("jwt")
if not jwt_str:
return {"status": "Error, JWT token missing"}
try:
claims = jwt.decode(jwt_str, truthseeker.app.config['SECRET_KEY'], algorithms=['HS256'])
except jwt.exceptions.InvalidTokenError as e:
print("Caught exception while decoding JWT token :", e)
return {"status": "Error, invalid JWT"}
return f(claims, *args, **kwargs)
return decorator
@routes_api.route("/createGame", methods=["GET", "POST"])
def create_game():
username = flask.request.values.get("username")
if username==None:
return {"status": "error, username not set"}
response = {}
response["status"] = "ok"
game = game_logic.create_game()
response["game_id"] = game.game_id
owner, owner_jwt = game.set_owner(username=username)
response["jwt"] = owner_jwt
return response
@routes_api.route("/joinGame", methods=["GET", "POST"])
def join_game():
game_id = flask.request.values.get("game_id")
username = flask.request.values.get("username")
if game_id==None or username==None:
return {"status": "error, username or game id not set"}
game = game_logic.get_game(game_id)
if game == None:
return {"status": "error, game does not exist"}
member, member_jwt = game.add_member(username)
response = {}
response["status"] = "ok"
response["jwt"] = member_jwt
return response
@routes_api.route("/getGameInfo", methods=["GET", "POST"])
def get_game_info(): # DEPRECATED, SHOULD BE REMOVED
response = {}
game_id = flask.request.values.get("game_id")
if game_id == None:
response["status"] = "No 'game_id' argument"
return response
game = game_logic.get_game_info(game_id)
if game == None:
response["status"] = "Game {} does not exist".format(game_id)
return response
else:
response["status"] = "ok"
response["game_id"] = game_id
response["token"] = game.start_token
return response
@routes_api.route("/startGame", methods=["GET", "POST"])
@jwt_required
def start_game(claims):
if not claims["owner"]:
return {"status": "Error, you are not the owner of this game"}
if game_logic.get_game(claims["game_id"]) == None:
return {"status": "Error, this game doesn't exist"}
return {"status": "ok"}