Wip

api/__init__.py

@@ -14,7 +14,7 @@
 from api.helpers.gdrive.gdrive import GoogleDriveTools
 
 gdrive = GoogleDriveTools()
-app = FastAPI()
+app = FastAPI(redoc_url=None, docs_url=None, openapi_url=None)
 config: dict = None
 mime = magic.Magic(mime=True)
 

api/models/auth.py

@@ -4,7 +4,9 @@
 
 class AddUserRequest(BaseModel):
     name: str
+    admin: str
     devices: List[str]
+    token: str
 
 
 class UserAddResponse(BaseModel):

api/routers/auth.py

@@ -11,15 +11,15 @@
 
 @router.post("/add_user")
 async def add_user(request: AddUserRequest) -> APIResponse:
-    # user = firebase.get_user(
-    #     username=request.admin, collection=firebase_collection_admin
-    # )
-
-    # if isinstance(user.to_dict(), dict):
-    #     if request.token not in user.to_dict()["token"]:
-    #         return APIResponse(status=404, message="USER_NOT_FOUND")
-    # else:
-    #     return APIResponse(status=404, message="USER_NOT_FOUND")
+    user = firebase.get_user(
+        username=request.admin, collection=firebase_collection_admin
+    )
+
+    if isinstance(user.to_dict(), dict):
+        if request.token not in user.to_dict()["token"]:
+            return APIResponse(status=404, message="USER_NOT_FOUND")
+    else:
+        return APIResponse(status=404, message="USER_NOT_FOUND")
 
     data = {"devices": request.devices, "token": uuid.uuid4().hex}
     firebase.create_user(collection="Users", username=request.name, data=data)