re-organized backend

server/py/fastapi_server/api.py

@@ -0,0 +1,63 @@
+from fastapi import Request, Response, HTTPException
+from fastapi.responses import JSONResponse, PlainTextResponse
+
+import aiosqlite
+
+from user_mgmt import *
+import db_setup as db
+
+async def register_endpoint(req: Request):
+    body = await req.json()
+    try:
+        create_user(body["first_name"], body["last_name"], body["email"], body["password"])
+    except KeyError:
+        raise HTTPException(400, "Missing args")
+
+
+async def login_endpoint(req: Request):
+    """
+    The login endpoint
+
+    Takes email and password
+    Returns 401 or 400 in error.
+    Returns the user's full info if correct
+    """
+    
+    body = await req.json()
+    #print(body)
+    try:
+        user_data = await fetch_user_info(body["email"], "email")
+        if (user_data is None): 
+            print("User doesn't exist")
+            raise HTTPException(status_code=401, detail="User doesn't exist")
+        else:
+            pwd = user_data.get("password")
+            if (bcrypt.checkpw(body["password"].encode(), user_data.get("password").encode())):
+                return user_data
+            else:
+                print("Wrong password")
+                raise HTTPException(status_code=401, detail="Wrong password")
+    except KeyError:
+        raise HTTPException(status_code=400)
+    
+
+
+async def me_get_endpoint(req: Request):
+    """
+    Retrieves user profile data.
+
+    Trusts Nuxt's backend to allow it or not.
+    """
+    args = dict(req.query_params)
+    me = None
+    try:
+    # Prefer id over e-mail
+        if "id" in args.keys(): me = fetch_user_info(args["id"], "id")
+        elif "email" in args.keys(): me = fetch_user_info(args["email"], "email")
+        else: raise HTTPException(400, "Provide either email or user id")
+    #print(args)
+        if (me is None):
+            raise HTTPException(404, "User doesn't exist")
+        else: return me
+    except KeyError:
+        raise HTTPException(400, "Please provide email")