import motor.motor_asyncio
from pymongo import MongoClient
from fastapi.responses import JSONResponse, HTMLResponse

# Mongodb setting
client = MongoClient("mongodb://localhost:27017/")
db = client.get_database("database")
user_collection = db.get_collection("users")


def login_user(username, password):
    query = {"username": {"$eq": username}}
    users = list(user_collection.find(query))
    if len(users) > 0:
        if password != str(users[0]['password']):
            response = "Incorrect password"
        else:
            response = "Login Successfully"
            return JSONResponse(content={"name": users[0]['username']}, status_code=200)
    else:
        response = "Incorrect username"
    return JSONResponse(status_code=404, content={"message": response})


def get_project_user(username):
    query = {"username": {"$eq": username}}
    users = list(user_collection.find(query))
    if len(users) > 0:
        return JSONResponse(content={"name": users[0]['projectname'], "path":users[0]["projectpath"]}, status_code=200)
    else:
        response = "Incorrect username"
    return JSONResponse(status_code=404, content={"message": response})