from scripts.database.mongo import mongo_database 
mongo_collection = mongo_database.get_collection("users")

def user_exists(user):
    try:
        res = mongo_collection.find_one({"name": user}, {"password": 1})
        if res is None:
            return {"success": True, "user": None}
        else:
            return {"success": True, "user": user, "password": res["password"]}
    except:
        return {"success": False, "error": "Request error, try again"}
    
def add_user(user, password):
    try:
        res = mongo_collection.insert_one({"name": user, "password": password})
        return {"success": True, "user": user}
    except:
        print("error")

def delete_single_user(user):
    try:
        mongo_collection.delete_one({"name": user})
        return {"success": True, "user": user}
    except:
        return {"success": False, "user": user}

def update_user(user, user_prefs):
    print(user, user_prefs, "in user handler")
    try:
        res = mongo_collection.update_one({"name": user}, {"$set": user_prefs})
        return {"success": True, "message": "user preferences successfully updated"}
    except Exception as e:
        return {"success": False, "message": str(e)}