import chalicelib.utils.TimeUTC
from chalicelib.utils import email_helper, captcha, helper
import secrets
from chalicelib.utils import pg_client

from chalicelib.core import users


def step1(data):
    print("====================== reset password 1 ===============")
    print(data)
    if helper.allow_captcha() and not captcha.is_valid(data["g-recaptcha-response"]):
        print("error: Invalid captcha.")
        return {"errors": ["Invalid captcha."]}
    if "email" not in data:
        return {"errors": ["email not found in body"]}

    a_users = users.get_by_email_only(data["email"])
    if len(a_users) > 1:
        print(f"multiple users found for [{data['email']}] please contact our support")
        return {"errors": ["multiple users, please contact our support"]}
    elif len(a_users) == 1:
        a_users = a_users[0]
        invitation_link=users.generate_new_invitation(user_id=a_users["id"])
        email_helper.send_forgot_password(recipient=data["email"], invitation_link=invitation_link)
    else:
        print(f"invalid email address [{data['email']}]")
        return {"errors": ["invalid email address"]}
    return {"data": {"state": "success"}}


# def step2(data):
#     print("====================== change password 2 ===============")
#     user = users.get_by_email_reset(data["email"], data["code"])
#     if not user:
#         print("error: wrong email or reset code")
#         return {"errors": ["wrong email or reset code"]}
#     users.update(tenant_id=user["tenantId"], user_id=user["id"],
#                  changes={"token": None, "password": data["password"], "generatedPassword": False})
#     return {"data": {"state": "success"}}