add passcode

games/game_angrybird.py

@@ -1,4 +1,4 @@
 import game_util
 
-def process(command: str):
-    return game_util.util_process("angrybird", command)
+def process(passcode: str, command: str):
+    return game_util.util_process("angrybird", passcode, command)

games/game_snake.py

@@ -1,4 +1,4 @@
 import game_util
 
-def process(command: str):
-    return game_util.util_process("snake", command)
+def process(passcode: str, command: str):
+    return game_util.util_process("snake", passcode, command)

games/game_util.py

@@ -2,8 +2,16 @@ import os
 import json
 import requests
 
+from fastapi import HTTPException, status
 
-def util_process(game: str, command: str):
+
+def util_process(game: str, passcode: str, command: str):
+    code = os.getenv("hfs_code")
+    if code is None:
+        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR)
+    if passcode != code:
+        raise HTTPException(status.HTTP_511_NETWORK_AUTHENTICATION_REQUIRED)        
+    
     url = f'https://{os.getenv("hfs_space")}.hf.space/games/{game}'
     # url = 'http://127.0.0.1:8080/nlp'
     myobj = {

main.py

@@ -12,10 +12,10 @@ def root():
 
 
 @app.post("/games/snake")
-def main_snake(command: str=Body(embed=True)):
-    return game_snake.process(command)
+def main_snake(passcode: str, command: str=Body(embed=True)):
+    return game_snake.process(passcode, command)
 
 
 @app.post("/games/angrybird")
-def main_angrybird(command: str=Body(embed=True)):
-    return game_angrybird.process(command)
+def main_angrybird(passcode: str, command: str=Body(embed=True)):
+    return game_angrybird.process(passcode, command)